Application: gvSIG desktop » gvSIG geoprocess

/*

 * 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.sextante.app.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.exception.CreateGeometryException;

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

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

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

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

import org.gvsig.sextante.app.algorithm.base.core.GeometryOperation;

import es.unex.sextante.core.Sextante;

/**

 * Translates all geometries

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

 */

public class XYShiftOperation extends GeometryOperation {

        private double                           x                = 0D;

        private double                           y                = 0D;

        private GeometryManager                  geometryManager  = null;

        public XYShiftOperation(double x, double y) {

                this.x = x;

                this.y = y;

                this.geometryManager = GeometryLocator.getGeometryManager();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.sextante.app.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(geomList == null) {

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

                                Geometry newGeom = shiftGeometry(oldGeom);

                                EditableFeature editFeat = feature.getEditable();

                                editFeat.setDefaultGeometry(newGeom);

                                persister.addFeature(feature, newGeom);

                        } else {

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

                                EditableFeature editFeat = null;

                                boolean first = true;

                                int nGeom = 0;

                                while(itGeom.hasNext()) {

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

                                        Geometry newGeom = shiftGeometry(oldGeom);

                                        editFeat = feature.getEditable();

                                        editFeat.setGeometry(nGeom, newGeom);

                                        nGeom ++;

                                        if(first) 

                                                persister.addFeature(editFeat, newGeom);

                                        else 

                                                persister.addGeometryToExistingFeature(editFeat, newGeom);

                                }

                        }

                } 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 {

                Geometry newGeom = null;

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

                        newGeom = shiftSurface((Surface)oldGeom);

                if(oldGeom.getType() == Geometry.TYPES.CURVE || oldGeom.getType() == Geometry.TYPES.MULTICURVE) 

                        newGeom = shiftCurve((Curve)oldGeom);

                if(oldGeom.getType() == Geometry.TYPES.POINT || oldGeom.getType() == Geometry.TYPES.MULTIPOINT) 

                        newGeom = shiftPoint((Point)oldGeom);

                return newGeom;

        }

        /**

         * 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, geom.getGeometryType().getSubType());

                GeneralPathX path = geom.getGeneralPath();

                double[] coords = path.getPointCoords();

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

                        if((coords[i] % 2) == 0) 

                                coords[i] = coords[i] + x;

                        else

                                coords[i] = coords[i] + y;

                }

                path.setPointCoords(coords);

                surface.setGeneralPath(path);

                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, geom.getGeometryType().getSubType());

                GeneralPathX path = geom.getGeneralPath();

                double[] coords = path.getPointCoords();

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

                        if((coords[i] % 2) == 0) 

                                coords[i] = coords[i] + x;

                        else

                                coords[i] = coords[i] + y;

                }

                path.setPointCoords(coords);

                curve.setGeneralPath(path);

                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;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.sextante.app.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) {

        }

}