bluray.c 55.5 KB
Newer Older
1 2 3
/*****************************************************************************
 * bluray.c: Blu-ray disc support plugin
 *****************************************************************************
Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
4
 * Copyright © 2010-2012 VideoLAN, VLC authors and libbluray AUTHORS
5 6
 *
 * Authors: Jean-Baptiste Kempf <jb@videolan.org>
Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
7
 *          Hugo Beauzée-Luyssen <hugo@videolan.org>
8
 *
9 10 11
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
12 13 14 15
 * (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
Jean-Baptiste Kempf's avatar
LGPL  
Jean-Baptiste Kempf committed
16
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17
 * GNU Lesser General Public License for more details.
18
 *
19 20 21
 * You should have received a copy of the GNU Lesser 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.
22 23 24 25 26 27 28
 *****************************************************************************/

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

#include <assert.h>
29
#include <limits.h>                         /* PATH_MAX */
Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
30

31
#if defined (HAVE_MNTENT_H) && defined(HAVE_SYS_STAT_H)
Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
32 33
# include <mntent.h>
# include <sys/stat.h>
34
#endif
35

36
#ifdef __APPLE__
Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
37 38 39 40
# include <sys/stat.h>
# include <sys/param.h>
# include <sys/ucred.h>
# include <sys/mount.h>
41 42
#endif

43 44
#include <vlc_common.h>
#include <vlc_plugin.h>
45 46
#include <vlc_demux.h>                      /* demux_t */
#include <vlc_input.h>                      /* Seekpoints, chapters */
47
#include <vlc_atomic.h>
48
#include <vlc_dialog.h>                     /* BD+/AACS warnings */
49
#include <vlc_vout.h>                       /* vout_PutSubpicture / subpicture_t */
50
#include <vlc_url.h>                        /* vlc_path2uri */
51 52 53 54 55
#include <vlc_iso_lang.h>

/* FIXME we should find a better way than including that */
#include "../../src/text/iso-639_def.h"

56 57

#include <libbluray/bluray.h>
58
#include <libbluray/bluray-version.h>
59
#include <libbluray/keys.h>
60
#include <libbluray/meta_data.h>
61
#include <libbluray/overlay.h>
62 63 64 65 66

/*****************************************************************************
 * Module descriptor
 *****************************************************************************/

Rafaël Carré's avatar
Rafaël Carré committed
67 68 69 70 71
#define BD_MENU_TEXT        N_("Blu-ray menus")
#define BD_MENU_LONGTEXT    N_("Use Blu-ray menus. If disabled, "\
                                "the movie will start directly")
#define BD_REGION_TEXT      N_("Region code")
#define BD_REGION_LONGTEXT  N_("Blu-Ray player region code. "\
72 73 74 75 76 77 78 79
                                "Some discs can be played only with a correct region code.")

static const char *const ppsz_region_code[] = {
    "A", "B", "C" };
static const char *const ppsz_region_code_text[] = {
    "Region A", "Region B", "Region C" };

#define REGION_DEFAULT   1   /* Index to region list. Actual region code is (1<<REGION_DEFAULT) */
80
#define LANGUAGE_DEFAULT ("eng")
81

82
/* Callbacks */
Rafaël Carré's avatar
Rafaël Carré committed
83 84
static int  blurayOpen (vlc_object_t *);
static void blurayClose(vlc_object_t *);
85 86

vlc_module_begin ()
Rafaël Carré's avatar
Rafaël Carré committed
87 88
    set_shortname(N_("Blu-ray"))
    set_description(N_("Blu-ray Disc support (libbluray)"))
89

Rafaël Carré's avatar
Rafaël Carré committed
90 91 92 93 94 95
    set_category(CAT_INPUT)
    set_subcategory(SUBCAT_INPUT_ACCESS)
    set_capability("access_demux", 200)
    add_bool("bluray-menu", false, BD_MENU_TEXT, BD_MENU_LONGTEXT, false)
    add_string("bluray-region", ppsz_region_code[REGION_DEFAULT], BD_REGION_TEXT, BD_REGION_LONGTEXT, false)
        change_string_list(ppsz_region_code, ppsz_region_code_text)
96

Rafaël Carré's avatar
Rafaël Carré committed
97
    add_shortcut("bluray", "file")
98

Rafaël Carré's avatar
Rafaël Carré committed
99
    set_callbacks(blurayOpen, blurayClose)
100 101
vlc_module_end ()

102 103
/* libbluray's overlay.h defines 2 types of overlay (bd_overlay_plane_e). */
#define MAX_OVERLAY 2
104

105 106 107 108 109 110 111 112
typedef enum OverlayStatus {
    Closed = 0,
    ToDisplay,  //Used to mark the overlay to be displayed the first time.
    Displayed,
    Outdated    //used to update the overlay after it has been sent to the vout
} OverlayStatus;

typedef struct bluray_overlay_t
113
{
114
    atomic_flag         released_once;
115 116 117 118 119 120 121 122 123
    vlc_mutex_t         lock;
    subpicture_t        *p_pic;
    OverlayStatus       status;
    subpicture_region_t *p_regions;
} bluray_overlay_t;

struct  demux_sys_t
{
    BLURAY              *bluray;
124 125

