/** Find the next feature */
    private void findNext() {
      if (nextFeature != null) {
        return;
      }

      while (nextFeature == null) {

        if (nextFC != null) {
          if (ite.hasNext()) {
            nextFeature = ite.next();
            continue;
          } else {
            nextFC = null;
            ite = null;
          }
        } else {
          if (firstPass) { // first pass iterate on the original FeatureCollection
            if (originalFI.hasNext()) {
              nextFC = modify2(originalFI.next(), firstPass, featureList);
              ite = nextFC.iterator();
            } else {
              firstPass = false;
            }
          } else {
            if (unionFI.hasNext()) { // second pass iterate on the union FeatureCollection
              nextFC = modify2(unionFI.next(), firstPass, featureList);
              ite = nextFC.iterator();
            } else {
              break;
            }
          }
        }
      }
    }
  @Test
  public void testRead() throws DataStoreException, NoSuchAuthorityCodeException, FactoryException {

    final ExtendedDataStore store = new ExtendedDataStore(dataStore);
    final Name name = DefaultName.valueOf("{http://www.geotoolkit.org/test}extsql");
    assertFalse(store.getNames().contains(name));

    // add a new query
    final Query query =
        QueryBuilder.language(
            JDBCDataStore.CUSTOM_SQL,
            "SELECT geometry as geo ,\"intProperty\" as it FROM custom",
            name);
    store.addQuery(query);
    assertTrue(store.getNames().contains(name));

    final FeatureType ft = store.getFeatureType(name);

    final FeatureCollection col =
        store.createSession(true).getFeatureCollection(QueryBuilder.all(name));
    assertEquals(ft, col.getFeatureType());
    assertEquals(3, col.size());

    final FeatureIterator ite = col.iterator();
    try {
      while (ite.hasNext()) {
        final Feature f = ite.next();
        assertEquals(f.getType(), ft);
      }
    } finally {
      ite.close();
    }
  }
