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.raster.tools.app.basic.tool.infobypoint.viewtool;

import java.awt.geom.Point2D;

import java.awt.image.BufferedImage;

import java.util.ArrayList;

import java.util.List;

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

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

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

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

import org.gvsig.fmap.dal.coverage.util.ColorConversion;

import org.gvsig.fmap.mapcontext.ViewPort;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.FLayers;

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.fmap.mapcontrol.tools.Events.PointEvent;

import org.gvsig.raster.fmap.layers.FLyrRaster;

import org.gvsig.raster.swing.infobypoint.InfoByPointDataModel;

import org.gvsig.raster.tools.app.basic.tool.selectrasterlayer.SelectImageListImpl;

/**

 * This tool writes in InfoByPointDataModel the information about the selected point.

 * 

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

 */

public class InfoByPixelPointViewTool extends SelectImageListImpl {

        private RasterManager            rManager            = RasterLocator.getManager();

        private InfoByPointDataModel     model               = null;

        private ColorConversion          conv                = null;

        private ViewPort                 viewPort            = null;

        private FLayers                  layers              = null;

        private List<String>             list                = null;

        /**

         * Contructor

         * @param mapCtrl

         */

        public InfoByPixelPointViewTool(MapControl mapCtrl, InfoByPointDataModel model) {

                super(mapCtrl);

                this.model = model;

                conv = rManager.getColorConversion();

                viewPort = mapCtrl.getViewPort();

                layers = mapCtrl.getMapContext().getLayers();

                list = new ArrayList<String>();

        }

        @SuppressWarnings("deprecation")

        public void point(PointEvent event) {

                super.point(event);

                Point2D point2D = event.getPoint();

                BufferedImage image = mapCtrl.getImage();

                int value = image.getRGB((int) point2D.getX(), (int) point2D.getY());

                int r = (value >> 16) & 0xff;

                int g = (value >> 8) & 0xff;

                int b = value & 0xff;

                double[] cmyk = conv.RGBtoCMYK(r & 0xff, g & 0xff, b & 0xff, 1D);

                double[] hsl = conv.RGBtoHSL(r & 0xff, g & 0xff, b & 0xff);

                hsl[0] = (int)(255.0 * hsl[0] / 360.0 + 0.5);

                hsl[2] = (int) (hsl[2] * 255. + 0.5);

                hsl[1] = (int) (hsl[1] * 255. + 0.5);

                Point2D mapPoint = viewPort.toMapPoint(point2D);

                model.setARGB(r, g, b);

                model.setCMYK(cmyk);

                model.setHSL(hsl[0], hsl[1], hsl[2]);

                model.setViewPoint(point2D.getX(), point2D.getY());

                model.setWorldPoint(mapPoint.getX(), mapPoint.getY());

                //Raster dependent

                list.clear();

                FLyrRaster selected = null;

                for (int i = layers.getLayersCount() - 1; i >= 0; i--) {

                        FLayer lyr = layers.getLayer(i);

                        if(lyr instanceof FLyrRaster) {

                                FLyrRaster raster = (FLyrRaster)lyr;

                                Point2D tmpPoint = mapPoint;

                                if(raster.getCoordTrans() != null){

                    tmpPoint = raster.getCoordTrans().getInverted().convert(mapPoint, null);

                                }

                                if(raster.isInside(tmpPoint)) {

                                    list.add(raster.getName());

                    if(raster.isActive())

                        selected = raster;

                                }

                        }

                }

                if(selected != null) {

                    if(selected.getCoordTrans() != null){

                        mapPoint = selected.getCoordTrans().getInverted().convert(mapPoint, null);

            }

                        Point2D rasterPoint = selected.getDataStore().worldToRaster(mapPoint);

                        model.setNumberOfBands(selected.getDataStore().getBandCount());

                        model.setBandValues(getBandValues(selected.getDataStore(), rasterPoint));

                        model.setPixelPoint(rasterPoint.getX(), rasterPoint.getY());

                } else {

                        list.clear();

                        list.add("...");

                        model.setNumberOfBands(0);

                        model.setBandValues(new double[]{Double.NaN});

                        model.setPixelPoint(-1, -1);

                }

                model.setLayerList(list);

                model.notifyObservers();

        }

        /**

         * Loads the values of a pixel point

         * @param dataStore

         * @param pxPoint

         * @return

         */

        private double[] getBandValues(RasterDataStore dataStore, Point2D pxPoint) {

                double[] res = new double[dataStore.getBandCount()];

                int pos = 0;

                try {

                        if(dataStore.getDataType()[0] >= 0 && dataStore.getDataType()[0] <= 3) {

                                for(int i = 0; i < dataStore.getBandCount(); i++) {

                                        int val = ((Integer)dataStore.getData((int)pxPoint.getX(), (int)pxPoint.getY(), i)).intValue();

                                        if(dataStore.getDataType()[0] == Buffer.TYPE_BYTE)

                                                res[pos] = (val & 0x000000ff);

                                        else

                                                res[pos] = val;

                                        pos ++;

                                }

                        }

                        if(dataStore.getDataType()[0] == 4) {

                                for(int i = 0; i < dataStore.getBandCount(); i++) {

                                        res[pos] = ((Float)dataStore.getData((int)pxPoint.getX(), (int)pxPoint.getY(), i)).doubleValue() ;

                                        pos ++;

                                }

                        }

                        if(dataStore.getDataType()[0] == 5) {

                                for(int i = 0; i < dataStore.getBandCount(); i++) {

                                        res[pos] = ((Double)dataStore.getData((int)pxPoint.getX(), (int)pxPoint.getY(), i)).doubleValue();

                                        pos ++;

                                }

                        }

                        return res;

                } catch (Exception e) {

                        for (int i = 0; i < dataStore.getBandCount(); i++) {

                                res[i] = Double.NaN;

                        }

                } 

                return res;

        }

}