    /* Titles */
126 127 128
    unsigned int        i_title;
    unsigned int        i_longest_title;
    input_title_t       **pp_title;
129

130 131 132 133
    vlc_mutex_t             pl_info_lock;
    BLURAY_TITLE_INFO      *p_pl_info;
    const BLURAY_CLIP_INFO *p_clip_info;

134
    /* Meta information */
135 136
    const META_DL       *p_meta;

137
    /* Menus */
138 139 140
    bluray_overlay_t    *p_overlays[MAX_OVERLAY];
    int                 current_overlay; // -1 if no current overlay;
    bool                b_menu;
Petri Hintukainen's avatar
Petri Hintukainen committed
141 142
    bool                b_menu_open;
    bool                b_popup_available;
143
    mtime_t             i_still_end_time;
144 145 146 147

    /* */
    input_thread_t      *p_input;
    vout_thread_t       *p_vout;
148

149
    /* TS stream */
150
    es_out_t            *p_out;
151
    vlc_array_t         es;
152
    int                 i_audio_stream; /* Selected audio stream. -1 if default */
153
    int                 i_spu_stream;   /* Selected subtitle stream. -1 if default */
154
    int                 i_video_stream;
155
    stream_t            *p_parser;
156
    bool                b_flushed;
157 158 159

    /* Used to store bluray disc path */
    char                *psz_bd_path;
160 161 162 163 164
};

struct subpicture_updater_sys_t
{
    bluray_overlay_t    *p_overlay;
165 166
};

167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203
/* Get a 3 char code
 * FIXME: partiallyy duplicated from src/input/es_out.c
 */
static const char *DemuxGetLanguageCode( demux_t *p_demux, const char *psz_var )
{
    const iso639_lang_t *pl;
    char *psz_lang;
    char *p;

    psz_lang = var_CreateGetString( p_demux, psz_var );
    if( !psz_lang )
        return LANGUAGE_DEFAULT;

    /* XXX: we will use only the first value
     * (and ignore other ones in case of a list) */
    if( ( p = strchr( psz_lang, ',' ) ) )
        *p = '\0';

    for( pl = p_languages; pl->psz_eng_name != NULL; pl++ )
    {
        if( *psz_lang == '\0' )
            continue;
        if( !strcasecmp( pl->psz_eng_name, psz_lang ) ||
            !strcasecmp( pl->psz_iso639_1, psz_lang ) ||
            !strcasecmp( pl->psz_iso639_2T, psz_lang ) ||
            !strcasecmp( pl->psz_iso639_2B, psz_lang ) )
            break;
    }

    free( psz_lang );

    if( pl->psz_eng_name != NULL )
        return pl->psz_iso639_2T;

    return LANGUAGE_DEFAULT;
}

204 205 206
/*****************************************************************************
 * Local prototypes
 *****************************************************************************/
Rafaël Carré's avatar
Rafaël Carré committed
207
static es_out_t *esOutNew(demux_t *p_demux);
208

Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
209 210
static int   blurayControl(demux_t *, int, va_list);
static int   blurayDemux(demux_t *);
211

212
static void  blurayInitTitles(demux_t *p_demux, int menu_titles);
Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
213
static int   bluraySetTitle(demux_t *p_demux, int i_title);
214

Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
215
static void  blurayOverlayProc(void *ptr, const BD_OVERLAY * const overlay);
216
static void  blurayArgbOverlayProc(void *ptr, const BD_ARGB_OVERLAY * const overlay);
217

Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
218 219
static int   onMouseEvent(vlc_object_t *p_vout, const char *psz_var,
                          vlc_value_t old, vlc_value_t val, void *p_data);
220

221 222
static void  blurayResetParser(demux_t *p_demux);

223 224 225
#define FROM_TICKS(a) (a*CLOCK_FREQ / INT64_C(90000))
#define TO_TICKS(a)   (a*INT64_C(90000)/CLOCK_FREQ)
#define CUR_LENGTH    p_sys->pp_title[p_demux->info.i_title]->i_length
226

227 228 229 230 231
/* */
static void FindMountPoint(char **file)
{
    char *device = *file;
#if defined (HAVE_MNTENT_H) && defined (HAVE_SYS_STAT_H)
232 233 234 235 236 237
    /* bd path may be a symlink (e.g. /dev/dvd -> /dev/sr0), so make sure
     * we look up the real device */
    char *bd_device = realpath(device, NULL);
    if (bd_device == NULL)
        return;

238
    struct stat st;
239
    if (lstat (bd_device, &st) == 0 && S_ISBLK (st.st_mode)) {
240 241 242 243 244 245 246 247 248 249 250 251 252
        FILE *mtab = setmntent ("/proc/self/mounts", "r");
        struct mntent *m, mbuf;
        char buf [8192];

        while ((m = getmntent_r (mtab, &mbuf, buf, sizeof(buf))) != NULL) {
            if (!strcmp (m->mnt_fsname, bd_device)) {
                free(device);
                *file = strdup(m->mnt_dir);
                break;
            }
        }
        endmntent (mtab);
    }
253 254
    free(bd_device);

255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274
#elif defined(__APPLE__)
    struct stat st;
    if (!stat (device, &st) && S_ISBLK (st.st_mode)) {
        int fs_count = getfsstat (NULL, 0, MNT_NOWAIT);
        if (fs_count > 0) {
            struct statfs mbuf[128];
            getfsstat (mbuf, fs_count * sizeof(mbuf[0]), MNT_NOWAIT);
            for (int i = 0; i < fs_count; ++i)
                if (!strcmp (mbuf[i].f_mntfromname, device)) {
                    free(device);
                    *file = strdup(mbuf[i].f_mntonname);
                    return;
                }
        }
    }
#else
# warning Disc device to mount point not implemented
#endif
}

275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295
/*****************************************************************************
 * cache current playlist (title) information
 *****************************************************************************/

