无人机java后台代码
blame | history | patch | commit | commitdiff | ignore whitespace
rain
2024-08-16 3430a3dd889ba51d06cbdb77014aa63d0a070e48 
 src/main/java/com/dji/sample/territory/service/impl/TbFjServiceImpl.java


@@ -1,27 +1,41 @@


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




import com.alibaba.fastjson.JSON;


import com.alibaba.fastjson.JSONObject;


import com.baomidou.dynamic.datasource.annotation.DS;


import com.dji.sample.media.model.MediaFileEntity;


import com.dji.sample.media.dao.IMarkMapper;


import com.dji.sample.media.model.*;


import com.dji.sample.media.service.IFileService;


import com.dji.sample.media.util.ImgUtil;


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


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


import com.dji.sample.territory.dao.ITbFjMapper;


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


import com.dji.sample.territory.pojo.TerritoryConfigPojo;


import com.dji.sample.territory.service.ITbFJService;


import com.dji.sample.territory.utils.VideoZipUtil;


import com.dji.sample.territory.utils.WaterMarkUtil;


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


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


import org.springframework.stereotype.Service;


import org.springframework.transaction.annotation.Transactional;




import javax.imageio.ImageIO;


import java.awt.image.BufferedImage;


import java.io.*;


import java.math.BigDecimal;


import java.math.RoundingMode;


import java.net.HttpURLConnection;


import java.net.URL;


import java.nio.file.Files;


import java.time.Instant;


import java.time.LocalDateTime;


import java.time.ZoneId;


import java.time.format.DateTimeFormatter;


import java.util.ArrayList;


import java.util.List;


import java.util.UUID;




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


import static com.dji.sample.media.service.impl.FileServiceImpl.uploadFile;


import static com.dji.sample.territory.utils.SM3.sm3;




@Service


@DS("sqlite-ret")


