#pragma once
#if ENABLE(B3_JIT)
#if ENABLE(MASM_PROBE)
#include "AirInst.h"
#include "AirSpecial.h"
#include "MacroAssemblerPrinter.h"
namespace JSC {
namespace Printer {
typedef Vector<B3::Air::Arg> ArgList;
template<typename T, typename U>
static constexpr auto IsSameOrReferenceHelper(int) -> std::enable_if_t<std::is_same<T, U>::value || std::is_same<T, U&>::value, std::true_type>;
template<typename T, typename U>
static constexpr std::false_type IsSameOrReferenceHelper(...);
template<class T, typename U>
struct IsSameOrReference : public std::is_same<decltype(IsSameOrReferenceHelper<T, U>(0)), std::true_type> { };
template<typename T, typename... Arguments, typename = std::enable_if_t<IsSameOrReference<T, B3::Air::Tmp>::value || IsSameOrReference<T, Reg>::value>>
inline void appendAirArg(B3::Air::Inst& inst, T&& arg)
{
inst.args.append(std::forward<T>(arg));
}
template<typename T, typename... Arguments, typename = std::enable_if_t<!IsSameOrReference<T, B3::Air::Tmp>::value && !IsSameOrReference<T, Reg>::value>>
inline void appendAirArg(B3::Air::Inst&, T&&, int = 0) { }
inline void appendAirArgs(B3::Air::Inst&) { }
template<typename T, typename... Arguments>
inline void appendAirArgs(B3::Air::Inst& inst, T&& t, Arguments&&... others)
{
appendAirArg(inst, std::forward<T>(t));
appendAirArgs(inst, std::forward<Arguments>(others)...);
}
void printAirArg(PrintStream&, Context&);
template<>
struct Printer<B3::Air::Tmp> : public PrintRecord {
Printer(B3::Air::Tmp&)
: PrintRecord(printAirArg)
{ }
};
template<>
struct Printer<Reg> : public PrintRecord {
Printer(Reg&)
: PrintRecord(printAirArg)
{ }
};
}
namespace B3 { namespace Air {
class PrintSpecial : public Special {
public:
PrintSpecial(Printer::PrintRecordList*);
~PrintSpecial();
static const GPRReg scratchRegister = GPRInfo::nonArgGPR0;
protected:
void forEachArg(Inst&, const ScopedLambda<Inst::EachArgCallback>&) final;
bool isValid(Inst&) final;
bool admitsStack(Inst&, unsigned argIndex) final;
bool admitsExtendedOffsetAddr(Inst&, unsigned) final;
void reportUsedRegisters(Inst&, const RegisterSet&) final;
CCallHelpers::Jump generate(Inst&, CCallHelpers&, GenerationContext&) final;
RegisterSet extraEarlyClobberedRegs(Inst&) final;
RegisterSet extraClobberedRegs(Inst&) final;
void dumpImpl(PrintStream&) const final;
void deepDumpImpl(PrintStream&) const final;
private:
static const unsigned specialArgOffset = 0;
static const unsigned numSpecialArgs = 1;
static const unsigned calleeArgOffset = numSpecialArgs;
static const unsigned numCalleeArgs = 1;
static const unsigned returnGPArgOffset = numSpecialArgs + numCalleeArgs;
static const unsigned numReturnGPArgs = 2;
static const unsigned returnFPArgOffset = numSpecialArgs + numCalleeArgs + numReturnGPArgs;
static const unsigned numReturnFPArgs = 1;
static constexpr unsigned argArgOffset =
numSpecialArgs + numCalleeArgs + numReturnGPArgs + numReturnFPArgs;
Printer::PrintRecordList* m_printRecordList;
};
} } }
#endif // ENABLE(MASM_PROBE)
#endif // ENABLE(B3_JIT)