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 | 219 +++++++++++++++++++++++++++++++++++++++++++++++-------
1 files changed, 188 insertions(+), 31 deletions(-)
diff --git a/src/main/java/com/lf/server/helper/GdalHelper.java b/src/main/java/com/lf/server/helper/GdalHelper.java
index 48bffcb..9c006c4 100644
--- a/src/main/java/com/lf/server/helper/GdalHelper.java
+++ b/src/main/java/com/lf/server/helper/GdalHelper.java
@@ -1,6 +1,8 @@
package com.lf.server.helper;
-import com.lf.server.entity.ctrl.LayerDto;
+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;
@@ -11,6 +13,8 @@
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;
@@ -42,6 +46,179 @@
// 涓轰簡浣垮睘鎬ц〃瀛楁鏀寔涓枃锛岃娣诲姞涓嬮潰杩欏彞锛欳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;
+ }
}
/**
@@ -78,29 +255,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) {
@@ -154,14 +311,14 @@
}
}
- public static LayerDto getLayerDto(String filePath) {
+ public static void getLayerDto(String filePath) {
org.gdal.ogr.Driver driver = ogr.GetDriverByName("OpenFileGDB");
if (driver == null) {
- return null;
+ return;
}
+
List<Map> list = new ArrayList<>();
List<Map<String, String>> mapList = new ArrayList<>();
- LayerDto layerDto = new LayerDto();
DataSource dataSource = null;
try {
dataSource = driver.Open(filePath, 0);
@@ -208,10 +365,10 @@
feature.delete();
} while (true);
}
- layerDto.setLayerName(strlayerName);
- layerDto.setList(list);
- layerDto.setCount(count);
- layerDto.setMap(mapList);
+ //layerDto.setLayerName(strlayerName);
+ //layerDto.setList(list);
+ //layerDto.setCount(count);
+ //layerDto.setMap(mapList);
}
} catch (Exception e) {
e.printStackTrace();
@@ -220,11 +377,11 @@
dataSource.delete();
}
}
- return layerDto;
}
/**
* 璇诲彇gdb鏂囦欢
+ *
* @param filePath
*/
public static void readGdb(String filePath) {
@@ -277,7 +434,7 @@
return propertyGetter.get(feature, index);
} catch (Exception ex) {
// ex.printStackTrace()
- log.error(ex.getMessage() + ex.getStackTrace() + "\n");
+ log.error(ex.getMessage(), ex);
return null;
}
--
Gitblit v1.9.3