|
@@ -353,7 +353,7 @@ public class BuildSceneServiceImpl implements IBuildSceneService {
|
|
|
aiService.detectScenePano(scenePlus, scenePlusExt, path);
|
|
|
|
|
|
//写入数据库
|
|
|
- this.updateDbPlus(scenePlus.getSceneSource(), space, videosJson.toJSONString(), message.getComputeTime(),isObj,scenePlusExt);
|
|
|
+ this.updateDbPlus(scenePlus, scenePlusExt, space, videosJson.toJSONString(), message.getComputeTime(),isObj);
|
|
|
|
|
|
Object[] editInfoArr = commonService.updateEditInfo(scenePlus);
|
|
|
SceneEditInfo sceneEditInfo = (SceneEditInfo)editInfoArr[0];
|
|
@@ -375,6 +375,7 @@ public class BuildSceneServiceImpl implements IBuildSceneService {
|
|
|
.sceneStatus(2)
|
|
|
.createTime(scenePlus.getCreateTime())
|
|
|
.shootCount(scenePlusExt.getShootCount())
|
|
|
+ .slamCount(scenePlusExt.getSlamCount())
|
|
|
.payStatus(scenePlus.getPayStatus())
|
|
|
.mixture(scenePlusExt.getMixture())
|
|
|
.version(SceneVersionType.V4.code()).build());
|
|
@@ -616,10 +617,10 @@ public class BuildSceneServiceImpl implements IBuildSceneService {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- private void updateDbPlus(int sceneSource,Long space,String videosJson, Long computeTime,boolean isObj,ScenePlusExt scenePlusExt){
|
|
|
+ private void updateDbPlus(ScenePlus scenePlus,ScenePlusExt scenePlusExt, Long space,String videosJson, Long computeTime,boolean isObj){
|
|
|
|
|
|
scenePlusExt.setSpace(space);
|
|
|
- scenePlusExt.setComputeTime(computeTime.toString());
|
|
|
+ scenePlusExt.setComputeTime(computeTime);
|
|
|
scenePlusExt.setAlgorithmTime(new Date());
|
|
|
scenePlusExt.setVideos(videosJson);
|
|
|
scenePlusExt.setIsObj(isObj ? 1 : 0);
|
|
@@ -628,7 +629,7 @@ public class BuildSceneServiceImpl implements IBuildSceneService {
|
|
|
scenePlusExt.setSceneScheme(3);
|
|
|
}
|
|
|
|
|
|
- switch (SceneSource.get(sceneSource)){
|
|
|
+ switch (SceneSource.get(scenePlus.getSceneSource())){
|
|
|
case BM:
|
|
|
scenePlusExt.setSceneResolution(SceneResolution.two_K.code());
|
|
|
scenePlusExt.setSceneFrom(SceneFrom.PRO.code());
|
|
@@ -656,10 +657,13 @@ public class BuildSceneServiceImpl implements IBuildSceneService {
|
|
|
// scenePlusExt.setModelKind(modelKind);
|
|
|
|
|
|
//统计点位数量
|
|
|
- Map<String, Integer> result = this.getShootCount(scenePlusExt);
|
|
|
- Integer shootCount = result.get("shootCount");
|
|
|
- Integer mixture = result.get("mixture");
|
|
|
- scenePlusExt.setShootCount(shootCount);
|
|
|
+ scenePlusExt.setShootCount(commonService.getShootCount(scenePlus.getNum()));
|
|
|
+ scenePlusExt.setSlamCount(commonService.getSlamCount(scenePlus.getNum()));
|
|
|
+ int mixture = CommonStatus.NO.code();
|
|
|
+ //如果点位数量和帧数都不为空,就是混合模式
|
|
|
+ if(Objects.nonNull(scenePlusExt.getShootCount()) && Objects.nonNull(scenePlusExt.getSlamCount())){
|
|
|
+ mixture = CommonStatus.YES.code();
|
|
|
+ }
|
|
|
scenePlusExt.setMixture(mixture);
|
|
|
|
|
|
scenePlusExtService.updateById(scenePlusExt);
|