From 9b4b8766818ef70c4498436cfd0ce86a1e039570 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期二, 01 十一月 2022 17:41:28 +0800
Subject: [PATCH] 1
---
src/main/java/com/lf/server/helper/ShpHelper.java | 196 ++++++++++++++++++++++++++++
src/main/java/com/lf/server/service/show/MarkService.java | 9
src/main/java/com/lf/server/helper/GdalHelper.java | 200 +---------------------------
src/main/java/com/lf/server/service/show/InquiryService.java | 4
4 files changed, 213 insertions(+), 196 deletions(-)
diff --git a/src/main/java/com/lf/server/helper/GdalHelper.java b/src/main/java/com/lf/server/helper/GdalHelper.java
index 9c006c4..5cc3c44 100644
--- a/src/main/java/com/lf/server/helper/GdalHelper.java
+++ b/src/main/java/com/lf/server/helper/GdalHelper.java
@@ -1,8 +1,5 @@
package com.lf.server.helper;
-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,8 +10,6 @@
import org.gdal.ogr.*;
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;
@@ -33,193 +28,20 @@
public class GdalHelper {
private final static Log log = LogFactory.getLog(GdalHelper.class);
- static {
- // 娉ㄥ唽鎵�鏈夌殑椹卞姩
- gdal.AllRegister();
+ public static void init() {
+ // 娉ㄥ唽鎵�鏈夌殑椹卞姩
+ gdal.AllRegister();
- // 涓轰簡鏀寔涓枃璺緞锛岃娣诲姞涓嬮潰杩欏彞浠ｇ爜
- gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "YES");
+ // 涓轰簡鏀寔涓枃璺緞锛岃娣诲姞涓嬮潰杩欏彞浠ｇ爜
+ gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "YES");
- // 閰嶇疆GDAL_DATA璺緞锛坓dal鏍圭洰褰曚笅鐨刡in\gdal-data锛�
- // gdal.SetConfigOption("GDAL_DATA", "E:\\terrait\\TianJin\\Zip\\release-1928-x64-dev\\release-1928-x64\\bin\\gdal-data");
- // gdal.SetConfigOption("PROJ_LIB", "E:\\terrait\\TianJin\\Zip\\release-1928-x64-dev\\release-1928-x64\\bin\\proj7\\share");
+ // 閰嶇疆GDAL_DATA璺緞锛坓dal鏍圭洰褰曚笅鐨刡in\gdal-data锛�
+ // gdal.SetConfigOption("GDAL_DATA", "E:\\terrait\\TianJin\\Zip\\release-1928-x64-dev\\release-1928-x64\\bin\\gdal-data");
+ // gdal.SetConfigOption("PROJ_LIB", "E:\\terrait\\TianJin\\Zip\\release-1928-x64-dev\\release-1928-x64\\bin\\proj7\\share");
- // 涓轰簡浣垮睘鎬ц〃瀛楁鏀寔涓枃锛岃娣诲姞涓嬮潰杩欏彞锛欳P936
- gdal.SetConfigOption("SHAPE_ENCODING", "");
- }
-
- /**
- * 璇诲彇Shp绗竴鏉¤褰曠殑WKT
- */
- public static ShpRecordEntity readShpFirstRecord(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);
- if (layer.GetFeatureCount() < 1) {
- return null;
- }
-
- SpatialReference spatialReference = layer.GetSpatialRef();
- String csid = spatialReference.GetAttrValue("AUTHORITY", 1);
-
- Feature feature = layer.GetFeature(0);
- String wkt = feature.GetGeometryRef().ExportToWkt();
-
- layer.delete();
- dataSource.delete();
- driver.delete();
-
- return new ShpRecordEntity(wkt, csid);
- } catch (Exception ex) {
- log.error(ex.getMessage(), ex);
- return null;
- }
- }
-
- /**
- * 璇诲彇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瀹炰綋绫婚泦鍚�
- * @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;
- }
- }
+ // 涓轰簡浣垮睘鎬ц〃瀛楁鏀寔涓枃锛岃娣诲姞涓嬮潰杩欏彞锛欳P936
+ gdal.SetConfigOption("SHAPE_ENCODING", "");
+ }
/**
* 璇诲彇tif鏂囦欢
diff --git a/src/main/java/com/lf/server/helper/ShpHelper.java b/src/main/java/com/lf/server/helper/ShpHelper.java
new file mode 100644
index 0000000..cfcac9c
--- /dev/null
+++ b/src/main/java/com/lf/server/helper/ShpHelper.java
@@ -0,0 +1,196 @@
+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;
+import org.gdal.ogr.*;
+import org.gdal.osr.SpatialReference;
+
+import java.io.File;
+import java.lang.reflect.Field;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * ShapeFile甯姪绫�
+ * @author WWW
+ */
+public class ShpHelper {
+ private final static Log log = LogFactory.getLog(ShpHelper.class);
+
+ /**
+ * 璇诲彇Shp绗竴鏉¤褰曠殑WKT
+ */
+ public static ShpRecordEntity readShpFirstRecord(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);
+ if (layer.GetFeatureCount() < 1) {
+ return null;
+ }
+
+ SpatialReference spatialReference = layer.GetSpatialRef();
+ String csid = spatialReference.GetAttrValue("AUTHORITY", 1);
+
+ Feature feature = layer.GetFeature(0);
+ String wkt = feature.GetGeometryRef().ExportToWkt();
+
+ layer.delete();
+ dataSource.delete();
+ driver.delete();
+
+ return new ShpRecordEntity(wkt, csid);
+ } catch (Exception ex) {
+ log.error(ex.getMessage(), ex);
+ return null;
+ }
+ }
+
+ /**
+ * 璇诲彇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瀹炰綋绫婚泦鍚�
+ * @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;
+ }
+ }
+}
diff --git a/src/main/java/com/lf/server/service/show/InquiryService.java b/src/main/java/com/lf/server/service/show/InquiryService.java
index e2b2e46..1cadf99 100644
--- a/src/main/java/com/lf/server/service/show/InquiryService.java
+++ b/src/main/java/com/lf/server/service/show/InquiryService.java
@@ -3,7 +3,7 @@
import com.lf.server.entity.ctrl.ShpRecordEntity;
import com.lf.server.entity.data.MetaFileEntity;
import com.lf.server.helper.FileHelper;
-import com.lf.server.helper.GdalHelper;
+import com.lf.server.helper.ShpHelper;
import com.lf.server.helper.StringHelper;
import org.springframework.stereotype.Service;
@@ -33,7 +33,7 @@
return null;
}
- ShpRecordEntity sr = GdalHelper.readShpFirstRecord(fileName);
+ ShpRecordEntity sr = ShpHelper.readShpFirstRecord(fileName);
FileHelper.deleteFiles(list);
return sr;
diff --git a/src/main/java/com/lf/server/service/show/MarkService.java b/src/main/java/com/lf/server/service/show/MarkService.java
index 55a96aa..52f654a 100644
--- a/src/main/java/com/lf/server/service/show/MarkService.java
+++ b/src/main/java/com/lf/server/service/show/MarkService.java
@@ -1,7 +1,6 @@
package com.lf.server.service.show;
import com.lf.server.entity.ctrl.MarkJsonEntity;
-import com.lf.server.entity.ctrl.ShpRecordEntity;
import com.lf.server.entity.data.DownloadEntity;
import com.lf.server.entity.data.MetaFileEntity;
import com.lf.server.entity.show.MarkEntity;
@@ -123,15 +122,15 @@
List<MarkJsonEntity> points = getMarkByType(list, "POINT");
if (points.size() > 0) {
- String pointFile = GdalHelper.createShp(points, path, "POINT");
+ String pointFile = ShpHelper.createShp(points, path, "POINT");
}
List<MarkJsonEntity> lines = getMarkByType(list, "LINESTRING");
if (lines.size() > 0) {
- String lineFile = GdalHelper.createShp(lines, path, "LINESTRING");
+ String lineFile = ShpHelper.createShp(lines, path, "LINESTRING");
}
List<MarkJsonEntity> polygons = getMarkByType(list, "POLYGON");
if (polygons.size() > 0) {
- String polygonFile = GdalHelper.createShp(polygons, path, "POLYGON");
+ String polygonFile = ShpHelper.createShp(polygons, path, "POLYGON");
}
return path;
@@ -217,7 +216,7 @@
return null;
}
- List<MarkJsonEntity> mjeList = GdalHelper.readShpForMarks(fileName);
+ List<MarkJsonEntity> mjeList = ShpHelper.readShpForMarks(fileName);
FileHelper.deleteFiles(list);
return mjeList;
--
Gitblit v1.9.3