source: trunk/src/lua/lua_map_area.cc @ 319

// Copyright (C) 2012-2014 ChaosForge Ltd
// http://chaosforge.org/
//
// This file is part of NV Libraries.
// For conditions of distribution and use, see copyright notice in nv.hh

#include "nv/lua/lua_map_area.hh"
#include "nv/core/flags.hh"
#include "nv/lua/lua_area.hh"
#include "nv/lua/lua_glm.hh"
#include "nv/lua/lua_values.hh"
#include "nv/lua/lua_raw.hh"

static const char* NLUA_MAP_AREA_METATABLE = "map_area";

typedef nv::flags<512> cell_set;

static nv::uint32 nlua_to_cell_id( lua_State* L, int index, nv::map_area* map )
{
        if ( lua_type( L, index ) == LUA_TSTRING ) 
                return map->string_to_id( lua_tostring( L, index ) );
        else
                return lua_tounsigned( L, index );
}

// static cell_set nlua_to_cell_set( lua_State* L, int index, nv::map_area* map )
// {
//      cell_set result;
//      switch ( lua_type( L, index ) )
//      {
//      case LUA_TTABLE :       
//              {
//                      lua_pushnil( L );
//                      while ( lua_next( L, index ) != 0 )
//                      {
//                              if ( lua_type( L, -1 ) == LUA_TSTRING )
//                                      result.set( map->string_to_id( lua_tostring( L, -1 ) ), true );
//                              else
//                                      result.set( lua_tounsigned( L, -1 ), true );
//                              lua_pop( L, 1 );
//                      }
//              } break;
//      case LUA_TSTRING : result.set( map->string_to_id( lua_tostring( L, index ) ), true ); break;
//      case LUA_TNUMBER : result.set( lua_tounsigned( L, index ), true ); break;
//      }
//      return result;
// }

bool nv::lua::detail::is_map_area( lua_State* L, int index )
{
        return luaL_testudata( L, index, NLUA_MAP_AREA_METATABLE ) != 0;
}

nv::map_area* nv::lua::detail::to_map_area( lua_State* L, int index )
{
        nv::map_area* o = nullptr;
        if ( lua_istable( L , index ) )
        {
                lua_pushstring( L, "__map_area_ptr" );
                lua_rawget( L, index );
                if ( lua_isuserdata( L, -1 ) )
                {
                        o = static_cast<nv::map_area*>( lua_touserdata( L, -1 ) );
                } 
                lua_pop( L, 1 );
        }
        else
        {
                return *(nv::map_area**)luaL_checkudata( L, index, NLUA_MAP_AREA_METATABLE );
        }
        return o;
}

using nv::lua::detail::is_map_area;
using nv::lua::detail::to_map_area;
using nv::lua::detail::push_map_area;

using nv::lua::detail::is_coord;
using nv::lua::detail::to_coord;
using nv::lua::detail::to_pcoord;
using nv::lua::detail::push_coord;

using nv::lua::detail::is_area;
using nv::lua::detail::to_area;
using nv::lua::detail::to_parea;
using nv::lua::detail::push_area;

void nv::lua::detail::push_map_area( lua_State* L, nv::map_area* c )
{
        nv::map_area** pm = (nv::map_area**) (lua_newuserdata(L, sizeof(nv::map_area*)));
        *pm = c;
        luaL_getmetatable( L, NLUA_MAP_AREA_METATABLE );
        lua_setmetatable( L, -2 );
}

static int nlua_map_area_tostring( lua_State* L )
{
        lua_pushstring( L, "map_area" );
        return 1;
}

static int nlua_map_area_gc( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        if ( ma != nullptr )
        {
                delete ma;
        }
        return 0;
}

static int nlua_map_area_get_area( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        nv::rectangle r  = ma->get_rectangle();
        r.lr.x -= 1;
        r.lr.y -= 1;
        push_area( L, r );
        return 1;
}

static int nlua_map_area_get_shift( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        push_coord( L, ma->get_shift() );
        return 1;
}

static int nlua_map_area_get_size( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        push_coord( L, ma->get_size() );
        return 1;
}

static int nlua_map_area_get_cell( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        lua_pushstring( L, ma->id_to_string( ma->get_cell( to_coord( L, 2 ) ) ).c_str() );
        return 1;
}

static int nlua_map_area_set_cell( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        ma->set_cell( to_coord( L, 2 ), nlua_to_cell_id( L, 3, ma ) );
        return 0;
}

static int nlua_map_area_raw_get_cell( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        lua_pushunsigned( L, ma->get_cell( to_coord( L, 2 ) ) );
        return 1;
}

static int nlua_map_area_raw_set_cell( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        ma->set_cell( to_coord( L, 2 ), lua_tounsigned( L, 3 ) );
        return 0;
}

static int nlua_map_area_index( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        if ( lua_type( L, 1 ) == LUA_TSTRING )
        {
                luaL_getmetafield( L, 1, "__functions" );
                lua_pushvalue( L, 2 );
                lua_rawget( L, -2 );
        }
        else
        {
                lua_pushunsigned( L, ma->get_cell( to_coord( L, 2 ) ) );
        }
        return 1;
}

static int nlua_map_area_newindex( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        ma->set_cell( to_coord( L, 2 ), lua_tounsigned( L, 3 ) );
        return 0;
}

static int nlua_map_area_new_sub_area( lua_State* L )
{
        nv::map_area* ma = to_map_area( L, 1 );
        push_map_area( L, ma->create_sub_area( to_area( L, 2 ) ) );
        return 1;
}

static const luaL_Reg nlua_map_area_f[] = {
        { "get_area",     nlua_map_area_get_area },
        { "get_shift",    nlua_map_area_get_shift },
        { "get_size",     nlua_map_area_get_size },
        { "get_cell",     nlua_map_area_get_cell },
        { "set_cell",     nlua_map_area_set_cell },
        { "raw_get_cell", nlua_map_area_raw_get_cell },
        { "raw_set_cell", nlua_map_area_raw_set_cell },
        { "new_sub_area", nlua_map_area_new_sub_area },
        {NULL, NULL}
};

static const luaL_Reg nlua_map_area_mt[] = {
        { "__newindex", nlua_map_area_newindex },
        { "__index",    nlua_map_area_index },
        { "__tostring", nlua_map_area_tostring },
        { "__gc",       nlua_map_area_gc },
        {NULL, NULL}
};

static int luaopen_map_area( lua_State * L )
{
        luaL_newmetatable( L, NLUA_MAP_AREA_METATABLE );
        nlua_register( L, nlua_map_area_mt, -1 );
        lua_createtable( L, 0, 0 );
        nlua_register( L, nlua_map_area_f, -1 );
        lua_setfield(L, -2, "__functions" );
        lua_pop( L, 1 );
        return 0;
}

void nv::lua::register_map_area_interface( lua_State* L, int index )
{
        nlua_register( L, nlua_map_area_f, index );
}

void nv::lua::register_map_area( lua_State* L )
{
        luaopen_map_area(L);
}

void nv::lua::register_map_area_instance( lua_State* L, ref object_index, map_area* area )
{
        lua_rawgeti( L, LUA_REGISTRYINDEX, object_index.get() );
        lua_pushstring( L, "__map_area_ptr" );
        lua_pushlightuserdata( L, (map_area*)area );
        lua_rawset( L, -3 );
        lua_pop( L, 1 );
}