Measure simulation time and startup time in the .h

BioDynaMo · Oct 12, 2024 · ef73746 · ef73746
1 parent dec2ac9
commit ef73746
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 3 deletions.
diff --git a/demo/pyramidal_cell/src/pyramidal_cell.h b/demo/pyramidal_cell/src/pyramidal_cell.h
@@ -171,12 +171,21 @@ inline void SaveNeuronMorphology(Simulation& sim) {
 
 inline int Simulate(int argc, const char** argv) {
   neuroscience::InitModule();
+
+  auto start = std::chrono::high_resolution_clock::now();
   Simulation simulation(argc, argv);
+  auto end = std::chrono::high_resolution_clock::now();
+  auto elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(end - start).count();
+
   AddInitialNeuron({150, 150, 0});
   CreateExtracellularSubstances(simulation.GetParam());
+
+  auto sim_start = std::chrono::high_resolution_clock::now();
   simulation.GetScheduler()->Simulate(500);
+  auto sim_end = std::chrono::high_resolution_clock::now();
+  auto sim_elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(sim_end - sim_start).count();
   SaveNeuronMorphology(simulation);
-  std::cout << "Simulation completed successfully!" << std::endl;
+  std::cout << "Simulation completed successfully! Startup: " << elapsed << " ns" << " Run: " << sim_elapsed << " ns" << std::endl;
   return 0;
 }
 

diff --git a/demo/soma_clustering/src/soma_clustering.h b/demo/soma_clustering/src/soma_clustering.h
@@ -45,7 +45,11 @@ inline int Simulate(int argc, const char** argv) {
     // param->unschedule_default_operations = {"mechanical forces"};
   };
 
+  auto start = std::chrono::high_resolution_clock::now();
   Simulation simulation(argc, argv, set_param);
+  auto end = std::chrono::high_resolution_clock::now();
+  auto elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(end - start).count();
+
 
   // Define initial model
   auto* param = simulation.GetParam();
@@ -99,7 +103,12 @@ inline int Simulate(int argc, const char** argv) {
                  "We recommend to run the simulation for roughly 6000 time "
                  "steps. Please change 'timesteps = ..'.");
   }
+
+  auto sim_start = std::chrono::high_resolution_clock::now();
   simulation.GetScheduler()->Simulate(timesteps);
+  auto sim_end = std::chrono::high_resolution_clock::now();
+  auto sim_elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(sim_end - sim_start).count();
+
 
   // Check if criterion is met
   real_t spatial_range = 15;
@@ -109,7 +118,7 @@ inline int Simulate(int argc, const char** argv) {
   } else {
     std::cout << "<SomaClustering> Clustering criterion not met!" << std::endl;
   }
-  std::cout << "Simulation completed successfully!\n";
+  std::cout << "Simulation completed successfully! Startup: " << elapsed << " ns" << " Run: " << sim_elapsed << " ns" << std::endl;
   return 0;
 }
 

diff --git a/demo/tumor_concept/src/tumor_concept.h b/demo/tumor_concept/src/tumor_concept.h
@@ -103,7 +103,11 @@ inline int Simulate(int argc, const char** argv) {
     param->max_bound = 100;  // cube of 100*100*100
   };
 
+  auto start = std::chrono::high_resolution_clock::now();
   Simulation simulation(argc, argv, set_param);
+  auto end = std::chrono::high_resolution_clock::now();
+  auto elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(end - start).count();
+
   auto* ctxt = simulation.GetExecutionContext();
   auto* param = simulation.GetParam();
   auto* myrand = simulation.GetRandom();
@@ -137,9 +141,13 @@ inline int Simulate(int argc, const char** argv) {
   ctxt->AddAgent(cell);  // put the created cell in our cells structure
 
   // Run simulation
+
+  auto sim_start = std::chrono::high_resolution_clock::now();
   simulation.GetScheduler()->Simulate(500);
+  auto sim_end = std::chrono::high_resolution_clock::now();
+  auto sim_elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(sim_end - sim_start).count();
 
-  std::cout << "Simulation completed successfully!" << std::endl;
+  std::cout << "Simulation completed successfully! Startup: " << elapsed << " ns" << " Run: " << sim_elapsed << " ns" << std::endl;
   return 0;
 }