| gvSIG desktop 1

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2007 IVER T.I. and Generalitat Valenciana.

 *

 * This program 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 2

 * 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, write to the Free Software

 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.raster.dataset.io.rmf;

import java.awt.geom.Point2D;

import java.io.FileNotFoundException;

import java.io.IOException;

import junit.framework.TestCase;

import org.gvsig.raster.RasterLibrary;

import org.gvsig.raster.buffer.BufferFactory;

import org.gvsig.raster.dataset.FileNotOpenException;

import org.gvsig.raster.dataset.NotSupportedExtensionException;

import org.gvsig.raster.dataset.RasterDataset;

import org.gvsig.raster.dataset.io.RasterDriverException;

import org.gvsig.raster.dataset.properties.DatasetColorInterpretation;

import org.gvsig.raster.dataset.properties.DatasetStatistics;

import org.gvsig.raster.dataset.serializer.ColorInterpretationRmfSerializer;

import org.gvsig.raster.dataset.serializer.GeoInfoRmfSerializer;

import org.gvsig.raster.dataset.serializer.GeoPointRmfSerializer;

import org.gvsig.raster.dataset.serializer.StatisticsRmfSerializer;

import org.gvsig.raster.datastruct.ColorTable;

import org.gvsig.raster.datastruct.Extent;

import org.gvsig.raster.datastruct.GeoPoint;

import org.gvsig.raster.datastruct.Histogram;

import org.gvsig.raster.datastruct.ViewPortData;

import org.gvsig.raster.datastruct.serializer.ColorTableRmfSerializer;

import org.gvsig.raster.datastruct.serializer.HistogramRmfSerializer;

import org.gvsig.raster.datastruct.serializer.NoDataRmfSerializer;

/**

 * Test de escritura de ficheros rmf.

 * Escribe un fichero rmf con distintos bloques y a continuaci?n le pasa un

 * test de lectura para comprobar que se ha generado bien.

 *

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class TestRmfWrite extends TestCase {

        private RmfBlocksManager manager = null;

        private String baseDir = "./test-images/";

        private String path = baseDir + "miniRaster25x24.tif";

        private String pathGif = baseDir + "gifTransparente.gif";

        private String pathJpg = baseDir + "03AUG23153350-M2AS-000000122423_01_P001-BROWSE.jpg";

        private Histogram histogram = null;

        private ColorTable colorTable = null;

        private RasterDataset f = null;

        private RasterDataset f2 = null;

        static {

                RasterLibrary.wakeUp();

        }

        public void start() {

                this.setUp();

                this.testStack();

        }

        public void setUp() {

                System.err.println("TestRmfWrite running...");

                RasterDataset f1 = null;

                try {

                        f = RasterDataset.open(null, path);

                        f1 = RasterDataset.open(null, pathGif);

                        f2 = RasterDataset.open(null, pathJpg);

                } catch (NotSupportedExtensionException e) {

                        return;

                } catch (RasterDriverException e) {

                        return;

                }

                BufferFactory ds = new BufferFactory(f);

                try {

                        ds.setAreaOfInterest();

                } catch (InterruptedException e1) {

                        e1.printStackTrace();

                } catch (RasterDriverException e) {

                        e.printStackTrace();

                }

                try {

                        histogram = f.getHistogram().getHistogram();

                        colorTable = f1.getColorTable();

                        f.getStatistics().calcFullStatistics();

                } catch (FileNotOpenException e) {

                        e.printStackTrace();

                } catch (RasterDriverException e) {

                        e.printStackTrace();

                } catch (InterruptedException e) {

                        e.printStackTrace();

                }

        }

        public void testStack(){

                manager = new RmfBlocksManager(baseDir + "writetest.rmf");

                //Histograma

                HistogramRmfSerializer ser = new HistogramRmfSerializer(histogram);

                manager.addClient(ser);

                //Tabla de color

                //colorTable.setName("Prueba Tabla de Color");

                ColorTableRmfSerializer ser1 = new ColorTableRmfSerializer(colorTable);

                manager.addClient(ser1);

                //Estadisticas

                DatasetStatistics stat = f.getStatistics();

                stat.setTailTrimValue(3.0, new Double(10.0));

                stat.setTailTrimValue(4.0, new Double(16.0));

                StatisticsRmfSerializer ser2 = new StatisticsRmfSerializer(stat);

                manager.addClient(ser2);

                //Georreferenciaci?n

                GeoInfoRmfSerializer ser3 = new GeoInfoRmfSerializer(f2);

                manager.addClient(ser3);

                //Puntos de control

                GeoPoint p1 = new GeoPoint();

                p1.pixelPoint = new Point2D.Double(10, 10);

                p1.mapPoint = new Point2D.Double(34223.3, 2344.2);

                p1.leftViewPort = new ViewPortData();

                p1.leftViewPort.setExtent(new Extent(30032.3, 2103.3, 50023.3, 1234.3));

                p1.rightViewPort = new ViewPortData();

                p1.rightViewPort.setExtent(new Extent(30032.3, 2103.3, 50023.3, 1234.3));

                p1.leftCenterPoint = new Point2D.Double(24223.3, 3244.2);

                p1.rightCenterPoint = new Point2D.Double(2433.3, 6244.2);

                GeoPoint p2 = new GeoPoint();

                p2.pixelPoint = new Point2D.Double(10, 10);

                p2.mapPoint = new Point2D.Double(34223.3, 2344.2);

                p2.leftViewPort = new ViewPortData();

                p2.leftViewPort.setExtent(new Extent(30032.3, 2103.3, 50023.3, 1234.3));

                p2.rightViewPort = new ViewPortData();

                p2.rightViewPort.setExtent(new Extent(30032.3, 2103.3, 50023.3, 1234.3));

                p2.leftCenterPoint = new Point2D.Double(24223.3, 3244.2);

                p2.rightCenterPoint = new Point2D.Double(2433.3, 6244.2);

                p1.leftViewPort.pxSize = new Point2D.Double(32, 34);

                GeoPointRmfSerializer ser4 = new GeoPointRmfSerializer(new GeoPoint[]{p1, p2}, p1.leftViewPort);

                manager.addClient(ser4);

                // Valor NoData

                NoDataRmfSerializer ser5 = new NoDataRmfSerializer(Double.valueOf(5450.0));

                manager.addClient(ser5);

                //Interpretaci?n de color

                DatasetColorInterpretation ci = f.getColorInterpretation();

                ci.setColorInterpValue(0, DatasetColorInterpretation.BLUE_BAND);

                ci.setColorInterpValue(2, DatasetColorInterpretation.RED_BAND);

                ColorInterpretationRmfSerializer ser6 = new ColorInterpretationRmfSerializer(f.getColorInterpretation());

                manager.addClient(ser6);

                try {

                        manager.write();

                } catch (FileNotFoundException e) {

                        e.printStackTrace();

                } catch (IOException e) {

                        e.printStackTrace();

                }

                //Pasamos el test de lectura para comprobar que se ha generado bien

                TestRmfRead t = new TestRmfRead();

                t.file = "writetest.rmf";

                t.start();

        }

}