static void setTitleInfo(demux_sys_t *p_sys, BLURAY_TITLE_INFO *info)
{
    vlc_mutex_lock(&p_sys->pl_info_lock);

    if (p_sys->p_pl_info) {
        bd_free_title_info(p_sys->p_pl_info);
    }
    p_sys->p_pl_info   = info;
    p_sys->p_clip_info = NULL;

    if (p_sys->p_pl_info && p_sys->p_pl_info->clip_count) {
        p_sys->p_clip_info = &p_sys->p_pl_info->clips[0];
    }

    vlc_mutex_unlock(&p_sys->pl_info_lock);
}

296 297 298
/*****************************************************************************
 * blurayOpen: module init function
 *****************************************************************************/
Rafaël Carré's avatar
Rafaël Carré committed
299
static int blurayOpen(vlc_object_t *object)
300
{
301 302
    demux_t *p_demux = (demux_t*)object;
    demux_sys_t *p_sys;
303

304
    const char *error_msg = NULL;
305
#define BLURAY_ERROR(s) do { error_msg = s; goto error; } while(0)
306

307
    if (strcmp(p_demux->psz_access, "bluray")) {
308
        // TODO BDMV support, once we figure out what to do in libbluray
309 310 311
        return VLC_EGENERIC;
    }

312
    /* */
313
    p_demux->p_sys = p_sys = calloc(1, sizeof(*p_sys));
Rafaël Carré's avatar
Rafaël Carré committed
314
    if (unlikely(!p_sys))
315
        return VLC_ENOMEM;
Rafaël Carré's avatar
Rafaël Carré committed
316

317
    p_sys->current_overlay = -1;
318
    p_sys->i_audio_stream = -1;
319
    p_sys->i_spu_stream = -1;
320
    p_sys->i_video_stream = -1;
321
    p_sys->i_still_end_time = 0;
322 323 324 325 326

    /* init demux info fields */
    p_demux->info.i_update    = 0;
    p_demux->info.i_title     = 0;
    p_demux->info.i_seekpoint = 0;
327

Rafaël Carré's avatar
Rafaël Carré committed
328
    TAB_INIT(p_sys->i_title, p_sys->pp_title);
329

330
    /* store current bd path */
Rafaël Carré's avatar
Rafaël Carré committed
331 332
    if (p_demux->psz_file)
        p_sys->psz_bd_path = strdup(p_demux->psz_file);
333

334
    /* If we're passed a block device, try to convert it to the mount point. */
335 336
    FindMountPoint(&p_sys->psz_bd_path);

337
    p_sys->bluray = bd_open(p_sys->psz_bd_path, NULL);
338
    if (!p_sys->bluray) {
339
        free(p_sys->psz_bd_path);
340 341 342 343
        free(p_sys);
        return VLC_EGENERIC;
    }

344 345
    vlc_mutex_init(&p_sys->pl_info_lock);

346 347
    /* Warning the user about AACS/BD+ */
    const BLURAY_DISC_INFO *disc_info = bd_get_disc_info(p_sys->bluray);
348 349

    /* Is it a bluray? */
350 351
    if (!disc_info->bluray_detected)
        BLURAY_ERROR(_("Path doesn't appear to be a Blu-ray"));
352

353 354
    msg_Info(p_demux, "First play: %i, Top menu: %i\n"
                      "HDMV Titles: %i, BD-J Titles: %i, Other: %i",
355 356 357 358 359 360
             disc_info->first_play_supported, disc_info->top_menu_supported,
             disc_info->num_hdmv_titles, disc_info->num_bdj_titles,
             disc_info->num_unsupported_titles);

    /* AACS */
    if (disc_info->aacs_detected) {
361
        msg_Dbg(p_demux, "Disc is using AACS");
362 363 364
        if (!disc_info->libaacs_detected)
            BLURAY_ERROR(_("This Blu-ray Disc needs a library for AACS decoding"
                      ", and your system does not have it."));
365
        if (!disc_info->aacs_handled) {
366 367
            if (disc_info->aacs_error_code) {
                switch (disc_info->aacs_error_code) {
368 369 370 371 372 373 374 375 376 377 378 379
                case BD_AACS_CORRUPTED_DISC:
                    BLURAY_ERROR(_("Blu-ray Disc is corrupted."));
                case BD_AACS_NO_CONFIG:
                    BLURAY_ERROR(_("Missing AACS configuration file!"));
                case BD_AACS_NO_PK:
                    BLURAY_ERROR(_("No valid processing key found in AACS config file."));
                case BD_AACS_NO_CERT:
                    BLURAY_ERROR(_("No valid host certificate found in AACS config file."));
                case BD_AACS_CERT_REVOKED:
                    BLURAY_ERROR(_("AACS Host certificate revoked."));
                case BD_AACS_MMC_FAILED:
                    BLURAY_ERROR(_("AACS MMC failed."));
380 381
                }
            }
382 383 384 385 386
        }
    }

    /* BD+ */
    if (disc_info->bdplus_detected) {
387
        msg_Dbg(p_demux, "Disc is using BD+");
388 389 390 391 392 393
        if (!disc_info->libbdplus_detected)
            BLURAY_ERROR(_("This Blu-ray Disc needs a library for BD+ decoding"
                      ", and your system does not have it."));
        if (!disc_info->bdplus_handled)
            BLURAY_ERROR(_("Your system BD+ decoding library does not work. "
                      "Missing configuration?"));
394 395
    }

396 397 398 399 400 401
    /* set player region code */
    char *psz_region = var_InheritString(p_demux, "bluray-region");
    unsigned int region = psz_region ? (psz_region[0] - 'A') : REGION_DEFAULT;
    free(psz_region);
    bd_set_player_setting(p_sys->bluray, BLURAY_PLAYER_SETTING_REGION_CODE, 1<<region);

402 403 404 405 406 407 408 409
    /* set preferred languages */
    const char *psz_code = DemuxGetLanguageCode( p_demux, "audio-language" );
    bd_set_player_setting_str(p_sys->bluray, BLURAY_PLAYER_SETTING_AUDIO_LANG, psz_code);
    psz_code = DemuxGetLanguageCode( p_demux, "sub-language" );
    bd_set_player_setting_str(p_sys->bluray, BLURAY_PLAYER_SETTING_PG_LANG,    psz_code);
    psz_code = DemuxGetLanguageCode( p_demux, "menu-language" );
    bd_set_player_setting_str(p_sys->bluray, BLURAY_PLAYER_SETTING_MENU_LANG,  psz_code);

410
    /* Get titles and chapters */
411 412
    p_sys->p_meta = bd_get_meta(p_sys->bluray);
    if (!p_sys->p_meta)
Rafaël Carré's avatar
Rafaël Carré committed
413
        msg_Warn(p_demux, "Failed to get meta info.");
414

415 416 417
    p_sys->b_menu = var_InheritBool(p_demux, "bluray-menu");

    blurayInitTitles(p_demux, disc_info->num_hdmv_titles + disc_info->num_bdj_titles + 1/*Top Menu*/ + 1/*First Play*/);
418

419 420 421 422 423
    /*
     * Initialize the event queue, so we can receive events in blurayDemux(Menu).
     */
    bd_get_event(p_sys->bluray, NULL);

424 425
    /* Registering overlay event handler */
    bd_register_overlay_proc(p_sys->bluray, p_demux, blurayOverlayProc);
426 427 428 429 430
    p_sys->p_input = demux_GetParentInput(p_demux);
    if (unlikely(!p_sys->p_input)) {
        msg_Err(p_demux, "Could not get parent input");
        goto error;
    }
431

Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
432
    if (p_sys->b_menu) {
433

434
        /* Register ARGB overlay handler for BD-J */
435
        if (disc_info->num_bdj_titles)
436 437
            bd_register_argb_overlay_proc(p_sys->bluray, p_demux, blurayArgbOverlayProc, NULL);

438
        /* libbluray will start playback from "First-Title" title */
439 440 441
        if (bd_play(p_sys->bluray) == 0)
            BLURAY_ERROR(_("Failed to start bluray playback. Please try without menu support."));

Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
442
    } else {
443
        /* set start title number */
444
        if (bluraySetTitle(p_demux, p_sys->i_longest_title) != VLC_SUCCESS) {
Rafaël Carré's avatar
Rafaël Carré committed
445
            msg_Err(p_demux, "Could not set the title %d", p_sys->i_longest_title);
446 447
            goto error;
        }
448 449
    }

450
    vlc_array_init(&p_sys->es);
Rafaël Carré's avatar
Rafaël Carré committed
451
    p_sys->p_out = esOutNew(p_demux);
Rafaël Carré's avatar
Rafaël Carré committed
452
    if (unlikely(p_sys->p_out == NULL))
453 454
        goto error;

Rafaël Carré's avatar
Rafaël Carré committed
455
    blurayResetParser(p_demux);
456 457
    if (!p_sys->p_parser) {
        msg_Err(p_demux, "Failed to create TS demuxer");
458
        goto error;
459 460 461 462
    }

    p_demux->pf_control = blurayControl;
    p_demux->pf_demux   = blurayDemux;
463 464

    return VLC_SUCCESS;
465 466 467

error:
    if (error_msg)
Christoph Miebach's avatar
Christoph Miebach committed
468
        dialog_Fatal(p_demux, _("Blu-ray error"), "%s", error_msg);
469 470
    blurayClose(object);
    return VLC_EGENERIC;
471
#undef BLURAY_ERROR
472 473 474 475 476 477
}