@@ -30,7 +44,14 @@


    private ITbFjMapper tbFjMapper;


    @Autowired


    private PatchesConfigPojo patchesConfigPojo;




    @Autowired


    private TerritoryConfigPojo territoryConfigPojo;


    @Autowired


    private MinioPojo pojo;


    @Autowired


    private IFileService fileService;


    @Autowired


    private WaterMarkUtil waterMarkUtil;


    /**


     * 按照地块编号所对应的信息和音视频文件存入sqlite数据库


     *


@@ -40,7 +61,7 @@


     * @throws IOException


     */


    @Transactional


    public int insertData(List<MediaFileEntity> mediaFile, LotInfo lotInfo) throws IOException {


    public int insertData(List<MediaFileEntity> mediaFile, LotInfo lotInfo) throws Exception {


        int count = 0;


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


        TbFjEntity tbFj;


@@ -54,6 +75,17 @@


        return count;


    }




    public int insertOneData(MediaFileEntity mediaFile, LotInfo lotInfo) throws Exception {


        TbFjEntity tbFj = dbConvertToEntity(mediaFile, lotInfo);


            tbFjMapper.insert(tbFj);


        return 0;


    }




    @Override


    public void deleteData() {


        tbFjMapper.delete(null);


    }




    /**


     * 将Media和Lotinfo实体类转化为TbFjEntity实体类


     *


@@ -62,7 +94,7 @@


     * @return


     * @throws IOException


     */


    private TbFjEntity dbConvertToEntity(MediaFileEntity mediaFile, LotInfo lotInfo) throws IOException {


    private TbFjEntity dbConvertToEntity(MediaFileEntity mediaFile, LotInfo lotInfo) throws Exception {


        File file1 = null;


        byte[] FJ = null;


        String jsonString = JSONObject.toJSONString(mediaFile.getMetadata());


@@ -70,87 +102,119 @@


        Double absoluteAltitude = jsonObject.getDouble("absoluteAltitude");


        Double relativeAltitude = jsonObject.getDouble("relativeAltitude");


        Double gimbalYawDegree = jsonObject.getDouble("gimbalYawDegree");


        Long sj = jsonObject.getLong("createdTime");


        JSONObject shootPosition = jsonObject.getJSONObject("shootPosition");


        Double lat = shootPosition.getDouble("lat");


        Double lng = shootPosition.getDouble("lng");


        double[] coordinates = extractCoordinates(pointWGStoCGCS(lng, lat));


        double x = Double.parseDouble(((formatCoordinate(coordinates[1]))));


        double y = Double.parseDouble(((formatCoordinate(coordinates[0]))));


        PositionDTO position=new PositionDTO();


        position.setLat(lat);


        position.setLng(lng);


//        double[] coordinates = extractCoordinates(pointWGStoCGCS(lng, lat));


//        double x = Double.parseDouble(((formatCoordinate(coordinates[1]))));


//        double y = Double.parseDouble(((formatCoordinate(coordinates[0]))));


        //本地数据库没有的数据在FJ表非空用0


        int psjd = 0;


        String fjhxz = "0";


        double psjd = WaterMarkUtil.convertAngle(gimbalYawDegree);


        int pshgj = 0;


        String jym = "0";


        String psry = "0";


        String zsdm = "0";


        String psry = "中图智绘无人机";


        String zsdm = "23C8CCC61E3042FBA6A658F319337B1A";


        String dklx = "0";


        String xzqdm = "0";


        int fjlx = 1;


        Long pssj = mediaFile.getCreateTime();


        String pstz = "Y";


        String pssj = convertTimestampToDateTime(sj);


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


        String fjmc = mediaFile.getFileName();


        String key = mediaFile.getObjectKey();


        String head = "http://dev.jxpskj.com:9000/cloud-bucket";


        String url = head + key;


        File file = downloadFile(url);


        JSONObject jsons = (JSONObject) (ImgUtil.getInfo(file));


        String str = jsons.toJSONString();


        String newstr = str.replace("+", "");


        JSONObject json = JSON.parseObject(newstr);


        double pitch = json.getDouble("GimbalPitchDegree");


        String focal = json.getString("FocalLength");


        String numberPart = focal.replaceAll("[^\\d]", "");


        int psjj = Integer.parseInt(numberPart);


        json.getString("GimbalYawDegree");


        BufferedImage image = ImageIO.read(file);


        int width = image.getWidth();


        int height = image.getHeight();


        String fjhxz = null;


        //对应图片和视频文件进行不同处理


        boolean endsWith = key.endsWith(".mp4");


        if (!endsWith) {


            file1 = WaterMarkUtil.addWatermark(file, patchesConfigPojo.getUnzip(), pssj, lat, lng,gimbalYawDegree);


            file1 = waterMarkUtil.addWatermark(file, sj, lat, lng, gimbalYawDegree,lotInfo.getType());


            byte[] bytesArray = Files.readAllBytes(file1.toPath());


            fjhxz= sm3(bytesArray);


            FJ = fileToByteArray(file1);


//            fileService.saveMarkFile(lotInfo.getWorkspaceId(),mediaFile,file1);


        } else {


            file1 = VideoZipUtil.compressVideo(file, 800000, 128000, 1280, 720);


            FJ = fileToByteArray(file1);


        }


        StringBuffer buffer=SM3.addStringBuffer(fjhxz,pssj,truncateToSevenDecimalPlaces(lng),truncateToSevenDecimalPlaces(lat),(int)pitch,(int)psjd,pshgj,psry,zsdm);


        String jyms= SM3.addJym(buffer);


        TbFjEntity.TbFjEntityBuilder builder = TbFjEntity.builder();


        if (lotInfo != null) {


            builder.bsm(bsm)


                    .dkbsm(lotInfo.getDkbh())


                    .xzqdm(xzqdm)


                    .dkbsm(lotInfo.getBsm())


                    .xzqdm(lotInfo.getXzqdm())


                    .dklx(dklx)


                    .xdgd(relativeAltitude)


                    .jdgd(absoluteAltitude)


                    .Latitude(x)


                    .longitude(y)


                    .xdgd(truncateToTwoDecimalPlaces(relativeAltitude))


                    .jdgd(truncateToTwoDecimalPlaces(absoluteAltitude))


                    .Latitude(truncateToSevenDecimalPlaces(lat))


                    .longitude(truncateToSevenDecimalPlaces(lng))


                    .fj(FJ)


                    .psjj(psjj)


                    .fjmc(fjmc)


                    .fjlx(fjlx)


                    .psfyj(gimbalYawDegree)


                    .pssj(String.valueOf(pssj))


                    .psjd(psjd)


                    .psfyj(pitch)


                    .pssj(pssj)


                    .psjd((int) psjd)


                    .fjhxz(fjhxz)


                    .pshgj(pshgj)


                    .zsdm(zsdm)


                    .psry(psry)


                    .jym(jym)


                    .jym(jyms)


                    .url(url)


                    .pstz(pstz)


                    .fjyskd(width)


                    .fjysgd(height)


                    .build();


        } else {


            throw new IllegalArgumentException("未匹配到相关地块");


        }


//        else {


////            throw new IllegalArgumentException("未匹配到相关地块");


//        }


        file.delete();


        return builder.build();


    }




    /**


     * 将音视频文件转换成字节流


     * 文件下载


     *


     * @param fileUrl


     * @return


     */


    public File downloadFile(String fileUrl) {


    public static File downloadFile(String fileUrl) {


        File downloadedFile = null;




        try {


            URL url = new URL(fileUrl);


            HttpURLConnection connection = (HttpURLConnection) url.openConnection();


            connection.setRequestMethod("GET");


            downloadedFile = new File(File.createTempFile("temp", ".jpeg").toURI());


            InputStream inputStream = connection.getInputStream();


            OutputStream outputStream = new FileOutputStream(downloadedFile);


            byte[] buffer = new byte[1024];


            int bytesRead;


            while ((bytesRead = inputStream.read(buffer)) != -1) {


                outputStream.write(buffer, 0, bytesRead);




            // 创建临时文件


            File tempFile = File.createTempFile("temp", ".jpeg");


            downloadedFile = tempFile;




            // 使用 try-with-resources 确保流被关闭


            try (InputStream inputStream = connection.getInputStream();


                 OutputStream outputStream = new FileOutputStream(downloadedFile)) {


                byte[] buffer = new byte[1024];


                int bytesRead;


                while ((bytesRead = inputStream.read(buffer)) != -1) {


                    outputStream.write(buffer, 0, bytesRead);


                }


            }


        } catch (IOException e) {


            e.printStackTrace();


@@ -173,4 +237,34 @@


        fis.close();


        return data;


    }


}


    public static String convertTimestampToDateTime(long timestamp) {


        // 将时间戳转换为 Instant 对象


        Instant instant = Instant.ofEpochMilli(timestamp);




        // 将 Instant 对象转换为 LocalDateTime 对象


        LocalDateTime dateTime = LocalDateTime.ofInstant(instant, ZoneId.systemDefault());




        // 格式化日期时间


        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");


        return dateTime.format(formatter);


    }




    public static Double truncateToTwoDecimalPlaces(Double value) {


        //小数点只保留两位


        if (value == null) {


            return null;  // 处理null值情况


        }


        BigDecimal bd = new BigDecimal(value).setScale(2, RoundingMode.DOWN);


        return bd.doubleValue();


    }




    public static Double truncateToSevenDecimalPlaces(Double value) {


        //小数点只保留七位


        if (value == null) {


            return null;  // 处理null值情况


        }


        BigDecimal bd = new BigDecimal(value).setScale(7, RoundingMode.DOWN);


        return bd.doubleValue();




    }


}