org.jgrasstools.gears.modules.r.mosaic.OmsMosaic 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 inFiles;
@Description(OMSMOSAIC_P_INTERPOLATION_DESCRIPTION)
@UI("combo:" + NEAREST_NEIGHTBOUR + "," + BILINEAR + "," + BICUBIC)
@In
public String pInterpolation = NEAREST_NEIGHTBOUR;
@Description(OMSMOSAIC_OUT_RASTER_DESCRIPTION)
@Out
public GridCoverage2D outRaster = null;
private CoordinateReferenceSystem crs;
@Execute
public void process() throws Exception {
if (!concatOr(outRaster == null, doReset)) {
return;
}
if (inFiles == null) {
throw new ModelsIllegalargumentException("No input data have been provided.", this, pm);
}
if (inFiles != null && inFiles.size() < 2) {
throw new ModelsIllegalargumentException("The patching module needs at least two maps to be patched.", this, pm);
}
GridGeometry2D referenceGridGeometry = null;
double n = Double.MIN_VALUE;
double s = Double.MAX_VALUE;
double e = Double.MIN_VALUE;
double w = Double.MAX_VALUE;
int np = Integer.MIN_VALUE;
int sp = Integer.MAX_VALUE;
int ep = Integer.MIN_VALUE;
int wp = Integer.MAX_VALUE;
pm.beginTask("Calculating final bounds...", inFiles.size());
for( File coverageFile : inFiles ) {
GridCoverage2D coverage = OmsRasterReader.readRaster(coverageFile.getAbsolutePath());
// pm.message(MessageFormat.format("Reading map: {0} with crs: {1}",
// coverageFile.getAbsolutePath(),
// CrsUtilities.getCodeFromCrs(crs)));
if (referenceGridGeometry == null) {
// take the first as reference
crs = coverage.getCoordinateReferenceSystem();
referenceGridGeometry = coverage.getGridGeometry();
pm.message("Using crs: " + CrsUtilities.getCodeFromCrs(crs));
}
Envelope2D worldEnv = coverage.getEnvelope2D();
GridEnvelope2D pixelEnv = referenceGridGeometry.worldToGrid(worldEnv);
int minPX = (int) pixelEnv.getMinX();
int minPY = (int) pixelEnv.getMinY();
int maxPX = (int) pixelEnv.getMaxX();
int maxPY = (int) pixelEnv.getMaxY();
if (minPX < wp)
wp = minPX;
if (minPY < sp)
sp = minPY;
if (maxPX > ep)
ep = maxPX;
if (maxPY > np)
np = maxPY;
double minWX = worldEnv.getMinX();
double minWY = worldEnv.getMinY();
double maxWX = worldEnv.getMaxX();
double maxWY = worldEnv.getMaxY();
if (minWX < w)
w = minWX;
if (minWY < s)
s = minWY;
if (maxWX > e)
e = maxWX;
if (maxWY > n)
n = maxWY;
pm.worked(1);
}
pm.done();
int endWidth = ep - wp;
int endHeight = np - sp;
WritableRaster outputWR = CoverageUtilities.createDoubleWritableRaster(endWidth, endHeight, null, null,
JGTConstants.doubleNovalue);
WritableRandomIter outputIter = RandomIterFactory.createWritable(outputWR, null);
int offestX = Math.abs(wp);
int offestY = Math.abs(sp);
int index = 1;
for( File coverageFile : inFiles ) {
GridCoverage2D coverage = OmsRasterReader.readRaster(coverageFile.getAbsolutePath());
RenderedImage renderedImage = coverage.getRenderedImage();
RandomIter randomIter = RandomIterFactory.create(renderedImage, null);
Envelope2D env = coverage.getEnvelope2D();
GridEnvelope2D repEnv = referenceGridGeometry.worldToGrid(env);
GridGeometry2D tmpGG = coverage.getGridGeometry();
GridEnvelope2D tmpEnv = tmpGG.worldToGrid(env);
int startX = (int) (repEnv.getMinX() + offestX);
int startY = (int) (repEnv.getMinY() + offestY);
double tmpW = tmpEnv.getWidth();
pm.beginTask("Patch map " + index++, (int) tmpW); //$NON-NLS-1$
for( int x = 0; x < tmpW; x++ ) {
for( int y = 0; y < tmpEnv.getHeight(); y++ ) {
double value = randomIter.getSampleDouble(x, y, 0);
outputIter.setSample(x + startX, y + startY, 0, value);
}
pm.worked(1);
}
pm.done();
randomIter.done();
}
HashMap envelopeParams = new HashMap();
envelopeParams.put(NORTH, n);
envelopeParams.put(SOUTH, s);
envelopeParams.put(WEST, w);
envelopeParams.put(EAST, e);
outRaster = CoverageUtilities.buildCoverage("patch", outputWR, envelopeParams, crs); //$NON-NLS-1$
}
}