/*****************************************************************************
 * blurayClose: module destroy function
 *****************************************************************************/
Rafaël Carré's avatar
Rafaël Carré committed
478
static void blurayClose(vlc_object_t *object)
479
{
480 481 482
    demux_t *p_demux = (demux_t*)object;
    demux_sys_t *p_sys = p_demux->p_sys;

483 484
    setTitleInfo(p_sys, NULL);

485 486 487
    /*
     * Close libbluray first.
     * This will close all the overlays before we release p_vout
Rafaël Carré's avatar
Rafaël Carré committed
488
     * bd_close(NULL) can crash
489 490 491 492
     */
    assert(p_sys->bluray);
    bd_close(p_sys->bluray);

493
    if (p_sys->p_vout != NULL) {
Rafaël Carré's avatar
Rafaël Carré committed
494 495
        var_DelCallback(p_sys->p_vout, "mouse-moved", onMouseEvent, p_demux);
        var_DelCallback(p_sys->p_vout, "mouse-clicked", onMouseEvent, p_demux);
496
        vlc_object_release(p_sys->p_vout);
497
    }
498 499
    if (p_sys->p_input != NULL)
        vlc_object_release(p_sys->p_input);
500 501
    if (p_sys->p_parser)
        stream_Delete(p_sys->p_parser);
502 503
    if (p_sys->p_out != NULL)
        es_out_Delete(p_sys->p_out);
Rafaël Carré's avatar
Rafaël Carré committed
504 505
    assert(vlc_array_count(&p_sys->es) == 0);
    vlc_array_clear(&p_sys->es);
506

507 508 509
    /* Titles */
    for (unsigned int i = 0; i < p_sys->i_title; i++)
        vlc_input_title_Delete(p_sys->pp_title[i]);
Rafaël Carré's avatar
Rafaël Carré committed
510
    TAB_CLEAN(p_sys->i_title, p_sys->pp_title);
511

512 513
    vlc_mutex_destroy(&p_sys->pl_info_lock);

514
    free(p_sys->psz_bd_path);
515 516 517
    free(p_sys);
}

