fix: using the person analysis filter in PublicTransportLegReaderFrom…

…Population (#203)
eqasim-org · Mar 12, 2024 · 3acf883 · 3acf883
1 parent 6082c28
commit 3acf883
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/core/src/main/java/org/eqasim/core/analysis/pt/PublicTransportLegReaderFromPopulation.java b/core/src/main/java/org/eqasim/core/analysis/pt/PublicTransportLegReaderFromPopulation.java
@@ -38,7 +38,10 @@ public Collection<PublicTransportLegItem> readPublicTransportLegs(String populat
     }
 
     public Collection<PublicTransportLegItem> readPublicTransportLegs(Population population) {
-        return population.getPersons().values().stream().flatMap(person -> getPublicTransportLegs(person).stream()).collect(Collectors.toList());
+        return population.getPersons().values().stream()
+                .filter(person -> personAnalysisFilter.analyzePerson(person.getId()))
+                .flatMap(person -> getPublicTransportLegs(person).stream())
+                .collect(Collectors.toList());
     }
 
     public Collection<PublicTransportLegItem> getPublicTransportLegs(Person person) {