dvander@41: /* vim: set ts=4 sw=4 tw=99 et:
dvander@41:  *
dvander@11:  * Copyright (C) 2012 David Anderson
dvander@11:  *
dvander@11:  * This file is part of JITCraft.
dvander@11:  *
dvander@11:  * JITCraft is free software: you can redistribute it and/or modify it under
dvander@11:  * the terms of the GNU General Public License as published by the Free
dvander@11:  * Software Foundation, either version 3 of the License, or (at your option)
dvander@11:  * any later version.
dvander@11:  * 
dvander@11:  * Foobar is distributed in the hope that it will be useful, but WITHOUT ANY
dvander@11:  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@11:  * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
dvander@11:  *
dvander@11:  * You should have received a copy of the GNU General Public License along with
dvander@11:  * JITCraft. If not, see http://www.gnu.org/licenses/.
dvander@11:  */
danderson@92: #include <string.h>
dvander@11: #include "Types.h"
dvander@11: #include "TypeManager.h"
dvander@14: #include "Zone.h"
dvander@54: #include "Strings.h"
danderson@92: #include "Spaces-inl.h"
danderson@92: #include "Heap-inl.h"
dvander@11: 
dvander@59: using namespace ke;
dvander@11: 
dvander@11: 
dvander@11: TypeManager::TypeManager(Zone *zone)
dvander@11:   : zone_(zone)
dvander@11: {
dvander@11: }
dvander@11: 
dvander@11: bool
dvander@11: TypeManager::initialize()
dvander@11: {
dvander@71:     voidType_ = Type::NewVoid(zone_);
dvander@71:     if (!voidType_)
dvander@71:         return false;
dvander@71: 
dvander@54:     Local<String> floatTag(zone_, zone_->makeSymbol("Float"));
dvander@54:     if (!floatTag)
dvander@54:         return false;
dvander@54: 
dvander@54:     Local<String> stringTag(zone_, zone_->makeSymbol("String"));
dvander@54:     if (!stringTag)
dvander@11:         return false;
dvander@54: 
dvander@54:     Local<String> nullTag(zone_);
dvander@54: 
dvander@54:     if ((primitiveTypes_[PrimitiveType_Int32] = Type::NewPrimitive(zone_, nullTag)) == NULL)
dvander@54:         return false;
dvander@54:     if ((primitiveTypes_[PrimitiveType_Float] = Type::NewPrimitive(zone_, floatTag)) == NULL)
dvander@11:         return false;
dvander@12:     if ((primitiveTypes_[PrimitiveType_Native] = Type::NewPrimitive(zone_, PrimitiveType_Native)) == NULL)
dvander@12:         return false;
dvander@54:     if ((primitiveTypes_[PrimitiveType_Char] = Type::NewPrimitive(zone_, stringTag)) == NULL)
dvander@21:         return false;
dvander@14: 
dvander@41:     Local<Type> type(zone_);
dvander@14:     for (unsigned i = 0; i < PrimitiveTypes_Total; i++) {
dvander@41:         type = primitiveTypes_[i];
dvander@67:         if ((referenceTypes_[i] = Type::NewReference(zone_, type, false)) == NULL)
dvander@14:             return false;
dvander@14:     }
dvander@14: 
dvander@11:     return true;
dvander@11: }
dvander@11: 
dvander@11: Type *
dvander@54: TypeManager::getPrimitiveForTag(Handle<String> tag)
dvander@11: {
dvander@54:     if (!tag)
dvander@11:         return getPrimitive(PrimitiveType_Int32);
dvander@54:     if (strcmp(tag->chars(), "Float") == 0)
dvander@11:         return getPrimitive(PrimitiveType_Float);
dvander@54:     if (strcmp(tag->chars(), "String") == 0)
dvander@21:         return getPrimitive(PrimitiveType_Char);
dvander@140: 
dvander@140:     // :TODO: cache this.
dvander@140:     return Type::NewPrimitive(zone_, tag);
dvander@11: }
dvander@11: 
dvander@11: Type *
dvander@67: TypeManager::unsafeArray(Handle<Type> contained, int elements, bool isConst)
dvander@11: {
dvander@11:     // :TODO: cache this.
dvander@67:     return Type::NewArray(zone_, contained, elements, isConst);
dvander@11: }
dvander@14: 
dvander@14: Type *
dvander@67: TypeManager::getReference(Handle<Type> type, bool isConst)
dvander@14: {
dvander@67:     if (!isConst && (type->isFloat() || (type->isInt32() && !type->tag())))
dvander@14:         return referenceTypes_[type->primitive()];
dvander@14: 
dvander@14:     // :TODO: cache this.
dvander@67:     return Type::NewReference(zone_, type, isConst);
dvander@14: }
dvander@14: 
dvander@14: Type *
dvander@41: TypeManager::newFunction(Handle<Type> returnType, Handle<FixedArray> args)
dvander@14: {
dvander@14:     // :TODO: cache this.
dvander@14:     return Type::NewFunction(zone_, returnType, args);
dvander@14: }
dvander@45: 
dvander@68: Type *
dvander@68: TypeManager::newNative(Handle<Type> returnType, Handle<FixedArray> args, bool variadic)
dvander@68: {
dvander@68:     // :TODO: cache this.
dvander@68:     return Type::NewNative(zone_, returnType, args, variadic);
dvander@68: }
dvander@68: 
dvander@45: void
dvander@45: TypeManager::accept(VirtualObjectVisitor *visitor)
dvander@45: {
dvander@72:     visitor->visit(voidType_.address());
dvander@45:     for (unsigned i = 0; i < PrimitiveTypes_Total; i++) {
dvander@45:         visitor->visit(primitiveTypes_[i].address());
dvander@45:         visitor->visit(referenceTypes_[i].address());
dvander@45:     }
dvander@45: }