package org.geoserver.wps.gs;
import java.io.IOException;
import org.geoserver.catalog.CoverageInfo;
import org.geotools.coverage.grid.GridCoverage2D;
import org.geotools.coverage.processing.Operations;
import org.geotools.data.simple.SimpleFeatureCollection;
import org.geotools.data.simple.SimpleFeatureIterator;
import org.geotools.factory.GeoTools;
import org.geotools.geometry.GeneralEnvelope;
import org.geotools.util.NullProgressListener;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.geometry.Envelope;
/**
* Test class for the contour process.
*
* @author Simone Giannecchini, GeoSolutions SAS
*
*/
public class ContourProcessTest extends BaseRasterToVectorTest {
/**
* Test basic capabilities for the contour process. It works on the DEM tiff
* and produces a shapefile. Nothing more nothing less.
*
* @throws Exception
*/
public void testProcessStandaloneBasicValues() throws Exception {
GridCoverage2D gc = extractCoverageSubset();
// extract just two isolines
final double levels[] = new double[2];
levels[0] = 1500;
levels[1] = 1700;
final ContourProcess process = new ContourProcess();
final SimpleFeatureCollection fc = process.execute(gc, 0, levels, null,
false, false, null, new NullProgressListener());
assertNotNull(fc);
assertTrue(fc.size() > 0);
SimpleFeatureIterator fi = fc.features();
while (fi.hasNext()) {
SimpleFeature sf = fi.next();
Double value = (Double) sf.getAttribute("value");
assertTrue(value == 1500.0 || value == 1700.0);
}
fi.close();
}
private GridCoverage2D extractCoverageSubset() throws IOException {
// get the coverage
CoverageInfo dem = getCatalog().getCoverageByName(DEM.getLocalPart());
GridCoverage2D gc = (GridCoverage2D) dem.getGridCoverage(null,
GeoTools.getDefaultHints());
// extract only a small part of it
Envelope fullEnvelope = gc.getEnvelope();
GeneralEnvelope subset = new GeneralEnvelope(
fullEnvelope.getCoordinateReferenceSystem());
double minX = fullEnvelope.getMinimum(0);
double minY = fullEnvelope.getMinimum(1);
double offsetX = fullEnvelope.getSpan(0) / 5;
double offsetY = fullEnvelope.getSpan(1) / 5;
subset.setEnvelope(minX + offsetX, minY + offsetY, minX + offsetX * 2,
minY + offsetY * 2);
gc = (GridCoverage2D) new Operations(null).crop(gc, subset);
return gc;
}
/**
* Test basic capabilities for the contour process. It works on the DEM tiff
* and produces a shapefile. Nothing more nothing less.
*
* @throws Exception
*/
public void testProcessStandaloneBasicInterval() throws Exception {
final GridCoverage2D gc = extractCoverageSubset();
final double step = 100;
final ContourProcess process = new ContourProcess();
final SimpleFeatureCollection fc = process.execute(gc, 0, null,
Double.valueOf(step), false, false, null,
new NullProgressListener());
assertNotNull(fc);
assertTrue(fc.size() > 0);
SimpleFeatureIterator fi = fc.features();
while (fi.hasNext()) {
SimpleFeature sf = fi.next();
Double value = (Double) sf.getAttribute("value");
assertTrue(value > 0);
}
fi.close();
}
}