iailab-plat.git

dengzedong

2024-12-24 c9dd12a0042b50af36edf2ae710d33430542385d

提交 \| 用户 \| 时间
7fd198	1	package com.iailab.module.model.mdk.sample;
潘	2
91343d	3	import com.iailab.module.data.api.ind.IndItemApi;
潘	4	import com.iailab.module.data.api.ind.dto.ApiIndItemDTO;
e78121	5	import com.iailab.module.data.api.plan.PlanItemApi;
潘	6	import com.iailab.module.data.api.plan.dto.ApiPlanItemDTO;
7fd198	7	import com.iailab.module.data.api.point.DataPointApi;
潘	8	import com.iailab.module.data.api.point.dto.ApiPointDTO;
e78121	9	import com.iailab.module.data.enums.DataPointFreqEnum;
潘	10	import com.iailab.module.data.enums.TimeGranularityEnum;
91343d	11	import com.iailab.module.data.enums.TimeGranularitySecEnum;
潘	12	import com.iailab.module.model.mcs.pre.service.MmPredictItemService;
e78121	13	import com.iailab.module.model.mdk.common.enums.ModelParamType;
7fd198	14	import com.iailab.module.model.mdk.sample.dto.ColumnItem;
潘	15	import com.iailab.module.model.mdk.sample.dto.ColumnItemPort;
	16	import com.iailab.module.model.mdk.sample.dto.SampleInfo;
	17	import org.springframework.beans.factory.annotation.Autowired;
	18
	19	import java.util.Calendar;
	20	import java.util.Date;
	21	import java.util.List;