예제 #3
0
  @Test
  public void testIntersection() throws ProcessException, NoSuchIdentifierException {

    // Inputs
    final FeatureCollection<?> featureList = buildFeatureList();
    final FeatureCollection<?> featureUnionList = buildFeatureUnionList();

    // Process
    ProcessDescriptor desc = ProcessFinder.getProcessDescriptor("vector", "union");

    ParameterValueGroup in = desc.getInputDescriptor().createValue();
    in.parameter("feature_in").setValue(featureList);
    in.parameter("feature_union").setValue(featureUnionList);
    in.parameter("input_geometry_name").setValue("geom1");
    org.geotoolkit.process.Process proc = desc.createProcess(in);

    // Features out
    final FeatureCollection<?> featureListOut =
        (FeatureCollection<?>) proc.call().parameter("feature_out").getValue();

    // Expected Features out
    final FeatureCollection<?> featureListResult = buildResultList();

    assertEquals(featureListOut.getFeatureType(), featureListResult.getFeatureType());
    assertEquals(featureListOut.getID(), featureListResult.getID());
    assertEquals(featureListOut.size(), featureListResult.size());
    assertTrue(featureListOut.containsAll(featureListResult));
  }
 public void testReadFeatures() throws Exception {
   FeatureCollection fc =
       dataStore.createSession(false).getFeatureCollection(QueryBuilder.all(nsname(PERSON)));
   assertEquals(2, fc.size());
   FeatureIterator<SimpleFeature> fr = fc.iterator();
   assertTrue(fr.hasNext());
   SimpleFeature f = fr.next();
   assertTrue(fr.hasNext());
   f = fr.next();
   assertFalse(fr.hasNext());
   fr.close();
 }
  public static FeatureCollection wrap(final FeatureCollection col, final Query query) {
    return new AbstractFeatureCollection("wrap", col.getSource()) {

      private FeatureType type = null;

      @Override
      public FeatureType getFeatureType() {
        if (type == null) {
          try (FeatureReader ite = iterator(null)) {
            type = ite.getFeatureType();
          }
        }
        return type;
      }

      @Override
      public FeatureReader iterator(Hints hints) throws FeatureStoreRuntimeException {
        final FeatureReader ite = (FeatureReader) col.iterator();
        try {
          return wrap(ite, query);
        } catch (DataStoreException ex) {
          throw new FeatureStoreRuntimeException(ex);
        }
      }

      @Override
      public void update(Filter filter, Map values) throws DataStoreException {
        throw new DataStoreException("Not supported.");
      }

      @Override
      public void remove(Filter filter) throws DataStoreException {
        throw new DataStoreException("Not supported.");
      }
    };
  }
  /**
   * Connect to the original FeatureConnection
   *
   * @param originalFC FeatureCollection
   * @param intersList FeatureCollection
   */
  public UnionFeatureCollection(
      final FeatureCollection<Feature> inputFC,
      final FeatureCollection<Feature> unionFC,
      final String inputGeomName,
      final String unionGeomName) {

    super(inputFC);
    this.unionFC = unionFC;

    /*
     * We ensure that inputGeomName and unionGeomName are not null and we get the CRS of
     * these geometry. This in order to genereate the new FeatureType based on two FeatureType
     * and with only one geometry (in this case inputGeomName)
     * If inputGeomName or unionGeomName is null, we use default geometry name and CRS.
     */
    CoordinateReferenceSystem geometryCRS;
    if (inputGeomName == null) {
      this.inputGeomName =
          inputFC.getFeatureType().getGeometryDescriptor().getName().getLocalPart();
      geometryCRS = inputFC.getFeatureType().getGeometryDescriptor().getCoordinateReferenceSystem();
    } else {
      this.inputGeomName = inputGeomName;
      final GeometryDescriptor buffDesc =
          (GeometryDescriptor) inputFC.getFeatureType().getDescriptor(inputGeomName);
      geometryCRS = buffDesc.getCoordinateReferenceSystem();
    }

    if (unionGeomName == null) {
      this.unionGeomName =
          unionFC.getFeatureType().getGeometryDescriptor().getName().getLocalPart();
    } else {
      this.unionGeomName = unionGeomName;
    }

    // Create the new FeatureType which concatenate two FeatureCollection FeatureType but with only
    // one geometry
    // (inputGeomName)
    this.newFeatureType =
        Union.mergeType(
            inputFC.getFeatureType(), unionFC.getFeatureType(), this.inputGeomName, geometryCRS);
  }
 /** {@inheritDoc } */
 @Override
 public FeatureIterator<Feature> iterator(final Hints hints) throws DataStoreRuntimeException {
   return new IntersectionFeatureIterator(
       getOriginalFeatureCollection().iterator(), unionFC.iterator());
 }
  /**
   * @param featureCollection
   * @param writer
   * @param fragment : true if we write in a stream, dont write start and end elements
   * @throws DataStoreException
   */
  public Element writeFeatureCollection(
      final FeatureCollection featureCollection, final boolean fragment, final boolean wfs)
      throws DataStoreException, ParserConfigurationException {

    DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
    // then we have to create document-loader:
    factory.setNamespaceAware(false);
    DocumentBuilder loader = factory.newDocumentBuilder();

    // creating a new DOM-document...
    Document document = loader.newDocument();

    // the XML header
    if (!fragment) {
      document.setXmlVersion("1.0");
      // writer.writeStartDocument("UTF-8", "1.0");
    }

    // the root Element
    final Element rootElement;
    if (wfs) {
      rootElement = document.createElementNS("http://www.opengis.net/wfs", "FeatureCollection");
      rootElement.setPrefix("wfs");
    } else {
      rootElement = document.createElementNS("http://www.opengis.net/gml", "FeatureCollection");
      rootElement.setPrefix("gml");
    }

    document.appendChild(rootElement);

    String collectionID = "";
    if (featureCollection.getID() != null) {
      collectionID = featureCollection.getID();
    }
    final Attr idAttribute = document.createAttributeNS(Namespaces.GML, "id");
    idAttribute.setValue(collectionID);
    idAttribute.setPrefix("gml");
    rootElement.setAttributeNodeNS(idAttribute);

    if (schemaLocation != null && !schemaLocation.equals("")) {
      rootElement.setAttributeNS(
          "http://www.w3.org/2001/XMLSchema-instance", "schemaLocation", schemaLocation);
    }

    /*FeatureType type = featureCollection.getFeatureType();
    if (type != null && type.getName() != null) {
        String namespace = type.getName().getNamespaceURI();
        if (namespace != null && !namespace.equals(Namespaces.GML)) {
            Prefix prefix    = getPrefix(namespace);
            writer.writeNamespace(prefix.prefix, namespace);
        }
    }*/
    /*
     * The boundedby part
     */
    final Element boundElement = writeBounds(featureCollection.getEnvelope(), document);
    if (boundElement != null) {
      rootElement.appendChild(boundElement);
    }

    // we write each feature member of the collection
    FeatureIterator iterator = featureCollection.iterator();
    try {
      while (iterator.hasNext()) {
        final Feature f = iterator.next();
        final Element memberElement = document.createElementNS(Namespaces.GML, "featureMember");
        memberElement.setPrefix("gml");
        memberElement.appendChild(writeFeature(f, document, true));
        rootElement.appendChild(memberElement);
      }

    } finally {
      // we close the stream
      iterator.close();
    }
    return rootElement;
  }
