Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 */

package org.gvsig.fmap.dal.coverage.dataset.io.rmf;

import java.awt.geom.AffineTransform;

import java.io.File;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.coverage.BaseTestCase;

import org.gvsig.fmap.dal.coverage.dataset.Buffer;

import org.gvsig.fmap.dal.coverage.datastruct.BufferHistogram;

import org.gvsig.fmap.dal.coverage.datastruct.HistogramClass;

import org.gvsig.fmap.dal.coverage.exception.ParsingException;

import org.gvsig.fmap.dal.coverage.store.RasterDataStore;

import org.gvsig.fmap.dal.coverage.store.props.ColorTable;

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemStoreParameters;

import org.gvsig.fmap.dal.spi.DataStoreProviderServices;

import org.gvsig.raster.impl.datastruct.BufferHistogramImpl;

import org.gvsig.raster.impl.datastruct.DefaultNoData;

import org.gvsig.raster.impl.datastruct.GeoPoint;

import org.gvsig.raster.impl.datastruct.GeoPointList;

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

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

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

import org.gvsig.raster.impl.provider.DefaultRasterProvider;

import org.gvsig.raster.impl.store.DefaultMultiRasterStore;

import org.gvsig.raster.impl.store.properties.DataStoreColorInterpretation;

import org.gvsig.raster.impl.store.properties.DataStoreStatistics;

import org.gvsig.raster.impl.store.rmf.IRmfBlock;

import org.gvsig.raster.impl.store.rmf.RmfBlocksManager;

import org.gvsig.raster.impl.store.serializer.ColorInterpretationRmfSerializer;

import org.gvsig.raster.impl.store.serializer.GeoInfoRmfSerializer;

import org.gvsig.raster.impl.store.serializer.GeoPointListRmfSerializer;

import org.gvsig.raster.impl.store.serializer.StatisticsRmfSerializer;

import org.gvsig.raster.lizardtech.io.LizardTechProvider;

/**

 * Test de lectura para ficheros rmf. Obtiene distintos tipos de bloques y

 * comprueba que el objeto que han de generar es correcto.

 *

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

 */

public class TestRmfRead extends BaseTestCase {

        private RmfBlocksManager blocksManager = null;

        public String            file    = "readtest.rmf";

        private DefaultRasterProvider      f2       = null;        

        protected Buffer           raster  = null;

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

        public void start() throws Exception {

                this.setUp();

                this.testStack();

        }

        protected void doSetUp() throws Exception {

                try {

                        super.doSetUp();

                } catch (Exception e) {

                        e.printStackTrace();

                }

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

                deleteRMF(path1);

                DefaultMultiRasterStore grmf = new DefaultMultiRasterStore();

                DataManager dm = DALLocator.getDataManager();

                FilesystemStoreParameters p = (FilesystemStoreParameters)dm.createStoreParameters(LizardTechProvider.NAME);

                p.setFile(new File(path1));

                f2 = (DefaultRasterProvider)loadProvider(p, (DataStoreProviderServices)grmf);

        }

