org.jgrasstools.gears.modules.v.rastercattofeatureattribute.OmsRasterCatToFeatureAttribute Maven / Gradle / Ivy
The newest version!
/*
* This file is part of JGrasstools (http://www.jgrasstools.org)
* (C) HydroloGIS - www.hydrologis.com
*
* JGrasstools is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see featureIterator = inVector.features();
int all = inVector.size();
pm.beginTask("Extracting raster information...", all);
while( featureIterator.hasNext() ) {
SimpleFeature feature = featureIterator.next();
Geometry geometry = (Geometry) feature.getDefaultGeometry();
double value = -1;
Coordinate c;
Coordinate[] coordinates = geometry.getCoordinates();
if (getGeometryType(geometry) == GeometryType.POINT || getGeometryType(geometry) == GeometryType.MULTIPOINT) {
c = coordinates[0];
if (!inCoverageEnvelope.contains(c.x, c.y)) {
continue;
}
value = getRasterValue(c);
if (fExt == null)
fExt = new FeatureExtender(featureType, //
new String[]{fNew}, //
new Class< ? >[]{Double.class});
SimpleFeature extendedFeature = fExt.extendFeature(feature, new Object[]{value});
((DefaultFeatureCollection) outVector).add(extendedFeature);
} else if (getGeometryType(geometry) == GeometryType.LINE || getGeometryType(geometry) == GeometryType.MULTILINE) {
if (pPos.trim().equalsIgnoreCase(START)) {
c = coordinates[0];
} else if (pPos.trim().equalsIgnoreCase(END)) {
c = coordinates[coordinates.length - 1];
} else {// (pPos.trim().equalsIgnoreCase(MIDDLE)) {
c = coordinates[coordinates.length / 2];
}
if (!inCoverageEnvelope.contains(c.x, c.y)) {
continue;
}
value = getRasterValue(c);
if (fExt == null)
fExt = new FeatureExtender(featureType, //
new String[]{fNew}, //
new Class< ? >[]{Double.class});
SimpleFeature extendedFeature = fExt.extendFeature(feature, new Object[]{value});
((DefaultFeatureCollection) outVector).add(extendedFeature);
} else if (getGeometryType(geometry) == GeometryType.POLYGON
|| getGeometryType(geometry) == GeometryType.MULTIPOLYGON) {
if (fExt == null) {
String max = fNew + "_max";
String min = fNew + "_min";
String sum = fNew + "_sum";
String avg = fNew + "_avg";
fExt = new FeatureExtender(featureType, //
new String[]{min, max, sum, avg}, //
new Class< ? >[]{Double.class, Double.class, Double.class, Double.class});
}
SimpleFeature singleFeature = FeatureUtilities.toDummyFeature(geometry, crs);
SimpleFeatureCollection newCollection = new DefaultFeatureCollection();
((DefaultFeatureCollection) newCollection).add(singleFeature);
OmsScanLineRasterizer raster = new OmsScanLineRasterizer();
raster.inVector = newCollection;
raster.inRaster = inRaster;
// raster.pCols = regionMap.getCols();
// raster.pRows = regionMap.getRows();
// raster.pNorth = regionMap.getNorth();
// raster.pSouth = regionMap.getSouth();
// raster.pEast = regionMap.getEast();
// raster.pWest = regionMap.getWest();
raster.pValue = 1.0;
raster.process();
GridCoverage2D rasterizedVector = raster.outRaster;
double[] minMaxAvgSum = OmsRasterSummary.getMinMaxAvgSum(rasterizedVector);
// Point centroid = geometry.getCentroid();
// if (geometry.contains(centroid)) {
// c = centroid.getCoordinate();
// } else {
// c = coordinates[0];
// }
SimpleFeature extendedFeature = fExt.extendFeature(feature, new Object[]{minMaxAvgSum[0], minMaxAvgSum[1],
minMaxAvgSum[2], minMaxAvgSum[3]});
((DefaultFeatureCollection) outVector).add(extendedFeature);
} else {
throw new ModelsIllegalargumentException("The Geometry type is not supported.", this, pm);
}
pm.worked(1);
}
featureIterator.close();
pm.done();
}
private double getRasterValue( Coordinate c ) throws TransformException {
double value;
GridCoordinates2D gridCoord = gridGeometry.worldToGrid(new DirectPosition2D(c.x, c.y));
value = inIter.getSampleDouble(gridCoord.x, gridCoord.y, 0);
// TODO make this better
if (isNovalue(value) || value >= Float.MAX_VALUE || value <= -Float.MAX_VALUE) {
value = -9999.0;
}
return value;
}
}