package com.iailab.module.model.mdk.predict.impl;
|
|
import com.alibaba.fastjson.JSON;
|
import com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity;
|
import com.iailab.module.model.mcs.pre.service.MmItemOutputService;
|
import com.iailab.module.model.mcs.pre.service.MmItemResultJsonService;
|
import com.iailab.module.model.mcs.sche.service.StAdjustResultService;
|
import com.iailab.module.model.mdk.common.exceptions.ItemInvokeException;
|
import com.iailab.module.model.mdk.factory.ItemEntityFactory;
|
import com.iailab.module.model.mdk.predict.PredictItemHandler;
|
import com.iailab.module.model.mdk.vo.ItemVO;
|
import com.iailab.module.model.mdk.vo.PredictResultVO;
|
import com.iailab.module.model.mdk.vo.StAdjustDeviationDTO;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.CollectionUtils;
|
|
import java.util.*;
|
|
/**
|
* @author PanZhibao
|
* @Description
|
* @createTime 2024年09月01日
|
*/
|
@Slf4j
|
@Component
|
public class PredictItemMergeHandlerImpl implements PredictItemHandler {
|
|
@Autowired
|
private ItemEntityFactory itemEntityFactory;
|
|
@Autowired
|
private MmItemResultJsonService mmItemResultJsonService;
|
|
@Autowired
|
private MmItemOutputService mmItemOutputService;
|
|
@Autowired
|
private StAdjustResultService stAdjustResultService;
|
|
/**
|
* 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();
|
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));
|
}
|
}
|
//是否为计算预测项
|
if (mathOutPutId.length > 1) {
|
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;
|
}
|
}
|
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);
|
} catch (Exception e) {
|
log.error("merge项预测失败，itemId：" + itemId);
|
e.printStackTrace();
|
throw e;
|
}
|
log.info("merge项预测完成，itemId：" + itemId + ",结果：" + JSON.toJSONString(predictResult));
|
return predictResult;
|
}
|
|
@Override
|
public PredictResultVO predictAdjust(Date predictTime, ItemVO predictItemDto, List<StAdjustDeviationDTO> deviationList)
|
throws ItemInvokeException {
|
PredictResultVO predictResult = new PredictResultVO();
|
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));
|
}
|
}
|
Map<String, double[]> predictValueMap = new HashMap<>();
|
for (int k = 0; k < mathOutPutId.length; k++) {
|
String outPutId = mathOutPutId[k];
|
double[] outPutValue = stAdjustResultService.getSimpleData(outPutId, predictTime, predictLength);
|
if (outPutValue == null) {
|
outPutValue = mmItemResultJsonService.getSimpleData(outPutId, predictTime, predictLength);
|
}
|
predictValueMap.put(outPutId, outPutValue);
|
}
|
|
//是否为计算预测项
|
if (mathOutPutId.length > 1) {
|
for (int i = 0; i < predictLength; i++) {
|
double 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;
|
}
|
}
|
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);
|
} catch (Exception e) {
|
log.error("merge项预测失败，itemId：" + itemId);
|
e.printStackTrace();
|
throw e;
|
}
|
log.info("merge项预测完成，itemId：" + itemId + ",结果：" + JSON.toJSONString(predictResult));
|
return predictResult;
|
}
|
}
|
