package com.gunshi.project.hsz.service;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gunshi.project.hsz.entity.vo.RsvrComplexVo;
import com.gunshi.project.hsz.mapper.StZqrlBMapper;
import com.gunshi.project.hsz.model.StRsvrR;
import com.gunshi.project.hsz.model.StZqrlB;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 描述: 水位流量关系曲线表
 * author: xusan
 * date: 2024-07-08 17:30:38
 */
@Service
@Slf4j
@Transactional(rollbackFor = Exception.class)
public class StZqrlBService extends ServiceImpl<StZqrlBMapper, StZqrlB> {

    /**
     * 计算溢洪量
     *
     * @param rz       计算溢洪的起始水位
     * @param seconds  溢洪时间，是两条水位数据的时间差，单位秒
     * @param zqrlList 水位流量关系曲线
     * @return 溢洪量
     * @author lyf
     */
    public BigDecimal calculateSpilledVolume(BigDecimal rz, long seconds, List<StZqrlB> zqrlList) {
        return getQFromZqrl(rz, zqrlList).multiply(BigDecimal.valueOf(seconds));
    }

    /**
     * 计算溢洪量
     *
     * @param rz      计算溢洪的起始水位
     * @param seconds 溢洪时间，是两条水位数据的时间差，单位秒
     * @return 溢洪量
     * @author lyf
     */
    public BigDecimal calculateSpilledVolume(BigDecimal rz, long seconds) {
        return calculateSpilledVolume(rz, seconds, list());
    }

    /**
     * 批量计算溢洪量
     *
     * @param rsvrList 要计算的水位数据
     * @param zqrlList 水位流量关系曲线
     * @return 计算后的水位数据（时间倒序），含每条水位对应的流量和与上一条水位对应的溢洪量
     * @author lyf
     */
    public List<RsvrComplexVo> calculateSpilledVolumeList(List<StRsvrR> rsvrList, List<StZqrlB> zqrlList) {
        rsvrList.sort(Comparator.comparing(StRsvrR::getTm).reversed());
        List<RsvrComplexVo> ret = new ArrayList<>();
        for (int i = 0; i < rsvrList.size(); i++) {
            StRsvrR rsvr = rsvrList.get(i);
            BigDecimal rz = new BigDecimal(rsvr.getRz());
            Date tm = rsvr.getTm();
            RsvrComplexVo vo = RsvrComplexVo.of(rz, tm);

            if (i == 0) {
                vo.setQ(getQFromZqrl(rz, zqrlList));
                vo.setSv(BigDecimal.ZERO);
            } else {
                StRsvrR rsvrPrev = rsvrList.get(i - 1);
                Date tmPrev = rsvrPrev.getTm();
                long seconds = tm.getTime() - tmPrev.getTime();
                vo.setQ(getQFromZqrl(rz, zqrlList));
                vo.setSv(calculateSpilledVolume(rz, seconds));
            }
            ret.add(vo);
        }
        return ret;
    }

    /**
     * 批量计算溢洪量
     *
     * @param rsvrList 要计算的水位数据
     * @return 计算后的水位数据（时间倒序），含每条水位对应的流量和与上一条水位对应的溢洪量
     * @author lyf
     */
    public List<RsvrComplexVo> calculateSpilledVolumeList(List<StRsvrR> rsvrList) {
        return calculateSpilledVolumeList(rsvrList, list());
    }

    /**
     * 根据水位查表得流量
     *
     * @param rz 水位
     * @return 查表流量
     * @author lyf
     */
    public BigDecimal getQFromZqrl(BigDecimal rz) {
        return getQFromZqrl(rz, list());
    }

    /**
     * 根据水位查表得流量
     *
     * @param rz       水位
     * @param zqrlList 水位流量关系曲线
     * @return 查表流量
     * @author lyf
     */
    public BigDecimal getQFromZqrl(BigDecimal rz, List<StZqrlB> zqrlList) {
        return toMap(zqrlList).getOrDefault(rz.setScale(3, RoundingMode.DOWN), BigDecimal.ZERO);
    }

    /**
     * 将水位流量曲线列表转为字典，水位作为键，注意BigDecimal做键的时候要匹配值和精度
     *
     * @param zqrlList 水位流量曲线列表
     * @return 水位流量曲线字典
     * @author lyf
     */
    private Map<BigDecimal, BigDecimal> toMap(List<StZqrlB> zqrlList) {
        return zqrlList.stream()
                .map(item -> Map.entry(item.getZ().setScale(3, RoundingMode.DOWN), item.getQ()))
                .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
    }

}