Application: gvSIG desktop » gvSIG tools: Vector editing

/**

 * 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.vectorediting.lib.prov.rectangularmatrix;

import java.awt.geom.AffineTransform;

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.primitive.Point;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.exception.BaseException;

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

/**

 * @author llmarques

 *

 */

public class RectangularMatrixEditingProvider extends AbstractEditingProvider

    implements EditingProvider {

    private FeatureStore featureStore;

    private Map<EditingServiceParameter, Object> values;

    private EditingServiceParameter selection;

    private EditingServiceParameter columns;

    private EditingServiceParameter rows;

    private EditingServiceParameter basePoint;

    private EditingServiceParameter columnSpacing;

    private EditingServiceParameter rowSpacing;

    /**

     * Default constructor.

     *

     * @param providerServices

     *            available services for this provider

     * @param parameters

     *            of this provider

     */

    public RectangularMatrixEditingProvider(DynObject parameters,

        ProviderServices services) {

        super(services);

        this.featureStore =

            (FeatureStore) parameters

                .getDynValue(EditingProviderFactory.FEATURE_STORE_FIELD);

        this.selection =

            new DefaultEditingServiceParameter("selection", "selection",

                EditingServiceParameter.TYPE.SELECTION);

        this.basePoint =

            new DefaultEditingServiceParameter("indicate_new_point", "indicate_new_point",

                EditingServiceParameter.TYPE.POSITION);

        this.columns =

            new DefaultEditingServiceParameter("columns_number", "columns_number",

                EditingServiceParameter.TYPE.VALUE);

        this.rows =

            new DefaultEditingServiceParameter("rows_number", "rows_number",

                EditingServiceParameter.TYPE.VALUE);

        this.columnSpacing =

            new DefaultEditingServiceParameter("space_between_columns",

                "spacing_between_columns", EditingServiceParameter.TYPE.VALUE,

                EditingServiceParameter.TYPE.POSITION);

        this.rowSpacing =

            new DefaultEditingServiceParameter("space_between_rows",

                "spacing_between_rows", EditingServiceParameter.TYPE.VALUE,

                EditingServiceParameter.TYPE.POSITION);

    }

    public EditingServiceParameter next() {

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

            return selection;

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

            return columns;

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

            return rows;

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

            return basePoint;

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

            return columnSpacing;

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

            return rowSpacing;

        }

        return null;

    }

    public DrawingStatus getDrawingStatus(Point mousePosition)

        throws DrawServiceException {

        DefaultDrawingStatus drawingStatus = new DefaultDrawingStatus();

        FeatureSelection featureSelection =

            (FeatureSelection) values.get(selection);

        Integer columnsValue = (Integer) values.get(columns);

        Integer rowsValue = (Integer) values.get(rows);

        Point basePointValue = (Point) values.get(basePoint);

        Double spacingColumnsValue = (Double) values.get(columnSpacing);

        Double spacingRowsValue = (Double) values.get(rowSpacing);

        if (featureSelection != null && columnsValue != null

            && rowsValue != null && basePointValue != null) {

            if (spacingColumnsValue == null) {

                spacingColumnsValue =

                    mousePosition.getX() - basePointValue.getX();

            }

            if (spacingRowsValue == null) {

                spacingRowsValue = mousePosition.getY() - basePointValue.getY();

            }

            DisposableIterator it = null;

            try {

                it = featureSelection.fastIterator();

                while (it.hasNext()) {

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

                    Geometry geometry = feature.getDefaultGeometry();

                    Geometry[][] matrix =

                        createRectangularMatrix(geometry, columnsValue,

                            rowsValue, basePointValue, spacingColumnsValue,

                            spacingRowsValue);

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

                        for (int j = 0; j < matrix[i].length; j++) {

                            drawingStatus.addGeometry(matrix[i][j]);

                        }

                    }

                }

            } catch (BaseException e) {

                throw new DrawServiceException(e);

            } finally {

                it.dispose();

            }

        }

        return drawingStatus;

    }

    private Geometry[][] createRectangularMatrix(Geometry geometry,

        int columnsValue, int rowsValue, Point basePointValue,

        double spacingColumns, double spacingRows) throws BaseException {

        Geometry[][] geometryMatrix = new Geometry[columnsValue][rowsValue];

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

            for (int j = 0; j < rowsValue; j++) {

                if (i == 0 && j == 0) {

                    continue;

                }

                AffineTransform at = new AffineTransform();

                at.translate(spacingColumns * i, spacingRows * j);

                Geometry clonedGeoemtry = geometry.cloneGeometry();

                clonedGeoemtry.transform(at);

                geometryMatrix[i][j] = clonedGeoemtry;

            }

        }

        return geometryMatrix;

    }

    public void stop() throws StopServiceException {

        values.clear();

    }

    public List<EditingServiceParameter> getParameters() {

        List<EditingServiceParameter> parameters =

            new ArrayList<EditingServiceParameter>();

        parameters.add(basePoint);

        parameters.add(columns);

        parameters.add(rows);

        parameters.add(columnSpacing);

        parameters.add(rowSpacing);

        return parameters;

    }

    public void setValue(Object value) throws InvalidEntryException {

        EditingServiceParameter parameter = next();

        validateAndInsertValue(parameter, value);

    }

    private void validateAndInsertValue(EditingServiceParameter parameter,

        Object value) throws InvalidEntryException {

        boolean insertedValue = false;

        if (parameter.equals(selection) && value instanceof FeatureSelection) {

            FeatureSelection featureSelection = (FeatureSelection) value;

            if (featureSelection.getSelectedCount() > 0) {

                values.put(parameter, featureSelection);

                insertedValue = true;

            }

        } else if (parameter.equals(columns) || parameter.equals(rows)) {

            if (value instanceof Double) {

                Double number = (Double) value;

                values.put(parameter, number.intValue());

                insertedValue = true;

            }

        } else if (parameter.equals(basePoint) && value instanceof Point) {

            values.put(basePoint, value);

            insertedValue = true;

        } else if (parameter.equals(columnSpacing)

            || parameter.equals(rowSpacing)) {

            if (value instanceof Double) {

                Double doubleValue = (Double) value;

                values.put(parameter, doubleValue);

                insertedValue = true;

            } else if (value instanceof Point) {

                Point basePointValue = (Point) values.get(basePoint);

                Point valuePoint = (Point) value;

                double distance = 0;

                if (parameter.equals(columnSpacing)) {

                    distance = valuePoint.getX() - basePointValue.getX();

                    values.put(columnSpacing, distance);

                } else if (parameter.equals(rowSpacing)) {

                    distance = valuePoint.getY() - basePointValue.getY();

                    values.put(rowSpacing, distance);

                }

                insertedValue = true;

            }

        }

        if (!insertedValue) {

            throw new InvalidEntryException(null);

        }

    }

    public Geometry finish() throws FinishServiceException {

        return null;

    }

    public void finishAndStore() throws FinishServiceException {

        FeatureSelection featureSelection =

            (FeatureSelection) values.get(selection);

        final Integer columnsValue = (Integer) values.get(columns);

        final Integer rowsValue = (Integer) values.get(rows);

        final Point basePointValue = (Point) values.get(basePoint);

        final Double spacingColumnsValue = (Double) values.get(columnSpacing);

        final Double spacingRowsValue = (Double) values.get(rowSpacing);

        final EditingProviderServices editingProviderServices =

            (EditingProviderServices) getProviderServices();

        if (featureSelection != null && columnsValue != null

            && rowsValue != null && basePointValue != null

            && spacingColumnsValue != null && spacingRowsValue != null) {

            try {

                featureSelection.accept(new Visitor() {

                    public void visit(Object obj)

                        throws VisitCanceledException, BaseException {

                        Feature feature = (Feature) obj;

                        Geometry geometry = feature.getDefaultGeometry();

                        Geometry[][] matrix =

                            createRectangularMatrix(geometry, columnsValue,

                                rowsValue, basePointValue, spacingColumnsValue,

                                spacingRowsValue);

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

                            for (int j = 0; j < matrix[i].length; j++) {

                                EditableFeature eFeature =

                                    editingProviderServices

                                        .getFeatureCopyWithoutPK(featureStore,

                                            feature);

                                eFeature.setDefaultGeometry(matrix[i][j]);

                                editingProviderServices

                                    .insertFeatureIntoFeatureStore(eFeature,

                                        featureStore);

                            }

                        }

                    }

                });

            } catch (BaseException e) {

                throw new FinishServiceException(e);

            }

        }

    }

    public void start() throws StartServiceException, InvalidEntryException {

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

        FeatureSelection selected = null;

        if (featureStore != null && selection == null) {

            try {

                selected = featureStore.getFeatureSelection();

            } catch (DataException e) {

                throw new StartServiceException(e);

            }

            if (selected.getSelectedCount() > 0) {

                try {

                    setValue(selected);

                } catch (InvalidEntryException e) {

                    throw new InvalidEntryException(e);

                }

            }

        }

    }

    public String getName() {

        return RectangularMatrixEditingProviderFactory.PROVIDER_NAME;

    }

}