package com.moral.api.pojo.vo.cityAQI;

import com.moral.api.pojo.dto.cityAQI.MonthlyPollutionLevel;
import com.moral.api.pojo.dto.cityAQI.PollutionDaysAndProportion;
import lombok.Data;
import org.springframework.util.ObjectUtils;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @ClassName BarChartOfPollutionLevelDTO
 * @Description 柱状图数据DTO
 * @Author 陈凯裕
 * @Date 2021/11/19 8:31
 * @Version TODO
 **/
@Data
public class BarChartOfPollutionLevelVO {
    /*时间集合*/
    List<String> time;

    /*污染信息，一共有5个Map，分别对应五个污染等级
     * Map中有三个字段
     * pollution表示污染等级
     * days存放天数集合
     * proportion存放天数占比的集合*/
    List<Map<String, Object>> values;

    public static BarChartOfPollutionLevelVO convert(List<MonthlyPollutionLevel> datas) {
        if(ObjectUtils.isEmpty(datas))
            return null;
        BarChartOfPollutionLevelVO vo = new BarChartOfPollutionLevelVO();
        List<String> time = new ArrayList<>();
        List<Map<String, Object>> values= new ArrayList<>();
        //6个污染等级循环获取数据
        for (int i = 0; i <= 5; i++) {
            Map<String,Object> map = new HashMap<>();
            List<String> proportion = new ArrayList<>();
            List<Integer> days = new ArrayList<>();
            String pollution = String.valueOf(i);
            for (MonthlyPollutionLevel data : datas) {
                if(!time.contains(data.getTime()))
                    time.add(data.getTime());
                List<PollutionDaysAndProportion> pollutionDaysAndProportions = data.getPollutionDaysAndProportions();
                for (PollutionDaysAndProportion pollutionDaysAndProportion : pollutionDaysAndProportions) {
                    if(pollutionDaysAndProportion.getPollutionLevel().equals(pollution)){
                        proportion.add(pollutionDaysAndProportion.getProportion());
                        days.add(pollutionDaysAndProportion.getDays());
                    }
                }
            }
            map.put("proportion",proportion);
            map.put("days",days);
            map.put("pollution",pollution);
            values.add(map);
        }
        vo.setTime(time);
        vo.setValues(values);
        return vo;
    }
}