518 519 520 521 522 523 524 525
/*****************************************************************************
 * Elementary streams handling
 *****************************************************************************/

struct es_out_sys_t {
    demux_t *p_demux;
};

526 527 528 529 530
typedef struct  fmt_es_pair {
    int         i_id;
    es_out_id_t *p_es;
}               fmt_es_pair_t;

Rafaël Carré's avatar
Rafaël Carré committed
531
static int  findEsPairIndex(demux_sys_t *p_sys, int i_id)
532
{
Rafaël Carré's avatar
Rafaël Carré committed
533 534
    for (int i = 0; i < vlc_array_count(&p_sys->es); ++i)
        if (((fmt_es_pair_t*)vlc_array_item_at_index(&p_sys->es, i))->i_id == i_id)
535
            return i;
Rafaël Carré's avatar
Rafaël Carré committed
536

537 538 539
    return -1;
}

Rafaël Carré's avatar
Rafaël Carré committed
540
static int  findEsPairIndexByEs(demux_sys_t *p_sys, es_out_id_t *p_es)
541
{
Rafaël Carré's avatar
Rafaël Carré committed
542 543
    for (int i = 0; i < vlc_array_count(&p_sys->es); ++i)
        if (((fmt_es_pair_t*)vlc_array_item_at_index(&p_sys->es, i))->p_es == p_es)
544
            return i;
Rafaël Carré's avatar
Rafaël Carré committed
545

546 547 548
    return -1;
}

549 550 551 552 553 554
static void setStreamLang(es_format_t *p_fmt,
                          const BLURAY_STREAM_INFO *p_streams, int i_stream_count)
{
    for (int i = 0; i < i_stream_count; i++) {
        if (p_fmt->i_id == p_streams[i].pid) {
            free(p_fmt->psz_language);
555
            p_fmt->psz_language = strndup((const char *)p_streams[i].lang, 3);
556 557 558 559 560
            return;
        }
    }
}

Rafaël Carré's avatar
Rafaël Carré committed
561
static es_out_id_t *esOutAdd(es_out_t *p_out, const es_format_t *p_fmt)
562
{
563 564 565 566
    demux_sys_t *p_sys = p_out->p_sys->p_demux->p_sys;
    es_format_t fmt;

    es_format_Copy(&fmt, p_fmt);
567 568 569

    vlc_mutex_lock(&p_sys->pl_info_lock);

Rafaël Carré's avatar
Rafaël Carré committed
570
    switch (fmt.i_cat) {
571
    case VIDEO_ES:
Rafaël Carré's avatar
Rafaël Carré committed
572
        if (p_sys->i_video_stream != -1 && p_sys->i_video_stream != p_fmt->i_id)
573
            fmt.i_priority = ES_PRIORITY_NOT_SELECTABLE;
574 575
        break ;
    case AUDIO_ES:
Rafaël Carré's avatar
Rafaël Carré committed
576
        if (p_sys->i_audio_stream != -1 && p_sys->i_audio_stream != p_fmt->i_id)
577
            fmt.i_priority = ES_PRIORITY_NOT_SELECTABLE;
578 579
        if (p_sys->p_clip_info)
            setStreamLang(&fmt, p_sys->p_clip_info->audio_streams, p_sys->p_clip_info->audio_stream_count);
580 581
        break ;
    case SPU_ES:
582 583
        if (p_sys->i_spu_stream != -1 && p_sys->i_spu_stream != p_fmt->i_id)
            fmt.i_priority = ES_PRIORITY_NOT_SELECTABLE;
584 585
        if (p_sys->p_clip_info)
            setStreamLang(&fmt, p_sys->p_clip_info->pg_streams, p_sys->p_clip_info->pg_stream_count);
586 587 588
        break ;
    }

589
    vlc_mutex_unlock(&p_sys->pl_info_lock);
590

Rafaël Carré's avatar
Rafaël Carré committed
591 592
    es_out_id_t *p_es = es_out_Add(p_out->p_sys->p_demux->out, &fmt);
    if (p_fmt->i_id >= 0) {
593
        /* Ensure we are not overriding anything */
594
        int idx = findEsPairIndex(p_sys, p_fmt->i_id);
Rafaël Carré's avatar
Rafaël Carré committed
595 596 597
        if (idx == -1) {
            fmt_es_pair_t *p_pair = malloc(sizeof(*p_pair));
            if (likely(p_pair != NULL)) {
598 599
                p_pair->i_id = p_fmt->i_id;
                p_pair->p_es = p_es;
Rafaël Carré's avatar
Rafaël Carré committed
600
                msg_Info(p_out->p_sys->p_demux, "Adding ES %d", p_fmt->i_id);
601
                vlc_array_append(&p_sys->es, p_pair);
602 603 604
            }
        }
    }
605
    es_format_Clean(&fmt);
606
    return p_es;
607 608
}

Rafaël Carré's avatar
Rafaël Carré committed
609
static int esOutSend(es_out_t *p_out, es_out_id_t *p_es, block_t *p_block)
610
{
Rafaël Carré's avatar
Rafaël Carré committed
611
    return es_out_Send(p_out->p_sys->p_demux->out, p_es, p_block);
612 613
}

