nlua_tk.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_tk.h"
 
#include "array.h"
#include "dialogue.h"
#include "input.h"
#include "land.h"
#include "land_outfits.h"
#include "log.h"
#include "nlua_colour.h"
#include "nlua_gfx.h"
#include "nlua_outfit.h"
#include "nluadef.h"
#include "toolkit.h"
 
/* Stuff for the custom toolkit. */
#define TK_CUSTOMDONE   "__customDone"
typedef struct custom_functions_s {
   lua_State *L; 
   int done; 
   /* Function references. */
   int update;
   int draw;
   int keyboard;
   int mouse;
   int resize;
   int textinput;
} custom_functions_t;
static int cust_update( double dt, void* data );
static void cust_render( double x, double y, double w, double h, void* data );
static int cust_event( unsigned int wid, SDL_Event *event, void* data );
static int cust_key( SDL_Keycode key, SDL_Keymod mod, int pressed, custom_functions_t *cf );
static int cust_text( const char *str, custom_functions_t *cf );
static int cust_mouse( int type, int button, double x, double y, custom_functions_t *cf );
static int cust_event_window( SDL_WindowEventID event, Sint32 w, Sint32 h, custom_functions_t *cf );
 
/* Toolkit methods. */
static int tkL_isOpen( lua_State *L );
static int tkL_query( lua_State *L );
static int tkL_msg( lua_State *L );
static int tkL_yesno( lua_State *L );
static int tkL_input( lua_State *L );
static int tkL_choice( lua_State *L );
static int tkL_list( lua_State *L );
static int tkL_merchantOutfit( lua_State *L );
static int tkL_custom( lua_State *L );
static int tkL_customRename( lua_State *L );
static int tkL_customFullscreen( lua_State *L );
static int tkL_customResize( lua_State *L );
static int tkL_customSize( lua_State *L );
static int tkL_customDone( lua_State *L );
static const luaL_Reg tkL_methods[] = {
   { "isOpen", tkL_isOpen },
   { "query", tkL_query },
   { "msg", tkL_msg },
   { "yesno", tkL_yesno },
   { "input", tkL_input },
   { "choice", tkL_choice },
   { "list", tkL_list },
   { "merchantOutfit", tkL_merchantOutfit },
   { "custom", tkL_custom },
   { "customRename", tkL_customRename },
   { "customFullscreen", tkL_customFullscreen },
   { "customResize", tkL_customResize },
   { "customSize", tkL_customSize },
   { "customDone", tkL_customDone },
   {0,0}
}; 
int nlua_loadTk( nlua_env env )
{
   nlua_register(env, "tk", tkL_methods, 0);
   nlua_loadCol(env);
   nlua_loadGFX(env);
   return 0;
}
 
static int tkL_isOpen( lua_State *L )
{
   lua_pushboolean(L,toolkit_isOpen());
   return 1;
}
 
static int tkL_query( lua_State *L )
{
   const char *wdwname, *wgtname;
   unsigned int wid;
   int bx, by, x, y, w, h;
   wdwname = luaL_checkstring( L, 1 );
   wgtname = luaL_optstring( L, 2, NULL );
   wid = window_get( wdwname );
   if (wid == 0)
      return 0;
   window_posWindow( wid, &bx, &by );
   if (wgtname == NULL) {
      x = bx;
      y = by;
      window_dimWindow( wid, &w, &h );
   }
   else {
      window_posWidget( wid, wgtname, &x, &y );
      window_dimWidget( wid, wgtname, &w, &h );
      x += bx;
      y += by;
   }
   lua_pushinteger( L, x );
   lua_pushinteger( L, y );
   lua_pushinteger( L, w );
   lua_pushinteger( L, h );
   return 4;
}
 
static int tkL_msg( lua_State *L )
{
   const char *title, *str, *img;
   int width, height;
 
   // Get fixed arguments : title, string to display and image filename
   title = luaL_checkstring(L,1);
   str   = luaL_checkstring(L,2);
 
   if (lua_gettop(L) > 2) {
      img   = luaL_checkstring(L,3);
 
      // Get optional arguments : width and height
      width  = (lua_gettop(L) < 4) ? -1 : luaL_checkinteger(L,4);
      height = (lua_gettop(L) < 5) ? -1 : luaL_checkinteger(L,5);
 
      dialogue_msgImgRaw( title, str, img, width, height );
      return 0;
   }
   dialogue_msgRaw( title, str );
   return 0;
}
static int tkL_yesno( lua_State *L )
{
   int ret;
   const char *title, *str;
 
   title = luaL_checkstring(L,1);
   str   = luaL_checkstring(L,2);
 
   ret = dialogue_YesNoRaw( title, str );
   lua_pushboolean(L,ret);
   return 1;
}
static int tkL_input( lua_State *L )
{
   const char *title, *str;
   char *ret;
   int min, max;
 
   title = luaL_checkstring(L,1);
   min   = luaL_checkint(L,2);
   max   = luaL_checkint(L,3);
   str   = luaL_checkstring(L,4);
 
   ret = dialogue_inputRaw( title, min, max, str );
   if (ret != NULL) {
      lua_pushstring(L, ret);
      free(ret);
   }
   else
      lua_pushnil(L);
   return 1;
}
 
