SimulationServer/SimulationServer/Component/TaskComponent.cs

using KYFramework;
using SimulationServer.Utils;

namespace SimulationServer;

public class TaskComponent : Component
{
    public static Weather Weather;
    public static int Speed = 100;
    // 救援任务列表
    public List<MHRescueMission> MHRescueMissions;
    public List<XHRescueMission> XHRescueMissions;
    public List<ZCRescueMission> ZCRescueMissions;
    public List<SeaSJRescueMission> SeaSJRescueMissions;
    public List<LandSJRescueMission> LandSJRescueMissions;

    public SJAllTotalTaskPerformanceComponent SSJAllTotalTaskPerformance = new SJAllTotalTaskPerformanceComponent();
    public Dictionary<string, Dictionary<string, List<string>>> alltotalReportSea = new();
    public Dictionary<string, Dictionary<string, List<string>>> alltotalReportLand = new();
    // <sheet,<指标名，值列表>>
    public Dictionary<string, Dictionary<string, List<string>>> totalReportSea = new();
    public Dictionary<string, Dictionary<string, List<string>>> totalReportLand = new();

    private int currentSeaExecuteCount = 1;
    private int currentLandExecuteCount = 1;
    public int ExecutionContext = 0;

    public int seaSuccessCount = 0;
    public int landSuccessCount = 0;

    public void Start()
    {
        MHRescueMissions?.ForEach(r => r.Reset());
        XHRescueMissions?.ForEach(r => r.Reset());

        MHRescueMissions?.ForEach(r => r.Start());
        XHRescueMissions?.ForEach(r => r.Start());
        ZCRescueMissions?.ForEach(r => r.Start());
        SeaSJRescueMissions?.ForEach(r => r.Start());
        LandSJRescueMissions?.ForEach(r => r.Start());
    }

    public void SaveSeaSJ(Dictionary<string, Dictionary<string, string>> totalReport)
    {
        //把totalReport保存到alltotalReport
        foreach (var kv in totalReport)
        {
            if (!alltotalReportSea.ContainsKey(kv.Key)) alltotalReportSea[kv.Key] = new Dictionary<string, List<string>>();


            foreach (var kv2 in kv.Value)
            {
                if (!alltotalReportSea[kv.Key].ContainsKey(kv2.Key)) alltotalReportSea[kv.Key][kv2.Key] = new List<string>();

                alltotalReportSea[kv.Key][kv2.Key].Add(kv2.Value);
            }
        }

        if (currentSeaExecuteCount == ExecutionContext)
        {
            // 求一个平均值放到最后一位
            foreach (var kv in alltotalReportSea)
            {
                foreach (var kv2 in kv.Value)
                {
                    var sum = 0f;
                    foreach (var value in kv2.Value)
                    {
                        bool isfloat = float.TryParse(value, out float f);
                        if (isfloat)
                            sum += float.Parse(value);
                        else
                            sum = -1f;
                    }
                    if (sum != -1f)
                        kv2.Value.Add((sum / kv2.Value.Count).ToString());
                    else
                        kv2.Value.Add("");
                }
            }
        }

        currentSeaExecuteCount++;
    }

    public void SaveLandSJ(Dictionary<string, Dictionary<string, string>> totalReport)
    {
        //把totalReport保存到alltotalReport
        foreach (var kv in totalReport)
        {
            if (!alltotalReportLand.ContainsKey(kv.Key)) alltotalReportLand[kv.Key] = new Dictionary<string, List<string>>();


            foreach (var kv2 in kv.Value)
            {
                if (!alltotalReportLand[kv.Key].ContainsKey(kv2.Key)) alltotalReportLand[kv.Key][kv2.Key] = new List<string>();

                alltotalReportLand[kv.Key][kv2.Key].Add(kv2.Value);
            }
        }

        if (currentLandExecuteCount == ExecutionContext)
        {
            // 求一个平均值放到最后一位
            foreach (var kv in alltotalReportLand)
            {
                foreach (var kv2 in kv.Value)
                {
                    var sum = 0f;
                    foreach (var value in kv2.Value)
                    {
                        bool isfloat = float.TryParse(value, out float f);
                        if (isfloat)
                            sum += float.Parse(value);
                        else
                            sum = -1f;
                    }
                    if (sum != -1f)
                        kv2.Value.Add((sum / kv2.Value.Count).ToString());
                    else
                        kv2.Value.Add("");
                }
            }
        }

        currentLandExecuteCount++;
    }

