From de821f0a99e234ce8e336b1775e90169e875cdd5 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Tue, 09 Apr 2024 09:44:38 +0800
Subject: [PATCH] 图斑生成kmz航线文件,入库并返回前端,优化图斑模块

---
 src/main/java/com/dji/sample/patches/service/impl/ShpToDataSourceServiceImpl.java |   48 ++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 40 insertions(+), 8 deletions(-)

diff --git a/src/main/java/com/dji/sample/patches/service/impl/ShpToDataSourceServiceImpl.java b/src/main/java/com/dji/sample/patches/service/impl/ShpToDataSourceServiceImpl.java
index 30cccc9..b8dfb92 100644
--- a/src/main/java/com/dji/sample/patches/service/impl/ShpToDataSourceServiceImpl.java
+++ b/src/main/java/com/dji/sample/patches/service/impl/ShpToDataSourceServiceImpl.java
@@ -1,31 +1,46 @@
 package com.dji.sample.patches.service.impl;
 
+import com.dji.sample.common.error.CommonErrorEnum;
+import com.dji.sample.component.mqtt.model.RequestsReply;
 import com.dji.sample.patches.dao.ShpToDataSourceMapper;
 import com.dji.sample.patches.model.entity.LotInfo;
 import com.dji.sample.patches.service.ShpToDataSourceService;
 import com.dji.sample.patches.utils.FormatConversionUtil;
+import com.dji.sample.patches.utils.GeoToolsUtil;
 import com.dji.sample.patches.utils.MultipartFileTOFileUtil;
 import com.dji.sample.patches.utils.ShapeFileUtil;
+import com.dji.sample.patches.xml.mode.XMLTemplateModel;
+import com.dji.sample.patches.xml.utils.CreateWaylineFileUtils;
+import com.dji.sample.wayline.service.IWaylineFileService;
+import org.locationtech.jts.geom.Coordinate;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
-
 import java.io.File;
-import java.io.IOException;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.UUID;
-
+import static com.dji.sample.patches.utils.FormatConversionUtil.getCentros;
+import static com.dji.sample.patches.utils.MultipartFileTOFileUtil.convert;
+import static com.dji.sample.patches.utils.ZipUtil.*;
 import static org.locationtech.jts.io.WKTConstants.POLYGON;
 
 @Service
 public class ShpToDataSourceServiceImpl implements ShpToDataSourceService {
     @Autowired
     private ShpToDataSourceMapper mapper;
+
+    @Autowired
+    private IWaylineFileService service;
     String head = POLYGON;
+    String sourceDir = "/home/drone/server/template/wpmz/"; // 源文件夹路径
+
+    MultipartFile multipartFile;
+
     String json;
-    public List<LotInfo> insertGeo(MultipartFile file ,String workspaceId) throws IOException {
-        List<LotInfo> list=new ArrayList<>();
+
+    public MultipartFile insertGeo(MultipartFile file, String workspaceId,String waylineName, double airportLat, double airportLon) throws Exception {
+        List<LotInfo> list = new ArrayList<>();
         MultipartFileTOFileUtil multipartFileTOFileUtil = new MultipartFileTOFileUtil();
         File file1 = multipartFileTOFileUtil.multipartFile2File(file);
         List<String> s = ShapeFileUtil.shpToGeoJson(file1);
@@ -40,12 +55,29 @@
             lotInfo.setBsm(bsm);
             lotInfo.setWorkspaceId(workspaceId);
             lotInfo.setDkfw(json);
-            lotInfo.setDkbh("dkbh"+i);
+            lotInfo.setDkbh("dkbh" + i);
+            list.add(lotInfo);
+            String strs=getCentros(list);
+            String end =strs.replaceAll("\\(","").replaceAll(", NaN\\)","");
+            String[] points=end.split(",");
+            lotInfo.setXzb(Double.valueOf(points[0]));
+            lotInfo.setYzb(Double.valueOf(points[1]));
             mapper.insert(lotInfo);
             list.add(lotInfo);
-            System.out.println(lotInfo);
         }
-        return list;
+        Coordinate[] coordinates = GeoToolsUtil.getRoutePointOrder(list, airportLat, airportLon);
+        XMLTemplateModel xmlModel = XMLTemplateModel.init(coordinates, list);
+        CreateWaylineFileUtils.createWaylineFile(xmlModel);
+        // 压缩文件夹中的内容
+        String destKMZFile = "/home/drone/server/template/kmz/"+waylineName+".kmz"; // 输出的KMZ文件路径
+        zipFolder(sourceDir, destKMZFile);
+        multipartFile = convert(new File("/home/drone/server/template/kmz/"+waylineName+".kmz"));
+        return multipartFile;
+
     }
 
 }
+
+
+
+

--
Gitblit v1.9.3