|
@@ -1,635 +0,0 @@
|
|
-package com.fdkankan.contro.mq.service.impl;
|
|
|
|
-
|
|
|
|
-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.ExpiredStatus;
|
|
|
|
-import com.fdkankan.common.constant.PayStatus;
|
|
|
|
-import com.fdkankan.common.constant.SceneStatus;
|
|
|
|
-import com.fdkankan.common.util.FileUtils;
|
|
|
|
-import com.fdkankan.contro.entity.CameraDetail;
|
|
|
|
-import com.fdkankan.contro.entity.IncrementType;
|
|
|
|
-import com.fdkankan.contro.entity.ScenePro;
|
|
|
|
-import com.fdkankan.contro.entity.UserIncrement;
|
|
|
|
-import com.fdkankan.contro.mq.service.IBuildSceneService;
|
|
|
|
-import com.fdkankan.contro.service.*;
|
|
|
|
-import com.fdkankan.contro.service.impl.FdkkV4Service;
|
|
|
|
-import com.fdkankan.fyun.config.FYunFileConfig;
|
|
|
|
-import com.fdkankan.fyun.constant.FYunTypeEnum;
|
|
|
|
-import com.fdkankan.fyun.face.FYunFileServiceInterface;
|
|
|
|
-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 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 BuildV3SceneServiceImpl implements IBuildSceneService {
|
|
|
|
-
|
|
|
|
- @Value("${queue.modeling.modeling-call}")
|
|
|
|
- private String queueModelingCall;
|
|
|
|
-
|
|
|
|
- @Value("${queue.modeling.v3.modeling-post}")
|
|
|
|
- private String queueV3ModelingPost;
|
|
|
|
-
|
|
|
|
- @Value("${model.type:#{null}}")
|
|
|
|
- private String modelSceneKind;
|
|
|
|
-
|
|
|
|
- @Autowired
|
|
|
|
- private RabbitMqProducer mqProducer;
|
|
|
|
-
|
|
|
|
- @Autowired
|
|
|
|
- private FYunFileServiceInterface fYunFileService;
|
|
|
|
- @Autowired
|
|
|
|
- private ICameraDetailService cameraDetailService;
|
|
|
|
-
|
|
|
|
- @Autowired
|
|
|
|
- private FYunFileConfig fYunFileConfig;
|
|
|
|
-
|
|
|
|
- @Autowired
|
|
|
|
- private ISceneProService sceneProService;
|
|
|
|
-
|
|
|
|
- @Autowired
|
|
|
|
- private IUserIncrementService userIncrementService;
|
|
|
|
-
|
|
|
|
- @Autowired
|
|
|
|
- private IFdkkLaserService fdkkLaserService;
|
|
|
|
-
|
|
|
|
- @Autowired
|
|
|
|
- private FdkkV4Service fdkkV4Service;
|
|
|
|
-
|
|
|
|
- @Autowired
|
|
|
|
- private IIncrementTypeService incrementTypeService;
|
|
|
|
-
|
|
|
|
- @Autowired
|
|
|
|
- private IBuildSceneDTService buildSceneDTService;
|
|
|
|
-
|
|
|
|
- @Override
|
|
|
|
- public void buildScenePre(BuildSceneCallMessage message) {
|
|
|
|
- boolean success = false;
|
|
|
|
- try {
|
|
|
|
- //根据相机类型,组装资源路径
|
|
|
|
- //下载资源到本地
|
|
|
|
- this.downLoadSource(message, message.getPath());
|
|
|
|
-
|
|
|
|
- if (!ObjectUtils.isEmpty(modelSceneKind)) {
|
|
|
|
- // 修改dataFdage文件
|
|
|
|
- JSONObject fdageJson = JSONObject.parseObject(FileUtils.readFile(message.getPath().concat("/capture/data.fdage")));
|
|
|
|
- fdageJson.put("modelType", modelSceneKind);
|
|
|
|
- FileUtils.writeFile(message.getPath().concat("/capture/data.fdage"), fdageJson.toJSONString());
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- message.getBuildContext().put("cameraType",message.getCameraType());
|
|
|
|
-
|
|
|
|
- message.setResultReceiverMqName(queueV3ModelingPost);
|
|
|
|
-
|
|
|
|
- //发送mq,就进行计算
|
|
|
|
- mqProducer.sendByWorkQueue(queueModelingCall, message);
|
|
|
|
-
|
|
|
|
- success = true;
|
|
|
|
-
|
|
|
|
- log.info("场景计算资源准备结束,场景码:{}", message.getSceneNum());
|
|
|
|
-
|
|
|
|
- }catch (Exception e){
|
|
|
|
- log.error("场景计算前置处理出错", e);
|
|
|
|
- }finally {
|
|
|
|
- //如果前置处理失败,发送钉钉消息
|
|
|
|
- if(!success){
|
|
|
|
- buildSceneDTService.handBaseFail("场景计算资源准备异常!", message.getPath(), message.getSceneNum(), "计算控制服务器");
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- @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.getBuildContext().get("cameraType").toString());
|
|
|
|
- String sceneCode = message.getBuildContext().get("sceneNum").toString();
|
|
|
|
- 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");
|
|
|
|
- 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("建模失败,修改状态为失败状态");
|
|
|
|
- sceneProService.update(new LambdaUpdateWrapper<ScenePro>()
|
|
|
|
- .set(ScenePro::getStatus, SceneStatus.FAILD.code())
|
|
|
|
- .eq(ScenePro::getNum, sceneCode));
|
|
|
|
- // 发送钉钉消息,计算失败
|
|
|
|
- buildSceneDTService.handModelFail("计算失败", message.getPath(), sceneCode, message.getHostName());
|
|
|
|
- return;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- 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);
|
|
|
|
-
|
|
|
|
- //写入数据库
|
|
|
|
- ScenePro scenePro = this.updateDbPlus(sceneCode, space, videosJson.toJSONString(), computeTime,fdageData.containsKey("exportMeshObj") && fdageData.getIntValue("exportMeshObj") == 1);
|
|
|
|
-
|
|
|
|
- //变更容量,arrearCap为true时,代表容量不足
|
|
|
|
- boolean arrearCap = this.updateSpace(scenePro.getCameraId(), space);
|
|
|
|
-
|
|
|
|
- //如果相机容量不足,需要把场景的paystatus改为容量不足状态
|
|
|
|
- if (arrearCap && cameraType != 14) {
|
|
|
|
- scenePro.setPayStatus(PayStatus.NO_CAPACITY.code());
|
|
|
|
- this.sealScene(arrearCap, scenePro.getId(), scenePro.getId());
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- //上传文件
|
|
|
|
- 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);
|
|
|
|
- });
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- //生成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,scenePro, arrearCap);
|
|
|
|
-
|
|
|
|
- //计算成功,通知APP
|
|
|
|
- Integer pushChannel = fdageData.getInteger("pushChannel");
|
|
|
|
- String pushToken = fdageData.getString("pushToken");
|
|
|
|
- this.pushMsgToApp(pushChannel,pushToken, cameraType, scenePro.getSceneName(), scenePro.getWebSite());
|
|
|
|
-
|
|
|
|
- if (cameraType == 14) {
|
|
|
|
- //计算成功 激光转台相机 同步 请求
|
|
|
|
- fdkkLaserService.syncBuildResult(scenePro.getNum(), scenePro.getDataSource(),scenePro.getCreateTime());
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- CreateObjUtil.deleteFile(path.replace(ConstantFilePath.BUILD_MODEL_PATH, "/") + "/capture");
|
|
|
|
-
|
|
|
|
- log.info("场景计算结果处理结束,场景码:{}", sceneCode);
|
|
|
|
-
|
|
|
|
- // 判断是否升级V4
|
|
|
|
- if(fdageData.containsKey("updateV4") && fdageData.getIntValue("updateV4") == 1){
|
|
|
|
- fdkkV4Service.upgradeToV4(sceneCode);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- }catch (Exception e){
|
|
|
|
- e.printStackTrace();
|
|
|
|
- buildSceneDTService.handBaseFail("场景计算结果处理出错!", message.getPath(), sceneCode, "计算控制服务器");
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- private Map<String, String> getUploadFiles(String path,Integer cameraType,JSONObject fdageData) throws Exception {
|
|
|
|
- ScenePro scenePro = sceneProService.getOne(
|
|
|
|
- new LambdaQueryWrapper<ScenePro>().like(ScenePro::getDataSource, path));
|
|
|
|
- if (ObjectUtils.isEmpty(scenePro)) {
|
|
|
|
- throw new Exception("未找到场景信息:" + path);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- String projectNum = scenePro.getNum();
|
|
|
|
-
|
|
|
|
-
|
|
|
|
- String dataViewPath = String.format(ConstantFilePath.DATA_PATH_FORMAT, projectNum);
|
|
|
|
- String imagesPath = String.format(ConstantFilePath.IMAGE_PATH_FORMAT, projectNum);
|
|
|
|
- String videoPath = String.format(ConstantFilePath.VIDEO_PATH_FORMAT, 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", imagesPath + ConstantFileName.modelUUID + "_50k.dam.lzma");
|
|
|
|
- map.put(path + File.separator + "results" + File.separator + ConstantFileName.modelUUID + "_50k.dam", imagesPath + 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);
|
|
|
|
-
|
|
|
|
- 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 writeSceneJson(String num, JSONObject videosJson, ScenePro scenePro,boolean arrearCap) throws Exception{
|
|
|
|
- String dataViewPath = String.format(ConstantFilePath.DATA_PATH_FORMAT, num);
|
|
|
|
-
|
|
|
|
- String sceneJsonKey = dataViewPath + "scene.json";
|
|
|
|
- boolean exist = fYunFileService.fileExist(sceneJsonKey);
|
|
|
|
- //如果云端没有scene.json文件,生成一份
|
|
|
|
- if(!exist){
|
|
|
|
- JSONObject sceneJson = JSONObject.parseObject(JSONObject.toJSONString(scenePro));
|
|
|
|
- sceneJson.put("videos",videosJson);
|
|
|
|
- if(arrearCap) {
|
|
|
|
- sceneJson.put("payStatus",PayStatus.NO_CAPACITY.code());
|
|
|
|
- }
|
|
|
|
- String sceneJsonStr = JSON.toJSONString(sceneJson);
|
|
|
|
- //上传sceneJson文件
|
|
|
|
- fYunFileService.uploadFile(sceneJsonStr.getBytes(), sceneJsonKey);
|
|
|
|
- }
|
|
|
|
- // 上传status JSON.
|
|
|
|
-
|
|
|
|
- JSONObject statusJson = new JSONObject();
|
|
|
|
- //临时将-2改成1,app还没完全更新
|
|
|
|
- statusJson.put("status", 1);
|
|
|
|
- statusJson.put("webSite", scenePro.getWebSite());
|
|
|
|
- statusJson.put("sceneNum", num);
|
|
|
|
- statusJson.put("thumb", scenePro.getThumb());
|
|
|
|
- statusJson.put("payStatus", scenePro.getPayStatus());
|
|
|
|
- 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(ConstantFilePath.DATA_PATH_FORMAT, 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(ConstantFilePath.DATA_PATH_FORMAT, 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(ConstantFilePath.DATA_PATH_FORMAT, 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){
|
|
|
|
- sceneProService.update(
|
|
|
|
- new LambdaUpdateWrapper<ScenePro>()
|
|
|
|
- .set(ScenePro::getPayStatus, PayStatus.NO_CAPACITY.code())
|
|
|
|
- .eq(ScenePro::getId, scenePlusId));
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- /**
|
|
|
|
- * <p>
|
|
|
|
- 双目场景更新数据库
|
|
|
|
- * </p>
|
|
|
|
- * @author dengsixing
|
|
|
|
- * @date 2022/3/21
|
|
|
|
- * @param num
|
|
|
|
- * @param space
|
|
|
|
- **/
|
|
|
|
- private void updateDb4Sm(String num, long space){
|
|
|
|
- List<ScenePro> ScenePlusList = sceneProService.list(
|
|
|
|
- new LambdaQueryWrapper<ScenePro>().select(ScenePro::getId).eq(ScenePro::getNum, num));
|
|
|
|
-
|
|
|
|
- if(CollUtil.isEmpty(ScenePlusList)){
|
|
|
|
- return ;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- List<Long> sceneIds = ScenePlusList.stream().map(ScenePro::getId).collect(Collectors.toList());
|
|
|
|
-
|
|
|
|
- //更新场景创建时间
|
|
|
|
- sceneProService.update(new LambdaUpdateWrapper<ScenePro>().in(ScenePro::getId, sceneIds)
|
|
|
|
- .set(ScenePro::getStatus, SceneStatus.NO_DISPLAY.code())
|
|
|
|
- .set(ScenePro::getSpace,space)
|
|
|
|
- .set(ScenePro::getCreateTime,new Date())
|
|
|
|
- );
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- private ScenePro updateDbPlus(String num, Long space, String videosJson, Long computeTime,boolean isObj){
|
|
|
|
- ScenePro scenePro = sceneProService.getByNum(num);
|
|
|
|
- if (ObjectUtils.isEmpty(scenePro)) {
|
|
|
|
- return null;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- sceneProService.update(new LambdaUpdateWrapper<ScenePro>()
|
|
|
|
- .in(ScenePro::getId, scenePro.getId())
|
|
|
|
- .set(ScenePro::getStatus, SceneStatus.NO_DISPLAY.code())
|
|
|
|
- .set(ScenePro::getUpdateTime, new Date())
|
|
|
|
- .set(ScenePro::getCreateTime, new Date())
|
|
|
|
- .set(ScenePro::getSpace, space)
|
|
|
|
- .set(ScenePro::getComputeTime, computeTime)
|
|
|
|
- .set(ScenePro::getVideos, videosJson)
|
|
|
|
- .set(ScenePro::getIsObj, isObj ? 1 : 0)
|
|
|
|
- .set(ScenePro::getPayStatus, PayStatus.PAY.code()));
|
|
|
|
-
|
|
|
|
- return scenePro;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- private boolean updateSpace(Long cameraId, Long space) throws Exception{
|
|
|
|
- //更新相机使用用量
|
|
|
|
- CameraDetail cameraDetail = cameraDetailService.updateCameraDetailByCameraIdAndSpace(cameraId, space);
|
|
|
|
- UserIncrement userIncrement = userIncrementService.getByCameraId(cameraId);
|
|
|
|
-
|
|
|
|
- if(!ObjectUtils.isEmpty(userIncrement)){
|
|
|
|
- IncrementType type = incrementTypeService.getById(userIncrement.getIncrementTypeId());
|
|
|
|
- if(type.getType().equals("cn") && userIncrement.getIsExpired().equals(ExpiredStatus.NOT_EXPIRED.code())){
|
|
|
|
- return false;
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- return cameraDetail.getTotalSpace().compareTo(cameraDetail.getUsedSpace()) == -1;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- 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(ConstantFilePath.DATA_PATH_FORMAT, 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);
|
|
|
|
- }
|
|
|
|
-}
|
|
|