Application: gvSIG desktop » gvSIG tools: Vector editing

package org.gvsig.vectorediting.lib.prov.scale;

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2014 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.

 */

import java.awt.Font;

import java.awt.geom.AffineTransform;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import net.sf.antcontrib.net.httpclient.GetMethodTask;

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.dal.feature.FeatureSelection;

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

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.operation.GeometryOperationException;

import org.gvsig.fmap.geom.operation.GeometryOperationNotSupportedException;

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

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

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.ViewPort;

import org.gvsig.fmap.mapcontext.rendering.symbols.ISymbol;

import org.gvsig.symbology.SymbologyLocator;

import org.gvsig.symbology.SymbologyManager;

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.text.ISimpleTextSymbol;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.service.spi.ProviderServices;

import org.gvsig.tools.visitor.VisitCanceledException;

import org.gvsig.tools.visitor.Visitor;

import org.gvsig.vectorediting.lib.api.DrawingStatus;

import org.gvsig.vectorediting.lib.api.EditingServiceParameter;

import org.gvsig.vectorediting.lib.api.EditingServiceParameter.TYPE;

import org.gvsig.vectorediting.lib.api.exceptions.DrawServiceException;

import org.gvsig.vectorediting.lib.api.exceptions.FinishServiceException;

import org.gvsig.vectorediting.lib.api.exceptions.InvalidEntryException;

import org.gvsig.vectorediting.lib.api.exceptions.StartServiceException;

import org.gvsig.vectorediting.lib.spi.AbstractEditingProvider;

import org.gvsig.vectorediting.lib.spi.DefaultDrawingStatus;

import org.gvsig.vectorediting.lib.spi.DefaultEditingServiceParameter;

import org.gvsig.vectorediting.lib.spi.EditingProvider;

import org.gvsig.vectorediting.lib.spi.EditingProviderFactory;

import org.gvsig.vectorediting.lib.spi.EditingProviderLocator;

import org.gvsig.vectorediting.lib.spi.EditingProviderManager;

import org.gvsig.vectorediting.lib.spi.EditingProviderServices;