Rafaël Carré's avatar
Rafaël Carré committed
614
static void esOutDel(es_out_t *p_out, es_out_id_t *p_es)
615
{
Rafaël Carré's avatar
Rafaël Carré committed
616 617 618
    int idx = findEsPairIndexByEs(p_out->p_sys->p_demux->p_sys, p_es);
    if (idx >= 0) {
        free(vlc_array_item_at_index(&p_out->p_sys->p_demux->p_sys->es, idx));
619 620
        vlc_array_remove(&p_out->p_sys->p_demux->p_sys->es, idx);
    }
Rafaël Carré's avatar
Rafaël Carré committed
621
    es_out_Del(p_out->p_sys->p_demux->out, p_es);
622 623
}

Rafaël Carré's avatar
Rafaël Carré committed
624
static int esOutControl(es_out_t *p_out, int i_query, va_list args)
625
{
Rafaël Carré's avatar
Rafaël Carré committed
626
    return es_out_vaControl(p_out->p_sys->p_demux->out, i_query, args);
627 628
}

Rafaël Carré's avatar
Rafaël Carré committed
629
static void esOutDestroy(es_out_t *p_out)
630
{
Rafaël Carré's avatar
Rafaël Carré committed
631 632
    for (int i = 0; i < vlc_array_count(&p_out->p_sys->p_demux->p_sys->es); ++i)
        free(vlc_array_item_at_index(&p_out->p_sys->p_demux->p_sys->es, i));
633
    vlc_array_clear(&p_out->p_sys->p_demux->p_sys->es);
Rafaël Carré's avatar
Rafaël Carré committed
634 635
    free(p_out->p_sys);
    free(p_out);
636 637
}

Rafaël Carré's avatar
Rafaël Carré committed
638
static es_out_t *esOutNew(demux_t *p_demux)
639
{
Rafaël Carré's avatar
Rafaël Carré committed
640 641 642
    assert(vlc_array_count(&p_demux->p_sys->es) == 0);
    es_out_t    *p_out = malloc(sizeof(*p_out));
    if (unlikely(p_out == NULL))
643 644
        return NULL;

Rafaël Carré's avatar
Rafaël Carré committed
645 646 647 648 649
    p_out->pf_add       = esOutAdd;
    p_out->pf_control   = esOutControl;
    p_out->pf_del       = esOutDel;
    p_out->pf_destroy   = esOutDestroy;
    p_out->pf_send      = esOutSend;
650

Rafaël Carré's avatar
Rafaël Carré committed
651 652 653
    p_out->p_sys = malloc(sizeof(*p_out->p_sys));
    if (unlikely(p_out->p_sys == NULL)) {
        free(p_out);
654 655 656 657 658 659
        return NULL;
    }
    p_out->p_sys->p_demux = p_demux;
    return p_out;
}

660 661 662
/*****************************************************************************
 * subpicture_updater_t functions:
 *****************************************************************************/
Rafaël Carré's avatar
Rafaël Carré committed
663
static int subpictureUpdaterValidate(subpicture_t *p_subpic,
664 665
                                      bool b_fmt_src, const video_format_t *p_fmt_src,
                                      bool b_fmt_dst, const video_format_t *p_fmt_dst,
Rafaël Carré's avatar
Rafaël Carré committed
666
                                      mtime_t i_ts)
667
{
Rafaël Carré's avatar
Rafaël Carré committed
668 669 670 671 672
    VLC_UNUSED(b_fmt_src);
    VLC_UNUSED(b_fmt_dst);
    VLC_UNUSED(p_fmt_src);
    VLC_UNUSED(p_fmt_dst);
    VLC_UNUSED(i_ts);
673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730

    subpicture_updater_sys_t *p_upd_sys = p_subpic->updater.p_sys;
    bluray_overlay_t         *p_overlay = p_upd_sys->p_overlay;

    vlc_mutex_lock(&p_overlay->lock);
    int res = p_overlay->status == Outdated;
    vlc_mutex_unlock(&p_overlay->lock);
    return res;
}

/* This should probably be moved to subpictures.c afterward */
static subpicture_region_t* subpicture_region_Clone(subpicture_region_t *p_region_src)
{
    if (!p_region_src)
        return NULL;
    subpicture_region_t *p_region_dst = subpicture_region_New(&p_region_src->fmt);
    if (unlikely(!p_region_dst))
        return NULL;

    p_region_dst->i_x      = p_region_src->i_x;
    p_region_dst->i_y      = p_region_src->i_y;
    p_region_dst->i_align  = p_region_src->i_align;
    p_region_dst->i_alpha  = p_region_src->i_alpha;

    p_region_dst->psz_text = p_region_src->psz_text ? strdup(p_region_src->psz_text) : NULL;
    p_region_dst->psz_html = p_region_src->psz_html ? strdup(p_region_src->psz_html) : NULL;
    if (p_region_src->p_style != NULL) {
        p_region_dst->p_style = malloc(sizeof(*p_region_dst->p_style));
        p_region_dst->p_style = text_style_Copy(p_region_dst->p_style,
                                                p_region_src->p_style);
    }

    //Palette is already copied by subpicture_region_New, we just have to duplicate p_pixels
    for (int i = 0; i < p_region_src->p_picture->i_planes; i++)
        memcpy(p_region_dst->p_picture->p[i].p_pixels,
               p_region_src->p_picture->p[i].p_pixels,
               p_region_src->p_picture->p[i].i_lines * p_region_src->p_picture->p[i].i_pitch);
    return p_region_dst;
}

