4dkankan-center-modeling-control/src/main/java/com/fdkankan/contro/service/impl/InnerServiceImpl.java

package com.fdkankan.contro.service.impl;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.http.ContentType;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.fdkankan.common.constant.*;
import com.fdkankan.common.exception.BusinessException;
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.service.*;
import com.fdkankan.fyun.config.FYunFileConfig;
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.SceneUtil;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.http.HttpHeaders;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.io.File;
import java.io.IOException;
import java.util.*;

@Service
public class InnerServiceImpl implements IInnerService {

    @Value("${model.type:#{null}}")
    private String modelType;

    @Autowired
    private IScenePlusService scenePlusService;
    @Autowired
    private IScenePlusExtService scenePlusExtService;
    @Autowired
    private ICommonService commonService;
    @Resource
    private FYunFileServiceInterface fYunFileService;
    @Autowired
    private IFdkkLaserService fdkkLaserService;
    @Autowired
    private ICameraDetailService cameraDetailService;
    @Autowired
    private ICompanyService companyService;
    @Resource
    private FYunFileConfig fYunFileConfig;

    @Override
    public void uploadArtificialResult(String num) throws Exception {

        ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
        if(Objects.isNull(scenePlus)){
            throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
        }
        ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());

        String path = scenePlusExt.getDataSource();

        JSONObject fdageData = commonService.getFdageData(path + File.separator + "capture" +File.separator+"data.fdage");

        Integer cameraType = 10;
        if(scenePlus.getSceneSource() == SceneSource.ZT.code()){
            cameraType = 13;
        }else{
            cameraType = 14;
        }

        Map<String, String> 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, num, cameraType);

        boolean isObj = fdageData.containsKey("exportMeshObj") && fdageData.getIntValue("exportMeshObj") == 1;

        //上传全景图俯视图
        this.uploadFloorCad(path, num, uploadFiles);

        //由于3dtiles算法mesh文件发生变化，所以这里需要先清除一下oss的mesh目录，避免存在旧算法obj文件
        fYunFileService.deleteFolder(String.format(UploadFilePath.DATA_VIEW_PATH, num) + "mesh");
        fYunFileService.deleteFolder(String.format(UploadFilePath.IMG_VIEW_PATH,  num) + ModelKind.THREE_D_TILE.code());
        //上传文件
        fYunFileService.uploadMulFiles(uploadFiles);

        //修改oss上dam的内容编码
        Map<String,String> 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, num) + ConstantFileName.modelUUID + "_50k.dam", damFileHeaders);

        //拷贝部分文件到编辑目录，用于用户编辑
        this.copyToEditDir(num);

        //计算完毕后，同步全景图到缓存目录
//            this.cachePanorama(path, sceneCode);

        //生成houseTypejson并上传
        boolean existHouseType = this.uploadHouseTypeJson(num, path);
        scenePlus.setHouseType(existHouseType ? CommonStatus.YES.code().intValue() : CommonStatus.NO.code().intValue());

        //生成floorpan.json
        commonService.uploadFloorplanJson(num, path);

        //重置异步操作记录
        commonService.removeSceneAsynOperLog(num);

        //清除用户编辑业务数据
        Set<String> bizs = new HashSet<>();
        bizs.add(UserEditDataType.BOX_MODEL.message());
        bizs.add(UserEditDataType.FLOORPLAN.message());
        bizs.add(UserEditDataType.FILTERS.message());
        commonService.initUserEditData(num, bizs, null);

        //上传计算结果文件
        commonService.uploadBuildResultData(num, path, SceneVersionType.V4.code());

        //容量统计
        Long space = commonService.getSpace(num);

        //写入数据库
        this.updateDbPlus(scenePlus.getSceneSource(), space, videosJson.toJSONString(), 0L,isObj,scenePlusExt);

        Object[] editInfoArr = commonService.updateEditInfo(scenePlus);
        SceneEditInfo sceneEditInfo = (SceneEditInfo)editInfoArr[0];
        SceneEditInfoExt sceneEditInfoExt = (SceneEditInfoExt)editInfoArr[1];
        SceneEditControls sceneEditControls = (SceneEditControls)editInfoArr[2];

        //更新场景主表
        //如果相机容量不足，需要把场景的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(num).dataSource(path)
                            .sceneStatus(2)
                            .createTime(scenePlus.getCreateTime())
                            .shootCount(scenePlusExt.getShootCount())
                            .payStatus(scenePlus.getPayStatus())
                            .mixture(scenePlusExt.getMixture())
                            .version(SceneVersionType.V4.code()).build());
            sceneEditControls.setShowMap(0);
        } else if (new File(path + "/results/laserData/vision_edit.txt").exists()) {
            fdkkLaserService.cloudPointBuild(num,path);
        }


        CameraDetail cameraDetail = cameraDetailService.getByCameraId(scenePlus.getCameraId());
        Company company = !ObjectUtils.isEmpty(cameraDetail.getCompanyId()) ? companyService.getById(cameraDetail.getCompanyId()) : null;
        //写scene.json
        commonService.writeSceneJson(num,sceneEditInfo, sceneEditInfoExt, sceneEditControls, scenePlus,scenePlusExt,company);

        this.uploadStatusJson(scenePlus, scenePlusExt);

        scenePlusService.updateById(scenePlus);
        scenePlusExtService.updateById(scenePlusExt);
    }

    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<String,String> 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 uploadFloorCad(String path, String num, Map<String, String> uploadFiles){

        //户型图上传
        String  dataViewPath = UploadFilePath.DATA_VIEW_PATH + "floor-cad-%s.%s";
        String floorCadPath = path + "/results/floorplan_cad";
        List<String> 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 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 (Map.Entry<String, String> entry : map.entrySet()) {
            fYunFileService.copyFileInBucket(entry.getValue(), entry.getKey());
        }
    }

    public boolean uploadHouseTypeJson(String num, String dataSource) {
        String floorPlanCardFilePath = dataSource + File.separator + "results/floorplan_cad.json";
        if (!new File(floorPlanCardFilePath).exists()) {
            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;
    }

    private void updateDbPlus(int sceneSource,Long space,String videosJson, Long computeTime,boolean isObj,ScenePlusExt scenePlusExt){

        scenePlusExt.setSpace(space);
        scenePlusExt.setComputeTime(computeTime.toString());
        scenePlusExt.setAlgorithmTime(new Date());
        scenePlusExt.setVideos(videosJson);
        scenePlusExt.setIsObj(isObj ? 1 : 0);

        if(ModelTypeEnums.TILE_CODE.equals(modelType)){
            scenePlusExt.setSceneScheme(3);
        }

        switch (SceneSource.get(sceneSource)){
            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);

        //统计点位数量
        Map<String, Integer> result = this.getShootCount(scenePlusExt);
        Integer shootCount = result.get("shootCount");
        Integer mixture = result.get("mixture");
        scenePlusExt.setShootCount(shootCount);
        scenePlusExt.setMixture(mixture);

        scenePlusExtService.updateById(scenePlusExt);
    }

    private Map<String, Integer> getShootCount(ScenePlusExt scenePlusExt){

        Map<String, Integer> 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;
    }
}