ClangInterpreter.cpp

Go to the documentation of this file.

#include "ClangInterpreter.hpp"
 
#include "Commentator.hpp"
 
#include "MayaFlux/Transitive/Platform/HostEnvironment.hpp"
 
#ifdef MAYAFLUX_COMPILER_MSVC
#pragma warning(push)
#pragma warning(disable : 4291) // no matching operator delete found
#pragma warning(disable : 4805) // unsafe mix of type 'unsigned int' and type 'bool' in operation
#endif
 
#include <llvm/ExecutionEngine/Orc/LLJIT.h>
#include <llvm/ExecutionEngine/Orc/ThreadSafeModule.h>
 
#include <llvm/Support/TargetSelect.h>
#include <llvm/TargetParser/Host.h>
 
#include <clang/AST/Type.h>
#include <clang/Frontend/CompilerInstance.h>
#include <clang/Interpreter/Interpreter.h>
 
#ifdef MAYAFLUX_COMPILER_MSVC
#pragma warning(pop)
#endif
 
#ifdef MAYAFLUX_PLATFORM_MACOS
#include <dispatch/dispatch.h>
#endif
 
#ifdef MAYAFLUX_PLATFORM_WINDOWS
#include "MayaFlux/Transitive/Parallel/Dispatch.hpp"
#endif
 
namespace Lila {
 
struct ClangInterpreter::Impl {
    std::unique_ptr<clang::Interpreter> interpreter;
 
    std::vector<std::string> include_paths;
    std::vector<std::string> compile_flags;
    std::string target_triple;
 
