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.thread.ThreadUtil;
import cn.hutool.core.util.ArrayUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.ZipUtil;
import cn.hutool.extra.qrcode.QrCodeUtil;
import cn.hutool.extra.qrcode.QrConfig;
import cn.hutool.http.ContentType;
import cn.hutool.http.HttpUtil;
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.*;
import com.fdkankan.common.util.FileUtils;
import com.fdkankan.contro.bean.SyncLaserResultBean;
import com.fdkankan.contro.constant.UserEditDataType;
import com.fdkankan.contro.entity.*;
import com.fdkankan.contro.mq.service.IBuildSceneService;
import com.fdkankan.contro.service.*;
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.enums.ModelTypeEnums;
import com.fdkankan.model.utils.CreateHouseJsonUtil;
import com.fdkankan.model.utils.CreateObjUtil;
import com.fdkankan.model.utils.SceneUtil;
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 com.fdkankan.redis.util.RedisUtil;
import com.fdkankan.rubbersheeting.ScalingService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.http.HttpHeaders;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cloud.context.config.annotation.RefreshScope;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.*;
import java.util.Map.Entry;
import java.util.stream.Collectors;
/**
*
* TODO
*
*
* @author dengsixing
* @since 2022/4/20
**/
@Slf4j
@Service
@RefreshScope
public class BuildSceneServiceImpl implements IBuildSceneService {
@Value("${queue.modeling.modeling-call}")
private String queueModelingCall;
@Value("${queue.modeling.single.modeling-call}")
private String singleModelingCall;
@Value("${model.type:#{null}}")
private String modelType;
@Value("${env:gn}")
private String env;
@Value("#{'${build.scene.post.not-delete-nas-nums:}'.split(',')}")
private List notDeleteNasNumList;
@Value("4dkk.laserService.bucket")
private String laserBucket;
@Autowired
private RabbitMqProducer mqProducer;
@Resource
private FYunFileServiceInterface fYunFileService;
@Autowired
private ICameraDetailService cameraDetailService;
@Autowired
private ISceneEditInfoService sceneEditInfoService;
@Autowired
private ISceneEditControlsService sceneEditControlsService;
@Autowired
private FYunFileConfig fYunFileConfig;
@Autowired
private RedisUtil redisUtil;
@Autowired
private IScenePlusService scenePlusService;
@Autowired
private IScenePlusExtService scenePlusExtService;
@Autowired
private ISceneEditInfoExtService sceneEditInfoExtService;
@Autowired
private IUserIncrementService userIncrementService;
@Autowired
private IFdkkLaserService fdkkLaserService;
@Autowired
private IBuildSceneDTService buildSceneDTService;
@Autowired
private IIncrementTypeService incrementTypeService;
@Autowired
private ICompanyService companyService;
@Autowired
private ISceneAsynOperLogService sceneAsynOperLogService;
@Autowired
private ICommonService commonService;
@Autowired
private ISceneBuildProcessLogService sceneBuildProcessLogService;
@Autowired
private ISceneColdStorageService sceneColdStorageService;
@Autowired
private ScalingService scalingService;
@Autowired
private RabbitMqProducer rabbitMqProducer;
@Autowired
private ILinkPanService linkPanService;
@Autowired
private IWbService wbService;
@Autowired
private IAiService aiService;
@Autowired
private IBuildService buildService;
@Override
public void buildScenePre(BuildSceneCallMessage message) throws Exception {
String num = message.getSceneNum();
try {
Map sceneStatusParam = new HashMap<>();
sceneStatusParam.put("num", num);
sceneStatusParam.put("status", 0);
commonService.sendUpdateSceneStatusMqToQueues(sceneStatusParam);
//重新计算时需要删除文件夹,否知使用缓存
if(new File(message.getPath() + File.separator + "results").exists()){
FileUtils.deleteDirectory(message.getPath() + File.separator + "results");
}
//由于刘强说caches会影响计算结果,所以这里删除caches
if(new File(message.getPath() + File.separator + "caches").exists()){
FileUtils.deleteDirectory(message.getPath() + File.separator + "caches");
}
//删除点位校准数据
if (Objects.nonNull(message.getExt())
&& message.getExt().containsKey("deleteExtras")
&& (Boolean) message.getExt().get("deleteExtras")) {
String extras = String.format(UploadFilePath.scene_result_data_path, num).concat("extras");
if(CollUtil.isNotEmpty(fYunFileService.listRemoteFiles(extras))){
fYunFileService.deleteFolder(extras);
}
}
//用户相机重新全量上传,需要解冻结
sceneColdStorageService.unfreeze(num, "用户相机重新全量上传", message.getPath());
//根据相机类型,组装资源路径
//下载资源到本地
this.downLoadSource(message, message.getPath());
List linkPanTargetList = linkPanService.genLinkPanPre(num);
if(CollUtil.isNotEmpty(linkPanTargetList)){
Map ext = message.getExt();
if(Objects.isNull(ext)){
ext = new HashMap<>();
message.setExt(ext);
}
ext.put("linkPanTargetList", linkPanTargetList);
}
JSONObject fdageJson = JSONObject.parseObject(FileUtils.readFile(message.getPath().concat("/capture/data.fdage")));
boolean rewrite = false;
// 兼容旧的数据,防止OnlyExportMeshObj标志未删除掉
if (fdageJson.containsKey("OnlyExportMeshObj")) {
log.info("data.fdage 包含 OnlyExportMeshObj,进行去除!");
// 写入data.fdage 防止重算
fdageJson.remove("OnlyExportMeshObj");
String ossPath = getOssPath(message.getPath());
fYunFileService.uploadFile(fdageJson.toJSONString().getBytes(), ossPath + "data.fdage");
rewrite = true;
}
if (!ObjectUtils.isEmpty(modelType)) {
// 修改dataFdage文件
fdageJson.put("modelType", modelType);
rewrite = true;
}
if (rewrite) {
FileUtils.writeFile(message.getPath().concat("/capture/data.fdage"), fdageJson.toJSONString());
}
message.getBuildContext().put("cameraType",message.getCameraType());
//查询是否超过比例50%,如果超过,则启动128G服务器弹性伸缩
Float maxRate = fdageJson.getFloat("maxRate");
if(env.equals("gn") && Objects.nonNull(maxRate) && maxRate > 50){
Map ext = message.getExt();
if(Objects.isNull(ext)){
ext = new HashMap<>();
message.setExt(ext);
}
ext.put("128G", 1);
}
message.setBizType("standard");
//写入算法参数文件
buildService.writeDataJson(message, fdageJson, null, null);
log.info("场景计算资源准备结束,场景码:{}", message.getSceneNum());
}catch (Exception e){
log.error("场景计算前置处理出错,num"+num, e);
buildSceneDTService.handBaseFail("场景计算资源准备异常!", message.getPath(), message.getSceneNum(), "计算控制服务器");
throw e;
}
}
private String getOssPath(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("/");
}
return ossPath;
}
@Override
public void downLoadSource(BuildSceneCallMessage buildSceneMqMessage,String path){
String ossPath = getOssPath(path);
fYunFileService.downloadFileByCommand(path + File.separator + "capture", ossPath);
}
@Override
public void buildScenePost(BuildSceneResultMqMessage message) throws Exception {
String sceneCode = message.getBuildContext().get("sceneNum").toString();
String path = message.getPath();
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 (!message.getBuildSuccess()) {
log.error("建模失败,修改状态为失败状态");
scenePlusService.update(new LambdaUpdateWrapper()
.set(ScenePlus::getSceneStatus, SceneStatus.FAILD.code())
.eq(ScenePlus::getNum, sceneCode));
wbService.sendMq(sceneCode, CommonSuccessStatus.FAIL.code());
// 发送钉钉消息,计算失败
buildSceneDTService.handModelFail("计算失败", message.getPath(), sceneCode, message.getHostName());
return;
}
JSONObject fdageData = commonService.getFdageData(path + File.separator + "capture" +File.separator+"data.fdage");
ScenePlus scenePlus = scenePlusService.getScenePlusByNum(sceneCode);
Integer cameraType = Integer.parseInt(message.getBuildContext().get("cameraType").toString());
Map uploadFiles = commonService.getUploadFiles(scenePlus,path,cameraType,fdageData);
scenePlus.setPayStatus(PayStatus.PAY.code());
scenePlus.setUpdateTime(new Date());
scenePlus.setSceneStatus(SceneStatus.NO_DISPLAY.code());
Integer videoVersion = fdageData.getInteger("videoVersion");
//读取计算结果文件生成videosJson
JSONObject videosJson = commonService.getVideosJson(path, videoVersion, sceneCode, cameraType);
ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
boolean isObj = fdageData.containsKey("exportMeshObj") && fdageData.getIntValue("exportMeshObj") == 1;
//上传全景图俯视图
this.uploadFloorCad(path, sceneCode, uploadFiles);
this.uploadFreespace(sceneCode, path, uploadFiles);
log.info("开始上传场景计算结果数据,num:{}", sceneCode);
//由于3dtiles算法mesh文件发生变化,所以这里需要先清除一下oss的mesh目录,避免存在旧算法obj文件
fYunFileService.deleteFolder(String.format(UploadFilePath.DATA_VIEW_PATH, sceneCode) + "mesh");
fYunFileService.deleteFolder(String.format(UploadFilePath.IMG_VIEW_PATH, sceneCode) + ModelKind.THREE_D_TILE.code());
//上传文件
fYunFileService.uploadMulFiles(uploadFiles);
//修改oss上dam的内容编码
Map damFileHeaders = new HashMap<>();
damFileHeaders.put("Content-Encoding","gzip");
String damPath = path + File.separator + "results" + File.separator + ConstantFileName.modelUUID + "_50k.dam";
fYunFileService.uploadFile(damPath, String.format(UploadFilePath.IMG_VIEW_PATH, sceneCode) + ConstantFileName.modelUUID + "_50k.dam", damFileHeaders);
//拷贝部分文件到编辑目录,用于用户编辑
this.copyToEditDir(sceneCode);
//计算完毕后,同步全景图到缓存目录
// this.cachePanorama(path, sceneCode);
//生成houseTypejson并上传
boolean existHouseType = this.uploadHouseTypeJson(sceneCode, path);
scenePlus.setHouseType(existHouseType ? CommonStatus.YES.code().intValue() : CommonStatus.NO.code().intValue());
//生成场景关联数据
Integer links = linkPanService.genLinkPanPost(sceneCode);
//生成floorpan.json
commonService.uploadFloorplanJson(sceneCode, path);
//上传平面图ai识别结果
commonService.uploadFloorplanAi(sceneCode, path);
//重置异步操作记录
commonService.removeSceneAsynOperLog(sceneCode);
//清除用户编辑业务数据
Set bizs = new HashSet<>();
bizs.add(UserEditDataType.BOX_MODEL.message());
bizs.add(UserEditDataType.FLOORPLAN.message());
bizs.add(UserEditDataType.FILTERS.message());
commonService.initUserEditData(sceneCode, bizs, null);
//上传计算结果文件
commonService.uploadBuildResultData(sceneCode, path, SceneVersionType.V4.code());
//容量统计
Long space = commonService.getSpace(sceneCode);
//ai识别
aiService.detectScenePano(scenePlus, scenePlusExt, path);
//写入数据库
this.updateDbPlus(scenePlus, scenePlusExt, space, videosJson.toJSONString(), message.getComputeTime(),isObj);
Object[] editInfoArr = commonService.updateEditInfo(scenePlus);
SceneEditInfo sceneEditInfo = (SceneEditInfo)editInfoArr[0];
SceneEditInfoExt sceneEditInfoExt = (SceneEditInfoExt)editInfoArr[1];
SceneEditControls sceneEditControls = (SceneEditControls)editInfoArr[2];
sceneEditInfoExt.setLinks(links);
sceneEditInfoExtService.updateById(sceneEditInfoExt);
//如果相机容量不足,需要把场景的paystatus改为容量不足状态
scenePlus.setPayStatus(commonService.getPayStatus(scenePlus.getCameraId(), space, fdageData));
//统计原始资源大小
scenePlusExt.setOrigSpace(FileUtil.size(new File(path.concat(File.separator).concat("capture"))));
if (cameraType == 14) {
//计算成功 激光转台相机 同步 请求
fdkkLaserService.syncBuildResult(
SyncLaserResultBean.builder()
.num(sceneCode).dataSource(path)
.sceneStatus(2)
.createTime(scenePlus.getCreateTime())
.shootCount(scenePlusExt.getShootCount())
.slamCount(scenePlusExt.getSlamCount())
.payStatus(scenePlus.getPayStatus())
.mixture(scenePlusExt.getMixture())
.version(SceneVersionType.V4.code()).build());
sceneEditControlsService.update(new LambdaUpdateWrapper().set(SceneEditControls::getShowMap,0)
.eq(SceneEditControls::getEditInfoId,sceneEditInfo.getId()));
sceneEditControls.setShowMap(0);
} else if (new File(path + "/results/laserData/vision_edit.txt").exists()) {
fdkkLaserService.cloudPointBuild(sceneCode,path);
}
log.info("生成scene.json上传oss并设置缓存,num:{}", sceneCode);
CameraDetail cameraDetail = cameraDetailService.getByCameraId(scenePlus.getCameraId());
Company company = !ObjectUtils.isEmpty(cameraDetail.getCompanyId()) ? companyService.getById(cameraDetail.getCompanyId()) : null;
//写scene.json
commonService.writeSceneJson(sceneCode,sceneEditInfo, sceneEditInfoExt, sceneEditControls, scenePlus,scenePlusExt,company);
String qrLogo = !ObjectUtils.isEmpty(company) && !ObjectUtils.isEmpty(company.getQrLogo()) ? company.getQrLogo() : null;
qrLogo = ObjectUtils.isEmpty(qrLogo) && !ObjectUtils.isEmpty(sceneEditInfoExt.getShareLogoImg()) ? fYunFileConfig.getHost().concat(sceneEditInfoExt.getShareLogoImg()) : null;
createQrCode(sceneCode, scenePlusExt, qrLogo);
//计算成功,通知APP
Integer pushChannel = fdageData.getInteger("pushChannel");
String pushToken = fdageData.getString("pushToken");
this.pushMsgToApp(pushChannel,pushToken, cameraType, scenePlus.getTitle(), scenePlusExt.getWebSite());
// //删除计算目录
if(CollUtil.isEmpty(notDeleteNasNumList) || !notDeleteNasNumList.contains(sceneCode)){
CreateObjUtil.deleteFile(path.replace(ConstantFilePath.BUILD_MODEL_PATH, "/"));
}
this.uploadStatusJson(scenePlus, scenePlusExt);
scenePlusService.updateById(scenePlus);
scenePlusExtService.updateById(scenePlusExt);
//国际环境需要发邮件通知
if("eur".equals(env) &&
!scenePlus.getSceneSource().equals(SceneSource.JG.code()) &&
!scenePlus.getSceneSource().equals(SceneSource.SG.code())){
commonService.sendEmail(sceneCode, "standar");
}
//发送到文保系统
wbService.sendMq(sceneCode, CommonSuccessStatus.SUCCESS.code());
//四川日报打包消费
this.sendMqToPackScene4Scrb(sceneCode);
//拜城公安需求
this.sendMqForBcgn(sceneCode);
log.info("场景计算结果处理结束,场景码:{}", sceneCode);
}catch (Exception e){
log.error("场景计算结果处理出错,num"+sceneCode, e);
buildSceneDTService.handBaseFail("场景计算结果处理出错!", message.getPath(), sceneCode, "计算控制服务器");
throw e;
} finally {
Map sceneStatusParam = new HashMap<>();
sceneStatusParam.put("num", sceneCode);
sceneStatusParam.put("status", message.getBuildSuccess() ? 1 : -1);
commonService.sendUpdateSceneStatusMqToQueues(sceneStatusParam);
}
}
void uploadFreespace(String num, String path, Map map){
String plyPath = path + "/results/laserData/cover/final_freespace.ply";
String pngPath = path + "/results/laserData/cover/final_freespace.png";
String infoJsonPath = path + "/results/laserData/cover/info.json";
if(FileUtil.exist(plyPath)){
map.put(plyPath, String.format(UploadFilePath.IMG_VIEW_PATH, num) + "cover/" + FileUtil.getName(plyPath));
}
if(FileUtil.exist(pngPath)){
map.put(pngPath, String.format(UploadFilePath.IMG_VIEW_PATH, num) + "cover/" + FileUtil.getName(pngPath));
}
if(FileUtil.exist(infoJsonPath)){
map.put(infoJsonPath, String.format(UploadFilePath.IMG_VIEW_PATH, num) + "cover/" + FileUtil.getName(infoJsonPath));
}
}
private void sendMqToPackScene4Scrb(String num){
ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
if(Objects.isNull(scenePlus.getCameraId())){
return;
}
CameraDetail cameraDetail = cameraDetailService.getByCameraId(scenePlus.getCameraId());
if(Objects.isNull(cameraDetail) || Objects.isNull(cameraDetail.getCompanyId()) || cameraDetail.getCompanyId() != 30){
return;
}
JSONObject jsonObject = new JSONObject();
jsonObject.put("num", num);
mqProducer.sendByWorkQueue("scrb-package-scene", jsonObject);
}
private void sendMqForBcgn(String num){
Map map = new HashMap<>();
map.put("num", num);
mqProducer.sendByWorkQueue("detect-queue", map);
}
private void uploadFloorCad(String path, String num, Map uploadFiles){
//户型图上传
String dataViewPath = UploadFilePath.DATA_VIEW_PATH + "floor-cad-%s.%s";
String floorCadPath = path + "/results/floorplan_cad";
List floorCadList = FileUtils.getFileList(floorCadPath);
if(CollUtil.isNotEmpty(floorCadList)){
floorCadList.stream().forEach(str->{
String substring = str.substring(str.lastIndexOf(File.separator) + 1);
String[] arr = substring.split("floor");
String[] arr2 = arr[1].split("\\.");
uploadFiles.put(str, String.format(dataViewPath, num, arr2[0], arr2[1]));
});
}
}
private void uploadStatusJson(ScenePlus scenePlus, ScenePlusExt scenePlusExt){
String num = scenePlus.getNum();
String dataViewPath = String.format(UploadFilePath.DATA_VIEW_PATH, num);
Integer status = 1;
if(scenePlus.getSceneSource() == 4 || scenePlus.getSceneSource() == 5){//如果是激光场景,需要激光系统那边完全处理好之后再发mq通知更新状态
status = 0;
}
// 上传status JSON.
JSONObject statusJson = new JSONObject();
//临时将-2改成1,app还没完全更新
statusJson.put("status", status);
statusJson.put("webSite", scenePlusExt.getWebSite());
statusJson.put("sceneNum", num);
statusJson.put("thumb", scenePlusExt.getThumb());
statusJson.put("payStatus", scenePlus.getPayStatus());
statusJson.put("sceneScheme", scenePlusExt.getSceneScheme());
FileUtils.writeFile(ConstantFilePath.SCENE_PATH + "data/data" + num + File.separator + "status.json", statusJson.toString());
Map headers = new HashMap<>();
headers.put(HttpHeaders.CONTENT_TYPE, ContentType.JSON.getValue());
fYunFileService.uploadFile(ConstantFilePath.SCENE_PATH + "data/data" + num + File.separator + "status.json", dataViewPath + "status.json", headers);
}
private void createQrCode(String num, ScenePlusExt scenePlusExt, String qrLogo) {
String localLogoPath = null;
if (!ObjectUtils.isEmpty(qrLogo)) {
try {
localLogoPath = ConstantFilePath.AGENT_PATH + qrLogo.substring(qrLogo.lastIndexOf("//") + 1);
HttpUtil.downloadFile(qrLogo, localLogoPath);
} catch (Exception e) {
log.error("公司logo下载失败:{}", qrLogo);
localLogoPath = null;
}
}
//生成二维码
String outPathZh = ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+ num +".png";
String outPathEn = ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+ num +"_en.png";
QrConfig qrConfig = QrConfig.create();
qrConfig.setWidth(1024);
qrConfig.setHeight(1024);
if(StrUtil.isNotEmpty(localLogoPath)){
qrConfig.setImg(localLogoPath);
}
QrCodeUtil.generate(scenePlusExt.getWebSite(), qrConfig, FileUtil.file(outPathZh));
QrCodeUtil.generate(scenePlusExt.getWebSite() + "&lang=en", qrConfig, FileUtil.file(outPathEn));
//上传二维码
fYunFileService.uploadFile(outPathZh, String.format(UploadFilePath.DOWNLOADS_QRCODE, num) + num + ".png");
fYunFileService.uploadFile(outPathEn, String.format(UploadFilePath.DOWNLOADS_QRCODE, num) + num + "_en.png");
if(!ObjectUtils.isEmpty(localLogoPath)){
FileUtils.deleteFile(localLogoPath);
}
}
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 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 entry : map.entrySet()) {
fYunFileService.copyFileInBucket(entry.getValue(), entry.getKey());
}
}
private void updateDbPlus(ScenePlus scenePlus,ScenePlusExt scenePlusExt, Long space,String videosJson, Long computeTime,boolean isObj){
scenePlusExt.setSpace(space);
scenePlusExt.setComputeTime(computeTime);
scenePlusExt.setAlgorithmTime(new Date());
scenePlusExt.setVideos(videosJson);
scenePlusExt.setIsObj(isObj ? 1 : 0);
if(ModelTypeEnums.TILE_CODE.equals(modelType)){
scenePlusExt.setSceneScheme(3);
}
switch (SceneSource.get(scenePlus.getSceneSource())){
case BM:
scenePlusExt.setSceneResolution(SceneResolution.two_K.code());
scenePlusExt.setSceneFrom(SceneFrom.PRO.code());
break;
case SM:
scenePlusExt.setSceneResolution(SceneResolution.one_k.code());
scenePlusExt.setSceneFrom(SceneFrom.LITE.code());
break;
case ZT:
scenePlusExt.setSceneResolution(SceneResolution.four_K.code());
scenePlusExt.setSceneFrom(SceneFrom.MINION.code());
break;
case JG:
scenePlusExt.setSceneResolution(SceneResolution.four_K.code());
scenePlusExt.setSceneFrom(SceneFrom.LASER.code());
break;
case SG:
scenePlusExt.setSceneResolution(SceneResolution.four_K.code());
scenePlusExt.setSceneFrom(SceneFrom.LASER.code());
break;
}
String sceneKind = scenePlusExt.getSceneScheme() == 3 ? SceneKind.FACE.code():SceneKind.TILES.code();
scenePlusExt.setSceneKind(sceneKind);
// scenePlusExt.setModelKind(modelKind);
//统计点位数量
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);
}
private Map getShootCount(ScenePlusExt scenePlusExt){
Map result = new HashMap<>();
Integer shootCount = 0;
Integer mixture = Objects.isNull(scenePlusExt.getMixture()) ? 0 : scenePlusExt.getMixture();
String homePath = SceneUtil.getHomePath(scenePlusExt.getDataSource());
JSONObject dataFdageObj = JSON.parseObject(fYunFileService.getFileContent(homePath.concat("data.fdage")));
if(Objects.nonNull(dataFdageObj)){
JSONArray points = dataFdageObj.getJSONArray("points");
if(CollUtil.isNotEmpty(points)){
shootCount = points.size();
}
}
if(Objects.nonNull(shootCount) && shootCount > 0){
if(Objects.nonNull(scenePlusExt.getLocation()) && scenePlusExt.getLocation() == 6){
mixture = CommonStatus.YES.code().intValue();
}
}else{
String slamDataStr = fYunFileService.getFileContent(homePath.concat("slam_data.json"));
JSONObject slamDataObj = JSON.parseObject(slamDataStr);
if(Objects.nonNull(slamDataObj)){
JSONArray viewsInfo = slamDataObj.getJSONArray("views_info");
if(CollUtil.isNotEmpty(viewsInfo)){
shootCount = viewsInfo.stream().mapToInt(info -> {
return ((JSONObject) info).getJSONArray("list_pose").size();
}).sum();
}
}
mixture = CommonStatus.NO.code().intValue();
}
result.put("shootCount", shootCount);
result.put("mixture", mixture);
return result;
}
public static void main(String[] args) {
JSONObject dataFdageObj = JSON.parseObject(null);
System.out.println(dataFdageObj);
}
public boolean uploadHouseTypeJson(String num, String dataSource) {
String floorPlanCardFilePath = dataSource + File.separator + "results/floorplan_cad.json";
if (!new File(floorPlanCardFilePath).exists()) {
log.warn("floorplan_cad.json 文件不存在,文件路径:{}", floorPlanCardFilePath);
return false;
}
JSONObject json = CreateHouseJsonUtil.createHouseTypeJsonByCad(floorPlanCardFilePath);
if(Objects.isNull(json)){
return false;
}
String hourseTypeJsonPath = String.format(UploadFilePath.USER_VIEW_PATH, num) + "houseType.json";
fYunFileService.uploadFile(json.toJSONString().getBytes(), hourseTypeJsonPath);
hourseTypeJsonPath = String.format(UploadFilePath.USER_EDIT_PATH, num) + "houseType.json";
fYunFileService.uploadFile(json.toJSONString().getBytes(), hourseTypeJsonPath);
return true;
}
}