org.opentripplanner.netex.mapping.GroupNetexMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of otp Show documentation

The OpenTripPlanner multimodal journey planning system

There is a newer version: 2.6.0

package org.opentripplanner.netex.mapping;

import com.google.common.collect.ArrayListMultimap;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import org.opentripplanner.graph_builder.DataImportIssueStore;
import org.opentripplanner.model.impl.OtpTransitServiceBuilder;
import org.opentripplanner.model.transfer.ConstrainedTransfer;
import org.opentripplanner.netex.mapping.support.FeedScopedIdFactory;
import org.rutebanken.netex.model.ServiceJourneyInterchange;

/**
 * Entities x-referencing entries in more than on independent file must be mapped AFTER all
 * individual files are processed. This class is responsible for the "post-process-mapping", and
 * delegate to mappers for each specific type.
 */
class GroupNetexMapper {

  private final FeedScopedIdFactory idFactory;
  private final DataImportIssueStore issueStore;
  private final OtpTransitServiceBuilder transitBuilder;
  private final List interchanges = new ArrayList<>();

  /**
   * A map from trip/serviceJourney id to an ordered list of scheduled stop point ids.
   */
  final ArrayListMultimap scheduledStopPointsIndex = ArrayListMultimap.create();

  GroupNetexMapper(
    FeedScopedIdFactory idFactory,
    DataImportIssueStore issueStore,
    OtpTransitServiceBuilder transitBuilder
  ) {
    this.idFactory = idFactory;
    this.issueStore = issueStore;
    this.transitBuilder = transitBuilder;
  }

  void addInterchange(Collection interchanges) {
    this.interchanges.addAll(interchanges);
  }

  void mapGroupEntries() {
    mapInterchanges();
  }

  private void mapInterchanges() {
    var mapper = new TransferMapper(
      idFactory,
      issueStore,
      scheduledStopPointsIndex,
      transitBuilder.getTripsById()
    );
    for (ServiceJourneyInterchange it : interchanges) {
      ConstrainedTransfer result = mapper.mapToTransfer(it);
      if (result != null) {
        transitBuilder.getTransfers().add(result);
      }
    }
  }
}