From 68e4ae3de4bc82cef2672e0e8e934a5a3915ec52 Mon Sep 17 00:00:00 2001
From: rakow <rakow@vsp.tu-berlin.de>
Date: Tue, 10 Dec 2024 20:28:39 +0100
Subject: [PATCH] and choice experiment policy

---
 .../policies/OpenBerlinChoiceExperiment.java  | 45 +++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 src/main/java/org/matsim/run/policies/OpenBerlinChoiceExperiment.java

diff --git a/src/main/java/org/matsim/run/policies/OpenBerlinChoiceExperiment.java b/src/main/java/org/matsim/run/policies/OpenBerlinChoiceExperiment.java
new file mode 100644
index 00000000..0907006f
--- /dev/null
+++ b/src/main/java/org/matsim/run/policies/OpenBerlinChoiceExperiment.java
@@ -0,0 +1,45 @@
+package org.matsim.run.policies;
+
+import org.matsim.api.core.v01.Scenario;
+import org.matsim.api.core.v01.TransportMode;
+import org.matsim.api.core.v01.population.Leg;
+import org.matsim.api.core.v01.population.Person;
+import org.matsim.api.core.v01.population.Plan;
+import org.matsim.application.MATSimApplication;
+import org.matsim.core.router.TripStructureUtils;
+import org.matsim.run.OpenBerlinScenario;
+import picocli.CommandLine;
+
+/**
+ * This class can be used to run some synthetic choice experiments on the OpenBerlin scenario.
+ */
+public class OpenBerlinChoiceExperiment extends OpenBerlinScenario {
+
+	@CommandLine.Option(names = "--bike-speed-factor", description = "Speed factor for bikes", defaultValue = "1.0")
+	private double bikeSpeedFactor = 1.0;
+
+	public static void main(String[] args) {
+		MATSimApplication.execute(OpenBerlinChoiceExperiment.class, args);
+	}
+
+	@Override
+	protected void prepareScenario(Scenario scenario) {
+
+		super.prepareScenario(scenario);
+
+		// If bike speed is adjusted, we need to remove all bike routes and travel times
+		// These times will be recalculated by the router
+		if (bikeSpeedFactor != 1.0) {
+			for (Person person : scenario.getPopulation().getPersons().values()) {
+				for (Plan plan : person.getPlans()) {
+					for (Leg leg : TripStructureUtils.getLegs(plan)) {
+						if (leg.getMode().equals(TransportMode.bike)) {
+							leg.setRoute(null);
+							leg.setTravelTimeUndefined();
+						}
+					}
+				}
+			}
+		}
+	}
+}