/*
 * Copyright (c) Huawei Technologies Co., Ltd. 2019-2021. All rights reserved.
 * Description: handle perf data
 * Author: xp
 * Create: 2019-10-13
 */

#ifndef MSPROFILER_API_PROF_ACL_API_H_
#define MSPROFILER_API_PROF_ACL_API_H_

// DataTypeConfig
#define PROF_ACL_API                0x00000001ULL
#define PROF_TASK_TIME              0x00000002ULL
#define PROF_AICORE_METRICS         0x00000004ULL
#define PROF_AICPU_TRACE            0x00000008ULL
#define PROF_L2CACHE                0x00000010ULL
#define PROF_HCCL_TRACE             0x00000020ULL
#define PROF_TRAINING_TRACE         0x00000040ULL
#define PROF_MSPROFTX               0x00000080ULL
#define PROF_RUNTIME_API            0x00000100ULL

// system profilinig switch
#define PROF_CPU                    0x00010000ULL
#define PROF_HARDWARE_MEMORY        0x00020000ULL
#define PROF_IO                     0x00040000ULL
#define PROF_INTER_CONNECTION       0x00080000ULL
#define PROF_DVPP                   0x00100000ULL
#define PROF_SYS_AICORE_SAMPLE      0x00200000ULL
#define PROF_AIVECTORCORE_SAMPLE    0x00400000ULL

#define PROF_MODEL_EXECUTE          0x0000001000000ULL
#define PROF_RUNTIME_TRACE          0x0000004000000ULL
#define PROF_SCHEDULE_TIMELINE      0x0000008000000ULL
#define PROF_SCHEDULE_TRACE         0x0000010000000ULL
#define PROF_AIVECTORCORE_METRICS   0x0000020000000ULL
#define PROF_SUBTASK_TIME           0x0000040000000ULL
#define PROF_OP_DETAIL              0x0000080000000ULL

#define PROF_MODEL_LOAD             0x8000000000000000ULL

#define PROF_TASK_TRACE             (PROF_MODEL_EXECUTE | PROF_RUNTIME_TRACE | PROF_TRAINING_TRACE | \
                                     PROF_HCCL_TRACE | PROF_TASK_TIME)

// DataTypeConfig MASK
#define PROF_ACL_API_MASK                0x00000001ULL
#define PROF_TASK_TIME_MASK              0x00000002ULL
#define PROF_AICORE_METRICS_MASK         0x00000004ULL
#define PROF_AICPU_TRACE_MASK            0x00000008ULL
#define PROF_L2CACHE_MASK                0x00000010ULL
#define PROF_HCCL_TRACE_MASK             0x00000020ULL
#define PROF_TRAINING_TRACE_MASK         0x00000040ULL
#define PROF_MSPROFTX_MASK               0x00000080ULL
#define PROF_RUNTIME_API_MASK            0x00000100ULL

// system profilinig mask
#define PROF_CPU_MASK                    0x00010000ULL
#define PROF_HARDWARE_MEMORY_MASK        0x00020000ULL
#define PROF_IO_MASK                     0x00040000ULL
#define PROF_INTER_CONNECTION_MASK       0x00080000ULL
#define PROF_DVPP_MASK                   0x00100000ULL
#define PROF_SYS_AICORE_SAMPLE_MASK      0x00200000ULL
#define PROF_AIVECTORCORE_SAMPLE_MASK    0x00400000ULL

#define PROF_MODEL_EXECUTE_MASK          0x0000001000000ULL
#define PROF_RUNTIME_TRACE_MASK          0x0000004000000ULL
#define PROF_SCHEDULE_TIMELINE_MASK      0x0000008000000ULL
#define PROF_SCHEDULE_TRACE_MASK         0x0000010000000ULL
#define PROF_AIVECTORCORE_METRICS_MASK   0x0000020000000ULL
#define PROF_SUBTASK_TIME_MASK           0x0000040000000ULL
#define PROF_OP_DETAIL_MASK              0x0000080000000ULL

#define PROF_MODEL_LOAD_MASK             0x8000000000000000ULL

#if (defined(_WIN32) || defined(_WIN64) || defined(_MSC_VER))
#define MSVP_PROF_API __declspec(dllexport)
#else
#define MSVP_PROF_API __attribute__((visibility("default")))
#endif

#include <cstdint>
#include <cstddef>

namespace Msprofiler {
namespace Api {
/**
 * @name  ProfGetOpExecutionTime
 * @brief get op execution time of specific part of data
 * @param data  [IN] data read from pipe
 * @param len   [IN] data length
 * @param index [IN] index of part(op)
 * @return op execution time (us)
 */
MSVP_PROF_API uint64_t ProfGetOpExecutionTime(const void *data, uint32_t len, uint32_t index);
}
}

#ifdef __cplusplus
extern "C" {
#endif

MSVP_PROF_API uint64_t ProfGetOpExecutionTime(const void *data, uint32_t len, uint32_t index);

typedef in32_t Status;
typedef struct aclprofSubscribeConfig aclprofSubscribeConfig1;
///
/// @ingroup AscendCL
/// @brief subscribe profiling data of graph
/// @param [in] graphId: the graph id subscribed
/// @param [in] profSubscribeConfig: pointer to config of model subscribe
/// @return Status result of function
///
MSVP_PROF_API Status aclgrphProfGraphSubscribe(const uint32_t graphId,
    const aclprofSubscribeConfig1 *profSubscribeConfig);

///
/// @ingroup AscendCL
/// @brief unsubscribe profiling data of graph
/// @param [in] graphId: the graph id subscribed
/// @return Status result of function
///
MSVP_PROF_API Status aclgrphProfGraphUnSubscribe(const uint32_t graphId);

/**
 * @ingroup AscendCL
 * @brief get graph id from subscription data
 *
 * @param  opInfo [IN]     pointer to subscription data
 * @param  opInfoLen [IN]  memory size of subscription data
 *
 * @retval graph id of subscription data
 * @retval 0 for failed
 */
MSVP_PROF_API size_t aclprofGetGraphId(const void *opInfo, size_t opInfoLen, uint32_t index);

/**
* @ingroup AscendCL
* @brief set stamp pay load
*
*
* @retval void
*/
MSVP_PROF_API int aclprofSetStampPayload(void *stamp, const int32_t type, void *value);

/**
* @ingroup AscendCL
* @brief set category and name
*
*
* @retval void
*/
MSVP_PROF_API int aclprofSetCategoryName(uint32_t category, const char *categoryName);

/**
* @ingroup AscendCL
* @brief set category to stamp
*
*
* @retval void
*/
MSVP_PROF_API int aclprofSetStampCategory(void *stamp, uint32_t category);
#ifdef __cplusplus
}
#endif

#endif  // MSPROFILER_API_PROF_ACL_API_H_