org.opentripplanner.netex.mapping.support.NetexMapperIndexes Maven / Gradle / Ivy
package org.opentripplanner.netex.mapping.support;
import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.Multimap;
import java.util.HashMap;
import java.util.Map;
import org.opentripplanner.model.StopTime;
import org.opentripplanner.netex.index.api.NetexEntityIndexReadOnlyView;
import org.opentripplanner.netex.index.api.ReadOnlyHierarchicalMapById;
import org.opentripplanner.transit.model.site.Station;
import org.rutebanken.netex.model.DatedServiceJourney;
/**
* The responsibility of this class is to cache and index NeTEx entities used by more than one
* mapper.
*
* The typical use case is that the same index is used in several mappers, so to avoid repeating the
* same logic and create the same index twice, we instead cache it here.
*
* Only local current level entities are cached, and the cache is thrown away for each set of files
* loaded (shared files, shared group files and group files).
*/
public class NetexMapperIndexes {
private final Multimap stationsByMultiModalStationRfs;
private final Map stopTimesByNetexId;
private final Multimap datedServiceJourneysBySjId;
private NetexMapperIndexes parent;
public NetexMapperIndexes(NetexEntityIndexReadOnlyView index, NetexMapperIndexes parent) {
this.parent = parent;
if (parent == null) {
this.datedServiceJourneysBySjId = indexDSJBySJId(index.getDatedServiceJourneys());
this.stationsByMultiModalStationRfs = ArrayListMultimap.create();
this.stopTimesByNetexId = new HashMap<>();
} else {
// Cached by level(shared files, shared group files and group files). If any entries exist at
// the current level, then they will hide entries at a higher level.
this.datedServiceJourneysBySjId = index.getDatedServiceJourneys().localKeys().isEmpty()
? parent.datedServiceJourneysBySjId
: indexDSJBySJId(index.getDatedServiceJourneys());
// Feed global instances. These fields contain mapping from a netex id to a OTP domain
// model object, hence we are not adding a lot of data to memory - only the id to object
// mapping.
this.stationsByMultiModalStationRfs = parent.stationsByMultiModalStationRfs;
this.stopTimesByNetexId = parent.stopTimesByNetexId;
}
}
public NetexMapperIndexes getParent() {
return parent;
}
public Multimap getStationsByMultiModalStationRfs() {
return stationsByMultiModalStationRfs;
}
public void addStationByMultiModalStationRfs(
Multimap stationByMultiModalStationRfs
) {
this.stationsByMultiModalStationRfs.putAll(stationByMultiModalStationRfs);
}
/**
* This is needed to assign a notice to a stop time. It is not part of the target
* OTPTransitService, so we need to temporally cache this here.
*/
public Map getStopTimesByNetexId() {
return stopTimesByNetexId;
}
public void addStopTimesByNetexId(Map stopTimesByNetexId) {
this.stopTimesByNetexId.putAll(stopTimesByNetexId);
}
public Multimap getDatedServiceJourneysBySjId() {
return datedServiceJourneysBySjId;
}
/* utility methods */
static Multimap indexDSJBySJId(
ReadOnlyHierarchicalMapById datedServiceJourneys
) {
Multimap dsjBySJId = ArrayListMultimap.create();
for (DatedServiceJourney dsj : datedServiceJourneys.localValues()) {
// The validation step ensure no NPE occurs here
String sjId = dsj.getJourneyRef().get(0).getValue().getRef();
dsjBySJId.put(sjId, dsj);
}
return dsjBySJId;
}
}