static int tkL_choice( lua_State *L )
{
   int ret, opts;
   const char *title, *str;
   char *result;
 
   /* Handle parameters. */
   opts  = lua_gettop(L) - 2;
   title = luaL_checkstring(L,1);
   str   = luaL_checkstring(L,2);
 
   /* Do an initial scan for invalid arguments. */
   for (int i=0; i<opts; i++)
      luaL_checkstring(L, i+3);
 
   /* Create dialogue. */
   dialogue_makeChoice( title, str, opts );
   for (int i=0; i<opts; i++)
      dialogue_addChoice( title, str, luaL_checkstring(L,i+3) );
   result = dialogue_runChoice();
   if (result == NULL) /* Something went wrong, return nil. */
      return 0;
 
   /* Handle results. */
   ret = -1;
   for (int i=0; i<opts && ret==-1; i++) {
      if (strcmp(result, luaL_checkstring(L,i+3)) == 0)
         ret = i+1; /* Lua uses 1 as first index. */
   }
 
   /* Push parameters. */
   lua_pushnumber(L,ret);
   lua_pushstring(L,result);
 
   /* Clean up. */
   free(result);
 
   return 2;
}
 
static int tkL_list( lua_State *L )
{
   int ret, opts;
   const char *title, *str;
   char **choices;
   NLUA_MIN_ARGS(3);
 
   /* Handle parameters. */
   opts  = lua_gettop(L) - 2;
   title = luaL_checkstring(L,1);
   str   = luaL_checkstring(L,2);
 
   /* Do an initial scan for invalid arguments. */
   for (int i=0; i<opts; i++)
      luaL_checkstring(L, i+3);
 
   /* Will be freed by the toolkit. */
   choices = malloc( sizeof(char*) * opts );
   for (int i=0; i<opts; i++)
      choices[i] = strdup( luaL_checkstring(L, i+3) );
 
   ret = dialogue_listRaw( title, choices, opts, str );
 
   /* Cancel returns -1, do nothing. */
   if (ret == -1)
      return 0;
 
   /* Push index and choice string. */
   lua_pushnumber(L, ret+1);
   lua_pushstring(L, choices[ret]);
 
   return 2;
}
 
static int tkL_merchantOutfit( lua_State *L )
{
   const Outfit **outfits;
   unsigned int wid;
   const char *name;
   int w, h;
 
   name = luaL_checkstring(L,1);
 
   if (!lua_istable(L,2))
      NLUA_INVALID_PARAMETER(L);
 
   outfits = array_create_size( const Outfit*, lua_objlen(L,2) );
   /* Iterate over table. */
   lua_pushnil(L);
   while (lua_next(L, -2) != 0) {
      array_push_back( &outfits, luaL_validoutfit(L, -1) );
      lua_pop(L,1);
   }
 
   /* Create window. */
   if (SCREEN_W < LAND_WIDTH || SCREEN_H < LAND_HEIGHT) {
      w = -1; /* Fullscreen. */
      h = -1;
   }
   else {
      w = LAND_WIDTH + 0.5 * (SCREEN_W - LAND_WIDTH);
      h = LAND_HEIGHT + 0.5 * (SCREEN_H - LAND_HEIGHT);
   }
   wid = window_create( "wdwMerchantOutfit", name, -1, -1, w, h );
   outfits_open( wid, outfits );
 
   return 0;
}
 
