org.opentripplanner.ext.vectortiles.layers.vehiclerental.VehicleRentalLayerBuilder Maven / Gradle / Ivy

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.5.0

package org.opentripplanner.ext.vectortiles.layers.vehiclerental;

import org.locationtech.jts.geom.Coordinate;
import org.locationtech.jts.geom.Envelope;
import org.locationtech.jts.geom.Geometry;
import org.locationtech.jts.geom.Point;
import org.opentripplanner.common.geometry.GeometryUtils;
import org.opentripplanner.ext.vectortiles.LayerBuilder;
import org.opentripplanner.ext.vectortiles.PropertyMapper;
import org.opentripplanner.ext.vectortiles.VectorTilesResource;
import org.opentripplanner.routing.vehicle_rental.VehicleRentalPlace;
import org.opentripplanner.routing.vehicle_rental.VehicleRentalStationService;
import org.opentripplanner.routing.graph.Graph;

import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

public class VehicleRentalLayerBuilder extends LayerBuilder {
  enum MapperType { Digitransit }

  static Map>> mappers = Map.of(
      MapperType.Digitransit, DigitransitVehicleRentalPropertyMapper::create
  );

  private final Graph graph;

  public VehicleRentalLayerBuilder(Graph graph, VectorTilesResource.LayerParameters layerParameters) {
    super(
        layerParameters.name(),
        mappers.get(MapperType.valueOf(layerParameters.mapper())).apply(graph)
    );

    this.graph = graph;
  }

  @Override
  protected List getGeometries(Envelope query) {
    VehicleRentalStationService service = graph.getVehicleRentalStationService();
    if (service == null) {return List.of();}
    return service.getVehicleRentalPlaces()
        .stream()
        .map(vehicleRentalStation -> {
          Coordinate coordinate = new Coordinate(vehicleRentalStation.getLongitude(), vehicleRentalStation.getLatitude());
          Point point = GeometryUtils.getGeometryFactory().createPoint(coordinate);
          point.setUserData(vehicleRentalStation);
          return point;
        })
        .collect(Collectors.toList());
  }
}