From 066f132c523fb2d8dbcaf20ea900db061b068910 Mon Sep 17 00:00:00 2001
From: xieb <vip_xiaobin810@163.com>
Date: Mon, 06 May 2024 14:40:38 +0800
Subject: [PATCH] 图斑航线按机场范围生成

---
 src/main/java/com/dji/sample/territory/service/impl/TbDkjbxxServiceImpl.java |   36 +++++++++++++++++++++++-------------
 1 files changed, 23 insertions(+), 13 deletions(-)

diff --git a/src/main/java/com/dji/sample/territory/service/impl/TbDkjbxxServiceImpl.java b/src/main/java/com/dji/sample/territory/service/impl/TbDkjbxxServiceImpl.java
index 82ec6a2..679665a 100644
--- a/src/main/java/com/dji/sample/territory/service/impl/TbDkjbxxServiceImpl.java
+++ b/src/main/java/com/dji/sample/territory/service/impl/TbDkjbxxServiceImpl.java
@@ -8,12 +8,14 @@
 import com.dji.sample.patches.model.entity.LotInfo;
 import com.dji.sample.patches.utils.GeoToolsUtil;
 import com.dji.sample.patches.utils.MultipartFileTOFileUtil;
+import com.dji.sample.patches.utils.PointPO;
 import com.dji.sample.patches.xml.mode.XMLTemplateModel;
 import com.dji.sample.patches.xml.utils.CreateWaylineFileUtils;
 import com.dji.sample.territory.dao.ITbDkjbxxMapper;
 import com.dji.sample.territory.dao.ITbFjMapper;
 import com.dji.sample.territory.model.entity.TbDkjbxxEntity;
 import com.dji.sample.territory.model.entity.TbFjEntity;
+import com.dji.sample.territory.pojo.TerritoryConfigPojo;
 import com.dji.sample.territory.service.ITbDkjbxxService;
 import org.locationtech.jts.geom.Coordinate;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -28,9 +30,12 @@
 import java.net.URL;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Objects;
 
 import static com.dji.sample.patches.utils.MultipartFileTOFileUtil.convert;
 import static com.dji.sample.patches.utils.ZipUtil.zipFolder;
+import static com.dji.sample.territory.utils.CoordinateSystemUtil.*;
+
 
 /**
  * @PROJECT_NAME: drone
@@ -41,11 +46,12 @@
 @Service
 @DS("sqlite-resource")
 public class TbDkjbxxServiceImpl extends ServiceImpl<ITbDkjbxxMapper, TbDkjbxxEntity> implements ITbDkjbxxService {
-    private static final String RESOURCE_FILE_PATH = "DB/resource_db.db";
     @Autowired
-    private ITbDkjbxxMapper mapper;
+    private ITbDkjbxxMapper iTbDkjbxxMapper;
     @Autowired
     private PatchesConfigPojo patchesConfigPojo;
+    @Autowired
+    private TerritoryConfigPojo territoryConfigPojo;
 
     /**
      * 上传DB文件并覆盖之前DB文件,自动读取数据到本地数据库
@@ -53,15 +59,19 @@
      * @param file
      * @return
      */
-    public ResponseResult<String> uploadFile(MultipartFile file) {
+    public ResponseResult uploadFile(MultipartFile file) {
         if (file.isEmpty()) {
-            return ResponseResult.error("所上传的文件为空");
+            throw new IllegalArgumentException("上传文件为空");
+        }
+        boolean fileName = Objects.requireNonNull(file.getOriginalFilename()).endsWith("db");
+        if (!fileName) {
+            throw new IllegalArgumentException("检查文件是否为db文件");
         }
         try {
             // 获取上传的文件输入流
             InputStream inputStream = file.getInputStream();
             // 创建输出流,将文件内容写入资源文件
-            OutputStream outputStream = new FileOutputStream(RESOURCE_FILE_PATH);
+            OutputStream outputStream = new FileOutputStream(territoryConfigPojo.getPath());
             byte[] buffer = new byte[1024];
             int bytesRead;
             while ((bytesRead = inputStream.read(buffer)) != -1) {
@@ -72,7 +82,7 @@
             return ResponseResult.success("文件上传成功");
         } catch (IOException e) {
             e.printStackTrace();
-            return ResponseResult.error(e.getMessage()); // 将异常信息包含在响应中返回给客户端
+            return ResponseResult.error(e.getMessage());
         }
     }
 
@@ -89,9 +99,9 @@
      */
     @Transactional
     public MultipartFile listFile(String workspaceId, String waylineName, double airportLat, double airportLon) throws IOException {
-        List<TbDkjbxxEntity> list = mapper.selectList(null);
+        List<TbDkjbxxEntity> list = iTbDkjbxxMapper.selectList(null);
         List<LotInfo> info = dbConvertToEntity(list);
-        Coordinate[] coordinates = GeoToolsUtil.getRoutePointOrder(info, airportLat, airportLon);
+        List<PointPO> coordinates = GeoToolsUtil.getRoutePointOrder(info, airportLat, airportLon);
         // 创建XML模板模型
         XMLTemplateModel xmlModel = XMLTemplateModel.init(coordinates, info);
         CreateWaylineFileUtils.createWaylineFile(xmlModel, patchesConfigPojo.getTemplate(), patchesConfigPojo.getTargetTemplate(), patchesConfigPojo.getWaylines(), patchesConfigPojo.getTargetWaylines());
@@ -99,8 +109,7 @@
         String destKMZFile = patchesConfigPojo.getDestKMZFile() + waylineName + ".kmz"; // 输出的KMZ文件路径
         zipFolder(patchesConfigPojo.getSourceDir(), destKMZFile);
         // 将压缩文件转换为MultipartFile对象
-        MultipartFile multipartFile = convert(new File(destKMZFile));
-        return multipartFile;
+        return convert(new File(destKMZFile));
     }
 
     /**
@@ -112,6 +121,7 @@
     private List<LotInfo> dbConvertToEntity(List<TbDkjbxxEntity> list) {
         List<LotInfo> infos = new ArrayList<>();
         for (TbDkjbxxEntity file : list) {
+            double[] coordinates = extractCoordinates(pointCGCStoWGS(file.getXzb(), file.getYzb()));
             // 使用Builder模式构建LotInfo对象
             LotInfo.LotInfoBuilder builder = LotInfo.builder();
             if (file != null) {
@@ -122,12 +132,12 @@
                         .dkmc(file.getDkmc())
                         .dkbh(file.getDkbh())
                         .xmc(file.getXmc())
-                        .xzb(file.getXzb())
+                        .xzb(coordinates[0])
                         .sfbhzdk(file.getSfbhzdk())
                         .sjlx(file.getSjlx())
-                        .dkfw(file.getDkfw())
+                        .dkfw(poylonCGCStoWGS(file.getDkfw()))
                         .xzqdm(file.getXzqdm())
-                        .yzb(file.getYzb())
+                        .yzb(coordinates[1])
                         .kzxx(file.getKzxx())
                         .dklx(file.getDklx())
                         .build();

--
Gitblit v1.9.3