/**
 * Copyright (c) 2025 Huawei Technologies Co., Ltd.
 * This program is free software, you can redistribute it and/or modify it under the terms and conditions of
 * CANN Open Software License Agreement Version 2.0 (the "License").
 * Please refer to the License for details. You may not use this file except in compliance with the License.
 * 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 FITNESS FOR A PARTICULAR PURPOSE.
 * See LICENSE in the root of the software repository for the full text of the License.
 */

#ifndef CCE_RUNTIME_RT_EXTERNAL_FFTS_H
#define CCE_RUNTIME_RT_EXTERNAL_FFTS_H

#include "rt_external_base.h"
#include "rt_external_ffts_define.h"
#include "rt_external_stars_define.h"

#if defined(__cplusplus) && !defined(COMPILE_OMG_PACKAGE)
extern "C" {
#endif

#pragma pack(push)
#pragma pack(1)

// context desc addr type
#define RT_FFTS_PLUS_CTX_DESC_ADDR_TYPE_HOST (0x0U)
#define RT_FFTS_PLUS_CTX_DESC_ADDR_TYPE_DEVICE (0x1U)

typedef struct tagFftsPlusDumpInfo {
    const void* loadDumpInfo;
    const void* unloadDumpInfo;
    uint32_t loadDumpInfolen;
    uint32_t unloadDumpInfolen;
} rtFftsPlusDumpInfo_t;

typedef struct tagFftsPlusTaskInfo {
    const rtFftsPlusSqe_t* fftsPlusSqe;
    const void* descBuf;                   // include total context
    size_t descBufLen;                     // the length of descBuf
    rtFftsPlusDumpInfo_t fftsPlusDumpInfo; // used only in the dynamic shape
    uint32_t descAddrType;                 // 0:host addr 1:device addr
    uint32_t argsHandleInfoNum;
    void** argsHandleInfoPtr;
} rtFftsPlusTaskInfo_t;

#pragma pack(pop)

RTS_API rtError_t rtFftsPlusTaskLaunch(rtFftsPlusTaskInfo_t* fftsPlusTaskInfo, rtStream_t stm);

RTS_API rtError_t rtFftsPlusTaskLaunchWithFlag(rtFftsPlusTaskInfo_t* fftsPlusTaskInfo, rtStream_t stm, uint32_t flag);

#if defined(__cplusplus) && !defined(COMPILE_OMG_PACKAGE)
}
#endif
#endif // CCE_RUNTIME_RT_EXTERNAL_FFTS_H