Search in sources :

Example 1 with DefaultFeatureMapper

use of org.geotoolkit.storage.memory.mapping.DefaultFeatureMapper in project geotoolkit by Geomatys.

the class FeatureStoreUtilities method decomposeByGeometryType.

/**
 * Split the collection by geometry types.
 * Multiple feature store can only support a limited number of geometry types.
 * This method will split the content of the given collection in collections with a
 * simple geometry type.
 *
 * Collection datas are not copied, result collections are filtered collections
 *
 * @param col
 * @param adaptType : ry to map types even if they do not match exactly.
 * list of adapt operations :
 * LineString -> MultiLineString
 * Polygon -> MultiPolygon
 * Point -> MultiPoint
 * @param geomClasses
 * @return splitted collections
 * @throws org.apache.sis.storage.DataStoreException
 */
public static FeatureCollection[] decomposeByGeometryType(FeatureCollection col, GenericName geomPropName, boolean adaptType, Class... geomClasses) throws DataStoreException {
    final FilterFactory FF = FilterUtilities.FF;
    final FeatureType baseType = col.getType();
    final GenericName name = baseType.getName();
    final PropertyType geomDesc = baseType.getProperty(geomPropName.toString());
    boolean setDefaultGeometryRole = false;
    try {
        IdentifiedType defaultGeometry = baseType.getProperty(AttributeConvention.GEOMETRY);
        setDefaultGeometryRole = defaultGeometry.equals(geomDesc);
        while (setDefaultGeometryRole == false && defaultGeometry instanceof Operation) {
            defaultGeometry = ((Operation) defaultGeometry).getResult();
            setDefaultGeometryRole = defaultGeometry.equals(geomDesc);
        }
    } catch (PropertyNotFoundException e) {
        LOGGER.log(Level.FINEST, "No SIS convention found in given data type", e);
    }
    final CoordinateReferenceSystem crs = FeatureExt.getCRS(geomDesc);
    final List<Class> lstClasses = Arrays.asList(geomClasses);
    final FeatureCollection[] cols = new FeatureCollection[geomClasses.length];
    for (int i = 0; i < geomClasses.length; i++) {
        final Class geomClass = geomClasses[i];
        Filter filter = FF.equal(FF.function("geometryType", FF.property(geomPropName.tip().toString())), FF.literal(geomClass.getSimpleName()));
        // check if we need to map another type
        if (adaptType) {
            if (geomClass == MultiPolygon.class && !lstClasses.contains(Polygon.class)) {
                filter = FF.or(filter, FF.equal(FF.function("geometryType", FF.property(geomPropName.tip().toString())), FF.literal(Polygon.class.getSimpleName())));
            } else if (geomClass == MultiLineString.class && !lstClasses.contains(LineString.class)) {
                filter = FF.or(filter, FF.equal(FF.function("geometryType", FF.property(geomPropName.tip().toString())), FF.literal(LineString.class.getSimpleName())));
            } else if (geomClass == MultiPoint.class && !lstClasses.contains(Point.class)) {
                filter = FF.or(filter, FF.equal(FF.function("geometryType", FF.property(geomPropName.tip().toString())), FF.literal(Point.class.getSimpleName())));
            }
        }
        cols[i] = col.subset(Query.filtered(name.toString(), filter));
        // retype the collection
        final FeatureTypeBuilder ftb = new FeatureTypeBuilder(baseType);
        ftb.setName(NamesExt.create(NamesExt.getNamespace(name), name.tip().toString() + '_' + geomClass.getSimpleName()));
        PropertyTypeBuilder geometryBuilder = null;
        final Iterator<PropertyTypeBuilder> it = ftb.properties().iterator();
        while (geometryBuilder == null && it.hasNext()) {
            final PropertyTypeBuilder next = it.next();
            if (next.getName().equals(geomPropName)) {
                geometryBuilder = next;
                it.remove();
            }
        }
        final AttributeTypeBuilder geomAttr = ftb.addAttribute(geomClasses[i]).setCRS(crs);
        if (setDefaultGeometryRole) {
            geomAttr.addRole(AttributeRole.DEFAULT_GEOMETRY);
        }
        cols[i] = new GenericMappingFeatureCollection(cols[i], new DefaultFeatureMapper(baseType, ftb.build()));
    }
    return cols;
}
Also used : MultiPoint(org.locationtech.jts.geom.MultiPoint) FeatureType(org.opengis.feature.FeatureType) FeatureTypeBuilder(org.apache.sis.feature.builder.FeatureTypeBuilder) PropertyNotFoundException(org.opengis.feature.PropertyNotFoundException) PropertyType(org.opengis.feature.PropertyType) Operation(org.opengis.feature.Operation) FilterFactory(org.opengis.filter.FilterFactory) Point(org.locationtech.jts.geom.Point) MultiPoint(org.locationtech.jts.geom.MultiPoint) DefaultFeatureMapper(org.geotoolkit.storage.memory.mapping.DefaultFeatureMapper) PropertyTypeBuilder(org.apache.sis.feature.builder.PropertyTypeBuilder) AttributeTypeBuilder(org.apache.sis.feature.builder.AttributeTypeBuilder) GenericName(org.opengis.util.GenericName) GenericMappingFeatureCollection(org.geotoolkit.storage.memory.GenericMappingFeatureCollection) Filter(org.opengis.filter.Filter) MultiPolygon(org.locationtech.jts.geom.MultiPolygon) MultiLineString(org.locationtech.jts.geom.MultiLineString) LineString(org.locationtech.jts.geom.LineString) CoordinateReferenceSystem(org.opengis.referencing.crs.CoordinateReferenceSystem) IdentifiedType(org.opengis.feature.IdentifiedType) GenericMappingFeatureCollection(org.geotoolkit.storage.memory.GenericMappingFeatureCollection)

Aggregations

AttributeTypeBuilder (org.apache.sis.feature.builder.AttributeTypeBuilder)1 FeatureTypeBuilder (org.apache.sis.feature.builder.FeatureTypeBuilder)1 PropertyTypeBuilder (org.apache.sis.feature.builder.PropertyTypeBuilder)1 GenericMappingFeatureCollection (org.geotoolkit.storage.memory.GenericMappingFeatureCollection)1 DefaultFeatureMapper (org.geotoolkit.storage.memory.mapping.DefaultFeatureMapper)1 LineString (org.locationtech.jts.geom.LineString)1 MultiLineString (org.locationtech.jts.geom.MultiLineString)1 MultiPoint (org.locationtech.jts.geom.MultiPoint)1 MultiPolygon (org.locationtech.jts.geom.MultiPolygon)1 Point (org.locationtech.jts.geom.Point)1 FeatureType (org.opengis.feature.FeatureType)1 IdentifiedType (org.opengis.feature.IdentifiedType)1 Operation (org.opengis.feature.Operation)1 PropertyNotFoundException (org.opengis.feature.PropertyNotFoundException)1 PropertyType (org.opengis.feature.PropertyType)1 Filter (org.opengis.filter.Filter)1 FilterFactory (org.opengis.filter.FilterFactory)1 CoordinateReferenceSystem (org.opengis.referencing.crs.CoordinateReferenceSystem)1 GenericName (org.opengis.util.GenericName)1