/*

 * Copyright (C) 2012 David Anderson

 *

 * This file is part of JITCraft.

 *

 * JITCraft 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.

 * 

 * Foobar 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

 * JITCraft. If not, see http://www.gnu.org/licenses/.

 */

#ifndef _include_name_binding_h_

#define _include_name_binding_h_

#include "PoolAllocator.h"

#include "Types.h"

#include "Scanner.h"

namespace ke {

class Variable;

class NamedFunction;

class Scope;

class Type;

class NamedEnum;

class NamedConstant;

class String;

class BoundName : public PoolObject

{

    Scope *scope_;

    ScopedRoot<String> name_;

    ScopedRoot<Type> type_;

  protected:

    SourcePosition pos_;

  public:

    enum Kind {

      VARIABLE,

      FUNCTION,

      ENUM,

      CONSTANT

    };

    BoundName(Scope *scope, Handle<String> name, const SourcePosition &pos);

    virtual Kind kind() const = 0;

    String *name() const {

        return name_;

    }

    Scope *scope() const {

        return scope_;

    }

    Variable *toVariable() {

        assert(kind() == VARIABLE);

        return reinterpret_cast<Variable *>(this);

    }

    NamedFunction *toNamedFunction() {

        assert(kind() == FUNCTION);

        return reinterpret_cast<NamedFunction *>(this);

    }

    NamedConstant *toNamedConstant() {

        assert(kind() == CONSTANT);

        return reinterpret_cast<NamedConstant *>(this);

    }

    Type *type() const {

        return type_;

    }

    void setType(Handle<Type> type) {

        type_ = type;

    }

    const SourcePosition &pos() {

        return pos_;

    }

};

class NamedFunction : public BoundName

{

    unsigned slot_;

    TokenKind token_;

    ScopedRoot<Type> forwardType_;

    bool isVariadicNative_;

  public:

    NamedFunction(Scope *scope, Handle<String> name, const SourcePosition &pos, TokenKind token);

    Kind kind() const {

        return FUNCTION;

    }

    void allocate(unsigned slot) {

        slot_ = slot;

    }

    bool isForward() const {

        return token_ == TOK_FORWARD;

    }

    bool isPublic() const {

        return token_ == TOK_PUBLIC;

    }

    bool isNative() const {

        return token_ == TOK_NATIVE;

    }

    void setForwardType(Handle<Type> type) {

        forwardType_ = type;

    }

    Type *forwardType() {

        return forwardType_;

    }

    void setImplemented(const SourcePosition &at) {

        token_ = TOK_PUBLIC;

        pos_ = at;

    }

    void markNativeVariadic() {

        assert(token_ == TOK_NATIVE);

        isVariadicNative_ = true;

    }

    bool isNativeVariadic() {

        assert(token_ == TOK_NATIVE);

        return isVariadicNative_;

    }

    unsigned slot() const {

        return slot_;

    }

};

class NamedEnum : public BoundName

{

  public:

    NamedEnum(Scope *scope, Handle<String> name, const SourcePosition &pos)

      : BoundName(scope, name, pos)

    {

    }

    Kind kind() const {

        return ENUM;

    }

};

class NamedConstant : public BoundName

{

    BoxedPrimitive value_;

    Type *type_;

  public:

    NamedConstant(Scope *scope, Handle<String> name, const SourcePosition &pos)

      : BoundName(scope, name, pos),

        type_(NULL)

    {

    }

    Kind kind() const {

        return CONSTANT;

    }

    void set(Type *type, const BoxedPrimitive &value) {

        value_ = value;

        type_ = type;

    }

    Type *type() const {

        return type_;

    }

    BoxedPrimitive value() const {

        return value_;

    }

};

class Variable : public BoundName

{

  public:

    enum Location {

        UNALLOCATED,

        STACK,

        PARAMETER,

        ENVIRONMENT

    };

    Variable(Scope *scope, Handle<String> name, const SourcePosition &pos);

    Kind kind() const {

        return VARIABLE;

    }

    bool unallocated() const {

        return location() == UNALLOCATED;

    }

    void allocate(Location location, unsigned slot) {

        location_ = location;

        slot_ = slot;

    }

    Location location() const {

        return location_;

    }

    unsigned slot() const {

        assert(!unallocated());

        return slot_;

    }

    bool needsHeapAllocation();

  private:

    Location location_;

    unsigned slot_;

};

}

#endif // _include_name_binding_h_