org.opentripplanner.netex.mapping.calendar.DatedServiceJourneyMapper Maven / Gradle / Ivy

Go to download
package org.opentripplanner.netex.mapping.calendar;

import org.opentripplanner.model.calendar.ServiceDate;
import org.opentripplanner.netex.index.api.ReadOnlyHierarchicalMapById;
import org.rutebanken.netex.model.DatedServiceJourney;
import org.rutebanken.netex.model.OperatingDay;
import org.rutebanken.netex.model.OperatingDayRefStructure;

import javax.annotation.Nullable;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import static org.opentripplanner.netex.mapping.support.ServiceAlterationFilter.isRunning;


/**
 * This class is responsible for indexing and mapping {@link DatedServiceJourney}.
 */
public class DatedServiceJourneyMapper {

  public static Map> indexDSJBySJId(
      ReadOnlyHierarchicalMapById datedServiceJourneys
  ) {
    Map> dsjBySJId = new HashMap<>();
    for (DatedServiceJourney dsj : datedServiceJourneys.localValues()) {
      dsjBySJId.computeIfAbsent(
          // The validation step ensure no NPE occurs here
          dsj.getJourneyRef().get(0).getValue().getRef(),
          it -> new ArrayList<>()
      ).add(dsj);
    }
    return dsjBySJId;
  }

  /**
   * Map a list of DSJs to a set of service days.
   */
  public static Collection mapToServiceDates(
      Iterable dsjs,
      ReadOnlyHierarchicalMapById operatingDayById
  ) {
    List result = new ArrayList<>();
    for (DatedServiceJourney dsj : dsjs) {

      // TODO This currently skips mapping of any trips containing ServiceAlteration CANCELLATION
      //      or REPLACED. In the future we will want to import these and allow them to be routed
      //      on if a parameter is set.
      if (!isRunning(dsj.getServiceAlteration())) { continue; }

      OperatingDay opDay = operatingDay(dsj, operatingDayById);
      if(opDay != null) {
        result.add(OperatingDayMapper.map(opDay));
      }
      else {
        // There is a validation on this, so we should never get here
        throw new NullPointerException(
            "DatedServiceJourney operating-day not found. DSJ Id: " + dsj.getId()
        );
      }
    }
    return result;
  }

  @Nullable
  private static OperatingDay operatingDay(
      DatedServiceJourney dsj,
      ReadOnlyHierarchicalMapById operatingDayById
  ) {
    OperatingDayRefStructure operatingDayRef = dsj.getOperatingDayRef();
    return operatingDayRef == null ? null : operatingDayById.lookup(operatingDayRef.getRef());
  }
}