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

   1 /*****************************************************************************
   2  * objects.c: Generic lua<->vlc object wrapper
   3  *****************************************************************************
   4  * Copyright (C) 2007 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/vlc.h>
  36
  37 #include <lua.h>        /* Low level lua C API */
  38 #include <lauxlib.h>    /* Higher level C API */
  39 #include <lualib.h>     /* Lua libs */
  40
  41 #include "vlc.h"
  42
  43 typedef struct
  44 {
  45     vlc_object_t *p_obj;
  46 } vlclua_object_t;
  47
  48 /*****************************************************************************
  49  * Generic vlc_object_t wrapper creation
  50  *****************************************************************************/
  51 int __vlclua_push_vlc_object( lua_State *L, vlc_object_t *p_obj,
  52                               lua_CFunction pf_gc )
  53 {
  54     vlclua_object_t *p_ud = (vlclua_object_t *)
  55         lua_newuserdata( L, sizeof( vlclua_object_t ) );
  56     p_ud->p_obj = p_obj;
  57     lua_newtable( L );
  58     /* Hide the metatable */
  59     lua_pushstring( L, "__metatable" );
  60     lua_pushstring( L, "none of your business" );
  61     lua_settable( L, -3 );
  62     if( pf_gc )
  63     {
  64         /* Set the garbage collector if needed */
  65         lua_pushstring( L, "__gc" );
  66         lua_pushcfunction( L, pf_gc );
  67         lua_settable( L, -3 );
  68     }
  69     lua_setmetatable( L, -2 );
  70     return 1;
  71 }
  72
  73 int vlclua_gc_release( lua_State *L )
  74 {
  75     vlclua_object_t *p_ud = (vlclua_object_t *)lua_touserdata( L, -1 );
  76     lua_pop( L, 1 );
  77     vlc_object_release( p_ud->p_obj );
  78     return 0;
  79 }
  80
  81 vlc_object_t *vlclua_checkobject( lua_State *L, int narg, int i_type )
  82 {
  83     vlclua_object_t *p_obj = (vlclua_object_t *)luaL_checkuserdata( L, narg, sizeof( vlclua_object_t ) );
  84     /* TODO: add some metatable based method to check that this isn't
  85      * any userdata, but really some vlc object */
  86     if( i_type )
  87     {
  88         if( p_obj->p_obj->i_object_type == i_type )
  89             return p_obj->p_obj;
  90         else
  91         {
  92             luaL_error( L, "VLC object type doesn't match requirements." );
  93             return NULL; /* luaL_error alread longjmp-ed out of here.
  94                           * This is to make gcc happy */
  95         }
  96     }
  97     else
  98     {
  99         return p_obj->p_obj;
 100     }
 101 }
 102
 103 static int vlc_object_type_from_string( const char *psz_name )
 104 {
 105     static const struct
 106     {
 107         int i_type;
 108         const char *psz_name;
 109     } pp_objects[] =
 110         { { VLC_OBJECT_GLOBAL, "global" },
 111           { VLC_OBJECT_LIBVLC, "libvlc" },
 112           { VLC_OBJECT_MODULE, "module" },
 113           { VLC_OBJECT_INTF, "intf" },
 114           { VLC_OBJECT_PLAYLIST, "playlist" },
 115           { VLC_OBJECT_INPUT, "input" },
 116           { VLC_OBJECT_DECODER, "decoder" },
 117           { VLC_OBJECT_VOUT, "vout" },
 118           { VLC_OBJECT_AOUT, "aout" },
 119           { VLC_OBJECT_SOUT, "sout" },
 120           { VLC_OBJECT_HTTPD, "httpd" },
 121           { VLC_OBJECT_PACKETIZER, "packetizer" },
 122           { VLC_OBJECT_ENCODER, "encoder" },
 123           { VLC_OBJECT_DIALOGS, "dialogs" },
 124           { VLC_OBJECT_VLM, "vlm" },
 125           { VLC_OBJECT_ANNOUNCE, "announce" },
 126           { VLC_OBJECT_DEMUX, "demux" },
 127           { VLC_OBJECT_ACCESS, "access" },
 128           { VLC_OBJECT_STREAM, "stream" },
 129           { VLC_OBJECT_OPENGL, "opengl" },
 130           { VLC_OBJECT_FILTER, "filter" },
 131           { VLC_OBJECT_XML, "xml" },
 132           { VLC_OBJECT_OSDMENU, "osdmenu" },
 133           { VLC_OBJECT_HTTPD_HOST, "httpd_host" },
 134           { VLC_OBJECT_META_ENGINE, "meta_engine" },
 135           { VLC_OBJECT_GENERIC, "generic" },
 136           { 0, "" } };
 137     int i;
 138     for( i = 0; pp_objects[i].i_type; i++ )
 139     {
 140         if( !strcmp( psz_name, pp_objects[i].psz_name ) )
 141             return pp_objects[i].i_type;
 142     }
 143     return 0;
 144 }
 145
 146 static int vlc_object_search_mode_from_string( const char *psz_name )
 147 {
 148     static const struct
 149     {
 150         int i_mode;
 151         const char *psz_name;
 152     } pp_modes[] =
 153         { { FIND_PARENT, "parent" },
 154           { FIND_CHILD, "child" },
 155           { FIND_ANYWHERE, "anywhere" },
 156           { 0, "" } };
 157     int i;
 158     for( i = 0; pp_modes[i].i_mode; i++ )
 159     {
 160         if( !strcmp( psz_name, pp_modes[i].psz_name ) )
 161             return pp_modes[i].i_mode;
 162     }
 163     return 0;
 164 }
 165
 166 int vlclua_object_find( lua_State *L )
 167 {
 168     const char *psz_type = luaL_checkstring( L, 2 );
 169     const char *psz_mode = luaL_checkstring( L, 3 );
 170
 171     vlc_object_t *p_this;
 172     int i_type = vlc_object_type_from_string( psz_type );
 173     int i_mode = vlc_object_search_mode_from_string( psz_mode );
 174     vlc_object_t *p_result;
 175
 176     if( !i_type )
 177         return luaL_error( L, "\"%s\" is not a valid object type.", psz_type );
 178     if( !i_mode )
 179         return luaL_error( L, "\"%s\" is not a valid search mode.", psz_mode );
 180
 181     if( lua_type( L, 1 ) == LUA_TNIL )
 182         p_this = vlclua_get_this( L );
 183     else
 184         p_this = vlclua_checkobject( L, 1, 0 );
 185
 186     p_result = vlc_object_find( p_this, i_type, i_mode );
 187     if( !p_result )
 188         lua_pushnil( L );
 189     else
 190         vlclua_push_vlc_object( L, p_result, vlclua_gc_release );
 191     return 1;
 192 }
 193
 194 int vlclua_object_find_name( lua_State *L )
 195 {
 196     const char *psz_name = luaL_checkstring( L, 2 );
 197     const char *psz_mode = luaL_checkstring( L, 3 );
 198
 199     vlc_object_t *p_this;
 200     int i_mode = vlc_object_search_mode_from_string( psz_mode );
 201     vlc_object_t *p_result;
 202
 203     if( !i_mode )
 204         return luaL_error( L, "\"%s\" is not a valid search mode.",
 205                            psz_mode );
 206
 207     if( lua_type( L, 1 ) == LUA_TNIL )
 208         p_this = vlclua_get_this( L );
 209     else
 210         p_this = vlclua_checkobject( L, 1, 0 );
 211
 212     p_result = vlc_object_find_name( p_this, psz_name, i_mode );
 213     if( !p_result )
 214         lua_pushnil( L );
 215     else
 216         vlclua_push_vlc_object( L, p_result, vlclua_gc_release );
 217     return 1;
 218 }