// Copyright 2021 yuzu Emulator Project
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

#include "shader_recompiler/backend/spirv/emit_spirv.h"
#include "shader_recompiler/frontend/ir/modifiers.h"

namespace Shader::Backend::SPIRV {
namespace {
void EmitMemoryBarrierImpl(EmitContext& ctx, spv::Scope scope) {
    const auto semantics{
        spv::MemorySemanticsMask::AcquireRelease | spv::MemorySemanticsMask::UniformMemory |
        spv::MemorySemanticsMask::WorkgroupMemory | spv::MemorySemanticsMask::AtomicCounterMemory |
        spv::MemorySemanticsMask::ImageMemory};
    ctx.OpMemoryBarrier(ctx.Constant(ctx.U32[1], static_cast<u32>(scope)),
                        ctx.Constant(ctx.U32[1], static_cast<u32>(semantics)));
}
} // Anonymous namespace

void EmitBarrier(EmitContext& ctx) {
    const auto execution{spv::Scope::Workgroup};
    const auto memory{spv::Scope::Workgroup};
    const auto memory_semantics{spv::MemorySemanticsMask::AcquireRelease |
                                spv::MemorySemanticsMask::WorkgroupMemory};
    ctx.OpControlBarrier(ctx.Constant(ctx.U32[1], static_cast<u32>(execution)),
                         ctx.Constant(ctx.U32[1], static_cast<u32>(memory)),
                         ctx.Constant(ctx.U32[1], static_cast<u32>(memory_semantics)));
}

void EmitMemoryBarrierWorkgroupLevel(EmitContext& ctx) {
    EmitMemoryBarrierImpl(ctx, spv::Scope::Workgroup);
}

void EmitMemoryBarrierDeviceLevel(EmitContext& ctx) {
    EmitMemoryBarrierImpl(ctx, spv::Scope::Device);
}

void EmitMemoryBarrierSystemLevel(EmitContext& ctx) {
    EmitMemoryBarrierImpl(ctx, spv::Scope::CrossDevice);
}

} // namespace Shader::Backend::SPIRV