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.util.ArrayList;

import java.util.HashMap;

import java.util.Iterator;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Map;

import java.util.logging.Level;

import java.util.logging.Logger;

import org.apache.commons.collections4.CollectionUtils;

import org.apache.commons.lang.BooleanUtils;

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.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.exception.CreateEnvelopeException;

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

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.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.i18n.I18nManager;

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 PasteFromClipboardEditingProvider extends AbstractEditingProvider implements

        EditingProvider {

    private final EditingServiceParameter clipboard;

    private final EditingServiceParameter location;

    private final EditingServiceParameter copyAlphanumericAttributes;

    private final FeatureStore featureStore;

    private Map<EditingServiceParameter, Object> values;

    private Point firstPoint;

    /**

     * Default constructor.

     *

     * @param services available services for this provider

     * @param parameters of this provider

     */

    public PasteFromClipboardEditingProvider(DynObject parameters,

            ProviderServices services) {

        super(services);

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        Map<String, String> options = new LinkedHashMap<>();

        options.put(i18nManager.getTranslation("_yes"), "true");

        options.put(i18nManager.getTranslation("_no"), "false");

        this.featureStore

                = (FeatureStore) parameters

                        .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.clipboard

                = new DefaultEditingServiceParameter("clipboard", "_Clipboard",

                        TYPE.CLIPBOARD);

        this.location

                = new DefaultEditingServiceParameter("location", "_Location",

                        TYPE.POSITION);

        this.copyAlphanumericAttributes =

                new DefaultEditingServiceParameter(

                    "copyAlphanumericAttributes", 

                    "_Copy_alphanumeric_attributes", 

                    options,

                    i18nManager.getTranslation("_yes"),

                    true,

                    TYPE.OPTION

                );

    }

    @Override

    public boolean isEnabled(EditingServiceParameter parameter) {

        return true;

    }

    @Override

    public EditingServiceParameter next() {

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

            return clipboard;

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

            return location;

//        } else if (values.get(copyAlphanumericAttributes) == null) {

//            return copyAlphanumericAttributes;

        }

        return null;

    }

    @Override

    public DrawingStatus getDrawingStatus(Point mousePosition)

            throws DrawServiceException {

        DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

        GeometryManager geometryManager = GeometryLocator.getGeometryManager();

        EditingProviderManager editingProviderManager

                = EditingProviderLocator.getProviderManager();

        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 polygonSymbolEditing = editingProviderManager.getSymbol("polygon-symbol-editing");

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

        if (values != null) {

            List<Feature> features = null;

            features = (List<Feature>) values.get(clipboard);

            if (features != null && firstPoint != null) {

                try {

                    AffineTransform at

                            = getMoveAffineTransform(firstPoint,

                                    mousePosition);

                    Iterator<Feature> it = features.iterator();

                    while (it.hasNext()) {

                        Geometry geometry = ((Feature) it.next()).getDefaultGeometry().cloneGeometry();

                        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, "");

                            }

                        } else {

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

                        }

                    }

                } catch (Exception ex) {

                    Logger.getLogger(PasteFromClipboardEditingProvider.class.getName()).log(Level.SEVERE, null, ex);

                }

            }

        }

        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(clipboard);

        parameters.add(location);

        parameters.add(copyAlphanumericAttributes);

        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 == clipboard) {

            if (value instanceof List) {

                List<Feature> features = (List<Feature>) value;

                if (CollectionUtils.isNotEmpty(features)) {

                    Iterator<Feature> it = features.iterator();

                    values.put(clipboard, features);

                    try {

                        Envelope envelope = getEnvelope(features);

                        firstPoint = null;

                        if (envelope != null) {

                            firstPoint = envelope.getLowerCorner();

                        }

                    } catch (CreateEnvelopeException ex) {

                    }

                }

            }

            if (values.get(clipboard) == null && value instanceof FeatureSelection) {

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

                    values.put(clipboard, value);

                }

            }

        } else if (parameter == location) {

            if (value == null) {

                values.put(parameter, firstPoint);

            }

            if (value instanceof Point) {

                values.put(parameter, value);

            }

        } else if (parameter == copyAlphanumericAttributes) {

            if (value instanceof String) {

                Boolean copy = BooleanUtils.toBooleanObject((String)value);

                if(copy == null){

                    I18nManager i18nManager = ToolsLocator.getI18nManager();

                    try {

                        copy = BooleanUtils.toBooleanObject(

                                (String)value, 

                                i18nManager.getTranslation("_yes"), 

                                i18nManager.getTranslation("_no"), 

                                "" 

                        );

                    } catch (Exception e) {

                        copy = Boolean.FALSE;

                    }

                }

                values.put(parameter, copy == null ? false : copy);

            }

        }

    }

    @Override

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    @Override

    public void finishAndStore() throws FinishServiceException {

        if (values != null) {

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

            List<EditableFeature> features = (List<EditableFeature>) values.get(clipboard);

            Boolean copyAttributes = (Boolean) values.get(copyAlphanumericAttributes);

            try {

                Iterator<EditableFeature> it = features.iterator();

                EditingProviderServices editingProviderServices = getProviderServices();

                AffineTransform at

                        = getMoveAffineTransform(firstPoint, p2);

                while (it.hasNext()) {

                    EditableFeature editableFeature = it.next();

                    Geometry geometry = editableFeature.getDefaultGeometry().cloneGeometry();

                    geometry.transform(at);

                    EditableFeature targetEditableFeature = null;

                    if(copyAttributes == null || copyAttributes){

                        targetEditableFeature = editableFeature;

                    } else {

                        targetEditableFeature = this.getProviderServices().createNewFeature(featureStore);

                    }

                    targetEditableFeature.setDefaultGeometry(geometry);

                    editingProviderServices.insertFeatureIntoFeatureStore(targetEditableFeature, featureStore);

                }

                featureStore.getFeatureSelection().deselectAll();

            } catch (Exception e) {

                throw new FinishServiceException(e);

            }

        }

    }

    public void start() throws StartServiceException, InvalidEntryException {

        values = new HashMap<>();

        EditingProviderManager manager = EditingProviderLocator.getProviderManager();

        List<EditableFeature> featuresFromClipBoard = manager.getFeaturesFromClipboard(this.getProviderServices(), featureStore);

        if (CollectionUtils.isNotEmpty(featuresFromClipBoard)) {

            try {

                setValue(featuresFromClipBoard);

            } catch (InvalidEntryException ex) {

            }

        }

        if(getValue(copyAlphanumericAttributes) == null){

            setValue(copyAlphanumericAttributes, "true");

        }

    }

    public String getName() {

        return PasteFromClipboardEditingProviderFactory.PROVIDER_NAME;

    }

    @Override

    public Object getValue(EditingServiceParameter parameter) {

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

    }

    private Envelope getEnvelope(List<Feature> features) throws CreateEnvelopeException {

        GeometryManager geomManager = GeometryLocator.getGeometryManager();

        Envelope envelope = geomManager.createEnvelope(GEOM2D);

        for (Feature feature : features) {

            envelope.add(feature.getDefaultGeometry());

        }

        return envelope;

    }

}