#include "vendor/SPIRV-Cross/spirv.hpp"
#include "vendor/SPIRV-Cross/spirv_cross_util.hpp"
#include "vendor/SPIRV-Cross/spirv_hlsl.hpp"
#include "vendor/SPIRV-Cross/spirv_msl.hpp"
#include "vendor/SPIRV-Cross/spirv_glsl.hpp"

typedef void ScInternalCompilerBase;
typedef void ScInternalCompilerHlsl;
typedef void ScInternalCompilerMsl;
typedef void ScInternalCompilerGlsl;

extern "C"
{
    enum ScInternalResult
    {
        Success,
        Unhandled,
        CompilationError,
    };

    typedef struct ScEntryPoint
    {
        char *name;
        spv::ExecutionModel execution_model;
        uint32_t work_group_size_x;
        uint32_t work_group_size_y;
        uint32_t work_group_size_z;
    } ScEntryPoint;

    typedef struct ScBufferRange
    {
        unsigned index;
        size_t offset;
        size_t range;
    } ScBufferRange;

    typedef struct ScCombinedImageSampler
    {
        uint32_t combined_id;
        uint32_t image_id;
        uint32_t sampler_id;
    } ScCombinedImageSampler;

    typedef struct ScHlslRootConstant
    {
        uint32_t start;
        uint32_t end;
        uint32_t binding;
        uint32_t space;
    } ScHlslRootConstant;

    typedef struct ScHlslCompilerOptions
    {
        int32_t shader_model;
        bool point_size_compat;
        bool point_coord_compat;
        bool vertex_transform_clip_space;
        bool vertex_invert_y;
        bool force_storage_buffer_as_uav;
        bool nonwritable_uav_texture_as_srv;
        bool force_zero_initialized_variables;
    } ScHlslCompilerOptions;

    typedef struct ScMslCompilerOptions
    {
        bool vertex_transform_clip_space;
        bool vertex_invert_y;
        uint8_t platform;
        uint32_t version;
        bool enable_point_size_builtin;
        bool disable_rasterization;
        uint32_t swizzle_buffer_index;
        uint32_t indirect_params_buffer_index;
        uint32_t shader_output_buffer_index;
        uint32_t shader_patch_output_buffer_index;
        uint32_t shader_tess_factor_buffer_index;
        uint32_t buffer_size_buffer_index;
        bool capture_output_to_buffer;
        bool swizzle_texture_samples;
        bool tess_domain_origin_lower_left;
        bool argument_buffers;
        bool pad_fragment_output_components;
        bool force_native_arrays;
        bool force_zero_initialized_variables;
    } ScMslCompilerOptions;

    typedef struct ScGlslCompilerOptions
    {
        bool vertex_transform_clip_space;
        bool vertex_invert_y;
        bool vertex_support_nonzero_base_instance;
        uint8_t fragment_default_float_precision;
        uint8_t fragment_default_int_precision;
        uint32_t version;
        bool es;
        bool force_temporary;
        bool vulkan_semantics;
        bool separate_shader_objects;
        bool flatten_multidimensional_arrays;
        bool enable_420_pack_extension;
        bool emit_push_constant_as_uniform_buffer;
        bool emit_uniform_buffer_as_plain_uniforms;
        bool emit_line_directives;
        bool enable_storage_image_qualifier_deduction;
        bool force_zero_initialized_variables;
    } ScGlslCompilerOptions;

    typedef struct ScResource
    {
        uint32_t id;
        uint32_t type_id;
        uint32_t base_type_id;
        char *name;
    } ScResource;

    typedef struct ScResourceArray
    {
        ScResource *data;
        size_t num;
    } ScResourceArray;

    typedef struct ScShaderResources
    {
        ScResourceArray uniform_buffers;
        ScResourceArray storage_buffers;
        ScResourceArray stage_inputs;
        ScResourceArray stage_outputs;
        ScResourceArray subpass_inputs;
        ScResourceArray storage_images;
        ScResourceArray sampled_images;
        ScResourceArray atomic_counters;
        ScResourceArray push_constant_buffers;
        ScResourceArray separate_images;
        ScResourceArray separate_samplers;
    } ScShaderResources;

    typedef struct ScSpecializationConstant
    {
        uint32_t id;
        uint32_t constant_id;
    } ScSpecializationConstant;

    typedef struct ScType
    {
        spirv_cross::SPIRType::BaseType type;
        uint32_t vecsize;
        uint32_t columns;
        uint32_t *member_types;
        size_t member_types_size;
        uint32_t *array;
        size_t array_size;
    } ScType;

    ScInternalResult sc_internal_get_latest_exception_message(const char **message);

#ifdef SPIRV_CROSS_WRAPPER_HLSL
    ScInternalResult sc_internal_compiler_hlsl_new(ScInternalCompilerHlsl **compiler, const uint32_t *ir, const size_t size);
    ScInternalResult sc_internal_compiler_hlsl_set_options(const ScInternalCompilerHlsl *compiler, const ScHlslCompilerOptions *options);
    ScInternalResult sc_internal_compiler_hlsl_set_root_constant_layout(const ScInternalCompilerHlsl *compiler, const ScHlslRootConstant *constants, size_t count);
#endif

#ifdef SPIRV_CROSS_WRAPPER_MSL
    typedef struct ScMslConstSamplerMapping {
        uint32_t desc_set;
        uint32_t binding;
        spirv_cross::MSLConstexprSampler sampler;
    } ScMslConstSamplerMapping;

    ScInternalResult sc_internal_compiler_msl_new(ScInternalCompilerMsl **compiler, const uint32_t *ir, const size_t size);
    ScInternalResult sc_internal_compiler_msl_set_options(const ScInternalCompilerMsl *compiler, const ScMslCompilerOptions *options);
    ScInternalResult sc_internal_compiler_msl_get_is_rasterization_disabled(const ScInternalCompilerMsl *compiler, bool *is_rasterization_disabled);
    ScInternalResult sc_internal_compiler_msl_compile(const ScInternalCompilerBase *compiler, const char **shader,
                                                      const spirv_cross::MSLShaderInput *p_vat_overrides, const size_t vat_override_count,
                                                      const spirv_cross::MSLResourceBinding *p_res_overrides, const size_t res_override_count,
                                                      const ScMslConstSamplerMapping *p_const_samplers, const size_t const_sampler_count);
#endif

#ifdef SPIRV_CROSS_WRAPPER_GLSL
    ScInternalResult sc_internal_compiler_glsl_new(ScInternalCompilerGlsl **compiler, const uint32_t *ir, const size_t size);
    ScInternalResult sc_internal_compiler_glsl_set_options(const ScInternalCompilerGlsl *compiler, const ScGlslCompilerOptions *options);
    ScInternalResult sc_internal_compiler_glsl_build_combined_image_samplers(const ScInternalCompilerBase *compiler);
    ScInternalResult sc_internal_compiler_glsl_get_combined_image_samplers(const ScInternalCompilerBase *compiler, const ScCombinedImageSampler **samplers, size_t *size);
    ScInternalResult sc_internal_compiler_glsl_add_header_line(const ScInternalCompilerBase *compiler, const char *str);
    ScInternalResult sc_internal_compiler_glsl_flatten_buffer_block(const ScInternalCompilerBase *compiler, const uint32_t id);
#endif

    ScInternalResult sc_internal_compiler_get_decoration(const ScInternalCompilerBase *compiler, uint32_t *result, const uint32_t id, const spv::Decoration decoration);
    ScInternalResult sc_internal_compiler_set_decoration(const ScInternalCompilerBase *compiler, const uint32_t id, const spv::Decoration decoration, const uint32_t argument);
    ScInternalResult sc_internal_compiler_unset_decoration(const ScInternalCompilerBase *compiler, const uint32_t id, const spv::Decoration decoration);
    ScInternalResult sc_internal_compiler_get_name(const ScInternalCompilerBase *compiler, const uint32_t id, const char **name);
    ScInternalResult sc_internal_compiler_set_name(const ScInternalCompilerBase *compiler, const uint32_t id, const char *name);
    ScInternalResult sc_internal_compiler_get_entry_points(const ScInternalCompilerBase *compiler, ScEntryPoint **entry_points, size_t *size);
    ScInternalResult sc_internal_compiler_get_active_buffer_ranges(const ScInternalCompilerBase *compiler, uint32_t id, ScBufferRange **active_buffer_ranges, size_t *size);
    ScInternalResult sc_internal_compiler_get_cleansed_entry_point_name(const ScInternalCompilerBase *compiler, const char *original_entry_point_name, const spv::ExecutionModel execution_model, const char **compiled_entry_point_name);
    ScInternalResult sc_internal_compiler_get_shader_resources(const ScInternalCompilerBase *compiler, ScShaderResources *shader_resources);
    ScInternalResult sc_internal_compiler_get_specialization_constants(const ScInternalCompilerBase *compiler, ScSpecializationConstant **constants, size_t *size);
    // `uint64_t` isn't supported in Emscripten without implicitly splitting the value into two `uint32_t` - instead do it explicitly
    ScInternalResult sc_internal_compiler_set_scalar_constant(const ScInternalCompilerBase *compiler, const uint32_t id, const uint32_t constant_high_bits, const uint32_t constant_low_bits);
    ScInternalResult sc_internal_compiler_get_type(const ScInternalCompilerBase *compiler, const uint32_t id, const ScType **spirv_type);
    ScInternalResult sc_internal_compiler_get_member_name(const ScInternalCompilerBase *compiler, const uint32_t id, const uint32_t index, const char **name);
    ScInternalResult sc_internal_compiler_get_member_decoration(const ScInternalCompilerBase *compiler, const uint32_t id, const uint32_t index, const spv::Decoration decoration, uint32_t *result);
    ScInternalResult sc_internal_compiler_set_member_decoration(const ScInternalCompilerBase *compiler, const uint32_t id, const uint32_t index, const spv::Decoration decoration, const uint32_t argument);
    ScInternalResult sc_internal_compiler_get_declared_struct_size(const ScInternalCompilerBase *compiler, const uint32_t id, uint32_t *result);
    ScInternalResult sc_internal_compiler_get_declared_struct_member_size(const ScInternalCompilerBase *compiler, const uint32_t id, const uint32_t index, uint32_t *result);
    ScInternalResult sc_internal_compiler_rename_interface_variable(const ScInternalCompilerBase *compiler, const ScResource *resources, const size_t resources_size, uint32_t location, const char *name);
    ScInternalResult sc_internal_compiler_get_work_group_size_specialization_constants(const ScInternalCompilerBase *compiler, ScSpecializationConstant **constants);
    ScInternalResult sc_internal_compiler_set_entry_point(const ScInternalCompilerBase *compiler, const char *name, const spv::ExecutionModel execution_model);
    ScInternalResult sc_internal_compiler_compile(const ScInternalCompilerBase *compiler, const char **shader);
    ScInternalResult sc_internal_compiler_delete(ScInternalCompilerBase *compiler);

    ScInternalResult sc_internal_free_pointer(void *pointer);
}