input.c 11.7 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
/*****************************************************************************
 * input.c
 *****************************************************************************
 * Copyright (C) 2007-2008 the VideoLAN team
 * $Id$
 *
 * Authors: Antoine Cellerier <dionoea at videolan tod org>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
#ifndef  _GNU_SOURCE
#   define  _GNU_SOURCE
#endif

#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

#include <vlc_common.h>
#include <vlc_meta.h>

#include <vlc_playlist.h>

#include <lua.h>        /* Low level lua C API */
#include <lauxlib.h>    /* Higher level C API */
Pierre's avatar
Pierre committed
42
#include <assert.h>
43 44 45 46 47

#include "input.h"
#include "playlist.h"
#include "../vlc.h"
#include "../libs.h"
48
#include "../extension.h"
49

50 51 52
static const luaL_Reg vlclua_input_reg[];
static const luaL_Reg vlclua_input_item_reg[];

53 54
input_thread_t * vlclua_get_input_internal( lua_State *L )
{
55 56 57 58
    extension_t *p_extension = vlclua_extension_get( L );
    if( p_extension )
    {
        input_thread_t *p_input = p_extension->p_sys->p_input;
59
        if( p_input )
60 61 62 63 64
        {
            vlc_object_hold(p_input);
            return p_input;
        }
    }
65
    playlist_t *p_playlist = vlclua_get_playlist_internal( L );
66
    input_thread_t *p_input = playlist_CurrentInput( p_playlist );
67
    vlclua_release_playlist_internal( p_playlist );
68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95
    return p_input;
}

static int vlclua_input_info( lua_State *L )
{
    input_thread_t * p_input = vlclua_get_input_internal( L );
    int i_cat;
    int i;
    if( !p_input ) return vlclua_error( L );
    //vlc_mutex_lock( &input_GetItem(p_input)->lock );
    i_cat = input_GetItem(p_input)->i_categories;
    lua_createtable( L, 0, i_cat );
    for( i = 0; i < i_cat; i++ )
    {
        info_category_t *p_category = input_GetItem(p_input)->pp_categories[i];
        int i_infos = p_category->i_infos;
        int j;
        lua_pushstring( L, p_category->psz_name );
        lua_createtable( L, 0, i_infos );
        for( j = 0; j < i_infos; j++ )
        {
            info_t *p_info = p_category->pp_infos[j];
            lua_pushstring( L, p_info->psz_name );
            lua_pushstring( L, p_info->psz_value );
            lua_settable( L, -3 );
        }
        lua_settable( L, -3 );
    }
96
    vlc_object_release( p_input );
97 98 99
    return 1;
}

100
static int vlclua_input_is_playing( lua_State *L )
101 102 103
{
    input_thread_t * p_input = vlclua_get_input_internal( L );
    lua_pushboolean( L, !!p_input );
104 105
    if( p_input )
        vlc_object_release( p_input );
106 107 108
    return 1;
}

109
static int vlclua_input_get_title( lua_State *L )
110 111 112 113 114 115 116 117 118 119 120 121
{
    input_thread_t *p_input = vlclua_get_input_internal( L );
    if( !p_input )
        lua_pushnil( L );
    else
    {
        lua_pushstring( L, input_GetItem(p_input)->psz_name );
        vlc_object_release( p_input );
    }
    return 1;
}