static void subpictureUpdaterUpdate(subpicture_t *p_subpic,
                                    const video_format_t *p_fmt_src,
                                    const video_format_t *p_fmt_dst,
                                    mtime_t i_ts)
{
    VLC_UNUSED(p_fmt_src);
    VLC_UNUSED(p_fmt_dst);
    VLC_UNUSED(i_ts);
    subpicture_updater_sys_t *p_upd_sys = p_subpic->updater.p_sys;
    bluray_overlay_t         *p_overlay = p_upd_sys->p_overlay;

    /*
     * When this function is called, all p_subpic regions are gone.
     * We need to duplicate our regions (stored internaly) to this subpic.
     */
    vlc_mutex_lock(&p_overlay->lock);

    subpicture_region_t *p_src = p_overlay->p_regions;
Rafaël Carré's avatar
Rafaël Carré committed
731
    if (!p_src) {
732
        vlc_mutex_unlock(&p_overlay->lock);
733
        return;
734
    }
735

Rafaël Carré's avatar
Rafaël Carré committed
736
    subpicture_region_t **p_dst = &p_subpic->p_region;
737 738 739
    while (p_src != NULL) {
        *p_dst = subpicture_region_Clone(p_src);
        if (*p_dst == NULL)
Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
740
            break;
Rafaël Carré's avatar
Rafaël Carré committed
741
        p_dst = &(*p_dst)->p_next;
742 743 744 745 746 747 748 749
        p_src = p_src->p_next;
    }
    if (*p_dst != NULL)
        (*p_dst)->p_next = NULL;
    p_overlay->status = Displayed;
    vlc_mutex_unlock(&p_overlay->lock);
}

750 751
static void blurayCleanOverlayStruct(bluray_overlay_t *);

752 753
static void subpictureUpdaterDestroy(subpicture_t *p_subpic)
{
754
    blurayCleanOverlayStruct(p_subpic->updater.p_sys->p_overlay);
Rafaël Carré's avatar
Rafaël Carré committed
755
    free(p_subpic->updater.p_sys);
756 757
}

758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780
/*****************************************************************************
 * User input events:
 *****************************************************************************/
static int onMouseEvent(vlc_object_t *p_vout, const char *psz_var, vlc_value_t old,
                        vlc_value_t val, void *p_data)
{
    demux_t     *p_demux = (demux_t*)p_data;
    demux_sys_t *p_sys   = p_demux->p_sys;
    mtime_t     now      = mdate();
    VLC_UNUSED(old);
    VLC_UNUSED(p_vout);

    if (psz_var[6] == 'm')   //Mouse moved
        bd_mouse_select(p_sys->bluray, now, val.coords.x, val.coords.y);
    else if (psz_var[6] == 'c') {
        bd_mouse_select(p_sys->bluray, now, val.coords.x, val.coords.y);
        bd_user_input(p_sys->bluray, now, BD_VK_MOUSE_ACTIVATE);
    } else {
        assert(0);
    }
    return VLC_SUCCESS;
}

781 782 783
static int sendKeyEvent(demux_sys_t *p_sys, unsigned int key)
{
    mtime_t now = mdate();
Rafaël Carré's avatar
Rafaël Carré committed
784
    if (bd_user_input(p_sys->bluray, now, key) < 0)
785
        return VLC_EGENERIC;
Rafaël Carré's avatar
Rafaël Carré committed
786

787 788 789
    return VLC_SUCCESS;
}

790 791 792
/*****************************************************************************
 * libbluray overlay handling:
 *****************************************************************************/
793
static void blurayCleanOverlayStruct(bluray_overlay_t *p_overlay)
794
{
795 796
    if (!atomic_flag_test_and_set(&p_overlay->released_once))
        return;
797 798 799 800 801
    /*
     * This will be called when destroying the picture.
     * Don't delete it again from here!
     */
    vlc_mutex_destroy(&p_overlay->lock);
Rafaël Carré's avatar
Rafaël Carré committed
802
    subpicture_region_ChainDelete(p_overlay->p_regions);
803 804 805
    free(p_overlay);
}

806 807 808 809 810 811
static void blurayCloseOverlay(demux_t *p_demux, int plane)
{
    demux_sys_t *p_sys = p_demux->p_sys;
    bluray_overlay_t *ov = p_sys->p_overlays[plane];

    if (ov != NULL) {
812 813
        if (p_sys->p_vout)
            vout_FlushSubpictureChannel(p_sys->p_vout, ov->p_pic->i_channel);
814 815 816
        blurayCleanOverlayStruct(ov);
        if (p_sys->current_overlay == plane)
            p_sys->current_overlay = -1;
817

818 819
        p_sys->p_overlays[plane] = NULL;
    }
820

821
    for (int i = 0; i < MAX_OVERLAY; i++)
822 823
        if (p_sys->p_overlays[i])
            return;
824

825
    /* All overlays have been closed */
826 827 828 829 830 831
    if (p_sys->p_vout != NULL) {
        var_DelCallback(p_sys->p_vout, "mouse-moved", onMouseEvent, p_demux);
        var_DelCallback(p_sys->p_vout, "mouse-clicked", onMouseEvent, p_demux);
        vlc_object_release(p_sys->p_vout);
        p_sys->p_vout = NULL;
    }
832 833 834 835 836 837 838 839 840
}

/*
 * Mark the overlay as "ToDisplay" status.
 * This will not send the overlay to the vout instantly, as the vout
 * may not be acquired (not acquirable) yet.
 * If is has already been acquired, the overlay has already been sent to it,
 * therefore, we only flag the overlay as "Outdated"
 */
