package com.moral.api.util; 
 | 
  
 | 
import lombok.extern.slf4j.Slf4j; 
 | 
import org.springframework.stereotype.Component; 
 | 
import org.springframework.util.ObjectUtils; 
 | 
import org.springframework.util.StringUtils; 
 | 
  
 | 
import java.math.BigDecimal; 
 | 
import java.util.Date; 
 | 
import java.util.HashMap; 
 | 
import java.util.List; 
 | 
import java.util.Map; 
 | 
import java.util.Optional; 
 | 
import java.util.Set; 
 | 
  
 | 
import com.alibaba.fastjson.JSON; 
 | 
import com.googlecode.aviator.AviatorEvaluator; 
 | 
import com.googlecode.aviator.Expression; 
 | 
import com.moral.api.entity.DeviceAdjustValue; 
 | 
import com.moral.constant.Constants; 
 | 
import com.moral.util.DateUtils; 
 | 
  
 | 
@Slf4j 
 | 
@Component 
 | 
public class AdjustDataUtils { 
 | 
    /** 
 | 
     * @param deviceData    设备数据 
 | 
     * @param adjustFormula 校准公式,包含所有因子校准公式,key为因子code,value为校准公式(各个时间段的公式List) 
 | 
     * @param aqiMap        与设备绑定的国控站点aqi数据 
 | 
     * @return Map<String, Object> 校准后数据 
 | 
     */ 
 | 
    public Map<String, Object> adjust(Map<String, Object> deviceData, Map<String, Object> adjustFormula, Map<String, Object> aqiMap) { 
 | 
        try { 
 | 
            Date time = DateUtils.getDate((String) deviceData.remove("DataTime"), DateUtils.yyyyMMddHHmmss_EN); 
 | 
            long finalTime = DateUtils.dataToTimeStampTime(time, DateUtils.HH_mm_ss_EN).getTime(); 
 | 
  
 | 
            //先校准烟气流速,再用流速校准烟气流量 
 | 
            if (deviceData.get(Constants.SENSOR_CODE_CURRENT_SPEED) != null) { 
 | 
                deviceData = adjustFlueSpeedAndFlow(deviceData, adjustFormula); 
 | 
            } 
 | 
  
 | 
            //硬件没有的传感器,但是需要显示的 
 | 
            Set<String> dateKey = deviceData.keySet(); 
 | 
            for (String adjustKey : adjustFormula.keySet()) { 
 | 
                if (!dateKey.contains(adjustKey)) { 
 | 
                    deviceData.put(adjustKey, "0"); 
 | 
                } 
 | 
            } 
 | 
  
 | 
            for (String key : deviceData.keySet()) { 
 | 
                if (!key.contains("Flag")) { 
 | 
                    //测量值 
 | 
                    Object measuredValue = deviceData.get(key); 
 | 
                    //单个因子校准公式 
 | 
                    List<DeviceAdjustValue> sensorFormulas = (List<DeviceAdjustValue>) adjustFormula.get(key); 
 | 
                    if (ObjectUtils.isEmpty(sensorFormulas)) { 
 | 
                        deviceData.put(key, measuredValue); 
 | 
                        continue; 
 | 
                    } 
 | 
                    //根据时间段筛选校准公式 
 | 
                    DeviceAdjustValue deviceAdjustValue = new DeviceAdjustValue(); 
 | 
                    Optional<DeviceAdjustValue> optional = sensorFormulas.stream() 
 | 
                            .filter(o -> o.getStartTime().getTime() <= finalTime && o.getEndTime().getTime() > finalTime) 
 | 
                            .findFirst(); 
 | 
                    if (optional.isPresent()) { 
 | 
                        deviceAdjustValue = optional.get(); 
 | 
                    } 
 | 
                    String formula = deviceAdjustValue.getValue(); 
 | 
                    if (StringUtils.isEmpty(formula)) { 
 | 
                        deviceData.put(key, measuredValue); 
 | 
                        continue; 
 | 
                    } 
 | 
  
 | 
                    Expression expression = AviatorEvaluator.compile(formula); 
 | 
                    Map<String, Object> env = new HashMap<>(); 
 | 
                    if (formula.contains("aqi")) { 
 | 
                        Object aqiValue = null; 
 | 
                        if (aqiMap != null) { 
 | 
                            aqiValue = aqiMap.get(key); 
 | 
                        } 
 | 
                        env.put("aqi", ObjectUtils.isEmpty(aqiValue) ? 0F : new BigDecimal(String.valueOf(aqiValue)).floatValue()); 
 | 
                    } 
 | 
                    if (formula.contains("vocs")) { 
 | 
                        Object vocsValue = ObjectUtils.isEmpty(deviceData.get(Constants.SENSOR_CODE_VOCS)) ? 0F : deviceData.get(Constants.SENSOR_CODE_VOCS); 
 | 
                        env.put("vocs", vocsValue); 
 | 
                    } 
 | 
                    if (formula.contains("cel")) { 
 | 
                        //env.put("cel", Float.parseFloat((String) measuredValue)); 
 | 
                        env.put("cel", new BigDecimal(String.valueOf(measuredValue)).floatValue()); 
 | 
                    } 
 | 
                    //校准 
 | 
                    measuredValue = expression.execute(env); 
 | 
                    //温度处理 
 | 
                    if (!Constants.SENSOR_CODE_TEMP.equals(measuredValue) && Float.parseFloat(measuredValue.toString()) < 0) { 
 | 
                        measuredValue = 0F; 
 | 
                    } 
 | 
                    deviceData.put(key, Double.parseDouble(String.format("%.4f", measuredValue))); 
 | 
                } 
 | 
            } 
 | 
        } catch (Exception e) { 
 | 
            log.error("param[0] deviceData:" + JSON.toJSONString(deviceData)); 
 | 
        } 
 | 
        return deviceData; 
 | 
    } 
 | 
  
