/**
* Copyright (C) 2007 - 2016 52°North Initiative for Geospatial Open Source
* Software GmbH
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 as published
* by the Free Software Foundation.
*
* If the program is linked with libraries which are licensed under one of
* the following licenses, the combination of the program with the linked
* library is not considered a "derivative work" of the program:
*
* • Apache License, version 2.0
* • Apache Software License, version 1.0
* • GNU Lesser General Public License, version 3
* • Mozilla Public License, versions 1.0, 1.1 and 2.0
* • Common Development and Distribution License (CDDL), version 1.0
*
* Therefore the distribution of the program linked with libraries licensed
* under the aforementioned licenses, is permitted by the copyright holders
* if the distribution is compliant with both the GNU General Public
* License version 2 and the aforementioned licenses.
*
* As an exception to the terms of the GPL, you may copy, modify,
* propagate, and distribute a work formed by combining 52°North WPS
* GeoTools Modules with the Eclipse Libraries, or a work derivative of
* such a combination, even if such copying, modification, propagation, or
* distribution would otherwise violate the terms of the GPL. Nothing in
* this exception exempts you from complying with the GPL in all respects
* for all of the code used other than the Eclipse Libraries. You may
* include this exception and its grant of permissions when you distribute
* 52°North WPS GeoTools Modules. Inclusion of this notice with such a
* distribution constitutes a grant of such permissions. If you do not wish
* to grant these permissions, remove this paragraph from your
* distribution. "52°North WPS GeoTools Modules" means the 52°North WPS
* modules using GeoTools functionality - software licensed under version 2
* or any later version of the GPL, or a work based on such software and
* licensed under the GPL. "Eclipse Libraries" means Eclipse Modeling
* Framework Project and XML Schema Definition software distributed by the
* Eclipse Foundation and licensed under the Eclipse Public License Version
* 1.0 ("EPL"), or a work based on such software and licensed under the EPL.
*
* 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.
*/
package org.n52.wps.io.datahandler.generator;
import java.io.File;
import java.io.IOException;
import java.io.Serializable;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import javax.xml.namespace.QName;
import org.geotools.data.DataStoreFactorySpi;
import org.geotools.data.DefaultTransaction;
import org.geotools.data.FeatureStore;
import org.geotools.data.Transaction;
import org.geotools.data.shapefile.ShapefileDataStore;
import org.geotools.data.shapefile.ShapefileDataStoreFactory;
import org.geotools.feature.DefaultFeatureCollections;
import org.geotools.feature.FeatureCollection;
import org.geotools.feature.FeatureIterator;
import org.geotools.referencing.CRS;
import org.n52.wps.io.GTHelper;
import org.n52.wps.io.SchemaRepository;
import org.n52.wps.io.data.IData;
import org.n52.wps.io.data.binding.complex.GTVectorDataBinding;
import org.opengis.feature.Feature;
import org.opengis.feature.IllegalAttributeException;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.referencing.FactoryException;
import org.opengis.referencing.NoSuchAuthorityCodeException;
import com.vividsolutions.jts.geom.Geometry;
public class GTBinDirectorySHPGenerator {
public File writeFeatureCollectionToDirectory(IData data)
throws IOException {
return writeFeatureCollectionToDirectory(data, null);
}
public File writeFeatureCollectionToDirectory(IData data, File parent) throws IOException {
GTVectorDataBinding binding = (GTVectorDataBinding) data;
FeatureCollection originalCollection = binding.getPayload();
FeatureCollection collection = createCorrectFeatureCollection(originalCollection);
return createShapefileDirectory(collection, parent);
}
private FeatureCollection createCorrectFeatureCollection(
FeatureCollection fc) {
FeatureCollection resultFeatureCollection = DefaultFeatureCollections.newCollection();
SimpleFeatureType featureType = null;
FeatureIterator iterator = fc.features();
String uuid = UUID.randomUUID().toString();
int i = 0;
while(iterator.hasNext()){
SimpleFeature feature = (SimpleFeature) iterator.next();
if(i==0){
featureType = GTHelper.createFeatureType(feature.getProperties(), (Geometry)feature.getDefaultGeometry(), uuid, feature.getFeatureType().getCoordinateReferenceSystem());
QName qname = GTHelper.createGML3SchemaForFeatureType(featureType);
SchemaRepository.registerSchemaLocation(qname.getNamespaceURI(), qname.getLocalPart());
}
Feature resultFeature = GTHelper.createFeature("ID"+i, (Geometry)feature.getDefaultGeometry(), featureType, feature.getProperties());
resultFeatureCollection.add(resultFeature);
i++;
}
return resultFeatureCollection;
}
/**
* Transforms the given {@link FeatureCollection} into a zipped SHP file
* (.shp, .shx, .dbf, .prj) and returs its Base64 encoding
*
* @param collection
* the collection to transform
* @return the zipped shapefile
* @throws IOException
* If an error occurs while creating the SHP file or encoding
* the shapefile
* @throws IllegalAttributeException
* If an error occurs while writing the features into the the
* shapefile
*/
private File createShapefileDirectory(FeatureCollection collection, File parent)
throws IOException, IllegalAttributeException {
if (parent == null) {
File tempBaseFile = File.createTempFile("resolveDir", ".tmp");
tempBaseFile.deleteOnExit();
parent = tempBaseFile.getParentFile();
}
if (parent == null || !parent.isDirectory()) {
throw new IllegalStateException("Could not find temporary file directory.");
}
File shpBaseDirectory = new File(parent, UUID.randomUUID().toString());
if (!shpBaseDirectory.mkdir()) {
throw new IllegalStateException("Could not create temporary shp directory.");
}
File tempSHPfile = File.createTempFile("shp", ".shp", shpBaseDirectory);
tempSHPfile.deleteOnExit();
DataStoreFactorySpi dataStoreFactory = new ShapefileDataStoreFactory();
Map<String, Serializable> params = new HashMap<String, Serializable>();
params.put("url", tempSHPfile.toURI().toURL());
params.put("create spatial index", Boolean.TRUE);
ShapefileDataStore newDataStore = (ShapefileDataStore) dataStoreFactory
.createNewDataStore(params);
newDataStore.createSchema((SimpleFeatureType) collection.getSchema());
if(collection.getSchema().getCoordinateReferenceSystem()==null){
try {
newDataStore.forceSchemaCRS(CRS.decode("4326"));
} catch (NoSuchAuthorityCodeException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (FactoryException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}else{
newDataStore.forceSchemaCRS(collection.getSchema()
.getCoordinateReferenceSystem());
}
Transaction transaction = new DefaultTransaction("create");
String typeName = newDataStore.getTypeNames()[0];
FeatureStore<SimpleFeatureType, SimpleFeature> featureStore = (FeatureStore<SimpleFeatureType, SimpleFeature>) newDataStore
.getFeatureSource(typeName);
featureStore.setTransaction(transaction);
try {
featureStore.addFeatures(collection);
transaction.commit();
} catch (Exception problem) {
transaction.rollback();
} finally {
transaction.close();
}
// Zip the shapefile
String path = tempSHPfile.getAbsolutePath();
String baseName = path.substring(0, path.length() - ".shp".length());
File shx = new File(baseName + ".shx");
File dbf = new File(baseName + ".dbf");
File prj = new File(baseName + ".prj");
// mark created files for delete
tempSHPfile.deleteOnExit();
shx.deleteOnExit();
dbf.deleteOnExit();
prj.deleteOnExit();
shpBaseDirectory.deleteOnExit();
return shpBaseDirectory;
}
}