#pragma once
#if ENABLE(DFG_JIT)
#include "CodeBlock.h"
#include "CompilationResult.h"
#include "DFGCommonData.h"
#include "DFGMinifiedGraph.h"
#include "DFGOSREntry.h"
#include "DFGOSRExit.h"
#include "DFGVariableEventStream.h"
#include "ExecutionCounter.h"
#include "JITCode.h"
#include <wtf/SegmentedVector.h>
namespace JSC {
class TrackedReferences;
namespace DFG {
class JITCompiler;
class JITCode : public DirectJITCode {
public:
JITCode();
virtual ~JITCode();
CommonData* dfgCommon() override;
JITCode* dfg() override;
OSREntryData* appendOSREntryData(unsigned bytecodeIndex, unsigned machineCodeOffset)
{
DFG::OSREntryData entry;
entry.m_bytecodeIndex = bytecodeIndex;
entry.m_machineCodeOffset = machineCodeOffset;
osrEntry.append(entry);
return &osrEntry.last();
}
OSREntryData* osrEntryDataForBytecodeIndex(unsigned bytecodeIndex)
{
return tryBinarySearch<OSREntryData, unsigned>(
osrEntry, osrEntry.size(), bytecodeIndex,
getOSREntryDataBytecodeIndex);
}
unsigned appendOSRExit(const OSRExit& exit)
{
unsigned result = osrExit.size();
osrExit.append(exit);
return result;
}
OSRExit& lastOSRExit()
{
return osrExit.last();
}
unsigned appendSpeculationRecovery(const SpeculationRecovery& recovery)
{
unsigned result = speculationRecovery.size();
speculationRecovery.append(recovery);
return result;
}
void reconstruct(
CodeBlock*, CodeOrigin, unsigned streamIndex, Operands<ValueRecovery>& result);
void reconstruct(
ExecState*, CodeBlock*, CodeOrigin, unsigned streamIndex, Operands<JSValue>& result);
#if ENABLE(FTL_JIT)
bool checkIfOptimizationThresholdReached(CodeBlock*);
void optimizeNextInvocation(CodeBlock*);
void dontOptimizeAnytimeSoon(CodeBlock*);
void optimizeAfterWarmUp(CodeBlock*);
void optimizeSoon(CodeBlock*);
void forceOptimizationSlowPathConcurrently(CodeBlock*);
void setOptimizationThresholdBasedOnCompilationResult(CodeBlock*, CompilationResult);
#endif // ENABLE(FTL_JIT)
void validateReferences(const TrackedReferences&) override;
void shrinkToFit();
RegisterSet liveRegistersToPreserveAtExceptionHandlingCallSite(CodeBlock*, CallSiteIndex) override;
#if ENABLE(FTL_JIT)
CodeBlock* osrEntryBlock() { return m_osrEntryBlock.get(); }
void setOSREntryBlock(VM& vm, const JSCell* owner, CodeBlock* osrEntryBlock) { m_osrEntryBlock.set(vm, owner, osrEntryBlock); }
void clearOSREntryBlock() { m_osrEntryBlock.clear(); }
#endif
static ptrdiff_t commonDataOffset() { return OBJECT_OFFSETOF(JITCode, common); }
std::optional<CodeOrigin> findPC(CodeBlock*, void* pc) override;
private:
friend class JITCompiler;
public:
CommonData common;
Vector<DFG::OSREntryData> osrEntry;
SegmentedVector<DFG::OSRExit, 8> osrExit;
Vector<DFG::SpeculationRecovery> speculationRecovery;
DFG::VariableEventStream variableEventStream;
DFG::MinifiedGraph minifiedDFG;
#if ENABLE(FTL_JIT)
uint8_t neverExecutedEntry { 1 };
UpperTierExecutionCounter tierUpCounter;
HashMap<unsigned, Vector<unsigned>> tierUpInLoopHierarchy;
HashMap<unsigned, unsigned, WTF::IntHash<unsigned>, WTF::UnsignedWithZeroKeyHashTraits<unsigned>> bytecodeIndexToStreamIndex;
enum class TriggerReason : uint8_t {
DontTrigger,
CompilationDone,
StartCompilation,
};
HashMap<unsigned, TriggerReason> tierUpEntryTriggers;
HashSet<unsigned, WTF::IntHash<unsigned>, WTF::UnsignedWithZeroKeyHashTraits<unsigned>> tierUpEntrySeen;
WriteBarrier<CodeBlock> m_osrEntryBlock;
unsigned osrEntryRetry;
bool abandonOSREntry;
#endif // ENABLE(FTL_JIT)
};
} }
#endif // ENABLE(DFG_JIT)