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 "naev.h"
#include "nlua_time.h"
 
#include "log.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;
}