Application: gvSIG desktop

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

 *

 * Copyright (C) 2004 IVER T.I. and 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.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

package org.gvsig.fmap.geom.primitive;

import java.awt.geom.AffineTransform;

import java.awt.geom.Point2D;

import java.util.ArrayList;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.Geometry.SUBTYPES;

import org.gvsig.fmap.geom.handler.AbstractHandler;

import org.gvsig.fmap.geom.handler.FinalHandler;

import org.gvsig.fmap.geom.handler.Handler;

import org.gvsig.fmap.geom.type.GeometryType;

/**

 * Spline2D.

 *

 * @author Vicente Caballero Navarro

 */

public class Spline2D extends Curve2D implements Spline {

        private static final long serialVersionUID = 1L;

        private static GeometryType geomType = GeometryLocator.getGeometryManager()

        .registerGeometryType(Spline2D.class, null, TYPES.SPLINE,  SUBTYPES.GEOM2D);

        public static int CODE = geomType.getId();

        private ArrayList points;

        /**

         * Constructor without arguments. It is necessary to create

         * geometries using the {@link GeometryType}{@link #create()}

         * method

         */

        public Spline2D() {

                super();

                points = new ArrayList();

                gp = new GeneralPathX();

        }

        /**

         * Crea un nuevo FPolyline2D.

         *

         * @param gpx GeneralPathX.

         */

