update after road pricing shp file creation

matsim-scenarios · Mar 13, 2024 · 2a8aabd · 2a8aabd
1 parent 91108c6
commit 2a8aabd
Showing 1 changed file with 31 additions and 5 deletions.
diff --git a/src/main/java/org/matsim/run/RoadPricingOptions.java b/src/main/java/org/matsim/run/RoadPricingOptions.java
@@ -9,9 +9,12 @@
 import org.matsim.contrib.roadpricing.RoadPricingUtils;
 import org.matsim.core.utils.geometry.geotools.MGC;
 import org.matsim.core.utils.misc.Time;
+import org.opengis.feature.simple.SimpleFeature;
 import picocli.CommandLine;
 
 import java.nio.file.Path;
+import java.util.ArrayList;
+import java.util.List;
 
 /**
  * This class bundles some run parameter options and functionalities connected to road-pricing-scenarios.
@@ -47,18 +50,41 @@ void configureAreaTollScheme(Scenario scenario) {
 			Time.parseTime("22:00:00"),
 			RoadPricingOptions.toll);
 
-		Geometry geometry = new ShpOptions(roadPricingAreaPath, null, null).getGeometry();
+		ShpOptions shp = new ShpOptions(roadPricingAreaPath, null, null);
+
+		List<SimpleFeature> features = shp.readFeatures();
 
 		for (Link link : scenario.getNetwork().getLinks().values()) {
 			if (link.getId().toString().contains("pt_")) {
 				continue;
 			}
 
-			boolean isInsideArea = MGC.coord2Point(link.getFromNode().getCoord()).within(geometry)
-				|| MGC.coord2Point(link.getToNode().getCoord()).within(geometry);
+			Geometry geometry;
+
+			for (SimpleFeature feat : features) {
+				geometry = (Geometry) feat.getDefaultGeometry();
+
+				boolean isInsideArea = MGC.coord2Point(link.getFromNode().getCoord()).within(geometry)
+					&& MGC.coord2Point(link.getToNode().getCoord()).within(geometry);
+
+				boolean isRoadType = true;
+
+				List<String> roadTypes = new ArrayList<>(List.of("highway.motorway", "highway.primary", "highway.trunk", "highway.secondary"));
+
+				if (features.size() > 1) {
+//					case avenidas principales
+					if (feat.getAttribute("name").toString().equals("avenida-gran-canal") || feat.getAttribute("name").toString().equals("avenida-canal-nacional")) {
+//						add highway tertiary for avenida gran canal and avenida canal nacional
+						roadTypes.add("highway.tertiary");
+					}
+					isRoadType = roadTypes.stream().anyMatch(link.getAttributes().getAttribute("type").toString()::contains);
+				}
 
-			if (isInsideArea) {
-				RoadPricingUtils.addLink(scheme, link.getId());
+				if (isInsideArea && isRoadType) {
+					RoadPricingUtils.addLink(scheme, link.getId());
+					link.getAttributes().putAttribute("roadPricing", "enabled");
+					break;
+				}
 			}
 		}
 	}