Пример #1
0
  public void addOSMEdgeToSpatialIndex(OSMEdge osmEdge) {
    // System.out.println("added edge with ID " + osmEdge.getId());
    // counter__edge.put(osmEdge.getId(), osmEdge.getGeometry());	// store edges in the hash map

    Geometry env = osmEdge.getGeometry().getBoundary();
    //  (minx, miny), (maxx, miny), (maxx, maxy), (minx, maxy), (minx, miny).

    if (env.getNumGeometries() > 0) {
      Point p1 = (Point) env.getGeometryN(0);
      Point p2 = (Point) env.getGeometryN(1);

      si.add(
          new Rectangle((float) p1.getX(), (float) p1.getY(), (float) p2.getX(), (float) p2.getY()),
          osmEdge.getId());
    }
  }
Пример #2
0
  public void dumpBuffer(List<OSMEdge> edges, String filename) throws SchemaException, IOException {

    final SimpleFeatureType TYPE =
        DataUtilities.createType(
            "route",
            "location:LineString:srid=4326,"
                + // <- the geometry attribute: Polyline type
                "name:String,"
                + // <- a String attribute
                "number:Integer" // a number attribute
            );

    // 1. build a feature
    SimpleFeatureBuilder featureBuilder = new SimpleFeatureBuilder(TYPE);
    ArrayList<SimpleFeature> features = new ArrayList<SimpleFeature>();

    Iterator<OSMEdge> iter = edges.iterator();
    while (iter.hasNext()) {
      OSMEdge o = iter.next();
      SimpleFeature feature = featureBuilder.buildFeature(null);
      feature.setDefaultGeometry(o.getGeometry());
      features.add(feature);
    }
    SimpleFeatureCollection collection =
        new ListFeatureCollection(TYPE, features); // FeatureCollections.newCollection();

    logger.info("Writing to shapefile " + filename);
    File newFile = new File(filename);
    // File newFile = getNewShapeFile(file);

    ShapefileDataStoreFactory dataStoreFactory = new ShapefileDataStoreFactory();

    Map<String, Serializable> params = new HashMap<String, Serializable>();
    params.put("url", newFile.toURI().toURL());
    params.put("create spatial index", Boolean.TRUE);

    ShapefileDataStore newDataStore =
        (ShapefileDataStore) dataStoreFactory.createNewDataStore(params);
    newDataStore.createSchema(TYPE);

    // You can comment out this line if you are using the createFeatureType method (at end of
    // class file) rather than DataUtilities.createType
    newDataStore.forceSchemaCRS(DefaultGeographicCRS.WGS84);

    Transaction transaction = new DefaultTransaction("create");

    String typeName = newDataStore.getTypeNames()[0];
    SimpleFeatureSource featureSource = newDataStore.getFeatureSource(typeName);

    if (featureSource instanceof SimpleFeatureStore) {
      SimpleFeatureStore featureStore = (SimpleFeatureStore) featureSource;

      featureStore.setTransaction(transaction);
      try {
        featureStore.addFeatures(collection);
        transaction.commit();
      } catch (Exception problem) {
        problem.printStackTrace();
        transaction.rollback();
      } finally {
        transaction.close();
      }
      // System.exit(0); // success!
    } else {
      logger.error(typeName + " does not support read/write access");
      // System.exit(1);	// exit program with status 1 (error)
    }
  }
Пример #3
0
  /**
   * Create a new graph, with linestrings read from the database (optionally using only a buffer
   * around a given track for the network)
   *
   * @param track GPS track consisting of a list of data points
   * @param bufferSize size of the buffer to select around the track
   * @param dumpFile if not empty, the path of a shapefile to dump the network into.
   */
  public void addLineStringsFromDatabase(
      ArrayList<Point> track, float bufferSize, String dumpFile) {
    setLineMergeGraphH4cked(new LineMergeGraphH4cked());
    distancesCalculated = false;

    Session session = HibernateUtil.getSessionFactory().getCurrentSession();
    session.beginTransaction();

    // Query gpsResults = session.createQuery("from" );

    if (track == null) {
      Query result = session.createQuery("from OSMEdge");

      @SuppressWarnings("unchecked")
      Iterator<OSMEdge> iter = result.iterate();
      while (iter.hasNext()) {
        OSMEdge o = iter.next();
        LineString g = o.getGeometry();
        addLineString(g, o.getId(), o.getEnvtype(), o.getCyktype(), o.getGroenm());
        // let us add this thing to the spatial index
        addOSMEdgeToSpatialIndex(o);
      }
    } else {
      // let us join the nodes of the track to a linestring ....

      Iterator<Point> trackIter = track.iterator();
      Coordinate[] coords = new Coordinate[track.size()];
      int i = 0;
      while (trackIter.hasNext()) {
        Point p = trackIter.next();
        coords[i] = p.getCoordinate();
        i++;
      }
      if (i > 1) {
        LineString l = fact.createLineString(coords);

        Criteria testCriteria = session.createCriteria(OSMEdge.class);
        if (bufferSize > 0.) { // ... build a buffer ...
          Geometry buffer = l.buffer(bufferSize);
          //				testCriteria.add(SpatialRestrictions.within("geometry", buffer));
          testCriteria.add(SpatialRestrictions.intersects("geometry", buffer));
        }
        @SuppressWarnings("unchecked")
        List<OSMEdge> result = testCriteria.list();

        logger.info("Spatial query selected " + result.size() + " edges");

        Iterator<OSMEdge> iter = result.iterator();
        i = 0;
        while (iter.hasNext()) {
          i++;
          OSMEdge o = iter.next();
          LineString g = o.getGeometry();
          addLineString(g, o.getId(), o.getEnvtype(), o.getCyktype(), o.getGroenm());
          // let us add this thing to the spatial index
          addOSMEdgeToSpatialIndex(o);
        }

        // if required, dump the graph to a shapefile:
        if (dumpFile != "") {
          try {
            this.dumpBuffer(result, dumpFile);
            logger.info("buffer dumped");
          } catch (Exception e) {
            logger.error("error dumping buffer: " + e);
          }
        }
      } else {
        logger.error("error: track has <2 coordinates");
      }
    }

    session.disconnect();
  }