add optional id filter

src/main/java/org/matsim/prepare/choices/ComputePlanChoices.java

@@ -37,6 +37,7 @@
 import java.nio.file.Path;
 import java.util.*;
 import java.util.concurrent.ConcurrentLinkedQueue;
+import java.util.regex.Pattern;
 
 
 @CommandLine.Command(
@@ -57,6 +58,10 @@ public class ComputePlanChoices implements MATSimAppCommand, PersonAlgorithm {
 	private int topK;
 	@CommandLine.Option(names = "--modes", description = "Modes to include in estimation", split = ",")
 	private Set<String> modes;
+
+	@CommandLine.Option(names = "--id-filter", description = "Filter for person ids")
+	private Pattern idFilter;
+
 	@CommandLine.Option(names = "--time-util-only", description = "Reset scoring for estimation and only use time utility", defaultValue = "false")
 	private boolean timeUtil;
 	@CommandLine.Option(names = "--calc-scores", description = "Perform pseudo scoring for each plan", defaultValue = "false")
@@ -112,6 +117,10 @@ public Integer call() throws Exception {
 			topK = 2;
 		}
 
+		if (idFilter != null) {
+			log.info("Using person id filter: {}", idFilter);
+		}
+
 		Controler controler = this.scenario.createControler();
 
 		controler.addOverridingModule(InformedModeChoiceModule.newBuilder()
@@ -209,6 +218,11 @@ public void run(Person person) {
 			return;
 		}
 
+		if (idFilter != null && !idFilter.matcher(person.getId().toString()).matches()) {
+			pb.step();
+			return;
+		}
+
 		Plan plan = person.getSelectedPlan();
 		PlanModel model = PlanModel.newInstance(plan);