Best code snippets using com.graphhopper.storage.GraphHopperStorage.getExtension(Showing top 7 results out of 315)

/**
 * Create a new storage from the specified one without copying the data.
 */
public static GraphHopperStorage newStorage(GraphHopperStorage store) {
  Directory outdir = guessDirectory(store);
  boolean is3D = store.getNodeAccess().is3D();
  return new GraphHopperStorage(store.getCHWeightings(), outdir, store.getEncodingManager(),
      is3D, store.getExtension()).
      create(store.getNodes());
}

@Test
public void testBasicTurnCosts() {
  GraphHopperStorage g = createStorage(createEncodingManager(false));
  initGraph(g);
  TurnCostExtension tcs = (TurnCostExtension) g.getExtension();
  Path p = createAlgo(g, AlgorithmOptions.start().
      weighting(createWeighting(carEncoder, tcs, 40)).
      traversalMode(TraversalMode.EDGE_BASED_1DIR).build()).
      calcPath(5, 1);
  // no restriction and costs
  EdgeIteratorState e3_6 = getEdge(g, 5, 6);
  e3_6.setDistance(2);
  assertEquals(Helper.createTList(5, 2, 3, 1), p.calcNodes());
  // now introduce some turn costs
  long tflags = carEncoder.getTurnFlags(false, 2);
  tcs.addTurnInfo(getEdge(g, 5, 2).getEdge(), 2, getEdge(g, 2, 3).getEdge(), tflags);
  p = createAlgo(g, AlgorithmOptions.start().
      weighting(createWeighting(carEncoder, tcs, 40)).
      traversalMode(TraversalMode.EDGE_BASED_1DIR).build()).
      calcPath(5, 1);
  assertEquals(Helper.createTList(5, 6, 3, 1), p.calcNodes());
}

@Test
public void testBasicTurnRestriction() {
  GraphHopperStorage g = createStorage(createEncodingManager(true));
  initGraph(g);
  TurnCostExtension tcs = (TurnCostExtension) g.getExtension();
  initTurnRestrictions(g, tcs, carEncoder);
  Path p = createAlgo(g, AlgorithmOptions.start().
      weighting(createWeighting(carEncoder, tcs, 40)).
      traversalMode(TraversalMode.EDGE_BASED_2DIR).build()).
      calcPath(5, 1);
  assertEquals(Helper.createTList(5, 2, 3, 4, 7, 6, 3, 1), p.calcNodes());
  // test 7-6-5 and reverse
  p = createAlgo(g, AlgorithmOptions.start().
      weighting(createWeighting(carEncoder, tcs, 40)).
      traversalMode(TraversalMode.EDGE_BASED_1DIR).build()).
      calcPath(5, 7);
  assertEquals(Helper.createTList(5, 6, 7), p.calcNodes());
  p = createAlgo(g, AlgorithmOptions.start().
      weighting(createWeighting(carEncoder, tcs, 40)).
      traversalMode(TraversalMode.EDGE_BASED_1DIR).build()).
      calcPath(7, 5);
  assertEquals(Helper.createTList(7, 6, 3, 2, 5), p.calcNodes());
}

@Test
public void testUTurns() {
  GraphHopperStorage g = createStorage(createEncodingManager(true));
  initGraph(g);
  TurnCostExtension tcs = (TurnCostExtension) g.getExtension();
  long tflags = carEncoder.getTurnFlags(true, 0);
  // force u-turn via lowering the cost for it
  EdgeIteratorState e3_6 = getEdge(g, 3, 6);
  e3_6.setDistance(0.1);
  getEdge(g, 3, 2).setDistance(864);
  getEdge(g, 1, 0).setDistance(864);
  tcs.addTurnInfo(getEdge(g, 7, 6).getEdge(), 6, getEdge(g, 6, 5).getEdge(), tflags);
  tcs.addTurnInfo(getEdge(g, 4, 3).getEdge(), 3, e3_6.getEdge(), tflags);
  AlgorithmOptions opts = AlgorithmOptions.start().
      weighting(createWeighting(carEncoder, tcs, 50)).
      traversalMode(TraversalMode.EDGE_BASED_2DIR_UTURN).build();
  Path p = createAlgo(g, opts).calcPath(7, 5);
  assertEquals(Helper.createTList(7, 6, 3, 6, 5), p.calcNodes());
  // no u-turn for 6-3
  opts = AlgorithmOptions.start().
      weighting(createWeighting(carEncoder, tcs, 100)).
      traversalMode(TraversalMode.EDGE_BASED_2DIR_UTURN).build();
  tcs.addTurnInfo(getEdge(g, 6, 3).getEdge(), 3, getEdge(g, 3, 6).getEdge(), tflags);
  p = createAlgo(g, opts).calcPath(7, 5);
  assertEquals(Helper.createTList(7, 6, 3, 2, 5), p.calcNodes());
}

for (String gtfsFile : gtfsFiles) {
  try {
    ((GtfsStorage) graphHopperStorage.getExtension()).loadGtfsFromFile("gtfs_" + id++, new ZipFile(gtfsFile));
  } catch (IOException e) {
    throw new RuntimeException(e);
  FakeWalkNetworkBuilder.buildWalkNetwork(((GtfsStorage) graphHopperStorage.getExtension()).getGtfsFeeds().values(), graphHopperStorage, ptFlagEncoder, Helper.DIST_EARTH);

final GraphHopperStorage g = createStorage(createEncodingManager(false));
initGraph(g);
TurnCostExtension tcs = (TurnCostExtension) g.getExtension();

/**
 * Create a new storage from the specified one without copying the data.
 */
public static GraphHopperStorage newStorage(GraphHopperStorage store) {
  Directory outdir = guessDirectory(store);
  boolean is3D = store.getNodeAccess().is3D();
  return new GraphHopperStorage(store.getCHWeightings(), outdir, store.getEncodingManager(),
      is3D, store.getExtension()).
      create(store.getNodes());
}