/*
 * Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
 *
 * SPDX-License-Identifier: MIT
 */

#pragma once

#ifndef ROC_D3D10_INTEROP_HPP_
#define ROC_D3D10_INTEROP_HPP_

#include "top.hpp"

#ifdef _WIN32

#include <d3d10.h>
#include <dxgi.h>

namespace amd {
namespace roc {

// Forward declarations
class Device;
class Memory;

namespace D3D10Interop {

/**
 * @brief Validate D3D10 device matches ROCr GPU device
 *
 * Performs validation: LUID matching via DXGI adapter
 *
 * @param device ROCr device to validate against
 * @param d3d10Device D3D10 device to associate
 * @param gfxContext Owning amd::Context*, used to cache the DXX extension
 * @param validateOnly When true, only check LUID compatibility and skip
 *        creating/caching the DXX extension objects (used by query paths)
 * @return true if devices can interoperate, false otherwise
 */
bool associateD3D10Device(
    const Device* device,
    ID3D10Device* d3d10Device,
    void* gfxContext,
    bool validateOnly
);

/**
 * @brief Dissociate D3D10 device from ROCr device
 *
 * Cleanup function called during context destruction
 *
 * @param device ROCr device
 */
void dissociateD3D10Device(const Device* device, void* const gfxDevice[], void* gfxContext);

/**
 * @brief Export D3D10 resource to HSA handle for interop
 *
 * Extracts shared handle from D3D10 resource and returns it
 * as HSA handle for memory mapping
 *
 * @param memory ROCr memory object
 * @param d3d10Obj D3D10 object to export
 * @param handle Output HSA handle
 * @param offset Output offset into resource
 * @return true if export succeeded, false otherwise
 */
bool Export(
    const Memory* memory,
    D3D10Object* d3d10Obj,
    hsa_handle_t* handle,
    int* offset,
    void* srd,
    UINT* srdSize,
    hsa_interop_map_flag_t* mapFlags
);

}  // namespace D3D10Interop
}  // namespace roc
}  // namespace amd

#endif  // _WIN32

#endif  // ROC_D3D10_INTEROP_HPP_