class to create distance matrix from shape file

src/main/java/org/matsim/run/RunLausitzScenario.java

@@ -26,6 +26,7 @@
 import org.matsim.core.controler.Controler;
 import org.matsim.run.analysis.CommunityFilter;
 import org.matsim.run.analysis.CommuterAnalysis;
+import org.matsim.run.analysis.DistanceMatrix;
 import picocli.CommandLine;
 
 import javax.annotation.Nullable;
@@ -40,7 +41,7 @@
 		CreateCountsFromBAStData.class
 })
 @MATSimApplication.Analysis({
-		TravelTimeAnalysis.class, LinkStats.class, CheckPopulation.class, CommuterAnalysis.class, CommunityFilter.class
+		TravelTimeAnalysis.class, LinkStats.class, CheckPopulation.class, CommuterAnalysis.class, CommunityFilter.class, DistanceMatrix.class
 })
 public class RunLausitzScenario extends MATSimApplication {
 

src/main/java/org/matsim/run/analysis/DistanceMatrix.java

@@ -0,0 +1,111 @@
+package org.matsim.run.analysis;
+
+import org.apache.commons.csv.CSVPrinter;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+import org.locationtech.jts.geom.Geometry;
+import org.locationtech.jts.geom.Point;
+import org.matsim.api.core.v01.Coord;
+import org.matsim.application.MATSimAppCommand;
+import org.matsim.application.options.CsvOptions;
+import org.matsim.application.options.ShpOptions;
+import org.matsim.core.utils.geometry.CoordUtils;
+import org.matsim.core.utils.geometry.geotools.MGC;
+import org.matsim.core.utils.gis.ShapeFileReader;
+import org.opengis.feature.simple.SimpleFeature;
+import picocli.CommandLine;
+
+import java.nio.file.Path;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.function.Predicate;
+import java.util.stream.Collectors;
+
+@CommandLine.Command(name = "distance-matrix", description = "creates a csv with commuity keys within shape")
+public class DistanceMatrix implements MATSimAppCommand{
+
+	@CommandLine.Option(names = "--output", description = "output csv filepath", required = true)
+	private static Path output;
+
+	@CommandLine.Option(names = "--dilution-area", description = "shape to filter zones", required = false)
+	private static Path dilutionArea;
+
+	@CommandLine.Mixin
+	CsvOptions csvOptions = new CsvOptions();
+
+	@CommandLine.Mixin
+	ShpOptions shp = new ShpOptions();
+
+	private final List<String> distances = new ArrayList<>();
+	private static final Logger logger = LogManager.getLogger(DistanceMatrix.class);
+
+	public static void main(String[] args) {
+		new DistanceMatrix().execute(args);
+	}
+
+	@Override
+	public Integer call() throws Exception {
+
+		logger.info("Read features.");
+		List<SimpleFeature> communities = shp.readFeatures();
+
+		ArrayList<SimpleFeature> copy = new ArrayList<>(communities); //to prevent RuntimeExceptions
+
+		Predicate<Geometry> filter = getFilter(dilutionArea);
+
+		logger.info("Calculate distance matrix.");
+		String delimiter = csvOptions.getFormat().getDelimiterString();
+		for(var community: communities){
+
+			if(!filter.test((Geometry) community.getDefaultGeometry()))
+				continue;
+
+			//copy.remove(community);
+			String nameFrom = (String) community.getAttribute("ARS");
+
+			Point centroid = ((Geometry) community.getDefaultGeometry()).getCentroid();
+			Coord from = MGC.point2Coord(centroid);
+			for(var target: copy){
+
+				String nameTo = (String) target.getAttribute("ARS");
+
+				Point centroid2 = ((Geometry) target.getDefaultGeometry()).getCentroid();
+				Coord to = MGC.point2Coord(centroid2);
+				double distance = CoordUtils.calcEuclideanDistance(from, to);
+
+				String distanceString = String.valueOf(distance).replace('.', ',');
+
+				distances.add(nameFrom + delimiter + nameTo + delimiter + distanceString);
+			}
+		}
+
+		logger.info("Print results to {}", output.toString());
+		CSVPrinter printer = csvOptions.createPrinter(output);
+		printer.print("from");
+		printer.print("to");
+		printer.print("distance");
+		printer.println();
+
+		for(String entry: distances){
+			for(String col: entry.split(csvOptions.getFormat().getDelimiterString()))
+				printer.print(col);
+			printer.println();
+		}
+		printer.close();
+
+		logger.info("Done!");
+		return 0;
+	}
+
+	private Predicate<Geometry> getFilter(Path path){
+
+		if(path == null)
+			return community -> true;
+
+		List<Geometry> geometries = ShapeFileReader.getAllFeatures(path.toString()).stream()
+				.map(feature -> (Geometry) feature.getDefaultGeometry())
+				.collect(Collectors.toList());
+
+		return community -> geometries.stream().anyMatch(geometry -> geometry.covers(community));
+	}
+}