    std::unordered_map<std::string, void*> symbol_table;
    int eval_counter = 0;
};
 
ClangInterpreter::ClangInterpreter()
    : m_impl(std::make_unique<Impl>())
{
    m_impl->target_triple = llvm::sys::getDefaultTargetTriple();
}
 
ClangInterpreter::~ClangInterpreter()
{
    shutdown();
}
 
ClangInterpreter::ClangInterpreter(ClangInterpreter&&) noexcept = default;
ClangInterpreter& ClangInterpreter::operator=(ClangInterpreter&&) noexcept = default;
 
bool ClangInterpreter::initialize(bool skip_host_library_load)
{
    LILA_INFO(Emitter::INTERPRETER, "Initializing Clang interpreter");
 
#ifdef MAYAFLUX_PLATFORM_WINDOWS
    llvm::sys::DynamicLibrary::LoadLibraryPermanently("msvcp140.dll");
    llvm::sys::DynamicLibrary::LoadLibraryPermanently("vcruntime140.dll");
    llvm::sys::DynamicLibrary::LoadLibraryPermanently("ucrtbase.dll");
#endif
 
    llvm::InitializeNativeTarget();
    llvm::InitializeNativeTargetAsmPrinter();
    llvm::InitializeNativeTargetAsmParser();
 
    m_impl->compile_flags.clear();
    m_impl->compile_flags.emplace_back("-std=c++23");
    m_impl->compile_flags.emplace_back("-DMAYASIMPLE");
 
#ifdef MAYAFLUX_PLATFORM_LINUX
    m_impl->compile_flags.emplace_back("-mcmodel=large");
    m_impl->compile_flags.emplace_back("-fPIC");
    m_impl->compile_flags.emplace_back("-fPIE");
#endif
 
    std::string pch_dir;
    if (std::filesystem::exists(MayaFlux::Config::PCH_RUNTIME_PATH)) {
        pch_dir = MayaFlux::Config::RUNTIME_DATA_DIR;
        LILA_DEBUG(Emitter::INTERPRETER,
            std::string("Using installed PCH from: ") + std::string(MayaFlux::Config::PCH_RUNTIME_PATH));
 
    } else if (std::filesystem::exists(MayaFlux::Config::PCH_SOURCE_PATH)) {
        pch_dir = std::string(MayaFlux::Config::SOURCE_DIR) + "/cmake";
        LILA_DEBUG(Emitter::INTERPRETER,
            std::string("Using source PCH from: ") + std::string(MayaFlux::Config::PCH_SOURCE_PATH));
 
    } else {
        m_last_error = "Cannot find pch.h in runtime or source locations";
        LILA_ERROR(Emitter::INTERPRETER, m_last_error);
        return false;
    }
 
    m_impl->compile_flags.push_back("-I" + pch_dir);
 
    const std::string& resource_dir = MayaFlux::Platform::SystemConfig::get_clang_resource_dir();
    if (!resource_dir.empty()) {
        m_impl->compile_flags.push_back("-resource-dir=" + resource_dir);
        LILA_DEBUG(Emitter::INTERPRETER,
            std::string("Using clang resource dir: ") + resource_dir);
    } else {
        m_impl->compile_flags.emplace_back("-resource-dir=/usr/lib/clang/21");
        LILA_WARN(Emitter::INTERPRETER,
            "Using default clang resource dir: /usr/lib/clang/21");
    }
 
    const auto& system_includes = MayaFlux::Platform::SystemConfig::get_system_includes();
    for (const auto& include : system_includes) {
        m_impl->compile_flags.push_back("-isystem" + include);
    }
 
#ifndef MAYAFLUX_PLATFORM_WINDOWS
    const auto& eigen_include = MayaFlux::Platform::SystemConfig::get_dep_includes("eigen");
    if (!eigen_include.empty()) {
        m_impl->compile_flags.push_back("-isystem" + eigen_include);
        LILA_DEBUG(Emitter::INTERPRETER,
            std::string("Added Eigen include path: ") + eigen_include);
    } else {
        LILA_WARN(Emitter::INTERPRETER,
            "Could not find Eigen include path - some features may not work");
    }
 
    const auto& freetype_include = MayaFlux::Platform::SystemConfig::get_dep_includes("freetype2");
    if (!freetype_include.empty()) {
        m_impl->compile_flags.push_back("-isystem" + freetype_include);
        LILA_DEBUG(Emitter::INTERPRETER,
            std::string("Added FreeType include path: ") + freetype_include);
    } else {
        LILA_WARN(Emitter::INTERPRETER,
            "Could not find FreeType include path - some features may not work");
    }
#endif
 
#ifdef MAYAFLUX_PLATFORM_MACOS
    // CRITICAL: JIT uses Homebrew LLVM but needs macOS SDK for system headers
    // Homebrew LLVM's libc++ requires pthread.h, sched.h, time.h, etc. from SDK
    std::string sdk_path = MayaFlux::Platform::SystemConfig::get_macos_sdk_path();
    if (!sdk_path.empty()) {
        m_impl->compile_flags.push_back("-isysroot" + sdk_path);
        LILA_DEBUG(Emitter::INTERPRETER, "Using macOS SDK: " + sdk_path);
    } else {
        LILA_WARN(Emitter::INTERPRETER,
            "Could not find macOS SDK - JIT may fail to find system headers");
    }
#endif
 
    for (const auto& path : m_impl->include_paths) {
        m_impl->compile_flags.push_back("-I" + path);
    }
 
#ifdef MAYAFLUX_PLATFORM_WINDOWS
    m_impl->compile_flags.emplace_back("-fno-function-sections");
    m_impl->compile_flags.emplace_back("-fno-data-sections");
    m_impl->compile_flags.emplace_back("-fno-unique-section-names");
#endif
 
    std::vector<const char*> args;
    for (const auto& flag : m_impl->compile_flags) {
        args.push_back(flag.c_str());
    }
 
    clang::IncrementalCompilerBuilder ICB;
    ICB.SetCompilerArgs(args);
 
    auto CI = ICB.CreateCpp();
    if (!CI) {
        m_last_error = "Failed to create CompilerInstance: " + llvm::toString(CI.takeError());
        LILA_ERROR(Emitter::INTERPRETER, m_last_error);
        return false;
    }
 
    auto interp = clang::Interpreter::create(std::move(*CI));
    if (!interp) {
        m_last_error = "Failed to create interpreter: " + llvm::toString(interp.takeError());
        LILA_ERROR(Emitter::INTERPRETER, m_last_error);
        return false;
    }
 
    m_impl->interpreter = std::move(*interp);
 
    LILA_INFO(Emitter::INTERPRETER, "Clang interpreter created successfully");
 
    if (skip_host_library_load) {
        LILA_DEBUG(Emitter::INTERPRETER,
            "Skipping MayaFluxLib load: host process already resident");
    } else {
        const auto& lib_to_load = MayaFlux::Platform::SystemConfig::find_dep_library(
            "MayaFluxLib", std::string(MayaFlux::Config::INSTALL_PREFIX));
 
        LILA_DEBUG(Emitter::INTERPRETER, "Loading MayaFlux library: " + lib_to_load);
 
        if (auto err = m_impl->interpreter->LoadDynamicLibrary(lib_to_load.c_str())) {
            m_last_error = "Failed to load " + lib_to_load + ": "
                + llvm::toString(std::move(err));
            LILA_ERROR(Emitter::INTERPRETER, m_last_error);
            return false;
        }
 
        LILA_DEBUG(Emitter::INTERPRETER, "Loaded " + lib_to_load);
    }
 
    const char* header_include = skip_host_library_load
        ? "#include \"pch.h\"\n"
          "#include \"MayaFlux/MayaFlux.hpp\"\n"
        : "#include \"pch.h\"\n"
          "#include \"Lila/LiveAid.hpp\"\n";
 
    auto result = m_impl->interpreter->ParseAndExecute(header_include);
 
    if (result) {
        std::string warning = "Failed to load MayaFlux headers: " + llvm::toString(std::move(result));
        LILA_WARN(Emitter::INTERPRETER, warning);
    } else {
        LILA_INFO(Emitter::INTERPRETER, "MayaFlux headers loaded successfully");
    }
 
    result = m_impl->interpreter->ParseAndExecute("std::cout << \"Ready for Live\" << std::flush;");
 
    return true;
}
 
ClangInterpreter::EvalResult ClangInterpreter::eval(const std::string& code)
{
#ifdef MAYAFLUX_PLATFORM_MACOS
    __block EvalResult result;
#else
    EvalResult result;
#endif
 
    if (!m_impl->interpreter) {
        result.error = "Interpreter not initialized";
        LILA_ERROR(Emitter::INTERPRETER, result.error);
        return result;
    }
 
    LILA_DEBUG(Emitter::INTERPRETER, "Evaluating code...");
 
#ifdef MAYAFLUX_PLATFORM_MACOS
    dispatch_sync(dispatch_get_main_queue(), ^{
        auto eval_result = m_impl->interpreter->ParseAndExecute(code);
 
        if (!eval_result) {
            result.success = true;
            LILA_DEBUG(Emitter::INTERPRETER, "Code evaluation succeeded");
        } else {
            result.success = false;
            result.error = "Execution failed: " + llvm::toString(std::move(eval_result));
            LILA_ERROR(Emitter::INTERPRETER, result.error);
        }
    });
 
#elif defined(MAYAFLUX_PLATFORM_WINDOWS)
    auto completed = std::make_shared<std::atomic<bool>>(false);
 
    auto* task = static_cast<std::function<void()>*>(
        HeapAlloc(GetProcessHeap(), 0, sizeof(std::function<void()>)));
    new (task) std::function<void()>([&, completed]() {
        auto eval_result = m_impl->interpreter->ParseAndExecute(code);
 
        if (!eval_result) {
            result.success = true;
            LILA_DEBUG(Emitter::INTERPRETER, "Code evaluation succeeded");
        } else {
            result.success = false;
            result.error = "Execution failed: " + llvm::toString(std::move(eval_result));
            LILA_ERROR(Emitter::INTERPRETER, result.error);
        }
 
        completed->store(true, std::memory_order_release);
    });
 
    PostThreadMessage(MayaFlux::Parallel::g_MainThreadId, MAYAFLUX_WM_DISPATCH, 0, (LPARAM)task);
 
    while (!completed->load(std::memory_order_acquire)) {
        std::this_thread::sleep_for(std::chrono::microseconds(100));
    }
 
#else
    auto eval_result = m_impl->interpreter->ParseAndExecute(code);
 
    if (!eval_result) {
        result.success = true;
        LILA_DEBUG(Emitter::INTERPRETER, "Code evaluation succeeded");
    } else {
        result.success = false;
        result.error = "Execution failed: " + llvm::toString(std::move(eval_result));
        LILA_ERROR(Emitter::INTERPRETER, result.error);
    }
#endif
 
    m_impl->eval_counter++;
    return result;
}
 
ClangInterpreter::EvalResult ClangInterpreter::eval_file(const std::string& filepath)
{
    EvalResult result;
 
    if (!std::filesystem::exists(filepath)) {
        result.error = "File does not exist: " + filepath;
        LILA_ERROR(Emitter::INTERPRETER, result.error);
        return result;
    }
 
    LILA_INFO(Emitter::INTERPRETER, std::string("Evaluating file: ") + filepath);
 
    std::string code = "#include \"" + filepath + "\"\n";
    return eval(code);
}
 
void* ClangInterpreter::get_symbol_address(const std::string& name)
{
    if (!m_impl->interpreter) {
        LILA_WARN(Emitter::INTERPRETER, "Cannot get symbol: interpreter not initialized");
        return nullptr;
    }
 
    auto symbol = m_impl->interpreter->getSymbolAddress(name);
    if (symbol) {
        LILA_DEBUG(Emitter::INTERPRETER, std::string("Found symbol: ") + name);
        return reinterpret_cast<void*>(symbol->getValue());
    }
 
    LILA_WARN(Emitter::INTERPRETER, std::string("Symbol not found: ") + name);
    return nullptr;
}
 
std::vector<std::string> ClangInterpreter::get_defined_symbols()
{
    std::vector<std::string> symbols;
    for (const auto& pair : m_impl->symbol_table) {
        symbols.push_back(pair.first);
    }
    return symbols;
}
 
void ClangInterpreter::add_include_path(const std::string& path)
{
    if (std::filesystem::exists(path)) {
        m_impl->include_paths.push_back(path);
        LILA_DEBUG(Emitter::INTERPRETER, std::string("Added include path: ") + path);
    } else {
        LILA_WARN(Emitter::INTERPRETER,
            std::string("Include path does not exist: ") + path);
    }
}
 
void ClangInterpreter::add_library_path(const std::string& path)
{
    LILA_DEBUG(Emitter::INTERPRETER,
        std::string("Library path noted (not yet implemented): ") + path);
}
 
void ClangInterpreter::add_compile_flag(const std::string& flag)
{
    m_impl->compile_flags.push_back(flag);
    LILA_DEBUG(Emitter::INTERPRETER, std::string("Added compile flag: ") + flag);
}
 
void ClangInterpreter::set_target_triple(const std::string& triple)
{
    m_impl->target_triple = triple;
    LILA_INFO(Emitter::INTERPRETER, std::string("Target triple set to: ") + triple);
}
 
void ClangInterpreter::reset()
{
    LILA_INFO(Emitter::INTERPRETER, "Resetting interpreter");
    shutdown();
    m_impl = std::make_unique<Impl>();
}
 
void ClangInterpreter::shutdown()
{
    if (m_impl->interpreter) {
        LILA_INFO(Emitter::INTERPRETER, "Shutting down interpreter");
        m_impl->interpreter.reset();
    }
}
 
} // namespace Lila