--- /dev/null
+package org.argeo.app.geo;
+
+import java.io.IOException;
+import java.io.Serializable;
+import java.io.Writer;
+import java.nio.file.Path;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import javax.measure.Quantity;
+import javax.measure.quantity.Area;
+
+import org.geotools.data.DefaultTransaction;
+import org.geotools.data.Transaction;
+import org.geotools.data.collection.ListFeatureCollection;
+import org.geotools.data.shapefile.ShapefileDataStore;
+import org.geotools.data.shapefile.ShapefileDataStoreFactory;
+import org.geotools.data.simple.SimpleFeatureCollection;
+import org.geotools.data.simple.SimpleFeatureIterator;
+import org.geotools.data.simple.SimpleFeatureSource;
+import org.geotools.data.simple.SimpleFeatureStore;
+import org.geotools.geometry.jts.JTS;
+import org.geotools.referencing.CRS;
+import org.geotools.referencing.crs.DefaultGeographicCRS;
+import org.locationtech.jts.geom.Coordinate;
+import org.locationtech.jts.geom.Point;
+import org.locationtech.jts.geom.Polygon;
+import org.opengis.feature.simple.SimpleFeature;
+import org.opengis.feature.simple.SimpleFeatureType;
+import org.opengis.geometry.MismatchedDimensionException;
+import org.opengis.referencing.FactoryException;
+import org.opengis.referencing.crs.CoordinateReferenceSystem;
+import org.opengis.referencing.operation.MathTransform;
+import org.opengis.referencing.operation.TransformException;
+
+import si.uom.SI;
+import tech.units.indriya.quantity.Quantities;
+
+/** Utilities around geographical format, mostly wrapping GeoTools patterns. */
+public class GeoUtils {
+
+ /** In square meters. */
+ public static Quantity<Area> calcArea(SimpleFeature feature) {
+ try {
+ Polygon p = (Polygon) feature.getDefaultGeometry();
+ Point centroid = p.getCentroid();
+ String code = "AUTO:42001," + centroid.getX() + "," + centroid.getY();
+ CoordinateReferenceSystem auto = CRS.decode(code);
+
+ MathTransform transform = CRS.findMathTransform(DefaultGeographicCRS.WGS84, auto);
+
+ Polygon projed = (Polygon) JTS.transform(p, transform);
+ return Quantities.getQuantity(projed.getArea(), SI.SQUARE_METRE);
+ } catch (MismatchedDimensionException | FactoryException | TransformException e) {
+ throw new IllegalStateException("Cannot claculate area of feature");
+ }
+ }
+
+ public static void exportToSvg(SimpleFeatureCollection features, Writer out, int width, int height) {
+ try {
+ double minY = Double.POSITIVE_INFINITY;
+ double maxY = Double.NEGATIVE_INFINITY;
+ double minX = Double.POSITIVE_INFINITY;
+ double maxX = Double.NEGATIVE_INFINITY;
+ List<String> shapes = new ArrayList<>();
+ for (SimpleFeatureIterator it = features.features(); it.hasNext();) {
+ SimpleFeature feature = it.next();
+ StringBuffer sb = new StringBuffer();
+ sb.append("<polyline style=\"stroke-width:1;stroke:#000000;fill:none;\" points=\"");
+
+ Polygon p = (Polygon) feature.getDefaultGeometry();
+ Point centroid = p.getCentroid();
+ String code = "AUTO:42001," + centroid.getX() + "," + centroid.getY();
+ CoordinateReferenceSystem auto = CRS.decode(code);
+
+ MathTransform transform = CRS.findMathTransform(DefaultGeographicCRS.WGS84, auto);
+
+ Polygon projed = (Polygon) JTS.transform(p, transform);
+
+ for (Coordinate coord : projed.getCoordinates()) {
+ double x = coord.x;
+ if (x < minX)
+ minX = x;
+ if (x > maxX)
+ maxX = x;
+ double y = -coord.y;
+ if (y < minY)
+ minY = y;
+ if (y > maxY)
+ maxY = y;
+ sb.append(x + "," + y + " ");
+ }
+ sb.append("\">");
+ sb.append("</polyline>\n");
+ shapes.add(sb.toString());
+
+ }
+ double viewportHeight = maxY - minY;
+ double viewportWidth = maxX - minX;
+ out.write("<svg xmlns=\"http://www.w3.org/2000/svg\"\n");
+ out.write(" width=\"" + width + "\"\n");
+ out.write(" height=\"" + height + "\"\n");
+ out.write(" viewBox=\"" + minX + " " + minY + " " + viewportWidth + " " + viewportHeight + "\"\n");
+ out.write(" preserveAspectRatio=\"xMidYMid meet\"\n");
+ out.write(">\n");
+ for (String shape : shapes) {
+ out.write(shape);
+ out.write("\n");
+ }
+ out.write("</svg>");
+ } catch (IOException | FactoryException | MismatchedDimensionException | TransformException e) {
+ throw new RuntimeException("Cannot export to SVG", e);
+ }
+ }
+
+ /** Write a list of simple features to a shapefile. */
+ public static void saveFeaturesAsShapefile(SimpleFeatureType featureType, List<SimpleFeature> features,
+ Path shpFile) {
+ try {
+ ShapefileDataStoreFactory dataStoreFactory = new ShapefileDataStoreFactory();
+
+ Map<String, Serializable> params = new HashMap<>();
+ params.put("url", shpFile.toUri().toURL());
+
+ params.put("create spatial index", Boolean.TRUE);
+
+ ShapefileDataStore newDataStore = (ShapefileDataStore) dataStoreFactory.createNewDataStore(params);
+ newDataStore.createSchema(featureType);
+
+ String typeName = newDataStore.getTypeNames()[0];
+ SimpleFeatureSource featureSource = newDataStore.getFeatureSource(typeName);
+ if (featureSource instanceof SimpleFeatureStore) {
+ SimpleFeatureStore featureStore = (SimpleFeatureStore) featureSource;
+ SimpleFeatureCollection collection = new ListFeatureCollection(featureType, features);
+
+ try (Transaction transaction = new DefaultTransaction("create")) {
+ try {
+ featureStore.setTransaction(transaction);
+ featureStore.addFeatures(collection);
+ transaction.commit();
+ } catch (Exception problem) {
+ transaction.rollback();
+ throw new RuntimeException("Cannot write shapefile " + shpFile, problem);
+ }
+ }
+ } else {
+ throw new IllegalArgumentException(typeName + " does not support read/write access");
+ }
+ } catch (IOException e) {
+ throw new RuntimeException("Cannot write shapefile " + shpFile, e);
+ }
+ }
+
+ /** Singleton. */
+ private GeoUtils() {
+ }
+}
projects / gpl / argeo-suite.git / blobdiff