org.opentripplanner.api.mapping.ItineraryMapper Maven / Gradle / Ivy

Go to download
package org.opentripplanner.api.mapping;

import java.util.Collection;
import java.util.GregorianCalendar;
import java.util.List;
import java.util.Locale;
import java.util.stream.Collectors;
import org.opentripplanner.api.model.ApiItinerary;
import org.opentripplanner.model.plan.Itinerary;

public class ItineraryMapper {

  private final LegMapper legMapper;
  private final FareMapper fareMapper;

  public ItineraryMapper(Locale locale, boolean addIntermediateStops) {
    this.legMapper = new LegMapper(locale, addIntermediateStops);
    this.fareMapper = new FareMapper(locale);
  }

  public List mapItineraries(Collection domain) {
    if (domain == null) {
      return null;
    }
    return domain.stream().map(this::mapItinerary).collect(Collectors.toList());
  }

  public ApiItinerary mapItinerary(Itinerary domain) {
    if (domain == null) {
      return null;
    }
    ApiItinerary api = new ApiItinerary();

    api.duration = domain.getDuration().toSeconds();
    api.startTime = GregorianCalendar.from(domain.startTime());
    api.endTime = GregorianCalendar.from(domain.endTime());
    api.walkTime = domain.getNonTransitDuration().toSeconds();
    api.transitTime = domain.getTransitDuration().toSeconds();
    api.waitingTime = domain.getWaitingDuration().toSeconds();
    api.walkDistance = domain.getNonTransitDistanceMeters();
    api.generalizedCost = domain.getGeneralizedCost();
    api.elevationLost = domain.getElevationLost();
    api.elevationGained = domain.getElevationGained();
    api.transfers = domain.getNumberOfTransfers();
    api.tooSloped = domain.isTooSloped();
    api.arrivedAtDestinationWithRentedBicycle = domain.isArrivedAtDestinationWithRentedVehicle();
    api.fare = fareMapper.mapFare(domain);
    api.legs = legMapper.mapLegs(domain.getLegs());
    api.systemNotices = SystemNoticeMapper.mapSystemNotices(domain.getSystemNotices());
    api.accessibilityScore = domain.getAccessibilityScore();

    return api;
  }
}