Application: gvSIG desktop » gvSIG geoprocess

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2012 gvSIG Association.

 *

 * 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.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.geoprocess.lib.sextante.dataObjects;

import java.awt.geom.Rectangle2D;

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.datastruct.Extent;

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

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

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

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

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

import org.gvsig.fmap.dal.coverage.grid.Grid;

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

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

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

import es.unex.sextante.core.AnalysisExtent;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.AbstractRasterLayer;

import es.unex.sextante.outputs.IOutputChannel;

/**

 * A wrapper for a gvSIG Raster layer. Allows only reading, but not writing to

 * it

 * 

 * @author volaya, nacho brodin (nachobrodin@gmail.com)

 * 

 */

public class FLyrRasterReadOnlyIRasterLayer extends AbstractRasterLayer {

    private Grid           m_LayerGrid    = null;

    private FLyrRaster     m_Layer        = null;

    private Buffer         m_Buffer       = null;

    private int            m_iDataType    = 0;

    private double         m_dNoDataValue = 0;

    private int            xTranslate     = 0;

    private int            yTranslate     = 0;

    private AnalysisExtent layerExtent    = null;

    public void create(final FLyrRaster obj) {

        m_Layer = obj;

    }

    public int getDataType() {

        return m_Layer.getDataStore().getDataType()[0];

    }

    public void setCellValue(final int x, final int y, final int iBand,

        final double dValue) {

    }

    public void setNoDataValue(final double dNoDataValue) {

        m_Layer.getNoDataValue().setValue(Double.valueOf(dNoDataValue));

    }

    @Override

    public void setNoData(final int x, final int y) {

    }

    public double getNoDataValue() {

        return m_dNoDataValue;

    }

    public double getCellValueInLayerCoords(final int x, final int y,

        final int iBand) {

            int newx = x - xTranslate;

            int newy = y - yTranslate;

        if (m_Buffer.isInside(newx, newy)) {

            switch (m_iDataType) {

            case Buffer.TYPE_BYTE:

                return m_Buffer.getElemByte(newy, newx, iBand);

            case Buffer.TYPE_SHORT:

                return m_Buffer.getElemShort(newy, newx, iBand);

            case Buffer.TYPE_INT:

                return m_Buffer.getElemInt(newy, newx, iBand);

            case Buffer.TYPE_FLOAT:

                return m_Buffer.getElemFloat(newy, newx, iBand);

            case Buffer.TYPE_DOUBLE:

            default:

                return m_Buffer.getElemDouble(newy, newx, iBand);

            }

        } else {

            return m_dNoDataValue;

        }

    }

    public int getBandsCount() {

        return m_Layer.getDataStore().getBandCount();

    }

    public String getName() {

        return m_Layer.getName();

    }

    public void postProcess() {

    }

    public void open() {

        try {

            m_LayerGrid = m_Layer.getReadOnlyFullGrid(false);

            m_Buffer = m_LayerGrid.getRasterBuf();

            m_iDataType = m_LayerGrid.getDataType();

            NoData nodata = m_LayerGrid.getNoDataValue();

            m_dNoDataValue = nodata != null ? 

                            (nodata.getValue() != null ? nodata.getValue().doubleValue() : 0.0d) : 

                            0.0d;

        } catch (GridException e) {

            Sextante.addErrorToLog(e);

        } catch (InterruptedException e) {

            Sextante.addErrorToLog(e);

        }

    }

    public void close() {

        m_LayerGrid = null;

    }

    public Rectangle2D getFullExtent() {

        return m_Layer.getFullRasterExtent().toRectangle2D();

    }

    public AnalysisExtent getLayerGridExtent() {

            if(layerExtent == null) {

                    try {

                            layerExtent = new AnalysisExtent();

                            layerExtent.setCellSize(m_Layer.getDataStore().getCellSize());

                            layerExtent.setXRange(m_Layer.getMinX(), m_Layer.getMaxX(), true);

                            layerExtent.setYRange(m_Layer.getMinY(), m_Layer.getMaxY(), true);

                    } catch (final Exception e) {

                            return null;

                    }

            }

            return layerExtent;

    }

    public void setWindowExtent(final AnalysisExtent extent) {

            super.setWindowExtent(extent);

            RasterManager rManager = RasterLocator.getManager();

        RasterDataStore store = m_Layer.getDataStore();

        RasterQuery query = rManager.createQuery();

        query.setReadOnly(true);

        int[] bands = new int[store.getBandCount()];

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

                        bands[i] = i;

                }

        query.setDrawableBands(bands);

        Extent ext = RasterLocator.getManager().getDataStructFactory().createExtent(

                        extent.getXMin(), 

                        extent.getYMax(), 

                        extent.getXMax(), 

                        extent.getYMin());

        query.setAreaOfInterest(ext, extent.getNX(), extent.getNY());

        Buffer entireBuf = m_Buffer;

        try {

                        m_Buffer = store.query(query);

                        //Calculamos la traslaci?n respecto a la ventana recortada pq Sextante pedir? 

                        //p?xeles en relaci?n a la imagen completa

                        double distx = extent.getXMin() - m_Layer.getMinX();

                        double disty = m_Layer.getMaxY() - extent.getYMax();

                        xTranslate = (int)Math.round((distx * extent.getNX()) / (extent.getXMax() - extent.getXMin()));

                        yTranslate = (int)Math.round((disty * extent.getNY()) / (extent.getYMax() - extent.getYMin()));

                } catch (RasterDriverException e) {

                        m_Buffer = entireBuf;

                } catch (ProcessInterruptedException e) {

                } catch (InvalidSetViewException e) {

                        m_Buffer = entireBuf;

                }

     }

    public double getLayerCellSize() {

        return m_Layer.getDataStore().getCellSize();

    }

    public Object getCRS() {

        return m_Layer.getProjection();

    }

    public void setName(final String name) {

        m_Layer.setName(name);

    }

    public void free() {

        m_Layer.dispose();

        m_Layer = null;

        m_LayerGrid = null;

    }

    public Object getBaseDataObject() {

        return m_Layer;

    }

    public IOutputChannel getOutputChannel() {

        return new IOutputChannel() {

            public String getAsCommandLineParameter() {

                return m_Layer.getName();

            }

        };

    }

}