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


@@ -3,28 +3,55 @@


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.google.gson.JsonObject;


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.media.service.impl.FileServiceImpl.uploadFile;


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




@Service


@DS("sqlite")


@DS("sqlite-ret")


public class TbFjServiceImpl implements ITbFJService {


    @Autowired


    private ITbFjMapper mapper;




    private ITbFjMapper tbFjMapper;


    @Autowired


    private PatchesConfigPojo patchesConfigPojo;


    @Autowired


    private TerritoryConfigPojo territoryConfigPojo;


    @Autowired


    private MinioPojo pojo;


    @Autowired


    private IFileService fileService;


    @Autowired


    private WaterMarkUtil waterMarkUtil;


    /**


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


     *


@@ -33,7 +60,8 @@


     * @return


     * @throws IOException


     */


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


    @Transactional


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


        int count = 0;


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


        TbFjEntity tbFj;


@@ -41,10 +69,21 @@


            MediaFileEntity file = mediaFile.get(i);


            tbFj = dbConvertToEntity(file, lotInfo);


            list.add(tbFj);


            mapper.insert(tbFj);


            tbFjMapper.insert(tbFj);


            count++;


        }


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


    }




    /**


@@ -55,77 +94,120 @@


     * @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());


        JSONObject jsonObject = JSONObject.parseObject(jsonString);


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


        int psjd = 0;


        String fjhxz = "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


        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";


        Long pssj = mediaFile.getCreateTime();


        int fjlx = 1;


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


        File file1=WaterMark.addWatermark(file,patchesConfigPojo.getUnzip(),pssj,lat,lng);


        byte[] FJ = fileToByteArray(file1);


        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, 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(lat)


                    .longitude(lng)


                    .xdgd(truncateToTwoDecimalPlaces(relativeAltitude))


                    .jdgd(truncateToTwoDecimalPlaces(absoluteAltitude))


                    .Latitude(truncateToSevenDecimalPlaces(lat))


                    .longitude(truncateToSevenDecimalPlaces(lng))


                    .fj(FJ)


                    .psjj(psjj)


                    .fjmc(fjmc)


                    .fjlx(1)


                    .psfyj(gimbalYawDegree)


                    .pssj(String.valueOf(pssj))


                    .psjd(psjd)


                    .fjlx(fjlx)


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


        String localFilePath =  patchesConfigPojo.getUnzip()+ "tmp.jpg";




        try {


            URL url = new URL(fileUrl);


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


            connection.setRequestMethod("GET");




            downloadedFile = new File(localFilePath);


            // 创建临时文件


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


@@ -134,9 +216,6 @@


                    outputStream.write(buffer, 0, bytesRead);


                }


            }




            System.out.println("File downloaded and saved at: " + downloadedFile.getAbsolutePath());




        } catch (IOException e) {


            e.printStackTrace();


        }


@@ -144,6 +223,13 @@


        return downloadedFile;


    }




    /**


     * 将文件转换为字节数组。


     *


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


     * @return 文件内容的字节数组。


     * @throws IOException 如果读取文件发生错误。


     */


    public static byte[] fileToByteArray(File file) throws IOException {


        FileInputStream fis = new FileInputStream(file);


        byte[] data = new byte[(int) file.length()];


@@ -151,5 +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();




    }


}