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.core;

import java.util.Date;

import javax.swing.JDialog;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.Utilities;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.andami.ui.mdiManager.IWindow;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.project.ProjectPreferences;

import org.gvsig.app.project.documents.view.gui.AbstractViewPanel;

import org.gvsig.fmap.geom.Geometry;

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

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

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

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

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

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

import es.unex.sextante.core.AnalysisExtent;

import es.unex.sextante.core.ITaskMonitor;

import es.unex.sextante.core.OutputFactory;

import es.unex.sextante.core.Sextante;

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 es.unex.sextante.outputs.NullOutputChannel;

import org.gvsig.raster.lib.buffer.api.exceptions.BufferException;

/**

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

 * outputs.

 *

 * @author volaya

 *

 */

public class DefaultOutputFactory extends OutputFactory {

    @Override

    public IVectorLayer getNewVectorLayer(final String name,

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

        final IOutputChannel channel, final Object crs, final int subtype)

        throws UnsupportedOutputChannelException {

        final int[] sizes = getSizesFromTypeClass(types);

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

            sizes, subtype);

    }

    @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 = getSizesFromTypeClass(types);

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

            sizes, Geometry.SUBTYPES.UNKNOWN);

    }

    private int[] getSizesFromTypeClass(final Class[] types) {

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

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

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

        }

        return sizes;

    }

    @Override

    public IVectorLayer getNewVectorLayer(final String sName,

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

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

        throws UnsupportedOutputChannelException {

        return getNewVectorLayer(sName, iShapeType, types, sFields,

        channel, crs, fieldSize, Geometry.SUBTYPES.UNKNOWN);

    }

    @Override

    public IVectorLayer getNewVectorLayer(final String sName,

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

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

        throws UnsupportedOutputChannelException {

        if( fieldSize==null ) {

            fieldSize = getSizesFromTypeClass(types);

        }

        if (channel != null) {

            if (channel instanceof CompositeSourceIOutputChannel) {

                FlyrVectIVectorLayer layer = new FlyrVectIVectorLayer();

                Object inputParams =

                    ((CompositeSourceIOutputChannel) channel).getParameters();

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

                    crs, fieldSize, subtype);

                return layer;

            } else if (channel instanceof FileOutputChannel) {

                FlyrVectIVectorLayer layer = new FlyrVectIVectorLayer();

                String fileName = ((FileOutputChannel) channel).getFilename();

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

                    fieldSize, subtype);

                return layer;

            } else if (channel instanceof NullOutputChannel) {

                return null;

            }

        }

        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 BufferWriteOnlyIRasterLayer layer = new BufferWriteOnlyIRasterLayer();

            try {

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

            } catch (BufferException e) {

                Sextante.addErrorToLog(e);            

            }

            return layer;

        } else {

            throw new UnsupportedOutputChannelException();

        }

    }

    public IRasterLayer getNewEmptyRORasterLayer(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 FLyrRasterIRasterLayer layer = new FLyrRasterIRasterLayer(sFilename);

                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 TableDocumentITable table = new TableDocumentITable();

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

            //Uses the active view

            IWindow window = PluginServices.getMDIManager().getActiveWindow();

            //If there is not active view then it looks for a view

            if(window == null || !(window instanceof AbstractViewPanel)) {

                    IWindow[] windowList = PluginServices.getMDIManager().getAllWindows();

                for (IWindow windowList1 : windowList) {

                    if (windowList1 instanceof AbstractViewPanel) {

                        window = windowList1;

                        break;

                    }

                }

            }

            if(window != null && window instanceof AbstractViewPanel) {

                    return ((AbstractViewPanel)window).getMapControl().getProjection();

            } else {

                    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;

                                }

    }

}