org.opentripplanner.routing.vehicle_rental.VehicleRentalStationService Maven / Gradle / Ivy

Go to download
package org.opentripplanner.routing.vehicle_rental;

import java.io.Serializable;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import org.locationtech.jts.geom.Coordinate;
import org.locationtech.jts.geom.Envelope;
import org.opentripplanner.model.FeedScopedId;

public class VehicleRentalStationService implements Serializable {
    private static final long serialVersionUID = -1288992939159246764L;

    private final Map vehicleRentalStations = new HashMap<>();

    public Collection getVehicleRentalPlaces() {
        return vehicleRentalStations.values();
    }

    public VehicleRentalPlace getVehicleRentalPlace(FeedScopedId id) {
        return vehicleRentalStations.get(id);
    }

    public List getVehicleRentalVehicles() {
        return vehicleRentalStations.values()
                .stream()
                .filter(vehicleRentalPlace -> vehicleRentalPlace instanceof VehicleRentalVehicle)
                .map(VehicleRentalVehicle.class::cast)
                .collect(Collectors.toList());
    }

    public VehicleRentalVehicle getVehicleRentalVehicle(FeedScopedId id) {
        VehicleRentalPlace vehicleRentalPlace = vehicleRentalStations.get(id);
        return vehicleRentalPlace instanceof VehicleRentalVehicle
                ? (VehicleRentalVehicle) vehicleRentalPlace
                : null;
    }

    public List getVehicleRentalStations() {
        return vehicleRentalStations.values()
                .stream()
                .filter(vehicleRentalPlace -> vehicleRentalPlace instanceof VehicleRentalStation)
                .map(VehicleRentalStation.class::cast)
                .collect(Collectors.toList());
    }

    public VehicleRentalStation getVehicleRentalStation(FeedScopedId id) {
        VehicleRentalPlace vehicleRentalPlace = vehicleRentalStations.get(id);
        return vehicleRentalPlace instanceof VehicleRentalStation
                ? (VehicleRentalStation) vehicleRentalPlace
                : null;
    }

    public void addVehicleRentalStation(VehicleRentalPlace vehicleRentalStation) {
        // Remove old reference first, as adding will be a no-op if already present
        vehicleRentalStations.remove(vehicleRentalStation.getId());
        vehicleRentalStations.put(vehicleRentalStation.getId(), vehicleRentalStation);
    }

    public void removeVehicleRentalStation(FeedScopedId vehicleRentalStationId) {
        vehicleRentalStations.remove(vehicleRentalStationId);
    }

    /**
     * Gets all the vehicle rental stations inside the envelope. This is currently done by iterating
     * over a set, but we could use a spatial index if the number of vehicle rental stations is high
     * enough for performance to be a concern.
     */
    public List getVehicleRentalStationForEnvelope(
        double minLon, double minLat, double maxLon, double maxLat
    ) {
        Envelope envelope = new Envelope(
            new Coordinate(minLon, minLat),
            new Coordinate(maxLon, maxLat)
        );

        return vehicleRentalStations
            .values()
            .stream()
            .filter(b -> envelope.contains(new Coordinate(b.getLongitude(), b.getLatitude())))
            .collect(Collectors.toList());
    }
}