git.sesse.net Git - vlc/blob - modules/misc/lua/libs/objects.c

   1 /*****************************************************************************
   2  * objects.c: Generic lua<->vlc object wrapper
   3  *****************************************************************************
   4  * Copyright (C) 2007-2008 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Antoine Cellerier <dionoea at videolan tod org>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #ifndef  _GNU_SOURCE
  28 #   define  _GNU_SOURCE
  29 #endif
  30
  31 #ifdef HAVE_CONFIG_H
  32 # include "config.h"
  33 #endif
  34
  35 #include <vlc_common.h>
  36
  37 #include <lua.h>        /* Low level lua C API */
  38 #include <lauxlib.h>    /* Higher level C API */
  39
  40 #include "../vlc.h"
  41 #include "../libs.h"
  42 #include "objects.h"
  43 #include "playlist.h"
  44 #include "input.h"
  45
  46 typedef struct
  47 {
  48     vlc_object_t *p_obj;
  49 } vlclua_object_t;
  50
  51 /*****************************************************************************
  52  * Generic vlc_object_t wrapper creation
  53  *****************************************************************************/
  54 int __vlclua_push_vlc_object( lua_State *L, vlc_object_t *p_obj,
  55                               lua_CFunction pf_gc )
  56 {
  57     vlc_object_t **udata = (vlc_object_t **)
  58         lua_newuserdata( L, sizeof( vlc_object_t * ) );
  59     *udata = p_obj;
  60
  61     if( luaL_newmetatable( L, "vlc_object" ) )
  62     {
  63         /* Hide the metatable */
  64         lua_pushstring( L, "none of your business" );
  65         lua_setfield( L, -2, "__metatable" );
  66         if( pf_gc ) /* FIXME */
  67         {
  68             /* Set the garbage collector if needed */
  69             lua_pushcfunction( L, pf_gc );
  70             lua_setfield( L, -2, "__gc" );
  71         }
  72     }
  73     lua_setmetatable( L, -2 );
  74     return 1;
  75 }
  76
  77 int vlclua_gc_release( lua_State *L )
  78 {
  79     vlc_object_t **p_obj = (vlc_object_t **)luaL_checkudata( L, 1, "vlc_object" );
  80     lua_pop( L, 1 );
  81     vlc_object_release( *p_obj );
  82     return 0;
  83 }
  84
  85 static int vlc_object_type_from_string( const char *psz_name )
  86 {
  87     static const struct
  88     {
  89         int i_type;
  90         const char *psz_name;
  91     } pp_objects[] =
  92         { { VLC_OBJECT_LIBVLC, "libvlc" },
  93           { VLC_OBJECT_MODULE, "module" },
  94           { VLC_OBJECT_INTF, "intf" },
  95           { VLC_OBJECT_INPUT, "input" },
  96           { VLC_OBJECT_DECODER, "decoder" },
  97           { VLC_OBJECT_VOUT, "vout" },
  98           { VLC_OBJECT_AOUT, "aout" },
  99           { VLC_OBJECT_PACKETIZER, "packetizer" },
 100           { VLC_OBJECT_ENCODER, "encoder" },
 101           { VLC_OBJECT_DIALOGS, "dialogs" },
 102           { VLC_OBJECT_ANNOUNCE, "announce" },
 103           { VLC_OBJECT_DEMUX, "demux" },
 104           { VLC_OBJECT_ACCESS, "access" },
 105           { VLC_OBJECT_STREAM, "stream" },
 106           { VLC_OBJECT_OPENGL, "opengl" },
 107           { VLC_OBJECT_FILTER, "filter" },
 108           { VLC_OBJECT_OSDMENU, "osdmenu" },
 109           { VLC_OBJECT_HTTPD_HOST, "httpd_host" },
 110           { VLC_OBJECT_INTERACTION, "interaction" },
 111           { VLC_OBJECT_GENERIC, "generic" },
 112           { 0, "" } };
 113     int i;
 114     for( i = 0; pp_objects[i].i_type; i++ )
 115     {
 116         if( !strcmp( psz_name, pp_objects[i].psz_name ) )
 117             return pp_objects[i].i_type;
 118     }
 119     return 0;
 120 }
 121
 122 static int vlc_object_search_mode_from_string( const char *psz_name )
 123 {
 124     static const struct
 125     {
 126         int i_mode;
 127         const char *psz_name;
 128     } pp_modes[] =
 129         { { FIND_PARENT, "parent" },
 130           { FIND_CHILD, "child" },
 131           { FIND_ANYWHERE, "anywhere" },
 132           { 0, "" } };
 133     int i;
 134     for( i = 0; pp_modes[i].i_mode; i++ )
 135     {
 136         if( !strcmp( psz_name, pp_modes[i].psz_name ) )
 137             return pp_modes[i].i_mode;
 138     }
 139     return 0;
 140 }
 141
 142 static int vlclua_object_find( lua_State *L )
 143 {
 144     const char *psz_type = luaL_checkstring( L, 2 );
 145     const char *psz_mode = luaL_checkstring( L, 3 );
 146
 147     vlc_object_t *p_this;
 148     int i_type = vlc_object_type_from_string( psz_type );
 149     int i_mode = vlc_object_search_mode_from_string( psz_mode );
 150     vlc_object_t *p_result;
 151
 152     if( !i_type )
 153         return luaL_error( L, "\"%s\" is not a valid object type.", psz_type );
 154     if( !i_mode )
 155         return luaL_error( L, "\"%s\" is not a valid search mode.", psz_mode );
 156
 157     if( lua_type( L, 1 ) == LUA_TNIL )
 158         p_this = vlclua_get_this( L );
 159     else
 160     {
 161         vlc_object_t **p_obj = luaL_checkudata( L, 1, "vlc_object" );
 162         p_this = *p_obj;
 163     }
 164
 165     p_result = vlc_object_find( p_this, i_type, i_mode );
 166     if( !p_result )
 167         lua_pushnil( L );
 168     else
 169         vlclua_push_vlc_object( L, p_result, vlclua_gc_release );
 170     return 1;
 171 }
 172
 173 static int vlclua_object_find_name( lua_State *L )
 174 {
 175     const char *psz_name = luaL_checkstring( L, 2 );
 176     const char *psz_mode = luaL_checkstring( L, 3 );
 177
 178     vlc_object_t *p_this;
 179     int i_mode = vlc_object_search_mode_from_string( psz_mode );
 180     vlc_object_t *p_result;
 181
 182     if( !i_mode )
 183         return luaL_error( L, "\"%s\" is not a valid search mode.",
 184                            psz_mode );
 185
 186     if( lua_type( L, 1 ) == LUA_TNIL )
 187         p_this = vlclua_get_this( L );
 188     else
 189     {
 190         vlc_object_t **p_obj = luaL_checkudata( L, 1, "vlc_object" );
 191         p_this = *p_obj;
 192     }
 193
 194     p_result = vlc_object_find_name( p_this, psz_name, i_mode );
 195     if( !p_result )
 196         lua_pushnil( L );
 197     else
 198         vlclua_push_vlc_object( L, p_result, vlclua_gc_release );
 199     return 1;
 200 }
 201
 202 static int vlclua_get_libvlc( lua_State *L )
 203 {
 204     vlclua_push_vlc_object( L, vlclua_get_this( L )->p_libvlc,
 205                             NULL );
 206     return 1;
 207 }
 208
 209 static int vlclua_get_playlist( lua_State *L )
 210 {
 211     playlist_t *p_playlist = vlclua_get_playlist_internal( L );
 212     if( p_playlist )
 213     {
 214         vlclua_push_vlc_object( L, p_playlist, vlclua_gc_release );
 215     }
 216     else lua_pushnil( L );
 217     return 1;
 218 }
 219
 220 static int vlclua_get_input( lua_State *L )
 221 {
 222     input_thread_t *p_input = vlclua_get_input_internal( L );
 223     if( p_input )
 224     {
 225         vlclua_push_vlc_object( L, p_input, vlclua_gc_release );
 226     }
 227     else lua_pushnil( L );
 228     return 1;
 229 }
 230
 231 /*****************************************************************************
 232  *
 233  *****************************************************************************/
 234 static const luaL_Reg vlclua_object_reg[] = {
 235     { "input", vlclua_get_input },
 236     { "playlist", vlclua_get_playlist },
 237     { "libvlc", vlclua_get_libvlc },
 238     { "find", vlclua_object_find },
 239     { "find_name", vlclua_object_find_name },
 240     { NULL, NULL }
 241 };
 242
 243 void luaopen_object( lua_State *L )
 244 {
 245     lua_newtable( L );
 246     luaL_register( L, NULL, vlclua_object_reg );
 247     lua_setfield( L, -2, "object" );
 248 }