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

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

import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.Multimap;
import java.time.LocalDateTime;
import java.util.Collection;
import java.util.stream.Collectors;
import org.opentripplanner.model.FareZone;
import org.opentripplanner.model.FeedScopedId;
import org.opentripplanner.netex.index.api.ReadOnlyHierarchicalVersionMapById;
import org.opentripplanner.netex.mapping.support.FeedScopedIdFactory;
import org.rutebanken.netex.model.TariffZone;
import org.rutebanken.netex.model.TariffZoneRef;

class TariffZoneMapper {
  private final LocalDateTime startOfPeriod;
  private final FeedScopedIdFactory idFactory;
  private final ReadOnlyHierarchicalVersionMapById tariffZonesById;
  private final Multimap deduplicateCache = ArrayListMultimap.create();


  TariffZoneMapper(
          LocalDateTime startOfPeriod,
          FeedScopedIdFactory idFactory,
          ReadOnlyHierarchicalVersionMapById tariffZonesById
  ) {
    this.startOfPeriod = startOfPeriod;
    this.idFactory = idFactory;
    this.tariffZonesById = tariffZonesById;
  }

  /**
   * Map all current TariffZones.
   */
  Collection listAllCurrentFareZones() {
    return tariffZonesById.localListCurrentVersionEntities(startOfPeriod).stream()
            .map(this::mapTariffZone)
            .collect(Collectors.toUnmodifiableList());
  }

  /**
   * Map Netex TariffZone to OTP TariffZone
   */
  FareZone findAndMapTariffZone(TariffZoneRef ref) {
    var tariffZone = tariffZonesById.lookup(ref, startOfPeriod);
    return (tariffZone == null) ? null : mapTariffZone(tariffZone);
  }

  /**
   * Map Netex TariffZone to OTP TariffZone
   */
  private FareZone mapTariffZone(org.rutebanken.netex.model.TariffZone tariffZone) {
    if(tariffZone == null) { return null; }

    FeedScopedId id = idFactory.createId(tariffZone.getId());
    String name = tariffZone.getName().getValue();
    return deduplicate(new FareZone(id, name));
  }

  private FareZone deduplicate(FareZone candidate) {
    var existing = deduplicateCache.get(candidate.getId()).stream()
            .filter(candidate::sameValueAs)
            .findFirst();

    if(existing.isPresent()) { return existing.get(); }

    deduplicateCache.put(candidate.getId(), candidate);
    return candidate;
  }
}