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

package com.fdkankan.contro.service.impl;

import cn.hutool.core.io.FileUtil;
import com.alibaba.fastjson.JSONObject;
import com.fdkankan.common.constant.SceneStatus;
import com.fdkankan.common.util.DateUtil;
import com.fdkankan.common.util.FileUtils;
import com.fdkankan.contro.entity.*;
import com.fdkankan.contro.service.IFdkkLaserService;
import com.fdkankan.contro.service.IScenePlusService;
import com.fdkankan.contro.service.ISceneProService;
import com.fdkankan.contro.service.IUserService;
import com.fdkankan.fyun.face.FYunFileServiceInterface;
import com.fdkankan.rabbitmq.util.RabbitMqProducer;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.ObjectUtils;

import java.io.File;
import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

@Service
@Slf4j
public class IFdkkLaserServiceImpl implements IFdkkLaserService {

    @Autowired
    private ISceneProService sceneProService;

    @Autowired
    private IScenePlusService scenePlusService;

    @Autowired
    private RabbitMqProducer rabbitMqProducer;

    @Value("${queue.application.laser.update-scene}")
    private String updateScene;

    @Value("${queue.application.laser.update-build-status}")
    private String updateBuildStatus;

    @Value("${queue.application.laser.cloud-point-build}")
    private String cloudPointBuild;

    @Value("${4dkk.laserService.bucket}")
    private String bucket;

    @Value("${4dkk.laserService.cloud-point-fyun-path}")
    private String cloudPointFyunPath;

    @Autowired
    private FYunFileServiceInterface fYunFileService;
    @Autowired
    private IUserService userService;


    public void updateSceneStatus(String sceneCode, int sceneStatus, String path, Date createTime, Date algorithmTime, Integer shootCount, Integer payStatus) {
        Map<String, Object> params = new HashMap<>();
        params.put("sceneCode", sceneCode);
        params.put("status", sceneStatus);
        params.put("version",getSceneVersion(sceneCode));
        params.put("createTime", DateUtil.date2String(createTime, null));
        params.put("algorithmTime", DateUtil.date2String(algorithmTime, null));
        if (!ObjectUtils.isEmpty(path)) {
            params.put("path", path);
        }
        params.put("shootCount", shootCount);
        params.put("payStatus", payStatus);
        rabbitMqProducer.sendByWorkQueue(updateScene, params);
    }

    public void syncBuildResult(String sceneNum, String dataSource,Date createTime, Date algorithmTime, Integer shootCount, Integer payStatus) {
        log.info("激光转台相机构建结果 同步 请求 ");
        try {
            String jgPath = dataSource;
            //创建目录
            if (dataSource.lastIndexOf("/") != -1) {
                jgPath = jgPath + "_laserData";
            } else {
                jgPath = jgPath.substring(0, jgPath.length() - 1) + "_laserData";
            }

            FileUtils.createDir(jgPath + "/extras");

            log.info("生成 激光相机目录 " + jgPath);
            //生成data.json
            JSONObject jgDataJson = new JSONObject();
            jgDataJson.put("split_type", "SPLIT_V15");
            jgDataJson.put("skybox_type", "SKYBOX_V5");
            jgDataJson.put("extras", null);
            FileUtils.writeFile(jgPath + File.separator + "data.json", jgDataJson.toString());

            // result/mesh/mesh.obj  --> jgPath/laserData/laser.obj
            log.info("开始拷贝obj文件");
            FileUtils.copyDirectiory(dataSource + "/results/mesh", jgPath + "/laserData/mesh");

            FileUtils.copyDirectiory(dataSource + "/results/laserData/cover", jgPath + "/extras");
            //删除旧的深度图
            if(FileUtil.exist( jgPath+"/laserData/depthmap/")){
                FileUtil.del(jgPath+"/laserData/depthmap/");
            }
            FileUtils.copyFile(dataSource + "/results/laserData", jgPath, true);

            updateSceneStatus(sceneNum, 2, jgPath + File.separator + "laserData",createTime, algorithmTime, shootCount, payStatus);
        } catch (Exception e) {
            log.error("激光转台相机同步失败", e);
        }
    }

    @Override
    public void pushBuildStatusToLaserSystem(String projectNum, String laserObjFilePath, Integer buildObjStatus) {
        Map<String, Object> params = new HashMap<>();
        params.put("sceneCode", projectNum);
        params.put("objPath", laserObjFilePath);
        params.put("buildObjStatus", buildObjStatus);
        rabbitMqProducer.sendByWorkQueue(updateBuildStatus, params);
    }


    public void saveScene(ScenePlus scenePlus, JSONObject fdageJson, Camera cameraEntity, String phone, boolean rebuild, Integer payStatus) {
        Map<String, Object> params = new HashMap<>();
        params.put("childName", cameraEntity.getChildName());
        params.put("createTime", DateUtil.date2String(scenePlus.getCreateTime(), null));
        params.put("phone", phone);
        params.put("sceneCode", scenePlus.getNum());
        params.put("snCode", cameraEntity.getSnCode());
        params.put("status", scenePlus.getSceneStatus());
        if (!rebuild) {
            if(Objects.nonNull(fdageJson)){
                params.put("password", fdageJson.getString("pwd"));
            }
        } else {
            params.put("status", 4);
        }
        params.put("version",getSceneVersion(scenePlus.getNum()));
        params.put("title", scenePlus.getTitle());
        params.put("userId", scenePlus.getUserId());
        params.put("sceneSource", scenePlus.getSceneSource());
        if(Objects.nonNull(fdageJson)){
            params.put("location", fdageJson.getInteger("location"));
        }
        params.put("payStatus", payStatus);
        rabbitMqProducer.sendByWorkQueue(updateScene, params);
    }

