/*
 *    GeoTools - The Open Source Java GIS Toolkit
 *    http://geotools.org
 *
 *    (C) 2002-2010, Open Source Geospatial Foundation (OSGeo)
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation;
 *    version 2.1 of the License.
 *
 *    This library 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
 *    Lesser General Public License for more details.
 *
 */
package org.geotools.gce.grassraster;

import java.awt.image.RenderedImage;
import java.io.File;
import java.io.IOException;
import java.net.URL;

import javax.media.jai.iterator.RectIter;
import javax.media.jai.iterator.RectIterFactory;

import junit.framework.TestCase;

import org.geotools.coverage.grid.GridCoverage2D;
import org.geotools.coverage.grid.GridEnvelope2D;
import org.geotools.coverage.grid.GridGeometry2D;
import org.geotools.coverage.grid.io.AbstractGridFormat;
import org.geotools.data.DataUtilities;
import org.geotools.gce.grassraster.format.GrassCoverageFormatFactory;
import org.geotools.resources.coverage.CoverageUtilities;
import org.opengis.coverage.grid.GridCoverageReader;
import org.opengis.coverage.grid.GridCoverageWriter;
import org.opengis.geometry.DirectPosition;
import org.opengis.geometry.Envelope;

/**
 * Test the grass raster reader abd writer.
 * 
 * @author Andrea Antonello (www.hydrologis.com)
 *
 * @source $URL: http://svn.osgeo.org/geotools/trunk/modules/plugin/grassraster/src/test/java/org/geotools/gce/grassraster/GrassRasterReaderWriterTest.java $
 */
public class GrassRasterReaderWriterTest extends TestCase {
    private double n = 5140020.0;
    private double s = 5139780.0;
    private double w = 1640650.0;
    private double e = 1640950.0;

    public static double[][] mapData = new double[][]{
            {800, 900, 1000, 1000, 1200, 1250, 1300, 1350, 1450, 1500},
            {600, 650, 750, 850, 860, 900, 1000, 1200, 1250, 1500},
            {500, 550, 700, 750, 800, 850, 900, 1000, 1100, 1500},
            {400, 410, 650, 700, 750, 800, 850, 800, 800, 1500},
            {450, 550, 430, 500, 600, 700, 800, 800, 800, 1500},
            {500, 600, 700, 750, 760, 770, 850, 1000, 1150, 1500},
            {600, 700, 750, 800, 780, 790, 1000, 1100, 1250, 1500},
            {800, 910, 980, 1001, 1150, 1200, 1250, 1300, 1450, 1500}};

    public void test() throws Exception {
        URL pitUrl = this.getClass().getClassLoader().getResource("testlocation/test/cell/pit");
        AbstractGridFormat format = (AbstractGridFormat) new GrassCoverageFormatFactory()
                .createFormat();
        File pitFile = DataUtilities.urlToFile(pitUrl);
        assertTrue(format.accepts(pitFile));

        GridCoverage2D gc = readGc(format, pitFile);

        File parentFile = pitFile.getParentFile();
        File newPitFile = new File(parentFile, "newpit");
        // writing it down
        GridCoverageWriter writer = format.getWriter(newPitFile);
        writer.write(gc, null);

        // check it again through reading
        readGc(format, newPitFile);
    }

    private GridCoverage2D readGc( AbstractGridFormat format, File fileToRead ) throws IOException {
        GridCoverageReader reader = format.getReader(fileToRead);
        GridCoverage2D gc = ((GridCoverage2D) reader.read(null));
        assertTrue(gc != null);
        assertTrue(CoverageUtilities.hasRenderingCategories(gc));

        checkMatrixEqual(gc.getRenderedImage(), mapData, 0);

        Envelope envelope = gc.getEnvelope();
        DirectPosition lowerCorner = envelope.getLowerCorner();
        double[] westSouth = lowerCorner.getCoordinate();
        DirectPosition upperCorner = envelope.getUpperCorner();
        double[] eastNorth = upperCorner.getCoordinate();

        GridGeometry2D gridGeometry = gc.getGridGeometry();
        GridEnvelope2D gridRange = gridGeometry.getGridRange2D();
        int height = gridRange.height;
        int width = gridRange.width;

        assertEquals(w, westSouth[0]);
        assertEquals(s, westSouth[1]);
        assertEquals(e, eastNorth[0]);
        assertEquals(n, eastNorth[1]);
        assertEquals(8, height);
        assertEquals(10, width);
        return gc;
    }

    protected void checkMatrixEqual( RenderedImage image, double[][] matrix, double delta ) {
        RectIter rectIter = RectIterFactory.create(image, null);
        int y = 0;
        do {
            int x = 0;
            do {
                double value = rectIter.getSampleDouble();
                double expectedResult = matrix[y][x];
                if (Double.isNaN(value)) {
                    assertTrue(x + " " + y, Double.isNaN(expectedResult));
                } else {
                    assertEquals(x + " " + y, expectedResult, value, delta);
                }
                x++;
            } while( !rectIter.nextPixelDone() );
            rectIter.startPixels();
            y++;
        } while( !rectIter.nextLineDone() );
    }

    public static final void main( String[] args ) throws Exception {
        junit.textui.TestRunner.run(GrassRasterReaderWriterTest.class);
    }

}