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 java.io.File;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.exception.InitializeException;

import org.gvsig.fmap.dal.exception.ProviderNotRegisteredException;

import org.gvsig.fmap.dal.exception.ValidateDataParametersException;

import org.gvsig.fmap.mapcontext.exceptions.LoadLayerException;

import es.unex.sextante.core.AnalysisExtent;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.AbstractRasterLayer;

import es.unex.sextante.outputs.IOutputChannel;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.raster.api.NewRasterStoreParameters;

import org.gvsig.fmap.dal.raster.api.RasterQuery;

import org.gvsig.fmap.dal.raster.api.RasterStore;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.exception.CreateEnvelopeException;

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.fmap.mapcontext.MapContextLocator;

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

import org.gvsig.fmap.mapcontext.raster.api.RasterLayer;

import org.gvsig.raster.lib.buffer.api.Buffer;

import org.gvsig.raster.lib.buffer.api.BufferLocator;

import org.gvsig.raster.lib.buffer.api.BufferManager;

import org.gvsig.raster.lib.buffer.api.NoData;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.locator.LocatorException;

/**

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

 * it

 * 

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

 * 

 */

public class FLyrRasterIRasterLayer extends AbstractRasterLayer {

    private RasterLayer      lyr            = null;

    private Buffer          m_Buffer       = null;

    private int             xTranslate     = 0;

    private int             yTranslate     = 0;

    private AnalysisExtent  layerExtent    = null;

    private String          fName          = null;

    public FLyrRasterIRasterLayer() {

    }

    public FLyrRasterIRasterLayer(String fName) {

            this.fName = fName;

    }

    public void create(final RasterLayer lyr) {

            this.lyr = lyr;

    }

    private RasterStore getDataStore() {

            return lyr == null ? null : lyr.getRasterStore();

    }

    @Override

    public int getDataType() {

        return getDataStore().getBandDescriptor(0).getDataType();

    }

    @Override

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

