Java tutorial
/* (c) 2014 - 2016 Open Source Geospatial Foundation - all rights reserved * (c) 2001 - 2013 OpenPlans * This code is licensed under the GPL 2.0 license, available at the root * application directory. */ package org.geoserver.wcs2_0; import static org.junit.Assert.assertArrayEquals; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNotNull; import java.io.File; import java.util.Map; import javax.xml.namespace.QName; import org.apache.commons.io.FileUtils; import org.geoserver.catalog.DimensionPresentation; import org.geoserver.catalog.ResourceInfo; import org.geoserver.data.test.CiteTestData; import org.geoserver.data.test.SystemTestData; import org.geoserver.ows.util.CaseInsensitiveMap; import org.geoserver.ows.util.KvpUtils; import org.geoserver.wcs2_0.kvp.WCS20GetCoverageRequestReader; import org.junit.Test; import org.springframework.mock.web.MockHttpServletResponse; import net.opengis.wcs20.GetCoverageType; import ucar.nc2.Dimension; import ucar.nc2.Variable; import ucar.nc2.dataset.NetcdfDataset; /** * Base support class for NetCDF wcs tests. * * @author Daniele Romagnoli, GeoSolutions * */ public class WCSNetCDFTest extends WCSNetCDFBaseTest { public static final double DELTA = 1e-6; public static QName POLYPHEMUS = new QName(CiteTestData.WCS_URI, "polyphemus", CiteTestData.WCS_PREFIX); public static QName NO2 = new QName(CiteTestData.WCS_URI, "NO2", CiteTestData.WCS_PREFIX); public static QName TEMPERATURE_SURFACE_NETCDF = new QName(CiteTestData.WCS_URI, "Temperature_surface_NetCDF", CiteTestData.WCS_PREFIX); public static QName TEMPERATURE_SURFACE_GRIB = new QName(CiteTestData.WCS_URI, "Temperature_surface", CiteTestData.WCS_PREFIX); public static QName SNOW_DEPTH_GRIB = new QName(CiteTestData.WCS_URI, "Snow_depth_water_equivalent_surface", CiteTestData.WCS_PREFIX); /** * Only setup coverages */ protected void setUpTestData(SystemTestData testData) throws Exception { super.setUpTestData(testData); } @SuppressWarnings("unchecked") protected GetCoverageType parse(String url) throws Exception { Map<String, Object> rawKvp = new CaseInsensitiveMap(KvpUtils.parseQueryString(url)); Map<String, Object> kvp = new CaseInsensitiveMap(parseKvp(rawKvp)); WCS20GetCoverageRequestReader reader = new WCS20GetCoverageRequestReader(); GetCoverageType gc = (GetCoverageType) reader.createRequest(); return (GetCoverageType) reader.read(gc, kvp, rawKvp); } @Override protected void onSetUp(SystemTestData testData) throws Exception { super.onSetUp(testData); testData.addRasterLayer(POLYPHEMUS, "pol.zip", null, null, this.getClass(), getCatalog()); setupRasterDimension(getLayerId(NO2), ResourceInfo.TIME, DimensionPresentation.LIST, null); setupRasterDimension(getLayerId(NO2), ResourceInfo.ELEVATION, DimensionPresentation.LIST, null); testData.addRasterLayer(TEMPERATURE_SURFACE_NETCDF, "rotated-pole.nc", null, null, this.getClass(), getCatalog()); testData.addRasterLayer(TEMPERATURE_SURFACE_GRIB, "rap-native.grib2", null, null, this.getClass(), getCatalog()); testData.addRasterLayer(SNOW_DEPTH_GRIB, "cosmo-eu.grib2", null, null, this.getClass(), getCatalog()); } /** * This test checks if an exception is not thrown when is requested an image with a total size lower than the maximum * geoserver output size. * */ @Test public void testOutputMemoryNotExceeded() throws Exception { // Setting of the output limit to 40 Kb setOutputLimit(40); // http response from the request inside the string MockHttpServletResponse response = getAsServletResponse("ows?request=GetCoverage&service=WCS&version=2.0.1" + "&coverageId=wcs__NO2&format=application/x-netcdf&subset=http://www.opengis.net/def/axis/OGC/0/elevation(450)"); // The status code should be correct assertEquals(200, response.getStatus()); // The output format should be netcdf assertEquals("application/x-netcdf", response.getContentType()); // Reset output limit setOutputLimit(-1); } /** * This test checks if an exception is thrown when is requested an image with a total size greater than the maximum * geoserver output memory allowed. * */ @Test public void testOutputMemoryExceeded() throws Exception { // Setting of the output limit to 40 Kb setOutputLimit(40); // http response from the request inside the string MockHttpServletResponse response = getAsServletResponse("ows?request=GetCoverage&service=WCS&version=2.0.1" + "&coverageId=wcs__NO2&format=application/x-netcdf"); // The output format should be xml because an exception must be thrown assertEquals("application/xml", response.getContentType()); // Reset output limit setOutputLimit(-1); } /** * This test checks if an exception is not thrown when is requested an image with a total size lower than the maximum * geoserver input size. * */ @Test public void testInputMemoryCorrect() throws Exception { // Setting of the input limit to 40 Kb setInputLimit(40); // http response from the request inside the string MockHttpServletResponse response = getAsServletResponse("ows?request=GetCoverage&service=WCS&version=2.0.1" + "&coverageId=wcs__NO2&format=application/x-netcdf&subset=http://www.opengis.net/def/axis/OGC/0/elevation(450)"); // The status code should be correct assertEquals(200, response.getStatus()); // The output format should be netcdf assertEquals("application/x-netcdf", response.getContentType()); // Reset input limit setInputLimit(-1); } /** * This test checks if an exception is thrown when is requested an image with a total size greater than the maximum * geoserver input memory allowed. * */ @Test public void testInputMemoryExceeded() throws Exception { // Setting of the input limit to 40 Kb setInputLimit(40); // http response from the request inside the string MockHttpServletResponse response = getAsServletResponse("ows?request=GetCoverage&service=WCS&version=2.0.1" + "&coverageId=wcs__NO2&format=application/x-netcdf"); // The output format should be xml because an exception must be thrown assertEquals("application/xml", response.getContentType()); // Reset input limit setInputLimit(-1); } /** * Test NetCDF output from a NetCDF file with a rotated pole projection. */ @Test public void testNetcdfRotatedPole() throws Exception { MockHttpServletResponse response = getAsServletResponse("ows?request=GetCoverage&service=WCS&version=2.0.1" + "&coverageid=wcs__Temperature_surface_NetCDF&format=application/x-netcdf"); assertEquals(200, response.getStatus()); assertEquals("application/x-netcdf", response.getContentType()); byte[] responseBytes = getBinary(response); File file = File.createTempFile("netcdf-rotated-pole-", "-wcs__Temperature_surface_NetCDF.nc", new File("./target")); FileUtils.writeByteArrayToFile(file, responseBytes); try (NetcdfDataset dataset = NetcdfDataset.openDataset(file.getAbsolutePath())) { assertNotNull(dataset); // check dimensions Dimension rlonDim = dataset.findDimension("rlon"); assertNotNull(rlonDim); assertEquals(7, rlonDim.getLength()); Dimension rlatDim = dataset.findDimension("rlat"); assertNotNull(rlatDim); assertEquals(5, rlatDim.getLength()); // check coordinate variables Variable rlonVar = dataset.findVariable("rlon"); assertNotNull(rlonVar); assertEquals(1, rlonVar.getDimensions().size()); assertEquals(rlonDim, rlonVar.getDimensions().get(0)); assertEquals("grid_longitude", rlonVar.findAttribute("long_name").getStringValue()); assertEquals("grid_longitude", rlonVar.findAttribute("standard_name").getStringValue()); assertEquals("degrees", rlonVar.findAttribute("units").getStringValue()); assertArrayEquals(new float[] { -30, -20, -10, 0, 10, 20, 30 }, (float[]) rlonVar.read().copyTo1DJavaArray(), (float) DELTA); Variable rlatVar = dataset.findVariable("rlat"); assertNotNull(rlatVar); assertEquals(1, rlatVar.getDimensions().size()); assertEquals(rlatDim, rlatVar.getDimensions().get(0)); assertEquals("grid_latitude", rlatVar.findAttribute("long_name").getStringValue()); assertEquals("grid_latitude", rlatVar.findAttribute("standard_name").getStringValue()); assertEquals("degrees", rlatVar.findAttribute("units").getStringValue()); assertArrayEquals(new float[] { -20, -10, 0, 10, 20 }, (float[]) rlatVar.read().copyTo1DJavaArray(), (float) DELTA); // check projection variable Variable projVar = dataset.findVariable("rotated_latitude_longitude"); assertNotNull(projVar); assertEquals("rotated_latitude_longitude", projVar.findAttribute("grid_mapping_name").getStringValue()); assertEquals(74.0, projVar.findAttribute("grid_north_pole_longitude").getNumericValue().doubleValue(), DELTA); assertEquals(36.0, projVar.findAttribute("grid_north_pole_latitude").getNumericValue().doubleValue(), DELTA); // check data variable Variable tempVar = dataset.findVariable("Temperature_surface_NetCDF"); assertNotNull(tempVar); assertEquals("rotated_latitude_longitude", tempVar.findAttribute("grid_mapping").getStringValue()); assertEquals("K", tempVar.findAttribute("units").getStringValue()); assertEquals(2, tempVar.getDimensions().size()); assertEquals(rlatDim, tempVar.getDimensions().get(0)); assertEquals(rlonDim, tempVar.getDimensions().get(1)); assertArrayEquals(new float[] { 300, 299, 298, 297, 296, 295, 294, 299, 300, 299, 298, 297, 296, 295, 298, 299, 300, 299, 298, 297, 296, 297, 298, 299, 300, 299, 298, 297, 296, 297, 298, 299, 300, 299, 298 }, (float[]) tempVar.read().copyTo1DJavaArray(), (float) DELTA); } finally { FileUtils.deleteQuietly(file); } } /** * Test NetCDF output from an RAP native GRIB2 file with a GDS template 32769 rotated pole projection. */ @Test public void testRapNativeGribRotatedPole() throws Exception { MockHttpServletResponse response = getAsServletResponse("ows?request=GetCoverage&service=WCS&version=2.0.1" + "&coverageid=wcs__Temperature_surface&format=application/x-netcdf"); assertEquals(200, response.getStatus()); assertEquals("application/x-netcdf", response.getContentType()); byte[] responseBytes = getBinary(response); File file = File.createTempFile("rap-native-grib-rotated-pole-", "-wcs__Temperature_surface.nc", new File("./target")); FileUtils.writeByteArrayToFile(file, responseBytes); try (NetcdfDataset dataset = NetcdfDataset.openDataset(file.getAbsolutePath())) { assertNotNull(dataset); // check dimensions Dimension rlonDim = dataset.findDimension("rlon"); assertNotNull(rlonDim); assertEquals(7, rlonDim.getLength()); Dimension rlatDim = dataset.findDimension("rlat"); assertNotNull(rlatDim); assertEquals(5, rlatDim.getLength()); // check coordinate variables Variable rlonVar = dataset.findVariable("rlon"); assertNotNull(rlonVar); assertEquals(1, rlonVar.getDimensions().size()); assertEquals(rlonDim, rlonVar.getDimensions().get(0)); assertEquals("grid_longitude", rlonVar.findAttribute("long_name").getStringValue()); assertEquals("grid_longitude", rlonVar.findAttribute("standard_name").getStringValue()); assertEquals("degrees", rlonVar.findAttribute("units").getStringValue()); assertArrayEquals(new float[] { -30, -20, -10, 0, 10, 20, 30 }, (float[]) rlonVar.read().copyTo1DJavaArray(), (float) DELTA); Variable rlatVar = dataset.findVariable("rlat"); assertNotNull(rlatVar); assertEquals(1, rlatVar.getDimensions().size()); assertEquals(rlatDim, rlatVar.getDimensions().get(0)); assertEquals("grid_latitude", rlatVar.findAttribute("long_name").getStringValue()); assertEquals("grid_latitude", rlatVar.findAttribute("standard_name").getStringValue()); assertEquals("degrees", rlatVar.findAttribute("units").getStringValue()); assertArrayEquals(new float[] { -20, -10, 0, 10, 20 }, (float[]) rlatVar.read().copyTo1DJavaArray(), (float) DELTA); // check projection variable Variable projVar = dataset.findVariable("rotated_latitude_longitude"); assertNotNull(projVar); assertEquals("rotated_latitude_longitude", projVar.findAttribute("grid_mapping_name").getStringValue()); assertEquals(74.0, projVar.findAttribute("grid_north_pole_longitude").getNumericValue().doubleValue(), DELTA); assertEquals(36.0, projVar.findAttribute("grid_north_pole_latitude").getNumericValue().doubleValue(), DELTA); // check data variable Variable dataVar = dataset.findVariable("Temperature_surface"); assertNotNull(dataVar); assertEquals("rotated_latitude_longitude", dataVar.findAttribute("grid_mapping").getStringValue()); assertEquals("K", dataVar.findAttribute("units").getStringValue()); assertEquals(2, dataVar.getDimensions().size()); assertEquals(rlatDim, dataVar.getDimensions().get(0)); assertEquals(rlonDim, dataVar.getDimensions().get(1)); assertArrayEquals(new float[] { 300, 299, 298, 297, 296, 295, 294, 299, 300, 299, 298, 297, 296, 295, 298, 299, 300, 299, 298, 297, 296, 297, 298, 299, 300, 299, 298, 297, 296, 297, 298, 299, 300, 299, 298 }, (float[]) dataVar.read().copyTo1DJavaArray(), (float) DELTA); } finally { FileUtils.deleteQuietly(file); } } /** * Test NetCDF output from a COSMO EU GRIB2 file with a GDS template 1 rotated pole projection. */ @Test public void testCosmoEuGribRotatedPole() throws Exception { MockHttpServletResponse response = getAsServletResponse("ows?request=GetCoverage&service=WCS&version=2.0.1" + "&coverageid=wcs__Snow_depth_water_equivalent_surface&format=application/x-netcdf"); assertEquals(200, response.getStatus()); assertEquals("application/x-netcdf", response.getContentType()); byte[] responseBytes = getBinary(response); File file = File.createTempFile("cosmo-eu-grib-rotated-pole-", "-wcs__Snow_depth_water_equivalent_surface.nc", new File("./target")); FileUtils.writeByteArrayToFile(file, responseBytes); try (NetcdfDataset dataset = NetcdfDataset.openDataset(file.getAbsolutePath())) { assertNotNull(dataset); // check dimensions Dimension rlonDim = dataset.findDimension("rlon"); assertNotNull(rlonDim); assertEquals(5, rlonDim.getLength()); Dimension rlatDim = dataset.findDimension("rlat"); assertNotNull(rlatDim); assertEquals(5, rlatDim.getLength()); // check coordinate variables Variable rlonVar = dataset.findVariable("rlon"); assertNotNull(rlonVar); assertEquals(1, rlonVar.getDimensions().size()); assertEquals(rlonDim, rlonVar.getDimensions().get(0)); assertEquals("grid_longitude", rlonVar.findAttribute("long_name").getStringValue()); assertEquals("grid_longitude", rlonVar.findAttribute("standard_name").getStringValue()); assertEquals("degrees", rlonVar.findAttribute("units").getStringValue()); assertArrayEquals(new float[] { -18, -8, 2, 12, 22 }, (float[]) rlonVar.read().copyTo1DJavaArray(), (float) DELTA); Variable rlatVar = dataset.findVariable("rlat"); assertNotNull(rlatVar); assertEquals(1, rlatVar.getDimensions().size()); assertEquals(rlatDim, rlatVar.getDimensions().get(0)); assertEquals("grid_latitude", rlatVar.findAttribute("long_name").getStringValue()); assertEquals("grid_latitude", rlatVar.findAttribute("standard_name").getStringValue()); assertEquals("degrees", rlatVar.findAttribute("units").getStringValue()); assertArrayEquals(new float[] { -20, -10, 0, 10, 20 }, (float[]) rlatVar.read().copyTo1DJavaArray(), (float) DELTA); // check projection variable Variable projVar = dataset.findVariable("rotated_latitude_longitude"); assertNotNull(projVar); assertEquals("rotated_latitude_longitude", projVar.findAttribute("grid_mapping_name").getStringValue()); assertEquals(-170.0, projVar.findAttribute("grid_north_pole_longitude").getNumericValue().doubleValue(), DELTA); assertEquals(40.0, projVar.findAttribute("grid_north_pole_latitude").getNumericValue().doubleValue(), DELTA); // check data variable Variable dataVar = dataset.findVariable("Snow_depth_water_equivalent_surface"); assertNotNull(dataVar); assertEquals("rotated_latitude_longitude", dataVar.findAttribute("grid_mapping").getStringValue()); assertEquals(2, dataVar.getDimensions().size()); assertEquals(rlatDim, dataVar.getDimensions().get(0)); assertEquals(rlonDim, dataVar.getDimensions().get(1)); assertArrayEquals( new float[] { 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124 }, (float[]) dataVar.read().copyTo1DJavaArray(), (float) DELTA); } finally { FileUtils.deleteQuietly(file); } } }