/*

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

#define _include_jitcraft_function_h_

#include "Heap.h"

#include "Barriers.h"

#include "Token.h"

#include "Types.h"

namespace ke {

class Code;

class Module;

class Type;

class String;

class Function : public Object

{

    Barriered<Code> code_;

    Barriered<Type> type_;

    Barriered<String> name_;

    SourcePosition pos_;

 public:

    static Function *New(Zone *zone, Handle<Code> code, Handle<Type> type, Handle<String> name,

                         const SourcePosition &pos);

    Code *code() const {

        return code_;

    }

    Type *type() const {

        return type_;

    }

    String *name() const {

        return name_;

    }

    const SourcePosition &pos() {

        return pos_;

    }

    unsigned nargs() {

        return type()->parameters()->length();

    }

    Type *argType(size_t index) {

        return type()->parameterAt(index);

    }

    // Begin GC Descriptor.

    static inline size_t offsetOfCode() {

        return offsetof(Function, code_);

    }

    static inline size_t offsetOfType() {

        return offsetof(Function, type_);

    }

    static inline size_t offsetOfName() {

        return offsetof(Function, name_);

    }

    // End GC Descriptor.

    static inline Function *cast(Object *obj) {

        assert(!obj || obj->is(MapKind_Function));

        return reinterpret_cast<Function *>(obj);

    }

};

// A Native is a function with a very specialized calling convention, intended

// for earlier ISourcePawnEngine API compatibility.

class Native : public Object

{

    Barriered<Type> type_;

    Barriered<String> name_;

    void *function_;

  public:

    static Native *New(Zone *zone, Handle<Type> type, Handle<String> name);

    void *function() {

        return function_;

    }

    Type *type() {

        return type_;

    }

    String *name() {

        return name_;

    }

    void bind(void *function) {

        function_ = function;

    }

    // Begin GC Descriptor.

    static inline size_t offsetOfType() {

        return offsetof(Native, type_);

    }

    static inline size_t offsetOfName() {

        return offsetof(Native, name_);

    }

    // End GC Descriptor.

    static inline Native *cast(Object *obj) {

        assert(!obj || obj->is(MapKind_Native));

        return reinterpret_cast<Native *>(obj);

    }

};

}

#endif // _include_jitcraft_function_h_