Application: gvSIG desktop » gvSIG geoprocess

package org.gvsig.geoprocess.core;

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.feature.FeatureStore;

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

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

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 = { "RasterStore",

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

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

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

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

        "RasterStore", "Shape", "GML", "DXF", "DGN", "DWG" };

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

                    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 DBFStoreParameters dbfParams = new DBFStoreParameters();

        dbfParams.setDBFFile(sFilename);

        final DataManager dm = DALLocator.getDataManager();

        try {

            final FeatureStore store =

                (FeatureStore) dm.openStore(dbfParams.getDataStoreName(),

                    dbfParams);

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

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

        if (layer != null) {

            return layer;

        }

        final TableDocument table = openTable(sFilename, sFilename);

        return table;

    }

}