// slang-ir-marshal-native-call.h
#pragma once

#include "../core/slang-basic.h"

namespace Slang
{
    class DiagnosticSink;
    struct IRModule;
    struct IRBuilder;
    struct IRType;
    struct IRFunc;
    struct IRFuncType;
    struct IRCall;
    struct IRInst;

    class NativeCallMarshallingContext
    {
    public:
        DiagnosticSink* diagnosticSink = nullptr;
    public:
        // Get a native type for `type` that can be used directly in a native function signature.
        IRType* getNativeType(IRBuilder& builder, IRType* type);

        // Get a native function type of `func`.
        IRFuncType* getNativeFuncType(IRBuilder& builder, IRFuncType* declaredFuncType);

        // Insert a call at builder's current position into a native func with original arguments.
        // `originalArgs` will be marshalled to native args before the actual call.
        // returns the managed result value of the call.
        // Note: additional insts maybe inserted after the call inst to marshal the native output values back
        // to non-native arguments/return values.
        IRInst* marshalNativeCall(
            IRBuilder& builder,
            IRFuncType* originalFuncType,
            IRFuncType* nativeFuncType,
            IRInst* nativeFunc,
            Int argCount,
            IRInst* const* originalArgs);

        void marshalRefManagedValueToNativeValue(
            IRBuilder& builder, IRInst* originalArg, List<IRInst*>& args);

        // Marshal a managed value to a native value for input into a native functions.
        void marshalManagedValueToNativeValue(
            IRBuilder& builder, IRType* originalParamType, IRInst* originalArg, List<IRInst*>& args);

        // Marshal a managed value to a native value for the return value of a native function.
        void marshalManagedValueToNativeResultValue(
            IRBuilder& builder, IRInst* originalArg, List<IRInst*>& args);

        IRInst* marshalNativeValueToManagedValue(
            IRBuilder& builder, IRInst* nativeValue);

        IRInst* marshalNativeArgToManagedArg(IRBuilder& builder, const List<IRInst*>& args, Index& consumeIndex, IRType* expectedManagedType);

        IRFunc* generateDLLExportWrapperFunc(IRBuilder& builder, IRFunc* originalFunc);
    };
}