pz.cpp

/*
 * Plasma in-memory representation
 * vim: ts=4 sw=4 et
 *
 * Copyright (C) Plasma Team
 * Distributed under the terms of the MIT license, see ../LICENSE.code
 */

#include "pz_common.h"

#include <stdio.h>
#include <string.h>

#include "pz_code.h"
#include "pz_data.h"
#include "pz_interp.h"

#include "pz.h"

namespace pz {
/*
 * PZ Programs
 *************/

PZ::PZ(const Options & options, Heap & heap)
    : AbstractGCTracer(heap)
    , m_options(options)
    , m_program(nullptr)
{}

// Defined here rather than the header even though it's a default destructor
// so that it can access the heap destructor.
PZ::~PZ() {}

Library * PZ::new_library(const String name, GCCapability & gc_cap)
{
    assert(!m_libraries[name]);
    m_libraries[name] = new (gc_cap) Library();
    return m_libraries[name];
}

void PZ::add_library(const String name, Library * library)
{
    assert(!m_libraries[name]);
    m_libraries[name] = library;
}

Library * PZ::lookup_library(const String name)
{
    auto iter = m_libraries.find(name);

    if (iter != m_libraries.end()) {
        return iter->second;
    } else {
        return nullptr;
    }
}

void PZ::add_program_lib(Library * program)
{
    assert(nullptr == m_program);
    m_program = program;
}

void PZ::do_trace(HeapMarkState * marker) const
{
    for (auto m : m_libraries) {
        marker->mark_root(m.first.ptr());
        marker->mark_root(m.second);
    }
    if (m_program) {
        marker->mark_root(m_program);
    }
}

}  // namespace pz