luau/CodeGen/include/Luau/IrRegAllocX64.h

// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
#pragma once

#include "Luau/IrData.h"
#include "Luau/RegisterX64.h"

#include <array>
#include <initializer_list>

namespace Luau
{
namespace CodeGen
{
namespace X64
{

struct IrRegAllocX64
{
    IrRegAllocX64(IrFunction& function);

    RegisterX64 allocGprReg(SizeX64 preferredSize);
    RegisterX64 allocXmmReg();

    RegisterX64 allocGprRegOrReuse(SizeX64 preferredSize, uint32_t index, std::initializer_list<IrOp> oprefs);
    RegisterX64 allocXmmRegOrReuse(uint32_t index, std::initializer_list<IrOp> oprefs);

    RegisterX64 takeReg(RegisterX64 reg);

    void freeReg(RegisterX64 reg);
    void freeLastUseReg(IrInst& target, uint32_t index);
    void freeLastUseRegs(const IrInst& inst, uint32_t index);

    bool isLastUseReg(const IrInst& target, uint32_t index) const;

    bool shouldFreeGpr(RegisterX64 reg) const;

    void assertFree(RegisterX64 reg) const;
    void assertAllFree() const;

    IrFunction& function;

    std::array<bool, 16> freeGprMap;
    std::array<bool, 16> freeXmmMap;
};

struct ScopedRegX64
{
    explicit ScopedRegX64(IrRegAllocX64& owner);
    ScopedRegX64(IrRegAllocX64& owner, SizeX64 size);
    ScopedRegX64(IrRegAllocX64& owner, RegisterX64 reg);
    ~ScopedRegX64();

    ScopedRegX64(const ScopedRegX64&) = delete;
    ScopedRegX64& operator=(const ScopedRegX64&) = delete;

    void alloc(SizeX64 size);
    void free();

    RegisterX64 release();

    IrRegAllocX64& owner;
    RegisterX64 reg;
};

} // namespace X64
} // namespace CodeGen
} // namespace Luau