Add set_rng to Simulation

Add dyn_rng method for SimulationRng wrapping

Signed-off-by: Tin Svagelj <tin.svagelj@live.com>

sim/src/input_modeling/mod.rs

@@ -13,4 +13,4 @@ pub use random_variable::Continuous as ContinuousRandomVariable;
 pub use random_variable::Discrete as DiscreteRandomVariable;
 pub use random_variable::Index as IndexRandomVariable;
 pub use thinning::Thinning;
-pub use uniform_rng::some_dyn_rng;
+pub use uniform_rng::{dyn_rng, some_dyn_rng};

sim/src/input_modeling/uniform_rng.rs

@@ -8,6 +8,10 @@ pub(crate) fn default_rng() -> DynRng {
     Rc::new(RefCell::new(rand_pcg::Pcg64Mcg::new(42)))
 }
 
+pub fn dyn_rng<Rng: SimulationRng + 'static>(rng: Rng) -> DynRng {
+    Rc::new(RefCell::new(rng))
+}
+
 pub fn some_dyn_rng<Rng: SimulationRng + 'static>(rng: Rng) -> Option<DynRng> {
-    Some(Rc::new(RefCell::new(rng)))
+    Some(dyn_rng(rng))
 }

sim/src/simulator/mod.rs

@@ -21,6 +21,7 @@ use std::rc::Rc;
 use serde::{Deserialize, Serialize};
 
 use crate::input_modeling::uniform_rng::SimulationRng;
+use crate::input_modeling::{dyn_rng, some_dyn_rng};
 use crate::models::{DevsModel, Model, ModelMessage, ModelRecord, Reportable};
 use crate::utils::errors::SimulationError;
 use crate::utils::set_panic_hook;
@@ -70,13 +71,17 @@ impl Simulation {
             models,
             connectors,
             services: Services {
-                global_rng: Rc::new(RefCell::new(global_rng)),
+                global_rng: dyn_rng(global_rng),
                 global_time: 0.0,
             },
             ..Self::default()
         }
     }
 
+    pub fn set_rng(&mut self, rng: impl SimulationRng + 'static) {
+        self.services.global_rng = dyn_rng(rng)
+    }
+
     /// This method sets the models and connectors of an existing simulation.
     pub fn put(&mut self, models: Vec<Model>, connectors: Vec<Connector>) {
         self.models = models;