/* 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 <stdlib.h>

#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()

  : stack_(thisFromCtor()),

    heap_(thisFromCtor()),

    types_(NULL),

    symbols_(NULL),

    globalRoots_(NULL),

    lastLocalRoot_(NULL),

    hasPendingException_(false),

    reporter_(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_Native] = allocateMap(MapKind_Native, 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));

}

bool

Zone::hasPendingException()

{

    return hasPendingException_;

}

void

Zone::reportOutOfMemory()

{

}

void

Zone::reportError(Message message, ...)

{

    va_list ap;

    va_start(ap, message);

    reportErrorVA(message, ap);

    va_end(ap);

}

#if defined(_MSC_VER)

# define VA_COPY(to, from)  to = from

#else

# define VA_COPY(to, from)  va_copy(to, from)

#endif

static char *

BuildErrorMessage(const char *format, va_list ap)

{

    va_list use;

    VA_COPY(use, ap);

    // We over-allocate and zero terminate ahead of time, since LIBCRT's

    // version of vsnprintf is screwy.

    size_t size = 255;

    char *buffer = (char *)calloc(size + 1, 1);

    if (!buffer)

        return NULL;

    for (;;) {

        int r = vsnprintf(buffer, size, format, use);

        if (r > -1 && size_t(r) < size)

            return buffer;

#if defined(_MSC_VER)

        if (r < 0)

            size *= 2;

#else

        if (r > -1)

            size = size_t(r) + 1;

        else

            size *= 2;

#endif

        free(buffer);

        if ((buffer = (char *)calloc(size + 1, 1)) == NULL)

            return NULL;

        VA_COPY(use, ap);

    }

}

void

Zone::reportErrorVA(Message message, va_list ap)

{

    char *report = BuildErrorMessage(Messages[message].format, ap);

    if (!report) {

        reportOutOfMemory();

        return;

    }

    if (reporter_) {

        // Reporters can re-enter. When we start having actual exception

        // objects that can be caught and inspected, this code will have to

        // save and restore the exception state.

        reporter_->onUncaughtException(report);

    }

    hasPendingException_ = true;

    free(report);

}

void

Zone::beginCompilationErrors(const char *file)

{

    if (reporter_)

        reporter_->beginCompilationErrors(file);

}

void

Zone::reportCompilationError(Message message, unsigned line, unsigned col, va_list ap)

{

    char *report = BuildErrorMessage(Messages[message].format, ap);

    if (!report) {

        reportOutOfMemory();

        return;

    }

    if (reporter_) {

        // Reporters can re-enter. When we start having actual exception

        // objects that can be caught and inspected, this code will have to

        // save and restore the exception state.

        reporter_->reportCompilationError(report, line, col);

    }

    free(report);

}

void

Zone::endCompilationErrors()

{

    if (reporter_)

        reporter_->endCompilationErrors();

}

void

Zone::installErrorReporter(Ke::IMessageReporter *reporter)

{

    reporter_ = reporter;

}

void

ZoneAllocatorPolicy::reportOutOfMemory()

{

    ZONE()->reportOutOfMemory();

}

void

ZoneAllocatorPolicy::reportAllocationOverflow()

{

    ZONE()->reportError(Message_AllocationOverflow);

}