/* 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 <string.h>

#include "Types.h"

#include "Zone.h"

#include "Strings.h"

#include "Array.h"

#include "Structures.h"

#include "Heap-inl.h"

using namespace ke;

void

Type::init(Kind kind, Type *root)

{

    this->kind_ = kind;

    this->root_ = root ? root : this;

}

Type *

Type::NewVoid(Zone *zone)

{

    Local<Type> type(zone, Type::cast(zone->allocate(MapKind_Type, sizeof(Type), Heap::Tenure_Old)));

    if (!type)

        return NULL;

    type->init(VOID);

    return type;

}

Type *

Type::NewImportable(Zone *zone)

{

    Local<Type> type(zone, Type::cast(zone->allocate(MapKind_Type, sizeof(Type), Heap::Tenure_Old)));

    if (!type)

        return NULL;

    type->init(IMPORTABLE);

    return type;

}

Type *

Type::NewString(Zone *zone)

{

    Local<Type> type(zone, Type::cast(zone->allocate(MapKind_Type, sizeof(Type), Heap::Tenure_Old)));

    if (!type)

        return NULL;

    type->init(STRING);

    return type;

}

Type *

Type::NewUnchecked(Zone *zone)

{

    Local<Type> type(zone, Type::cast(zone->allocate(MapKind_Type, sizeof(Type), Heap::Tenure_Old)));

    if (!type)

        return NULL;

    type->init(UNCHECKED);

    return type;

}

Type *

Type::NewType(Zone *zone)

{

    Local<Type> type(zone, Type::cast(zone->allocate(MapKind_Type, sizeof(Type), Heap::Tenure_Old)));

    if (!type)

        return NULL;

    type->init(TYPE);

    return type;

}

Type *

Type::NewPrimitive(Zone *zone, PrimitiveType prim)

{

    Local<Type> type(zone, Type::cast(zone->allocate(MapKind_Type, sizeof(Type), Heap::Tenure_Old)));

    if (!type)

        return NULL;

    type->init(PRIMITIVE);

    type->primitive_ = prim;

    return type;

}

Type *

Type::NewQualified(Zone *zone, TypeQualifiers qual, Handle<Type> root)

{

    Local<Type> type(zone, Type::cast(zone->allocate(MapKind_Type, sizeof(Type), Heap::Tenure_Old)));

    if (!type)

        return NULL;

    type->init(QUALTYPE, root);

    type->quals_ = qual;

    return type;

}

ArrayType *

ArrayType::New(Zone *zone, Handle<Type> contained, int elements)

{

    if (contained->isArray() &&

        ArrayType::cast(contained)->levels() == MAX_ARRAY_DEPTH)

    {

        zone->reportError(Message_MaximumArrayDepth);

        return NULL;

    }

    Local<ArrayType> type(zone, (ArrayType *)zone->allocate(MapKind_ArrayType, sizeof(ArrayType), Heap::Tenure_Old));

    if (!type)

        return NULL;

    type->init(ARRAY);

    type->contained_ = contained;

    type->elements_ = elements;

    if (!contained->isArray())

        type->levels_ = 1;

    else

        type->levels_ = ArrayType::cast(contained)->levels() + 1;

    type->newMap_ = ArrayMap::New(zone, type);

    if (!type->newMap_)

        return NULL;

    return type;

}

ArrayType *

ArrayType::NewExternal(Zone *zone, Handle<Type> contained)

{

    Local<ArrayType> type(zone, (ArrayType *)zone->allocate(MapKind_ArrayType, sizeof(ArrayType), Heap::Tenure_Old));

    if (!type)

        return NULL;

    assert(contained->isPrimitive());

    type->init(EXTERNAL_ARRAY);

    type->contained_ = contained;

    type->elements_ = DYNAMIC_ARRAY_SIZE;

    type->levels_ = 1;

    return type;

}

ReferenceType *

ReferenceType::New(Zone *zone, Handle<Type> contained)

{

    Local<ReferenceType> type(zone, (ReferenceType *)zone->allocate(MapKind_ReferenceType, sizeof(ReferenceType), Heap::Tenure_Old));

    if (!type)

        return NULL;

    type->init(REFERENCE);

    type->contained_ = contained;

    return type;

}

EnumType *

EnumType::New(Zone *zone, Handle<String> name)

{

    Local<EnumType> type(zone, (EnumType *)zone->allocate(MapKind_EnumType, sizeof(EnumType), Heap::Tenure_Old));

    if (!type)

        return NULL;

    type->init(ENUM);

    type->name_ = name;

    type->fields_ = NULL;

    return type;

}

StructType *

StructType::New(Zone *zone, Handle<String> name, Handle<FixedArray> fields)

{

    Local<StructType> type(zone, (StructType *)zone->allocate(MapKind_StructType, sizeof(StructType), Heap::Tenure_Old));

    if (!type)

        return NULL;

    type->init(STRUCT);

    type->newMap_ = NULL;

    type->fields_ = fields;

    type->name_ = name;

    // Fill in field offsets.

    StructureBufferBuilder builder;

    Local<Type> fieldType(zone);

    for (size_t i = 0; i < fields->length(); i++) {

        Descriptor *desc = Descriptor::cast(fields->at(i));

        fieldType = desc->type();

        unsigned offset;

        if (!builder.add(fieldType, &offset))

            return NULL;

        desc->setOffset(offset);

    }

    Local<StructMap> map(zone, StructMap::New(zone, builder, type));

    if (!map)

        return NULL;

    type->newMap_ = map;

    return type;

}

Descriptor *

StructType::lookupField(String *name, unsigned *index)

{

    // :TODO: assert no GC

    for (unsigned i = 0; i < fields_->length(); i++) {

        Descriptor *desc = Descriptor::cast(fields_->at(i));

        if (desc->name() == name) {

            if (index)

                *index = i;

            return desc;

        }

    }

    return NULL;

}

FunctionType *

FunctionType::New(Zone *zone, Handle<String> name, TokenKind token)

{

    Local<FunctionType> type(zone, (FunctionType *)zone->allocate(MapKind_FunctionType, sizeof(FunctionType), Heap::Tenure_Old));

    if (!type)

        return NULL;

    type->init(FUNCTION);

    type->parameters_ = NULL;

    type->defaults_ = NULL;

    type->returnType_ = NULL;

    type->name_ = name;

    type->token_ = token;

    return type;

}

static size_t

FormatVA(char *buffer, size_t maxlen, const char *text, va_list ap)

{

    size_t len = vsnprintf(buffer, maxlen, text, ap);

    if (len >= maxlen) {

        buffer[maxlen - 1] = '\0';

        return maxlen - 1;

    }

    return len;

}

static size_t

Format(char *buffer, size_t maxlen, const char *text, ...)

{

    va_list ap;

    va_start(ap, text);

    size_t len = FormatVA(buffer, maxlen, text, ap);

    va_end(ap);

    return len;

}

static size_t

FormatTypeName(Type *type, char *buffer, size_t maxlen)

{

    if (type->isArray()) {

        ArrayType *atype = ArrayType::cast(type);

        Type *innermost = atype->innermost();

        size_t len = FormatTypeName(innermost, buffer, maxlen);

        for (;;) {

            if (atype->isFixedLength())

                len += Format(buffer + len, maxlen - len, "[%d]", atype->fixedLength());

            else

                len += Format(buffer + len, maxlen - len, "[]");

            if (!atype->contained()->isArray())

                break;

            atype = ArrayType::cast(atype->contained());

        }

        return len;

    }

    if (type->hasTypeQualifiers()) {

        size_t len = 0;

        if (type->quals() & TypeQual_Const)

            len += Format(buffer, maxlen, "const ");

        return len + FormatTypeName(type->unqualified(), buffer + len, maxlen - len);

    }

    if (type->isFunction())

        return Format(buffer, maxlen, "function");

    if (type->isVoid())

        return Format(buffer, maxlen, "void");

    if (type->isUnchecked())

        return Format(buffer, maxlen, "unchecked");

    if (type->isImportable())

        return Format(buffer, maxlen, "import");

    if (type->isEnum())

        return Format(buffer, maxlen, "%s", EnumType::cast(type)->name()->chars());

    if (type->isStruct())

        return Format(buffer, maxlen, "%s", StructType::cast(type)->name()->chars());

    if (type->isReference()) {

        ReferenceType *ref = ReferenceType::cast(type);

        size_t len = Format(buffer, maxlen, "&");

        return len + FormatTypeName(ref->contained(), buffer + len, maxlen - len);

    }

    switch (type->primitive()) {

      case PrimitiveType_Bool:

        return Format(buffer, maxlen, "bool");

      case PrimitiveType_Char:

        return Format(buffer, maxlen, "char");

      case PrimitiveType_Int32:

        return Format(buffer, maxlen, "int");

      case PrimitiveType_Float:

        return Format(buffer, maxlen, "float");

      default:

        return Format(buffer, maxlen, "<UNKNOWN>");

    }

}

const char *

ke::GetTypeName(Type *type, char *buffer, size_t maxlen)

{

    FormatTypeName(type, buffer, maxlen);

    return buffer;

}