/*
 * -------------------------------------------------------------------------
 * This file is part of the MindStudio project.
 * Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 *
 * 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.
 * -------------------------------------------------------------------------
 */

#ifndef PROFILER_SERVER_OPERATORPROTOCOLRESPONSE_H
#define PROFILER_SERVER_OPERATORPROTOCOLRESPONSE_H

#include <set>
#include <vector>
#include "ProtocolDefs.h"
#include "ProtocolMessage.h"

namespace Dic::Protocol {
constexpr int64_t INT_MIN_VALUE = std::numeric_limits<int64_t>::min();
;
constexpr double DOUBLE_MIN_VALUE = std::numeric_limits<double>::min();
// 饼图中按算子类型耗时统计和按计算单元分类耗时统计
struct OperatorDurationRes {
    std::string name;
    double duration;
};

// 按Operator Type、Input Shape展示时算子统计信息
struct OperatorStatisticInfoRes {
    std::string opType;
    std::string opName;
    std::string inputShape;
    std::string accCore;
    std::string totalTime;
    std::string count;
    std::string avgTime;
    std::string maxTime;
    std::string minTime;
};

// 按Operator展示算子详细信息，和See More响应
struct OperatorDetailInfoRes {
    std::string rankId;
    std::string stepId;
    std::string name;
    std::string type;
    std::string accCore;
    std::string startTime;
    std::string duration;
    std::string waitTime;
    std::string blockNum;
    std::string inputShape;
    std::string inputType;
    std::string inputFormat;
    std::string outputShape;
    std::string outputType;
    std::string outputFormat;
    std::map<std::string, std::string> pmuDatas;
};

struct OperatorStatisticCmpInfoRes {
    OperatorStatisticInfoRes diff;
    OperatorStatisticInfoRes baseline;
    OperatorStatisticInfoRes compare;
};

struct OperatorDetailCmpInfoRes {
    OperatorDetailInfoRes diff;
    OperatorDetailInfoRes baseline;
    OperatorDetailInfoRes compare;
};

// 获取按算子类型统计耗时信息以绘制饼图的响应
struct OperatorCategoryInfoResponse : public Response {
    OperatorCategoryInfoResponse() : Response(REQ_RES_OPERATOR_CATEGORY_INFO) {}
    std::vector<OperatorDurationRes> data;
};

// 获取按计算单元统计耗时信息以绘制饼图的响应
struct OperatorComputeUnitInfoResponse : public Response {
    OperatorComputeUnitInfoResponse() : Response(REQ_RES_OPERATOR_COMPUTE_UNIT_INFO) {}
    std::vector<OperatorDurationRes> data;
};

// 获取按算子类型统计耗时信息以填充表格的响应
struct OperatorStatisticInfoResponse : public Response {
    OperatorStatisticInfoResponse() : Response(REQ_RES_OPERATOR_STATISTIC_INFO) {};
    int64_t total{0};
    std::vector<OperatorStatisticCmpInfoRes> data;
};

// 获取按算子详情信息以填充表格的响应
struct OperatorDetailInfoResponse : public Response {
    OperatorDetailInfoResponse() : Response(REQ_RES_OPERATOR_DETAIL_INFO) {};
    int64_t total{0};
    std::string level; // l0, l1, l2
    std::set<std::string> pmuHeaders;
    std::vector<OperatorDetailCmpInfoRes> data;
};

// 获取See More时算子详情信息的响应
struct OperatorMoreInfoResponse : public Response {
    OperatorMoreInfoResponse() : Response(REQ_RES_OPERATOR_MORE_INFO) {};
    int64_t total{0};
    std::string level; // l0, l1, l2
    std::set<std::string> pmuHeaders;
    std::vector<OperatorDetailInfoRes> data;
};

// 导出算子全量信息的响应
struct OperatorExportDetailsResponse : public Response {
    OperatorExportDetailsResponse() : Response(REQ_RES_OPERATOR_EXPORT_DETAILS) {};
    bool exceedingFileLimit{false};
    std::string filePath;
};
}

#endif // PROFILER_SERVER_OPERATORPROTOCOLRESPONSE_H