package com.qianwen.smartman.modules.system.vo;

import cn.hutool.core.collection.ListUtil;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import com.qianwen.smartman.modules.cps.entity.Workstation;

/* loaded from: blade-api.jar:BOOT-INF/classes/org/springblade/modules/system/vo/WorkstationChartSeriesDataVO.class */
public class WorkstationChartSeriesDataVO<T extends Serializable> extends ChartSeriesDataVO<T> {
    public void generateCategories(List<Workstation> workstationList) {
        generateCategories(workstationList, 1);
    }

    public void generateCategories(List<Workstation> workstationList, int defaultCategory) {
        List<String> nameList = new ArrayList<>();
        List<String> codeList = new ArrayList<>();
        List<String> nameAndCodeList = new ArrayList<>();
        workstationList.forEach(station -> {
            nameList.add(station == null ? "" : station.getName());
            codeList.add(station == null ? "" : station.getCode());
            nameAndCodeList.add(station == null ? "" : station.getName() + "-" + station.getCode());
        });
        setCategories(defaultCategory == 1 ? nameList : defaultCategory == 2 ? codeList : nameAndCodeList);
        setCategoriesExtra(ListUtil.toList(new List[]{nameList, codeList, nameAndCodeList}));
    }
}