841
static void blurayActivateOverlay(demux_t *p_demux, int plane)
842 843
{
    demux_sys_t *p_sys = p_demux->p_sys;
844
    bluray_overlay_t *ov = p_sys->p_overlays[plane];
845 846 847 848 849

    /*
     * If the overlay is already displayed, mark the picture as outdated.
     * We must NOT use vout_PutSubpicture if a picture is already displayed.
     */
850 851 852 853
    vlc_mutex_lock(&ov->lock);
    if (ov->status >= Displayed && p_sys->p_vout) {
        ov->status = Outdated;
        vlc_mutex_unlock(&ov->lock);
Jean-Baptiste Kempf's avatar
Jean-Baptiste Kempf committed
854
        return;
855
    }
856

857 858 859 860 861
    /*
     * Mark the overlay as available, but don't display it right now.
     * the blurayDemuxMenu will send it to vout, as it may be unavailable when
     * the overlay is computed
     */
862
    p_sys->current_overlay = plane;
863 864
    ov->status = ToDisplay;
    vlc_mutex_unlock(&ov->lock);
865 866
}

867
static void blurayInitOverlay(demux_t *p_demux, int plane, int width, int height)
868 869 870
{
    demux_sys_t *p_sys = p_demux->p_sys;

871
    assert(p_sys->p_overlays[plane] == NULL);
872

873 874
    bluray_overlay_t *ov = calloc(1, sizeof(*ov));
    if (unlikely(ov == NULL))
875 876 877
        return;

    subpicture_updater_sys_t *p_upd_sys = malloc(sizeof(*p_upd_sys));
878
    if (unlikely(p_upd_sys == NULL)) {
879
        free(ov);
880 881
        return;
    }
882
    /* two references: vout + demux */
883
    ov->released_once = ATOMIC_FLAG_INIT;
884

885
    p_upd_sys->p_overlay = ov;
886 887 888 889 890 891
    subpicture_updater_t updater = {
        .pf_validate = subpictureUpdaterValidate,
        .pf_update   = subpictureUpdaterUpdate,
        .pf_destroy  = subpictureUpdaterDestroy,
        .p_sys       = p_upd_sys,
    };
892

893
    ov->p_pic = subpicture_New(&updater);
894 895 896 897 898 899
    if (ov->p_pic == NULL) {
        free(p_upd_sys);
        free(ov);
        return;
    }

900 901 902 903
    ov->p_pic->i_original_picture_width = width;
    ov->p_pic->i_original_picture_height = height;
    ov->p_pic->b_ephemer = true;
    ov->p_pic->b_absolute = true;
904 905 906 907

    vlc_mutex_init(&ov->lock);

    p_sys->p_overlays[plane] = ov;
908 909 910 911 912 913 914 915 916 917
}

/**
 * Destroy every regions in the subpicture.
 * This is done in two steps:
 * - Wiping our private regions list
 * - Flagging the overlay as outdated, so the changes are replicated from
 *   the subpicture_updater_t::pf_update
 * This doesn't destroy the subpicture, as the overlay may be used again by libbluray.
 */
918
static void blurayClearOverlay(demux_t *p_demux, int plane)
919 920
{
    demux_sys_t *p_sys = p_demux->p_sys;
921 922 923
    bluray_overlay_t *ov = p_sys->p_overlays[plane];

    vlc_mutex_lock(&ov->lock);
924

925 926 927
    subpicture_region_ChainDelete(ov->p_regions);
    ov->p_regions = NULL;
    ov->status = Outdated;
928

929
    vlc_mutex_unlock(&ov->lock);
930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963
}

/*
 * This will draw to the overlay by adding a region to our region list
 * This will have to be copied to the subpicture used to render the overlay.
 */
static void blurayDrawOverlay(demux_t *p_demux, const BD_OVERLAY* const ov)
{
    demux_sys_t *p_sys = p_demux->p_sys;

    /*
     * Compute a subpicture_region_t.
     * It will be copied and sent to the vout later.
     */
    if (!ov->img)
        return;

    vlc_mutex_lock(&p_sys->p_overlays[ov->plane]->lock);

    /* Find a region to update */
    subpicture_region_t *p_reg = p_sys->p_overlays[ov->plane]->p_regions;
    subpicture_region_t *p_last = NULL;
    while (p_reg != NULL) {
        p_last = p_reg;
        if (p_reg->i_x == ov->x && p_reg->i_y == ov->y &&
                p_reg->fmt.i_width == ov->w && p_reg->fmt.i_height == ov->h)
            break;
        p_reg = p_reg->p_next;
    }

    /* If there is no region to update, create a new one. */
    if (!p_reg) {
        video_format_t fmt;
        video_format_Init(&fmt, 0);
964
        video_format_Setup(&fmt, VLC_CODEC_YUVP, ov->w, ov->h, ov->w, ov->h, 1, 1);
965 966 967 968 969 970 971 972 973 974 975 976 977

        p_reg = subpicture_region_New(&fmt);
        p_reg->i_x = ov->x;
        p_reg->i_y = ov->y;
        /* Append it to our list. */
        if (p_last != NULL)
            p_last->p_next = p_reg;
        else /* If we don't have a last region, then our list empty */
            p_sys->p_overlays[ov->plane]->p_regions = p_reg;
    }

    /* Now we can update the region, regardless it's an update or an insert */
    const BD_PG_RLE_ELEM *img = ov->img;
Rafaël Carré's avatar
Rafaël Carré committed
978
    for (int y = 0; y < ov->h; y++)
979
        for (int x = 0; x < ov->w;) {
980 981
            plane_t *p = &p_reg->p_picture->p[0];
            memset(&p->p_pixels[y * p->i_pitch + x], img->color, img->len);