无人机java后台代码
blame | history | patch | commit | commitdiff | ignore whitespace
zrj
2024-08-20 8b7b86f12fab51fe21b5575f4ffd524fbeaf412b 
 src/main/java/com/dji/sample/patches/service/impl/ShpToDataSourceServiceImpl.java


@@ -1,7 +1,14 @@


package com.dji.sample.patches.service.impl;




import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;


import com.dji.sample.patches.config.pojo.PatchesConfigPojo;


import com.dji.sample.patches.constant.FileSufConstant;


import com.dji.sample.patches.dao.ShpToDataSourceMapper;


import com.dji.sample.patches.kml.KmlData;


import com.dji.sample.patches.kml.KmlLine;


import com.dji.sample.patches.kml.KmlPoint;


import com.dji.sample.patches.kml.KmlPolygon;


import com.dji.sample.patches.model.dto.ShpDTO;


import com.dji.sample.patches.model.entity.LotInfo;


import com.dji.sample.patches.service.ShpToDataSourceService;


import com.dji.sample.patches.utils.*;


@@ -9,7 +16,11 @@


import com.dji.sample.patches.xml.utils.CreateWaylineFileUtils;


import com.dji.sample.territory.model.entity.TbDkjbxxEntity;


import com.dji.sample.wayline.service.IWaylineFileService;


import org.locationtech.jts.geom.Coordinate;


import de.micromata.opengis.kml.v_2_2_0.Coordinate;


import lombok.extern.slf4j.Slf4j;


import org.jetbrains.annotations.NotNull;


import org.jetbrains.annotations.Nullable;


import org.locationtech.jts.io.ParseException;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.beans.factory.annotation.Value;


import org.springframework.stereotype.Service;


@@ -17,17 +28,22 @@


import org.springframework.web.multipart.MultipartFile;




import java.io.File;


import java.io.IOException;


import java.io.InputStream;


import java.util.ArrayList;


import java.util.Collections;


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.TimerUtil.getNowDay;


import static com.dji.sample.patches.utils.ZipUtil.*;


import static com.dji.sample.territory.utils.CoordinateSystemUtil.*;


import static org.locationtech.jts.io.WKTConstants.POLYGON;




@Service


@Slf4j


public class ShpToDataSourceServiceImpl implements ShpToDataSourceService {




    @Autowired


@@ -42,50 +58,190 @@


    private PatchesConfigPojo patchesConfigPojo;




    @Transactional


