Make events writing configurable

Author: janekdererste

src/simulation/config.rs

@@ -43,6 +43,7 @@ impl Config {
             config.set_output(Output {
                 output_dir: out_dir,
                 profiling: config.output().profiling,
+                write_events: Default::default(),
             });
         }
         config
@@ -90,6 +91,7 @@ impl Config {
             let default = Output {
                 output_dir: "./".to_string(),
                 profiling: Profiling::None,
+                write_events: Default::default(),
             };
             self.modules
                 .borrow_mut()
@@ -157,6 +159,8 @@ pub struct Output {
     pub output_dir: String,
     #[serde(default)]
     pub profiling: Profiling,
+    #[serde(default)]
+    pub write_events: WriteEvents,
 }
 
 #[derive(Serialize, Deserialize, Clone)]
@@ -242,6 +246,13 @@ pub enum Profiling {
     CSV(ProfilingLevel),
 }
 
+#[derive(PartialEq, Debug, Clone, Serialize, Deserialize, Default)]
+pub enum WriteEvents {
+    #[default]
+    None,
+    Proto,
+}
+
 #[derive(PartialEq, Debug, Clone, Serialize, Deserialize, Default)]
 pub struct ProfilingLevel {
     #[serde(default = "default_profiling_level")]

src/simulation/controller.rs

@@ -9,7 +9,9 @@ use mpi::traits::{Communicator, CommunicatorCollectives};
 use nohash_hasher::IntMap;
 use tracing::info;
 
-use crate::simulation::config::{CommandLineArgs, Config, PartitionMethod, RoutingMode};
+use crate::simulation::config::{
+    CommandLineArgs, Config, PartitionMethod, RoutingMode, WriteEvents,
+};
 use crate::simulation::io::proto_events::ProtoEventsWriter;
 use crate::simulation::messaging::communication::communicators::{
     ChannelSimCommunicator, MpiSimCommunicator, SimCommunicator,
@@ -126,12 +128,13 @@ fn execute_partition<C: SimCommunicator + 'static>(comm: C, args: &CommandLineAr
 
     let mut events = EventsPublisher::new();
 
-    let events_file = format!("events.{rank}.binpb");
-    let events_path = output_path.join(events_file);
-    events.add_subscriber(Box::new(ProtoEventsWriter::new(&events_path)));
+    if config.output().write_events == WriteEvents::Proto {
+        let events_file = format!("events.{rank}.binpb");
+        let events_path = output_path.join(events_file);
+        events.add_subscriber(Box::new(ProtoEventsWriter::new(&events_path)));
+    }
     let travel_time_collector = Box::new(TravelTimeCollector::new());
     events.add_subscriber(travel_time_collector);
-    //events.add_subscriber(Box::new(EventsLogger {}));
 
     let rc = Rc::new(comm);