예제 #9
0
  private static FeatureCollection<?> buildResultList() {

    try {
      type = createSimpleResultType();
    } catch (NoSuchAuthorityCodeException ex) {
      Logger.getLogger(UnionTest.class.getName()).log(Level.SEVERE, null, ex);
    } catch (FactoryException ex) {
      Logger.getLogger(UnionTest.class.getName()).log(Level.SEVERE, null, ex);
    }

    final FeatureCollection<Feature> featureList = DataUtilities.collection("", type);

    Feature myFeature;
    LinearRing ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(4, 7),
              new Coordinate(6, 7),
              new Coordinate(6, 5),
              new Coordinate(4, 5),
              new Coordinate(4, 7)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature1");
    sfb.set("color", "red");
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    sfb.set("att", 20);
    myFeature = sfb.buildFeature("id-01-id-11");
    featureList.add(myFeature);

    LineString str =
        geometryFactory.createLineString(
            new Coordinate[] {new Coordinate(3, 5), new Coordinate(3, 6)});
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature1");
    sfb.set("color", "grey");
    sfb.set("geom1", str);
    sfb.set("att", 12);
    myFeature = sfb.buildFeature("id-01-id-15");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(3, 5),
              new Coordinate(3, 6),
              new Coordinate(3, 7),
              new Coordinate(4, 7),
              new Coordinate(4, 5),
              new Coordinate(3, 5)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature1");
    sfb.set("color", null);
    sfb.set("att", null);
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature = sfb.buildFeature("id-01");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(7, 5),
              new Coordinate(8, 5),
              new Coordinate(8, 4),
              new Coordinate(7, 4),
              new Coordinate(7, 5)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature3");
    sfb.set("color", "blue");
    sfb.set("att", 20);
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature = sfb.buildFeature("id-03-id-12");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(6, 4),
              new Coordinate(6, 5),
              new Coordinate(7, 5),
              new Coordinate(7, 4),
              new Coordinate(6, 4)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature3");
    sfb.set("color", "red");
    sfb.set("att", 20);
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature = sfb.buildFeature("id-03-id-11");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(6, 2),
              new Coordinate(6, 4),
              new Coordinate(8, 4),
              new Coordinate(8, 2),
              new Coordinate(6, 2)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature3");
    sfb.set("color", "grey");
    sfb.set("att", 10);
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature = sfb.buildFeature("id-03-id-13");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(7, 7),
              new Coordinate(8, 7),
              new Coordinate(8, 5),
              new Coordinate(7, 5),
              new Coordinate(7, 7)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature2");
    sfb.set("color", "blue");
    sfb.set("att", 20);
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature = sfb.buildFeature("id-02-id-12");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(6, 5),
              new Coordinate(6, 7),
              new Coordinate(7, 7),
              new Coordinate(7, 5),
              new Coordinate(6, 5)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature2");
    sfb.set("color", "red");
    sfb.set("att", 20);
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature = sfb.buildFeature("id-02-id-11");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(2, 3),
              new Coordinate(2, 4),
              new Coordinate(3, 4),
              new Coordinate(3, 3),
              new Coordinate(2, 3)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature4");
    sfb.set("color", null);
    sfb.set("att", null);
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature = sfb.buildFeature("id-04");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(7, 7),
              new Coordinate(7, 8),
              new Coordinate(9, 8),
              new Coordinate(9, 4),
              new Coordinate(8, 4),
              new Coordinate(8, 5),
              new Coordinate(8, 7),
              new Coordinate(7, 7)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature12");
    sfb.set("color", "blue");
    sfb.set("att", 20);
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature = sfb.buildFeature("id-12");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(8, 4),
              new Coordinate(9, 4),
              new Coordinate(9, 2),
              new Coordinate(8, 2),
              new Coordinate(8, 4)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature13");
    sfb.set("color", "grey");
    sfb.set("att", 10);
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature = sfb.buildFeature("id-13");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(4, 2),
              new Coordinate(4, 3),
              new Coordinate(5, 3),
              new Coordinate(5, 2),
              new Coordinate(4, 2)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature14");
    sfb.set("color", "grey");
    sfb.set("att", 12);
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature = sfb.buildFeature("id-14");
    featureList.add(myFeature);

    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(4, 4),
              new Coordinate(4, 5),
              new Coordinate(6, 5),
              new Coordinate(6, 4),
              new Coordinate(4, 4)
            });
    LinearRing ring2 =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(6, 7),
              new Coordinate(4, 7),
              new Coordinate(4, 8),
              new Coordinate(7, 8),
              new Coordinate(7, 7),
              new Coordinate(6, 7)
            });
    Polygon poly1 = geometryFactory.createPolygon(ring, null);
    Polygon poly2 = geometryFactory.createPolygon(ring2, null);
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature11");
    sfb.set("color", "red");
    sfb.set("att", 20);
    sfb.set("geom1", geometryFactory.createMultiPolygon(new Polygon[] {poly1, poly2}));
    myFeature = sfb.buildFeature("id-11");
    featureList.add(myFeature);

    Feature myFeature5;
    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(2.0, 5.0),
              new Coordinate(2.0, 6.0),
              new Coordinate(3.0, 6.0),
              new Coordinate(3.0, 5.0),
              new Coordinate(2.0, 5.0)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature15");
    sfb.set("color", "grey");
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    sfb.set("att", 12);
    myFeature5 = sfb.buildFeature("id-15");
    featureList.add(myFeature5);

    return featureList;
  }