        public void testStack(){

                blocksManager = new RmfBlocksManager(baseDir + file);

                if (!blocksManager.checkRmf())

                        assertEquals(0, 1);

                HistogramRmfSerializer ser = new HistogramRmfSerializer();

                ColorTableRmfSerializer ser1 = new ColorTableRmfSerializer();

                StatisticsRmfSerializer ser2 = new StatisticsRmfSerializer();

                GeoInfoRmfSerializer ser3 = new GeoInfoRmfSerializer(f2);

                GeoPointListRmfSerializer ser4 = new GeoPointListRmfSerializer();

                NoDataRmfSerializer ser5 = new NoDataRmfSerializer();

                ColorInterpretationRmfSerializer ser6 = new ColorInterpretationRmfSerializer((DataStoreColorInterpretation)f2.getColorInterpretation());

                blocksManager.addClient(ser);

                blocksManager.addClient(ser1);

                blocksManager.addClient(ser2);

                blocksManager.addClient(ser3);

                blocksManager.addClient(ser4);

                blocksManager.addClient(ser5);

                blocksManager.addClient(ser6);

                try {

                        blocksManager.read(null);

                        for (int i = 0; i < blocksManager.getClientsCount(); i++) {

                                IRmfBlock client = blocksManager.getClient(i);

                                if (client instanceof HistogramRmfSerializer)

                                        testHistogram((BufferHistogram) client.getResult());

                                if (client instanceof ColorTableRmfSerializer) {

                                        ColorTable colorTable = (ColorTable) client.getResult();

                                        colorTable.setInterpolated(false);

                                        testColorTable(colorTable);

                                }

                                if (client instanceof StatisticsRmfSerializer)

                                        testStatistics((DataStoreStatistics) client.getResult());

                                if (client instanceof NoDataRmfSerializer)

                                        testNoData((DefaultNoData) client.getResult());

//                                if (client instanceof GeoInfoRmfSerializer)

//                                        testGeoInfo((RasterDataset) client.getResult());

                                if (client instanceof GeoPointListRmfSerializer)

                                        testGeoPoints((GeoPointList) client.getResult());

                                if (client instanceof ColorInterpretationRmfSerializer) {

                                        DataStoreColorInterpretation ci = (DataStoreColorInterpretation) client.getResult();

                                        testColorInterpretation(ci);

                                }

                        }

                } catch (ParsingException e) {

                        e.printStackTrace();

                }

        }

        public void testColorInterpretation(DataStoreColorInterpretation ci) {

                for (int j = 0; j < ci.length(); j++) {

                        String value = ci.get(j);

//                        System.out.println(value);

                        if (j == 0)

                                assertEquals(value, "Red");

                        if (j == 1)

                                assertEquals(value, "Green");

                        if (j == 2)

                                assertEquals(value, "Blue");

                }

        }

        public void testHistogram(BufferHistogram h) {

                BufferHistogram h2 = BufferHistogramImpl.convertHistogramToRGB(h);

                HistogramClass[][] classes = h2.getHistogram();

                for (int iBand = 0; iBand < classes.length; iBand++)

                        for (int iValue = 0; iValue < classes[iBand].length; iValue++) {

                                assertEquals((int) classes[iBand][iValue].getMin(), iValue);

                                assertEquals((int) classes[iBand][iValue].getMax(), iValue + 1);

                        }

                assertEquals((int) h.getHistogramValue(0, 0), 16);

                assertEquals((int) h.getHistogramValue(0, 1), 4);

                assertEquals((int) h.getHistogramValue(0, 23), 6);

                assertEquals((int) h.getHistogramValue(0, 48), 4);

                assertEquals((int) h.getHistogramValue(1, 0), 2);

                assertEquals((int) h.getHistogramValue(1, 4), 1);

                assertEquals((int) h.getHistogramValue(1, 7), 8);

                assertEquals((int) h.getHistogramValue(1, 20), 4);

                assertEquals((int) h.getHistogramValue(2, 0), 25);

                assertEquals((int) h.getHistogramValue(2, 2), 1);

                assertEquals((int) h.getHistogramValue(2, 13), 4);

                assertEquals((int) h.getHistogramValue(2, 21), 2);

        }

        public void testColorTable(ColorTable ct) {

//                byte[][] color = ct.getColorTableByBand();

//                assertEquals(ct.getName(), "Prueba Tabla de Color");

                assertEquals(ct.getRGBAByBand(15)[0] & 0xff, 95);

                assertEquals(ct.getRGBAByBand(15)[1] & 0xff, 95);

                assertEquals(ct.getRGBAByBand(15)[2] & 0xff, 95);

                assertEquals(ct.getRGBAByBand(22)[0] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(22)[1] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(22)[2] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(24)[0] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(24)[1] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(24)[2] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(28)[0] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(28)[1] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(28)[2] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(0)[0] & 0xff, 255);

                assertEquals(ct.getRGBAByBand(0)[1] & 0xff, 255);

                assertEquals(ct.getRGBAByBand(0)[2] & 0xff, 255);

                assertEquals(ct.getRGBAByBand(1)[0] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(1)[1] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(1)[2] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(2)[0] & 0xff, 0);

                assertEquals(ct.getRGBAByBand(2)[1] & 0xff, 102);

                assertEquals(ct.getRGBAByBand(2)[2] & 0xff, 255);

//                System.out.println(ct.getName());

//                for (int i = 0; i < color.length; i++) {

//                        System.out.println((color[i][0] & 0xff) + " " + (color[i][1] & 0xff) + " " + (color[i][2] & 0xff));

//                }

        }