static int tkL_custom( lua_State *L )
{
   int w = luaL_checkinteger(L, 2);
   int h = luaL_checkinteger(L, 3);
   const char *caption = luaL_checkstring(L, 1);
   custom_functions_t *cf = calloc( 1, sizeof(custom_functions_t) );
 
   /* Set up custom function pointers. */
   cf->L = L;
#define GETFUNC( address, pos ) \
do { \
   lua_pushvalue( L, (pos) ); \
   luaL_checktype(L, -1, LUA_TFUNCTION); \
   (address) = luaL_ref(L, LUA_REGISTRYINDEX); \
} while (0)
   GETFUNC( cf->update,    4 );
   GETFUNC( cf->draw,      5 );
   GETFUNC( cf->keyboard,  6 );
   GETFUNC( cf->mouse,     7 );
   GETFUNC( cf->resize,    8 );
   GETFUNC( cf->textinput, 9 );
#undef GETFUNC
 
   /* Set done condition. */
   lua_pushboolean(L, 0);
   lua_setglobal(L, TK_CUSTOMDONE );
 
   /* Create the dialogue. */
   dialogue_custom( caption, w, h, cust_update, cust_render, cust_event, cf, 1 );
 
   /* Clean up. */
   cf->done = 1;
   luaL_unref(L, LUA_REGISTRYINDEX, cf->update);
   luaL_unref(L, LUA_REGISTRYINDEX, cf->draw);
   luaL_unref(L, LUA_REGISTRYINDEX, cf->keyboard);
   luaL_unref(L, LUA_REGISTRYINDEX, cf->mouse);
   luaL_unref(L, LUA_REGISTRYINDEX, cf->resize);
   luaL_unref(L, LUA_REGISTRYINDEX, cf->textinput);
 
   return 0;
}
 
static int tkL_customRename( lua_State *L )
{
   const char *s = luaL_checkstring(L,1);
   unsigned int wid = window_get( "dlgMsg" );
   if (wid == 0)
      NLUA_ERROR(L, _("custom dialogue not open"));
   window_setDisplayname( wid, s );
   return 0;
}
 
static int tkL_customFullscreen( lua_State *L )
{
   int enable = lua_toboolean(L,1);
   unsigned int wid = window_get( "dlgMsg" );
   if (wid == 0)
      NLUA_ERROR(L, _("custom dialogue not open"));
   dialogue_customFullscreen( enable );
   return 0;
}
 
static int tkL_customResize( lua_State *L )
{
   int w, h;
   unsigned int wid = window_get( "dlgMsg" );
   if (wid == 0)
      NLUA_ERROR(L, _("custom dialogue not open"));
   w = luaL_checkinteger(L,1);
   h = luaL_checkinteger(L,2);
   dialogue_customResize( w, h );
   return 0;
}
 
static int tkL_customSize( lua_State *L )
{
   int w, h;
   unsigned int wid = window_get( "dlgMsg" );
   if (wid == 0)
      NLUA_ERROR(L, _("custom dialogue not open"));
   window_dimWindow( wid, &w, &h );
   lua_pushinteger(L,w);
   lua_pushinteger(L,h);
   return 2;
}
 
static int tkL_customDone( lua_State *L )
{
   unsigned int wid = window_get( "dlgMsg" );
   if (wid == 0)
      NLUA_ERROR(L, _("custom dialogue not open"));
   lua_pushboolean(L, 1);
   lua_setglobal(L, TK_CUSTOMDONE );
   return 0;
}
 