    public void saveScene(ScenePro scenePro, String scenePassword, Camera cameraEntity, String phone, boolean rebuild, Integer payStatus) {
        Map<String, Object> params = new HashMap<>();
        params.put("childName", cameraEntity.getChildName());
        if(ObjectUtils.isEmpty(scenePro.getCreateTime())){
            scenePro.setCreateTime(new Date());
        }
        params.put("createTime", DateUtil.date2String(scenePro.getCreateTime(), null));
        params.put("phone", phone);
        params.put("sceneCode", scenePro.getNum());
        params.put("snCode", cameraEntity.getSnCode());
        params.put("status", scenePro.getStatus());
        if (!rebuild) {
            params.put("password", scenePassword);
        } else {
            params.put("status", 4);
        }
        params.put("version",getSceneVersion(scenePro.getNum()));
        params.put("title", scenePro.getSceneName());
        params.put("userId", scenePro.getUserId());
        params.put("payStatus", payStatus);
        rabbitMqProducer.sendByWorkQueue(updateScene, params);
    }

    private String getSceneVersion(String num) {
        String version = "V3";
        ScenePro scenepro = sceneProService.getByNum(num);
        if (scenepro == null || scenepro.getIsUpgrade() == 1) {
            ScenePlus plus = scenePlusService.getScenePlusByNum(num);
            if (plus != null) {
                version = "V4";
            }
        }
        return version;
    }

    @Override
    public void cloudPointBuild(String sceneCode, String path) {
        log.info("开始同步点云编辑文件");
        // 删除旧的文件及目录
        try {
            fYunFileService.deleteFile(cloudPointFyunPath + "vision_edit.txt");
            fYunFileService.deleteFolder(cloudPointFyunPath + "uuidcloud");
        } catch (IOException e) {
            log.error("删除文件失败", e);
            e.printStackTrace();
        }

        // 上传点云编辑文件，并通知激光系统
        fYunFileService.uploadFile(bucket,path + "/results/laserData/vision_edit.txt", String.format(cloudPointFyunPath,sceneCode,sceneCode) + "vision_edit.txt");
        fYunFileService.uploadFileByCommand(bucket,path + "/results/laserData/uuidcloud", String.format(cloudPointFyunPath,sceneCode,sceneCode) + "uuidcloud");

        Map<String, Object> params = new HashMap<>();
        params.put("sceneNum", sceneCode);
        params.put("businessType", 0);
        rabbitMqProducer.sendByWorkQueue(cloudPointBuild, params);
    }

    public void sendE57ToLaser(ScenePlus scenePlus, ScenePlusExt scenePlusExt, String path) {
        User user = userService.getById(scenePlus.getUserId());
        Map<String, Object> params = new HashMap<>();
        params.put("createTime", DateUtil.date2String(scenePlus.getCreateTime(), null));
        params.put("phone", user.getUserName());
        params.put("sceneCode", scenePlus.getNum());
        params.put("status", -scenePlus.getSceneStatus());
        params.put("version",getSceneVersion(scenePlus.getNum()));
        params.put("title", scenePlus.getTitle());
        params.put("userId", scenePlus.getUserId());
        params.put("sceneSource", scenePlus.getSceneSource());
        params.put("location", scenePlusExt.getLocation());
        params.put("payStatus", scenePlus.getPayStatus());
        params.put("shootCount", scenePlusExt.getShootCount());
        params.put("path", path);
        if(scenePlus.getSceneStatus() == SceneStatus.wait.code()){
            rabbitMqProducer.sendByWorkQueue("laser-e57-init-scene", params);
        }
        if(scenePlus.getSceneStatus() == SceneStatus.NO_DISPLAY.code()){
            rabbitMqProducer.sendByWorkQueue("laser-e57-update-scene", params);
        }
    }

    public String syncBuildResult4E57(String dataSource) {
        log.info("激光转台相机构建结果 同步 请求 ");
        try {
            String jgPath = dataSource;
            //创建目录
            if (dataSource.lastIndexOf("/") != -1) {
                jgPath = jgPath + "_laserData";
            } else {
                jgPath = jgPath.substring(0, jgPath.length() - 1) + "_laserData";
            }

            FileUtils.createDir(jgPath + "/extras");

            log.info("生成 激光相机目录 " + jgPath);
            //生成data.json
            JSONObject jgDataJson = new JSONObject();
            jgDataJson.put("split_type", "SPLIT_V15");
            jgDataJson.put("skybox_type", "SKYBOX_V5");
            jgDataJson.put("extras", null);
            FileUtils.writeFile(jgPath + File.separator + "data.json", jgDataJson.toString());

            // result/mesh/mesh.obj  --> jgPath/laserData/laser.obj
            log.info("开始拷贝obj文件");
            FileUtils.copyDirectiory(dataSource + "/results/mesh", jgPath + "/laserData/mesh");

            FileUtils.copyDirectiory(dataSource + "/results/laserData/cover", jgPath + "/extras");
            //删除旧的深度图
            if(FileUtil.exist( jgPath+"/laserData/depthmap/")){
                FileUtil.del(jgPath+"/laserData/depthmap/");
            }
            FileUtils.copyFile(dataSource + "/results/laserData", jgPath, true);

            return jgPath + "/laserData";

        } catch (Exception e) {
            log.error("激光转台相机同步失败", e);
        }
    }
}