Application: gvSIG desktop

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 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.fmap.geom.jts.primitive.curve.spline;

import java.util.ArrayList;

import com.vividsolutions.jts.geom.Coordinate;

import com.vividsolutions.jts.geom.CoordinateSequence;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryException;

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

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

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

import org.gvsig.fmap.geom.jts.aggregate.MultiLine3DM;

import org.gvsig.fmap.geom.jts.aggregate.MultiPoint3DM;

import org.gvsig.fmap.geom.jts.aggregate.MultiPolygon3DM;

import org.gvsig.fmap.geom.jts.primitive.curve.line.Line3DM;

import org.gvsig.fmap.geom.jts.primitive.curve.spline.AbstractSpline.Spline;

import org.gvsig.fmap.geom.jts.primitive.point.Point3D;

import org.gvsig.fmap.geom.jts.primitive.point.Point3DM;

import org.gvsig.fmap.geom.jts.primitive.surface.polygon.Polygon3DM;

import org.gvsig.fmap.geom.jts.util.ArrayListCoordinateSequence;

import org.gvsig.fmap.geom.jts.util.JTSUtils;

import org.gvsig.fmap.geom.jts.util.ReadOnlyCoordinates;

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

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

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

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

/**

 * @author fdiaz

 *

 */

public abstract class BaseSpline3DM extends AbstractSpline {

    /**

     *

     */

    private static final long serialVersionUID = -5436087623278868159L;

    /**

     * @param type

     */

    public BaseSpline3DM(int type) {

        super(type, Geometry.SUBTYPES.GEOM3DM);

        this.coordinates = new ArrayListCoordinateSequence(new ArrayList<>());

    }

    /**

     * @param type

     * @param coordinates

     */

    public BaseSpline3DM(int type, Coordinate[] coordinates) {

        super(type, Geometry.SUBTYPES.GEOM3DM);

        this.coordinates = new ArrayListCoordinateSequence(new ReadOnlyCoordinates(coordinates));

        if (coordinates.length < 1) {

            anyVertex = new Point3DM(0, 0, 0, 0);

        } else {

            Coordinate coordinate = coordinates[0];

            anyVertex = new Point3DM(coordinate.x, coordinate.y, coordinate.z, coordinate.getOrdinate(CoordinateSequence.M));

        }

    }

    public BaseSpline3DM(int type, ArrayListCoordinateSequence coordinates) {

        super(type, Geometry.SUBTYPES.GEOM3DM);

        this.coordinates = coordinates;

    }

    public OrientablePrimitive addVertex(double x, double y) {

        return this.addVertex(new Point3DM(x, y, 0, 0));

    }

    public OrientablePrimitive addVertex(double x, double y, double z) {

        return this.addVertex(new Point3DM(x, y, z, 0));

    }

    @Override

    protected Point fixPoint(Point point) {

        if (point instanceof Point3DM) {

            return point;

        } else if (point instanceof Point3D) {

            return new Point3DM(point.getX(), point.getY(), ((Point3D)point).getZ(), 0);

        } else {

            return new Point3DM(point.getX(), point.getY(), 0, 0);

        }

    }

    @Override

    protected ArrayListCoordinateSequence getSplineCoordinates() {

        ArrayListCoordinateSequence splineCoordinates = new ArrayListCoordinateSequence();

        if (splineCoordinates == null || splineCoordinates.size() == 0) {

            int num = coordinates.size();

            double[] px = new double[num];

            double[] py = new double[num];

            double[] pz = new double[num];

            double[] pm = new double[num];

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

                Coordinate coord = coordinates.get(i);

                px[i] = coord.x;

                py[i] = coord.y;

                pz[i] = coord.z;

                pm[i] = coord.getOrdinate(CoordinateSequence.M);

            }

            Spline splineX = new Spline(px);

            Spline splineY = new Spline(py);

            Spline splineZ = new Spline(pz);

            Spline splineM = new Spline(pm);

            splineCoordinates.add(coordinates.get(0));

            for (int i = 0; i < coordinates.size() - 1; i++) {

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

                    if ((t == SUBSEGMENTS) && (i == (coordinates.size() - 2))) {

                        // We don't calculate the last point to avoid a possible

                        // error precision with floating point numbers.

                        splineCoordinates.add(JTSUtils.createMCoordinate(px[px.length - 1], py[px.length - 1], pz[px.length - 1], pm[px.length - 1]));

                    } else {

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

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

                        double z1 = splineZ.fn(i, ((double) t) / SUBSEGMENTS);

                        double m1 = splineM.fn(i, ((double) t) / SUBSEGMENTS);

                        splineCoordinates.add(JTSUtils.createMCoordinate(x1, y1, z1,m1));

                    }

                }

            }

        }

        return splineCoordinates;

    }

    @Override

    public MultiPoint toPoints() throws GeometryException {

        MultiPoint multiPoint = new MultiPoint3DM();

        Coordinate[] theCoordinates = getJTS().getCoordinates();

        multiPoint.ensureCapacity(theCoordinates.length);

        for (Coordinate theCoordinate : theCoordinates) {

            multiPoint.addPoint(new Point3DM(this.getProjection(), theCoordinate));

        }

        return multiPoint;

    }

    @Override

    public MultiLine toLines() throws GeometryException {

        MultiLine multiLine = new MultiLine3DM();

        Line line = new Line3DM(getJTS().getCoordinates());

        multiLine.addPrimitive(line);

        return multiLine;

    }

    @Override

    public MultiPolygon toPolygons() throws GeometryException {

        MultiPolygon multiPolygon = new MultiPolygon3DM();

        Polygon polygon = new Polygon3DM(getJTS().getCoordinates());

        multiPolygon.addPrimitive(polygon);

        return multiPolygon;

    }

    public Point getVertex(int index) {

        Point3DM vertex = new Point3DM(this.getProjection(), this.coordinates.get(index));

        anyVertex = vertex;

        return vertex;

    }

}