    public MultipartFile insertGeo(MultipartFile file, String workspaceId, String waylineName, double airportLat, double airportLon) throws Exception {


        List<LotInfo> list = new ArrayList<>();


        File file1 = MultipartFileTOFileUtil.multipartFile2File(file, patchesConfigPojo.getUnzip());


        List<String> s = ShapeFileUtil.shpToGeoJson(file1);


        String[] arr1 = FormatConversionUtil.formatConversion(s);


        for (int i = 0; i < arr1.length; i++) {


    public MultipartFile insertGeo(MultipartFile file,


                                   String workspaceId,


                                   String waylineName,


                                   double airportLat,


                                   double airportLon,


                                   String creator){


        // zip 格式


        if (file.getOriginalFilename().endsWith(FileSufConstant.FILE_SUFFIX_ZIP)) {


            List<LotInfo> list = new ArrayList<>();


            File file1 = MultipartFileTOFileUtil.multipartFile2File(file, patchesConfigPojo.getUnzip());


            List<ShpDTO> shpData = null;


            try {


                shpData = ShapeFileUtil.shpToGeoJson(file1);


            } catch (IOException | ParseException e) {


                throw new RuntimeException(e);


            }


            for (ShpDTO shpDatum : shpData) {


                LotInfo lotInfo = new LotInfo();


                String uuid = UUID.randomUUID().toString();


                String bsm = uuid.replaceAll("-", "");


                lotInfo.setBsm(bsm);


                lotInfo.setUserName(creator);


                lotInfo.setWorkspaceId(workspaceId);


                lotInfo.setDkfw(poylonCGCStoWGS(convertToWKT(shpDatum.getDKFW())));


                lotInfo.setDkbh(shpDatum.getDKBH());


                lotInfo.setXzb(shpDatum.getXZB());


                lotInfo.setYzb(shpDatum.getYZB());


                lotInfo.setXzqdm(shpDatum.getXZQDM());


                lotInfo.setXmc(DistrictCodeUtils.codeToName(lotInfo.getXzqdm()));


                list.add(lotInfo);


                shpToDataSourceMapper.insert(lotInfo);


            }


            MultipartFile multipartFile = createWaylineFile(waylineName, airportLat, airportLon, list);


            MultipartFileTOFileUtil.deleteFile(file1);


            return  multipartFile;


        }


        // kmz 格式


        if (file.getOriginalFilename().endsWith(FileSufConstant.FILE_SUFFIX_KMZ)){


            try {


                InputStream inputStream = file.getInputStream();


                // 转换 kmz 文件并获取数据


                KmlData kmlData = ParsingKmlUtil.parseKMZFile(inputStream);


                log.info("kmlData:{}",kmlData);


                MultipartFile multipartFile = insertLOTAndGetMultipartFile(workspaceId, waylineName, airportLat, airportLon, creator, kmlData);


                // 返回


                if (multipartFile != null) return multipartFile;


            } catch (IOException e) {


                throw new RuntimeException(e);


            }


        }


        // kml 格式


        if (file.getOriginalFilename().endsWith(FileSufConstant.FILE_SUFFIX_KML)) {


            try {


                InputStream inputStream = file.getInputStream();


                KmlData kmlData = ParsingKmlUtil.parseKmlByInputstream(inputStream);


                log.info("kmlData:{}",kmlData);


                MultipartFile multipartFile = insertLOTAndGetMultipartFile(workspaceId, waylineName, airportLat, airportLon, creator, kmlData);


                // 返回


                if (multipartFile != null) return multipartFile;


            } catch (IOException e) {


                throw new RuntimeException(e);


            }


        }


        return null;


    }




    /**


     * 保存图斑信息和生成航线文件


     * @param workspaceId


     * @param waylineName


     * @param airportLat


     * @param airportLon


     * @param creator


     * @param kmlData


     * @return


     * @throws IOException


     */


    @Nullable


    public MultipartFile insertLOTAndGetMultipartFile(String workspaceId,


                                                      String waylineName,


                                                      double airportLat,


                                                      double airportLon,


                                                      String creator,


                                                      KmlData kmlData


    ) {


        if (null!=kmlData){


            List<LotInfo> list = new ArrayList<>();


            // 点、线暂不处理


            if (null!=kmlData.getKmlPoints() && kmlData.getKmlPoints().size()>0){


                List<KmlPoint> kmlPointList = kmlData.getKmlPoints();


            }


            if (null!=kmlData.getKmlLines() && kmlData.getKmlLines().size()>0){


                List<KmlLine> kmlLineList = kmlData.getKmlLines();


            }


            // 处理面状数据


            if (null!=kmlData.getKmlPolygons() && kmlData.getKmlPolygons().size()>0){


                List<KmlPolygon> kmlPolygonList = kmlData.getKmlPolygons();


                // 保存图斑信息


                this.saveLot(workspaceId, creator, list, kmlPolygonList);


                // 生成航线文件


                MultipartFile multipartFile = createWaylineFile(waylineName, airportLat, airportLon, list);


                // 返回


                return  multipartFile;


            }


        }


        return null;


    }




    /**


     * 创建航线文件


     * @param waylineName 航线名称


     * @param airportLat


     * @param airportLon


     * @param list


     * @return


     * @throws IOException


     */


    @NotNull


    public MultipartFile createWaylineFile(String waylineName, double airportLat, double airportLon, List<LotInfo> list) {


        try {


            List<PointPO> coordinates = GeoToolsUtil.getRoutePointOrder(list, airportLat, airportLon);


            XMLTemplateModel xmlModel = XMLTemplateModel.init(coordinates, list);


            CreateWaylineFileUtils.createWaylineFile(xmlModel, patchesConfigPojo.getTemplate(), patchesConfigPojo.getTargetTemplate(), patchesConfigPojo.getWaylines(), patchesConfigPojo.getTargetWaylines());


            // 压缩文件夹中的内容


            String destKMZFile = patchesConfigPojo.getDestKMZFile() + waylineName + ".kmz"; // 输出的KMZ文件路径


            zipFolder(patchesConfigPojo.getSourceDir(), destKMZFile);


            return convert(new File(destKMZFile));


        }catch (IOException e){


            throw new RuntimeException(e);


        }


    }




    /**


     * 保存图斑信息


     * @param workspaceId


     * @param creator


     * @param list


     * @param kmlPolygonList


     */


    public void saveLot(String workspaceId, String creator, List<LotInfo> list, List<KmlPolygon> kmlPolygonList) {


        for (KmlPolygon kmlPolygon : kmlPolygonList) {


            LotInfo lotInfo = new LotInfo();


            String temp = arr1[i].trim();


            temp = FormatConversionUtil.modifySpacesAndCommas(temp);


            json = head + temp;


            String uuid = UUID.randomUUID().toString();


            String bsm = uuid.replaceAll("-", "");


            lotInfo.setBsm(bsm);


            lotInfo.setUserName(creator);


            lotInfo.setWorkspaceId(workspaceId);


            lotInfo.setDkfw(json);


            lotInfo.setDkbh("dkbh" + i);


            lotInfo.setDkbh(getNowDay());


            lotInfo.setDkfw(handlePolygon(kmlPolygon.getPoints()));


            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]));


            shpToDataSourceMapper.insert(lotInfo);


        }


        List<PointPO> coordinates = GeoToolsUtil.getRoutePointOrder(list, airportLat, airportLon);


        XMLTemplateModel xmlModel = XMLTemplateModel.init(coordinates, list);