    public void ReportAllSJSea()
    {
        string data = DateTime.Now.ToString("yyyy-MM-dd");
        string path = $"Reports/SSJ/{data}";
        if (!Directory.Exists(path)) Directory.CreateDirectory(path);

        string totalPath = $"{path}/{"海上搜救任务总体指标报告"}.xls";
        DataTableExtensions.SaveToExcel(totalPath, alltotalReportSea, true);


        string totalPath1 = $"{path}/{"搜救任务总体指标报告"}.xls";
        DataTableExtensions.SaveToExcel(totalPath1, totalReportSea);
    }

    public void ReportAllSJLand()
    {
        string data = DateTime.Now.ToString("yyyy-MM-dd");
        string path = $"Reports/LSX/{data}";
        if (!Directory.Exists(path)) Directory.CreateDirectory(path);

        string totalPath = $"{path}/{"陆上搜寻任务总体指标报告"}.xls";
        DataTableExtensions.SaveToExcel(totalPath, alltotalReportLand, true);


        string totalPath1 = $"{path}/{"搜救任务总体指标报告"}.xls";
        DataTableExtensions.SaveToExcel(totalPath1, totalReportLand);
    }

    public void SaveTotalMHSea()
    {
        if (!totalReportSea.ContainsKey("总任务表现")) totalReportSea["总任务表现"] = new Dictionary<string, List<string>>();
        if (!totalReportSea["总任务表现"].ContainsKey("任务成功率")) totalReportSea["总任务表现"]["任务成功率"] = new List<string>();
        totalReportSea["总任务表现"]["任务成功率"].Add((seaSuccessCount * 1f / ExecutionContext).ToString());
    }

    public void SaveTotalMHLand()
    {
        if (!totalReportLand.ContainsKey("总任务表现")) totalReportLand["总任务表现"] = new Dictionary<string, List<string>>();
        if (!totalReportLand["总任务表现"].ContainsKey("任务成功率")) totalReportLand["总任务表现"]["任务成功率"] = new List<string>();
        totalReportLand["总任务表现"]["任务成功率"].Add((landSuccessCount * 1f / ExecutionContext).ToString());
    }
}

[ObjectSystem]
public class TaskComponentAwakeSystem : AwakeSystem<TaskComponent>
{
    public override void Awake(TaskComponent self)
    {
        self.MHRescueMissions = new List<MHRescueMission>();
        self.ZCRescueMissions = new List<ZCRescueMission>();
        self.XHRescueMissions = new List<XHRescueMission>();
        self.SeaSJRescueMissions = new List<SeaSJRescueMission>();
        self.LandSJRescueMissions = new List<LandSJRescueMission>();
    }
}

[ObjectSystem]
public class TaskComponentUpdateSystem : UpdateSystem<TaskComponent>
{
    public override void Update(TaskComponent self)
    {
        if (self.SeaSJRescueMissions.Count > 0)
        {
            if (self.SeaSJRescueMissions.All(m => m.IsOver))
            {
                for (int i = 0; i < self.ExecutionContext; i++)
                {
                    self.SSJAllTotalTaskPerformance.FillData(self.SeaSJRescueMissions);
                    var report = self.SSJAllTotalTaskPerformance.GetReport();
                    if (report["总任务表现"]["任务是否成功"] == "1")
                        self.seaSuccessCount++;
                    self.SaveSeaSJ(report);
                }
                self.SaveTotalMHSea();
                self.ReportAllSJSea();
                self.SeaSJRescueMissions.Clear();
            }

        }

        if (self.LandSJRescueMissions.Count > 0)
        {
            if (self.LandSJRescueMissions.All(m => m.IsOver))
            {
                for (int i = 0; i < self.ExecutionContext; i++)
                {
                    self.SSJAllTotalTaskPerformance.FillData1(self.LandSJRescueMissions);
                    var report = self.SSJAllTotalTaskPerformance.GetReport1();
                    if (report["总任务表现"]["任务是否成功"] == "1")
                        self.landSuccessCount++;
                    self.SaveLandSJ(report);
                }
                self.SaveTotalMHLand();
                self.ReportAllSJLand();
                self.LandSJRescueMissions.Clear();
            }
        }
    }
}