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.algorithm.reproject;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.crs.CRSFactory;

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

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

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

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

import org.gvsig.geoprocess.lib.sextante.AbstractSextanteGeoProcess;

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

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.IVectorLayer;

import es.unex.sextante.exceptions.GeoAlgorithmExecutionException;

import es.unex.sextante.exceptions.RepeatedParameterNameException;

import es.unex.sextante.gui.algorithm.GeoAlgorithmParametersPanel;

import es.unex.sextante.outputs.OutputVectorLayer;

/**

 * Reproject algorithm

 * @author <a href="mailto:nachobrodin@gmail.com">Nacho Brodin</a>

 */

public class ReprojectAlgorithm extends AbstractSextanteGeoProcess {

        public static final String        RESULT_POL        = "RESULT_POL";

        public static final String        RESULT_POINT      = "RESULT_POINT";

        public static final String        RESULT_LINE       = "RESULT_LINE";

        public static final String        LAYER             = "LAYER";

        public static final String        SELECTED_GEOM     = "SELECTED_GEOM";

        public static final String        DST_PROJECTION    = "DST_PROJECTION";

        /*

         * (non-Javadoc)

         * @see es.unex.sextante.core.GeoAlgorithm#defineCharacteristics()

         */

        public void defineCharacteristics() {

        setName(getTranslation("Reproject"));

        setGroup(getTranslation("basic_vect_algorithms"));

        // setGeneratesUserDefinedRasterOutput(false);

                try {

                        m_Parameters.addInputVectorLayer(LAYER, 

                getTranslation("Input_layer"),

                                                                                                IVectorLayer.SHAPE_TYPE_WRONG, 

                                                                                                true);

                        m_Parameters.addBoolean(SELECTED_GEOM, 

                getTranslation("Selected_geometries"),

                                                                        false);

            m_Parameters

                .addString(DST_PROJECTION, getTranslation("Projection"));

            addOutputVectorLayer(RESULT_POL, getTranslation("Reproject_polygon"),

                                        OutputVectorLayer.SHAPE_TYPE_POLYGON);

                        addOutputVectorLayer(RESULT_LINE, getTranslation("Reproject_line"),

                                        OutputVectorLayer.SHAPE_TYPE_LINE);

                        addOutputVectorLayer(RESULT_POINT, getTranslation("Reproject_point"),

                                        OutputVectorLayer.SHAPE_TYPE_POINT);

                } catch (RepeatedParameterNameException e) {

                        Sextante.addErrorToLog(e);

                }

        }

        /*

         * (non-Javadoc)

         * @see es.unex.sextante.core.GeoAlgorithm#processAlgorithm()

         */

        public boolean processAlgorithm() throws GeoAlgorithmExecutionException {

                if(existsOutPutFile(RESULT_POL, 0) || 

                        existsOutPutFile(RESULT_LINE, 0) ||

                        existsOutPutFile(RESULT_POINT, 0)) {

                    throw new GeoAlgorithmExecutionException(getTranslation("file_exists"));

            }

                IVectorLayer layer = m_Parameters.getParameterValueAsVectorLayer(LAYER);

                boolean selectedGeom = m_Parameters.getParameterValueAsBoolean(SELECTED_GEOM);

                String dstProj = m_Parameters.getParameterValueAsString(DST_PROJECTION);

                FeatureStore storeLayer = null;

                if(layer instanceof FlyrVectIVectorLayer)

                        storeLayer = ((FlyrVectIVectorLayer)layer).getFeatureStore();

                else

                        return false;

                try {

                        FeatureSet features = null;

                        features = storeLayer.getFeatureSet();

                        FeatureType featureType = features.getDefaultFeatureType();

            /*FeatureStore outFeatStore =

                buildOutPutStore(featureType, layer.getShapeType(),

                    getTranslation("Reproject"), RESULT);

                        IProjection projOutput = CRSFactory.getCRS(dstProj);

                        ReprojectOperation operation = new ReprojectOperation(((IProjection)layer.getCRS()), projOutput, this);

            operation.setTaskStatus(getStatus());

                        operation.computesFeatureOperation(storeLayer, outFeatStore, attrNames, selectedGeom, true);*/

                        IProjection projOutput = CRSFactory.getCRS(dstProj);

                        ReprojectOperation operation = new ReprojectOperation(((IProjection)layer.getCRS()), projOutput, this);

            operation.setTaskStatus(getStatus());

            if (isPolygon(storeLayer) || isUndefined(storeLayer)) {

                                FeatureStore outFeatStore =

                                        buildOutPutStore(featureType, IVectorLayer.SHAPE_TYPE_POLYGON,

                                                        getTranslation("Reproject_polygon"), RESULT_POL);

                                operation.computesFeatureOperation(storeLayer, outFeatStore, attrNames, selectedGeom, true);

                        } else {

                                buildOutPutStore(featureType, IVectorLayer.SHAPE_TYPE_POLYGON,

                                                getTranslation("Null_polygon"), RESULT_POL);

                        }

                        if (isLine(storeLayer) || isUndefined(storeLayer)) {

                                FeatureStore outFeatStore =

                                        buildOutPutStore(featureType, IVectorLayer.SHAPE_TYPE_LINE,

                                                        getTranslation("Reproject_line"), RESULT_LINE);

                                operation.computesFeatureOperation(storeLayer, outFeatStore, attrNames, selectedGeom, true);

                        } else {

                                buildOutPutStore(featureType.getCopy(), IVectorLayer.SHAPE_TYPE_LINE,

                                                getTranslation("Null_line"), RESULT_LINE);

                        }

                        if (isPoint(storeLayer) || isUndefined(storeLayer)) {

                                FeatureStore outFeatStore =

                                        buildOutPutStore(featureType.getCopy(), IVectorLayer.SHAPE_TYPE_POINT,

                                                        getTranslation("Reproject_point"), RESULT_POINT);

                                operation.computesFeatureOperation(storeLayer, outFeatStore, attrNames, selectedGeom, true);

                        } else {

                                buildOutPutStore(featureType.getCopy(), IVectorLayer.SHAPE_TYPE_POINT,

                                                getTranslation("Null_point"), RESULT_POINT);

                        }

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                        return false;

                }

                if(getTaskMonitor().isCanceled())

                        return false;

                return true;

        }

    @Override

    public Class<? extends GeoAlgorithmParametersPanel> getCustomParametersPanelClass() {

        return ReprojectParametersPanel.class;

    }

}