public class ScaleEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    private I18nManager i18nManager = ToolsLocator.getI18nManager();

    private EditingServiceParameter selectionParameter;

    private EditingServiceParameter originPointParameter;

    private EditingServiceParameter scaleFactorOrReferencePointParameter;

    private EditingServiceParameter secondScalePointParameter;

    private Map<EditingServiceParameter, Object> values;

    private FeatureStore featureStore;

    private MapContext mapContext;

    public ScaleEditingProvider(ProviderServices providerServices,

        DynObject parameters) {

        super(providerServices);

        this.mapContext =

            (MapContext) parameters

                .getDynValue(EditingProviderFactory.MAPCONTEXT_FIELD);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selectionParameter =

            new DefaultEditingServiceParameter("selection",

                i18nManager.getTranslation("selection"), TYPE.SELECTION);

        this.originPointParameter =

            new DefaultEditingServiceParameter("origin_point",

                i18nManager.getTranslation("origin_point"), TYPE.VALUE, TYPE.POSITION);

        this.scaleFactorOrReferencePointParameter =

            new DefaultEditingServiceParameter("scale_factor_or_reference_point",

                i18nManager.getTranslation("scale_factor_or_reference_point"), TYPE.VALUE, TYPE.POSITION);

        this.secondScalePointParameter =

            new DefaultEditingServiceParameter("second_scale_point",

                i18nManager.getTranslation("second_scale_point"), TYPE.POSITION);

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> list =

            new ArrayList<EditingServiceParameter>();

        list.add(selectionParameter);

        list.add(originPointParameter);

        list.add(secondScalePointParameter);

        return list;

    }

    private void validateAndInsertValue(EditingServiceParameter param,

        Object value) throws InvalidEntryException {

        if (param == selectionParameter) {

            if (value instanceof FeatureSelection) {

                values.put(param, value);

                return;

            }

        } else if (param == originPointParameter) {

            if (value instanceof Point) {

                values.put(param, value);

                return;

            }

        } else if (param == scaleFactorOrReferencePointParameter) {

            if (value instanceof Point) {

                if (((Point) value).getX() == ((Point)values.get(originPointParameter)).getX()){

                    throw new InvalidEntryException(null);

                }

                values.put(param, value);

                return;

            } else if (value instanceof Double) {

                if ((Double)value == 0.0){

                    throw new InvalidEntryException(null);

                }

                values.put(param, value);

                return;

            }

        } else if (param == secondScalePointParameter) {

            if (value instanceof Point) {

                values.put(param, value);

                return;

            }

        }

    }

    public EditingServiceParameter next() {

        if (values.get(selectionParameter) == null) {

            return this.selectionParameter;

        } else if (values.get(originPointParameter) == null) {

            return this.originPointParameter;

        } else if (values.get(scaleFactorOrReferencePointParameter) == null) {

            return this.scaleFactorOrReferencePointParameter;

        } else {

            if (values.get(scaleFactorOrReferencePointParameter) instanceof Point ) {

                if (values.get(secondScalePointParameter) == null) {

                    return this.secondScalePointParameter;

                }

            }

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        EditingProviderManager editingProviderManager =

            EditingProviderLocator.getProviderManager();

        ISymbol ruleMarkerSymbol = editingProviderManager.getSymbol("rule-marker-symbol");

        ViewPort vp = this.mapContext.getViewPort();

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        FeatureSelection selected =

            (FeatureSelection) values.get(this.selectionParameter);

        try {

            int subType =

                editingProviderServices

                .getSubType(featureStore);

            if ((selected != null) && !selected.isEmpty()) {

                Point origin = (Point) values.get(this.originPointParameter);

                if (origin != null) {

                    Object scaleFactorOrReferencePointValue =

                        values.get(this.scaleFactorOrReferencePointParameter);

                    Double scale = 1.0;

                    if (scaleFactorOrReferencePointValue != null) {

                        if (scaleFactorOrReferencePointValue instanceof Double) {

                            scale = (Double) scaleFactorOrReferencePointValue;

                        } else if (scaleFactorOrReferencePointValue instanceof Point) {

                            double unity = ((Point) scaleFactorOrReferencePointValue).getX()-origin.getX();

                            // Unidad de referencia

                            createRule(drawingStatus, origin, unity, unity, subType); //+(Math.signum(unity)*0.01), subType);

                            Object secondScalePointParameter =

                                values.get(this.secondScalePointParameter);

                            if (secondScalePointParameter != null

                                && secondScalePointParameter instanceof Point) {

                                double rawScale = ((Point) secondScalePointParameter).getX()-origin.getX();

                                scale = rawScale / unity;

                                createRule(drawingStatus, origin, unity, rawScale, subType);

                                Point marker = null;

                                marker = geometryManager.createPoint(origin.getX()+rawScale,origin.getY(), subType);

                                drawingStatus.addStatus(marker, ruleMarkerSymbol, "");

                                ISimpleTextSymbol textSymbol = getTextSymbol();

                                Point textPoint = geometryManager.createPoint(marker.getX()-vp.toMapDistance(10), marker.getY()-vp.toMapDistance(20), subType);

                                drawingStatus.addStatus(textPoint, textSymbol, String.valueOf(Math.round(scale*100.0)/100.0));

                            } else {

                                double rawScale = mousePosition.getX()-origin.getX();

                                scale = rawScale / unity;

                                createRule(drawingStatus, origin, unity, rawScale, subType);

                                Point marker = null;

                                marker = geometryManager.createPoint(origin.getX()+rawScale,origin.getY(), subType);

                                drawingStatus.addStatus(marker, ruleMarkerSymbol, "");

                                ISimpleTextSymbol textSymbol = getTextSymbol();

                                Point textPoint = geometryManager.createPoint(marker.getX()-vp.toMapDistance(10), marker.getY()-vp.toMapDistance(20), subType);

                                drawingStatus.addStatus(textPoint, textSymbol, String.valueOf(Math.round(scale*100.0)/100.0));

                                AffineTransform at;

                                try {

                                    at = getScaleAffineTransform(origin, scale);

                                } catch (Exception e) {

                                    throw new DrawServiceException(e);

                                }

                                applyTransformToDrawingStatus(drawingStatus, selected, at);

                            }

                        }

                    } else {

                        double unity = mousePosition.getX()-origin.getX();

                        createRule(drawingStatus, origin, unity, unity, subType); //+(Math.signum(unity)*0.01), subType);

                    }

                }

                return drawingStatus;

            }

        } catch (Exception e) {

            throw new DrawServiceException(e);

        }

        return null;

    }

    private Line createHorizontalLineFromAPointWithLenght(Point origin,

        double length, int subType) throws CreateGeometryException {

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        Line line = geometryManager.createLine(subType);

        line.addVertex(origin);

        Point p3 =

            geometryManager.createPoint(origin.getX() + length,

                origin.getY(), subType);

        line.addVertex(p3);

        return line;

    }

    private ISimpleTextSymbol getTextSymbol(){

        SymbologyManager symbologyManager = SymbologyLocator.getSymbologyManager();

        ISimpleTextSymbol textSymbol = symbologyManager.createSimpleTextSymbol();

        textSymbol.setFontSize(10);

        return textSymbol;

    }

    private void createRule(DefaultDrawingStatus drawingStatus, Point point, double unity, double length, int subType) throws CreateGeometryException {

        if(unity==0.0 || length==0.0){

            return;

        }

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        EditingProviderManager editingProviderManager =

            EditingProviderLocator.getProviderManager();

        ISymbol ruleUnitySymbol = editingProviderManager.getSymbol("rule-unity-symbol");

        ISymbol ruleDecimalSymbol = editingProviderManager.getSymbol("rule-decimal-symbol");

        ISymbol ruleAxisSymbol = editingProviderManager.getSymbol("rule-axis-symbol");

        ViewPort vp = this.mapContext.getViewPort();

        Line line =

            createHorizontalLineFromAPointWithLenght(point, length, subType);

        drawingStatus.addStatus(line, ruleAxisSymbol, "");

        double pos = 0.0;

        int i = 0;

        // Se le a?ade 0.01 del length a la condici?n de parada del bucle para evitar

        // que por problemas de precisi?n de los n?meros en punto flotante

        // no se termine de dibujar la regla

        while (Math.abs(pos)<=Math.abs(length)*1.01) {

            Point mark = null;

            mark = geometryManager.createPoint(point.getX()+pos,point.getY(), subType);

            if (i%10==0){

                drawingStatus.addStatus(mark, ruleUnitySymbol, "");

                ISimpleTextSymbol textSymbol = getTextSymbol();

                Point textPoint = geometryManager.createPoint(mark.getX()-vp.toMapDistance(3), mark.getY()-vp.toMapDistance(10), subType);

                drawingStatus.addStatus(textPoint, textSymbol, String.valueOf(((int)Math.signum(length*unity))*i/10));

            } else {

                drawingStatus.addStatus(mark, ruleDecimalSymbol, "");

            }

            pos += Math.signum(length)*Math.signum(unity)*unity/10;

            i++;

        }

    }

    private void applyTransformToDrawingStatus(

        final DefaultDrawingStatus geometries, FeatureSelection selected,

        final AffineTransform at) throws BaseException {

        selected.accept(new Visitor() {

            public void visit(Object obj) throws VisitCanceledException,

                BaseException {

                Feature feature = (Feature) obj;

                Geometry geom = feature.getDefaultGeometry().cloneGeometry();

                geom.transform(at);

                geometries.addGeometry(geom);

            }

        });

    }

    private AffineTransform getScaleAffineTransform(Point axisP1, Double scale)

        throws GeometryOperationNotSupportedException,

        GeometryOperationException {

        AffineTransform translate =

            AffineTransform

                .getTranslateInstance(-axisP1.getX(), -axisP1.getY());

        AffineTransform scaleTransform = AffineTransform.getScaleInstance(scale,scale);

        AffineTransform inverseTranslate =

            AffineTransform.getTranslateInstance(axisP1.getX(), axisP1.getY());

        AffineTransform at = new AffineTransform(translate);

        at.preConcatenate(scaleTransform);

        at.preConcatenate(inverseTranslate);

        return at;

    }

    public void stop() {

        values.clear();

    }

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter param = next();

        validateAndInsertValue(param, value);

    }

    public void finishAndStore() throws FinishServiceException {

        FeatureSelection selected =

            (FeatureSelection) values.get(selectionParameter);

        try {

            if (!selected.isEmpty()) {

                Point origin = (Point) values.get(this.originPointParameter);

                Double scale = null;

                if (origin != null) {

                    Object scaleFactorOrReferencePointValue =

                        values.get(this.scaleFactorOrReferencePointParameter);

                    if (scaleFactorOrReferencePointValue != null) {

                        if (scaleFactorOrReferencePointValue instanceof Double) {

                            scale = (Double) scaleFactorOrReferencePointValue;

                        } else if (scaleFactorOrReferencePointValue instanceof Point) {

                            Object secondScalePointParameter =

                                values.get(this.secondScalePointParameter);

                            if (secondScalePointParameter != null

                                && secondScalePointParameter instanceof Point) {

                                double unity = ((Point) scaleFactorOrReferencePointValue).getX()-origin.getX();

                                double rawScale = ((Point) secondScalePointParameter).getX()-origin.getX();

                                scale = rawScale / unity;

                            }

                        }

                    }

                }

                if ((origin != null) && (scale != null)) {

                    final AffineTransform at;

                    try {

                        at = getScaleAffineTransform(origin, scale);

                    } catch (GeometryOperationNotSupportedException e) {

                        throw new FinishServiceException(e);

                    } catch (GeometryOperationException e) {

                        throw new FinishServiceException(e);

                    }

                    try {

                        selected.accept(new Visitor() {

                            public void visit(Object obj)

                                throws VisitCanceledException, BaseException {

                                Feature feature = (Feature) obj;

                                Geometry geom =

                                    feature.getDefaultGeometry()

                                        .cloneGeometry();

                                geom.transform(at);

                                // Se sustituye la geometr?a original por la

                                // calculada

                                EditableFeature editableFeature =

                                    feature.getEditable();

                                editableFeature.setDefaultGeometry(geom);

                                ((EditingProviderServices) getProviderServices())

                                    .updateFeatureInFeatureStore(

                                        editableFeature, featureStore);

                            }

                        });

                    } catch (BaseException e) {

                        throw new FinishServiceException(e);

                    }

                    featureStore.getFeatureSelection().deselectAll();

                }

            }

        } catch (DataException e) {

            throw new FinishServiceException(e);

        }

    }

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    public void start() throws StartServiceException {

        this.values = new HashMap<EditingServiceParameter, Object>();

        FeatureSelection selected = null;

        if (featureStore != null) {

            try {

                selected =

                    (FeatureSelection) featureStore.getFeatureSelection()

                        .clone();

            } catch (DataException e) {

                throw new StartServiceException(e);

            } catch (CloneNotSupportedException e) {

                // Do nothing

            }

            if ((selected != null) && (selected.getSelectedCount() > 0)) {

                values.put(selectionParameter, selected);

            }

        }

    }

    public String getName() {

        return ScaleEditingProviderFactory.PROVIDER_NAME;

    }

}