/* vim: set ts=4 sw=4 tw=99 et:

 *

 * Copyright (C) 2012 David Anderson

 *

 * This file is part of JITCraft.

 *

 * JITCraft is free software: you can redistribute it and/or modify it under

 * the terms of the GNU General Public License as published by the Free

 * Software Foundation, either version 3 of the License, or (at your option)

 * any later version.

 * 

 * Foobar is distributed in the hope that it will be useful, but WITHOUT ANY

 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License along with

 * JITCraft. If not, see http://www.gnu.org/licenses/.

 */

#include "Zone.h"

#include "StringPool.h"

#include "TypeManager.h"

#include "GlobalRoots.h"

#include "AllocatorPolicies.h"

#include "Array.h"

using namespace ke;

Zone *ke::SingleThreadedZone;

Zone::Zone()

  : heap_(this),

    types_(NULL),

    symbols_(NULL),

    globalRoots_(NULL),

    lastLocalRoot_(NULL)

{

}

Zone::~Zone()

{

    delete types_;

    delete symbols_;

    delete globalRoots_;

}

bool

Zone::initialize()

{

    if (!stack_.initialize())

        return false;

    if (!heap_.initialize())

        return false;

    symbols_ = new StringPool(this);

    if (!symbols_ || !symbols_->initialize())

        return false;

    globalRoots_ = new GlobalRoots(this);

    if (!globalRoots_ || !globalRoots_->initialize())

        return false;

    maps_[MapKind_Meta] = heap_.metaMap();

    if ((maps_[MapKind_Type] = allocateMap(MapKind_Type, sizeof(Map))) == NULL)

        return false;

    if ((maps_[MapKind_FixedArray] = allocateMap(MapKind_FixedArray, sizeof(Map))) == NULL)

        return false;

    if ((maps_[MapKind_ByteArray] = allocateMap(MapKind_ByteArray, sizeof(Map))) == NULL)

        return false;

    if ((maps_[MapKind_Code] = allocateMap(MapKind_Code, sizeof(Map))) == NULL)

        return false;

    if ((maps_[MapKind_Reference] = allocateMap(MapKind_Reference, sizeof(Map))) == NULL)

        return false;

    if ((maps_[MapKind_MetaArray] = allocateMap(MapKind_MetaArray, sizeof(Map))) == NULL)

        return false;

    if ((maps_[MapKind_Module] = allocateMap(MapKind_Module, sizeof(Map))) == NULL)

        return false;

    if ((maps_[MapKind_Function] = allocateMap(MapKind_Function, sizeof(Map))) == NULL)

        return false;

    if ((maps_[MapKind_String] = allocateMap(MapKind_String, sizeof(Map))) == NULL)

        return false;

    if ((maps_[MapKind_FreeRegion] = allocateMap(MapKind_FreeRegion, sizeof(Map))) == NULL)

        return false;

    types_ = new TypeManager(this);

    if (!types_ || !types_->initialize())

        return false;

    // Create the cached array maps for generic arrays. This is for the API

    // version of Invoke(), which needs to make arrays very quickly from C++.

    {

        // Generic "cell" array (contents do not matter, just non-gc things).

        Local<Type> type(this, types_->getPrimitive(PrimitiveType_Int32));

        if (!type)

            return false;

        type = types_->unsafeArray(type, Type::UNKNOWN_SIZE, false);

        if (!type)

            return false;

        Local<ArrayMap> contained(this);

        genericArrayMap_ = ArrayMap::New(this, type, contained);

        if (!genericArrayMap_)

            return false;

        type = types_->getPrimitive(PrimitiveType_Char);

        if (!type)

            return false;

        type = types_->unsafeArray(type, Type::UNKNOWN_SIZE, false);

        if (!type)

            return false;

        charArrayMap_ = ArrayMap::New(this, type, contained);

        if (!charArrayMap_)

            return false;

    }

    return true;

}

Map *

Zone::allocateMap(MapKind kind, size_t bytes, MapKind parent)

{

    assert(maps_[parent]);

    Map *map = heap_.allocateMap(kind, bytes);

    if (!map)

        return NULL;

    map->init(maps_[parent]);

    return map;

}

void

Zone::accept(VirtualObjectVisitor *visitor)

{

    for (unsigned i = 0; i < MapKinds_Total; i++)

        visitor->visit(maps_[i].address());

    visitor->visit(charArrayMap_.address());

    visitor->visit(genericArrayMap_.address());

    types()->accept(visitor);

    Local<Object> *local = lastLocalRoot_;

    while (local) {

        visitor->visit(local->address());

        local = local->prev();

    }

    globalRoots_->accept(visitor);

    rootScopes_.accept(visitor);

    stack_.accept(visitor);

}

String *

Zone::makeSymbol(const char *str, size_t length)

{

    return symbols_->add(str, length);

}

String *

Zone::makeSymbol(const char *str)

{

    return makeSymbol(str, strlen(str));

}

void

ZoneAllocatorPolicy::reportOutOfMemory()

{

    // :TODO: report

    assert(false);

}

void

ZoneAllocatorPolicy::reportAllocationOverflow()

{

    // :TODO: report

    assert(false);

}