org.opentripplanner.routing.algorithm.strategies.MultiTargetTerminationStrategy 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.routing.algorithm.strategies;

import java.util.HashSet;
import java.util.Set;

import org.opentripplanner.routing.core.RoutingRequest;
import org.opentripplanner.routing.core.State;
import org.opentripplanner.routing.graph.Vertex;
import org.opentripplanner.routing.spt.ShortestPathTree;

/**
 * A termination strategy that terminates after multiple targets have been reached.
 * 
 * Useful for implementing a restricted batch search - i.e. doing one-to-many search 
 * without building a full shortest path tree.
 * 
 * @author avi
 */
public class MultiTargetTerminationStrategy implements SearchTerminationStrategy {

    private final Set unreachedTargets;
    private final Set reachedTargets;
    
    public MultiTargetTerminationStrategy(Set targets) {
        unreachedTargets = new HashSet(targets);
        reachedTargets = new HashSet(targets.size());
    }
    
    /**
     * Updates the list of reached targets and returns True if all the
     * targets have been reached.
     */
    @Override
    public boolean shouldSearchTerminate(Vertex origin, Vertex target, State current,
                                         ShortestPathTree spt, RoutingRequest traverseOptions) {
        Vertex currentVertex = current.getVertex();
                
        // TODO(flamholz): update this to handle vertices that are not in the graph
        // but rather along edges in the graph.
        if (unreachedTargets.contains(currentVertex)) {
            unreachedTargets.remove(currentVertex);
            reachedTargets.add(currentVertex);
        }
        return unreachedTargets.size() == 0;
    }

}