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.

 */

/*

 * gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2010 Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.geoprocess.algorithm.xyshift;

import java.util.Iterator;

import java.util.List;

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

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

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.aggregate.MultiCurve;

import org.gvsig.fmap.geom.aggregate.MultiPoint;

import org.gvsig.fmap.geom.aggregate.MultiSurface;

import org.gvsig.fmap.geom.exception.CreateGeometryException;

import org.gvsig.fmap.geom.primitive.Curve;

import org.gvsig.fmap.geom.primitive.GeneralPathX;

import org.gvsig.fmap.geom.primitive.OrientablePrimitive;

import org.gvsig.fmap.geom.primitive.Point;

import org.gvsig.fmap.geom.primitive.Surface;

import org.gvsig.geoprocess.algorithm.base.core.GeometryOperation;

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

import es.unex.sextante.core.Sextante;

/**

 * Translates all geometries

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

@SuppressWarnings("deprecation")

public class XYShiftOperation extends GeometryOperation {

        private double                           x                = 0D;

        private double                           y                = 0D;

        private GeometryManager                  geometryManager  = null;

        public XYShiftOperation(double x, double y, AbstractSextanteGeoProcess p) {

                super(p);

                this.x = x;

                this.y = y;

                this.geometryManager = GeometryLocator.getGeometryManager();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.geoprocess.algorithm.base.core.GeometryOperation#invoke(org.gvsig.fmap.geom.Geometry, org.gvsig.fmap.dal.feature.Feature)

         */

        @SuppressWarnings("unchecked")

        public EditableFeature invoke(org.gvsig.fmap.geom.Geometry g, Feature feature) {

                List geomList = feature.getGeometries();

                try {

                        if(!persister.isCompatibleType(feature.getDefaultGeometry()))

                                return lastEditFeature;

                        if(geomList == null) {

                                org.gvsig.fmap.geom.Geometry oldGeom = feature.getDefaultGeometry();

                                Geometry newGeom = shiftGeometry(oldGeom);

                                lastEditFeature = persister.addFeature(feature, newGeom);

                        } else {

                                Iterator<org.gvsig.fmap.geom.Geometry> itGeom = geomList.iterator();

                                EditableFeature editFeat = null;

                                int nGeom = 0;

                                while(itGeom.hasNext()) {

                                        org.gvsig.fmap.geom.Geometry oldGeom = itGeom.next();

                                        Geometry newGeom = shiftGeometry(oldGeom);

                                        nGeom ++;

                                        if(editFeat == null) 

                                                editFeat = persister.addFeature(feature, newGeom);

                                        else 

                                                persister.addGeometryToExistingFeature(editFeat, newGeom);

                                        lastEditFeature = editFeat;

                                }

                        }

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                } catch (CreateGeometryException e1) {

                        Sextante.addErrorToLog(e1);

                }

                return lastEditFeature;

        }

        /**

         * Shifts all coordinates in a geometry

         * @param oldGeom

         * @return

         * @throws CreateGeometryException

         */

        private Geometry shiftGeometry(Geometry oldGeom) throws CreateGeometryException {

                if(oldGeom.getType() == Geometry.TYPES.SURFACE)

                        return shiftSurface((Surface)oldGeom);

                else if(oldGeom.getType() == Geometry.TYPES.MULTISURFACE)

                        return shiftMultiSurface((MultiSurface)oldGeom);

                else if(oldGeom.getType() == Geometry.TYPES.CURVE) 

                        return shiftCurve((Curve)oldGeom);

                else if(oldGeom.getType() == Geometry.TYPES.MULTICURVE)

                        return shiftMultiCurve((MultiCurve)oldGeom);

                else if(oldGeom.getType() == Geometry.TYPES.POINT) 

                        return shiftPoint((Point)oldGeom);

                else if(oldGeom.getType() == Geometry.TYPES.MULTIPOINT)

                        return shiftMultiPoint((MultiPoint)oldGeom);

                return oldGeom;

        }

        /**

         * Shifts all coordinates in a surface

         * @param geom

         * @return

         * @throws CreateGeometryException

         */

        private Surface shiftSurface(Surface geom) throws CreateGeometryException {

                Surface surface = (Surface)geometryManager.create(Geometry.TYPES.SURFACE, Geometry.SUBTYPES.GEOM2D);

                return (Surface)shiftGeom(geom, surface);

        }

        /**

         * Shifts all coordinates in a multisurface

         * @param geom

         * @return

         * @throws CreateGeometryException

         */

        private MultiSurface shiftMultiSurface(MultiSurface geom) throws CreateGeometryException {

                MultiSurface surface = (MultiSurface)geometryManager.create(Geometry.TYPES.MULTISURFACE, geom.getGeometryType().getSubType());

                for (int i = 0; i < geom.getPrimitivesNumber(); i++) {

                        Surface s = geom.getSurfaceAt(i);

                        Surface newSurface = shiftSurface(s);

                        surface.addSurface(newSurface);

                }

                return surface;

        }

        /**

         * Shifts all coordinates in a curve

         * @param geom

         * @return

         * @throws CreateGeometryException

         */

        private Curve shiftCurve(Curve geom) throws CreateGeometryException {

                Curve curve = (Curve)geometryManager.create(Geometry.TYPES.CURVE, Geometry.SUBTYPES.GEOM2D);

                return (Curve)shiftGeom(geom, curve);

        }

        private OrientablePrimitive shiftGeom(OrientablePrimitive src, OrientablePrimitive dst) throws CreateGeometryException {

                GeneralPathX gp = (GeneralPathX)src.getGeneralPath().clone();

                double[] pcoords = gp.getPointCoords();

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

                        if((i % 2) == 0)

                                pcoords[i] = pcoords[i] + x;

                        else 

                                pcoords[i] = pcoords[i] + y;

                }

                gp.setPointCoords(pcoords);

                dst.setGeneralPath(gp);

                return dst;

        }

        /**

         * Shifts all coordinates in a multicurve

         * @param geom

         * @return

         * @throws CreateGeometryException

         */

        private MultiCurve shiftMultiCurve(MultiCurve geom) throws CreateGeometryException {

                MultiCurve curve = (MultiCurve)geometryManager.create(Geometry.TYPES.MULTICURVE, geom.getGeometryType().getSubType());

                for (int i = 0; i < geom.getPrimitivesNumber(); i++) {

                        Curve c = geom.getCurveAt(i);

                        Curve newCurve = shiftCurve(c);

                        curve.addCurve(newCurve);

                }

                return curve;

        }

        /**

         * Shifts a point

         * @param geom

         * @return

         * @throws CreateGeometryException

         */

        private Point shiftPoint(Point geom) throws CreateGeometryException {

                Point point = (Point)geometryManager.create(Geometry.TYPES.POINT, geom.getGeometryType().getSubType());

                point.setX(geom.getX() + x);

                point.setY(geom.getY() + y);

                return point;

        }

        /**

         * Shifts all coordinates in a multisurface

         * @param geom

         * @return

         * @throws CreateGeometryException

         */

        private MultiPoint shiftMultiPoint(MultiPoint geom) throws CreateGeometryException {

                MultiPoint point = (MultiPoint)geometryManager.create(Geometry.TYPES.MULTIPOINT, geom.getGeometryType().getSubType());

                for (int i = 0; i < geom.getPrimitivesNumber(); i++) {

                        Point p = geom.getPointAt(i);

                        Point newPoint = shiftPoint(p);

                        point.addPoint(newPoint);

                }

                return point;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.geoprocess.algorithm.base.core.GeometryOperation#invoke(org.gvsig.fmap.geom.Geometry, org.gvsig.fmap.dal.feature.EditableFeature)

         */

        public void invoke(org.gvsig.fmap.geom.Geometry g, EditableFeature feature) {

        }

}