|
@@ -328,7 +328,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];
|
|
@@ -348,6 +348,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());
|
|
@@ -566,10 +567,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);
|
|
@@ -578,7 +579,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());
|
|
@@ -606,10 +607,20 @@ 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);
|
|
|
+ Integer shootCount = commonService.getShootCount(scenePlus.getNum());
|
|
|
+ if(Objects.nonNull(shootCount)){
|
|
|
+ scenePlusExt.setShootCount(shootCount);
|
|
|
+ }
|
|
|
+ Integer slamCount = commonService.getSlamCount(scenePlus.getNum());
|
|
|
+ if(Objects.nonNull(slamCount)){
|
|
|
+ scenePlusExt.setSlamCount(slamCount);
|
|
|
+ }
|
|
|
+ int mixture = CommonStatus.NO.code();
|
|
|
+ //如果点位数量和帧数都不为空,就是混合模式
|
|
|
+ if(Objects.nonNull(scenePlusExt.getShootCount()) && scenePlusExt.getShootCount() > 0
|
|
|
+ && Objects.nonNull(scenePlusExt.getSlamCount()) && scenePlusExt.getSlamCount() > 0){
|
|
|
+ mixture = CommonStatus.YES.code();
|
|
|
+ }
|
|
|
scenePlusExt.setMixture(mixture);
|
|
|
|
|
|
scenePlusExtService.updateById(scenePlusExt);
|