        final double dValue) {

    }

    public void setNoDataValue(final double dNoDataValue, final int iBand) {

        //getDataStore().getBandDescriptor(iBand)

        if (m_Buffer.getBand(iBand).getNoData()== null) {

            NoData nodata = BufferLocator.getBufferManager().createNoData(dNoDataValue, null);

            m_Buffer.getBand(iBand).getNoData().setValue(nodata.getValue());

        } else {

            m_Buffer.getBand(iBand).getNoData().setValue(dNoDataValue);

        }     

    }

    @Override

    public void setNoDataValue(final double dNoDataValue) {

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

           setNoDataValue(dNoDataValue, i);

        }

    }

    @Override

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

    }

    public double getNoDataValue(final int iBand) {

        NoData noData = getDataStore().getBandDescriptor(0).getNoData();

            if(noData!=null && noData.getValue() != null)

                return noData.getValue().doubleValue();

            return -99999.0;

    }

    @Override

    public double getNoDataValue() {

        return getNoDataValue(0);

    }

    @Override

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

                    final int iBand) {

            int newx = x - xTranslate;

            int newy = y - yTranslate;

            if(m_Buffer == null)

                    open();

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

        switch (getDataStore().getBandDescriptor(iBand).getDataType()) {

        case BufferManager.TYPE_BYTE: 

                return (m_Buffer.getBandByte(iBand).getValue(newy, newx) & 0xff);

        case BufferManager.TYPE_SHORT:

                return m_Buffer.getBandShort(iBand).getValue(newy, newx);

        case BufferManager.TYPE_INT:

                return m_Buffer.getBandInt(iBand).getValue(newy, newx);

        case BufferManager.TYPE_FLOAT:

                return m_Buffer.getBandFloat(iBand).getValue(newy, newx);

        case BufferManager.TYPE_DOUBLE:

        default:

                return m_Buffer.getBandDouble(iBand).getValue(newy, newx);            

            }

//        }

//            else {

//                    return getNoDataValue();

//            }

    }    

    @Override

    public int getBandsCount() {

        return getDataStore().getBands();

    }

    @Override

    public String getName() {

            if(lyr != null) {

                    return lyr.getName();

            }

            if(getDataStore() != null) {

                    String uri = getDataStore().getName();

                    if(uri.contains(File.separator)) {

                            int index = uri.lastIndexOf(File.separator);

                            return uri.substring(index + 1);

                    }

                    return uri;

            }

            return null;

    }

    @Override

    public void postProcess() {

    }

    @Override

    public void open() {

            if(getDataStore() == null) {

            try {

                    setBaseDataObject(fName);

            } catch (LoadLayerException e) {

                Sextante.addErrorToLog(e);

            } catch (ValidateDataParametersException e) {

                Sextante.addErrorToLog(e);

            }

            RasterQuery query = getDataStore().createRasterQuery();

            try {

                    m_Buffer = getDataStore().getRasterSet(query);

            } catch (DataException e) { 

                    Sextante.addErrorToLog(e);

            } 

        }

    }

    @Override

    public void close() {

            if(m_Buffer != null)

                    m_Buffer.dispose();

    }

    @Override

    public Rectangle2D getFullExtent() {

        try {

            return getDataStore().getEnvelope().getGeometry().getBounds().getBounds2D();

        } catch (DataException e) {

            Sextante.addErrorToLog(e);

            return null;

        } catch (LocatorException e) {

            Sextante.addErrorToLog(e);

            return null;

        } catch (CreateEnvelopeException e) {

            Sextante.addErrorToLog(e);

            return null;

        }

    }

    @Override

    public AnalysisExtent getLayerGridExtent() {

            if(layerExtent == null) {

                    try {

                            layerExtent = new AnalysisExtent();

                            layerExtent.setCellSize(getLayerCellSize());

                            Envelope bbox = getDataStore().getEnvelope();

                            layerExtent.setXRange(bbox.getLowerCorner().getX(), bbox.getUpperCorner().getX(), true);

                            layerExtent.setYRange(bbox.getLowerCorner().getY(), bbox.getUpperCorner().getY(), true);

                    } catch (final DataException e) {

                            Sextante.addErrorToLog(e);

                    } catch (final CreateEnvelopeException e) {

                        Sextante.addErrorToLog(e);                            

                        layerExtent = new AnalysisExtent();

                } catch (final LocatorException e) {

                        Sextante.addErrorToLog(e);

                }

            }

            return layerExtent;

    }

    @Override

    public void setWindowExtent(final AnalysisExtent extent) {

            super.setWindowExtent(extent);

            RasterQuery query = getDataStore().createRasterQuery();

        //int[] bands = new int[getDataStore().getBands()];

        //for (int i = 0; i < getDataStore().getBands(); i++) {

        //                bands[i] = i;

        //        }

        //query.setDrawableBands(bands);

        Envelope ext;

        try {

            ext = GeometryLocator.getGeometryManager().createEnvelope(

                    extent.getXMin(),

                    extent.getYMin(),

                    extent.getXMax(),

                    extent.getYMax(),

                    Geometry.SUBTYPES.GEOM2D);

        } catch (CreateEnvelopeException e) {

            Sextante.addErrorToLog(e);

            ext = null;

        }

        query.setClip(ext);

        Buffer entireBuf = m_Buffer;

        try {

                        m_Buffer = getDataStore().getRasterSet(query);

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

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

                        Rectangle2D bbox = getDataStore().getEnvelope().getGeometry().getBounds().getBounds2D();

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

                        double disty = bbox.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 (final Exception e) {

                        m_Buffer = entireBuf;                                                                                                     

                }

     }

    @Override

    public double getLayerCellSize() {

        try {

            //return getDataStore().getCellSize();

            return getDataStore().getDimensions().getPixelSizeX();

        } catch (InitializeException e) {

           Sextante.addErrorToLog(e);

        }

        return 0;

    }

    @Override

    public Object getCRS() {

        try {

            return getDataStore().getRasterSet().getProjection();

        } catch (DataException e) {

            Sextante.addErrorToLog(e);

        }

        return null;

    }

    @Override

    public void setName(final String name) {

            lyr.setName(name);

    }

    @Override

    public void free() {

        DisposeUtils.disposeQuietly(m_Buffer);

        DisposeUtils.disposeQuietly(lyr);

    }

    @Override

    public Object getBaseDataObject() {

        return lyr;

    }

    public void setBaseDataObject(RasterLayer lyr) {

        this.lyr = lyr;

    }

    public void setBaseDataObject(RasterStore store) throws LoadLayerException {

        //DefaultRasterLayer newlyr = new DefaultRasterLayer();

        FLayer newlyr = MapContextLocator.getMapContextManager().createLayer(store.getName(), store);

//        try {

//                        newlyr.setDataStore(store);

//                } catch (LoadLayerException e) {

//                        Sextante.addErrorToLog(e);

//                        return;

//                }

        this.lyr = (RasterLayer) newlyr;

    }

    public void setBaseDataObject(String fileName) throws LoadLayerException, ValidateDataParametersException {

            DataManager dataManager = DALLocator.getDataManager();      

            try {

                    NewRasterStoreParameters params;

                params = (NewRasterStoreParameters)dataManager.createStoreParameters("Gdal Store");

                    params.setDynValue("uri",new File(fileName).toURI());

                RasterStore dataStore;

                dataStore = (RasterStore)dataManager.openStore(params.getDataStoreName(), params);

                        setBaseDataObject(dataStore);

                } catch (InitializeException e) {

                        Sextante.addErrorToLog(e);

                } catch (ProviderNotRegisteredException e) {

                        Sextante.addErrorToLog(e);

                }

    }

    @Override

    public IOutputChannel getOutputChannel() {

        return new IOutputChannel() {

            @Override

            public String getAsCommandLineParameter() {

                return lyr.getName();

            }

        };

    }

}