Final touches

UofUEpiBio · Oct 29, 2024 · 4ceca3e · 4ceca3e
1 parent e7449ca
commit 4ceca3e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/epiworld.hpp b/epiworld.hpp
@@ -6485,7 +6485,7 @@ class Model {
      * @param s Seed
      */
     ///@{
-    void set_rand_engine(std::mt19937 & eng);
+    void set_rand_engine(std::shared_ptr< std::mt19937 > & eng);
     std::shared_ptr< std::mt19937 > & get_rand_endgine();
     void seed(size_t s);
     void set_rand_norm(epiworld_double mean, epiworld_double sd);
@@ -17031,7 +17031,7 @@ inline ModelSURV<TSeq>::ModelSURV(
 
         // How many will we find
         std::binomial_distribution<> bdist(m->size(), m->par("Surveilance prob."));
-        int nsampled = bdist(m->get_rand_endgine());
+        int nsampled = bdist(*m->get_rand_endgine());
 
         int to_go = nsampled + 1;
 

diff --git a/include/epiworld/model-bones.hpp b/include/epiworld/model-bones.hpp
@@ -286,7 +286,7 @@ class Model {
      * @param s Seed
      */
     ///@{
-    void set_rand_engine(std::mt19937 & eng);
+    void set_rand_engine(std::shared_ptr< std::mt19937 > & eng);
     std::shared_ptr< std::mt19937 > & get_rand_endgine();
     void seed(size_t s);
     void set_rand_norm(epiworld_double mean, epiworld_double sd);

diff --git a/include/epiworld/models/surveillance.hpp b/include/epiworld/models/surveillance.hpp
@@ -227,7 +227,7 @@ inline ModelSURV<TSeq>::ModelSURV(
 
         // How many will we find
         std::binomial_distribution<> bdist(m->size(), m->par("Surveilance prob."));
-        int nsampled = bdist(m->get_rand_endgine());
+        int nsampled = bdist(*m->get_rand_endgine());
 
         int to_go = nsampled + 1;