/* 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());

 }