#pragma once
#if ENABLE(JIT)
#include "CodeLocation.h"
#include <wtf/StdLibExtras.h>
#include <wtf/Vector.h>
namespace JSC {
class JITCodeMap {
public:
static_assert(std::is_trivially_destructible_v<BytecodeIndex>);
static_assert(std::is_trivially_destructible_v<CodeLocationLabel<JSEntryPtrTag>>);
static_assert(alignof(CodeLocationLabel<JSEntryPtrTag>) >= alignof(BytecodeIndex), "Putting CodeLocationLabel vector first since we can avoid alignment consideration of BytecodeIndex vector");
JITCodeMap() = default;
JITCodeMap(Vector<BytecodeIndex>&& indexes, Vector<CodeLocationLabel<JSEntryPtrTag>>&& codeLocations)
: m_size(indexes.size())
{
ASSERT(indexes.size() == codeLocations.size());
m_pointer = MallocPtr<uint8_t>::malloc(sizeof(CodeLocationLabel<JSEntryPtrTag>) * m_size + sizeof(BytecodeIndex) * m_size);
std::copy(codeLocations.begin(), codeLocations.end(), this->codeLocations());
std::copy(indexes.begin(), indexes.end(), this->indexes());
}
CodeLocationLabel<JSEntryPtrTag> find(BytecodeIndex bytecodeIndex) const
{
auto* index = binarySearch<BytecodeIndex, BytecodeIndex>(indexes(), m_size, bytecodeIndex, [] (BytecodeIndex* index) { return *index; });
if (!index)
return CodeLocationLabel<JSEntryPtrTag>();
return codeLocations()[index - indexes()];
}
explicit operator bool() const { return m_size; }
private:
CodeLocationLabel<JSEntryPtrTag>* codeLocations() const
{
return bitwise_cast<CodeLocationLabel<JSEntryPtrTag>*>(m_pointer.get());
}
BytecodeIndex* indexes() const
{
return bitwise_cast<BytecodeIndex*>(m_pointer.get() + sizeof(CodeLocationLabel<JSEntryPtrTag>) * m_size);
}
MallocPtr<uint8_t> m_pointer;
unsigned m_size { 0 };
};
class JITCodeMapBuilder {
WTF_MAKE_NONCOPYABLE(JITCodeMapBuilder);
public:
JITCodeMapBuilder() = default;
void append(BytecodeIndex bytecodeIndex, CodeLocationLabel<JSEntryPtrTag> codeLocation)
{
m_indexes.append(bytecodeIndex);
m_codeLocations.append(codeLocation);
}
JITCodeMap finalize()
{
return JITCodeMap(WTFMove(m_indexes), WTFMove(m_codeLocations));
}
private:
Vector<BytecodeIndex> m_indexes;
Vector<CodeLocationLabel<JSEntryPtrTag>> m_codeLocations;
};
}
#endif // ENABLE(JIT)