Merge pull request #3 from Geode-solutions/feat/rand_engine

feat(RandomEngine): implement random engine.

Author: MelchiorSchuh

.gitignore

@@ -1,2 +1,3 @@
 build
 .settings
+bindings/python/requirements.txt

bindings/python/CMakeLists.txt

@@ -29,7 +29,7 @@ add_geode_python_wheel(
     DESCRIPTION 
         "A module of the OpenGeode framework for stochastic modeling, simulation and bayesian inference"
     MODULES
-        "geometry.py"
+        "stochastic.py"
     LICENSE "MIT"
 )
 

bindings/python/geometry.py

@@ -18,4 +18,4 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-add_subdirectory(geometry)
+add_subdirectory(stochastic)

bindings/python/requirements.txt

@@ -19,9 +19,10 @@
 # SOFTWARE.
 
 add_geode_python_binding(
-    NAME "py_geometry"
+    NAME "py_stochastic"
     SOURCES
-        "geometry.cpp"
+#        "random_engine.cpp"
+        "stochastic.cpp"
     DEPENDENCIES
-        ${PROJECT_NAME}::geometry
+        ${PROJECT_NAME}::stochastic
 )

bindings/python/src/CMakeLists.txt

@@ -0,0 +1,52 @@
+/*
+ * Copyright (c) 2019 - 2025 Geode-solutions
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ *
+ */
+
+#include <geode/stochastic/random_engine.hpp>
+
+namespace geode
+{
+    template < typename type >
+    type python_sample_uniform( pybind11::class_< RandomEngine >& rand_engine,
+        const std::string& suffix )
+    {
+        const auto read_suffix = absl::StrCat( "sample_uniform_", suffix );
+        manager.def(
+            read_suffix.c_str(), &RandomEngine::sample_uniform< type > );
+    }
+
+    void define_random_engine( pybind11::module& module )
+    {
+        pybind11::class_< RandomEngine > rand_engine( module, "RandomEngine" );
+        rand_engine.def( pybind11::init<>() )
+            .def( "set_seed", &RandomEngine::set_seed )
+            .def( "sample_gaussian", &RandomEngine::sample_gaussian )
+            .def( "sample_bernoulli", &AttributeManager::sample_bernoulli )
+            .def( "attribute_exists", &AttributeManager::attribute_exists );
+        python_sample_uniform< index_t >( rand_engine, "index_t" );
+        python_sample_uniform< local_index_t >( rand_engine, "local_index_t" );
+        python_sample_uniform< signed_index_t >(
+            rand_engine, "signed_index_t" );
+        python_sample_uniform< float >( rand_engine, "float" );
+        python_sample_uniform< double >( rand_engine, "double" );
+    }
+} // namespace geode

…dings/python/src/geometry/CMakeLists.txt …ngs/python/src/stochastic/CMakeLists.txtbindings/python/src/geometry/CMakeLists.txt renamed to bindings/python/src/stochastic/CMakeLists.txt bindings/python/src/geometry/CMakeLists.txt renamed to bindings/python/src/stochastic/CMakeLists.txt

@@ -23,13 +23,14 @@
 
 #include <pybind11/pybind11.h>
 
-#include <geode/stochastic/geometry/hello_world.hpp>
+#include <geode/stochastic/hello_world.hpp>
+#include <geode/stochastic/sampling/random_engine.hpp>
 
-PYBIND11_MODULE( opengeode_stochastic_py_geometry, module )
+PYBIND11_MODULE( opengeode_stochastic_py_stochastic, module )
 {
-    module.doc() = "OpenGeode-Stochastic Python binding for geometry";
-    pybind11::class_< geode::StochasticGeometryLibrary >(
-        module, "StochasticGeometryLibrary" )
-        .def( "initialize", &geode::StochasticGeometryLibrary::initialize );
+    module.doc() = "OpenGeode-Stochastic Python binding";
+    pybind11::class_< geode::StochasticLibrary >( module, "StochasticLibrary" )
+        .def( "initialize", &geode::StochasticLibrary::initialize );
     module.def( "hello_world", &geode::hello_world );
+    //    geode::define_random_engine( module );
 }

bindings/python/src/stochastic/random_engine.cpp

@@ -18,9 +18,8 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-add_geode_test(
-    SOURCE "test-hello-world.cpp"
-    DEPENDENCIES
-        OpenGeode::basic
-        ${PROJECT_NAME}::geometry
-)
+import opengeode
+
+from opengeode_stochastic_py_stochastic import *
+
+StochasticLibrary.initialize()

bindings/python/src/geometry/geometry.cpp …ngs/python/src/stochastic/stochastic.cppbindings/python/src/geometry/geometry.cpp renamed to bindings/python/src/stochastic/stochastic.cpp bindings/python/src/geometry/geometry.cpp renamed to bindings/python/src/stochastic/stochastic.cpp

@@ -18,4 +18,4 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-add_subdirectory(geometry)
+add_subdirectory(stochastic)