4dkankan-center-modeling-control/src/main/java/com/fdkankan/contro/mq/service/impl/BuildSceneServiceImpl.java

package com.fdkankan.contro.mq.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.file.FileNameUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.ZipUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.fdkankan.common.constant.*;
import com.fdkankan.common.util.FileUtils;
import com.fdkankan.contro.bean.SceneJsonBean;
import com.fdkankan.contro.entity.*;
import com.fdkankan.contro.mq.service.IBuildSceneService;
import com.fdkankan.contro.service.*;
import com.fdkankan.contro.vo.SceneEditControlsVO;
import com.fdkankan.fyun.config.FYunFileConfig;
import com.fdkankan.fyun.constant.FYunTypeEnum;
import com.fdkankan.fyun.face.FYunFileServiceInterface;
import com.fdkankan.image.MatrixToImageWriterUtil;
import com.fdkankan.model.constants.ConstantFileName;
import com.fdkankan.model.constants.ConstantFilePath;
import com.fdkankan.model.constants.UploadFilePath;
import com.fdkankan.model.utils.CreateHouseJsonUtil;
import com.fdkankan.model.utils.CreateObjUtil;
import com.fdkankan.push.config.PushMessageConfig;
import com.fdkankan.push.utils.PushMsgUtil;
import com.fdkankan.rabbitmq.bean.BuildSceneCallMessage;
import com.fdkankan.rabbitmq.bean.BuildSceneResultMqMessage;
import com.fdkankan.rabbitmq.util.RabbitMqProducer;
import com.fdkankan.redis.constant.RedisKey;
import com.fdkankan.redis.util.RedisUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.io.File;
import java.io.IOException;
import java.util.*;
import java.util.Map.Entry;
import java.util.stream.Collectors;


/**
 * <p>
 * TODO
 * </p>
 *
 * @author dengsixing
 * @since 2022/4/20
 **/
@Slf4j
@Service
public class BuildSceneServiceImpl implements IBuildSceneService {

    @Value("${main.url}")
    private String mainUrl;
    @Value("${scene.pro.new.url}")
    private String sceneProNewUrl;

    @Value("${queue.modeling.modeling-call}")
    private String queueModelingCall;

    @Autowired
    private RabbitMqProducer mqProducer;

    @Autowired
    private FYunFileServiceInterface fYunFileService;

    @Autowired
    private ISceneFileBuildService sceneFileBuildService;
    @Autowired
    private ICameraDetailService cameraDetailService;
    @Autowired
    private ISceneEditInfoService sceneEditInfoService;
    @Autowired
    private ISceneEditControlsService sceneEditControlsService;

    @Autowired
    private FYunFileConfig fYunFileConfig;

    @Autowired
    private RedisUtil redisUtil;
    @Autowired
    private IScenePlusService scenePlusService;
    @Autowired
    private IScenePlusExtService scenePlusExtService;
    @Autowired
    private ISceneEditInfoExtService sceneEditInfoExtService;

    @Autowired
    private IUserIncrementService userIncrementService;

    @Autowired
    private IFdkkLaserService fdkkLaserService;

    @Override
    public void buildScenePre(BuildSceneCallMessage message) {
        boolean success = false;
        try {
            //根据相机类型，组装资源路径
            String path = message.getDataSource();

            //下载资源到本地
            this.downLoadSource(message, path);

            //发送mq，就进行计算
            mqProducer.sendByWorkQueue(queueModelingCall, message);

            success = true;

            log.info("场景计算资源准备结束，场景码：{}", message.getSceneNum());

        }catch (Exception e){
            log.error("场景计算前置处理出错", e);
        }finally {
            //如果前置处理失败，发送钉钉消息
            if(!success){
                String serverPath = message.getPath().substring(0,message.getPath().lastIndexOf("/")+1).concat(message.getPrefix());
//            buildSceneDTService.handFail("场景计算前置处理出错", serverPath, message.getSceneNum(), BuildScenePreServiceImpl.hostName, null);
            }

        }
    }

    @Override
    public void downLoadSource(BuildSceneCallMessage buildSceneMqMessage,String path){
        String ossPath = ConstantFilePath.OSS_PREFIX
                + path.replace(ConstantFilePath.BUILD_MODEL_PATH, "")
                .replace(ConstantFilePath.BUILD_MODEL_LASER_PATH, "");
        if (!ossPath.endsWith("/")) {
            ossPath = ossPath.concat("/");
        }
        fYunFileService.downloadFileByCommand(path + File.separator + "capture", ossPath);
    }


