mirror of https://github.com/RPCS3/rpcs3.git
74 lines
1.3 KiB
C++
74 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#ifdef LLVM_AVAILABLE
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
#include "types.h"
|
|
|
|
#include "restore_new.h"
|
|
#ifdef _MSC_VER
|
|
#pragma warning(push, 0)
|
|
#endif
|
|
#include "llvm/IR/LLVMContext.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/ExecutionEngine/ExecutionEngine.h"
|
|
#ifdef _MSC_VER
|
|
#pragma warning(pop)
|
|
#endif
|
|
#include "define_new_memleakdetect.h"
|
|
|
|
extern llvm::LLVMContext g_llvm_ctx;
|
|
|
|
// Temporary compiler interface
|
|
class jit_compiler final
|
|
{
|
|
// Execution instance
|
|
std::unique_ptr<llvm::ExecutionEngine> m_engine;
|
|
|
|
// Compiled functions
|
|
std::unordered_map<std::string, std::uintptr_t> m_map;
|
|
|
|
// Linkage cache
|
|
std::unordered_map<std::string, std::uintptr_t> m_link;
|
|
|
|
// Arch
|
|
std::string m_cpu;
|
|
|
|
// Internal
|
|
void init();
|
|
|
|
public:
|
|
jit_compiler(std::unordered_map<std::string, std::uintptr_t>);
|
|
~jit_compiler();
|
|
|
|
// Compile module
|
|
void make(std::unique_ptr<llvm::Module>, std::string);
|
|
|
|
// Load object
|
|
void load(std::unique_ptr<llvm::Module>, std::unique_ptr<llvm::object::ObjectFile>);
|
|
|
|
// Get compiled function address
|
|
std::uintptr_t get(const std::string& name) const
|
|
{
|
|
const auto found = m_map.find(name);
|
|
|
|
if (found != m_map.end())
|
|
{
|
|
return found->second;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
// Get CPU info
|
|
const std::string& cpu() const
|
|
{
|
|
return m_cpu;
|
|
}
|
|
};
|
|
|
|
#endif
|