// SPDX-FileCopyrightText: Copyright 2024 shadPS4 Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later

#pragma once

#include <boost/container/flat_map.hpp>
#include "shader_recompiler/pools.h"
#include "shader_recompiler/ir/basic_block.h"
#include "shader_recompiler/ir/program.h"

namespace Shader::IR {

struct SubProgram {
    SubProgram(Program* super_program, Pools& pools);

    Block* AddBlock(Block* orig_block);
    Inst* AddInst(Inst* orig_inst);

    Block* GetBlock(Block* orig_block);
    Inst* GetInst(Inst* orig_inst);

    Program GetSubProgram();
private:
    void AddPhi(Inst* orig_phi, Inst* phi);

    void SetArg(Inst* inst, size_t index, const Value& arg);
    void AddPhiOperand(Inst* phi, Block* block, const Value& arg);

    void BuildBlockListAndASL(Program& sub_program);

    Program* super_program;
    Pools& pools;
    boost::container::flat_map<Block*, Block*> orig_block_to_block;
    boost::container::flat_map<Inst*, Inst*> orig_inst_to_inst;
};

} // namespace Shader::IR