src/NativeTable.h
author David Anderson <dvander@alliedmods.net>
Sun Nov 18 15:50:41 2012 -0800 (2012-11-18)
changeset 217 c22e983b3b6e
parent 196 9904633b1dfc
child 263 ba85a47ee414
permissions -rw-r--r--
Get preinc working again.
dvander@195
     1
/* vim: set ts=4 sw=4 tw=99 et:
dvander@195
     2
 *
dvander@195
     3
 * Copyright (C) 2012 David Anderson
dvander@195
     4
 *
dvander@195
     5
 * This file is part of SourcePawn.
dvander@195
     6
 *
dvander@195
     7
 * SourcePawn is free software: you can redistribute it and/or modify it under
dvander@195
     8
 * the terms of the GNU General Public License as published by the Free
dvander@195
     9
 * Software Foundation, either version 3 of the License, or (at your option)
dvander@195
    10
 * any later version.
dvander@195
    11
 * 
dvander@195
    12
 * SourcePawn is distributed in the hope that it will be useful, but WITHOUT ANY
dvander@195
    13
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@195
    14
 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
dvander@195
    15
 *
dvander@195
    16
 * You should have received a copy of the GNU General Public License along with
dvander@195
    17
 * SourcePawn. If not, see http://www.gnu.org/licenses/.
dvander@195
    18
 */
dvander@195
    19
#ifndef _include_sp2_native_table_h_
dvander@195
    20
#define _include_sp2_native_table_h_
dvander@195
    21
dvander@196
    22
#include <string.h>
dvander@195
    23
#include "AllocatorPolicies.h"
dvander@195
    24
#include "Strings.h"
dvander@195
    25
#include "HashTable.h"
dvander@195
    26
#include "Vector.h"
dvander@195
    27
#include "VirtualObjectVisitor.h"
dvander@195
    28
dvander@195
    29
namespace ke {
dvander@195
    30
dvander@195
    31
class NativeTable
dvander@195
    32
{
dvander@195
    33
  public:
dvander@195
    34
    NativeTable(Zone *zone);
dvander@195
    35
    ~NativeTable();
dvander@195
    36
dvander@195
    37
  private:
dvander@195
    38
    struct NativeInfo {
dvander@195
    39
        Unbarriered<String> name;
dvander@195
    40
        void *address;
dvander@195
    41
        size_t index;
dvander@195
    42
dvander@195
    43
        NativeInfo(String *name, void *address, size_t index)
dvander@195
    44
          : address(address),
dvander@195
    45
            index(index)
dvander@195
    46
        {
dvander@195
    47
            this->name = name;
dvander@195
    48
        }
dvander@195
    49
    };
dvander@195
    50
dvander@195
    51
    struct Policy
dvander@195
    52
    {
dvander@195
    53
        typedef NativeInfo * Payload;
dvander@195
    54
dvander@195
    55
        static uint32 hash(const CharsAndLength &key) {
dvander@195
    56
            return HashCharSequence(key.str(), key.length());
dvander@195
    57
        }
dvander@195
    58
        
dvander@195
    59
        static bool matches(const CharsAndLength &key, NativeInfo *info) {
dvander@195
    60
            if (info->name->length() != key.length())
dvander@195
    61
                return false;
dvander@195
    62
            return memcmp(info->name->chars(), key.str(), key.length() * sizeof(char)) == 0;
dvander@195
    63
        }
dvander@195
    64
    };
dvander@195
    65
dvander@195
    66
    typedef HashTable<Policy, ZoneAllocatorPolicy> NativeMap;
dvander@195
    67
dvander@195
    68
  public:
dvander@195
    69
    bool initialize();
dvander@195
    70
dvander@195
    71
    bool bind(const char *name, void *address);
dvander@195
    72
    bool add(Handle<String> name, size_t *indexp);
dvander@195
    73
    String *name(size_t index) const {
dvander@195
    74
        return natives_[index]->name;
dvander@195
    75
    }
dvander@195
    76
    void *address(size_t index) const {
dvander@195
    77
        return natives_[index]->address;
dvander@195
    78
    }
dvander@195
    79
dvander@195
    80
    void accept(VirtualObjectVisitor *visitor);
dvander@195
    81
dvander@195
    82
  private:
dvander@195
    83
    Zone *zone_;
dvander@195
    84
    NativeMap nativeMap_;
dvander@195
    85
    Vector<NativeInfo *> natives_;
dvander@195
    86
};
dvander@195
    87
dvander@195
    88
}
dvander@195
    89
dvander@195
    90
#endif // _include_sp2_native_table_h_