潘志宝
2024-12-23 5bf42aa9950058f391805e6fb8d7376f4378924b
提交 | 用户 | 时间
7fd198 1 package com.iailab.module.model.mdk.sample;
2
8bf553 3 import com.iailab.module.data.api.plan.PlanItemApi;
D 4 import com.iailab.module.data.api.plan.dto.ApiPlanItemDTO;
07890e 5 import com.iailab.module.data.api.point.DataPointApi;
D 6 import com.iailab.module.data.api.point.dto.ApiPointDTO;
7fd198 7 import com.iailab.module.model.mcs.pre.entity.MmModelParamEntity;
91343d 8 import com.iailab.module.model.mcs.pre.service.MmModelParamService;
9 import com.iailab.module.model.mcs.pre.service.MmPredictItemService;
7fd198 10 import com.iailab.module.model.mcs.pre.service.MmPredictModelService;
07890e 11 import com.iailab.module.model.mdk.common.enums.ModelParamType;
7fd198 12 import com.iailab.module.model.mdk.sample.dto.ColumnItem;
13 import com.iailab.module.model.mdk.sample.dto.ColumnItemPort;
50084d 14 import com.iailab.module.model.mdk.sample.dto.SampleInfo;
7fd198 15 import org.springframework.beans.factory.annotation.Autowired;
16 import org.springframework.stereotype.Component;
17 import org.springframework.util.CollectionUtils;
18
50084d 19 import java.util.*;
07890e 20 import java.util.function.Function;
D 21 import java.util.stream.Collectors;
7fd198 22
23 /**
24  * @author PanZhibao
25  * @Description
26  * @createTime 2024年09月03日
27  */
28 @Component
29 public class PredictSampleInfoConstructor extends SampleInfoConstructor {
30
31     @Autowired
32     private MmPredictModelService mmPredictModelService;
33
34     @Autowired
91343d 35     private MmModelParamService mmModelParamService;
7fd198 36
37     @Autowired
91343d 38     private MmPredictItemService mmPredictItemService;
07890e 39
D 40     @Autowired
41     private DataPointApi dataPointApi;
7fd198 42
8bf553 43     @Autowired
D 44     private PlanItemApi planItemApi;
45
7fd198 46     /**
47      * 返回样本矩阵的列数
48      *
49      * @param modelId
50      * @return
51      */
52     @Override
53     protected Integer getSampleColumn(String modelId) {
54         return mmPredictModelService.getSampleLength(modelId).intValue();
55     }
56
57     /**
58      * 样本的列信息
59      *
60      * @param modelId
61      * @param predictTime
62      * @return
63      */
64     @Override
50084d 65     protected SampleInfo getColumnInfo(String modelId, Date predictTime) {
D 66         SampleInfo sampleInfo = new SampleInfo();
7fd198 67         List<ColumnItemPort> resultList = new ArrayList<>();
68         List<ColumnItem> columnItemList = new ArrayList<>();
69         ColumnItem columnInfo = new ColumnItem();
70         ColumnItemPort curPort = new ColumnItemPort();  //当前端口
45520a 71         List<MmModelParamEntity> modelInputParamEntityList = mmModelParamService.getByModelidFromCache(modelId);
7fd198 72         if (CollectionUtils.isEmpty(modelInputParamEntityList)) {
73             return null;
74         }
75         //设置当前端口号,初始值为最小端口(查询结果按端口号从小到达排列)
76         int curPortOrder = modelInputParamEntityList.get(0).getModelparamportorder();
77         //设置当前查询数据长度,初始值为最小端口数据长度
78         int curDataLength = modelInputParamEntityList.get(0).getDatalength();
07890e 79         // 统一获取测点的信息
50084d 80         Set<String> pointIds = modelInputParamEntityList.stream().filter(e -> ModelParamType.getEumByCode(e.getModelparamtype()).equals(ModelParamType.DATAPOINT)).map(MmModelParamEntity::getModelparamid).collect(Collectors.toSet());
07890e 81         List<ApiPointDTO> points = dataPointApi.getInfoByIds(pointIds);
50084d 82         Map<String, ApiPointDTO> pointMap = points.stream().collect(Collectors.toMap(ApiPointDTO::getId, Function.identity(), (e1,e2) -> e1));
8bf553 83         // 统一获取计划数据的信息
D 84         Set<String> planIds = modelInputParamEntityList.stream().filter(e -> ModelParamType.getEumByCode(e.getModelparamtype()).equals(ModelParamType.PLAN)).map(MmModelParamEntity::getModelparamid).collect(Collectors.toSet());
85         List<ApiPlanItemDTO> plans = planItemApi.getInfoByIds(planIds);
86         Map<String, ApiPlanItemDTO> planMap = plans.stream().collect(Collectors.toMap(ApiPlanItemDTO::getId, Function.identity(), (e1,e2) -> e1));
07890e 87
7fd198 88         for (MmModelParamEntity entry : modelInputParamEntityList) {
89             columnInfo.setParamType(entry.getModelparamtype());
1a2b62 90             columnInfo.setParamId(entry.getModelparamid());
7fd198 91             columnInfo.setDataLength(entry.getDatalength());
92             columnInfo.setModelParamOrder(entry.getModelparamorder());
93             columnInfo.setModelParamPortOrder(entry.getModelparamportorder());
8bf553 94             columnInfo.setStartTime(getStartTime(columnInfo, predictTime,pointMap,planMap));
D 95             columnInfo.setEndTime(getEndTime(columnInfo, predictTime,pointMap,planMap));
96             columnInfo.setGranularity(super.getGranularity(columnInfo,pointMap,planMap));
7fd198 97
98             //对每一个爪进行数据项归并
99             if (curPortOrder != entry.getModelparamportorder()){
100                 //当数据项端口号不为当前端口号时,封装上一个端口类,操作下一个端口类
101                 curPort.setColumnItemList(columnItemList);
102                 curPort.setDataLength(curDataLength);
103                 curPort.setPortOrder(curPortOrder);
104                 resultList.add(curPort);
105                 curPort = new ColumnItemPort(); //对象重新初始化,防止引用拷贝导致数据覆盖
106                 //封装上一个端口类后更新当前的各个参数
107                 columnItemList = new ArrayList<>();
108                 curDataLength = entry.getDatalength();
109                 curPortOrder = entry.getModelparamportorder();
110             }
111             columnItemList.add(columnInfo);
112             columnInfo = new ColumnItem();    //对象重新初始化,防止引用拷贝导致数据覆盖
113         }
114         //当迭代到最后一个项的时候,封装最后一个端口的信息
115         curPort.setColumnItemList(columnItemList);
116         curPort.setDataLength(curDataLength);
117         curPort.setPortOrder(curPortOrder);
118         resultList.add(curPort);
50084d 119         sampleInfo.setColumnInfo(resultList);
D 120         sampleInfo.setPointMap(pointMap);
8bf553 121         sampleInfo.setPlanMap(planMap);
50084d 122         return sampleInfo;
7fd198 123     }
124
125     /**
126      * 样本的采样周期
127      *
128      * @param modelId
129      * @return
130      */
131     @Override
132     protected Integer getSampleCycle(String modelId) {
b82ba2 133         return mmPredictItemService.getItemByIdFromCache(mmPredictModelService.getInfoFromCatch(modelId).getItemid()).getGranularity();
7fd198 134     }
135
136
137 }