        CreateWaylineFileUtils.createWaylineFile(xmlModel, patchesConfigPojo.getTemplate(), patchesConfigPojo.getTargetTemplate(), patchesConfigPojo.getWaylines(), patchesConfigPojo.getTargetWaylines());


        // 压缩文件夹中的内容


        String destKMZFile = patchesConfigPojo.getDestKMZFile() + waylineName + ".kmz"; // 输出的KMZ文件路径


        zipFolder(patchesConfigPojo.getSourceDir(), destKMZFile);


        multipartFile = convert(new File(destKMZFile));


        MultipartFileTOFileUtil.deleteFile(file1);


        return multipartFile;


    }




    /**


     * 处理面状数据


     * @param coordinates


     * @return


     */


    private String handlePolygon(List<de.micromata.opengis.kml.v_2_2_0.Coordinate> coordinates) {


        // 转换为WKT格式


        StringBuilder wkt = new StringBuilder("POLYGON((");


        for (Coordinate coordinate : coordinates) {


            wkt.append(coordinate.getLongitude()).append(" ").append(coordinate.getLatitude()).append(",");


        }


        // 添加第一个点的坐标以闭合多边形


        wkt.append(coordinates.get(0).getLongitude()).append(" ").append(coordinates.get(0).getLatitude()).append("))");


        // 返回


        return wkt.toString();


    }




    @Transactional


    public void savaInMysql(List<TbDkjbxxEntity> list, String workspaceId) {


    public void savaInMysql(List<TbDkjbxxEntity> list, String workspaceId,String id,String name) {


        for (int i = 0; i < list.size(); i++) {


            LotInfo lotInfo = new LotInfo();


            lotInfo.setWorkspaceId(workspaceId);


            lotInfo = dbConvertToEntity(list.get(i), workspaceId);


            lotInfo.setTaskId(id);


            lotInfo.setType(2);


            lotInfo.setTaskName(name);


            lotInfo = dbConvertToEntity(list.get(i), workspaceId, id, name);


            shpToDataSourceMapper.insert(lotInfo);


        }


    }






    /**


     * 将SQlite数据库中DKJBXX对象转换为LotInfo对象。


@@ -93,27 +249,57 @@


     * @param file 需要转换的DKJBXX对象。


     * @return 返回一个构建好的LotInfo对象，包含从数据库实体中转换来的信息。


     */


    private LotInfo dbConvertToEntity(TbDkjbxxEntity file, String workspaceId) {


        double[] coordinates = extractCoordinates(pointCGCStoWGS(file.getXzb(), file.getYzb()));


    private LotInfo dbConvertToEntity(TbDkjbxxEntity file, String workspaceId,String id,String name) {


        LotInfo.LotInfoBuilder builder = LotInfo.builder();


        builder.bsm(file.getBsm())


                .bz(file.getBz())


                .dkmj(file.getDkmj())


                .dkmc(file.getDkmc())


                .dkbh(file.getDkbh())


                .xmc(file.getXmc())


                .xzb(coordinates[0])


                .sfbhzdk(file.getSfbhzdk())


                .sjlx(file.getSjlx())


                .dkfw(poylonCGCStoWGS(file.getDkfw()))


                .xzqdm(file.getXzqdm())


                .yzb(coordinates[1])


                .kzxx(file.getKzxx())


                .dklx(file.getDklx())


        builder.bsm(file.getFId())


                .xzqdm(file.getFXzqdm())


                .dkbh(file.getFTbbh())


                .dkfw(file.getFShape())


                .workspaceId(workspaceId)


                .isPlan(0)


                .isPush(0)


                .xmc(DistrictCodeUtils.nameToCode(file.getFXzqdm()))


                .investigate(0)


                .taskId(id)


                .taskName(name)


                .build();


        return builder.build();


    }




//    private LotInfo shpDtoToLotInfo(ShpDTO shpDTO){


//        LotInfo.LotInfoBuilder builder=LotInfo.builder();


//        builder.


//


//    }




    public List<LotInfo> getNoPlan(){


   return shpToDataSourceMapper.selectList(new LambdaQueryWrapper<LotInfo>().eq(LotInfo::getIsPlan,0));


    }


    public static String convertToWKT(String coordinates) {


        // Remove outermost square brackets and split by comma


        String cleanedCoordinates = coordinates.substring(4, coordinates.length() - 4);


        String[] polygons = cleanedCoordinates.split("\\]\\],\\[\\[");


        StringBuilder wkt = new StringBuilder("MULTIPOLYGON(");




        for (String polygon : polygons) {


            wkt.append("((");


            String[] points = polygon.split("\\],\\[");


            for (String point : points) {


                String[] coords = point.split(",");


                double x = Double.parseDouble(coords[0]);


                double y = Double.parseDouble(coords[1]);


                wkt.append(x).append(" ").append(y).append(", ");


            }


            // Remove the last comma and space


            wkt.setLength(wkt.length() - 2);


            wkt.append(")), ");


        }


        // Remove the last comma and space, and close the polygon


        wkt.setLength(wkt.length() - 2);


        wkt.append(")");




        return wkt.toString();


    }


}