/* -------------------------------------------------------------------------
* Copyright (c) 2026 Huawei Technologies Co., Ltd.
 * This file is part of the MindStudio project.
 *
 * MindStudio is licensed under Mulan PSL v2.
 * You can use this software according to the terms and conditions of the Mulan PSL v2.
 * You may obtain a copy of Mulan PSL v2 at:
 *
 *    http://license.coscl.org.cn/MulanPSL2
 *
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
 * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
 * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 * See the Mulan PSL v2 for more details.
 * -------------------------------------------------------------------------*/

#include "analysis/csrc/application/timeline/biu_perf_assembler.h"
#include "analysis/csrc/domain/entities/viewer_data/system/include/biu_perf_data.h"
#include "analysis/csrc/domain/services/environment/context.h"
#include "analysis/csrc/domain/services/constant/time_unit_constant.h"

namespace Analysis {
namespace Application {
BiuPerfAssembler::BiuPerfAssembler() : JsonAssembler(PROCESS_BIU_PERF, {{MSPROF_JSON_FILE, FileCategory::MSPROF}}) {}


std::map<std::pair<uint16_t, std::string>, std::vector<BiuPerfData>> groupBiuPerfData(const std::vector<BiuPerfData>& biuPerfData) {
    // 用 map 存储分组，天然保证键的有序性（先 groupId，后 core_type）
    std::map<std::pair<uint16_t, std::string>, std::vector<BiuPerfData>> groupedMap;

    if (biuPerfData.empty()) {
        return groupedMap;
    }

    // 遍历所有元素，按组合键分组
    for (const auto& data : biuPerfData) {
        std::pair<uint16_t, std::string> key = {data.groupId, data.coreType};
        groupedMap[key].emplace_back(data); // 同组元素加入对应列表
    }

    return groupedMap;
}


void GenerateBiuPerfTrace(std::vector<BiuPerfData> &biuPerfData, const std::unordered_map<uint16_t, uint32_t> &pidMap,
                         std::vector<std::shared_ptr<TraceEvent>> &res)
{
    //group by (group_id, core_type)
    std::map<std::pair<uint16_t, std::string>, std::vector<BiuPerfData>> groupedData = groupBiuPerfData(biuPerfData);
    std::shared_ptr<BiuPerfEvent> event;
    std::shared_ptr<MetaDataNameEvent> threadNameEvent;
    uint16_t tid = 0;
    // if use device sep, get pid from pidMap through deviceId
    uint32_t pid = pidMap.begin() -> second;
    for (auto it = groupedData.begin(); it != groupedData.end(); ++it) {
        const auto& key = it->first;
        MAKE_SHARED_RETURN_VOID(threadNameEvent, MetaDataNameEvent, pid, tid,
                                 META_DATA_THREAD_NAME, "Group" + std::to_string(key.first) + "-" + key.second);
        res.emplace_back(std::move(threadNameEvent));
        // 后续打印逻辑同上
        for (const auto &data : it->second) {
            MAKE_SHARED_RETURN_VOID(event, BiuPerfEvent, pid, tid, data.duration / US_TO_MS,
                DivideByPowersOfTenWithPrecision(data.timestamp), data.instruction, data.coreType, data.blockId);
            res.emplace_back(std::move(event));
        }
        tid++;
    }
}

uint8_t BiuPerfAssembler::AssembleData(DataInventory &dataInventory, JsonWriter &ostream, const std::string &profPath)
{
    auto biuPerf = dataInventory.GetPtr<std::vector<BiuPerfData>>();
    if (biuPerf == nullptr) {
        WARN("Can't get biuPerf data from dataInventory");
        return DATA_NOT_EXIST;
    }
    std::unordered_map<uint16_t, uint32_t> pidMap;
    auto layerInfo = GetLayerInfo(PROCESS_BIU_PERF);
    auto deviceList = File::GetFilesWithPrefix(profPath, DEVICE_PREFIX);
    for (const auto& devicePath: deviceList) {
        auto deviceId = GetDeviceIdByDevicePath(devicePath);
        auto pid = Environment::Context::GetInstance().GetPidFromInfoJson(deviceId, profPath);
        uint32_t formatPid = GetFormatPid(pid, layerInfo.sortIndex, deviceId);
        pidMap[deviceId] = formatPid;
    }
    if (pidMap.empty()) {
        WARN("Can't found any device info.");
        return DATA_NOT_EXIST;
    }
    GenerateHWMetaData(pidMap, layerInfo, res_);
    GenerateBiuPerfTrace(*biuPerf, pidMap, res_);
    if (res_.empty()) {
        ERROR("Can't Generate any biu perf process data");
        return ASSEMBLE_FAILED;
    }
    for (const auto &node : res_) {
        node->DumpJson(ostream);
    }
    // 为了让下一个写入的内容形成正确的JSON格式，需要补一个","
    ostream << ",";
    return ASSEMBLE_SUCCESS;
}
}
}