예제 #10
0
  private static FeatureCollection<?> buildFeatureUnionList() {

    try {
      type = createSimpleType2();
    } catch (NoSuchAuthorityCodeException ex) {
      Logger.getLogger(UnionTest.class.getName()).log(Level.SEVERE, null, ex);
    } catch (FactoryException ex) {
      Logger.getLogger(UnionTest.class.getName()).log(Level.SEVERE, null, ex);
    }

    final FeatureCollection<Feature> featureList = DataUtilities.collection("", type);

    Feature myFeature1;
    LinearRing ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(4.0, 4.0),
              new Coordinate(4.0, 8.0),
              new Coordinate(7.0, 8.0),
              new Coordinate(7.0, 4.0),
              new Coordinate(4.0, 4.0)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature11");
    sfb.set("color", "red");
    sfb.set("geom3", geometryFactory.createPolygon(ring, null));
    sfb.set("att", 20);
    myFeature1 = sfb.buildFeature("id-11");
    featureList.add(myFeature1);

    Feature myFeature2;
    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(7.0, 4.0),
              new Coordinate(7.0, 8.0),
              new Coordinate(9.0, 8.0),
              new Coordinate(9.0, 4.0),
              new Coordinate(7.0, 4.0)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature12");
    sfb.set("color", "blue");
    sfb.set("geom3", geometryFactory.createPolygon(ring, null));
    sfb.set("att", 20);
    myFeature2 = sfb.buildFeature("id-12");
    featureList.add(myFeature2);

    Feature myFeature3;
    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(6.0, 2.0),
              new Coordinate(6.0, 4.0),
              new Coordinate(9.0, 4.0),
              new Coordinate(9.0, 2.0),
              new Coordinate(6.0, 2.0)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature13");
    sfb.set("color", "grey");
    sfb.set("geom3", geometryFactory.createPolygon(ring, null));
    sfb.set("att", 10);
    myFeature3 = sfb.buildFeature("id-13");
    featureList.add(myFeature3);

    Feature myFeature4;
    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(4.0, 2.0),
              new Coordinate(4.0, 3.0),
              new Coordinate(5.0, 3.0),
              new Coordinate(5.0, 2.0),
              new Coordinate(4.0, 2.0)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature14");
    sfb.set("color", "grey");
    sfb.set("geom3", geometryFactory.createPolygon(ring, null));
    sfb.set("att", 12);
    myFeature4 = sfb.buildFeature("id-14");
    featureList.add(myFeature4);

    Feature myFeature5;
    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(2.0, 5.0),
              new Coordinate(2.0, 6.0),
              new Coordinate(3.0, 6.0),
              new Coordinate(3.0, 5.0),
              new Coordinate(2.0, 5.0)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature15");
    sfb.set("color", "grey");
    sfb.set("geom3", geometryFactory.createPolygon(ring, null));
    sfb.set("att", 12);
    myFeature5 = sfb.buildFeature("id-15");
    featureList.add(myFeature5);

    return featureList;
  }
