allow packages to be loaded from previous sim

nuclearkatie · May 14, 2024 · bbc3c0b · bbc3c0b
1 parent 2559d4d
commit bbc3c0b
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/sim_init.cc b/src/sim_init.cc
@@ -190,6 +190,23 @@ void SimInit::LoadRecipes() {
   }
 }
 
+void SimInit::LoadPackages() {
+  QueryResult qr;
+  try {
+    qr = b_->Query("Packages", NULL);
+  } catch (std::exception err) {
+    return;
+  }  // table doesn't exist (okay)
+
+  for (int i = 0; i < qr.rows.size(); ++i) {
+    std::string package = qr.GetVal<std::string>("Package", i);
+    double fill_min = qr.GetVal<int>("FillMin", i);
+    double fill_max = qr.GetVal<int>("FillMax", i);
+    std::string strategy = qr.GetVal<std::string>("Strategy", i);
+    ctx_->AddPackage(package, fill_min, fill_max, strategy);
+  }
+}
+
 void* SimInit::LoadPreconditioner(std::string name) {
   using std::map;
   using std::string;

diff --git a/src/sim_init.h b/src/sim_init.h
@@ -80,6 +80,7 @@ class SimInit {
 
   void LoadInfo();
   void LoadRecipes();
+  void LoadPackages();
   void LoadSolverInfo();
   void LoadPrototypes();
   void LoadInitialAgents();