07890e	22	import java.util.Map;
7fd198	23
潘	24	/**
	25	* @author PanZhibao
	26	* @Description
	27	* @createTime 2024年09月03日
	28	*/
	29	abstract class SampleInfoConstructor {
	30
	31	@Autowired
	32	private DataPointApi dataPointApi;
	33
	34	@Autowired
91343d	35	private PlanItemApi planItemApi;
e78121	36
潘	37	@Autowired
91343d	38	private IndItemApi indItemApi;
潘	39
	40	@Autowired
	41	private MmPredictItemService mmPredictItemService;
7fd198	42
潘	43	/**
	44	* prepareSampleInfo
	45	*
	46	* @param modelId
	47	* @param predictTime
	48	* @return
	49	*/
45520a	50	protected SampleInfo prepareSampleInfo(String modelId, Date predictTime) {
7fd198	51	//样本的列信息
50084d	52	return getColumnInfo(modelId, predictTime);
7fd198	53	}
潘	54
	55	/**
	56	* 返回样本矩阵的列数
	57	*
	58	* @param modelId
	59	* @return
	60	*/
	61	protected abstract Integer getSampleColumn(String modelId);
45520a	62
D	63
	64	/**
	65	* 样本的列信息
	66	*
	67	* @param modelId
	68	* @param predictTime
	69	* @return
	70	*/
50084d	71	protected abstract SampleInfo getColumnInfo(String modelId, Date predictTime);
45520a	72
D	73	/**
	74	* 样本的采样周期
	75	*
	76	* @param modelId
	77	* @return
	78	*/
	79	protected abstract Integer getSampleCycle(String modelId);
7fd198	80
潘	81	/**
	82	* 获取开始时间
	83	*
	84	* @param columnItem
	85	* @param originalTime
	86	* @return
	87	* @throws Exception
	88	*/
8bf553	89	protected Date getStartTime(ColumnItem columnItem, Date originalTime, Map<String, ApiPointDTO> pointMap, Map<String, ApiPlanItemDTO> planMap) {
7fd198	90	Date dateTime = new Date();
潘	91	Calendar calendar = Calendar.getInstance();
	92	calendar.setTime(originalTime);
e78121	93	switch (ModelParamType.getEumByCode(columnItem.getParamType())) {
潘	94	case DATAPOINT:
07890e	95	ApiPointDTO dataPoint = pointMap.get(columnItem.getParamId());
7fd198	96	if (dataPoint == null) {
潘	97	return null;
	98	}
e78121	99	dateTime = calculateTime(originalTime, true, columnItem.getDataLength(), DataPointFreqEnum.getEumByCode(dataPoint.getMinfreqid()).getValue());
7fd198	100	break;
19a4fb	101	case NORMALITEM:
D	102	case MERGEITEM:
c9dd12	103	dateTime = calculateTime(originalTime, true, columnItem.getDataLength(), 60);
7fd198	104	break;
e78121	105	case IND:
7fd198	106	dateTime = calculateTime(originalTime, true, columnItem.getDataLength(), 60);
潘	107	break;
e78121	108	case PLAN:
8bf553	109	ApiPlanItemDTO apiPlanItemDTO = planMap.get(columnItem.getParamId());
D	110	if (apiPlanItemDTO == null) {
	111	return null;
	112	}
	113	dateTime = calculateTime(originalTime, true, columnItem.getDataLength(), TimeGranularitySecEnum.getEumByCode(apiPlanItemDTO.getTimeGranularity()).getValue());
7fd198	114	break;
潘	115	default:
	116	break;
	117	}
	118	return dateTime;
	119	}
	120
	121	/**
	122	* 获取结束时间
	123	*
	124	* @param columnItem
	125	* @param originalTime
	126	* @return
	127	* @throws Exception
	128	*/
8bf553	129	protected Date getEndTime(ColumnItem columnItem, Date originalTime,Map<String, ApiPointDTO> pointMap, Map<String, ApiPlanItemDTO> planMap) {
7fd198	130	Date dateTime = new Date();
潘	131	Calendar calendar = Calendar.getInstance();
	132	calendar.setTime(originalTime);
e78121	133	switch (ModelParamType.getEumByCode(columnItem.getParamType())) {
潘	134	case DATAPOINT:
07890e	135	ApiPointDTO dataPoint = pointMap.get(columnItem.getParamId());
7fd198	136	if (dataPoint == null) {
潘	137	return null;
	138	}
b2aca2	139	dateTime = calendar.getTime();
7fd198	140	break;
19a4fb	141	case NORMALITEM:
D	142	case MERGEITEM:
b2aca2	143	dateTime = calendar.getTime();
7fd198	144	break;
e78121	145	case IND:
7fd198	146	dateTime = calendar.getTime();
潘	147	break;
e78121	148	case PLAN:
8bf553	149	ApiPlanItemDTO apiPlanItemDTO = planMap.get(columnItem.getParamId());
D	150	if (apiPlanItemDTO == null) {
e78121	151	return null;
潘	152	}
8bf553	153	dateTime = calculateDate(originalTime, false, columnItem.getDataLength(), TimeGranularitySecEnum.getEumByCode(apiPlanItemDTO.getTimeGranularity()).getValue());
7fd198	154	break;
潘	155	default:
	156	break;
	157	}
	158	return dateTime;
	159	}
	160
	161	/**
	162	* 获取粒度，s
	163	*
	164	* @param columnItem
	165	* @return
	166	* @throws Exception
	167	*/
8bf553	168	protected Integer getGranularity(ColumnItem columnItem,Map<String, ApiPointDTO> pointMap, Map<String, ApiPlanItemDTO> planMap) {
91343d	169	// 默认60s
7fd198	170	Integer granularity = 60;
91343d	171	switch (ModelParamType.getEumByCode(columnItem.getParamType())) {
潘	172	case DATAPOINT:
8bf553	173	ApiPointDTO dataPoint = pointMap.get(columnItem.getParamId());
e78121	174	granularity = DataPointFreqEnum.getEumByCode(dataPoint.getMinfreqid()).getValue();
7fd198	175	break;
19a4fb	176	case NORMALITEM:
D	177	case MERGEITEM:
50084d	178	granularity = mmPredictItemService.getItemByOutPutId(columnItem.getParamId()).getGranularity();
7fd198	179	break;
91343d	180	case IND:
潘	181	ApiIndItemDTO indItemDTO = indItemApi.getInfoById(columnItem.getParamId());
	182	granularity = TimeGranularitySecEnum.getEumByCode(indItemDTO.getTimeGranularity()).getValue();;
7fd198	183	break;
91343d	184	case PLAN:
8bf553	185	ApiPlanItemDTO apiPlanItemDTO = planMap.get(columnItem.getParamId());
91343d	186	granularity = TimeGranularitySecEnum.getEumByCode(apiPlanItemDTO.getTimeGranularity()).getValue();
7fd198	187	break;
潘	188	default:
	189	break;
	190	}
	191	return granularity;
	192	}
	193
	194	/**
	195	* 计算取值的时间
	196	*
	197	* @param originalTime
	198	* @param backward
	199	* @param dataLength
	200	* @param granularity
	201	* @return
	202	*/
	203	public Date calculateTime(Date originalTime, Boolean backward, int dataLength, int granularity) {
	204	int timeLength;
	205	if (backward) {
	206	timeLength = (-1) * dataLength;
	207	} else {
	208	timeLength = dataLength - 1;
	209	}
	210	Date desTime = originalTime;
	211	Calendar calendar = Calendar.getInstance();
	212	calendar.setTime(desTime);
	213	calendar.set(Calendar.MILLISECOND, 0);
	214	// 数据长度 * 粒度
	215	calendar.add(Calendar.SECOND, timeLength * granularity);
	216	return calendar.getTime();
	217	}
e78121	218
潘	219	public Date calculateDate(Date originalTime, Boolean backward, int dataLength, int field) {
	220	int timeLength;
	221	if (backward) {
	222	timeLength = (-1) * dataLength;
	223	} else {
	224	timeLength = dataLength - 1;
	225	}
	226	Date desTime = originalTime;
	227	Calendar calendar = Calendar.getInstance();
	228	calendar.setTime(desTime);
	229	calendar.set(Calendar.MILLISECOND, 0);
	230	// 数据长度 * 粒度
	231	calendar.add(Calendar.SECOND, timeLength);
	232	return calendar.getTime();
	233	}
7fd198	234	}