Application: gvSIG desktop » gvSIG Linear Reference System

/* 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.lrs.lib.impl;

import java.io.File;

import java.util.Iterator;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

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

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

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

import org.gvsig.fmap.dal.store.shp.SHPNewStoreParameters;

import org.gvsig.fmap.dal.store.shp.SHPStoreProvider;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

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

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

import org.gvsig.fmap.geom.type.GeometryType;

import org.gvsig.lrs.lib.api.LrsAlgorithm;

import org.gvsig.lrs.lib.api.LrsAlgorithmParams;

import org.gvsig.lrs.lib.api.LrsCreateRouteAlgorithmParams;

import org.gvsig.lrs.lib.api.LrsGenerateDynamicSegmentationAlgorithmParams;

import org.gvsig.lrs.lib.api.exceptions.LrsCreateRouteException;

import org.gvsig.lrs.lib.api.exceptions.LrsException;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.DataType;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.locator.LocatorException;

import org.gvsig.tools.service.Manager;

import org.gvsig.tools.task.SimpleTaskStatus;

import org.gvsig.tools.visitor.VisitCanceledException;

import org.gvsig.tools.visitor.Visitor;

/**

 * @author fdiaz

 *

 */

public class LrsGenerateDynamicSegmentationAlgorithm implements LrsAlgorithm {

    private static final Logger logger = LoggerFactory.getLogger(LrsGenerateDynamicSegmentationAlgorithm.class);

    private LrsGenerateDynamicSegmentationAlgorithmParams parameters;

    /**

     *

     */