        public Spline2D(String id, IProjection projection, Point2D[] ps) {

                super(id, projection, getGeneralPathX(ps));

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

                        points.add(ps[i]);

                }

        }

        private static GeneralPathX getGeneralPathX(Point2D[] ps) {

                GeneralPathX gpx=new GeneralPathX();

                int num=ps.length;

                double[] px=new double[num];

            double[] py=new double[num];

            for (int i=0;i<num;i++) {

                    Point2D p=ps[i];

                    px[i]=p.getX();

                    py[i]=p.getY();

            }

            Spline splineX = new Spline(px);

        Spline splineY = new Spline(py);

        gpx.moveTo(px[0],py[0]);

        for (int i = 0; i < px.length - 1; i++) {

            for (int t = 1; t < 31; t++) {

                double x1 = splineX.fn(i, ((double) t) / 30.0);

                double y1 = splineY.fn(i, ((double) t) / 30.0);

                gpx.lineTo(x1,y1);

            }

        }

        if (ps[0].getX()==ps[ps.length-1].getX() && ps[0].getY()==ps[ps.length-1].getY())

                gpx.closePath();

                return gpx;

        }

        private static GeneralPathX getGeneralPathX(ArrayList ps) {

                Point2D[] _ps  = new Point2D[ps.size()];

                for (int i=0 ; i<ps.size() ; i++){

                        _ps[i] = (Point2D)ps.get(i);

                }

                return getGeneralPathX(_ps);

        }

        /**

         * @see org.gvsig.fmap.geom.primitive.FShape#getShapeType()

         */

        public int getShapeType() {

                return FShape.LINE;

        }

        /* (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.core.FShape#cloneFShape()

         */

        public FShape cloneFShape() {

                Spline2D curve = new Spline2D();

                for (int i=0;i<points.size();i++){

                        curve.setCoordinateAt(i, 0, ((Point)points.get(i)).getX());

                        curve.setCoordinateAt(i, 1, ((Point)points.get(i)).getY());

                }

                return (FShape)curve;

        }

        /* (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.core.FShape#getStretchingHandlers()

         */

        public Handler[] getStretchingHandlers() {

                ArrayList handlers = new ArrayList();

                for (int i=0;i<points.size();i++) {

                        handlers.add(new PointHandler(i, ((Point)points.get(i)).getX(), ((Point)points.get(i)).getY()));

                }

                return (Handler[]) handlers.toArray(new Handler[0]);

        }

        /* (non-Javadoc)

         * @see com.iver.cit.gvsig.fmap.core.FShape#getSelectHandlers()

         */

        public Handler[] getSelectHandlers() {

                ArrayList handlers = new ArrayList();

                for (int i=0;i<points.size();i++) {

                        Point2D p=((Point2D)points.get(i));

                        handlers.add(new PointSelHandler(i, p.getX(), p.getY()));

                }

                return (Handler[]) handlers.toArray(new Handler[0]);

        }

        /**

         * DOCUMENT ME!

         *

         * @author Vicente Caballero Navarro

         */

        class PointHandler extends AbstractHandler implements FinalHandler{

                /**

                 * Crea un nuevo PointHandler.

                 *

                 * @param x DOCUMENT ME!

                 * @param y DOCUMENT ME!

                 */

                public PointHandler(int i,double x, double y) {

                        point = new Point2D.Double(x,y);

                        index=i;

                }

                /**

                 * DOCUMENT ME!

                 *

                 * @param x DOCUMENT ME!

                 * @param y DOCUMENT ME!

                 *

                 * @return DOCUMENT ME!

                 */

                public void move(double x, double y) {

                        point.setLocation(point.getX()+x,point.getY()+y);

                        //TODO falta actualizar el GeneralPathX

                }

                /**

                 * @see org.gvsig.fmap.geom.handler.Handler#set(double, double)

                 */

                public void set(double x, double y) {

                        point.setLocation(x,y);

                        //TODO falta actualizar el GeneralPathX

                }

        }

        /**

         * DOCUMENT ME!

         *

         * @author Vicente Caballero Navarro

         */

        class PointSelHandler extends AbstractHandler implements FinalHandler{

                /**

                 * Crea un nuevo PointHandler.

                 *

                 * @param x DOCUMENT ME!

                 * @param y DOCUMENT ME!

                 */

                public PointSelHandler(int i,double x, double y) {

                        point = new Point2D.Double(x,y);

                        index=i;

                }

                /**

                 * DOCUMENT ME!

                 *

                 * @param x DOCUMENT ME!

                 * @param y DOCUMENT ME!

                 *

                 * @return DOCUMENT ME!

                 */

                public void move(double x, double y) {

                        point.setLocation(point.getX()+x,point.getY()+y);

                        ((Point)points.get(index)).setX(point.getX());

                        ((Point)points.get(index)).setY(point.getY());

                        gp = getGeneralPathX(points);

                }

                /**

                 * @see org.gvsig.fmap.geom.handler.Handler#set(double, double)

                 */

                public void set(double x, double y) {

                        point.setLocation(x,y);

                        ((Point)points.get(index)).setX(point.getX());

                        ((Point)points.get(index)).setY(point.getY());

                        gp=getGeneralPathX(points);

                }

        }

         static class Spline {

                    private double y[];

                    private double y2[];

                    /**

                     * The constructor calculates the second derivatives of the interpolating function

                     * at the tabulated points xi, with xi = (i, y[i]).

                     * Based on numerical recipes in C, http://www.library.cornell.edu/nr/bookcpdf/c3-3.pdf .

                     * @param y Array of y coordinates for cubic-spline interpolation.

                     */

                    public Spline(double y[]) {

                            this.y = y;

                            int n = y.length;

                            y2 = new double[n];

                            double u[] = new double[n];

                            for (int i = 1; i < n - 1; i++) {

                                    y2[i] = -1.0 / (4.0 + y2[i - 1]);

                                    u[i] = (6.0 * (y[i + 1] - 2.0 * y[i] + y[i - 1]) - u[i - 1]) / (4.0 + y2[i - 1]);

                            }

                            for (int i = n - 2; i >= 0; i--) {

                                    y2[i] = y2[i] * y2[i + 1] + u[i];

                            }

                    }

                    /**

                     * Returns a cubic-spline interpolated value y for the point between

                     * point (n, y[n]) and (n+1, y[n+1), with t ranging from 0 for (n, y[n])

                     * to 1 for (n+1, y[n+1]).

                     * @param n The start point.

                     * @param t The distance to the next point (0..1).

                     * @return A cubic-spline interpolated value.

                     */

                    public double fn(int n, double t) {

                            return t * y[n + 1] - ((t - 1.0) * t * ((t - 2.0) * y2[n] - (t + 1.0) * y2[n + 1])) / 6.0 + y[n] - t * y[n];

                    }

            }

         public void transform(AffineTransform at) {

                        for (int i=0;i<points.size();i++) {

                                Point2D p= (Point2D)points.get(i);

                                at.transform(p, p);

                        }

                        gp.transform(at);

                }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.geom.primitive.Curve2D#addPoint(org.gvsig.fmap.geom.primitive.Point)

         */

        public void addVertex(Point point) {

                points.add(new java.awt.geom.Point2D.Double(point.getX(), point.getY()));

                gp = getGeneralPathX(points);

        }

}