package com.iailab.module.model.mdk.predict.impl;
|
|
import com.iailab.module.data.api.point.DataPointApi;
|
import com.iailab.module.data.api.point.dto.ApiPointDTO;
|
import com.iailab.module.data.enums.DataPointFreqEnum;
|
import com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity;
|
import com.iailab.module.model.mcs.pre.enums.ItemRunStatusEnum;
|
import com.iailab.module.model.mcs.pre.service.MmItemOutputService;
|
import com.iailab.module.model.mcs.pre.service.MmItemResultService;
|
import com.iailab.module.model.mdk.common.enums.ItemPredictStatus;
|
import com.iailab.module.model.mdk.common.exceptions.ItemInvokeException;
|
import com.iailab.module.model.mdk.factory.ItemEntityFactory;
|
import com.iailab.module.model.mdk.factory.PredictItemFactory;
|
import com.iailab.module.model.mdk.predict.PredictItemHandler;
|
import com.iailab.module.model.mdk.predict.PredictResultHandler;
|
import com.iailab.module.model.mdk.vo.DataValueVO;
|
import com.iailab.module.model.mdk.vo.ItemVO;
|
import com.iailab.module.model.mdk.vo.PredictResultVO;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.CollectionUtils;
|
|
import java.sql.Timestamp;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* @author PanZhibao
|
* @Description
|
* @createTime 2024年09月01日
|
*/
|
@Slf4j
|
@Component
|
public class PredictItemMergeHandlerImpl implements PredictItemHandler {
|
|
@Autowired
|
private ItemEntityFactory itemEntityFactory;
|
|
@Autowired
|
private DataPointApi dataPointApi;
|
|
@Autowired
|
private PredictItemFactory predictItemFactory;
|
|
@Autowired
|
private PredictResultHandler predictResultHandler;
|
|
@Autowired
|
private MmItemResultService mmItemResultService;
|
|
@Autowired
|
private MmItemOutputService mmItemOutputService;
|
|
/**
|
* MergeItem预测
|
*
|
* @param predictTime
|
* @param predictItemDto
|
* @return
|
* @throws ItemInvokeException
|
*/
|
@Override
|
public PredictResultVO predict(Date predictTime, ItemVO predictItemDto, Map<String, double[]> predictValueMap)
|
throws ItemInvokeException {
|
PredictResultVO predictResult = new PredictResultVO();
|
ItemPredictStatus itemStatus = ItemPredictStatus.PREDICTING;
|
String itemId = predictItemDto.getId();
|
try {
|
String expression = itemEntityFactory.getMergeItem(itemId).getExpression();
|
int predictLength = itemEntityFactory.getItemById(itemId).getPredictLength();
|
double[] predictResultMat = new double[predictLength];
|
String[] mathOutPutId = expression.split("[\\+ \\-]");
|
ArrayList<Character> operator = new ArrayList<>();
|
for (int i = 0; i < expression.length(); i++) {
|
if (expression.charAt(i)=='+' || expression.charAt(i)=='-'){
|
operator.add(expression.charAt(i));
|
}
|
}
|
// String[] compositionItem = expression.split(String.valueOf("&".toCharArray()));
|
//是否为计算预测项
|
if (mathOutPutId.length > 1) {
|
// Map<String, List<DataValueVO>> predictValueMap = new HashMap<>();
|
// for (String outPutId : mathOutPutId) {
|
// if (outPutId.length() > 4) {
|
// Date endTime = predictTime;
|
//// ItemVO itemEntity = itemEntityFactory.getItemByItemNo(itemNo);
|
//// List<MmItemOutputEntity> outPutList = itemEntityFactory.getOutPutByItemId(itemEntity.getId());
|
// MmItemOutputEntity outPut = mmItemOutputService.getOutPutById(outPutId);
|
// ApiPointDTO pointEntity = dataPointApi.getInfoById(outPut.getPointid());
|
//
|
// Calendar calendar = Calendar.getInstance();
|
// calendar.setTime(endTime);
|
// calendar.add(Calendar.SECOND, (predictLength - 1) * DataPointFreqEnum.getEumByCode(pointEntity.getMinfreqid()).getValue());
|
// endTime = new Timestamp(calendar.getTime().getTime());
|
//// List<DataValueVO> predictValueList = predictResultHandler.getPredictValueByItemNo(itemNo, predictTime, endTime);
|
// List<DataValueVO> predictValueList = mmItemResultService.getPredictValue(outPutId, predictTime, endTime);
|
// if (predictValueList.size() != predictLength) {
|
// log.debug("merge项融合失败：缺少子项预测数据,对应子项outPutId=" + outPutId);
|
// return null;
|
// }
|
// predictValueMap.put(outPutId, predictValueList);
|
// }
|
// }
|
|
for (Integer i = 0; i < predictLength; i++) {
|
double sum =0.0;
|
sum = predictValueMap.get(mathOutPutId[0])[i];
|
for (int j = 1; j < mathOutPutId.length; j++) {
|
if (operator.get(j-1)=='+')
|
{sum += predictValueMap.get(mathOutPutId[j])[i];}
|
if (operator.get(j-1)=='-')
|
{sum -= predictValueMap.get(mathOutPutId[j])[i];}
|
}
|
predictResultMat[i] = sum;
|
}
|
}
|
//是否为组合预测项
|
// if (compositionItem.length > 1) {
|
// Map<String, PredictResultVO> predictResultMap = new HashMap<>();
|
// Integer columnTotalNumber = 0;
|
// Integer rowNumber = 0;
|
// for (String itemNo : compositionItem) {
|
// PredictItemHandler predictItem = (PredictItemHandler) predictItemFactory.create(itemEntityFactory.
|
// getItemByItemNo(itemNo).getId());
|
// predictResult = predictItem.predict(predictTime, predictItemDto);
|
// columnTotalNumber += Integer.valueOf(predictResult.getPredictMatrix().length);
|
// predictResultMap.put(itemNo, predictItem.predict(predictTime, predictItemDto));
|
// }
|
// double[][] matrix = new double[columnTotalNumber][1];
|
// for (String itemNo : compositionItem) {
|
// for (Integer i = 0; i < predictResultMap.get(itemNo).getPredictMatrix().length; i++) {
|
// matrix[rowNumber][0] = predictResultMap.get(itemNo).getPredictMatrix()[i][0];
|
// rowNumber++;
|
// }
|
// }
|
// predictResult.setPredictMatrix(matrix);
|
// }
|
log.info("计算预测模型结果：" + predictResultMat);
|
predictResult.setPredictId(itemId);
|
List<MmItemOutputEntity> outputServiceByItemid = mmItemOutputService.getByItemid(itemId);
|
if (CollectionUtils.isEmpty(outputServiceByItemid)) {
|
Map<MmItemOutputEntity, double[]> predictMatrixs = new HashMap<>();
|
predictMatrixs.put(outputServiceByItemid.get(0),predictResultMat);
|
predictResult.setPredictMatrixs(predictMatrixs);
|
}
|
predictResult.setPredictTime(predictTime);
|
//预测项预测成功的状态
|
itemStatus = ItemPredictStatus.SUCCESS;
|
} catch (Exception e) {
|
//预测项预测失败的状态
|
itemStatus = ItemPredictStatus.FAILED;
|
log.debug("merge项预测失败，itemId：" + itemId);
|
throw e;
|
}
|
log.debug("预测完成，itemId：" + itemId + ",itemStatus：" + itemStatus.getValue());
|
return predictResult;
|
}
|
}
|