    @Override
    public void buildScenePost(BuildSceneResultMqMessage message) throws Exception {
        Integer cameraType = Integer.parseInt(message.getCameraType());
        String sceneCode = message.getSceneCode();
        String fileId = message.getFileId();
        Long computeTime = message.getComputeTime();

        String path = message.getPath();

        Boolean buildSuccess = message.getBuildSuccess();
        JSONObject fdageData = getFdageData(path + File.separator + "capture" +File.separator+"data.fdage");
        Integer videoVersion = fdageData.getInteger("videoVersion");


        String prefix = message.getPrefix();
        try {
            // 上传计算日志
            //如果是重复计算，没有走到计算逻辑，不需要上传日志文件
            log.info("开始上传计算日志");
            String buildLogPath = String.format(UploadFilePath.BUILD_LOG_PATH, sceneCode);
            fYunFileService.uploadFile(path + File.separator + "console.log", buildLogPath + "console.log");
            log.info("计算日志上传完成");

            if (!buildSuccess) {
                log.error("建模失败，修改状态为失败状态");
                scenePlusService.update(new LambdaUpdateWrapper<ScenePlus>()
                        .set(ScenePlus::getSceneStatus, SceneStatus.FAILD.code())
                        .eq(ScenePlus::getNum, sceneCode));

                SceneFileBuild sceneFileBuild = sceneFileBuildService.findByFileId(fileId);
                if (Objects.nonNull(sceneFileBuild)) {
                    sceneFileBuild.setBuildStatus(BuildStatus.fail.code());
                    sceneFileBuildService.updateById(sceneFileBuild);
                }
                throw new RuntimeException("建模失败!");
            }

            Map<String, String> uploadFiles = getUploadFiles(path,cameraType,fdageData);

            //建模成功走以下逻辑
            log.info("cameraType:{}",cameraType);

            //计算场景消耗磁盘空间
            long space = this.calUseSpace(uploadFiles);

            if(cameraType < 3){
                this.updateDb4Sm(sceneCode, space);
                fYunFileService.uploadMulFiles(uploadFiles);
                return;
            }

            //读取计算结果文件生成videosJson
            JSONObject videosJson = this.getVideosJson(path, videoVersion, sceneCode, cameraType);

            //写入数据库
            ScenePlus scenePlus = this.updateDbPlus(sceneCode, space, videosJson.toJSONString(), computeTime, fileId,fdageData.containsKey("exportMeshObj") && fdageData.getIntValue("exportMeshObj") == 1);
            ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());

            //变更容量，arrearCap为true时，代表容量不足
            boolean arrearCap = this.updateSpace(scenePlus.getCameraId(), space);

            //如果相机容量不足，需要把场景的paystatus改为容量不足状态
            this.sealScene(arrearCap, scenePlus.getId(), scenePlus.getId());

            Object[] editInfoArr = this.updateEditInfo(scenePlus);
            SceneEditInfo sceneEditInfo = (SceneEditInfo)editInfoArr[0];
            SceneEditInfoExt sceneEditInfoExt = (SceneEditInfoExt)editInfoArr[1];
            SceneEditControls sceneEditControls = (SceneEditControls)editInfoArr[2];

            String outPathZh = ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+sceneCode+".png";
            String outPathEn = ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+sceneCode+"_en.png";
            //生成新的分享的二维码-中文版本
            MatrixToImageWriterUtil.createQRCode(scenePlusExt.getWebSite(), outPathZh, false,null);
            //生成新的分享的二维码-英文版本
            MatrixToImageWriterUtil.createQRCode(scenePlusExt.getWebSite() + "&lang=en", outPathEn, false, null);
            //上传二维码
            fYunFileService.uploadFile(outPathZh, String.format(UploadFilePath.DOWNLOADS_QRCODE, sceneCode) + sceneCode + ".png");
            fYunFileService.uploadFile(outPathEn, String.format(UploadFilePath.DOWNLOADS_QRCODE, sceneCode) + sceneCode + "_en.png");

            //上传全景图俯视图
            this.uploadFloorCad(path, sceneCode, uploadFiles);

            //上传文件
            log.info("开始上传场景计算结果数据，num:{}", sceneCode);
            fYunFileService.uploadMulFiles(uploadFiles);

            Map<String,String> damFileHeaders = new HashMap<>();
            damFileHeaders.put("Content-Encoding","gzip");

            if (!fYunFileService.getFyunType().equals(FYunTypeEnum.LOCAL.code())) {
                // dam 文件设置请求头
                uploadFiles.entrySet().stream().filter(entry -> FileNameUtil.extName(entry.getKey()).equals("dam"))
                        .filter(entry -> new File(entry.getKey()).exists())
                        .forEach(entry -> {
                            // gzip压缩
                            FileUtil.writeBytes(ZipUtil.gzip(new File(entry.getKey())), entry.getKey() + ".gzip");
                            // 重命名
                            FileUtil.rename(new File(entry.getKey() + ".gzip"), entry.getKey(), true);
                            fYunFileService.uploadFile(entry.getKey(), entry.getValue(), damFileHeaders);
                        });
            }

            //拷贝部分文件到编辑目录，用于用户编辑
            this.copyToEditDir(sceneCode);

            //生成houseTypejson并上传
            uploadFiles.entrySet().stream().filter(entry-> FileNameUtil.getName(entry.getKey()).equals("floorplan_cad.json"))
                    .forEach(entry-> uploadHouseTypeJson(sceneCode,entry.getKey()));

            //写scene.json
            log.info("生成scene.json上传oss并设置缓存，num:{}", sceneCode);
            this.writeSceneJson(sceneCode, videosJson,sceneEditInfo, sceneEditInfoExt, sceneEditControls, scenePlus,
                    scenePlusExt, arrearCap);

            //计算成功，通知APP
            Integer pushChannel = fdageData.getInteger("pushChannel");
            String pushToken = fdageData.getString("pushToken");
            this.pushMsgToApp(pushChannel,pushToken, cameraType, scenePlus.getTitle(), scenePlusExt.getWebSite());

            if (cameraType == 14) {
                //计算成功  激光转台相机 同步 请求
                fdkkLaserService.syncBuildResult(scenePlus, scenePlusExt);
            }

            CreateObjUtil.deleteFile(path.replace(ConstantFilePath.BUILD_MODEL_PATH, "/") + "/capture");

            log.info("场景计算结果处理结束，场景码：{}", message.getSceneCode());

        }catch (Exception e){
            String errorMsg = "场景计算结果处理出错！";
            log.error(errorMsg);
            e.printStackTrace();
            if(!buildSuccess){
                errorMsg = "场景构建失败！";
            }
            //发送钉钉
            String serverPath = message.getPath().substring(0,message.getPath().lastIndexOf("/")+1).concat(prefix);
//            buildSceneDTService.handFail(errorMsg, serverPath, sceneCode, BuildScenePreServiceImpl.hostName, BuildScenePreServiceImpl.contentExt);
        }
    }

    private Map<String, String> getUploadFiles(String path,Integer cameraType,JSONObject fdageData) throws Exception {
        ScenePlus scenePlus = scenePlusService.getByFileId(path);
        if (ObjectUtils.isEmpty(scenePlus)) {
            throw new Exception("未找到场景信息：" + path);
        }

        String projectNum = scenePlus.getNum();


        String dataViewPath = String.format(UploadFilePath.DATA_VIEW_PATH, projectNum);
        String imagesPath = String.format(UploadFilePath.IMG_VIEW_PATH, projectNum);
        String videoPath = String.format(UploadFilePath.VIDEOS_VIEW_PATH, projectNum);
        String resultsPath = path + File.separator + "results" + File.separator;

        String uploadJsonPath = resultsPath + "upload.json";
        String uploadData = FileUtils.readFile(uploadJsonPath);
        log.info("upload.json 文件路径：{}, 内容：{}", uploadJsonPath, uploadData);
        JSONArray array = JSONObject.parseObject(uploadData).getJSONArray("upload");


        JSONObject fileJson = null;
        String fileName = "";

        Map<String, String> map = new HashMap();

        for (int i = 0; i < array.size(); ++i) {
            fileJson = array.getJSONObject(i);
            fileName = fileJson.getString("file");
            String filePath = resultsPath + fileName;

            if (!(new File(filePath)).exists()) {
                throw new Exception(filePath + "文件不存在");
            }

            if ("vision2.txt".equals(fileName)) {
                CreateObjUtil.convertTxtToVisionmodeldata(resultsPath + "vision2.txt", resultsPath + "vision2.modeldata");
                map.put(resultsPath + "vision2.modeldata", imagesPath + "vision2.modeldata");
                map.put(resultsPath + "vision2.txt", imagesPath + "vision2.txt");
            }

            if (fileJson.getIntValue("clazz") == 2) {
                map.put(filePath, imagesPath + ConstantFileName.modelUUID + "_50k_texture_jpg_high1/" + fileName.replace("tex/", ""));
            } else if (fileJson.getIntValue("clazz") == 3) {
                map.put(filePath, imagesPath + "pan/high/" + fileName.replace("high/", ""));
            } else if (fileJson.getIntValue("clazz") == 4) {
                map.put(filePath, imagesPath + "pan/low/" + fileName.replace("low/", ""));
            } else if (fileJson.getIntValue("clazz") == 5) {
                map.put(filePath, imagesPath + fileName);
            } else if (fileJson.getIntValue("clazz") == 7) {
                map.put(filePath, imagesPath + fileName);
            } else if (fileJson.getIntValue("clazz") == 10) {
                String updown = FileUtils.readFile(filePath);
                JSONObject updownJson = JSONObject.parseObject(updown);
                String mappingOssPath = String.format("scene_edit_data/%s/data/", projectNum) + fileName.replace("updown", "mapping");
                map.put(filePath, mappingOssPath);
            } else {
                if (fileJson.getIntValue("clazz") == 11 || fileJson.getIntValue("clazz") == 12) {
                    map.put(filePath, videoPath + fileName.replace("videos/", ""));
                    if (fileName.contains(".mp4")) {
                        map.put(resultsPath + fileName.replace("mp4", "flv"), videoPath + fileName.replace("videos/", "").replace("mp4", "flv"));
                    }
                }

                if (fileJson.getIntValue("clazz") == 16) {
                    map.put(filePath, dataViewPath + fileName);
                }

                if (fileJson.getIntValue("clazz") == 18) {
                    map.put(filePath, imagesPath + fileName);
                }
            }
        }


        if ((fdageData.containsKey("exportMeshObj") && fdageData.getIntValue("exportMeshObj") == 1)
                || (!ObjectUtils.isEmpty(cameraType) && (!cameraType.equals(14)))) {
            CreateObjUtil.convertTxtToDam(path + File.separator + "results" + File.separator + "tex" + File.separator + "modeldata.txt", path + File.separator + "results" + File.separator + ConstantFileName.modelUUID + "_50k.dam");
            CreateObjUtil.convertDamToLzma(path + File.separator + "results/");
            CreateObjUtil.convertTxtToDam(path + File.separator + "results" + File.separator + "tex" + File.separator + "modeldata.txt", path + File.separator + "results" + File.separator + ConstantFileName.modelUUID + "_50k.dam");
            map.put(path + File.separator + "results" + File.separator + ConstantFileName.modelUUID + "_50k.dam.lzma", "images/images" + projectNum + "/" + ConstantFileName.modelUUID + "_50k.dam.lzma");
            map.put(path + File.separator + "results" + File.separator + ConstantFileName.modelUUID + "_50k.dam", "images/images" + projectNum + "/" + ConstantFileName.modelUUID + "_50k.dam");
        }

        CreateObjUtil.convertTxtToVisionmodeldata(resultsPath + "vision.txt", resultsPath + "vision.modeldata");
        map.put(resultsPath + "vision.txt", imagesPath + "vision.txt");
        map.put(resultsPath + "vision.modeldata", imagesPath + "vision.modeldata");

        log.info("数据转换完成：" + projectNum);

        map.put(resultsPath + ConstantFileName.modelUUID + "_50k.dam.lzma", imagesPath + ConstantFileName.modelUUID + "_50k.dam.lzma");
        map.put(resultsPath + ConstantFileName.modelUUID + "_50k.dam", imagesPath + ConstantFileName.modelUUID + "_50k.dam");
        FileUtil.touch("/mnt/4Dkankan/scene/data" + File.separator + "data" + projectNum);
        map.put(resultsPath + "floorplan.json", dataViewPath + "floor.json");
        map.put(resultsPath + "floorplan_cad.json", dataViewPath + "floorplan_cad.json");
        map.put(path + File.separator + "capture/stitch_params.txt", dataViewPath + "stitch_params.txt");
        map.put(path + File.separator + "capture/Up.xml", dataViewPath + "Up.xml");
        map.put(path + File.separator + "capture/Up2.xml", dataViewPath + "Up2.xml");
        map.put(path + File.separator + "capture/Up.txt", dataViewPath + "Up.txt");
        map.put(path + File.separator + "capture/Up2.txt", dataViewPath + "Up2.txt");
        return map;

    }

    private JSONObject getFdageData(String dataFdagePath) {
        log.info("dataFdagePath 文件路径 ：{}", dataFdagePath);
        String data = FileUtils.readFile(dataFdagePath);
        //获取data.fdage的内容
        JSONObject dataJson = new JSONObject();
        if(data!=null){
            dataJson = JSONObject.parseObject(data);
        }
        return dataJson;
    }

    private void uploadFloorCad(String path, String num, Map<String, String> uploadFiles){

        //户型图上传
        String  userEditPath = UploadFilePath.USER_EDIT_PATH + "floor-cad-%s.%s";
        String  userViewPath = UploadFilePath.USER_VIEW_PATH + "floor-cad-%s.%s";
        String floorCadPath = path + "/results/floorplan_cad";
        List<String> floorCadList = FileUtils.getFileList(floorCadPath);
        if(CollUtil.isNotEmpty(floorCadList)){
            floorCadList.stream().forEach(str->{
                String substring = str.substring(str.lastIndexOf(File.separator) + 1);
                String[] arr = substring.split("floor");
                String[] arr2 = arr[1].split("\\.");
                uploadFiles.put(str, String.format(userEditPath, num, arr2[0], arr2[1]));
                uploadFiles.put(str, String.format(userViewPath, num, arr2[0], arr2[1]));
            });
        }

    }

    private void writeSceneJson(String num, JSONObject videosJson, SceneEditInfo sceneEditInfo, SceneEditInfoExt sceneEditInfoExt,
                                SceneEditControls sceneEditControls, ScenePlus scenePlus, ScenePlusExt scenePlusExt, boolean arrearCap)  throws Exception{
        String dataViewPath = String.format(UploadFilePath.DATA_VIEW_PATH, num);

        String sceneJsonKey = dataViewPath + "scene.json";
        boolean exist = fYunFileService.fileExist(sceneJsonKey);
        //如果云端没有scene.json文件，生成一份
        if(!exist){
            SceneJsonBean sceneJson = new SceneJsonBean();
            BeanUtil.copyProperties(sceneEditInfoExt, sceneJson);
            BeanUtil.copyProperties(sceneEditInfo, sceneJson);
            SceneEditControlsVO sceneEditControlsVO = BeanUtil.copyProperties(sceneEditControls, SceneEditControlsVO.class);
            sceneJson.setControls(sceneEditControlsVO);
            sceneJson.setNum(num);
            sceneJson.setCreateTime(scenePlus.getCreateTime());
            sceneJson.setSceneResolution(scenePlusExt.getSceneResolution());
            sceneJson.setSceneFrom(scenePlusExt.getSceneFrom());
            sceneJson.setVideos(JSON.toJSONString(videosJson));
            if(arrearCap) {
                sceneJson.setPayStatus(PayStatus.NO_CAPACITY.code());
            }

            String sceneJsonStr = JSON.toJSONString(sceneJson);
            //上传sceneJson文件
            fYunFileService.uploadFile(sceneJsonStr.getBytes(), sceneJsonKey);
            //scenejson写入缓存
            redisUtil.set(String.format(RedisKey.SCENE_JSON, num), sceneJsonStr);
        }
        // 上传status JSON.

        JSONObject statusJson = new JSONObject();
        //临时将-2改成1，app还没完全更新
        statusJson.put("status", 1);
        statusJson.put("webSite", scenePlusExt.getWebSite());
        statusJson.put("sceneNum", num);
        statusJson.put("thumb", scenePlusExt.getThumb());
        statusJson.put("payStatus", 1);
        FileUtils.writeFile(ConstantFilePath.SCENE_PATH + "data/data" + num + File.separator + "status.json", statusJson.toString());

        fYunFileService.uploadFile(ConstantFilePath.SCENE_PATH + "data/data" + num + File.separator + "status.json",
                dataViewPath + "status.json");
    }

    private void pushMsgToApp(Integer pushChannel, String pushToken, int cameraType, String sceneName, String webSite){
        log.info("推送消息，渠道是 {}, 手机token是 {}", pushChannel, pushToken);
        if(Objects.isNull(pushChannel) && StrUtil.isBlank(pushToken)){
            return;
        }

        String title = sceneName + "计算完成";
        String body = "您上传的" + sceneName + "计算完成，点击查看";

        try{
            if(FYunTypeEnum.AWS.code().equals(fYunFileService.getFyunType())){
                PushMsgUtil.googlePushMsg(ConstantFilePath.BASE_PATH + "/refreshToken.json", pushToken,
                        title, body , webSite);
                return;
            }

            PushMessageConfig pushConfig = null;
            if(pushChannel == 0){
                if(cameraType == 10 || cameraType == 13){
                    //ios
                    pushConfig = new PushMessageConfig(PushMessageConfig.IOS_KEY_Z, PushMessageConfig.IOS_SECRET_Z);
                    pushConfig.sendIOSUnicast(pushToken,  "四维看看Minion",title, body, webSite);
                }else {
                    //ios
                    pushConfig = new PushMessageConfig(PushMessageConfig.IOS_KEY, PushMessageConfig.IOS_SECRET);
                    pushConfig.sendIOSUnicast(pushToken, "四维看看Pro",title, body, webSite);
                }
            }else {
                if(cameraType == 10 || cameraType == 13){
                    //ios
                    //安卓
                    pushConfig = new PushMessageConfig(PushMessageConfig.ANDROID_KEY_Z, PushMessageConfig.ANDROID_SECRET_Z);
                    pushConfig.sendAndroidUnicast2(pushToken, "四维看看Minion",title, body, webSite);
                }else {
                    //安卓
                    pushConfig = new PushMessageConfig(PushMessageConfig.ANDROID_KEY, PushMessageConfig.ANDROID_SECRET);
                    pushConfig.sendAndroidUnicast(pushToken, "四维看看Pro",title, body, webSite);
                }
            }
            log.info("消息推送结束！");
        }catch (Exception e){
            log.error("推送消息失败：", e);
        }
    }

    private void copyToEditDir(String num) throws IOException {

        String editImagesPath = String.format(UploadFilePath.IMG_EDIT_PATH, num);
        String viewImagesPath = String.format(UploadFilePath.IMG_VIEW_PATH, num);

        String editDataPath = String.format(UploadFilePath.DATA_EDIT_PATH, num);
        String viewDataPath = String.format(UploadFilePath.DATA_VIEW_PATH, num);

        Map<String, String> map = new HashMap<>();
        map.put(editImagesPath + "vision.modeldata", viewImagesPath + "vision.modeldata");
        map.put(editImagesPath + "vision2.modeldata", viewImagesPath + "vision2.modeldata");
        map.put(editDataPath + "floorplan_cad.json", viewDataPath + "floorplan_cad.json");

        for (Entry<String, String> entry : map.entrySet()) {
                fYunFileService.copyFileInBucket(entry.getValue(), entry.getKey());
        }
    }

    private JSONObject getVideosJson(String path, Integer videoVersion, String projectNum, int cameraType) throws Exception {
        //读取videos_hdr_param.json, 保存点位视频的value
        Map<String, Object> videoMap = new HashMap<>();
        String videosHdr = FileUtils.readFile(path + File.separator + "results/videos/videos_hdr_param.json");
        JSONArray videoArray = null;
        if(StringUtils.isNotEmpty(videosHdr)){
            videoArray = JSONObject.parseObject(videosHdr).getJSONArray("hdr_param");
        }
        if(videoArray != null){
            for(int i = 0, len = videoArray.size(); i < len; i++) {
                videoMap.put(videoArray.getJSONObject(i).getString("name"), videoArray.getJSONObject(i).getString("value"));
                if(videoArray.getJSONObject(i).containsKey("fov")){
                    videoMap.put(videoArray.getJSONObject(i).getString("name") + "_fov", videoArray.getJSONObject(i).getString("fov"));
                }
            }
        }

        //获取upload中的video视频名称
        String uploadData = FileUtils.readFile(path + File.separator + "results" +File.separator+"upload.json");
        JSONObject uploadJson = null;
        JSONArray array = null;
        if(uploadData!=null) {
            uploadJson = JSONObject.parseObject(uploadData);
            array = uploadJson.getJSONArray("upload");
        }
        JSONObject fileJson = null;
        String fileName = "";

        //计算ts文件的大小，并拼接成json格式
        JSONArray jsonArray = new JSONArray();
        JSONObject videoJson = null;
        JSONObject videosJson = new JSONObject();
        long videoSize = 0L;
        for(int i = 0, len = array.size(); i < len; i++) {
            fileJson = array.getJSONObject(i);
            fileName = fileJson.getString("file");
            if(fileJson.getIntValue("clazz") == 11 && fileName.contains(".mp4") && !fileName.contains("-ios.mp4")){
                videoJson = new JSONObject();
                videoJson.put("id", fileName.substring(
                    0, fileName.lastIndexOf(".")).replace("videos/", ""));

                //如果ts文件存在，就计算ts大小
                if(new File(path + File.separator + "results" +File.separator+ fileName.replace(".mp4", ".ts")).exists()){
                    videoSize = new File(path + File.separator + "results" +File.separator+ fileName.replace(".mp4", ".ts")).length();
                    videoJson.put("tsSize", videoSize);
                }
                if(videoMap.containsKey(videoJson.get("id"))){
                    videoJson.put("value", videoMap.get(videoJson.get("id")));
                }
                if(videoMap.containsKey(videoJson.get("id") + "_fov")){
                    videoJson.put("blend_fov", videoMap.get(videoJson.get("id") + "_fov"));
                }else {
                    videoJson.put("blend_fov", 7);
                }
                jsonArray.add(videoJson);
            }
        }

        videosJson.put("data", jsonArray);
        if(Objects.nonNull(videoVersion) && videoVersion >= 4){
            videosJson.put("version", 3);
            videosJson.put("upPath", fYunFileConfig.getHost() + String.format(UploadFilePath.DATA_VIEW_PATH, projectNum) + "Up.xml");
            if(cameraType == 13){
                //转台相机
                videosJson.put("upPath", videosJson.getString("upPath").replace(".xml", ".txt"));
            }
        }else {
            videosJson.put("version", 1);
            videosJson.put("upPath", fYunFileConfig.getHost() + String.format(UploadFilePath.DATA_VIEW_PATH, projectNum) + "Up2.xml");
            if(cameraType == 13){
                //转台相机
                videosJson.put("upPath", videosJson.getString("upPath").replace(".xml", ".txt"));
            }
        }

        if(cameraType == 5 || cameraType == 6){
            videosJson.put("version", 1);
            videosJson.put("upPath", fYunFileConfig.getHost() + String.format(UploadFilePath.DATA_VIEW_PATH, projectNum) + "stitch_params.txt");
        }

        return videosJson;
    }

    private Long calUseSpace(Map<String, String> uploadFile) {
        return uploadFile.keySet().stream().map(File::new).filter(File::exists).mapToLong(File::length).sum();
    }

    private void sealScene(boolean arrearCap, Long scenePlusId, Long sceneProId){
        if(!arrearCap){
            return;
        }
        scenePlusService.update(
            new LambdaUpdateWrapper<ScenePlus>()
                .set(ScenePlus::getPayStatus, PayStatus.NO_CAPACITY.code())
                .eq(ScenePlus::getId, scenePlusId));
    }

    /**
     * <p>
     双目场景更新数据库
     * </p>
     * @author dengsixing
     * @date 2022/3/21
     * @param num
     * @param space
     **/
    private void updateDb4Sm(String num, long space){
        List<ScenePlus> ScenePlusList = scenePlusService.list(
            new LambdaQueryWrapper<ScenePlus>().select(ScenePlus::getId).eq(ScenePlus::getNum, num));

        if(CollUtil.isEmpty(ScenePlusList)){
            return ;
        }

        List<Long> sceneIds = ScenePlusList.stream().map(ScenePlus::getId).collect(Collectors.toList());

        //更新场景创建时间
        scenePlusService.update(new LambdaUpdateWrapper<ScenePlus>().in(ScenePlus::getId, sceneIds)
            .set(ScenePlus::getSceneStatus, SceneStatus.NO_DISPLAY.code()));

        //更新使用容量
        scenePlusExtService.update(new LambdaUpdateWrapper<ScenePlusExt>().in(ScenePlusExt::getPlusId, sceneIds)
                .set(ScenePlusExt::getSpace, space)
                .set(ScenePlusExt::getAlgorithmTime, Calendar.getInstance().getTime()));
    }

    private ScenePlus updateDbPlus(String num, Long space, String videosJson, Long computeTime, String fileId,boolean isObj){
        ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
        ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
        List<ScenePlus> list = scenePlusService.list(new LambdaQueryWrapper<ScenePlus>()
                .select(ScenePlus::getId).eq(ScenePlus::getNum, num));
        if (CollUtil.isEmpty(list)) {
            return null;
        }

        //修改场景状态 空间 支付状态 计算时间
        List<Long> scenePlusIds = list.stream().map(ScenePlus::getId).collect(Collectors.toList());

        scenePlusService.update(new LambdaUpdateWrapper<ScenePlus>()
            .in(ScenePlus::getId, scenePlusIds)
            .set(ScenePlus::getSceneStatus, SceneStatus.NO_DISPLAY.code())
            .set(ScenePlus::getUpdateTime, new Date())
            .set(ScenePlus::getPayStatus, PayStatus.PAY.code()));

        SceneFileBuild sceneFileBuild = sceneFileBuildService.findByFileId(fileId);
        if(sceneFileBuild != null){
            sceneFileBuild.setBuildStatus(BuildStatus.success.code());
            sceneFileBuildService.updateById(sceneFileBuild);
        }

        scenePlusExt.setSpace(space);
        scenePlusExt.setComputeTime(computeTime.toString());
        scenePlusExt.setAlgorithmTime(new Date());
        scenePlusExt.setVideos(videosJson);
        scenePlusExt.setIsObj(isObj ? 1 : 0);

        SceneSource sceneSource = SceneSource.get(scenePlus.getSceneSource());
        switch (sceneSource){
            case BM:
                scenePlusExt.setSceneResolution(SceneResolution.TILES_2K.code());
                scenePlusExt.setSceneFrom(SceneFrom.PRO.code());
                break;
            case SM:
                scenePlusExt.setSceneResolution(SceneResolution.TILES_1K.code());
                scenePlusExt.setSceneFrom(SceneFrom.LITE.code());
                break;
            case ZT:
                scenePlusExt.setSceneResolution(SceneResolution.TILES_4K.code());
                scenePlusExt.setSceneFrom(SceneFrom.MINION.code());
                break;
            case JG:
                scenePlusExt.setSceneResolution(SceneResolution.TILES_4K.code());
                scenePlusExt.setSceneFrom(SceneFrom.LASER.code());
                break;
        }
        scenePlusExtService.updateById(scenePlusExt);


        return scenePlus;
    }

    private boolean updateSpace(Long cameraId, Long space) throws Exception{
        //更新相机使用用量
        CameraDetail cameraDetail = cameraDetailService.updateCameraDetailByCameraIdAndSpace(cameraId, space);
        UserIncrement userIncrement = userIncrementService.getByCameraId(cameraId);
        if( userIncrement == null || userIncrement.getIsExpired() == ExpiredStatus.Expired.code()) {
            // 新上传的场景，如果总容量小于使用容量，则该大场景保留在临时存储空间30天
            if (cameraDetail.getTotalSpace().compareTo(cameraDetail.getUsedSpace()) == -1){
                return true;
            }
        }
        return false;
    }

    private Object[] updateEditInfo(ScenePlus scenePlus){
        SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());
        SceneEditControls sceneEditControls = null;
        SceneEditInfoExt sceneEditInfoExt = null;
        if(sceneEditInfo == null){
            sceneEditInfo = new SceneEditInfo();
            sceneEditInfo.setScenePlusId(scenePlus.getId());
            sceneEditInfo.setDescription(scenePlus.getDescription());
            sceneEditInfo.setTitle(scenePlus.getTitle());
            sceneEditInfoService.save(sceneEditInfo);

        }else{
            sceneEditControls = sceneEditControlsService.getBySceneEditId(sceneEditInfo.getId());
            sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfo.getId());
            sceneEditInfo.setVersion(sceneEditInfo.getVersion() + 1);
            sceneEditInfoService.updateById(sceneEditInfo);
        }
        if(sceneEditControls == null){
            sceneEditControls = new SceneEditControls();
            sceneEditControls.setEditInfoId(sceneEditInfo.getId());
            sceneEditControlsService.save(sceneEditControls);
        }
        if(sceneEditInfoExt == null){
            sceneEditInfoExt = new SceneEditInfoExt();
            sceneEditInfoExt.setScenePlusId(scenePlus.getId());
            sceneEditInfoExt.setEditInfoId(sceneEditInfo.getId());
            sceneEditInfoExtService.save(sceneEditInfoExt);
        }
        return new Object[]{sceneEditInfo, sceneEditInfoExt, sceneEditControls};
    }

    public void uploadHouseTypeJson(String num, String floorPlanCardFilePath) {
        if (!new File(floorPlanCardFilePath).exists()) {
            log.error("floorplan_cad.json 文件不存在，文件路径：{}", floorPlanCardFilePath);
            return;
        }
        JSONObject json = CreateHouseJsonUtil.createHouseTypeJsonByCad(floorPlanCardFilePath);
        String hourseTypeJsonPath = String.format(UploadFilePath.DATA_VIEW_PATH, num) + "houseType.json";
        fYunFileService.uploadFile(json.toJSONString().getBytes(), hourseTypeJsonPath);
        hourseTypeJsonPath = String.format(UploadFilePath.DATA_EDIT_PATH, num) + "houseType.json";
        fYunFileService.uploadFile(json.toJSONString().getBytes(), hourseTypeJsonPath);
    }
}