예제 #11
0
  private static FeatureCollection<?> buildFeatureList() {

    try {
      type = createSimpleType();
    } catch (NoSuchAuthorityCodeException ex) {
      Logger.getLogger(UnionTest.class.getName()).log(Level.SEVERE, null, ex);
    } catch (FactoryException ex) {
      Logger.getLogger(UnionTest.class.getName()).log(Level.SEVERE, null, ex);
    }

    final FeatureCollection<Feature> featureList = DataUtilities.collection("", type);

    Feature myFeature1;
    LinearRing ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(3.0, 5.0),
              new Coordinate(3.0, 7.0),
              new Coordinate(6.0, 7.0),
              new Coordinate(6.0, 5.0),
              new Coordinate(3.0, 5.0)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature1");
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature1 = sfb.buildFeature("id-01");
    featureList.add(myFeature1);

    Feature myFeature2;
    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(6.0, 5.0),
              new Coordinate(6.0, 7.0),
              new Coordinate(8.0, 7.0),
              new Coordinate(8.0, 5.0),
              new Coordinate(6.0, 5.0)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature2");
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature2 = sfb.buildFeature("id-02");
    featureList.add(myFeature2);

    Feature myFeature3;
    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(6.0, 2.0),
              new Coordinate(6.0, 5.0),
              new Coordinate(8.0, 5.0),
              new Coordinate(8.0, 2.0),
              new Coordinate(6.0, 2.0)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature3");
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    // sfb.set("geom2", line);
    myFeature3 = sfb.buildFeature("id-03");
    featureList.add(myFeature3);

    Feature myFeature4;
    ring =
        geometryFactory.createLinearRing(
            new Coordinate[] {
              new Coordinate(2.0, 3.0),
              new Coordinate(2.0, 4.0),
              new Coordinate(3.0, 4.0),
              new Coordinate(3.0, 3.0),
              new Coordinate(2.0, 3.0)
            });
    sfb = new SimpleFeatureBuilder(type);
    sfb.set("name", "feature4");
    sfb.set("geom1", geometryFactory.createPolygon(ring, null));
    myFeature4 = sfb.buildFeature("id-04");
    featureList.add(myFeature4);

    return featureList;
  }