nlua_time.c

Go to the documentation of this file.

/*
 * See Licensing and Copyright notice in naev.h
 */
#include <lauxlib.h>
#include <stdlib.h>
 
#include "nlua_time.h"
 
#include "nluadef.h"
#include "ntime.h"
 
/* Time methods. */
static int timeL_new( lua_State *L );
static int timeL_add( lua_State *L );
static int timeL_add__( lua_State *L );
static int timeL_sub( lua_State *L );
static int timeL_sub__( lua_State *L );
static int timeL_eq( lua_State *L );
static int timeL_lt( lua_State *L );
static int timeL_le( lua_State *L );
static int timeL_get( lua_State *L );
static int timeL_str( lua_State *L );
static int timeL_inc( lua_State *L );
static int timeL_tonumber( lua_State *L );
static int timeL_fromnumber( lua_State *L );
 
static const luaL_Reg time_methods[] = { { "new", timeL_new },
                                         { "add", timeL_add__ },
                                         { "__add", timeL_add },
                                         { "sub", timeL_sub__ },
                                         { "__sub", timeL_sub },
                                         { "__eq", timeL_eq },
                                         { "__lt", timeL_lt },
                                         { "__le", timeL_le },
                                         { "get", timeL_get },
                                         { "str", timeL_str },
                                         { "__tostring", timeL_str },
                                         { "inc", timeL_inc },
                                         { "tonumber", timeL_tonumber },
                                         { "fromnumber", timeL_fromnumber },
                                         { 0, 0 } }; 

int nlua_loadTime( nlua_env env )
{
   nlua_register( env, TIME_METATABLE, time_methods, 1 );
   return 0; /* No error */
}

ntime_t *lua_totime( lua_State *L, int ind )
{
   return (ntime_t *)lua_touserdata( L, ind );
}

ntime_t *luaL_checktime( lua_State *L, int ind )
{
   if ( lua_istime( L, ind ) )
      return lua_totime( L, ind );
   luaL_typerror( L, ind, TIME_METATABLE );
   return NULL;
}

ntime_t luaL_validtime( lua_State *L, int ind )
{
   return *luaL_checktime( L, ind );
}

ntime_t *lua_pushtime( lua_State *L, ntime_t time )
{
   ntime_t *p = (ntime_t *)lua_newuserdata( L, sizeof( ntime_t ) );
   *p         = time;
   luaL_getmetatable( L, TIME_METATABLE );
   lua_setmetatable( L, -2 );
   return p;
}

int lua_istime( lua_State *L, int ind )
{
   int ret;
 
   if ( lua_getmetatable( L, ind ) == 0 )
      return 0;
   lua_getfield( L, LUA_REGISTRYINDEX, TIME_METATABLE );
 
   ret = 0;
   if ( lua_rawequal( L, -1, -2 ) ) /* does it have the correct mt? */
      ret = 1;
 
   lua_pop( L, 2 ); /* remove both metatables */
   return ret;
}

static int timeL_new( lua_State *L )
{
   int cycles, periods, seconds;
 
   /* Parameters. */
   cycles  = luaL_checkint( L, 1 );
   periods = luaL_checkint( L, 2 );
   seconds = luaL_checkint( L, 3 );
 
   /* Create the time. */
   lua_pushtime( L, ntime_create( cycles, periods, seconds ) );
   return 1;
}

static int timeL_add( lua_State *L )
{
   ntime_t t1, t2;
 
   /* Parameters. */
   t1 = luaL_validtime( L, 1 );
   t2 = luaL_validtime( L, 2 );
 
   /* Add them. */
   lua_pushtime( L, t1 + t2 );
   return 1;
}
 
/*
 * Method version of time_add that modifies the first time.
 */
static int timeL_add__( lua_State *L )
{
   ntime_t *t1, t2;
 
   /* Parameters. */
   t1 = luaL_checktime( L, 1 );
   t2 = luaL_validtime( L, 2 );
 
   /* Add them. */
   *t1 += t2;
   lua_pushtime( L, *t1 );
   return 1;
}

static int timeL_sub( lua_State *L )
{
   ntime_t t1, t2;
 
   /* Parameters. */
   t1 = luaL_validtime( L, 1 );
   t2 = luaL_validtime( L, 2 );
 
   /* Sub them. */
   lua_pushtime( L, t1 - t2 );
   return 1;
}
 
/*
 * Method version of time_sub that modifies the first time.
 */
static int timeL_sub__( lua_State *L )
{
   ntime_t *t1, t2;
 
   /* Parameters. */
   t1 = luaL_checktime( L, 1 );
   t2 = luaL_validtime( L, 2 );
 
   /* Sub them. */
   *t1 -= t2;
   lua_pushtime( L, *t1 );
   return 1;
}

static int timeL_eq( lua_State *L )
{
   ntime_t t1, t2;
   t1 = luaL_validtime( L, 1 );
   t2 = luaL_validtime( L, 2 );
   lua_pushboolean( L, t1 == t2 );
   return 1;
}

static int timeL_lt( lua_State *L )
{
   ntime_t t1, t2;
   t1 = luaL_validtime( L, 1 );
   t2 = luaL_validtime( L, 2 );
   lua_pushboolean( L, t1 < t2 );
   return 1;
}

static int timeL_le( lua_State *L )
{
   ntime_t t1, t2;
   t1 = luaL_validtime( L, 1 );
   t2 = luaL_validtime( L, 2 );
   lua_pushboolean( L, t1 <= t2 );
   return 1;
}

static int timeL_get( lua_State *L )
{
   lua_pushtime( L, ntime_get() );
   return 1;
}

static int timeL_str( lua_State *L )
{
   ntime_t t;
   char    nt[64];
   int     d;
 
   /* Parse parameters. */
   if ( !lua_isnoneornil( L, 1 ) )
      t = luaL_validtime( L, 1 );
   else
      t = ntime_get();
   d = luaL_optinteger( L, 2, 2 ); /* Defaults to 2 decimals. */
 
   /* Push string. */
   ntime_prettyBuf( nt, sizeof( nt ), t, d );
   lua_pushstring( L, nt );
   return 1;
}

static int timeL_inc( lua_State *L )
{
   ntime_inc( luaL_validtime( L, 1 ) );
   return 0;
}

static int timeL_tonumber( lua_State *L )
{
   ntime_t t = luaL_validtime( L, 1 );
   lua_pushnumber( L, t );
   return 1;
}

static int timeL_fromnumber( lua_State *L )
{
   ntime_t t = (ntime_t)luaL_checknumber( L, 1 );
   lua_pushtime( L, t );
   return 1;
}