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

import es.unex.sextante.core.Sextante;

import org.cresques.cts.IProjection;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.project.ProjectManager;

import org.gvsig.app.project.ProjectPreferences;

import org.gvsig.app.project.documents.table.TableDocument;

import org.gvsig.app.project.documents.table.TableManager;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStoreParameters;

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

import org.gvsig.fmap.dal.feature.FeatureStore;

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

import org.gvsig.fmap.dal.store.dbf.DBFStoreParameters;

import org.gvsig.fmap.dal.store.dbf.DBFStoreProvider;

import org.gvsig.fmap.mapcontext.MapContextLocator;

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

public class FileTools {

    public final static String[] RASTER3D_EXT_IN = { "asc3d" };

    public final static String[] RASTER_EXT_IN = { "tif", "asc", "dat", "tiff",

        "bmp", "gif", "img", "jpg", "png", "vrt", "lan", "gis", "pix", "aux",

        "adf", "mpr", "mpl", "map", "hdr" };

    public final static String[] RASTER_DRIVERS_IN = { "RasterStore",

        "RasterStore", "RasterStore", "RasterStore", "RasterStore",

        "RasterStore", "RasterStore", "RasterStore", "RasterStore",

        "RasterStore", "RasterStore", "RasterStore", "RasterStore",

        "RasterStore", "RasterStore", "RasterStore", "RasterStore",

        "RasterStore", "RasterStore" };

    public final static String[] VECTOR_EXT_IN = { "shp", "gml", "dxf", "dgn",

        "dwg" };

    public final static String[] VECTOR_DRIVERS_IN = { "GML", "DXF", "DGN",

        "DWG" };

    public final static String[] TABLE_EXT = { "dbf" };

    public static final String[] LAYERS_EXT_IN = { "tif", "asc", "dat", "tiff",

        "bmp", "gif", "img", "jpg", "png", "vrt", "lan", "gis", "pix", "aux",

        "adf", "mpr", "mpl", "map", "shp", "gml", "dxf", "dgn", "dwg" };

    public static final String[] LAYER_DRIVERS_IN = { "Gdal Store",

        "Gdal Store", "Gdal Store", "Gdal Store", "Gdal Store", "Gdal Store",

        "Gdal Store", "Gdal Store", "Gdal Store", "Gdal Store", "Gdal Store",

        "Gdal Store", "Gdal Store", "Gdal Store", "Gdal Store", "Gdal Store",

        "Gdal Store", "Gdal Store", "Shape", "GML", "DXF", "DGN", "DWG" };

    public static final boolean[] LAYER_RASTER = { true, true, true, true,

        true, true, true, true, true, true, true, true, true, true, true, true,

        true, true, false, false, false, false, false, };

    public static FLayer openLayer(final String sFilename,

        final String sName, final IProjection projection) {

        final String sExtension =

            sFilename.substring(sFilename.lastIndexOf('.') + 1,

                sFilename.length());

        for (int i = 0; i < LAYERS_EXT_IN.length; i++) {

            if (sExtension.equals(LAYERS_EXT_IN[i])) {

                try {

                    FLayer layer;

                    final DataManager dm = DALLocator.getDataManager();

                    final DataStoreParameters params =

                        dm.createStoreParameters(FileTools.LAYER_DRIVERS_IN[i]);

                    ((FilesystemStoreParameters) params).setFile(new File(sFilename));

                    if(!(params instanceof RasterDataParameters)) {

                            params.setDynValue("crs", projection);

                    }

                    layer = MapContextLocator.getMapContextManager().createLayer(sName, params);

                    if ((layer != null) && layer.isAvailable()) {

                        return layer;

                    } else {

                        return null;

                    }

                } catch (final Exception e) {

                    Sextante.addErrorToLog(e);

                    return null;

                }

            }

        }

        return null;

    }

    public static TableDocument openTable(final String sFilename,

        final String sName) {

        final DataManager dm = DALLocator.getDataManager();

        try {

            DataStoreParameters storeParameters =

                dm.createStoreParameters(DBFStoreProvider.NAME);

            storeParameters.setDynValue(

                DBFStoreParameters.DBFFILE_PARAMTER_NAME, sFilename);

            final FeatureStore store =

                (FeatureStore) dm.openStore(DBFStoreProvider.NAME,

                    storeParameters);

            final TableDocument pt =

                (TableDocument) ProjectManager.getInstance().createDocument(

                    TableManager.TYPENAME, sName);

            pt.setStore(store);

            return pt;

        } catch (final Exception e) {

            Sextante.addErrorToLog(e);

            return null;

        }

    }

    public static Object open(final String sFilename) {

        final IProjection projection =

            ((ProjectPreferences) ApplicationLocator.getManager()

                .getPreferences("project")).getDefaultProjection();

        FLayer layer = openLayer(sFilename, sFilename, projection);

        if (layer != null) {

            return layer;

        }

        final TableDocument table = openTable(sFilename, sFilename);

        return table;

    }

}