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.

*

*/

/*

* AUTHORS (In addition to CIT):

* 2009 IVER T.I   {{Task}}

*/

package org.gvsig.raster.gdal.io;

import java.awt.geom.AffineTransform;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.NewDataStoreParameters;

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

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

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

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

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

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

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

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

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

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

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

import org.gvsig.fmap.dal.serverexplorer.filesystem.impl.AbstractFilesystemServerExplorerProvider;

import org.gvsig.fmap.dal.serverexplorer.filesystem.spi.FilesystemServerExplorerProviderServices;

import org.gvsig.raster.impl.store.AbstractRasterFileDataParameters;

import org.gvsig.tools.locator.LocatorException;

public class GdalFilesystemServerExplorer extends AbstractFilesystemServerExplorerProvider {

        public boolean canCreate() {

                return false;

        }

        public boolean canCreate(NewDataStoreParameters parameters) {

                return false;

        }

        public void create(NewDataStoreParameters parameters, boolean overwrite)

                        throws CreateException {

                NewRasterStoreParameters p = null;

                if(parameters instanceof NewRasterStoreParameters)

                         p = (NewRasterStoreParameters)parameters;

                DataServerWriter dataWriter = p.getDataServer();

                if(dataWriter == null)

                        dataWriter = RasterLocator.getManager().createDataServerWriter();

                dataWriter.setBuffer(p.getBuffer(), p.getBand());

                Params params;

                try {

                        if(p.getDriverParams() != null)

                                params = p.getDriverParams();

                        else

                                params = RasterLocator.getManager().createWriter(p.getFileName()).getParams();

                        AffineTransform affineTransform = p.getAffineTransform();

                        if(affineTransform == null) {

                                if(p.getBuffer().getDataExtent() != null) {

                                        double x = p.getBuffer().getDataExtent().getMinX();

                                        double y = p.getBuffer().getDataExtent().getMaxY();

                                        double pixelSizeX = p.getBuffer().getDataExtent().getWidth() / p.getBuffer().getWidth();

                                        double pixelSizeY = p.getBuffer().getDataExtent().getHeight() / p.getBuffer().getHeight();

                                        affineTransform = new AffineTransform(pixelSizeX, 0, 0, -pixelSizeY, x, y);

                                } else {

                                        affineTransform = new AffineTransform();

                                }

                        }

                        RasterWriter writer = RasterLocator.getManager().createWriter(

                                        dataWriter, 

                                        p.getPath() + File.separator + p.getFileName(),

                                        p.getBand() < 0 ? p.getBuffer().getBandCount() : 1, 

                                        affineTransform, 

                                        p.getBuffer().getWidth(),

                                        p.getBuffer().getHeight(), 

                                        p.getBuffer().getDataType(), 

                                        params, 

                                        null);

                        if(p.getColorInterpretation() != null)

                                writer.setColorBandsInterpretation(p.getColorInterpretation());

                        writer.setWkt(p.getWktProjection());

                        writer.dataWrite();

                        writer.writeClose();

                } catch (LocatorException e) {

                        throw new CreateException("", e);

                } catch (NotSupportedExtensionException e) {

                        throw new CreateException("", e);

                } catch (RasterDriverException e) {

                        throw new CreateException("", e);

                } catch (ProcessInterruptedException e) {

                        //Fin del proceso

                } catch (IOException e) {

                        throw new CreateException("", e);

                }

        }

        public NewDataStoreParameters getCreateParameters() throws DataException {

                return new GdalNewRasterStoreParameters();

        }

        public void initialize(FilesystemServerExplorerProviderServices serverExplorer) {

        }

        public void remove(DataStoreParameters parameters) throws RemoveException {

                throw new UnsupportedOperationException();

        }

        public String getDataStoreProviderName() {

                return GdalProvider.NAME;

        }

        public boolean accept(File pathname) {

                if (pathname.getParentFile() != null && 

                        pathname.getParentFile().getName().equals("cellhd")) {

                        if (pathname.getName().endsWith(".rmf")

                                        || pathname.getName().endsWith(".rmf~")) {

                                return false;

                        }

                        return true;

                }

                // Comprobamos que no sea un rmf propio, osea, que contenga xml

                if (pathname.getName().toLowerCase().endsWith(".rmf")) {

                        FileInputStream reader = null;

                        try {

                                reader = new FileInputStream(pathname);

                                String xml = "";

                                for (int i = 0; i < 6; i++) {

                                        xml += (char) reader.read();

                                }

                                if (xml.equals("<?xml ")) {

                                        return false;

                                }

                        } catch (Exception e) {

                        } finally {

                                try {

                                        reader.close();

                                } catch (Exception e) {

                                }

                        }

                }

                return RasterLocator.getManager().getProviderServices().isExtensionSupported(

                                pathname.getAbsolutePath(), 

                                GdalProvider.class);

        }

        public String getDescription() {

                return GdalProvider.DESCRIPTION;

        }

        public DataStoreParameters getParameters(File file) throws DataException {

                DataManager manager = DALLocator.getDataManager();

                AbstractRasterFileDataParameters params = (AbstractRasterFileDataParameters) manager

                                .createStoreParameters(this.getDataStoreProviderName());

                params.setFile(file);

                return params;

        }

        public int getMode() {

                return DataServerExplorer.MODE_RASTER;

        }

}