122 123 124 125 126 127 128 129 130 131 132
static int vlclua_input_metas_internal( lua_State *L, input_item_t *p_item )
{
    if( !p_item )
    {
        lua_pushnil( L );
        return 1;
    }

    lua_newtable( L );
    char *psz_meta;

133 134 135 136
    psz_meta = input_item_GetName( p_item );
    lua_pushstring( L, psz_meta );
    lua_setfield( L, -2, "filename" );
    free( psz_meta );
137

138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163
#define PUSH_META( n, m ) \
    psz_meta = input_item_GetMeta( p_item, vlc_meta_ ## n ); \
    lua_pushstring( L, psz_meta ); \
    lua_setfield( L, -2, m ); \
    free( psz_meta )

    PUSH_META( Title, "title" );
    PUSH_META( Artist, "artist" );
    PUSH_META( Genre, "genre" );
    PUSH_META( Copyright, "copyright" );
    PUSH_META( Album, "album" );
    PUSH_META( TrackNumber, "track_number" );
    PUSH_META( Description, "description" );
    PUSH_META( Rating, "rating" );
    PUSH_META( Date, "date" );
    PUSH_META( Setting, "setting" );
    PUSH_META( URL, "url" );
    PUSH_META( Language, "language" );
    PUSH_META( NowPlaying, "now_playing" );
    PUSH_META( Publisher, "publisher" );
    PUSH_META( EncodedBy, "encoded_by" );
    PUSH_META( ArtworkURL, "artwork_url" );
    PUSH_META( TrackID, "track_id" );

#undef PUSH_META

164 165 166 167 168 169 170 171 172 173 174 175 176 177
    vlc_mutex_lock(&p_item->lock);
    if (p_item->p_meta) {
        char ** names = vlc_meta_CopyExtraNames(p_item->p_meta);
        for(int i = 0; names[i]; i++)
        {
            const char *meta = vlc_meta_GetExtra(p_item->p_meta, names[i]);
            lua_pushstring( L, meta );
            lua_setfield( L, -2, names[i] );
            free(names[i]);
        }
        free(names);
    }
    vlc_mutex_unlock(&p_item->lock);

178 179 180 181 182 183 184 185 186 187 188 189 190 191
    return 1;
}

static int vlclua_input_metas( lua_State *L )
{
    input_thread_t *p_input = vlclua_get_input_internal( L );
    input_item_t *p_item = p_input && p_input->p
                         ? input_GetItem( p_input ) : NULL;
    vlclua_input_metas_internal( L, p_item );
    if( p_input )
        vlc_object_release( p_input );
    return 1;
}

192 193 194
static int vlclua_input_stats( lua_State *L )
{
    input_thread_t *p_input = vlclua_get_input_internal( L );
195 196
    input_item_t *p_item = p_input && p_input->p
                         ? input_GetItem( p_input ) : NULL;
197 198 199
    lua_newtable( L );
    if( p_item )
    {
200
        vlc_mutex_lock( &p_item->p_stats->lock );
201 202 203 204 205 206 207 208
#define STATS_INT( n ) lua_pushinteger( L, p_item->p_stats->i_ ## n ); \
                       lua_setfield( L, -2, #n );
#define STATS_FLOAT( n ) lua_pushnumber( L, p_item->p_stats->f_ ## n ); \
                         lua_setfield( L, -2, #n );
        STATS_INT( read_bytes )
        STATS_FLOAT( input_bitrate )
        STATS_INT( demux_read_bytes )
        STATS_FLOAT( demux_bitrate )
ivoire's avatar
ivoire committed
209 210
        STATS_INT( demux_corrupted )
        STATS_INT( demux_discontinuity )
211 212 213 214 215 216 217 218 219 220 221
        STATS_INT( decoded_video )
        STATS_INT( displayed_pictures )
        STATS_INT( lost_pictures )
        STATS_INT( decoded_audio )
        STATS_INT( played_abuffers )
        STATS_INT( lost_abuffers )
        STATS_INT( sent_packets )
        STATS_INT( sent_bytes )
        STATS_FLOAT( send_bitrate )
#undef STATS_INT
#undef STATS_FLOAT
222
        vlc_mutex_unlock( &p_item->p_stats->lock );
223
    }
224 225
    if( p_input )
        vlc_object_release( p_input );
226 227 228
    return 1;
}

229 230 231 232 233 234 235 236 237 238 239 240 241
static int vlclua_input_add_subtitle( lua_State *L )
{
    input_thread_t *p_input = vlclua_get_input_internal( L );
    if( !p_input )
        return luaL_error( L, "can't add subtitle: no current input" );
    if( !lua_isstring( L, 1 ) )
        return luaL_error( L, "vlc.input.add_subtitle() usage: (url)" );
    const char *psz_url = luaL_checkstring( L, 1 );
    input_AddSubtitle( p_input, psz_url, false );
    vlc_object_release( p_input );
    return 1;
}

242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271
/*****************************************************************************
 * Input items
 *****************************************************************************/

static input_item_t* vlclua_input_item_get_internal( lua_State *L )
{
    input_item_t **pp_item = luaL_checkudata( L, 1, "input_item" );
    input_item_t *p_item = *pp_item;

    if( !p_item )
        luaL_error( L, "script went completely foobar" );

    return p_item;
}

/* Garbage collection of an input_item_t */
static int vlclua_input_item_delete( lua_State *L )
{
    input_item_t **pp_item = luaL_checkudata( L, 1, "input_item" );
    input_item_t *p_item = *pp_item;

    if( !p_item )
        return luaL_error( L, "script went completely foobar" );

    *pp_item = NULL;
    vlc_gc_decref( p_item );

    return 1;
}

272
static int vlclua_input_item_get( lua_State *L, input_item_t *p_item )
273 274 275 276 277 278 279 280 281 282 283 284 285 286
{
    vlc_gc_incref( p_item );
    input_item_t **pp = lua_newuserdata( L, sizeof( void* ) );
    *pp = p_item;

    if( luaL_newmetatable( L, "input_item" ) )
    {
        lua_newtable( L );
        luaL_register( L, NULL, vlclua_input_item_reg );
        lua_setfield( L, -2, "__index" );
        lua_pushcfunction( L, vlclua_input_item_delete );
        lua_setfield( L, -2, "__gc" );
    }

287
    lua_setmetatable(L, -2);
288

289 290 291 292 293 294 295 296 297 298 299 300 301 302 303
    return 1;
}

static int vlclua_input_item_get_current( lua_State *L )
{
    input_thread_t *p_input = vlclua_get_input_internal( L );
    input_item_t *p_item = ( p_input && p_input->p ) ? input_GetItem( p_input ) : NULL;
    if( !p_item )
    {
        lua_pushnil( L );
        if( p_input ) vlc_object_release( p_input );
        return 1;
    }

    vlclua_input_item_get( L, p_item );
304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321

    if( p_input ) vlc_object_release( p_input );
    return 1;
}

static int vlclua_input_item_metas( lua_State *L )
{
    vlclua_input_metas_internal( L, vlclua_input_item_get_internal( L ) );
    return 1;
}

static int vlclua_input_item_set_meta( lua_State *L )
{
    input_item_t *p_item = vlclua_input_item_get_internal( L );
    lua_settop( L, 1 + 2 ); // two arguments
    const char *psz_name = luaL_checkstring( L, 2 ),
               *psz_value = luaL_checkstring( L, 3 );

322
#define META_TYPE( n, s ) { s, vlc_meta_ ## n },
323 324 325 326 327
    static const struct
    {
        const char *psz_name;
        vlc_meta_type_t type;
    } pp_meta_types[] = {
328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344
        META_TYPE( Title, "title" )
        META_TYPE( Artist, "artist" )
        META_TYPE( Genre, "genre" )
        META_TYPE( Copyright, "copyright" )
        META_TYPE( Album, "album" )
        META_TYPE( TrackNumber, "track_number" )
        META_TYPE( Description, "description" )
        META_TYPE( Rating, "rating" )
        META_TYPE( Date, "date" )
        META_TYPE( Setting, "setting" )
        META_TYPE( URL, "url" )
        META_TYPE( Language, "language" )
        META_TYPE( NowPlaying, "now_playing" )
        META_TYPE( Publisher, "publisher" )
        META_TYPE( EncodedBy, "encoded_by" )
        META_TYPE( ArtworkURL, "artwork_url" )
        META_TYPE( TrackID, "track_id" )
345 346 347 348 349 350 351 352 353
    };
#undef META_TYPE

    vlc_meta_type_t type = vlc_meta_Title;
    for( unsigned i = 0; i < VLC_META_TYPE_COUNT; i++ )
    {
        if( !strcasecmp( pp_meta_types[i].psz_name, psz_name ) )
        {
            type = pp_meta_types[i].type;
354 355
            input_item_SetMeta( p_item, type, psz_value );
            return 1;
356 357 358
        }
    }

359
    vlc_meta_AddExtra( p_item->p_meta, psz_name, psz_value );
360 361 362
    return 1;
}

363
/*****************************************************************************
364
 * Lua bindings
365 366 367
 *****************************************************************************/
static const luaL_Reg vlclua_input_reg[] = {
    { "info", vlclua_input_info },
368 369 370
    { "is_playing", vlclua_input_is_playing },
    { "get_title", vlclua_input_get_title },
    { "metas", vlclua_input_metas },
371
    { "item", vlclua_input_item_get_current },
372
    { "stats", vlclua_input_stats },
373
    { "add_subtitle", vlclua_input_add_subtitle },
374 375 376 377 378 379 380 381 382
    { NULL, NULL }
};

void luaopen_input( lua_State *L )
{
    lua_newtable( L );
    luaL_register( L, NULL, vlclua_input_reg );
    lua_setfield( L, -2, "input" );
}
383 384 385 386 387 388

static const luaL_Reg vlclua_input_item_reg[] = {
    { "metas", vlclua_input_item_metas },
    { "set_meta", vlclua_input_item_set_meta },
    { NULL, NULL }
};
389 390 391 392


void luaopen_input_item( lua_State *L, input_item_t *item )
{
Pierre's avatar
Pierre committed
393
    assert(item);
394 395 396
    vlclua_input_item_get( L, item );
    lua_setfield( L, -2, "item" );
}