 | 
    /** 
 | 
     * 烟道流速,烟道流量校准 
 | 
     * 
 | 
     * @param deviceData    数据,其中包含各传感器值,和数据时间(用于不同时间段不同校准公式校准) 
 | 
     * @param adjustFormula 该数据所属设备所有因子的校准公式,key为因子code 
 | 
     */ 
 | 
    public Map<String, Object> adjustFlueSpeedAndFlow(Map<String, Object> deviceData, Map<String, Object> adjustFormula) { 
 | 
        Date time = DateUtils.getDate((String) deviceData.remove("DataTime"), DateUtils.yyyyMMddHHmmss_EN); 
 | 
        long finalTime = DateUtils.dataToTimeStampTime(time, DateUtils.HH_mm_ss_EN).getTime(); 
 | 
        //流速测量值 
 | 
        Object measuredValue = deviceData.get(Constants.SENSOR_CODE_CURRENT_SPEED); 
 | 
  
 | 
  
 | 
        //流速校准 
 | 
        List<DeviceAdjustValue> speedFormulas = (List<DeviceAdjustValue>) adjustFormula.get(Constants.SENSOR_CODE_CURRENT_SPEED); 
 | 
        if (!ObjectUtils.isEmpty(speedFormulas)) { 
 | 
            //根据时间段筛选校准公式 
 | 
            DeviceAdjustValue deviceAdjustValue = new DeviceAdjustValue(); 
 | 
            Optional<DeviceAdjustValue> optional = speedFormulas.stream() 
 | 
                    .filter(o -> o.getStartTime().getTime() <= finalTime && o.getEndTime().getTime() > finalTime) 
 | 
                    .findFirst(); 
 | 
            if (optional.isPresent()) { 
 | 
                deviceAdjustValue = optional.get(); 
 | 
            } 
 | 
  
 | 
            String formula = deviceAdjustValue.getValue(); 
 | 
            if (!StringUtils.isEmpty(formula)) { 
 | 
  
 | 
                Expression expression = AviatorEvaluator.compile(formula); 
 | 
                Map<String, Object> env = new HashMap<>(); 
 | 
                env.put("cel", Float.parseFloat((String) measuredValue)); 
 | 
                //校准 
 | 
                measuredValue = expression.execute(env); 
 | 
                deviceData.put(Constants.SENSOR_CODE_CURRENT_SPEED, Double.parseDouble(String.format("%.4f", measuredValue))); 
 | 
            } 
 | 
  
 | 
  
 | 
            //流量校准 
 | 
            List<DeviceAdjustValue> flowFormulas = (List<DeviceAdjustValue>) adjustFormula.get(Constants.SENSOR_CODE_CURRENT_FLOW); 
 | 
            //根据时间段筛选校准公式 
 | 
            optional = flowFormulas.stream() 
 | 
                    .filter(o -> o.getStartTime().getTime() <= finalTime && o.getEndTime().getTime() > finalTime) 
 | 
                    .findFirst(); 
 | 
            if (optional.isPresent()) { 
 | 
                deviceAdjustValue = optional.get(); 
 | 
            } 
 | 
  
 | 
            formula = deviceAdjustValue.getValue(); 
 | 
            if (!StringUtils.isEmpty(formula)) { 
 | 
                Expression expression = AviatorEvaluator.compile(formula); 
 | 
                Map<String, Object> env = new HashMap<>(); 
 | 
                String currentSpeed = deviceData.get(Constants.SENSOR_CODE_CURRENT_SPEED).toString(); 
 | 
                env.put("currentSpeed", Float.parseFloat(currentSpeed)); 
 | 
                deviceData.put(Constants.SENSOR_CODE_CURRENT_FLOW, Double.parseDouble(String.format("%.4f", expression.execute(env)))); 
 | 
            } 
 | 
        } 
 | 
        return deviceData; 
 | 
    } 
 | 
} 
 |