From 69184dabefb7ebeb874501207f73ceb056c5f500 Mon Sep 17 00:00:00 2001
From: janekdererste <janek.laudan@mailbox.org>
Date: Mon, 5 Feb 2024 14:08:36 +0100
Subject: [PATCH] Add run class to run barebones rvr scenario for timing agains
 rust qsim

---
 pom.xml                                       |   2 +-
 .../org/matsim/run/RunQasimComparison.java    | 100 ++++++++++++++++++
 2 files changed, 101 insertions(+), 1 deletion(-)
 create mode 100644 src/main/java/org/matsim/run/RunQasimComparison.java

diff --git a/pom.xml b/pom.xml
index 1ae70bc..25fbdff 100644
--- a/pom.xml
+++ b/pom.xml
@@ -147,7 +147,7 @@
         <dependency>
             <groupId>org.apache.commons</groupId>
             <artifactId>commons-csv</artifactId>
-            <version>1.6</version>
+            <version>1.10.0</version>
         </dependency>
 
         <dependency>
diff --git a/src/main/java/org/matsim/run/RunQasimComparison.java b/src/main/java/org/matsim/run/RunQasimComparison.java
new file mode 100644
index 0000000..18177e3
--- /dev/null
+++ b/src/main/java/org/matsim/run/RunQasimComparison.java
@@ -0,0 +1,100 @@
+package org.matsim.run;
+
+import org.apache.commons.csv.CSVFormat;
+import org.apache.commons.csv.CSVPrinter;
+import com.google.inject.Inject;
+import org.matsim.api.core.v01.TransportMode;
+import org.matsim.application.MATSimApplication;
+import org.matsim.contrib.bicycle.BicycleConfigGroup;
+import org.matsim.contrib.bicycle.BicycleModule;
+import org.matsim.contrib.bicycle.Bicycles;
+import org.matsim.core.config.Config;
+import org.matsim.core.config.ConfigUtils;
+import org.matsim.core.controler.AbstractModule;
+import org.matsim.core.controler.Controler;
+import org.matsim.core.controler.OutputDirectoryHierarchy;
+import org.matsim.core.mobsim.framework.events.MobsimBeforeCleanupEvent;
+import org.matsim.core.mobsim.framework.events.MobsimInitializedEvent;
+import org.matsim.core.mobsim.framework.listeners.MobsimBeforeCleanupListener;
+import org.matsim.core.mobsim.framework.listeners.MobsimInitializedListener;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Paths;
+import java.time.Duration;
+import java.time.Instant;
+
+public class RunQasimComparison extends RunMetropoleRuhrScenario {
+
+    public static void main(String[] args) {
+        MATSimApplication.run(RunQasimComparison.class, args);
+    }
+
+    @Override
+    protected Config prepareConfig(Config config) {
+
+        config.controler().setOverwriteFileSetting(OutputDirectoryHierarchy.OverwriteFileSetting.deleteDirectoryIfExists);
+        config.controler().setFirstIteration(0);
+        config.controler().setLastIteration(0);
+
+        config.transit().setUseTransit(false);
+        config.transit().setTransitScheduleFile(null);
+        config.transit().setVehiclesFile(null);
+
+        BicycleConfigGroup bikeConfigGroup = ConfigUtils.addOrGetModule(config, BicycleConfigGroup.class);
+        bikeConfigGroup.setBicycleMode(TransportMode.bike);
+
+        return config;
+    }
+
+    @Override
+    protected void prepareControler(Controler controler) {
+
+        controler.addOverridingModule(new BicycleModule());
+        controler.addOverridingModule(new AbstractModule() {
+            @Override
+            public void install() {
+                addMobsimListenerBinding().to(MobsimTimer.class);
+            }
+        });
+    }
+
+
+    private static class MobsimTimer implements MobsimInitializedListener, MobsimBeforeCleanupListener {
+
+        private Instant start;
+
+        @Inject
+        private Config config;
+
+        @Inject
+        private OutputDirectoryHierarchy outDir;
+
+        @Override
+        public void notifyMobsimInitialized(MobsimInitializedEvent e) {
+            start = Instant.now();
+        }
+
+        @Override
+        public void notifyMobsimBeforeCleanup(MobsimBeforeCleanupEvent e) {
+            var now = Instant.now();
+            var duration = Duration.between(start, now);
+            var size = config.qsim().getNumberOfThreads();
+            var filename = Paths.get(outDir.getOutputFilename("instrument-mobsim.csv"));
+            try (var writer = Files.newBufferedWriter(filename); var p = new CSVPrinter(writer, createWriteFormat("timestamp", "func", "duration", "size"))) {
+                p.printRecord(Instant.now().getNano(), "org.matsim.core.mobsim.qsim.run", duration.toNanos(), size);
+            } catch (IOException ex) {
+                throw new RuntimeException(ex);
+            }
+        }
+
+        public static CSVFormat createWriteFormat(String... header) {
+            return CSVFormat.DEFAULT.builder()
+                    .setHeader(header)
+                    .setSkipHeaderRecord(false)
+                    .build();
+        }
+    }
+
+
+}