From ae3ca6829a5bae6088b6f6a005377f11fee03df8 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 31 十月 2022 17:17:37 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/helper/GdalHelper.java |  100 +++++++++++++++++++++++++++++++++++++++-----------
 1 files changed, 78 insertions(+), 22 deletions(-)

diff --git a/src/main/java/com/lf/server/helper/GdalHelper.java b/src/main/java/com/lf/server/helper/GdalHelper.java
index 1d0f0df..5d64ca8 100644
--- a/src/main/java/com/lf/server/helper/GdalHelper.java
+++ b/src/main/java/com/lf/server/helper/GdalHelper.java
@@ -1,5 +1,6 @@
 package com.lf.server.helper;
 
+import com.lf.server.entity.ctrl.MarkJsonEntity;
 import com.lf.server.entity.ctrl.ShpRecordEntity;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
@@ -11,6 +12,7 @@
 import org.gdal.ogr.*;
 import org.gdal.osr.SpatialReference;
 
+import java.io.File;
 import java.sql.Time;
 import java.sql.Timestamp;
 import java.time.LocalDate;
@@ -72,8 +74,81 @@
 
             return new ShpRecordEntity(wkt, csid);
         } catch (Exception ex) {
-            ex.printStackTrace();
+            log.error(ex.getMessage(), ex);
             return null;
+        }
+    }
+
+    /**
+     * 鍒涘缓ShapeFile鏂囦欢
+     *
+     * @param list 鏍囩粯JSON瀹炰綋绫婚泦鍚�
+     * @param path 璺緞
+     * @param type 绫诲瀷锛歅OINT,LINESTRING,POLYGON
+     * @return ShapeFile鏂囦欢鍚�
+     */
+    public static String createShp(List<MarkJsonEntity> list, String path, String type) {
+        try {
+            org.gdal.ogr.Driver driver = ogr.GetDriverByName("ESRI shapefile");
+            if (driver == null) {
+                return null;
+            }
+
+            String filePath = path + File.separator + type.toLowerCase() + ".shp";
+            //DataSource ds = driver.Open(filePath, 0);
+            DataSource ds = driver.CreateDataSource(filePath, null);
+
+            SpatialReference sr = new SpatialReference();
+            sr.ImportFromEPSG(4326);
+
+            int geoType = getGeometryType(type);
+            Layer layer = ds.CreateLayer(type.toLowerCase(), sr, geoType);
+
+            FieldDefn fdName = new FieldDefn("name", ogr.OFTString);
+            fdName.SetWidth(50);
+            layer.CreateField(fdName, 0);
+
+            FieldDefn fdProps = new FieldDefn("props", ogr.OFTString);
+            fdProps.SetWidth(1024);
+            // layer.DeleteField(layer.FindFieldIndex("name", 1));
+            layer.CreateField(fdProps, 1);
+
+            FeatureDefn featureDefn = layer.GetLayerDefn();
+            for (MarkJsonEntity mje : list) {
+                Geometry geo = Geometry.CreateFromWkt(mje.getWkt());
+
+                Feature f = new Feature(featureDefn);
+                f.SetField(0, mje.getName());
+                f.SetField(1, mje.getProps());
+                f.SetGeometry(geo);
+
+                layer.CreateFeature(f);
+            }
+
+            layer.delete();
+            ds.delete();
+            driver.delete();
+
+            return filePath;
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+            return null;
+        }
+    }
+
+    /**
+     * 鑾峰彇鍥惧舰绫诲瀷
+     */
+    private static int getGeometryType(String type) {
+        switch (type) {
+            case "POINT":
+                return 1;
+            case "LINESTRING":
+                return 2;
+            case "POLYGON":
+                return 3;
+            default:
+                return -1;
         }
     }
 
@@ -111,29 +186,9 @@
         gdal.GDALDestroyDriverManager();
     }
 
-    private static void writeShp(String filePath) {
-        try {
-            org.gdal.ogr.Driver driver = ogr.GetDriverByName("ESRI shapefile");
-            if (driver == null) {
-                System.out.println(" ESRI shapefile椹卞姩涓嶅彲鐢紒\n");
-                System.out.println("fail");
-            }
-
-            // Open()鐨勭浜屼釜鍙傛暟榛樿涓�0锛屾槸浠ュ彧璇绘柟寮忔墦寮�鏂囦欢锛�1鏄鍐欐柟寮忔墦寮�
-            DataSource dSource = driver.Open(filePath, 0);
-
-            Layer layer = dSource.GetLayerByIndex(0);
-            FieldDefn fieldDefn = new FieldDefn("name", ogr.OFTString);
-            fieldDefn.SetWidth(50);
-            layer.CreateField(fieldDefn, 1);
-            layer.DeleteField(layer.FindFieldIndex("name", 1));
-        } catch (Exception ex) {
-            ex.printStackTrace();
-        }
-    }
-
     /**
      * 璇诲彇shp鏂囦欢
+     *
      * @param filePath
      */
     public static void readShp(String filePath) {
@@ -257,6 +312,7 @@
 
     /**
      * 璇诲彇gdb鏂囦欢
+     *
      * @param filePath
      */
     public static void readGdb(String filePath) {

--
Gitblit v1.9.3