org.opentripplanner.graph_builder.module.shapefile.CompositeStreetTraversalPermissionConverter Maven / Gradle / Ivy

package org.opentripplanner.graph_builder.module.shapefile;

import java.util.ArrayList;
import java.util.Collection;

import org.opengis.feature.simple.SimpleFeature;
import org.opentripplanner.common.model.P2;
import org.opentripplanner.graph_builder.services.shapefile.SimpleFeatureConverter;
import org.opentripplanner.routing.edgetype.StreetTraversalPermission;

/**
 * A converter which is a composite of other converters. It can combine them with an "and" or "or"
 * strategy. The orPermissions variable controls that.
 * 
 * @author rob
 * 
 */
public class CompositeStreetTraversalPermissionConverter implements SimpleFeatureConverter> {

    private Collection>> converters = new ArrayList>>();

    private boolean orPermissions = false;

    public CompositeStreetTraversalPermissionConverter() {
    }

    /**
     * Is the or combination strategy being used?
     * 
     * @return whether the or combination strategy is used
     */
    public boolean isOrPermissions() {
        return orPermissions;
    }

    public void setOrPermissions(boolean orPermissions) {
        this.orPermissions = orPermissions;
    }

    /**
     * set the list of converters used to the passed in parameter
     * 
     * @param converters
     *            list of converters to use
     */
    public void setConverters(
            Collection>> converters) {
        this.converters = converters;
    }

    /**
     * use the permission combination strategy to combine the results of the list of converters
     */
    @Override
    public P2 convert(SimpleFeature feature) {
        P2 result = null;
        for (SimpleFeatureConverter> converter : converters) {
            P2 value = converter.convert(feature);
            if (result == null) {
                result = value;
            } else {
                StreetTraversalPermission first, second;
                if (orPermissions) {
                    first = result.first.add(value.first);
                    second = result.second.add(value.second);
                } else {
                    first = StreetTraversalPermission.get(result.first.code & value.first.code);
                    second = StreetTraversalPermission.get(result.second.code & value.second.code);
                }
                result = new P2(first, second);
            }

        }
        return result;
    }

    /**
     * add a converter to the list to be applied
     * @param converter the new converter
     */
    public void add(SimpleFeatureConverter> converter) {
        converters.add(converter);
    }

}