src/TypeManager.cpp
author David Anderson <dvander@alliedmods.net>
Tue Dec 25 18:13:02 2012 -0800 (2012-12-25)
changeset 240 e3c851a14f16
parent 235 43287eef2fff
child 241 18f3010a358c
permissions -rw-r--r--
Add weak hashtable to manage qual types.
     1 /* vim: set ts=4 sw=4 tw=99 et:
     2  *
     3  * Copyright (C) 2012 David Anderson
     4  *
     5  * This file is part of JITCraft.
     6  *
     7  * JITCraft is free software: you can redistribute it and/or modify it under
     8  * the terms of the GNU General Public License as published by the Free
     9  * Software Foundation, either version 3 of the License, or (at your option)
    10  * any later version.
    11  * 
    12  * Foobar is distributed in the hope that it will be useful, but WITHOUT ANY
    13  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
    14  * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
    15  *
    16  * You should have received a copy of the GNU General Public License along with
    17  * JITCraft. If not, see http://www.gnu.org/licenses/.
    18  */
    19 #include <string.h>
    20 #include "Types.h"
    21 #include "TypeManager.h"
    22 #include "Zone.h"
    23 #include "Strings.h"
    24 #include "Spaces-inl.h"
    25 #include "Heap-inl.h"
    26 
    27 using namespace ke;
    28 
    29 
    30 TypeManager::TypeManager(Zone *zone)
    31   : zone_(zone),
    32     qualTable_(ZoneAllocatorPolicy())
    33 {
    34 }
    35 
    36 bool
    37 TypeManager::initialize()
    38 {
    39     voidType_ = Type::NewVoid(zone_);
    40     if (!voidType_)
    41         return false;
    42 
    43     uncheckedType_ = Type::NewUnchecked(zone_);
    44     if (!uncheckedType_)
    45         return false;
    46 
    47     stringType_ = Type::NewString(zone_);
    48     if (!stringType_)
    49         return false;
    50 
    51     if ((primitiveTypes_[PrimitiveType_Int32] = Type::NewPrimitive(zone_, PrimitiveType_Int32)) == NULL)
    52         return false;
    53     if ((primitiveTypes_[PrimitiveType_Float] = Type::NewPrimitive(zone_, PrimitiveType_Float)) == NULL)
    54         return false;
    55     if ((primitiveTypes_[PrimitiveType_Native] = Type::NewPrimitive(zone_, PrimitiveType_Native)) == NULL)
    56         return false;
    57     if ((primitiveTypes_[PrimitiveType_Char] = Type::NewPrimitive(zone_, PrimitiveType_Char)) == NULL)
    58         return false;
    59     if ((primitiveTypes_[PrimitiveType_Bool] = Type::NewPrimitive(zone_, PrimitiveType_Bool)) == NULL)
    60         return false;
    61 
    62     Local<Type> type(zone_);
    63     for (unsigned i = 0; i < PrimitiveTypes_Total; i++) {
    64         type = primitiveTypes_[i];
    65         if ((referenceTypes_[i] = ReferenceType::New(zone_, type)) == NULL)
    66             return false;
    67     }
    68 
    69     if (!qualTable_.init(32))
    70         return false;
    71 
    72     return true;
    73 }
    74 
    75 ArrayType *
    76 TypeManager::newArray(Handle<Type> contained, int elements)
    77 {
    78     // :TODO: cache this.
    79     return ArrayType::New(zone_, contained, elements);
    80 }
    81 
    82 ArrayType *
    83 TypeManager::newExternalArray(Handle<Type> contained)
    84 {
    85     return ArrayType::NewExternal(zone_, contained);
    86 }
    87 
    88 ReferenceType *
    89 TypeManager::getReference(Handle<Type> type, bool isConst)
    90 {
    91     if (!isConst && type->isPrimitive())
    92         return referenceTypes_[type->primitive()];
    93 
    94     // :TODO: cache this.
    95     return ReferenceType::New(zone_, type);
    96 }
    97 
    98 EnumType *
    99 TypeManager::newEnum(Handle<String> name)
   100 {
   101     return EnumType::New(zone_, name);
   102 }
   103 
   104 Type *
   105 TypeManager::qualify(Handle<Type> type, TypeQualifiers quals)
   106 {
   107     assert(!type->hasTypeQualifiers());
   108 
   109     LookupKey key(type, quals);
   110 
   111     // We can't use findForAdd because a GC could rehash the table.
   112     TypeHashTable::Result r = qualTable_.find(key);
   113     if (r.found())
   114         return *r;
   115 
   116     Type *quald = Type::NewQualified(zone_, quals, type);
   117     if (!quald)
   118         return NULL;
   119 
   120     // :TODO: assert-no-gc
   121     TypeHashTable::Insert i = qualTable_.findForAdd(key);
   122     assert(!i.found());
   123 
   124     if (!qualTable_.add(i, quald))
   125         return NULL;
   126 
   127     return quald;
   128 }
   129 
   130 void
   131 TypeManager::accept(VirtualObjectVisitor *visitor)
   132 {
   133     visitor->visit(voidType_.address());
   134     visitor->visit(uncheckedType_.address());
   135     visitor->visit(stringType_.address());
   136     for (unsigned i = 0; i < PrimitiveTypes_Total; i++) {
   137         visitor->visit(primitiveTypes_[i].address());
   138         visitor->visit(referenceTypes_[i].address());
   139     }
   140 }
   141 
   142 void
   143 TypeManager::sweep()
   144 {
   145     for (TypeHashTable::iterator i(&qualTable_); !i.empty(); i.next()) {
   146         Type *obj = *i;
   147         MarkBit bits = HeapChunk::GetMarkBit(obj);
   148         if (HeapChunk::IsWhite(bits))
   149             i.erase();
   150         else
   151             assert(HeapChunk::IsBlack(bits));
   152     }
   153     qualTable_.checkDensity();
   154 }