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

 *

 * Copyright (C) 2012 David Anderson

 *

 * This file is part of SourcePawn.

 *

 * SourcePawn 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.

 * 

 * SourcePawn 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

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

 */

#include "Zone.h"

#include "NativeTable.h"

#include "StringPool.h"

#include "Heap-inl.h"

using namespace ke;

NativeTable::NativeTable(Zone *zone)

  : zone_(zone),

    nativeMap_(ZoneAllocatorPolicy())

{

}

NativeTable::~NativeTable()

{

    for (size_t i = 0; i < natives_.length(); i++)

        delete natives_[i];

}

bool

NativeTable::initialize()

{

    return nativeMap_.init(16);

}

bool

NativeTable::add(Handle<String> name, size_t *indexp)

{

    // :TODO: assert no GC

    CharsAndLength key(name->chars(), name->length());

    NativeMap::Insert p = nativeMap_.findForAdd(key);

    if (p.found()) {

        *indexp = (*p)->index;

        return true;

    }

    NativeInfo *info = new NativeInfo(name, NULL, natives_.length());

    if (!info || !natives_.append(info))

        return false;

    if (!nativeMap_.add(p, info))

        return false;

    *indexp = info->index;

    return true;

}

bool

NativeTable::bind(const char *name, void *address)

{

    CharsAndLength key(name, strlen(name));

    NativeMap::Insert p = nativeMap_.findForAdd(key);

    if (p.found()) {

        (*p)->address = address;

        return true;

    }

    Local<String> str(zone_, zone_->symbols()->add(key.str(), key.length()));

    if (!str)

        return false;

    // :TODO: assert no GC

    NativeInfo *info = new NativeInfo(str, address, natives_.length());

    if (!info || !natives_.append(info))

        return false;

    if (!nativeMap_.add(p, info))

        return false;

    return true;

}

void

NativeTable::accept(VirtualObjectVisitor *visitor)

{

    for (size_t i = 0; i < natives_.length(); i++)

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

}