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

import java.net.URL;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Properties;

import java.util.function.Predicate;

import javax.json.JsonArray;

import javax.json.JsonObject;

import javax.json.JsonValue;

import org.apache.commons.lang3.StringUtils;

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

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

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

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

import org.gvsig.fmap.geom.DataTypes;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryCoercionContext;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryUtils;

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

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

import org.gvsig.json.Json;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.Coercion;

import org.gvsig.tools.dataTypes.DataTypeUtils;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.service.Service;

import org.gvsig.tools.service.ServiceException;

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

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

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

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

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.ToolsSwingManager;

import org.gvsig.tools.swing.icontheme.IconTheme;

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

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

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class DefaultEditingProviderManager extends AbstractProviderManager

implements EditingProviderManager {

    private static final Logger logger = LoggerFactory

        .getLogger(DefaultEditingProviderManager.class);

    Map<String, ISymbol> symbols = new HashMap<>();

    Map<String, Map<String, String>> defaultValues = new HashMap<>();

    private String clipboardValue;

    private List<Geometry> geometriesFromClipboard;

    private List<EditableFeature> featuresFromClipboard;

    public ProviderServices createProviderServices(Service service) {

        // TODO Auto-generated method stub

        return null;

    }

    @Override

    protected String getRegistryKey() {

        // TODO Auto-generated method stub

        return null;

    }

    @Override

    protected String getRegistryDescription() {

        // TODO Auto-generated method stub

        return null;

    }

    @Override

    public Provider createProvider(DynObject serviceParameters,

        ProviderServices providerServices) throws ServiceException {

        String providerName =

            (String) serviceParameters

            .getDynValue(EditingProviderFactory.PROVIDER_NAME_FIELD);

        ProviderFactory factory = getProviderFactory(providerName);

        return factory == null ? null : factory.create(serviceParameters,

            providerServices);

    }

    public EditingServiceInfo getServiceInfo(String name)

        throws ServiceInformationException {

        EditingProviderFactory factory;

        try {

            factory = (EditingProviderFactory) getProviderFactory(name);

            return (factory == null ? null : factory.getServiceInfo());

        } catch (Exception e) {

            throw new ServiceInformationException("Can't get service info of "

                + name, e);

        }

    }

    public void registerIcon(String group, String name, ClassLoader loader,

        String provider) {

        String resourceName;

        IconTheme iconTheme =

            ToolsSwingLocator.getIconThemeManager().getCurrent();

        if ((group == null) || (group.trim().length() == 0)) {

            resourceName = "images/" + name + ".png";

        } else {

            resourceName = "images/" + group + "/" + name + ".png";

        }

        URL resource = null;

        try {

            resource = loader.getResource(resourceName);

            iconTheme.registerDefault(provider, group, name, null, resource);

        } catch (Throwable e) {

            logger.info("Can't register icon '" + name + "'.", e);

        }

    }

    public void registerSymbol(String name, ISymbol symbol) {

        symbols.put(name, symbol);

    }

    public ISymbol getSymbol(String name) {

        return symbols.get(name);

    }

    public void registerDefaultValues(String provider, Map values) {

        if (!this.defaultValues.containsKey(provider)) {

            this.defaultValues.put(provider, values);

        }

    }

    public void registerDefaultValues(String provider, ClassLoader loader) {

        if (!this.defaultValues.containsKey(provider)) {

            String resourceName =

                "defaultvalues/" + provider + "/defaultvalues.properties";

            HashMap<String, String> providerDefaultValues =

                new HashMap<String, String>();

            URL resource = null;

            try {

                resource = loader.getResource(resourceName);

                Properties properties = new Properties();

                properties.load(resource.openStream());

                for (Iterator it = properties.keySet().iterator(); it.hasNext();) {

                    String key = (String) it.next();

                    providerDefaultValues.put(key, (String) properties.get(key));

                }

                this.defaultValues.put(provider, providerDefaultValues);

            } catch (Throwable e) {

                logger.info("Can't register default values from '" + provider

                    + "'.", e);

            }

        }

    }

    @Override

    public Map<String, String> getDefaultValues(String name) {

        return this.defaultValues.get(name);

    }

    @Override

    public List<Geometry> getGeometriesFromClipboard(){

        ToolsSwingManager manager = ToolsSwingLocator.getToolsSwingManager();

        String clipboardValue = manager.getFromClipboard();

        if(StringUtils.isNotBlank(clipboardValue)

                && StringUtils.isNotBlank(this.clipboardValue)

                && StringUtils.equals(clipboardValue, this.clipboardValue)){

            return this.geometriesFromClipboard;

        }

        this.geometriesFromClipboard = new ArrayList<>();

        String geometryName = null;

        try {

            JsonArray jsonArray = Json.createArray(clipboardValue);

            Coercion toGeometry = ToolsLocator.getDataTypesManager().getCoercion(DataTypes.GEOMETRY);

            for (JsonValue item : jsonArray) {

                JsonObject jsonItem = (JsonObject) item;

                if (StringUtils.isBlank(geometryName)) {

                    for (Map.Entry<String, JsonValue> field : jsonItem.entrySet()) {

                        Geometry geometry;

                        try {

                            geometry = (Geometry) toGeometry.coerce(field.getValue().toString().replace("\"", ""));

                            if (geometry != null) {

                                geometryName = field.getKey();

                                this.geometriesFromClipboard.add(geometry);

                                break;

                            }

                        } catch (Exception e) {

                            //DO NOTHING

                        }

                    }

                } else {

                    Geometry geometry;

                    try {

                        String strGeom = jsonItem.getString(geometryName);

                        geometry = (Geometry) toGeometry.coerce(strGeom);

                        if (geometry != null) {

                            this.geometriesFromClipboard.add(geometry);

                        }

                    } catch (Exception e) {

                        //DO NOTHING

    //                        logger.warn("Can't coerce geometry", e);

                    }

                }

            }

        } catch (Exception ex) {

            //DO NOTHING

        }

        return this.geometriesFromClipboard;

    }

    @Override

    public List<EditableFeature> getFeaturesFromClipboard(EditingProviderServices providerServices, FeatureStore store) {

        ToolsSwingManager manager = ToolsSwingLocator.getToolsSwingManager();

        String clipboardValue = manager.getFromClipboard();

        if(StringUtils.isNotBlank(clipboardValue)

                && StringUtils.isNotBlank(this.clipboardValue)

                && StringUtils.equals(clipboardValue, this.clipboardValue)){

            return this.featuresFromClipboard;

        }

        this.featuresFromClipboard = new ArrayList<>();

        GeometryType storeGeomType = store.getDefaultFeatureTypeQuietly().getDefaultGeometryAttribute().getGeomType();

        GeometryCoercionContext context = GeometryLocator.getGeometryManager().createGeometryCoercionContext();

        context.setGeometryType(storeGeomType);

        context.setMode(GeometryCoercionContext.MODE_ONERROR_NULL);

        Coercion toGeometry = DataTypeUtils.getCoercion(DataTypes.GEOMETRY);

        String geometryName = null;

        try {

            JsonArray jsonArray = Json.createArray(clipboardValue);

            for (JsonValue item : jsonArray) {

                JsonObject jsonItem = (JsonObject) item;

                EditableFeature feat = providerServices.createNewFeature(store);

                feat.copyFrom(jsonItem, (FeatureAttributeDescriptor t) -> !t.isPrimaryKey());

                Geometry geometry = feat.getDefaultGeometry();

                if(geometry == null){

                    if (StringUtils.isBlank(geometryName)) {

                        for (Map.Entry<String, JsonValue> field : jsonItem.entrySet()) {

                            try {

                                geometry = (Geometry) toGeometry.coerce(field.getValue().toString().replace("\"", ""), context);

                                if (geometry != null) {

                                    geometryName = field.getKey();

                                    break;

                                }

                            } catch (Exception e) {

                                //DO NOTHING

                            }

                        }

                    } else {

                        try {

                            String strGeom = jsonItem.getString(geometryName);

                            geometry = (Geometry) toGeometry.coerce(strGeom, context);

                        } catch (Exception e) {

                            //DO NOTHING

                        }

                    }

                } else {

                    geometry = (Geometry) toGeometry.coerce(geometry,context);

                }

                if (geometry != null) {

                    feat.setDefaultGeometry(geometry);

                    this.featuresFromClipboard.add(feat);

                }

            }

        } catch (Exception ex) {

            //DO NOTHING

        }

        return this.featuresFromClipboard;

    }

}