write directions to performance file

Tehforsch · Sep 19, 2023 · 4b67578 · 4b67578
1 parent 7ccd8a2
commit 4b67578
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/sweep/mod.rs b/src/sweep/mod.rs
@@ -140,6 +140,7 @@ impl SubsweepPlugin for SweepPlugin {
                 hydrogen_ionization_mass_average_system
                     .before(num_particles_at_timestep_levels_system::<HydrogenOnly>),
             )
+            .add_startup_system_to_stage(StartupStages::InitSweep, show_num_directions_system)
             .add_system_to_stage(
                 Stages::AfterSweep,
                 num_particles_at_timestep_levels_system::<HydrogenOnly>,
@@ -729,3 +730,10 @@ pub fn initialize_sweep_test_components_system(
         ));
     }
 }
+
+fn show_num_directions_system(
+    parameters: Res<SweepParameters>,
+    mut performance_data: ResMut<Performance>,
+) {
+    performance_data.record_number("num_sweep_directions", parameters.directions.num());
+}