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

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

diff --git a/src/main/java/com/lf/server/helper/GdalHelper.java b/src/main/java/com/lf/server/helper/GdalHelper.java
index 5d64ca8..9c006c4 100644
--- a/src/main/java/com/lf/server/helper/GdalHelper.java
+++ b/src/main/java/com/lf/server/helper/GdalHelper.java
@@ -2,6 +2,7 @@
 
 import com.lf.server.entity.ctrl.MarkJsonEntity;
 import com.lf.server.entity.ctrl.ShpRecordEntity;
+import com.lf.server.entity.data.MetaFileEntity;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.gdal.gdal.Band;
@@ -13,6 +14,7 @@
 import org.gdal.osr.SpatialReference;
 
 import java.io.File;
+import java.lang.reflect.Field;
 import java.sql.Time;
 import java.sql.Timestamp;
 import java.time.LocalDate;
@@ -80,6 +82,73 @@
     }
 
     /**
+     * 璇诲彇ShapeFile鏂囦欢鑾峰彇Mark瀹炰綋绫�
+     */
+    public static List<MarkJsonEntity> readShpForMarks(String filePath) {
+        try {
+            org.gdal.ogr.Driver driver = ogr.GetDriverByName("ESRI shapefile");
+            if (driver == null) {
+                return null;
+            }
+
+            DataSource dataSource = driver.Open(filePath);
+            Layer layer = dataSource.GetLayer(0);
+
+            FeatureDefn featureDefn = layer.GetLayerDefn();
+            Map<Integer, Field> fieldMap = new HashMap(5);
+            for (int i = 0, count = featureDefn.GetFieldCount(); i < count; i++) {
+                try {
+                    FieldDefn fieldDefn = featureDefn.GetFieldDefn(i);
+                    Field field = MarkJsonEntity.class.getDeclaredField(fieldDefn.GetName().toLowerCase());
+                    field.setAccessible(true);
+                    fieldMap.put(i, field);
+                } catch (Exception e) {
+                    //
+                }
+            }
+
+            long count = layer.GetFeatureCount();
+            if (count == 0 || fieldMap.size() == 0) {
+                return null;
+            }
+
+            List<MarkJsonEntity> list = new ArrayList<MarkJsonEntity>();
+            for (int i = 0; i < count; i++) {
+                Feature f = layer.GetFeature(i);
+                MarkJsonEntity me = new MarkJsonEntity(i + 1);
+                for (Integer key : fieldMap.keySet()) {
+                    Field field = fieldMap.get(key);
+                    switch (field.getType().toString()) {
+                        case "double":
+                            field.set(me, f.GetFieldAsDouble(i));
+                            break;
+                        case "long":
+                            field.set(me, f.GetFieldAsInteger64(i));
+                            break;
+                        case "int":
+                            field.set(me, f.GetFieldAsInteger(i));
+                            break;
+                        default:
+                            field.set(me, f.GetFieldAsString(i));
+                            break;
+                    }
+                }
+                me.setWkt(f.GetGeometryRef().ExportToWkt());
+
+                list.add(me);
+            }
+
+            layer.delete();
+            driver.delete();
+
+            return list;
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+            return null;
+        }
+    }
+
+    /**
      * 鍒涘缓ShapeFile鏂囦欢
      *
      * @param list 鏍囩粯JSON瀹炰綋绫婚泦鍚�

--
Gitblit v1.9.3