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.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.exception.ProcessInterruptedException;

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

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

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

import org.gvsig.fmap.dal.coverage.store.parameter.RasterFileStoreParameters;

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

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 org.gvsig.raster.fmap.layers.DefaultFLyrRaster;

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 Nacho Brodin (nachobrodin@gmail.com)

 * 

 */

public class FLyrRasterIRasterLayer extends AbstractRasterLayer {

    private FLyrRaster      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 FLyrRaster lyr) {

            this.lyr = lyr;

    }

    private RasterDataStore getDataStore() {

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

    }

    public int getDataType() {

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

    }

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

        final double dValue) {

    }

    public void setNoDataValue(final double dNoDataValue) {

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

    }

    @Override

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

    }

    public double getNoDataValue() {

            if(getDataStore().getNoDataValue().getValue() != null)

                    return getDataStore().getNoDataValue().getValue().doubleValue();

            return -99999.0;

    }

    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().getDataType()[0]) {

                    case Buffer.TYPE_BYTE:

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

                    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 getNoDataValue();

            }

    }

    public int getBandsCount() {

        return getDataStore().getBandCount();

    }

    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;

    }

    public void postProcess() {

    }

    public void open() {

            if(getDataStore() == null)

                    setBaseDataObject(fName);

            RasterQuery query = RasterLocator.getManager().createQuery();

                query.setAllDrawableBands();

                try {

                        query.setAreaOfInterest();

                        query.setReadOnly(true);

                        m_Buffer = getDataStore().query(query);

                } catch (QueryException e) {

                         Sextante.addErrorToLog(e);

                } catch (ProcessInterruptedException e) {

                         Sextante.addErrorToLog(e);

                } 

    }

    public void close() {

            if(m_Buffer != null)

                    m_Buffer.dispose();

    }

    public Rectangle2D getFullExtent() {

        return getDataStore().getExtent().toRectangle2D();

    }

    public AnalysisExtent getLayerGridExtent() {

            if(layerExtent == null) {

                    try {

                            layerExtent = new AnalysisExtent();

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

                            Extent bbox = getDataStore().getExtent();

                            layerExtent.setXRange(bbox.getMin().getX(), bbox.getMax().getX(), true);

                            layerExtent.setYRange(bbox.getMin().getY(), bbox.getMax().getY(), true);

                    } catch (final Exception e) {

                            return null;

                    }

            }

            return layerExtent;

    }

    public void setWindowExtent(final AnalysisExtent extent) {

            super.setWindowExtent(extent);

            RasterManager rManager = RasterLocator.getManager();

        RasterQuery query = rManager.createQuery();

        query.setReadOnly(true);

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

        for (int i = 0; i < getDataStore().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 = getDataStore().query(query);

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

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

                        Extent bbox = getDataStore().getExtent();

                        double distx = extent.getXMin() - bbox.getMin().getX();

                        double disty = bbox.getMax().getY() - 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 (QueryException e) {

                        m_Buffer = entireBuf;

                } catch (ProcessInterruptedException e) {

                } 

     }

    public double getLayerCellSize() {

        return getDataStore().getCellSize();

    }

    public Object getCRS() {

        return getDataStore().getProjection();

    }

    public void setName(final String name) {

            lyr.setName(name);

    }

    public void free() {

            if(m_Buffer != null)

                    m_Buffer.dispose();

        try {

                        getDataStore().close();

                } catch (CloseException e) {

                        Sextante.addErrorToLog(e);

                }

    }

    public Object getBaseDataObject() {

        return lyr;

    }

    public void setBaseDataObject(FLyrRaster lyr) {

        this.lyr = lyr;

    }

    public void setBaseDataObject(RasterDataStore store) {

            DefaultFLyrRaster lyr = new DefaultFLyrRaster();

        try {

                        lyr.setDataStore(store);

                } catch (LoadLayerException e) {

                        Sextante.addErrorToLog(e);

                        return;

                }

        this.lyr = lyr;

    }

    public void setBaseDataObject(String fileName) {

            DataManager dataManager = DALLocator.getDataManager();

            try {

                    RasterFileStoreParameters params = 

                        (RasterFileStoreParameters)dataManager.createStoreParameters("Gdal Store");

                    params.setURI(new File(fileName).toURI());

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

                        setBaseDataObject(dataStore);

                } catch (ValidateDataParametersException e) {

                        Sextante.addErrorToLog(e);

                } catch (InitializeException e) {

                        Sextante.addErrorToLog(e);

                } catch (ProviderNotRegisteredException e) {

                        Sextante.addErrorToLog(e);

                }

    }

    public IOutputChannel getOutputChannel() {

        return new IOutputChannel() {

            public String getAsCommandLineParameter() {

                return lyr.getName();

            }

        };

    }

}