Store step info in local db

NVE · Oct 3, 2024 · 6ee1223 · 6ee1223
1 parent ee31416
commit 6ee1223
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/src/local_db.jl b/src/local_db.jl
@@ -143,6 +143,7 @@ get_dist_mp(db::LocalDB) = db.dist_mp
 get_dist_sp(db::LocalDB) = db.dist_sp
 get_core_main(db::LocalDB) = db.core_main
 get_div(db::LocalDB) = db.div
+get_t(db::LocalDB) = db.div["t"]::TuLiPa.ProbTime # TODO: Remove?
 
 get_ifm(db::LocalDB) = db.ifm
 get_dist_ifm(db::LocalDB) = db.dist_ifm

diff --git a/src/prob_util.jl b/src/prob_util.jl
@@ -147,6 +147,13 @@ function get_states(modelobjects::Vector)
     return states
 end
 
+# Step info-----------------------------------------------------------------------
+function update_stepinfo(t)
+    db = get_local_db()
+    db.div["t"] = t
+    return
+end
+
 # Startstates-------------------------------------------------------------------------------------
 function update_startstates(stepnr, t)
     db = get_local_db()

diff --git a/src/run_serial.jl b/src/run_serial.jl
@@ -745,6 +745,14 @@ function step_jules(t, steplength, stepnr, skipmed)
             end
         end
     end
+
+    println(t)
+    println("Step info")
+    @time begin
+        @sync for core in cores
+            @spawnat core update_stepinfo(t)
+        end
+    end
 
     println("Startstates")
     @time begin