    public LrsGenerateDynamicSegmentationAlgorithm(LrsGenerateDynamicSegmentationAlgorithmParams parameters) {

        this.parameters = parameters;

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.tools.service.Service#getManager()

     */

    public Manager getManager() {

        return null;

    }

    /*

     * (non-Javadoc)

     *

     * @see org.gvsig.lrs.lib.api.LrsAlgorithm#getParams()

     */

    public  LrsAlgorithmParams getParams() {

        return this.parameters;

    }

    /*

     * (non-Javadoc)

     * @see org.gvsig.lrs.lib.api.LrsAlgorithm#execute(org.gvsig.tools.task.SimpleTaskStatus)

     */

    public void execute(final SimpleTaskStatus taskStatus) throws LrsException {

        NewFeatureStoreParameters newFeatureStoreParameters = parameters.getNewFeatureStoreParameters();

        final FeatureStore sourceFeatureStore = parameters.getSourceFeatureStore();

        final FeatureAttributeDescriptor idRouteField = parameters.getIdRouteField();

        final FeatureAttributeDescriptor idTableRouteField = parameters.getValueTableIdRouteField();

        FeatureStore tableFeatureStore = parameters.getValueTableFeatureStore();

        final FeatureAttributeDescriptor landmarkField = parameters.getLandmarkField();

        final FeatureAttributeDescriptor finalLandmarkField = parameters.getFinalLandmarkField();

        final FeatureAttributeDescriptor valueField = parameters.getValueField();

        logger.info(parameters.toString());

        taskStatus.setTitle(parameters.getName());

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        taskStatus.message(i18nManager.getTranslation("processing"));

        try {

            final String tableRouteFieldName=idTableRouteField.getName();

            final String fromFieldName;

            final DataType fromDataType;

            final String toFieldName;

            final DataType toDataType;

            final int geomType;

            if (landmarkField!=null){

                fromDataType=landmarkField.getDataType();

                if (finalLandmarkField!=null){

                    fromFieldName="PK INICIAL";

                    toFieldName="PK FINAL";

                    toDataType=finalLandmarkField.getDataType();

                    geomType = Geometry.TYPES.MULTILINE;

                }else{

                    fromFieldName="PK";

                    toFieldName=null;

                    toDataType=null;

                    geomType = Geometry.TYPES.POINT;

                }

            }else{

                //TODO: Si no hay fromFieldName ?salir por patas?

                fromFieldName=null;

                fromDataType=null;

                toFieldName=null;

                toDataType=null;

                geomType = Geometry.TYPES.POINT;

            }

            final FeatureStore newFeatureStore = createNewDataStore(

                newFeatureStoreParameters,

                idRouteField,

                fromFieldName,

                toFieldName,

                valueField,

                geomType

             );

            FeatureSet tableFeatures;

            tableFeatures=tableFeatureStore.getFeatureSet();

            taskStatus.setRangeOfValues(0, tableFeatures.getSize()-1);

            newFeatureStore.edit(FeatureStore.MODE_FULLEDIT);

            tableFeatures.accept(new Visitor() {

                int taskCount=0;

                public void visit(Object obj) throws VisitCanceledException, BaseException {

                    Feature feature = (Feature) obj;

                    String routeName = (String) feature.get(tableRouteFieldName);

                    Object objFrom = null;

                    Object objTo = null;

                    if (fromFieldName != null) {

                        objFrom = feature.get(landmarkField.getName());

                        double fromValue = LrsAlgorithmUtils.getAsDouble(objFrom, fromDataType);

                        Object value = feature.get(valueField.getName());

                        if (toFieldName == null || toFieldName.isEmpty()) {

                            addSegmentatePointsFeatures(newFeatureStore, sourceFeatureStore, idRouteField, routeName, fromFieldName, fromValue, valueField, value);

                        } else {

                            objTo = feature.get(finalLandmarkField.getName());

                            double toValue = LrsAlgorithmUtils.getAsDouble(objTo, toDataType);

                            addSegmentateLinesFeatures(newFeatureStore, sourceFeatureStore, idRouteField, routeName,

                                fromFieldName, fromValue, toFieldName, toValue, valueField, value);

                        }

                    } else {

                        //TODO: ?Exception?

                    }

                    taskStatus.setCurValue(taskCount++);

                }

                private void addSegmentatePointsFeatures(final FeatureStore featureStore,

                    FeatureStore sourceFeatureStore, final FeatureAttributeDescriptor idRouteField,

                    final String routeName, final String fromFieldName, final double fromValue, final FeatureAttributeDescriptor valueField, final Object value)

                    throws LocatorException, BaseException {

                    DataManager dataManager = DALLocator.getDataManager();

                    FeatureQuery query = sourceFeatureStore.createFeatureQuery();

                    String expression = StringUtils.replace(StringUtils.replace("NOMBRE like 'a%'", "NOMBRE", idRouteField.getName()), "a%", routeName);

                    query.setFilter( dataManager.createExpresion(expression) );

                    FeatureSet features = sourceFeatureStore.getFeatureSet(query);

                    features.accept(new Visitor() {

                        public void visit(Object obj) throws VisitCanceledException, BaseException {

                            Feature feature = (Feature)obj;

                            Geometry geom = feature.getDefaultGeometry();

                            List<Point>points = LrsAlgorithmUtils.getPointsWithM(geom, fromValue);

                            for (Iterator<Point> iterator = points.iterator(); iterator.hasNext();) {

                                Point point = (Point) iterator.next();

                                EditableFeature newFeature = featureStore.createNewFeature(true);

                                newFeature.set(idRouteField.getName(), routeName);

                                newFeature.set(fromFieldName, fromValue);

                                newFeature.set(valueField.getName(), value);

                                newFeature.setDefaultGeometry(point);

                                featureStore.insert(newFeature);

                            }

                        }

                    });

                }

                private void addSegmentateLinesFeatures(final FeatureStore featureStore,

                    FeatureStore sourceFeatureStore, final FeatureAttributeDescriptor idRouteField,

                    final String routeName, final String fromFieldName, final double fromValue,

                    final String toFieldName, final double toValue, final FeatureAttributeDescriptor valueField, final Object value) throws LocatorException, BaseException {

                    DataManager dataManager = DALLocator.getDataManager();

                    FeatureQuery query = sourceFeatureStore.createFeatureQuery();

                    String expression = StringUtils.replace(StringUtils.replace("NOMBRE like 'a%'", "NOMBRE", idRouteField.getName()), "a%", routeName);

                    query.setFilter( dataManager.createExpresion(expression) );

                    FeatureSet features = sourceFeatureStore.getFeatureSet(query);

                    features.accept(new Visitor() {

                        public void visit(Object obj) throws VisitCanceledException, BaseException {

                            Feature feature = (Feature) obj;

                            Geometry geom = feature.getDefaultGeometry();

                            MultiLine lines = LrsAlgorithmUtils.getLinesBetweenTwoM(geom, fromValue, toValue);

                            if (lines != null && lines.getPrimitivesNumber()>0) {

                                EditableFeature newFeature = featureStore.createNewFeature(true);

                                newFeature.set(idRouteField.getName(), routeName);

                                newFeature.set(fromFieldName, fromValue);

                                newFeature.set(toFieldName, toValue);

                                newFeature.set(valueField.getName(), value);

                                newFeature.setDefaultGeometry(lines);

                                featureStore.insert(newFeature);

                            }

                        }

                    });

                }

            });

            newFeatureStore.finishEditing();

        } catch (Exception e1) {

            taskStatus.abort();

            throw new LrsCreateRouteException("Error creating routes", e1);

        }

        taskStatus.terminate();

    }

    /* (non-Javadoc)

     * @see org.gvsig.lrs.lib.api.LrsAlgorithm#setParams(org.gvsig.lrs.lib.api.LrsAlgorithmParams)

     */

    public void setParams(LrsAlgorithmParams params) throws IllegalArgumentException {

        if(!(params instanceof LrsCreateRouteAlgorithmParams)){

            throw new IllegalArgumentException("params should be LrsCreateRouteAlgorithmParams type.");

        }

        this.parameters = (LrsGenerateDynamicSegmentationAlgorithmParams) params;

    }

    private FeatureStore createNewDataStore(

        NewFeatureStoreParameters newFeatureStoreParameters,

        FeatureAttributeDescriptor idRouteField,

        String fromFieldName,

        String toFieldName,

        FeatureAttributeDescriptor valueField,

        int type) throws Exception{

        try {

            SHPNewStoreParameters shapeStoreParams = (SHPNewStoreParameters)newFeatureStoreParameters;

            File file=shapeStoreParams.getFile();

            String filePath=file.getPath().substring(0, file.getPath().lastIndexOf(File.separator));

            DataManager dataManager = DALLocator.getDataManager();

            DataServerExplorerParameters serverParams =

                dataManager.createServerExplorerParameters("FilesystemExplorer");

            serverParams.setDynValue("initialpath", filePath);

            DataServerExplorer serverExplorer =

                dataManager.openServerExplorer(serverParams.getExplorerName(), serverParams);

            EditableFeatureType featureType = (EditableFeatureType)shapeStoreParams.getDefaultFeatureType();

            featureType.add(idRouteField.getName(), idRouteField.getType(), idRouteField.getSize());

            featureType.add(fromFieldName, DataTypes.DOUBLE);

            if(toFieldName!=null){

                featureType.add(toFieldName, DataTypes.DOUBLE);

            }

            featureType.add(valueField.getName(), valueField.getType(), valueField.getSize());

            EditableFeatureAttributeDescriptor geometryField = featureType.add("Geometry", DataTypes.GEOMETRY);

            GeometryType geometryType = GeometryLocator.getGeometryManager().getGeometryType(type, Geometry.SUBTYPES.GEOM2DM);

            geometryField.setGeometryType(geometryType);

            featureType.setDefaultGeometryAttributeName("Geometry");

            shapeStoreParams.setDefaultFeatureType(featureType);

            serverExplorer.add("Shape", shapeStoreParams, true);

            DataStore store = dataManager.openStore(SHPStoreProvider.NAME, shapeStoreParams);

            return (FeatureStore)store;

        } catch (Exception e) {

            throw new LrsCreateRouteException("Error creating new dataStore", e);

        }

    }

}