Application: gvSIG desktop » gvSIG tools: Vector editing

/**

 * 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.

 */

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

import java.awt.geom.AffineTransform;

import java.text.DecimalFormat;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

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.GeometryUtils;

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

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

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

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

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

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

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

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.dispose.DisposableIterator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.exception.BaseException;

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

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.api.exceptions.StopServiceException;

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;

/**

 * @author llmarques

 *

 */

public class DuplicateEditingProvider extends AbstractEditingProvider implements

    EditingProvider {

    private final EditingServiceParameter selection;

    private final EditingServiceParameter firstPoint;

    private final EditingServiceParameter secondPoint;

    private final EditingServiceParameter equidistantCopies;

    private final FeatureStore featureStore;

    private Map<EditingServiceParameter, Object> values;

    /**

     * Default constructor.

     *

     * @param services

     *            available services for this provider

     * @param parameters

     *            of this provider

     */

    public DuplicateEditingProvider(DynObject parameters,

        ProviderServices services) {

        super(services);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selection =

            new DefaultEditingServiceParameter("selection", "selection",

                TYPE.SELECTION);

        this.firstPoint =

            new DefaultEditingServiceParameter("first_point", "first_point",

                TYPE.POSITION);

        this.secondPoint =

            new DefaultEditingServiceParameter("second_point", "second_point",

                TYPE.POSITION);

        this.equidistantCopies = new DefaultEditingServiceParameter("equidistant_copies", "equidistant_copies", true, TYPE.VALUE);

        this.equidistantCopies.setDefaultValue(1);

    }

    @Override

    public EditingServiceParameter next() {

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

            return selection;

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

            return firstPoint;

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

            return secondPoint;

        }

        return null;

    }

    @Override

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        EditingProviderManager editingProviderManager =

            EditingProviderLocator.getProviderManager();

        EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        int subtype;

        try {

            subtype = editingProviderServices.getSubType(featureStore);

        } catch (DataException e2) {

            throw new DrawServiceException(e2);

        }

        ISymbol lineSymbolEditing = editingProviderManager.getSymbol("line-symbol-editing");

        ISymbol auxiliarylineSymbolEditing = editingProviderManager.getSymbol("auxiliary-line-symbol-editing");

        ISymbol polygonSymbolEditing = editingProviderManager.getSymbol("polygon-symbol-editing");

        ISymbol auxiliaryPointSymbolEditing = editingProviderManager.getSymbol("auxiliary-point-symbol-editing");

        if (values != null) {

            FeatureSelection featureSelection =

                (FeatureSelection) values.get(selection);

            Point firstPointValue = (Point) values.get(firstPoint);

            Number equidistantCopiesNumberValue = ((Number) values.get(this.equidistantCopies));

            int equidistantCopiesValue = equidistantCopiesNumberValue != null ? equidistantCopiesNumberValue.intValue() : ((Number)this.equidistantCopies.getDefaultValue()).intValue();

            if (featureSelection != null && firstPointValue != null) {

                drawingStatus.addStatus(firstPointValue, auxiliaryPointSymbolEditing, "");

                Point offset = GeometryUtils.createPoint(

                        mousePosition.getX() - firstPointValue.getX(),

                        mousePosition.getY() - firstPointValue.getY()

                );

                DisposableIterator it = null;

                try {

                    it = featureSelection.fastIterator();

                    Point pos = null;

                    while (it.hasNext()) {

                        Feature feature = (Feature) it.next();

                        ISymbol previewSymbol = this.getPreviewSymbol(feature);

                        Point antPos = firstPointValue;

                        for (int c = 1; c <= equidistantCopiesValue; c++) {

                            pos = GeometryUtils.createPoint(

                                    firstPointValue.getX() + (c * offset.getX()),

                                    firstPointValue.getY() + (c * offset.getY())

                            );

                            Geometry geometry

                                    = feature.getDefaultGeometry().cloneGeometry();

                            AffineTransform at

                                    = getMoveAffineTransform(firstPointValue,

                                            pos);

                            geometry.transform(at);

                            ISymbol symbol = null;

                            if (geometry instanceof Curve || geometry instanceof MultiCurve) {

                                symbol = lineSymbolEditing;

                            } else if (geometry instanceof Surface || geometry instanceof MultiSurface) {

                                symbol = polygonSymbolEditing;

                            } else if (geometry instanceof Point || geometry instanceof MultiPoint) {

                                symbol = auxiliaryPointSymbolEditing;

                            }

                            if (geometry instanceof Aggregate) {

                                int primitivesNumber = ((Aggregate) geometry).getPrimitivesNumber();

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

                                    drawingStatus.addStatus(((Aggregate) geometry).getPrimitiveAt(i), symbol, "");

                                    drawingStatus.addStatus(((Aggregate) geometry).getPrimitiveAt(i), previewSymbol, "");

                                }

                            } else {

                                drawingStatus.addStatus(geometry, symbol, "");

                                drawingStatus.addStatus(geometry, previewSymbol, "");

                            }

                            antPos = pos;

                        }

                    }

                    if(pos != null) {

                        Line auxLine = GeometryUtils.createLine(firstPointValue, pos, subtype);

                        drawingStatus.addStatus(auxLine, auxiliarylineSymbolEditing, "");

                    }

                    Point pointText = GeometryUtils.createPoint(

                                firstPointValue.getX() + (0.5 * offset.getX()),

                                firstPointValue.getY() + (0.5 * offset.getY())

                        );

                    ISimpleTextSymbol textSymbol = getTextSymbol();

                    drawingStatus.addStatus(

                            pointText, 

                            textSymbol, 

                            new DecimalFormat("#.0#").format(Math.sqrt(Math.pow(offset.getX(),2)+Math.pow(offset.getY(),2)))

                    );

                } catch (BaseException e) {

                    throw new DrawServiceException(e);

                } finally {

                    DisposeUtils.disposeQuietly(it);

                }

            }

        }

        return drawingStatus;

    }

    private AffineTransform getMoveAffineTransform(Point p1, Point p2)

        throws GeometryOperationNotSupportedException,

        GeometryOperationException {

        AffineTransform translate =

            AffineTransform.getTranslateInstance(p2.getX() - p1.getX(),

                p2.getY() - p1.getY());

        return translate;

    }

    @Override

    public void stop() throws StopServiceException {

        if (values != null) {

            values.clear();

        }

    }

    @Override

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<>();

        parameters.add(selection);

        parameters.add(firstPoint);

        parameters.add(secondPoint);

        parameters.add(equidistantCopies);

        return parameters;

    }

    @Override

    public void setValue(EditingServiceParameter parameter, Object value) throws InvalidEntryException {

        validateAndInsertValue(parameter, value);

    }

    @Override

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter parameter = next();

        validateAndInsertValue(parameter, value);

    }

    private void validateAndInsertValue(EditingServiceParameter parameter,

        Object value) {

        if (parameter == selection) {

            if (value instanceof FeatureSelection) {

                if (((FeatureSelection) value).getSelectedCount() > 0) {

                    values.put(selection, value);

                }

            }

        } else if (parameter == firstPoint || parameter == secondPoint) {

            if (value instanceof Point) {

                values.put(parameter, value);

            }

        } else if (parameter == equidistantCopies) {

            if (value instanceof Number) {

                int intValue = ((Number) value).intValue();

                if(intValue >= 1) {

                    values.put(parameter, ((Number) value).intValue());

                }

            }

        }

    }

    @Override

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    @Override

    public void finishAndStore() throws FinishServiceException {

        if (values != null) {

            final Point p1 = (Point) values.get(firstPoint);

            final Point p2 = (Point) values.get(secondPoint);

            Number equidistantCopiesNumberValue = ((Number) values.get(this.equidistantCopies));

            int equidistantCopiesValue = equidistantCopiesNumberValue != null ? equidistantCopiesNumberValue.intValue() : ((Number)this.equidistantCopies.getDefaultValue()).intValue();

            FeatureSelection featureSelection =

                (FeatureSelection) values.get(selection);

//            ToolsLocator.getDisposableManager().bind(featureSelection);

            try {

                featureSelection.accept((Object obj) -> {

                    Feature feature = (Feature) obj;

                    EditingProviderServices editingProviderServices =

                            (EditingProviderServices) getProviderServices();

                    Point offset = GeometryUtils.createPoint(

                            p2.getX() - p1.getX(),

                            p2.getY() - p1.getY()

                    );

                    for(int c = 1; c <= equidistantCopiesValue; c++){

                        EditableFeature editable =

                                editingProviderServices.getFeatureCopyWithoutUniqueIndex(

                                        featureStore, feature);

                        Geometry geometry =

                                feature.getDefaultGeometry().cloneGeometry();

                        Point pos = GeometryUtils.createPoint(

                                p1.getX() + (c * offset.getX()),

                                p1.getY() + (c * offset.getY())

                        );

                        geometry.transform(getMoveAffineTransform(p1, pos));

                        editable.setDefaultGeometry(geometry);

                        editingProviderServices.insertFeatureIntoFeatureStore(

                                editable, featureStore);

                    }

                });

//                featureStore.getFeatureSelection().deselectAll();

//                featureSelection.dispose();

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            }

        }

    }

    @Override

    public void start() throws StartServiceException, InvalidEntryException {

        values = new HashMap<>();

        FeatureSelection selected = null;

        if (featureStore != null) {

            try {

                selected = featureStore.getFeatureSelection();

            } catch (DataException e) {

                throw new StartServiceException(e);

            }

            try {

                setValue(selected);

            } catch (InvalidEntryException e) {

                throw new InvalidEntryException(e);

            }

        }

    }

    @Override

    public void restart() throws StartServiceException, InvalidEntryException, StopServiceException {

        values.put(secondPoint,null);

    }

    @Override

    public String getName() {

        return DuplicateEditingProviderFactory.PROVIDER_NAME;

    }

    @Override

    public Object getValue(EditingServiceParameter parameter) {

        return values!=null?values.get(parameter):null;

    }

    private ISimpleTextSymbol getTextSymbol(){

        SymbologyManager symbologyManager = SymbologyLocator.getSymbologyManager();

        ISimpleTextSymbol textSymbol = symbologyManager.createSimpleTextSymbol();

        textSymbol.setFontSize(10);

        return textSymbol;

    }

    @Override

    public boolean isEnabled(EditingServiceParameter parameter) {

        if (parameter == equidistantCopies) {

            return true;

        }

        return true;

    }

}