BuildV3SceneServiceImpl.java 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601
  1. package com.fdkankan.contro.mq.service.impl;
  2. import cn.hutool.core.collection.CollUtil;
  3. import cn.hutool.core.io.FileUtil;
  4. import cn.hutool.core.io.file.FileNameUtil;
  5. import cn.hutool.core.util.StrUtil;
  6. import cn.hutool.core.util.ZipUtil;
  7. import com.alibaba.fastjson.JSON;
  8. import com.alibaba.fastjson.JSONArray;
  9. import com.alibaba.fastjson.JSONObject;
  10. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  11. import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
  12. import com.fdkankan.common.constant.ExpiredStatus;
  13. import com.fdkankan.common.constant.PayStatus;
  14. import com.fdkankan.common.constant.SceneStatus;
  15. import com.fdkankan.common.util.FileUtils;
  16. import com.fdkankan.contro.entity.*;
  17. import com.fdkankan.contro.mq.service.IBuildSceneService;
  18. import com.fdkankan.contro.service.*;
  19. import com.fdkankan.contro.service.impl.FdkkV4Service;
  20. import com.fdkankan.fyun.config.FYunFileConfig;
  21. import com.fdkankan.fyun.constant.FYunTypeEnum;
  22. import com.fdkankan.fyun.face.FYunFileServiceInterface;
  23. import com.fdkankan.model.constants.ConstantFileName;
  24. import com.fdkankan.model.constants.ConstantFilePath;
  25. import com.fdkankan.model.constants.UploadFilePath;
  26. import com.fdkankan.model.utils.CreateHouseJsonUtil;
  27. import com.fdkankan.model.utils.CreateObjUtil;
  28. import com.fdkankan.push.config.PushMessageConfig;
  29. import com.fdkankan.push.utils.PushMsgUtil;
  30. import com.fdkankan.rabbitmq.bean.BuildSceneCallMessage;
  31. import com.fdkankan.rabbitmq.bean.BuildSceneResultMqMessage;
  32. import com.fdkankan.rabbitmq.util.RabbitMqProducer;
  33. import lombok.extern.slf4j.Slf4j;
  34. import org.apache.commons.lang3.ObjectUtils;
  35. import org.apache.commons.lang3.StringUtils;
  36. import org.springframework.beans.factory.annotation.Autowired;
  37. import org.springframework.beans.factory.annotation.Value;
  38. import org.springframework.stereotype.Service;
  39. import java.io.File;
  40. import java.io.IOException;
  41. import java.util.*;
  42. import java.util.Map.Entry;
  43. import java.util.stream.Collectors;
  44. /**
  45. * <p>
  46. * TODO
  47. * </p>
  48. *
  49. * @author dengsixing
  50. * @since 2022/4/20
  51. **/
  52. @Slf4j
  53. @Service
  54. public class BuildV3SceneServiceImpl implements IBuildSceneService {
  55. @Value("${queue.modeling.modeling-call}")
  56. private String queueModelingCall;
  57. @Value("${queue.modeling.v3.modeling-post}")
  58. private String queueV3ModelingPost;
  59. @Value("${model.type:#{null}}")
  60. private String modelSceneKind;
  61. @Autowired
  62. private RabbitMqProducer mqProducer;
  63. @Autowired
  64. private FYunFileServiceInterface fYunFileService;
  65. @Autowired
  66. private ICameraDetailService cameraDetailService;
  67. @Autowired
  68. private FYunFileConfig fYunFileConfig;
  69. @Autowired
  70. private ISceneProService sceneProService;
  71. @Autowired
  72. private IUserIncrementService userIncrementService;
  73. @Autowired
  74. private IFdkkLaserService fdkkLaserService;
  75. @Autowired
  76. private FdkkV4Service fdkkV4Service;
  77. @Autowired
  78. private IIncrementTypeService incrementTypeService;
  79. @Autowired
  80. private IBuildSceneDTService buildSceneDTService;
  81. @Autowired
  82. private ISceneProEditService sceneProEditService;
  83. @Override
  84. public void buildScenePre(BuildSceneCallMessage message) {
  85. boolean success = false;
  86. try {
  87. //根据相机类型,组装资源路径
  88. //下载资源到本地
  89. this.downLoadSource(message, message.getPath());
  90. if (!ObjectUtils.isEmpty(modelSceneKind)) {
  91. // 修改dataFdage文件
  92. JSONObject fdageJson = JSONObject.parseObject(FileUtils.readFile(message.getPath().concat("/capture/data.fdage")));
  93. fdageJson.put("modelType", modelSceneKind);
  94. FileUtils.writeFile(message.getPath().concat("/capture/data.fdage"), fdageJson.toJSONString());
  95. }
  96. message.getBuildContext().put("cameraType",message.getCameraType());
  97. message.setResultReceiverMqName(queueV3ModelingPost);
  98. //发送mq,就进行计算
  99. mqProducer.sendByWorkQueue(queueModelingCall, message);
  100. success = true;
  101. log.info("场景计算资源准备结束,场景码:{}", message.getSceneNum());
  102. }catch (Exception e){
  103. log.error("场景计算前置处理出错", e);
  104. }finally {
  105. //如果前置处理失败,发送钉钉消息
  106. if(!success){
  107. buildSceneDTService.handBaseFail("场景计算资源准备异常!", message.getPath(), message.getSceneNum(), "计算控制服务器");
  108. }
  109. }
  110. }
  111. @Override
  112. public void downLoadSource(BuildSceneCallMessage buildSceneMqMessage,String path){
  113. String ossPath = ConstantFilePath.OSS_PREFIX
  114. + path.replace(ConstantFilePath.BUILD_MODEL_PATH, "")
  115. .replace(ConstantFilePath.BUILD_MODEL_LASER_PATH, "");
  116. if (!ossPath.endsWith("/")) {
  117. ossPath = ossPath.concat("/");
  118. }
  119. fYunFileService.downloadFileByCommand(path + File.separator + "capture", ossPath);
  120. }
  121. @Override
  122. public void buildScenePost(BuildSceneResultMqMessage message) throws Exception {
  123. Integer cameraType = Integer.parseInt(message.getBuildContext().get("cameraType").toString());
  124. String sceneCode = message.getBuildContext().get("sceneNum").toString();
  125. Long computeTime = message.getComputeTime();
  126. String path = message.getPath();
  127. Boolean buildSuccess = message.getBuildSuccess();
  128. JSONObject fdageData = getFdageData(path + File.separator + "capture" +File.separator+"data.fdage");
  129. Integer videoVersion = fdageData.getInteger("videoVersion");
  130. try {
  131. // 上传计算日志
  132. //如果是重复计算,没有走到计算逻辑,不需要上传日志文件
  133. log.info("开始上传计算日志");
  134. String buildLogPath = String.format(UploadFilePath.BUILD_LOG_PATH, sceneCode);
  135. fYunFileService.uploadFile(path + File.separator + "console.log", buildLogPath + "console.log");
  136. log.info("计算日志上传完成");
  137. if (!buildSuccess) {
  138. log.error("建模失败,修改状态为失败状态");
  139. sceneProService.update(new LambdaUpdateWrapper<ScenePro>()
  140. .set(ScenePro::getStatus, SceneStatus.FAILD.code())
  141. .eq(ScenePro::getNum, sceneCode));
  142. // 发送钉钉消息,计算失败
  143. buildSceneDTService.handModelFail("计算失败", message.getPath(), sceneCode, message.getHostName());
  144. return;
  145. }
  146. Map<String, String> uploadFiles = getUploadFiles(path,cameraType,fdageData);
  147. //建模成功走以下逻辑
  148. log.info("cameraType:{}",cameraType);
  149. //计算场景消耗磁盘空间
  150. long space = this.calUseSpace(uploadFiles);
  151. //读取计算结果文件生成videosJson
  152. JSONObject videosJson = this.getVideosJson(path, videoVersion, sceneCode, cameraType);
  153. //写入数据库
  154. ScenePro scenePro = this.updateDbPlus(sceneCode, space, videosJson.toJSONString(), computeTime,fdageData.containsKey("exportMeshObj") && fdageData.getIntValue("exportMeshObj") == 1);
  155. scenePro.setPayStatus(PayStatus.PAY.code());
  156. //如果相机容量不足,需要把场景的paystatus改为容量不足状态
  157. if (cameraType != 14) {
  158. scenePro.setPayStatus(this.getPayStatus(scenePro.getCameraId(), space));
  159. sceneProService.update(
  160. new LambdaUpdateWrapper<ScenePro>()
  161. .set(ScenePro::getPayStatus, scenePro.getPayStatus())
  162. .eq(ScenePro::getId, scenePro.getId()));
  163. }
  164. //上传文件
  165. log.info("开始上传场景计算结果数据,num:{}", sceneCode);
  166. fYunFileService.uploadMulFiles(uploadFiles);
  167. Map<String,String> damFileHeaders = new HashMap<>();
  168. damFileHeaders.put("Content-Encoding","gzip");
  169. if (!fYunFileService.getFyunType().equals(FYunTypeEnum.LOCAL.code())) {
  170. // dam 文件设置请求头
  171. uploadFiles.entrySet().stream().filter(entry -> FileNameUtil.extName(entry.getKey()).equals("dam"))
  172. .filter(entry -> new File(entry.getKey()).exists())
  173. .forEach(entry -> {
  174. // gzip压缩
  175. FileUtil.writeBytes(ZipUtil.gzip(new File(entry.getKey())), entry.getKey() + ".gzip");
  176. // 重命名
  177. FileUtil.rename(new File(entry.getKey() + ".gzip"), entry.getKey(), true);
  178. fYunFileService.uploadFile(entry.getKey(), entry.getValue(), damFileHeaders);
  179. });
  180. }
  181. //生成houseTypejson并上传
  182. uploadFiles.entrySet().stream().filter(entry-> FileNameUtil.getName(entry.getKey()).equals("floorplan_cad.json"))
  183. .forEach(entry-> uploadHouseTypeJson(sceneCode,entry.getKey()));
  184. //写scene.json
  185. log.info("生成scene.json上传oss并设置缓存,num:{}", sceneCode);
  186. this.writeSceneJson(sceneCode, videosJson,scenePro);
  187. //计算成功,通知APP
  188. Integer pushChannel = fdageData.getInteger("pushChannel");
  189. String pushToken = fdageData.getString("pushToken");
  190. this.pushMsgToApp(pushChannel,pushToken, cameraType, scenePro.getSceneName(), scenePro.getWebSite());
  191. SceneProEdit sceneProEdit = sceneProEditService.getByProId(scenePro.getId());
  192. sceneProEdit.setVersion(ObjectUtils.isEmpty(sceneProEdit.getVersion()) ? 0 : sceneProEdit.getVersion() + 1);
  193. sceneProEdit.setImagesVersion(ObjectUtils.isEmpty(sceneProEdit.getImagesVersion()) ? 0 : sceneProEdit.getImagesVersion() + 1);
  194. if (cameraType == 14) {
  195. //计算成功 激光转台相机 同步 请求
  196. fdkkLaserService.syncBuildResult(scenePro.getNum(), scenePro.getDataSource(),scenePro.getCreateTime());
  197. sceneProEdit.setMapVisi(0);
  198. }
  199. sceneProEditService.updateById(sceneProEdit);
  200. CreateObjUtil.deleteFile(path.replace(ConstantFilePath.BUILD_MODEL_PATH, "/") + "/capture");
  201. log.info("场景计算结果处理结束,场景码:{}", sceneCode);
  202. // 判断是否升级V4
  203. if(fdageData.containsKey("updateV4") && fdageData.getIntValue("updateV4") == 1){
  204. fdkkV4Service.upgradeToV4(sceneCode);
  205. }
  206. }catch (Exception e){
  207. e.printStackTrace();
  208. buildSceneDTService.handBaseFail("场景计算结果处理出错!", message.getPath(), sceneCode, "计算控制服务器");
  209. }
  210. }
  211. private Map<String, String> getUploadFiles(String path,Integer cameraType,JSONObject fdageData) throws Exception {
  212. ScenePro scenePro = sceneProService.getOne(
  213. new LambdaQueryWrapper<ScenePro>().like(ScenePro::getDataSource, path));
  214. if (ObjectUtils.isEmpty(scenePro)) {
  215. throw new Exception("未找到场景信息:" + path);
  216. }
  217. String projectNum = scenePro.getNum();
  218. String dataViewPath = String.format(ConstantFilePath.DATA_PATH_FORMAT, projectNum);
  219. String imagesPath = String.format(ConstantFilePath.IMAGE_PATH_FORMAT, projectNum);
  220. String videoPath = String.format(ConstantFilePath.VIDEO_PATH_FORMAT, projectNum);
  221. String resultsPath = path + File.separator + "results" + File.separator;
  222. String uploadData = FileUtils.readFile(resultsPath + "upload.json");
  223. JSONArray array = JSONObject.parseObject(uploadData).getJSONArray("upload");
  224. JSONObject fileJson = null;
  225. String fileName = "";
  226. Map<String, String> map = new HashMap();
  227. for (int i = 0; i < array.size(); ++i) {
  228. fileJson = array.getJSONObject(i);
  229. fileName = fileJson.getString("file");
  230. String filePath = resultsPath + fileName;
  231. if (!(new File(filePath)).exists()) {
  232. throw new Exception(filePath + "文件不存在");
  233. }
  234. if ("vision2.txt".equals(fileName)) {
  235. CreateObjUtil.convertTxtToVisionmodeldata(resultsPath + "vision2.txt", resultsPath + "vision2.modeldata");
  236. map.put(resultsPath + "vision2.modeldata", imagesPath + "vision2.modeldata");
  237. map.put(resultsPath + "vision2.txt", imagesPath + "vision2.txt");
  238. }
  239. if (fileJson.getIntValue("clazz") == 2) {
  240. map.put(filePath, imagesPath + ConstantFileName.modelUUID + "_50k_texture_jpg_high1/" + fileName.replace("tex/", ""));
  241. } else if (fileJson.getIntValue("clazz") == 3) {
  242. map.put(filePath, imagesPath + "pan/high/" + fileName.replace("high/", ""));
  243. } else if (fileJson.getIntValue("clazz") == 4) {
  244. map.put(filePath, imagesPath + "pan/low/" + fileName.replace("low/", ""));
  245. } else if (fileJson.getIntValue("clazz") == 5) {
  246. map.put(filePath, imagesPath + fileName);
  247. } else if (fileJson.getIntValue("clazz") == 7) {
  248. map.put(filePath, imagesPath + fileName);
  249. } else if (fileJson.getIntValue("clazz") == 10) {
  250. String mappingOssPath = String.format("scene_edit_data/%s/data/", projectNum) + fileName.replace("updown", "mapping");
  251. map.put(filePath, mappingOssPath);
  252. } else {
  253. if (fileJson.getIntValue("clazz") == 11 || fileJson.getIntValue("clazz") == 12) {
  254. map.put(filePath, videoPath + fileName.replace("videos/", ""));
  255. if (fileName.contains(".mp4")) {
  256. map.put(resultsPath + fileName.replace("mp4", "flv"), videoPath + fileName.replace("videos/", "").replace("mp4", "flv"));
  257. }
  258. }
  259. if (fileJson.getIntValue("clazz") == 16) {
  260. map.put(filePath, dataViewPath + fileName);
  261. }
  262. if (fileJson.getIntValue("clazz") == 18) {
  263. map.put(filePath, imagesPath + fileName);
  264. }
  265. }
  266. }
  267. if ((fdageData.containsKey("exportMeshObj") && fdageData.getIntValue("exportMeshObj") == 1)
  268. || (!ObjectUtils.isEmpty(cameraType) && (!cameraType.equals(14)))) {
  269. CreateObjUtil.convertTxtToDam(path + File.separator + "results" + File.separator + "tex" + File.separator + "modeldata.txt", path + File.separator + "results" + File.separator + ConstantFileName.modelUUID + "_50k.dam");
  270. CreateObjUtil.convertDamToLzma(path + File.separator + "results/");
  271. CreateObjUtil.convertTxtToDam(path + File.separator + "results" + File.separator + "tex" + File.separator + "modeldata.txt", path + File.separator + "results" + File.separator + ConstantFileName.modelUUID + "_50k.dam");
  272. map.put(path + File.separator + "results" + File.separator + ConstantFileName.modelUUID + "_50k.dam.lzma", imagesPath + ConstantFileName.modelUUID + "_50k.dam.lzma");
  273. map.put(path + File.separator + "results" + File.separator + ConstantFileName.modelUUID + "_50k.dam", imagesPath + ConstantFileName.modelUUID + "_50k.dam");
  274. }
  275. CreateObjUtil.convertTxtToVisionmodeldata(resultsPath + "vision.txt", resultsPath + "vision.modeldata");
  276. map.put(resultsPath + "vision.txt", imagesPath + "vision.txt");
  277. map.put(resultsPath + "vision.modeldata", imagesPath + "vision.modeldata");
  278. log.info("数据转换完成:" + projectNum);
  279. FileUtil.touch("/mnt/4Dkankan/scene/data" + File.separator + "data" + projectNum);
  280. map.put(resultsPath + "floorplan.json", dataViewPath + "floor.json");
  281. map.put(resultsPath + "floorplan_cad.json", dataViewPath + "floorplan_cad.json");
  282. map.put(path + File.separator + "capture/stitch_params.txt", dataViewPath + "stitch_params.txt");
  283. map.put(path + File.separator + "capture/Up.xml", dataViewPath + "Up.xml");
  284. map.put(path + File.separator + "capture/Up2.xml", dataViewPath + "Up2.xml");
  285. map.put(path + File.separator + "capture/Up.txt", dataViewPath + "Up.txt");
  286. map.put(path + File.separator + "capture/Up2.txt", dataViewPath + "Up2.txt");
  287. return map;
  288. }
  289. private JSONObject getFdageData(String dataFdagePath) {
  290. log.info("dataFdagePath 文件路径 :{}", dataFdagePath);
  291. String data = FileUtils.readFile(dataFdagePath);
  292. //获取data.fdage的内容
  293. JSONObject dataJson = new JSONObject();
  294. if(data!=null){
  295. dataJson = JSONObject.parseObject(data);
  296. }
  297. return dataJson;
  298. }
  299. private void writeSceneJson(String num, JSONObject videosJson, ScenePro scenePro) throws Exception{
  300. String dataViewPath = String.format(ConstantFilePath.DATA_PATH_FORMAT, num);
  301. //如果云端没有scene.json文件,生成一份
  302. JSONObject sceneJson = JSONObject.parseObject(JSONObject.toJSONString(scenePro));
  303. sceneJson.put("videos",videosJson);
  304. sceneJson.put("payStatus",scenePro.getPayStatus());
  305. String sceneJsonStr = JSON.toJSONString(sceneJson);
  306. //上传sceneJson文件
  307. fYunFileService.uploadFile(sceneJsonStr.getBytes(), dataViewPath + "scene.json");
  308. // 上传status JSON.
  309. JSONObject statusJson = new JSONObject();
  310. //临时将-2改成1,app还没完全更新
  311. statusJson.put("status", 1);
  312. statusJson.put("webSite", scenePro.getWebSite());
  313. statusJson.put("sceneNum", num);
  314. statusJson.put("thumb", scenePro.getThumb());
  315. statusJson.put("payStatus", scenePro.getPayStatus());
  316. FileUtils.writeFile(ConstantFilePath.SCENE_PATH + "data/data" + num + File.separator + "status.json", statusJson.toString());
  317. fYunFileService.uploadFile(ConstantFilePath.SCENE_PATH + "data/data" + num + File.separator + "status.json",
  318. dataViewPath + "status.json");
  319. }
  320. private void pushMsgToApp(Integer pushChannel, String pushToken, int cameraType, String sceneName, String webSite){
  321. log.info("推送消息,渠道是 {}, 手机token是 {}", pushChannel, pushToken);
  322. if(Objects.isNull(pushChannel) && StrUtil.isBlank(pushToken)){
  323. return;
  324. }
  325. String title = sceneName + "计算完成";
  326. String body = "您上传的" + sceneName + "计算完成,点击查看";
  327. try{
  328. if(FYunTypeEnum.AWS.code().equals(fYunFileService.getFyunType())){
  329. PushMsgUtil.googlePushMsg(ConstantFilePath.BASE_PATH + "/refreshToken.json", pushToken,
  330. title, body , webSite);
  331. return;
  332. }
  333. PushMessageConfig pushConfig = null;
  334. if(pushChannel == 0){
  335. if(cameraType == 10 || cameraType == 13){
  336. //ios
  337. pushConfig = new PushMessageConfig(PushMessageConfig.IOS_KEY_Z, PushMessageConfig.IOS_SECRET_Z);
  338. pushConfig.sendIOSUnicast(pushToken, "四维看看Minion",title, body, webSite);
  339. }else {
  340. //ios
  341. pushConfig = new PushMessageConfig(PushMessageConfig.IOS_KEY, PushMessageConfig.IOS_SECRET);
  342. pushConfig.sendIOSUnicast(pushToken, "四维看看Pro",title, body, webSite);
  343. }
  344. }else {
  345. if(cameraType == 10 || cameraType == 13){
  346. //ios
  347. //安卓
  348. pushConfig = new PushMessageConfig(PushMessageConfig.ANDROID_KEY_Z, PushMessageConfig.ANDROID_SECRET_Z);
  349. pushConfig.sendAndroidUnicast2(pushToken, "四维看看Minion",title, body, webSite);
  350. }else {
  351. //安卓
  352. pushConfig = new PushMessageConfig(PushMessageConfig.ANDROID_KEY, PushMessageConfig.ANDROID_SECRET);
  353. pushConfig.sendAndroidUnicast(pushToken, "四维看看Pro",title, body, webSite);
  354. }
  355. }
  356. log.info("消息推送结束!");
  357. }catch (Exception e){
  358. log.error("推送消息失败:", e);
  359. }
  360. }
  361. private JSONObject getVideosJson(String path, Integer videoVersion, String projectNum, int cameraType) throws Exception {
  362. //读取videos_hdr_param.json, 保存点位视频的value
  363. Map<String, Object> videoMap = new HashMap<>();
  364. String videosHdr = FileUtils.readFile(path + File.separator + "results/videos/videos_hdr_param.json");
  365. JSONArray videoArray = null;
  366. if(StringUtils.isNotEmpty(videosHdr)){
  367. videoArray = JSONObject.parseObject(videosHdr).getJSONArray("hdr_param");
  368. }
  369. if(videoArray != null){
  370. for(int i = 0, len = videoArray.size(); i < len; i++) {
  371. videoMap.put(videoArray.getJSONObject(i).getString("name"), videoArray.getJSONObject(i).getString("value"));
  372. if(videoArray.getJSONObject(i).containsKey("fov")){
  373. videoMap.put(videoArray.getJSONObject(i).getString("name") + "_fov", videoArray.getJSONObject(i).getString("fov"));
  374. }
  375. }
  376. }
  377. //获取upload中的video视频名称
  378. String uploadData = FileUtils.readFile(path + File.separator + "results" +File.separator+"upload.json");
  379. JSONObject uploadJson = null;
  380. JSONArray array = null;
  381. if(uploadData!=null) {
  382. uploadJson = JSONObject.parseObject(uploadData);
  383. array = uploadJson.getJSONArray("upload");
  384. }
  385. JSONObject fileJson = null;
  386. String fileName = "";
  387. //计算ts文件的大小,并拼接成json格式
  388. JSONArray jsonArray = new JSONArray();
  389. JSONObject videoJson = null;
  390. JSONObject videosJson = new JSONObject();
  391. long videoSize = 0L;
  392. for(int i = 0, len = array.size(); i < len; i++) {
  393. fileJson = array.getJSONObject(i);
  394. fileName = fileJson.getString("file");
  395. if(fileJson.getIntValue("clazz") == 11 && fileName.contains(".mp4") && !fileName.contains("-ios.mp4")){
  396. videoJson = new JSONObject();
  397. videoJson.put("id", fileName.substring(
  398. 0, fileName.lastIndexOf(".")).replace("videos/", ""));
  399. //如果ts文件存在,就计算ts大小
  400. if(new File(path + File.separator + "results" +File.separator+ fileName.replace(".mp4", ".ts")).exists()){
  401. videoSize = new File(path + File.separator + "results" +File.separator+ fileName.replace(".mp4", ".ts")).length();
  402. videoJson.put("tsSize", videoSize);
  403. }
  404. if(videoMap.containsKey(videoJson.get("id"))){
  405. videoJson.put("value", videoMap.get(videoJson.get("id")));
  406. }
  407. if(videoMap.containsKey(videoJson.get("id") + "_fov")){
  408. videoJson.put("blend_fov", videoMap.get(videoJson.get("id") + "_fov"));
  409. }else {
  410. videoJson.put("blend_fov", 7);
  411. }
  412. jsonArray.add(videoJson);
  413. }
  414. }
  415. videosJson.put("data", jsonArray);
  416. if(Objects.nonNull(videoVersion) && videoVersion >= 4){
  417. videosJson.put("version", 3);
  418. videosJson.put("upPath", fYunFileConfig.getHost() + String.format(ConstantFilePath.DATA_PATH_FORMAT, projectNum) + "Up.xml");
  419. if(cameraType == 13){
  420. //转台相机
  421. videosJson.put("upPath", videosJson.getString("upPath").replace(".xml", ".txt"));
  422. }
  423. }else {
  424. videosJson.put("version", 1);
  425. videosJson.put("upPath", fYunFileConfig.getHost() + String.format(ConstantFilePath.DATA_PATH_FORMAT, projectNum) + "Up2.xml");
  426. if(cameraType == 13){
  427. //转台相机
  428. videosJson.put("upPath", videosJson.getString("upPath").replace(".xml", ".txt"));
  429. }
  430. }
  431. if(cameraType == 5 || cameraType == 6){
  432. videosJson.put("version", 1);
  433. videosJson.put("upPath", fYunFileConfig.getHost() + String.format(ConstantFilePath.DATA_PATH_FORMAT, projectNum) + "stitch_params.txt");
  434. }
  435. return videosJson;
  436. }
  437. private Long calUseSpace(Map<String, String> uploadFile) {
  438. return uploadFile.keySet().stream().map(File::new).filter(File::exists).mapToLong(File::length).sum();
  439. }
  440. /**
  441. * <p>
  442. 双目场景更新数据库
  443. * </p>
  444. * @author dengsixing
  445. * @date 2022/3/21
  446. * @param num
  447. * @param space
  448. **/
  449. private void updateDb4Sm(String num, long space){
  450. List<ScenePro> ScenePlusList = sceneProService.list(
  451. new LambdaQueryWrapper<ScenePro>().select(ScenePro::getId).eq(ScenePro::getNum, num));
  452. if(CollUtil.isEmpty(ScenePlusList)){
  453. return ;
  454. }
  455. List<Long> sceneIds = ScenePlusList.stream().map(ScenePro::getId).collect(Collectors.toList());
  456. //更新场景创建时间
  457. sceneProService.update(new LambdaUpdateWrapper<ScenePro>().in(ScenePro::getId, sceneIds)
  458. .set(ScenePro::getStatus, SceneStatus.NO_DISPLAY.code())
  459. .set(ScenePro::getSpace,space)
  460. .set(ScenePro::getCreateTime,new Date())
  461. );
  462. }
  463. private ScenePro updateDbPlus(String num, Long space, String videosJson, Long computeTime,boolean isObj){
  464. ScenePro scenePro = sceneProService.getByNum(num);
  465. if (ObjectUtils.isEmpty(scenePro)) {
  466. return null;
  467. }
  468. sceneProService.update(new LambdaUpdateWrapper<ScenePro>()
  469. .in(ScenePro::getId, scenePro.getId())
  470. .set(ScenePro::getStatus, SceneStatus.NO_DISPLAY.code())
  471. .set(ScenePro::getUpdateTime, new Date())
  472. .set(ScenePro::getCreateTime, new Date())
  473. .set(ScenePro::getSpace, space)
  474. .set(ScenePro::getComputeTime, computeTime)
  475. .set(ScenePro::getVideos, videosJson)
  476. .set(ScenePro::getIsObj, isObj ? 1 : 0)
  477. .set(ScenePro::getPayStatus, PayStatus.PAY.code()));
  478. return scenePro;
  479. }
  480. private int getPayStatus(Long cameraId, Long space) throws Exception{
  481. //更新相机使用用量
  482. CameraDetail cameraDetail = cameraDetailService.updateCameraDetailByCameraIdAndSpace(cameraId, space);
  483. Long totalSpace = cameraDetail.getTotalSpace();
  484. UserIncrement userIncrement = userIncrementService.getByCameraId(cameraId);
  485. if(!ObjectUtils.isEmpty(userIncrement) && userIncrement.getIsExpired().equals(ExpiredStatus.NOT_EXPIRED.code())){
  486. IncrementType type = incrementTypeService.getById(userIncrement.getIncrementTypeId());
  487. if (ObjectUtils.isEmpty(type) || type.getType().equals("cn")) {
  488. return PayStatus.PAY.code();
  489. } else if (type.getType().equals("eur")) {
  490. totalSpace = type.getCameraCapacity() * 1024 * 1024L;
  491. }
  492. }
  493. return totalSpace.compareTo(cameraDetail.getUsedSpace()) == -1 ? PayStatus.NO_CAPACITY.code() : PayStatus.PAY.code();
  494. }
  495. public void uploadHouseTypeJson(String num, String floorPlanCardFilePath) {
  496. if (!new File(floorPlanCardFilePath).exists()) {
  497. log.error("floorplan_cad.json 文件不存在,文件路径:{}", floorPlanCardFilePath);
  498. return;
  499. }
  500. JSONObject json = CreateHouseJsonUtil.createHouseTypeJsonByCad(floorPlanCardFilePath);
  501. String hourseTypeJsonPath = String.format(ConstantFilePath.DATA_PATH_FORMAT, num) + "houseType.json";
  502. fYunFileService.uploadFile(json.toJSONString().getBytes(), hourseTypeJsonPath);
  503. hourseTypeJsonPath = String.format(UploadFilePath.DATA_EDIT_PATH, num) + "houseType.json";
  504. fYunFileService.uploadFile(json.toJSONString().getBytes(), hourseTypeJsonPath);
  505. }
  506. }