#include "anaglym.h"
#include "ag.h"
#include "Video.h"
#include <SDL.h>
#include "wfobj/WFObj.hh"
#include <unistd.h>                 /* usleep() */
#include <lua.hpp>
#include <iostream>
#include <string>
#include <GL/gl.h>
#include <GL/glu.h>
#ifdef _WIN32
#include <windows.h>                /* Sleep() */
#endif
using namespace std;

namespace ag
{
    void register_functions(lua_State * L)
    {
        static const luaL_Reg functions[] = {
            { "print", print },
            { "println", println },
            { "loadModel", loadModel },
            { "loadStaticModel", loadStaticModel },
            { "sleep", sleep },
            { "startFrame", startFrame },
            { "endFrame", endFrame },
            { "setCamera", setCamera },
            { "elapsedTime", elapsedTime },
            { "doPhysics", doPhysics },
            { NULL, NULL }
        };
        luaL_register(L, "ag", functions);
    }

    static int print_val(lua_State * L, int index)
    {
        int type = lua_type(L, index);
        switch (type)
        {
            case LUA_TNUMBER:
                cout << lua_tonumber(L, index);
                break;
            case LUA_TSTRING:
                cout << lua_tostring(L, index);
                break;
            case LUA_TTABLE:
                cout << "{ ";

                /* traverse the table and print the keys/values */
                lua_checkstack(L, 3);
                lua_pushnil(L);
                for (bool first = true; lua_next(L, index) != 0; first = false)
                {
                    if (!first)
                        cout << ", ";
                    cout << '[';
                    print_val(L, lua_gettop(L) - 1);
                    cout << ']';
                    cout << " => ";
                    print_val(L, lua_gettop(L));
                    lua_pop(L, 1);
                }

                cout << " }";
                break;
            case LUA_TFUNCTION:
                cout << "<function>";
                break;
            case LUA_TUSERDATA:
                cout << "<userdata>";
                break;
            case LUA_TTHREAD:
                cout << "<thread>";
                break;
            case LUA_TLIGHTUSERDATA:
                cout << "<lightuserdata>";
                break;
        }
    }

    int print(lua_State * L)
    {
        int argc = lua_gettop(L);

        for ( int n = 1; n <= argc; n++ )
        {
            print_val(L, n);
        }
        return 0;
    }

    int println(lua_State * L)
    {
        int ret = print(L);
        cout << endl;
        return ret;
    }

    static int loadModelSpecify(lua_State * L, bool static_data)
    {
        int argc = lua_gettop(L);

        if (argc == 1 && lua_type(L, -1) == LUA_TSTRING)
        {
            string modelname = lua_tostring(L, -1);
            size_t pos = modelname.find_first_not_of(FILENAME_SAFE_CHARS);
            if (pos == string::npos)
            {
                string path = g_engine->locateResource(modelname + ".obj");
                string physpath = g_engine->locateResource(modelname + ".phy");
                if (path != "")
                {
                    WFObj * obj = new WFObj();

                    if (obj->load(path))
                    {
                        int id = g_engine->addObject(obj, static_data);
                        lua_newtable(L);
                        lua_pushinteger(L, id);
                        lua_setfield(L, -2, "id");
                        lua_pushcfunction(L, object::draw);
                        lua_setfield(L, -2, "draw");
                        lua_pushcfunction(L, object::setPosition);
                        lua_setfield(L, -2, "setPosition");
                        lua_pushcfunction(L, object::getPosition);
                        lua_setfield(L, -2, "getPosition");
                        if (physpath != "")
                        {
                            Engine::Object * obj = g_engine->getObject(id);
                            if (obj != NULL)
                            {
                                obj->loadPhy(physpath);
                            }
                        }
                        return 1;
                    }
                    else
                    {
                        delete obj;
                        cerr << "error loading object" << endl;
                    }
                }
            }
        }

        lua_pushnil(L);
        return 1;
    }

    int loadModel(lua_State * L)
    {
        loadModelSpecify(L, false);
    }

    int loadStaticModel(lua_State * L)
    {
        loadModelSpecify(L, true);
    }

    int sleep(lua_State * L)
    {
        int argc = lua_gettop(L);

        if (argc == 1)
        {
            if (lua_type(L, -1) == LUA_TNUMBER)
            {
                double seconds = lua_tonumber(L, -1);
#ifdef _WIN32
                Sleep((DWORD)(seconds * 1000));
#else
                usleep((useconds_t) (seconds * 1000000));
#endif
            }
        }

        return 0;
    }

    int startFrame(lua_State * L)
    {
        return g_engine->startFrame(L);
    }

    int endFrame(lua_State * L)
    {
        return g_engine->endFrame(L);
    }

    int setCamera(lua_State * L)
    {
        return g_engine->setCamera(L);
    }

    int elapsedTime(lua_State * L)
    {
        lua_pushinteger(L, SDL_GetTicks());
        return 1;
    }

    int doPhysics(lua_State * L)
    {
        g_engine->doPhysics();
        return 0;
    }

    namespace object
    {
        static Engine::Object * getObject(lua_State * L, int index)
        {
            Engine::Object * ret = NULL;

            lua_getfield(L, index, "id");
            if (lua_type(L, -1) == LUA_TNUMBER)
            {
                int id = lua_tointeger(L, -1);
                ret = g_engine->getObject(id);
            }
            lua_pop(L, 1);

            return ret;
        }

        int draw(lua_State * L)
        {
            int argc = lua_gettop(L);
            if (argc == 1 && lua_type(L, -1) == LUA_TTABLE)
            {
                Engine::Object * obj = getObject(L, -1);
                if (obj != NULL)
                    obj->draw();
            }
            return 0;
        }

        int setPosition(lua_State * L)
        {
            int argc = lua_gettop(L);

            if (argc == 4)
            {
                Engine::Object * obj = getObject(L, 1);
                if (obj != NULL)
                {
                    double position[3];
                    for (int i = 0; i < 3; i++)
                    {
                        int type = lua_type(L, i + 2);
                        if (type == LUA_TNUMBER  || type == LUA_TSTRING)
                        {
                            position[i] = lua_tonumber(L, i + 2);
                        }
                    }
                    obj->setPosition(position[0], position[1], position[2]);
                }
            }

            return 0;
        }

        int getPosition(lua_State * L)
        {
            int argc = lua_gettop(L);
            if (argc == 1)
            {
                Engine::Object * obj = getObject(L, 1);
                if (obj != NULL)
                {
                    double x, y, z;
                    obj->getPosition(&x, &y, &z);
                    lua_pushnumber(L, x);
                    lua_pushnumber(L, y);
                    lua_pushnumber(L, z);
                    return 3;
                }
            }
            return 0;
        }
    }
}