org.opentripplanner.ext.fares.impl.HSLFareServiceFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of application Show documentation
The newest version!
package org.opentripplanner.ext.fares.impl;

import java.util.Collection;
import java.util.Map;
import org.opentripplanner.ext.fares.model.FareAttribute;
import org.opentripplanner.ext.fares.model.FareRule;
import org.opentripplanner.ext.fares.model.FareRuleSet;
import org.opentripplanner.routing.core.FareType;
import org.opentripplanner.routing.fares.FareService;
import org.opentripplanner.transit.model.framework.FeedScopedId;
import org.opentripplanner.transit.model.network.Route;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class HSLFareServiceFactory extends DefaultFareServiceFactory {

  private static final Logger LOG = LoggerFactory.getLogger(HSLFareService.class);

  @Override
  protected void fillFareRules(
    Collection fareAttributes,
    Collection fareRules,
    Map fareRuleSet
  ) {
    /*
     * Create an empty FareRuleSet for each FareAttribute, as some FareAttribute may have no
     * rules attached to them.
     */
    for (FareAttribute fare : fareAttributes) {
      FeedScopedId id = fare.getId();
      FareRuleSet fareRule = fareRuleSet.get(id);
      if (fareRule == null) {
        fareRule = new FareRuleSet(fare);
        fareRuleSet.put(id, fareRule);
      }
    }

    /*
     * For each fare rule, add it to the FareRuleSet of the fare.
     */
    for (FareRule rule : fareRules) {
      FareAttribute fare = rule.getFare();
      FeedScopedId id = fare.getId();
      FareRuleSet fareRule = fareRuleSet.get(id);
      if (fare.getAgency() != null) {
        fareRule.setAgency(fare.getAgency());
      }

      if (fareRule == null) {
        // Should never happen by design
        LOG.error("Inexistant fare ID in fare rule: " + id);
        continue;
      }
      String contains = rule.getContainsId();
      if (contains != null) {
        fareRule.addContains(contains);
      }

      String origin = rule.getOriginId();
      String destination = rule.getDestinationId();
      Route route = rule.getRoute();

      if (route != null) {
        FeedScopedId routeId = route.getId();
        if (origin != null && destination != null) {
          fareRule.addRouteOriginDestination(routeId.toString(), origin, destination);
        } else {
          fareRule.addRoute(routeId);
        }
      } else {
        if (origin != null || destination != null) {
          fareRule.addOriginDestination(origin, destination);
        }
      }
    }
  }

  public FareService makeFareService() {
    HSLFareService fareService = new HSLFareService();
    fareService.addFareRules(FareType.regular, regularFareRules.values());
    if (LOG.isDebugEnabled()) {
      for (FareRuleSet ruleSet : regularFareRules.values()) {
        LOG.debug("farerule {}", ruleSet);
        LOG.debug("ruleattr {}", ruleSet.getFareAttribute());
        LOG.debug("hasAgency {}", ruleSet.hasAgencyDefined());
        LOG.debug("agency {}", ruleSet.getAgency());
      }
    }
    return fareService;
  }
}