        public void testNoData(DefaultNoData noData) {

                assertEquals((int) noData.getValue(), 5450);

                assertEquals(noData.getType(), 2);

        }

        public void testStatistics(DataStoreStatistics ds) {

                int bandCount = ds.getBandCount();

                double[] max = ds.getMaxByteUnsigned();

                double[] min = ds.getMinByteUnsigned();

                double[] secondMax = ds.getSecondMaxByteUnsigned();

                double[] secondMin = ds.getSecondMinByteUnsigned();

                double[] mean = ds.getMean();

                double[] variance = ds.getVariance();

                assertEquals(bandCount, 3);

                for (int i = 0; i < bandCount; i++)

                        switch (i) {

                                case 0:

                                        assertEquals((int) max[i], 250);

                                        assertEquals((int) min[i], 0);

                                        assertEquals((int) secondMax[i], 248);

                                        assertEquals((int) secondMin[i], 1);

                                        assertEquals((int) mean[i], 95);

                                        assertEquals((int) variance[i], 3050);

                                        break;

                                case 1:

                                        assertEquals((int) max[i], 255);

                                        assertEquals((int) min[i], 0);

                                        assertEquals((int) secondMax[i], 254);

                                        assertEquals((int) secondMin[i], 3);

                                        assertEquals((int) mean[i], 105);

                                        assertEquals((int) variance[i], 2689);

                                        break;

                                case 2:

                                        assertEquals((int) max[i], 254);

                                        assertEquals((int) min[i], 0);

                                        assertEquals((int) secondMax[i], 250);

                                        assertEquals((int) secondMin[i], 1);

                                        assertEquals((int) mean[i], 85);

                                        assertEquals((int) variance[i], 3218);

                                        break;

                        }

//                System.out.println(ct.getName());

//                System.out.println(ct.getType());

//                for (int i = 0; i < color.length; i++) {

//                        System.out.print(range[i] + ": ");

//                        System.out.println(color[i][0] + " " + color[i][1] + " " + color[i][2]);

//                }

        }

        public void testGeoInfo(RasterDataStore dataset) {

        //TODO: TEST: Terminar test de acceso a la georreferenciaci?n.

//                String proj = "";

//                if(dataset.getProjection() != null)

//                        proj = dataset.getProjection().getAbrev();

                AffineTransform at = dataset.getAffineTransform();

                assertEquals((int) at.getTranslateX(), 5000);

                assertEquals((int) at.getTranslateY(), 5000);

                assertEquals((int) at.getScaleX(), 2);

                assertEquals((int) at.getScaleY(), -2);

        }

        public void testGeoPoints(GeoPointList gpList) {

                GeoPoint p = gpList.get(0);

                assertEquals((int) p.pixelPoint.getX(), 10);

                assertEquals((int) p.pixelPoint.getY(), 10);

                assertEquals((int) p.mapPoint.getX(), 34223);

                assertEquals((int) p.mapPoint.getY(), 2344);

                assertEquals(p.active, true);

//                assertEquals((int) p.leftCenterPoint.getX(), 24223);

//                assertEquals((int) p.leftCenterPoint.getY(), 3244);

//                assertEquals((int) p.rightCenterPoint.getX(), 2433);

//                assertEquals((int) p.rightCenterPoint.getY(), 6244);

//

//                assertEquals((int) p.leftViewPort.getExtent().minX(), 30032);

//                assertEquals((int) p.leftViewPort.getExtent().maxY(), 2103);

//

//                assertEquals((int) p.rightViewPort.getExtent().minX(), 30032);

//                assertEquals((int) p.rightViewPort.getExtent().maxY(), 2103);

                assertEquals((int) p.zoomLeft, 1);

                assertEquals((int) p.zoomRight, 1);

        }

}