static int cust_pcall( lua_State *L, int nargs, int nresults )
{
   int errf, ret;
 
#if DEBUGGING
   errf = lua_gettop(L) - nargs;
   lua_pushcfunction(L, nlua_errTrace);
   lua_insert(L, errf);
#else /* DEBUGGING */
   errf = 0;
#endif /* DEBUGGING */
 
   ret = lua_pcall( L, nargs, nresults, errf );
 
#if DEBUGGING
   lua_remove(L, errf);
#endif /* DEBUGGING */
 
   return ret;
}
static int cust_update( double dt, void* data )
{
   int ret;
   custom_functions_t *cf = (custom_functions_t*) data;
   if (cf->done)
      return 1; /* Mark done. */
   lua_State *L = cf->L;
   lua_rawgeti(L, LUA_REGISTRYINDEX, cf->update);
   lua_pushnumber(L, dt);
   if (cust_pcall( L, 1, 0 )) {
      cf->done = 1;
      WARN(_("Custom dialogue internal error: %s"), lua_tostring(L,-1));
      lua_pop(L,1);
      return 1;
   }
   /* Check if done. */
   lua_getglobal(L, TK_CUSTOMDONE );
   ret = lua_toboolean(L, -1);
   lua_pop(L, 1);
   return ret;
}
static void cust_render( double x, double y, double w, double h, void* data )
{
   custom_functions_t *cf = (custom_functions_t*) data;
   if (cf->done)
      return;
   lua_State *L = cf->L;
   lua_rawgeti(L, LUA_REGISTRYINDEX, cf->draw);
   lua_pushnumber(L, x);
   lua_pushnumber(L, y);
   lua_pushnumber(L, w);
   lua_pushnumber(L, h);
   if (cust_pcall( L, 4, 0 )) {
      cf->done = 1;
      WARN(_("Custom dialogue internal error: %s"), lua_tostring(L,-1));
      lua_pop(L,1);
   }
}
static int cust_event( unsigned int wid, SDL_Event *event, void* data )
{
   (void) wid;
   double x, y;
   custom_functions_t *cf = (custom_functions_t*) data;
   if (cf->done)
      return 0;
 
   /* Handle all the events. */
   switch (event->type) {
      case SDL_MOUSEBUTTONDOWN:
         x = gl_screen.x;
         y = gl_screen.y;
         return cust_mouse( 1, event->button.button, event->button.x+x, event->button.y+y, cf );
      case SDL_MOUSEBUTTONUP:
         x = gl_screen.x;
         y = gl_screen.y;
         return cust_mouse( 2, event->button.button, event->button.x+x, event->button.y+y, cf );
      case SDL_MOUSEMOTION:
         x = gl_screen.x;
         y = gl_screen.y;
         return cust_mouse( 3, -1, event->button.x+x, event->button.y+y, cf );
      case SDL_MOUSEWHEEL:
         return cust_mouse( 4, -1, event->wheel.x, event->wheel.y, cf );
 
      case SDL_KEYDOWN:
         return cust_key( event->key.keysym.sym, event->key.keysym.mod, 1, cf );
      case SDL_KEYUP:
         return cust_key( event->key.keysym.sym, event->key.keysym.mod, 0, cf );
 
      case SDL_TEXTINPUT:
         return cust_text( event->text.text, cf );
 
      case SDL_WINDOWEVENT:
         return cust_event_window( event->window.event, event->window.data1, event->window.data2, cf );
 
      default:
         return 0;
   }
 
   return 0;
}
static int cust_key( SDL_Keycode key, SDL_Keymod mod, int pressed, custom_functions_t *cf )
{
   int b;
   lua_State *L = cf->L;
   lua_rawgeti(L, LUA_REGISTRYINDEX, cf->keyboard);
   lua_pushboolean(L, pressed );
   lua_pushstring(L, SDL_GetKeyName(key));
   lua_pushstring(L, input_modToText(mod));
   if (cust_pcall( L, 3, 1 )) {
      cf->done = 1;
      WARN(_("Custom dialogue internal error: %s"), lua_tostring(L,-1));
      lua_pop(L,1);
      return 0;
   }
   b = lua_toboolean(L, -1);
   lua_pop(L,1);
   return b;
}
static int cust_text( const char *str, custom_functions_t *cf )
{
   int b;
   lua_State *L = cf->L;
   lua_rawgeti(L, LUA_REGISTRYINDEX, cf->textinput);
   lua_pushstring(L, str);
   if (cust_pcall( L, 1, 1 )) {
      cf->done = 1;
      WARN(_("Custom dialogue internal error: %s"), lua_tostring(L,-1));
      lua_pop(L,1);
      return 0;
   }
   b = lua_toboolean(L, -1);
   lua_pop(L,1);
   return b;
}
static int cust_mouse( int type, int button, double x, double y, custom_functions_t *cf )
{
   int b, nargs = 3;
   lua_State *L = cf->L;
   lua_rawgeti(L, LUA_REGISTRYINDEX, cf->mouse);
   lua_pushnumber(L, x);
   lua_pushnumber(L, y);
   lua_pushnumber(L, type);
   if (type < 3) {
      switch (button) {
         case SDL_BUTTON_LEFT:  button=1; break;
         case SDL_BUTTON_RIGHT: button=2; break;
         default:               button=3; break;
      }
      lua_pushnumber(L, button);
      nargs++;
   }
   if (cust_pcall( L, nargs, 1 )) {
      cf->done = 1;
      WARN(_("Custom dialogue internal error: %s"), lua_tostring(L,-1));
      lua_pop(L,1);
      return 0;
   }
   b = lua_toboolean(L, -1);
   lua_pop(L,1);
   return b;
}
static int cust_event_window( SDL_WindowEventID event, Sint32 w, Sint32 h, custom_functions_t *cf )
{
   int b;
   lua_State *L = cf->L;
 
   if (event == SDL_WINDOWEVENT_RESIZED)
      return 1;
 
   lua_rawgeti(L, LUA_REGISTRYINDEX, cf->resize);
   lua_pushinteger(L, w );
   lua_pushinteger(L, h );
   if (cust_pcall( L, 2, 1 )) {
      cf->done = 1;
      WARN(_("Custom dialogue internal error: %s"), lua_tostring(L,-1));
      lua_pop(L,1);
      return 0;
   }
   b = lua_toboolean(L, -1);
   lua_pop(L,1);
   return b;
}