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.image.DataBuffer;

import java.io.File;

import es.unex.sextante.core.AnalysisExtent;

import es.unex.sextante.core.Sextante;

import org.cresques.cts.IProjection;

import org.gvsig.andami.Utilities;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

import org.gvsig.fmap.dal.DataStore;

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

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

import org.gvsig.fmap.geom.GeometryLocator;

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

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

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.raster.lib.buffer.api.exceptions.BufferException;

public class RasterDriver {

    public static final int RASTER_DATA_TYPE_FLOAT = DataBuffer.TYPE_FLOAT;

    public static final int RASTER_DATA_TYPE_DOUBLE = DataBuffer.TYPE_DOUBLE;

    public static final int RASTER_DATA_TYPE_INT = DataBuffer.TYPE_INT;

    public static final int RASTER_DATA_TYPE_SHORT = DataBuffer.TYPE_SHORT;

    public static final int RASTER_DATA_TYPE_BYTE = DataBuffer.TYPE_BYTE;

    private final AnalysisExtent m_GridExtent;

    private Buffer buf = null;

    private String name = null;

    public RasterDriver(final AnalysisExtent ae, final int iDataType,

        final int iNumBands) {

        super();

        IProjection projection = null; //TODO check create bufefr without projection

        int[] listDataType; //TODO check how listDataType is made.

        listDataType = new int[1];

        listDataType[0] = (iDataType);

        //buf = BufferLocator.getBufferManager().createBuffer(ae.getNY(),ae.getNX(), listDataType, projection);

        Envelope enve;

        try {

            enve = GeometryLocator.getGeometryManager().createEnvelope(ae.getXMin(), ae.getYMin(), ae.getXMax(), ae.getYMax(), 2);

        } catch (CreateEnvelopeException e) {

            Sextante.addErrorToLog(e);

            enve = null;

        }

        try {

            buf = BufferLocator.getBufferManager().createBuffer(ae.getNY(), ae.getNX(), listDataType, null, projection, enve);

        } catch (BufferException e) {

            Sextante.addErrorToLog(e);

            buf = null;

        }

        m_GridExtent = ae;

    }

    public RasterDriver(final AnalysisExtent ae, final int iDataType) throws BufferException {

        this(ae, iDataType, 1);

    }

    public AnalysisExtent getGridExtent() {

        return m_GridExtent;

    }

    public Buffer getRasterBuf() {

        return this.buf;

    }

    public void reset() {

        free();

    }

    public void setCellValue(final int x, final int y, final double dValue) {

        setCellValue(x, y, 0, dValue);

    }

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

        setNoData(x, y, 0);

    }

    private void setNoData(final int x, final int y, final int iBand) {

        setCellValue(x, y, iBand, getNoDataValue());

    }

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

        final double dValue) {

        if (isInGrid(x, y) && (iBand < buf.getBandCount())) {

            switch (buf.getBand(iBand).getDataType()) {

            case BufferManager.TYPE_BYTE:

                buf.getBand(iBand).set(y,x, (byte) dValue);

                break;

            case BufferManager.TYPE_SHORT:

                buf.getBand(iBand).set(y, x,(short) dValue);

                break;

            case BufferManager.TYPE_INT:

                buf.getBand(iBand).set(y, x, (int) dValue);

                break;

            case BufferManager.TYPE_FLOAT:

                buf.getBand(iBand).set(y, x,(float) dValue);

                break;

            case BufferManager.TYPE_DOUBLE:

            default:

                buf.getBand(iBand).set(y, x, dValue);

                break;

            }

        }

    }

    public double getNoDataValue() {

        NoData nodata;

        nodata = buf.getBandNoData()[0];

        return nodata != null ? (nodata.getValue() != null ? nodata.getValue()

            .doubleValue() : 0.0d) : 0.0d;

    }

    public void setNoDataValue(final double dNoDataValue) {

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

                if (buf.getBand(i).getNoData()== null) {

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

                    buf.getBand(i).getNoData().setValue(nodata.getValue());

                } else {

                    buf.getBand(i).getNoData().setValue(dNoDataValue);

                }

        }

    }

    public double getCellValue(final int x, final int y) {

        return getCellValue(x, y, 0);

    }

    public double getCellValue(final int x, final int y, final int iBand) {

        if (isInGrid(x, y) && (iBand < buf.getBandCount())) {

            switch (buf.getBand(iBand).getDataType()) {

            case BufferManager.TYPE_BYTE:

                return buf.getBandByte(iBand).getValue(y, x);

            case BufferManager.TYPE_SHORT:

                return buf.getBandShort(iBand).getValue(y, x);

            case BufferManager.TYPE_INT:

                return buf.getBandInt(iBand).getValue(y,x);

            case BufferManager.TYPE_FLOAT:

                return buf.getBandFloat(iBand).getValue(y,x);

            case BufferManager.TYPE_DOUBLE:

                return buf.getBandDouble(iBand).getValue(y,x);

            default:

                return getNoDataValue();

            }

        } else {

            return getNoDataValue();

        }

    }

    public boolean isNoDataValue(final double dNoDataValue) {

        return (getNoDataValue() == dNoDataValue);

    }

    public boolean isInGrid(final int x, final int y) {

        if ((x < 0) || (y < 0)) {

            return false;

        }

        if ((x >= m_GridExtent.getNX()) || (y >= m_GridExtent.getNY())) {

            return false;

        }

        return true;

    }

    public double getCellSize() {

        return m_GridExtent.getCellSize();

    }

    public static String getFilename(String sRoot, final String sExtension) {

        String sFilename;

        int i = 1;

        sRoot = sRoot.toLowerCase();

        sRoot = sRoot.replaceAll(" ", "_");

        sRoot = sRoot.replaceAll("\\)", "");

        sRoot = sRoot.replaceAll("\\(", "_");

        sRoot = sRoot.replaceAll("\\[", "_");

        sRoot = sRoot.replaceAll("\\]", "");

        sRoot = sRoot.replaceAll("<", "_");

        sRoot = sRoot.replaceAll(">", "_");

        sRoot = sRoot.replaceAll("__", "_");

        while (true) {

            sFilename =

                Utilities.createTempDirectory() + File.separator + sRoot

                    + Integer.toString(i) + "." + sExtension;

            final File file = new File(sFilename);

            if (file.exists()) {

                i++;

            } else {

                return sFilename;

            }

        }

    }

    public boolean export(final String sFilename, final IProjection projection) {

        try {

            final DataManager dm = DALLocator.getDataManager();

            DataServerExplorerParameters serverParameters = dm.createServerExplorerParameters(FilesystemServerExplorer.NAME);

            FilesystemServerExplorer server = (FilesystemServerExplorer) dm.openServerExplorer(FilesystemServerExplorer.NAME, serverParameters);

            File file = new File(sFilename);

            NewRasterStoreParameters parameters = (NewRasterStoreParameters) server.getAddParameters(file);

            parameters.setDynValue(DataStore.METADATA_CRS, projection);

            parameters.setBuffer(buf);

            server.add(parameters.getDataStoreName(), parameters, true);

            return true;

        } catch (Exception e) {

            Sextante.addErrorToLog(e);

        }

        return false;

    }

    public void setName(final String name) {

        this.name = name;

    }

    public String getName() {

        return this.name;

    }

    public void free() {

            if(buf != null) {

                    buf.dispose();

                    buf = null;

            }

    }

}