Application: gvSIG desktop » gvSIG geoprocess

package org.gvsig.geoprocess.lib.sextante.core;

import java.util.Date;

import javax.swing.JDialog;

import es.unex.sextante.core.AnalysisExtent;

import es.unex.sextante.core.ITaskMonitor;

import es.unex.sextante.core.OutputFactory;

import es.unex.sextante.dataObjects.IRasterLayer;

import es.unex.sextante.dataObjects.ITable;

import es.unex.sextante.dataObjects.IVectorLayer;

import es.unex.sextante.exceptions.UnsupportedOutputChannelException;

import es.unex.sextante.gui.core.DefaultTaskMonitor;

import es.unex.sextante.outputs.FileOutputChannel;

import es.unex.sextante.outputs.IOutputChannel;

import org.gvsig.andami.Utilities;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.project.ProjectPreferences;

import org.gvsig.geoprocess.lib.sextante.dataObjects.FileTools;

import org.gvsig.geoprocess.lib.sextante.dataObjects.gvRasterLayerWrite;

import org.gvsig.geoprocess.lib.sextante.dataObjects.gvTable;

import org.gvsig.geoprocess.lib.sextante.dataObjects.gvVectorLayer;

import org.gvsig.geoprocess.lib.sextante.outputs.CompositeSourceOutputChannel;

/**

 * An OutputFactory based on the gvSIG data model. Supports only file-based

 * outputs.

 * 

 * @author volaya

 * 

 */

public class gvOutputFactory extends OutputFactory {

    @Override

    public IVectorLayer getNewVectorLayer(final String name,

        final int shapeType, final Class[] types, final String[] fields,

        final IOutputChannel channel, final Object crs)

        throws UnsupportedOutputChannelException {

        final int[] sizes = new int[types.length];

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

            sizes[i] = getSizeFromTypeClass(types[i]);

        }

        return getNewVectorLayer(name, shapeType, types, fields, channel, crs,

            sizes);

    }

    @Override

    public IVectorLayer getNewVectorLayer(final String sName,

        final int iShapeType, final Class[] types, final String[] sFields,

        final IOutputChannel channel, final Object crs, final int[] fieldSize)

        throws UnsupportedOutputChannelException {

        if (channel != null) {

            if (channel instanceof CompositeSourceOutputChannel) {

                gvVectorLayer layer = new gvVectorLayer();

                Object inputParams =

                    ((CompositeSourceOutputChannel) channel).getParameters();

                layer.create(sName, inputParams, iShapeType, types, sFields,

                    crs, fieldSize);

                return layer;

            } else

                if (channel instanceof FileOutputChannel) {

                    gvVectorLayer layer = new gvVectorLayer();

                    String fileName =

                        ((FileOutputChannel) channel).getFilename();

                    layer.create(sName, fileName, iShapeType, types, sFields,

                        crs, fieldSize);

                    return layer;

                }

        }

        throw new UnsupportedOutputChannelException();

    }

    @Override

    public IRasterLayer getNewRasterLayer(final String sName,

        final int iDataType, final AnalysisExtent extent, final int iBands,

        final IOutputChannel channel, final Object crs)

        throws UnsupportedOutputChannelException {

        if (channel instanceof FileOutputChannel) {

            final String sFilename =

                ((FileOutputChannel) channel).getFilename();

            final gvRasterLayerWrite layer = new gvRasterLayerWrite();

            layer.create(sName, sFilename, extent, iDataType, iBands, crs);

            return layer;

        } else {

            throw new UnsupportedOutputChannelException();

        }

    }

    @Override

    public ITable getNewTable(final String sName, final Class types[],

        final String[] sFields, final IOutputChannel channel)

        throws UnsupportedOutputChannelException {

        if (channel instanceof FileOutputChannel) {

            final String sFilename =

                ((FileOutputChannel) channel).getFilename();

            final gvTable table = new gvTable();

            table.create(sName, sFilename, types, sFields);

            return table;

        } else {

            throw new UnsupportedOutputChannelException();

        }

    }

    @Override

    public String getTempFolder() {

        return Utilities.createTempDirectory();

    }

    @Override

    public String[] getRasterLayerOutputExtensions() {

        return FileTools.RASTER_EXT_IN;

    }

    @Override

    public String[] getVectorLayerOutputExtensions() {

        return new String[] { "shp", "dxf" };

    }

    @Override

    public String[] getTableOutputExtensions() {

        return new String[] { "dbf" };

    }

    public void addMessage(final String s) {

        NotificationManager.addInfo(s, null);

    }

    @Override

    public ITaskMonitor getTaskMonitor(final String sTitle,

        final boolean bDeterminate, final JDialog parent) {

        return new DefaultTaskMonitor(sTitle, bDeterminate, parent);

    }

    @Override

    public Object getDefaultCRS() {

        return ((ProjectPreferences) ApplicationLocator.getManager()

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

    }

    private int getSizeFromTypeClass(final Class<?> type) {

        if (type.equals(Integer.class)) {

            return 10;

        } else

            if (type.equals(Long.class)) {

                return 20;

            } else

                if (type.equals(Double.class)) {

                    return 20;

                } else

                    if (type.equals(Float.class)) {

                        return 15;

                    } else

                        if (type.equals(String.class)) {

                            return 80;

                        } else

                            if (type.equals(Date.class)) {

                                return 10;

                            } else

                                if (type.equals(Boolean.class)) {

                                    return 1;

                                } else {

                                    return 20;

                                }

    }

}