]> git.sesse.net Git - vlc/blob - modules/access/bluray.c
bluray: improve readability by using local variables
[vlc] / modules / access / bluray.c
1 /*****************************************************************************
2  * bluray.c: Blu-ray disc support plugin
3  *****************************************************************************
4  * Copyright © 2010-2012 VideoLAN, VLC authors and libbluray AUTHORS
5  *
6  * Authors: Jean-Baptiste Kempf <jb@videolan.org>
7  *          Hugo Beauzée-Luyssen <hugo@videolan.org>
8  *
9  * This program is free software; you can redistribute it and/or modify it
10  * under the terms of the GNU Lesser General Public License as published by
11  * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public License
20  * along with this program; if not, write to the Free Software Foundation,
21  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22  *****************************************************************************/
23
24 #ifdef HAVE_CONFIG_H
25 # include "config.h"
26 #endif
27
28 #include <assert.h>
29 #include <limits.h>                         /* PATH_MAX */
30
31 #if defined (HAVE_MNTENT_H) && defined(HAVE_SYS_STAT_H)
32 # include <mntent.h>
33 # include <sys/stat.h>
34 #endif
35
36 #ifdef __APPLE__
37 # include <sys/stat.h>
38 # include <sys/param.h>
39 # include <sys/ucred.h>
40 # include <sys/mount.h>
41 #endif
42
43 #include <vlc_common.h>
44 #include <vlc_plugin.h>
45 #include <vlc_demux.h>                      /* demux_t */
46 #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 #include <vlc_iso_lang.h>
52
53 /* FIXME we should find a better way than including that */
54 #include "../../src/text/iso-639_def.h"
55
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  * Module descriptor
65  *****************************************************************************/
66
67 #define BD_MENU_TEXT        N_("Blu-ray menus")
68 #define BD_MENU_LONGTEXT    N_("Use Blu-ray menus. If disabled, "\
69                                 "the movie will start directly")
70 #define BD_REGION_TEXT      N_("Region code")
71 #define BD_REGION_LONGTEXT  N_("Blu-Ray player region code. "\
72                                 "Some discs can be played only with a correct region code.")
73
74 static const char *const ppsz_region_code[] = {
75     "A", "B", "C" };
76 static const char *const ppsz_region_code_text[] = {
77     "Region A", "Region B", "Region C" };
78
79 #define REGION_DEFAULT   1   /* Index to region list. Actual region code is (1<<REGION_DEFAULT) */
80 #define LANGUAGE_DEFAULT ("eng")
81
82 /* Callbacks */
83 static int  blurayOpen (vlc_object_t *);
84 static void blurayClose(vlc_object_t *);
85
86 vlc_module_begin ()
87     set_shortname(N_("Blu-ray"))
88     set_description(N_("Blu-ray Disc support (libbluray)"))
89
90     set_category(CAT_INPUT)
91     set_subcategory(SUBCAT_INPUT_ACCESS)
92     set_capability("access_demux", 200)
93     add_bool("bluray-menu", false, BD_MENU_TEXT, BD_MENU_LONGTEXT, false)
94     add_string("bluray-region", ppsz_region_code[REGION_DEFAULT], BD_REGION_TEXT, BD_REGION_LONGTEXT, false)
95         change_string_list(ppsz_region_code, ppsz_region_code_text)
96
97     add_shortcut("bluray", "file")
98
99     set_callbacks(blurayOpen, blurayClose)
100 vlc_module_end ()
101
102 /* libbluray's overlay.h defines 2 types of overlay (bd_overlay_plane_e). */
103 #define MAX_OVERLAY 2
104
105 typedef enum OverlayStatus {
106     Closed = 0,
107     ToDisplay,  //Used to mark the overlay to be displayed the first time.
108     Displayed,
109     Outdated    //used to update the overlay after it has been sent to the vout
110 } OverlayStatus;
111
112 typedef struct bluray_overlay_t
113 {
114     atomic_flag         released_once;
115     vlc_mutex_t         lock;
116     subpicture_t        *p_pic;
117     OverlayStatus       status;
118     subpicture_region_t *p_regions;
119 } bluray_overlay_t;
120
121 struct  demux_sys_t
122 {
123     BLURAY              *bluray;
124
125     /* Titles */
126     unsigned int        i_title;
127     unsigned int        i_longest_title;
128     input_title_t       **pp_title;
129
130     vlc_mutex_t             pl_info_lock;
131     BLURAY_TITLE_INFO      *p_pl_info;
132     const BLURAY_CLIP_INFO *p_clip_info;
133
134     /* Meta information */
135     const META_DL       *p_meta;
136
137     /* Menus */
138     bluray_overlay_t    *p_overlays[MAX_OVERLAY];
139     int                 current_overlay; // -1 if no current overlay;
140     bool                b_menu;
141     bool                b_menu_open;
142     bool                b_popup_available;
143     mtime_t             i_still_end_time;
144
145     /* */
146     input_thread_t      *p_input;
147     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     /* Used to store bluray disc path */
159     char                *psz_bd_path;
160 };
161
162 struct subpicture_updater_sys_t
163 {
164     bluray_overlay_t    *p_overlay;
165 };
166
167 /* Get a 3 char code
168  * FIXME: partiallyy duplicated from src/input/es_out.c
169  */
170 static const char *DemuxGetLanguageCode( demux_t *p_demux, const char *psz_var )
171 {
172     const iso639_lang_t *pl;
173     char *psz_lang;
174     char *p;
175
176     psz_lang = var_CreateGetString( p_demux, psz_var );
177     if( !psz_lang )
178         return LANGUAGE_DEFAULT;
179
180     /* XXX: we will use only the first value
181      * (and ignore other ones in case of a list) */
182     if( ( p = strchr( psz_lang, ',' ) ) )
183         *p = '\0';
184
185     for( pl = p_languages; pl->psz_eng_name != NULL; pl++ )
186     {
187         if( *psz_lang == '\0' )
188             continue;
189         if( !strcasecmp( pl->psz_eng_name, psz_lang ) ||
190             !strcasecmp( pl->psz_iso639_1, psz_lang ) ||
191             !strcasecmp( pl->psz_iso639_2T, psz_lang ) ||
192             !strcasecmp( pl->psz_iso639_2B, psz_lang ) )
193             break;
194     }
195
196     free( psz_lang );
197
198     if( pl->psz_eng_name != NULL )
199         return pl->psz_iso639_2T;
200
201     return LANGUAGE_DEFAULT;
202 }
203
204 /*****************************************************************************
205  * Local prototypes
206  *****************************************************************************/
207 static es_out_t *esOutNew(demux_t *p_demux);
208
209 static int   blurayControl(demux_t *, int, va_list);
210 static int   blurayDemux(demux_t *);
211
212 static void  blurayInitTitles(demux_t *p_demux, int menu_titles);
213 static int   bluraySetTitle(demux_t *p_demux, int i_title);
214
215 static void  blurayOverlayProc(void *ptr, const BD_OVERLAY * const overlay);
216 static void  blurayArgbOverlayProc(void *ptr, const BD_ARGB_OVERLAY * const overlay);
217
218 static int   onMouseEvent(vlc_object_t *p_vout, const char *psz_var,
219                           vlc_value_t old, vlc_value_t val, void *p_data);
220
221 static void  blurayResetParser(demux_t *p_demux);
222
223 #define FROM_TICKS(a) (a*CLOCK_FREQ / INT64_C(90000))
224 #define TO_TICKS(a)   (a*INT64_C(90000)/CLOCK_FREQ)
225 #define CUR_LENGTH    p_sys->pp_title[p_demux->info.i_title]->i_length
226
227 /* */
228 static void FindMountPoint(char **file)
229 {
230     char *device = *file;
231 #if defined (HAVE_MNTENT_H) && defined (HAVE_SYS_STAT_H)
232     struct stat st;
233     if (!stat (device, &st) && S_ISBLK (st.st_mode)) {
234         FILE *mtab = setmntent ("/proc/self/mounts", "r");
235         struct mntent *m, mbuf;
236         char buf [8192];
237         /* bd path may be a symlink (e.g. /dev/dvd -> /dev/sr0), so make
238          * sure we look up the real device */
239         char *bd_device = realpath(device, NULL);
240         if (!bd_device)
241             bd_device = strdup(device);
242
243         while ((m = getmntent_r (mtab, &mbuf, buf, sizeof(buf))) != NULL) {
244             if (!strcmp (m->mnt_fsname, bd_device)) {
245                 free(device);
246                 *file = strdup(m->mnt_dir);
247                 break;
248             }
249         }
250         free(bd_device);
251         endmntent (mtab);
252     }
253 #elif defined(__APPLE__)
254     struct stat st;
255     if (!stat (device, &st) && S_ISBLK (st.st_mode)) {
256         int fs_count = getfsstat (NULL, 0, MNT_NOWAIT);
257         if (fs_count > 0) {
258             struct statfs mbuf[128];
259             getfsstat (mbuf, fs_count * sizeof(mbuf[0]), MNT_NOWAIT);
260             for (int i = 0; i < fs_count; ++i)
261                 if (!strcmp (mbuf[i].f_mntfromname, device)) {
262                     free(device);
263                     *file = strdup(mbuf[i].f_mntonname);
264                     return;
265                 }
266         }
267     }
268 #else
269 # warning Disc device to mount point not implemented
270 #endif
271 }
272
273 /*****************************************************************************
274  * cache current playlist (title) information
275  *****************************************************************************/
276
277 static void setTitleInfo(demux_sys_t *p_sys, BLURAY_TITLE_INFO *info)
278 {
279     vlc_mutex_lock(&p_sys->pl_info_lock);
280
281     if (p_sys->p_pl_info) {
282         bd_free_title_info(p_sys->p_pl_info);
283     }
284     p_sys->p_pl_info   = info;
285     p_sys->p_clip_info = NULL;
286
287     if (p_sys->p_pl_info && p_sys->p_pl_info->clip_count) {
288         p_sys->p_clip_info = &p_sys->p_pl_info->clips[0];
289     }
290
291     vlc_mutex_unlock(&p_sys->pl_info_lock);
292 }
293
294 /*****************************************************************************
295  * blurayOpen: module init function
296  *****************************************************************************/
297 static int blurayOpen(vlc_object_t *object)
298 {
299     demux_t *p_demux = (demux_t*)object;
300     demux_sys_t *p_sys;
301
302     const char *error_msg = NULL;
303 #define BLURAY_ERROR(s) do { error_msg = s; goto error; } while(0)
304
305     if (strcmp(p_demux->psz_access, "bluray")) {
306         // TODO BDMV support, once we figure out what to do in libbluray
307         return VLC_EGENERIC;
308     }
309
310     /* */
311     p_demux->p_sys = p_sys = calloc(1, sizeof(*p_sys));
312     if (unlikely(!p_sys))
313         return VLC_ENOMEM;
314
315     p_sys->current_overlay = -1;
316     p_sys->i_audio_stream = -1;
317     p_sys->i_spu_stream = -1;
318     p_sys->i_video_stream = -1;
319     p_sys->i_still_end_time = 0;
320
321     /* init demux info fields */
322     p_demux->info.i_update    = 0;
323     p_demux->info.i_title     = 0;
324     p_demux->info.i_seekpoint = 0;
325
326     TAB_INIT(p_sys->i_title, p_sys->pp_title);
327
328     /* store current bd path */
329     if (p_demux->psz_file)
330         p_sys->psz_bd_path = strdup(p_demux->psz_file);
331
332     /* If we're passed a block device, try to convert it to the mount point. */
333     FindMountPoint(&p_sys->psz_bd_path);
334
335     p_sys->bluray = bd_open(p_sys->psz_bd_path, NULL);
336     if (!p_sys->bluray) {
337         free(p_sys->psz_bd_path);
338         free(p_sys);
339         return VLC_EGENERIC;
340     }
341
342     /* Warning the user about AACS/BD+ */
343     const BLURAY_DISC_INFO *disc_info = bd_get_disc_info(p_sys->bluray);
344
345     /* Is it a bluray? */
346     if (!disc_info->bluray_detected)
347         BLURAY_ERROR(_("Path doesn't appear to be a Blu-ray"));
348
349     msg_Info(p_demux, "First play: %i, Top menu: %i\n"
350                       "HDMV Titles: %i, BD-J Titles: %i, Other: %i",
351              disc_info->first_play_supported, disc_info->top_menu_supported,
352              disc_info->num_hdmv_titles, disc_info->num_bdj_titles,
353              disc_info->num_unsupported_titles);
354
355     /* AACS */
356     if (disc_info->aacs_detected) {
357         msg_Dbg(p_demux, "Disc is using AACS");
358         if (!disc_info->libaacs_detected)
359             BLURAY_ERROR(_("This Blu-ray Disc needs a library for AACS decoding"
360                       ", and your system does not have it."));
361         if (!disc_info->aacs_handled) {
362             if (disc_info->aacs_error_code) {
363                 switch (disc_info->aacs_error_code) {
364                 case BD_AACS_CORRUPTED_DISC:
365                     BLURAY_ERROR(_("Blu-ray Disc is corrupted."));
366                 case BD_AACS_NO_CONFIG:
367                     BLURAY_ERROR(_("Missing AACS configuration file!"));
368                 case BD_AACS_NO_PK:
369                     BLURAY_ERROR(_("No valid processing key found in AACS config file."));
370                 case BD_AACS_NO_CERT:
371                     BLURAY_ERROR(_("No valid host certificate found in AACS config file."));
372                 case BD_AACS_CERT_REVOKED:
373                     BLURAY_ERROR(_("AACS Host certificate revoked."));
374                 case BD_AACS_MMC_FAILED:
375                     BLURAY_ERROR(_("AACS MMC failed."));
376                 }
377             }
378         }
379     }
380
381     /* BD+ */
382     if (disc_info->bdplus_detected) {
383         msg_Dbg(p_demux, "Disc is using BD+");
384         if (!disc_info->libbdplus_detected)
385             BLURAY_ERROR(_("This Blu-ray Disc needs a library for BD+ decoding"
386                       ", and your system does not have it."));
387         if (!disc_info->bdplus_handled)
388             BLURAY_ERROR(_("Your system BD+ decoding library does not work. "
389                       "Missing configuration?"));
390     }
391
392     /* set player region code */
393     char *psz_region = var_InheritString(p_demux, "bluray-region");
394     unsigned int region = psz_region ? (psz_region[0] - 'A') : REGION_DEFAULT;
395     free(psz_region);
396     bd_set_player_setting(p_sys->bluray, BLURAY_PLAYER_SETTING_REGION_CODE, 1<<region);
397
398     /* set preferred languages */
399     const char *psz_code = DemuxGetLanguageCode( p_demux, "audio-language" );
400     bd_set_player_setting_str(p_sys->bluray, BLURAY_PLAYER_SETTING_AUDIO_LANG, psz_code);
401     psz_code = DemuxGetLanguageCode( p_demux, "sub-language" );
402     bd_set_player_setting_str(p_sys->bluray, BLURAY_PLAYER_SETTING_PG_LANG,    psz_code);
403     psz_code = DemuxGetLanguageCode( p_demux, "menu-language" );
404     bd_set_player_setting_str(p_sys->bluray, BLURAY_PLAYER_SETTING_MENU_LANG,  psz_code);
405
406     /* Get titles and chapters */
407     p_sys->p_meta = bd_get_meta(p_sys->bluray);
408     if (!p_sys->p_meta)
409         msg_Warn(p_demux, "Failed to get meta info.");
410
411     p_sys->b_menu = var_InheritBool(p_demux, "bluray-menu");
412
413     blurayInitTitles(p_demux, disc_info->num_hdmv_titles + disc_info->num_bdj_titles + 1/*Top Menu*/ + 1/*First Play*/);
414
415     /*
416      * Initialize the event queue, so we can receive events in blurayDemux(Menu).
417      */
418     bd_get_event(p_sys->bluray, NULL);
419
420     /* Registering overlay event handler */
421     bd_register_overlay_proc(p_sys->bluray, p_demux, blurayOverlayProc);
422     p_sys->p_input = demux_GetParentInput(p_demux);
423     if (unlikely(!p_sys->p_input)) {
424         msg_Err(p_demux, "Could not get parent input");
425         goto error;
426     }
427
428     if (p_sys->b_menu) {
429
430         /* Register ARGB overlay handler for BD-J */
431         if (disc_info->num_bdj_titles)
432             bd_register_argb_overlay_proc(p_sys->bluray, p_demux, blurayArgbOverlayProc, NULL);
433
434         /* libbluray will start playback from "First-Title" title */
435         if (bd_play(p_sys->bluray) == 0)
436             BLURAY_ERROR(_("Failed to start bluray playback. Please try without menu support."));
437
438     } else {
439         /* set start title number */
440         if (bluraySetTitle(p_demux, p_sys->i_longest_title) != VLC_SUCCESS) {
441             msg_Err(p_demux, "Could not set the title %d", p_sys->i_longest_title);
442             goto error;
443         }
444     }
445
446     vlc_array_init(&p_sys->es);
447     p_sys->p_out = esOutNew(p_demux);
448     if (unlikely(p_sys->p_out == NULL))
449         goto error;
450
451     blurayResetParser(p_demux);
452     if (!p_sys->p_parser) {
453         msg_Err(p_demux, "Failed to create TS demuxer");
454         goto error;
455     }
456
457     vlc_mutex_init(&p_sys->pl_info_lock);
458
459     p_demux->pf_control = blurayControl;
460     p_demux->pf_demux   = blurayDemux;
461
462     return VLC_SUCCESS;
463
464 error:
465     if (error_msg)
466         dialog_Fatal(p_demux, _("Blu-ray error"), "%s", error_msg);
467     blurayClose(object);
468     return VLC_EGENERIC;
469 #undef BLURAY_ERROR
470 }
471
472
473 /*****************************************************************************
474  * blurayClose: module destroy function
475  *****************************************************************************/
476 static void blurayClose(vlc_object_t *object)
477 {
478     demux_t *p_demux = (demux_t*)object;
479     demux_sys_t *p_sys = p_demux->p_sys;
480
481     setTitleInfo(p_sys, NULL);
482
483     /*
484      * Close libbluray first.
485      * This will close all the overlays before we release p_vout
486      * bd_close(NULL) can crash
487      */
488     assert(p_sys->bluray);
489     bd_close(p_sys->bluray);
490
491     if (p_sys->p_vout != NULL) {
492         var_DelCallback(p_sys->p_vout, "mouse-moved", onMouseEvent, p_demux);
493         var_DelCallback(p_sys->p_vout, "mouse-clicked", onMouseEvent, p_demux);
494         vlc_object_release(p_sys->p_vout);
495     }
496     if (p_sys->p_input != NULL)
497         vlc_object_release(p_sys->p_input);
498     if (p_sys->p_parser)
499         stream_Delete(p_sys->p_parser);
500     if (p_sys->p_out != NULL)
501         es_out_Delete(p_sys->p_out);
502     assert(vlc_array_count(&p_sys->es) == 0);
503     vlc_array_clear(&p_sys->es);
504
505     /* Titles */
506     for (unsigned int i = 0; i < p_sys->i_title; i++)
507         vlc_input_title_Delete(p_sys->pp_title[i]);
508     TAB_CLEAN(p_sys->i_title, p_sys->pp_title);
509
510     vlc_mutex_destroy(&p_sys->pl_info_lock);
511
512     free(p_sys->psz_bd_path);
513     free(p_sys);
514 }
515
516 /*****************************************************************************
517  * Elementary streams handling
518  *****************************************************************************/
519
520 struct es_out_sys_t {
521     demux_t *p_demux;
522 };
523
524 typedef struct  fmt_es_pair {
525     int         i_id;
526     es_out_id_t *p_es;
527 }               fmt_es_pair_t;
528
529 static int  findEsPairIndex(demux_sys_t *p_sys, int i_id)
530 {
531     for (int i = 0; i < vlc_array_count(&p_sys->es); ++i)
532         if (((fmt_es_pair_t*)vlc_array_item_at_index(&p_sys->es, i))->i_id == i_id)
533             return i;
534
535     return -1;
536 }
537
538 static int  findEsPairIndexByEs(demux_sys_t *p_sys, es_out_id_t *p_es)
539 {
540     for (int i = 0; i < vlc_array_count(&p_sys->es); ++i)
541         if (((fmt_es_pair_t*)vlc_array_item_at_index(&p_sys->es, i))->p_es == p_es)
542             return i;
543
544     return -1;
545 }
546
547 static void setStreamLang(es_format_t *p_fmt,
548                           const BLURAY_STREAM_INFO *p_streams, int i_stream_count)
549 {
550     for (int i = 0; i < i_stream_count; i++) {
551         if (p_fmt->i_id == p_streams[i].pid) {
552             free(p_fmt->psz_language);
553             p_fmt->psz_language = strndup((const char *)p_streams[i].lang, 3);
554             return;
555         }
556     }
557 }
558
559 static es_out_id_t *esOutAdd(es_out_t *p_out, const es_format_t *p_fmt)
560 {
561     demux_sys_t *p_sys = p_out->p_sys->p_demux->p_sys;
562     es_format_t fmt;
563
564     es_format_Copy(&fmt, p_fmt);
565
566     vlc_mutex_lock(&p_sys->pl_info_lock);
567
568     switch (fmt.i_cat) {
569     case VIDEO_ES:
570         if (p_sys->i_video_stream != -1 && p_sys->i_video_stream != p_fmt->i_id)
571             fmt.i_priority = ES_PRIORITY_NOT_SELECTABLE;
572         break ;
573     case AUDIO_ES:
574         if (p_sys->i_audio_stream != -1 && p_sys->i_audio_stream != p_fmt->i_id)
575             fmt.i_priority = ES_PRIORITY_NOT_SELECTABLE;
576         if (p_sys->p_clip_info)
577             setStreamLang(&fmt, p_sys->p_clip_info->audio_streams, p_sys->p_clip_info->audio_stream_count);
578         break ;
579     case SPU_ES:
580         if (p_sys->i_spu_stream != -1 && p_sys->i_spu_stream != p_fmt->i_id)
581             fmt.i_priority = ES_PRIORITY_NOT_SELECTABLE;
582         if (p_sys->p_clip_info)
583             setStreamLang(&fmt, p_sys->p_clip_info->pg_streams, p_sys->p_clip_info->pg_stream_count);
584         break ;
585     }
586
587     vlc_mutex_unlock(&p_sys->pl_info_lock);
588
589     es_out_id_t *p_es = es_out_Add(p_out->p_sys->p_demux->out, &fmt);
590     if (p_fmt->i_id >= 0) {
591         /* Ensure we are not overriding anything */
592         int idx = findEsPairIndex(p_sys, p_fmt->i_id);
593         if (idx == -1) {
594             fmt_es_pair_t *p_pair = malloc(sizeof(*p_pair));
595             if (likely(p_pair != NULL)) {
596                 p_pair->i_id = p_fmt->i_id;
597                 p_pair->p_es = p_es;
598                 msg_Info(p_out->p_sys->p_demux, "Adding ES %d", p_fmt->i_id);
599                 vlc_array_append(&p_sys->es, p_pair);
600             }
601         }
602     }
603     es_format_Clean(&fmt);
604     return p_es;
605 }
606
607 static int esOutSend(es_out_t *p_out, es_out_id_t *p_es, block_t *p_block)
608 {
609     return es_out_Send(p_out->p_sys->p_demux->out, p_es, p_block);
610 }
611
612 static void esOutDel(es_out_t *p_out, es_out_id_t *p_es)
613 {
614     int idx = findEsPairIndexByEs(p_out->p_sys->p_demux->p_sys, p_es);
615     if (idx >= 0) {
616         free(vlc_array_item_at_index(&p_out->p_sys->p_demux->p_sys->es, idx));
617         vlc_array_remove(&p_out->p_sys->p_demux->p_sys->es, idx);
618     }
619     es_out_Del(p_out->p_sys->p_demux->out, p_es);
620 }
621
622 static int esOutControl(es_out_t *p_out, int i_query, va_list args)
623 {
624     return es_out_vaControl(p_out->p_sys->p_demux->out, i_query, args);
625 }
626
627 static void esOutDestroy(es_out_t *p_out)
628 {
629     for (int i = 0; i < vlc_array_count(&p_out->p_sys->p_demux->p_sys->es); ++i)
630         free(vlc_array_item_at_index(&p_out->p_sys->p_demux->p_sys->es, i));
631     vlc_array_clear(&p_out->p_sys->p_demux->p_sys->es);
632     free(p_out->p_sys);
633     free(p_out);
634 }
635
636 static es_out_t *esOutNew(demux_t *p_demux)
637 {
638     assert(vlc_array_count(&p_demux->p_sys->es) == 0);
639     es_out_t    *p_out = malloc(sizeof(*p_out));
640     if (unlikely(p_out == NULL))
641         return NULL;
642
643     p_out->pf_add       = esOutAdd;
644     p_out->pf_control   = esOutControl;
645     p_out->pf_del       = esOutDel;
646     p_out->pf_destroy   = esOutDestroy;
647     p_out->pf_send      = esOutSend;
648
649     p_out->p_sys = malloc(sizeof(*p_out->p_sys));
650     if (unlikely(p_out->p_sys == NULL)) {
651         free(p_out);
652         return NULL;
653     }
654     p_out->p_sys->p_demux = p_demux;
655     return p_out;
656 }
657
658 /*****************************************************************************
659  * subpicture_updater_t functions:
660  *****************************************************************************/
661 static int subpictureUpdaterValidate(subpicture_t *p_subpic,
662                                       bool b_fmt_src, const video_format_t *p_fmt_src,
663                                       bool b_fmt_dst, const video_format_t *p_fmt_dst,
664                                       mtime_t i_ts)
665 {
666     VLC_UNUSED(b_fmt_src);
667     VLC_UNUSED(b_fmt_dst);
668     VLC_UNUSED(p_fmt_src);
669     VLC_UNUSED(p_fmt_dst);
670     VLC_UNUSED(i_ts);
671
672     subpicture_updater_sys_t *p_upd_sys = p_subpic->updater.p_sys;
673     bluray_overlay_t         *p_overlay = p_upd_sys->p_overlay;
674
675     vlc_mutex_lock(&p_overlay->lock);
676     int res = p_overlay->status == Outdated;
677     vlc_mutex_unlock(&p_overlay->lock);
678     return res;
679 }
680
681 /* This should probably be moved to subpictures.c afterward */
682 static subpicture_region_t* subpicture_region_Clone(subpicture_region_t *p_region_src)
683 {
684     if (!p_region_src)
685         return NULL;
686     subpicture_region_t *p_region_dst = subpicture_region_New(&p_region_src->fmt);
687     if (unlikely(!p_region_dst))
688         return NULL;
689
690     p_region_dst->i_x      = p_region_src->i_x;
691     p_region_dst->i_y      = p_region_src->i_y;
692     p_region_dst->i_align  = p_region_src->i_align;
693     p_region_dst->i_alpha  = p_region_src->i_alpha;
694
695     p_region_dst->psz_text = p_region_src->psz_text ? strdup(p_region_src->psz_text) : NULL;
696     p_region_dst->psz_html = p_region_src->psz_html ? strdup(p_region_src->psz_html) : NULL;
697     if (p_region_src->p_style != NULL) {
698         p_region_dst->p_style = malloc(sizeof(*p_region_dst->p_style));
699         p_region_dst->p_style = text_style_Copy(p_region_dst->p_style,
700                                                 p_region_src->p_style);
701     }
702
703     //Palette is already copied by subpicture_region_New, we just have to duplicate p_pixels
704     for (int i = 0; i < p_region_src->p_picture->i_planes; i++)
705         memcpy(p_region_dst->p_picture->p[i].p_pixels,
706                p_region_src->p_picture->p[i].p_pixels,
707                p_region_src->p_picture->p[i].i_lines * p_region_src->p_picture->p[i].i_pitch);
708     return p_region_dst;
709 }
710
711 static void subpictureUpdaterUpdate(subpicture_t *p_subpic,
712                                     const video_format_t *p_fmt_src,
713                                     const video_format_t *p_fmt_dst,
714                                     mtime_t i_ts)
715 {
716     VLC_UNUSED(p_fmt_src);
717     VLC_UNUSED(p_fmt_dst);
718     VLC_UNUSED(i_ts);
719     subpicture_updater_sys_t *p_upd_sys = p_subpic->updater.p_sys;
720     bluray_overlay_t         *p_overlay = p_upd_sys->p_overlay;
721
722     /*
723      * When this function is called, all p_subpic regions are gone.
724      * We need to duplicate our regions (stored internaly) to this subpic.
725      */
726     vlc_mutex_lock(&p_overlay->lock);
727
728     subpicture_region_t *p_src = p_overlay->p_regions;
729     if (!p_src) {
730         vlc_mutex_unlock(&p_overlay->lock);
731         return;
732     }
733
734     subpicture_region_t **p_dst = &p_subpic->p_region;
735     while (p_src != NULL) {
736         *p_dst = subpicture_region_Clone(p_src);
737         if (*p_dst == NULL)
738             break;
739         p_dst = &(*p_dst)->p_next;
740         p_src = p_src->p_next;
741     }
742     if (*p_dst != NULL)
743         (*p_dst)->p_next = NULL;
744     p_overlay->status = Displayed;
745     vlc_mutex_unlock(&p_overlay->lock);
746 }
747
748 static void blurayCleanOverlayStruct(bluray_overlay_t *);
749
750 static void subpictureUpdaterDestroy(subpicture_t *p_subpic)
751 {
752     blurayCleanOverlayStruct(p_subpic->updater.p_sys->p_overlay);
753     free(p_subpic->updater.p_sys);
754 }
755
756 /*****************************************************************************
757  * User input events:
758  *****************************************************************************/
759 static int onMouseEvent(vlc_object_t *p_vout, const char *psz_var, vlc_value_t old,
760                         vlc_value_t val, void *p_data)
761 {
762     demux_t     *p_demux = (demux_t*)p_data;
763     demux_sys_t *p_sys   = p_demux->p_sys;
764     mtime_t     now      = mdate();
765     VLC_UNUSED(old);
766     VLC_UNUSED(p_vout);
767
768     if (psz_var[6] == 'm')   //Mouse moved
769         bd_mouse_select(p_sys->bluray, now, val.coords.x, val.coords.y);
770     else if (psz_var[6] == 'c') {
771         bd_mouse_select(p_sys->bluray, now, val.coords.x, val.coords.y);
772         bd_user_input(p_sys->bluray, now, BD_VK_MOUSE_ACTIVATE);
773     } else {
774         assert(0);
775     }
776     return VLC_SUCCESS;
777 }
778
779 static int sendKeyEvent(demux_sys_t *p_sys, unsigned int key)
780 {
781     mtime_t now = mdate();
782     if (bd_user_input(p_sys->bluray, now, key) < 0)
783         return VLC_EGENERIC;
784
785     return VLC_SUCCESS;
786 }
787
788 /*****************************************************************************
789  * libbluray overlay handling:
790  *****************************************************************************/
791 static void blurayCleanOverlayStruct(bluray_overlay_t *p_overlay)
792 {
793     if (!atomic_flag_test_and_set(&p_overlay->released_once))
794         return;
795     /*
796      * This will be called when destroying the picture.
797      * Don't delete it again from here!
798      */
799     vlc_mutex_destroy(&p_overlay->lock);
800     subpicture_region_ChainDelete(p_overlay->p_regions);
801     free(p_overlay);
802 }
803
804 static void blurayCloseOverlay(demux_t *p_demux, int plane)
805 {
806     demux_sys_t *p_sys = p_demux->p_sys;
807     bluray_overlay_t *ov = p_sys->p_overlays[plane];
808
809     if (ov != NULL) {
810         if (p_sys->p_vout)
811             vout_FlushSubpictureChannel(p_sys->p_vout, ov->p_pic->i_channel);
812         blurayCleanOverlayStruct(ov);
813         if (p_sys->current_overlay == plane)
814             p_sys->current_overlay = -1;
815
816         p_sys->p_overlays[plane] = NULL;
817     }
818
819     for (int i = 0; i < MAX_OVERLAY; i++)
820         if (p_sys->p_overlays[i])
821             return;
822
823     /* All overlays have been closed */
824     if (p_sys->p_vout != NULL) {
825         var_DelCallback(p_sys->p_vout, "mouse-moved", onMouseEvent, p_demux);
826         var_DelCallback(p_sys->p_vout, "mouse-clicked", onMouseEvent, p_demux);
827         vlc_object_release(p_sys->p_vout);
828         p_sys->p_vout = NULL;
829     }
830 }
831
832 /*
833  * Mark the overlay as "ToDisplay" status.
834  * This will not send the overlay to the vout instantly, as the vout
835  * may not be acquired (not acquirable) yet.
836  * If is has already been acquired, the overlay has already been sent to it,
837  * therefore, we only flag the overlay as "Outdated"
838  */
839 static void blurayActivateOverlay(demux_t *p_demux, int plane)
840 {
841     demux_sys_t *p_sys = p_demux->p_sys;
842     bluray_overlay_t *ov = p_sys->p_overlays[plane];
843
844     /*
845      * If the overlay is already displayed, mark the picture as outdated.
846      * We must NOT use vout_PutSubpicture if a picture is already displayed.
847      */
848     vlc_mutex_lock(&ov->lock);
849     if (ov->status >= Displayed && p_sys->p_vout) {
850         ov->status = Outdated;
851         vlc_mutex_unlock(&ov->lock);
852         return;
853     }
854
855     /*
856      * Mark the overlay as available, but don't display it right now.
857      * the blurayDemuxMenu will send it to vout, as it may be unavailable when
858      * the overlay is computed
859      */
860     p_sys->current_overlay = plane;
861     ov->status = ToDisplay;
862     vlc_mutex_unlock(&ov->lock);
863 }
864
865 static void blurayInitOverlay(demux_t *p_demux, int plane, int width, int height)
866 {
867     demux_sys_t *p_sys = p_demux->p_sys;
868
869     assert(p_sys->p_overlays[plane] == NULL);
870
871     p_sys->p_overlays[plane] = calloc(1, sizeof(**p_sys->p_overlays));
872     if (unlikely(!p_sys->p_overlays[plane]))
873         return;
874
875     bluray_overlay_t *ov = p_sys->p_overlays[plane];
876
877     subpicture_updater_sys_t *p_upd_sys = malloc(sizeof(*p_upd_sys));
878     if (unlikely(!p_upd_sys)) {
879         free(ov);
880         p_sys->p_overlays[plane] = NULL;
881         return;
882     }
883     /* two references: vout + demux */
884     ov->released_once = ATOMIC_FLAG_INIT;
885
886     p_upd_sys->p_overlay = ov;
887     subpicture_updater_t updater = {
888         .pf_validate = subpictureUpdaterValidate,
889         .pf_update   = subpictureUpdaterUpdate,
890         .pf_destroy  = subpictureUpdaterDestroy,
891         .p_sys       = p_upd_sys,
892     };
893     vlc_mutex_init(&ov->lock);
894     ov->p_pic = subpicture_New(&updater);
895     ov->p_pic->i_original_picture_width = width;
896     ov->p_pic->i_original_picture_height = height;
897     ov->p_pic->b_ephemer = true;
898     ov->p_pic->b_absolute = true;
899 }
900
901 /**
902  * Destroy every regions in the subpicture.
903  * This is done in two steps:
904  * - Wiping our private regions list
905  * - Flagging the overlay as outdated, so the changes are replicated from
906  *   the subpicture_updater_t::pf_update
907  * This doesn't destroy the subpicture, as the overlay may be used again by libbluray.
908  */
909 static void blurayClearOverlay(demux_t *p_demux, int plane)
910 {
911     demux_sys_t *p_sys = p_demux->p_sys;
912     bluray_overlay_t *ov = p_sys->p_overlays[plane];
913
914     vlc_mutex_lock(&ov->lock);
915
916     subpicture_region_ChainDelete(ov->p_regions);
917     ov->p_regions = NULL;
918     ov->status = Outdated;
919
920     vlc_mutex_unlock(&ov->lock);
921 }
922
923 /*
924  * This will draw to the overlay by adding a region to our region list
925  * This will have to be copied to the subpicture used to render the overlay.
926  */
927 static void blurayDrawOverlay(demux_t *p_demux, const BD_OVERLAY* const ov)
928 {
929     demux_sys_t *p_sys = p_demux->p_sys;
930
931     /*
932      * Compute a subpicture_region_t.
933      * It will be copied and sent to the vout later.
934      */
935     if (!ov->img)
936         return;
937
938     vlc_mutex_lock(&p_sys->p_overlays[ov->plane]->lock);
939
940     /* Find a region to update */
941     subpicture_region_t *p_reg = p_sys->p_overlays[ov->plane]->p_regions;
942     subpicture_region_t *p_last = NULL;
943     while (p_reg != NULL) {
944         p_last = p_reg;
945         if (p_reg->i_x == ov->x && p_reg->i_y == ov->y &&
946                 p_reg->fmt.i_width == ov->w && p_reg->fmt.i_height == ov->h)
947             break;
948         p_reg = p_reg->p_next;
949     }
950
951     /* If there is no region to update, create a new one. */
952     if (!p_reg) {
953         video_format_t fmt;
954         video_format_Init(&fmt, 0);
955         video_format_Setup(&fmt, VLC_CODEC_YUVP, ov->w, ov->h, ov->w, ov->h, 1, 1);
956
957         p_reg = subpicture_region_New(&fmt);
958         p_reg->i_x = ov->x;
959         p_reg->i_y = ov->y;
960         /* Append it to our list. */
961         if (p_last != NULL)
962             p_last->p_next = p_reg;
963         else /* If we don't have a last region, then our list empty */
964             p_sys->p_overlays[ov->plane]->p_regions = p_reg;
965     }
966
967     /* Now we can update the region, regardless it's an update or an insert */
968     const BD_PG_RLE_ELEM *img = ov->img;
969     for (int y = 0; y < ov->h; y++)
970         for (int x = 0; x < ov->w;) {
971             plane_t *p = &p_reg->p_picture->p[0];
972             memset(&p->p_pixels[y * p->i_pitch + x], img->color, img->len);
973             x += img->len;
974             img++;
975         }
976
977     if (ov->palette) {
978         p_reg->fmt.p_palette->i_entries = 256;
979         for (int i = 0; i < 256; ++i) {
980             p_reg->fmt.p_palette->palette[i][0] = ov->palette[i].Y;
981             p_reg->fmt.p_palette->palette[i][1] = ov->palette[i].Cb;
982             p_reg->fmt.p_palette->palette[i][2] = ov->palette[i].Cr;
983             p_reg->fmt.p_palette->palette[i][3] = ov->palette[i].T;
984         }
985     }
986
987     vlc_mutex_unlock(&p_sys->p_overlays[ov->plane]->lock);
988     /*
989      * /!\ The region is now stored in our internal list, but not in the subpicture /!\
990      */
991 }
992
993 static void blurayOverlayProc(void *ptr, const BD_OVERLAY *const overlay)
994 {
995     demux_t *p_demux = (demux_t*)ptr;
996     demux_sys_t *p_sys = p_demux->p_sys;
997
998     if (!overlay) {
999         msg_Info(p_demux, "Closing overlays.");
1000         p_sys->current_overlay = -1;
1001         if (p_sys->p_vout)
1002             for (int i = 0; i < MAX_OVERLAY; i++)
1003                 blurayCloseOverlay(p_demux, i);
1004         return;
1005     }
1006
1007     switch (overlay->cmd) {
1008     case BD_OVERLAY_INIT:
1009         msg_Info(p_demux, "Initializing overlay");
1010         blurayInitOverlay(p_demux, overlay->plane, overlay->w, overlay->h);
1011         break;
1012     case BD_OVERLAY_CLOSE:
1013         blurayClearOverlay(p_demux, overlay->plane);
1014         blurayCloseOverlay(p_demux, overlay->plane);
1015         break;
1016     case BD_OVERLAY_CLEAR:
1017         blurayClearOverlay(p_demux, overlay->plane);
1018         break;
1019     case BD_OVERLAY_FLUSH:
1020         blurayActivateOverlay(p_demux, overlay->plane);
1021         break;
1022     case BD_OVERLAY_DRAW:
1023         blurayDrawOverlay(p_demux, overlay);
1024         break;
1025     default:
1026         msg_Warn(p_demux, "Unknown BD overlay command: %u", overlay->cmd);
1027         break;
1028     }
1029 }
1030
1031 /*
1032  * ARGB overlay (BD-J)
1033  */
1034 static void blurayInitArgbOverlay(demux_t *p_demux, int plane, int width, int height)
1035 {
1036     demux_sys_t *p_sys = p_demux->p_sys;
1037
1038     blurayInitOverlay(p_demux, plane, width, height);
1039
1040     if (!p_sys->p_overlays[plane]->p_regions) {
1041         video_format_t fmt;
1042         video_format_Init(&fmt, 0);
1043         video_format_Setup(&fmt, VLC_CODEC_RGBA, width, height, width, height, 1, 1);
1044
1045         p_sys->p_overlays[plane]->p_regions = subpicture_region_New(&fmt);
1046     }
1047 }
1048
1049 static void blurayDrawArgbOverlay(demux_t *p_demux, const BD_ARGB_OVERLAY* const ov)
1050 {
1051     demux_sys_t *p_sys = p_demux->p_sys;
1052
1053     vlc_mutex_lock(&p_sys->p_overlays[ov->plane]->lock);
1054
1055     /* Find a region to update */
1056     subpicture_region_t *p_reg = p_sys->p_overlays[ov->plane]->p_regions;
1057     if (!p_reg) {
1058         vlc_mutex_unlock(&p_sys->p_overlays[ov->plane]->lock);
1059         return;
1060     }
1061
1062     /* Now we can update the region */
1063     const uint32_t *src0 = ov->argb;
1064     uint8_t        *dst0 = p_reg->p_picture->p[0].p_pixels +
1065                            p_reg->p_picture->p[0].i_pitch * ov->y +
1066                            ov->x * 4;
1067
1068     for (int y = 0; y < ov->h; y++) {
1069         // XXX: add support for this format ? Should be possible with OPENGL/VDPAU/...
1070         // - or add libbluray option to select the format ?
1071         for (int x = 0; x < ov->w; x++) {
1072             dst0[x*4  ] = src0[x]>>16; /* R */
1073             dst0[x*4+1] = src0[x]>>8;  /* G */
1074             dst0[x*4+2] = src0[x];     /* B */
1075             dst0[x*4+3] = src0[x]>>24; /* A */
1076         }
1077
1078         src0 += ov->stride;
1079         dst0 += p_reg->p_picture->p[0].i_pitch;
1080     }
1081
1082     vlc_mutex_unlock(&p_sys->p_overlays[ov->plane]->lock);
1083     /*
1084      * /!\ The region is now stored in our internal list, but not in the subpicture /!\
1085      */
1086 }
1087
1088 static void blurayArgbOverlayProc(void *ptr, const BD_ARGB_OVERLAY *const overlay)
1089 {
1090     demux_t *p_demux = (demux_t*)ptr;
1091
1092     switch (overlay->cmd) {
1093     case BD_ARGB_OVERLAY_INIT:
1094         blurayInitArgbOverlay(p_demux, overlay->plane, overlay->w, overlay->h);
1095         break;
1096     case BD_ARGB_OVERLAY_CLOSE:
1097         blurayClearOverlay(p_demux, overlay->plane);
1098         blurayCloseOverlay(p_demux, overlay->plane);
1099         break;
1100     case BD_ARGB_OVERLAY_FLUSH:
1101         blurayActivateOverlay(p_demux, overlay->plane);
1102         break;
1103     case BD_ARGB_OVERLAY_DRAW:
1104         blurayDrawArgbOverlay(p_demux, overlay);
1105         break;
1106     default:
1107         msg_Warn(p_demux, "Unknown BD ARGB overlay command: %u", overlay->cmd);
1108         break;
1109     }
1110 }
1111
1112 static void bluraySendOverlayToVout(demux_t *p_demux)
1113 {
1114     demux_sys_t *p_sys = p_demux->p_sys;
1115
1116     assert(p_sys->current_overlay >= 0 &&
1117            p_sys->p_overlays[p_sys->current_overlay] != NULL &&
1118            p_sys->p_overlays[p_sys->current_overlay]->p_pic != NULL);
1119
1120     p_sys->p_overlays[p_sys->current_overlay]->p_pic->i_start =
1121         p_sys->p_overlays[p_sys->current_overlay]->p_pic->i_stop = mdate();
1122     p_sys->p_overlays[p_sys->current_overlay]->p_pic->i_channel =
1123         vout_RegisterSubpictureChannel(p_sys->p_vout);
1124     /*
1125      * After this point, the picture should not be accessed from the demux thread,
1126      * as it is held by the vout thread.
1127      * This must be done only once per subpicture, ie. only once between each
1128      * blurayInitOverlay & blurayCloseOverlay call.
1129      */
1130     vout_PutSubpicture(p_sys->p_vout, p_sys->p_overlays[p_sys->current_overlay]->p_pic);
1131     /*
1132      * Mark the picture as Outdated, as it contains no region for now.
1133      * This will make the subpicture_updater_t call pf_update
1134      */
1135     p_sys->p_overlays[p_sys->current_overlay]->status = Outdated;
1136 }
1137
1138 static void blurayUpdateTitleInfo(input_title_t *t, BLURAY_TITLE_INFO *title_info)
1139 {
1140     t->i_length = FROM_TICKS(title_info->duration);
1141
1142     if (!t->i_seekpoint) {
1143         for (unsigned int j = 0; j < title_info->chapter_count; j++) {
1144             seekpoint_t *s = vlc_seekpoint_New();
1145             if (!s) {
1146                 break;
1147             }
1148             s->i_time_offset = title_info->chapters[j].offset;
1149
1150             TAB_APPEND(t->i_seekpoint, t->seekpoint, s);
1151         }
1152     }
1153 }
1154
1155 static void blurayInitTitles(demux_t *p_demux, int menu_titles)
1156 {
1157     demux_sys_t *p_sys = p_demux->p_sys;
1158 #if BLURAY_VERSION < BLURAY_VERSION_CODE(0,5,0)
1159     int64_t duration = 0;
1160 #endif
1161
1162     /* get and set the titles */
1163     unsigned i_title = menu_titles;
1164
1165     if (!p_sys->b_menu) {
1166         i_title = bd_get_titles(p_sys->bluray, TITLES_RELEVANT, 60);
1167 #if BLURAY_VERSION >= BLURAY_VERSION_CODE(0,5,0)
1168         p_sys->i_longest_title = bd_get_main_title(p_sys->bluray);
1169 #endif
1170     }
1171
1172     for (unsigned int i = 0; i < i_title; i++) {
1173         input_title_t *t = vlc_input_title_New();
1174         if (!t)
1175             break;
1176
1177         if (!p_sys->b_menu) {
1178             BLURAY_TITLE_INFO *title_info = bd_get_title_info(p_sys->bluray, i, 0);
1179             blurayUpdateTitleInfo(t, title_info);
1180             bd_free_title_info(title_info);
1181
1182 #if BLURAY_VERSION < BLURAY_VERSION_CODE(0,5,0)
1183             if (t->i_length > duration) {
1184                 duration = t->i_length;
1185                 p_sys->i_longest_title = i;
1186             }
1187 #endif
1188         } else if (i == 0) {
1189             t->psz_name = strdup(_("Top Menu"));
1190         } else if (i == i_title - 1) {
1191             t->psz_name = strdup(_("First Play"));
1192         }
1193
1194         TAB_APPEND(p_sys->i_title, p_sys->pp_title, t);
1195     }
1196 }
1197
1198 static void blurayResetParser(demux_t *p_demux)
1199 {
1200     /*
1201      * This is a hack and will have to be removed.
1202      * The parser should be flushed, and not destroy/created each time
1203      * we are changing title.
1204      */
1205     demux_sys_t *p_sys = p_demux->p_sys;
1206     if (p_sys->p_parser)
1207         stream_Delete(p_sys->p_parser);
1208
1209     p_sys->p_parser = stream_DemuxNew(p_demux, "ts", p_sys->p_out);
1210
1211     if (!p_sys->p_parser)
1212         msg_Err(p_demux, "Failed to create TS demuxer");
1213 }
1214
1215 /*****************************************************************************
1216  * bluraySetTitle: select new BD title
1217  *****************************************************************************/
1218 static int bluraySetTitle(demux_t *p_demux, int i_title)
1219 {
1220     demux_sys_t *p_sys = p_demux->p_sys;
1221
1222     if (p_sys->b_menu) {
1223         if (i_title <= 0) {
1224             msg_Dbg(p_demux, "Playing TopMenu Title");
1225         } else if (i_title >= (int)p_sys->i_title - 1) {
1226             msg_Dbg(p_demux, "Playing FirstPlay Title");
1227             i_title = BLURAY_TITLE_FIRST_PLAY;
1228         } else {
1229             msg_Dbg(p_demux, "Playing Title %i", i_title);
1230         }
1231
1232         if (bd_play_title(p_sys->bluray, i_title) == 0) {
1233             msg_Err(p_demux, "cannot play bd title '%d'", i_title);
1234             return VLC_EGENERIC;
1235         }
1236
1237         return VLC_SUCCESS;
1238     }
1239
1240     /* Looking for the main title, ie the longest duration */
1241     if (i_title < 0)
1242         i_title = p_sys->i_longest_title;
1243     else if ((unsigned)i_title > p_sys->i_title)
1244         return VLC_EGENERIC;
1245
1246     msg_Dbg(p_demux, "Selecting Title %i", i_title);
1247
1248     if (bd_select_title(p_sys->bluray, i_title) == 0) {
1249         msg_Err(p_demux, "cannot select bd title '%d'", i_title);
1250         return VLC_EGENERIC;
1251     }
1252
1253     blurayResetParser(p_demux);
1254
1255     return VLC_SUCCESS;
1256 }
1257
1258 /*****************************************************************************
1259  * blurayControl: handle the controls
1260  *****************************************************************************/
1261 static int blurayControl(demux_t *p_demux, int query, va_list args)
1262 {
1263     demux_sys_t *p_sys = p_demux->p_sys;
1264     bool     *pb_bool;
1265     int64_t  *pi_64;
1266
1267     switch (query) {
1268     case DEMUX_CAN_SEEK:
1269     case DEMUX_CAN_PAUSE:
1270     case DEMUX_CAN_CONTROL_PACE:
1271          pb_bool = (bool*)va_arg(args, bool *);
1272          *pb_bool = true;
1273          break;
1274
1275     case DEMUX_GET_PTS_DELAY:
1276         pi_64 = (int64_t*)va_arg(args, int64_t *);
1277         *pi_64 = INT64_C(1000) * var_InheritInteger(p_demux, "disc-caching");
1278         break;
1279
1280     case DEMUX_SET_PAUSE_STATE:
1281         /* Nothing to do */
1282         break;
1283
1284     case DEMUX_SET_TITLE:
1285     {
1286         int i_title = (int)va_arg(args, int);
1287         if (bluraySetTitle(p_demux, i_title) != VLC_SUCCESS)
1288             return VLC_EGENERIC;
1289         break;
1290     }
1291     case DEMUX_SET_SEEKPOINT:
1292     {
1293         int i_chapter = (int)va_arg(args, int);
1294         bd_seek_chapter(p_sys->bluray, i_chapter);
1295         p_demux->info.i_update = INPUT_UPDATE_SEEKPOINT;
1296         break;
1297     }
1298
1299     case DEMUX_GET_TITLE_INFO:
1300     {
1301         input_title_t ***ppp_title = (input_title_t***)va_arg(args, input_title_t***);
1302         int *pi_int             = (int*)va_arg(args, int*);
1303         int *pi_title_offset    = (int*)va_arg(args, int*);
1304         int *pi_chapter_offset  = (int*)va_arg(args, int*);
1305
1306         /* */
1307         *pi_title_offset   = 0;
1308         *pi_chapter_offset = 0;
1309
1310         /* Duplicate local title infos */
1311         *pi_int = p_sys->i_title;
1312         *ppp_title = malloc(p_sys->i_title * sizeof(input_title_t *));
1313         for (unsigned int i = 0; i < p_sys->i_title; i++)
1314             (*ppp_title)[i] = vlc_input_title_Duplicate(p_sys->pp_title[i]);
1315
1316         return VLC_SUCCESS;
1317     }
1318
1319     case DEMUX_GET_LENGTH:
1320     {
1321         int64_t *pi_length = (int64_t*)va_arg(args, int64_t *);
1322         *pi_length = p_demux->info.i_title < (int)p_sys->i_title ? CUR_LENGTH : 0;
1323         return VLC_SUCCESS;
1324     }
1325     case DEMUX_SET_TIME:
1326     {
1327         int64_t i_time = (int64_t)va_arg(args, int64_t);
1328         bd_seek_time(p_sys->bluray, TO_TICKS(i_time));
1329         return VLC_SUCCESS;
1330     }
1331     case DEMUX_GET_TIME:
1332     {
1333         int64_t *pi_time = (int64_t*)va_arg(args, int64_t *);
1334         *pi_time = (int64_t)FROM_TICKS(bd_tell_time(p_sys->bluray));
1335         return VLC_SUCCESS;
1336     }
1337
1338     case DEMUX_GET_POSITION:
1339     {
1340         double *pf_position = (double*)va_arg(args, double *);
1341         *pf_position = p_demux->info.i_title < (int)p_sys->i_title && CUR_LENGTH > 0 ?
1342                       (double)FROM_TICKS(bd_tell_time(p_sys->bluray))/CUR_LENGTH : 0.0;
1343         return VLC_SUCCESS;
1344     }
1345     case DEMUX_SET_POSITION:
1346     {
1347         double f_position = (double)va_arg(args, double);
1348         bd_seek_time(p_sys->bluray, TO_TICKS(f_position*CUR_LENGTH));
1349         return VLC_SUCCESS;
1350     }
1351
1352     case DEMUX_GET_META:
1353     {
1354         vlc_meta_t *p_meta = (vlc_meta_t *) va_arg (args, vlc_meta_t*);
1355         const META_DL *meta = p_sys->p_meta;
1356         if (meta == NULL)
1357             return VLC_EGENERIC;
1358
1359         if (!EMPTY_STR(meta->di_name)) vlc_meta_SetTitle(p_meta, meta->di_name);
1360
1361         if (!EMPTY_STR(meta->language_code)) vlc_meta_AddExtra(p_meta, "Language", meta->language_code);
1362         if (!EMPTY_STR(meta->filename)) vlc_meta_AddExtra(p_meta, "Filename", meta->filename);
1363         if (!EMPTY_STR(meta->di_alternative)) vlc_meta_AddExtra(p_meta, "Alternative", meta->di_alternative);
1364
1365         // if (meta->di_set_number > 0) vlc_meta_SetTrackNum(p_meta, meta->di_set_number);
1366         // if (meta->di_num_sets > 0) vlc_meta_AddExtra(p_meta, "Discs numbers in Set", meta->di_num_sets);
1367
1368         if (meta->thumb_count > 0 && meta->thumbnails) {
1369             char *psz_thumbpath;
1370             if (asprintf(&psz_thumbpath, "%s" DIR_SEP "BDMV" DIR_SEP "META" DIR_SEP "DL" DIR_SEP "%s",
1371                           p_sys->psz_bd_path, meta->thumbnails[0].path) > 0) {
1372                 char *psz_thumburl = vlc_path2uri(psz_thumbpath, "file");
1373                 if (unlikely(psz_thumburl == NULL)) {
1374                     free(psz_thumbpath);
1375                     return VLC_ENOMEM;
1376                 }
1377
1378                 vlc_meta_SetArtURL(p_meta, psz_thumburl);
1379                 free(psz_thumburl);
1380             }
1381             free(psz_thumbpath);
1382         }
1383
1384         return VLC_SUCCESS;
1385     }
1386
1387     case DEMUX_NAV_ACTIVATE:
1388         if (p_sys->b_popup_available && !p_sys->b_menu_open) {
1389             return sendKeyEvent(p_sys, BD_VK_POPUP);
1390         }
1391         return sendKeyEvent(p_sys, BD_VK_ENTER);
1392     case DEMUX_NAV_UP:
1393         return sendKeyEvent(p_sys, BD_VK_UP);
1394     case DEMUX_NAV_DOWN:
1395         return sendKeyEvent(p_sys, BD_VK_DOWN);
1396     case DEMUX_NAV_LEFT:
1397         return sendKeyEvent(p_sys, BD_VK_LEFT);
1398     case DEMUX_NAV_RIGHT:
1399         return sendKeyEvent(p_sys, BD_VK_RIGHT);
1400
1401     case DEMUX_CAN_RECORD:
1402     case DEMUX_GET_FPS:
1403     case DEMUX_SET_GROUP:
1404     case DEMUX_HAS_UNSUPPORTED_META:
1405     case DEMUX_GET_ATTACHMENTS:
1406         return VLC_EGENERIC;
1407     default:
1408         msg_Warn(p_demux, "unimplemented query (%d) in control", query);
1409         return VLC_EGENERIC;
1410     }
1411     return VLC_SUCCESS;
1412 }
1413
1414 /*****************************************************************************
1415  * libbluray event handling
1416  *****************************************************************************/
1417
1418 static void streamFlush( demux_sys_t *p_sys )
1419 {
1420     /*
1421      * MPEG-TS demuxer does not flush last video frame if size of PES packet is unknown.
1422      * Packet is flushed only when TS packet with PUSI flag set is received.
1423      *
1424      * Fix this by emitting (video) ts packet with PUSI flag set.
1425      * Add video sequence end code to payload so that also video decoder is flushed.
1426      * Set PES packet size in the payload so that it will be sent to decoder immediately.
1427      */
1428
1429     if (p_sys->b_flushed)
1430         return;
1431
1432     block_t *p_block = block_Alloc(192);
1433     if (!p_block)
1434         return;
1435
1436     static const uint8_t seq_end_pes[] = {
1437         0x00, 0x00, 0x01, 0xe0, 0x00, 0x07, 0x80, 0x00, 0x00,  /* PES header */
1438         0x00, 0x00, 0x01, 0xb7,                                /* PES payload: sequence end */
1439     };
1440     static const uint8_t vid_pusi_ts[] = {
1441         0x00, 0x00, 0x00, 0x00,                /* TP extra header (ATC) */
1442         0x47, 0x50, 0x11, 0x30,                /* TP header */
1443         (192 - (4 + 5) - sizeof(seq_end_pes)), /* adaptation field length */
1444         0x80,                                  /* adaptation field: discontinuity indicator */
1445     };
1446
1447     memset(p_block->p_buffer, 0, 192);
1448     memcpy(p_block->p_buffer, vid_pusi_ts, sizeof(vid_pusi_ts));
1449     memcpy(p_block->p_buffer + 192 - sizeof(seq_end_pes), seq_end_pes, sizeof(seq_end_pes));
1450     p_block->i_buffer = 192;
1451
1452     /* set correct sequence end code */
1453     vlc_mutex_lock(&p_sys->pl_info_lock);
1454     if (p_sys->p_clip_info != NULL) {
1455         if (p_sys->p_clip_info->video_streams[0].coding_type > 2) {
1456             /* VC1 / H.264 sequence end */
1457             p_block->p_buffer[191] = 0x0a;
1458         }
1459     }
1460     vlc_mutex_unlock(&p_sys->pl_info_lock);
1461
1462     stream_DemuxSend(p_sys->p_parser, p_block);
1463     p_sys->b_flushed = true;
1464 }
1465
1466 static void blurayResetStillImage( demux_t *p_demux )
1467 {
1468     demux_sys_t *p_sys = p_demux->p_sys;
1469
1470     if (p_sys->i_still_end_time) {
1471         p_sys->i_still_end_time = 0;
1472
1473         blurayResetParser(p_demux);
1474         es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
1475     }
1476 }
1477
1478 static void blurayStillImage( demux_t *p_demux, unsigned i_timeout )
1479 {
1480     demux_sys_t *p_sys = p_demux->p_sys;
1481
1482     /* time period elapsed ? */
1483     if (p_sys->i_still_end_time > 0 && p_sys->i_still_end_time <= mdate()) {
1484         msg_Dbg(p_demux, "Still image end");
1485         bd_read_skip_still(p_sys->bluray);
1486
1487         blurayResetStillImage(p_demux);
1488         return;
1489     }
1490
1491     /* show last frame as still image */
1492     if (!p_sys->i_still_end_time) {
1493         if (i_timeout) {
1494             msg_Dbg(p_demux, "Still image (%d seconds)", i_timeout);
1495             p_sys->i_still_end_time = mdate() + i_timeout * CLOCK_FREQ;
1496         } else {
1497             msg_Dbg(p_demux, "Still image (infinite)");
1498             p_sys->i_still_end_time = -1;
1499         }
1500
1501         /* flush demuxer and decoder (there won't be next video packet starting with ts PUSI) */
1502         streamFlush(p_sys);
1503
1504         /* stop buffering */
1505         bool b_empty;
1506         es_out_Control( p_demux->out, ES_OUT_GET_EMPTY, &b_empty );
1507     }
1508
1509     /* avoid busy loops (read returns no data) */
1510     msleep( 40000 );
1511 }
1512
1513 static void blurayStreamSelect(demux_t *p_demux, uint32_t i_type, uint32_t i_id)
1514 {
1515     demux_sys_t *p_sys = p_demux->p_sys;
1516     int i_pid = -1;
1517
1518     vlc_mutex_lock(&p_sys->pl_info_lock);
1519
1520     if (p_sys->p_clip_info) {
1521
1522         /* The param we get is the real stream id, not an index, ie. it starts from 1 */
1523         i_id--;
1524         if (i_type == BD_EVENT_AUDIO_STREAM) {
1525             if (i_id < p_sys->p_clip_info->audio_stream_count) {
1526                 i_pid = p_sys->p_clip_info->audio_streams[i_id].pid;
1527                 p_sys->i_audio_stream = i_pid;
1528             }
1529         } else if (i_type == BD_EVENT_PG_TEXTST_STREAM) {
1530             if (i_id < p_sys->p_clip_info->pg_stream_count) {
1531                 i_pid = p_sys->p_clip_info->pg_streams[i_id].pid;
1532                 p_sys->i_spu_stream = i_pid;
1533             }
1534         }
1535     }
1536
1537     vlc_mutex_unlock(&p_sys->pl_info_lock);
1538
1539     if (i_pid > 0) {
1540         int i_idx = findEsPairIndex(p_sys, i_pid);
1541         if (i_idx >= 0) {
1542             es_out_id_t *p_es = vlc_array_item_at_index(&p_sys->es, i_idx);
1543             es_out_Control(p_demux->out, ES_OUT_SET_ES, p_es);
1544         }
1545     }
1546 }
1547
1548 static void blurayUpdatePlaylist(demux_t *p_demux, unsigned i_playlist)
1549 {
1550     demux_sys_t *p_sys = p_demux->p_sys;
1551
1552     blurayResetParser(p_demux);
1553
1554     /* read title info and init some values */
1555     if (!p_sys->b_menu)
1556         p_demux->info.i_title = bd_get_current_title(p_sys->bluray);
1557     p_demux->info.i_seekpoint = 0;
1558     p_demux->info.i_update |= INPUT_UPDATE_TITLE | INPUT_UPDATE_SEEKPOINT;
1559
1560     BLURAY_TITLE_INFO *p_title_info = bd_get_playlist_info(p_sys->bluray, i_playlist, 0);
1561     if (p_title_info) {
1562         blurayUpdateTitleInfo(p_sys->pp_title[p_demux->info.i_title], p_title_info);
1563     }
1564     setTitleInfo(p_sys, p_title_info);
1565
1566     blurayResetStillImage(p_demux);
1567 }
1568
1569 static void blurayUpdateCurrentClip(demux_t *p_demux, uint32_t clip)
1570 {
1571     demux_sys_t *p_sys = p_demux->p_sys;
1572
1573     vlc_mutex_lock(&p_sys->pl_info_lock);
1574
1575     p_sys->p_clip_info = NULL;
1576     p_sys->i_video_stream = -1;
1577
1578     if (p_sys->p_pl_info && clip < p_sys->p_pl_info->clip_count) {
1579
1580         p_sys->p_clip_info = &p_sys->p_pl_info->clips[clip];
1581
1582     /* Let's assume a single video track for now.
1583      * This may brake later, but it's enough for now.
1584      */
1585         assert(p_sys->p_clip_info->video_stream_count >= 1);
1586         p_sys->i_video_stream = p_sys->p_clip_info->video_streams[0].pid;
1587     }
1588
1589     vlc_mutex_unlock(&p_sys->pl_info_lock);
1590
1591     blurayResetStillImage(p_demux);
1592 }
1593
1594 static void blurayHandleEvent(demux_t *p_demux, const BD_EVENT *e)
1595 {
1596     demux_sys_t *p_sys = p_demux->p_sys;
1597
1598     switch (e->event) {
1599     case BD_EVENT_TITLE:
1600         if (e->param == BLURAY_TITLE_FIRST_PLAY)
1601             p_demux->info.i_title = p_sys->i_title - 1;
1602         else
1603             p_demux->info.i_title = e->param;
1604         /* this is feature title, we don't know yet which playlist it will play (if any) */
1605         setTitleInfo(p_sys, NULL);
1606         /* reset title infos here ? */
1607         p_demux->info.i_update |= INPUT_UPDATE_TITLE | INPUT_UPDATE_SEEKPOINT; /* might be BD-J title with no video */
1608         break;
1609     case BD_EVENT_PLAYLIST:
1610         /* Start of playlist playback (?????.mpls) */
1611         blurayUpdatePlaylist(p_demux, e->param);
1612         break;
1613     case BD_EVENT_PLAYITEM:
1614         blurayUpdateCurrentClip(p_demux, e->param);
1615         break;
1616     case BD_EVENT_CHAPTER:
1617         p_demux->info.i_update |= INPUT_UPDATE_SEEKPOINT;
1618         p_demux->info.i_seekpoint = e->param;
1619         break;
1620     case BD_EVENT_ANGLE:
1621         break;
1622     case BD_EVENT_MENU:
1623         p_sys->b_menu_open = e->param;
1624         break;
1625     case BD_EVENT_POPUP:
1626         p_sys->b_popup_available = e->param;
1627         /* TODO: show / hide pop-up menu button in gui ? */
1628         break;
1629
1630     /*
1631      * stream selection events
1632      */
1633     case BD_EVENT_AUDIO_STREAM:
1634     case BD_EVENT_PG_TEXTST_STREAM:
1635         blurayStreamSelect(p_demux, e->event, e->param);
1636         break;
1637     case BD_EVENT_IG_STREAM:
1638         break;
1639
1640     /*
1641      * playback control events
1642      */
1643     case BD_EVENT_STILL_TIME:
1644         blurayStillImage(p_demux, e->param);
1645         break;
1646     case BD_EVENT_DISCONTINUITY:
1647         /* reset demuxer (partially decoded PES packets must be dropped) */
1648         blurayResetParser(p_demux);
1649         break;
1650     case BD_EVENT_IDLE:
1651         /* nothing to do (ex. BD-J is preparing menus, waiting user input or running animation) */
1652         /* avoid busy loop (bd_read() returns no data) */
1653         msleep( 40000 );
1654         break;
1655
1656     default:
1657         msg_Warn(p_demux, "event: %d param: %d", e->event, e->param);
1658         break;
1659     }
1660 }
1661
1662 #define BD_TS_PACKET_SIZE (192)
1663 #define NB_TS_PACKETS (200)
1664
1665 static int blurayDemux(demux_t *p_demux)
1666 {
1667     demux_sys_t *p_sys = p_demux->p_sys;
1668     BD_EVENT e;
1669
1670     block_t *p_block = block_Alloc(NB_TS_PACKETS * (int64_t)BD_TS_PACKET_SIZE);
1671     if (!p_block)
1672         return -1;
1673
1674     int nread;
1675
1676     if (p_sys->b_menu == false) {
1677         while (bd_get_event(p_sys->bluray, &e))
1678             blurayHandleEvent(p_demux, &e);
1679
1680         nread = bd_read(p_sys->bluray, p_block->p_buffer,
1681                         NB_TS_PACKETS * BD_TS_PACKET_SIZE);
1682     } else {
1683         nread = bd_read_ext(p_sys->bluray, p_block->p_buffer,
1684                             NB_TS_PACKETS * BD_TS_PACKET_SIZE, &e);
1685         while (e.event != BD_EVENT_NONE) {
1686             blurayHandleEvent(p_demux, &e);
1687             bd_get_event(p_sys->bluray, &e);
1688         }
1689     }
1690
1691     if (p_sys->current_overlay != -1) {
1692         bluray_overlay_t *ov = p_sys->p_overlays[p_sys->current_overlay];
1693         vlc_mutex_lock(&ov->lock);
1694         bool display = ov->status == ToDisplay;
1695         vlc_mutex_unlock(&ov->lock);
1696         if (display) {
1697             if (p_sys->p_vout == NULL)
1698                 p_sys->p_vout = input_GetVout(p_sys->p_input);
1699             if (p_sys->p_vout != NULL) {
1700                 var_AddCallback(p_sys->p_vout, "mouse-moved", onMouseEvent, p_demux);
1701                 var_AddCallback(p_sys->p_vout, "mouse-clicked", onMouseEvent, p_demux);
1702                 bluraySendOverlayToVout(p_demux);
1703             }
1704         }
1705     }
1706
1707     if (nread <= 0) {
1708         block_Release(p_block);
1709         if (nread < 0)
1710             return -1;
1711         return 1;
1712     }
1713
1714     p_block->i_buffer = nread;
1715
1716     stream_DemuxSend(p_sys->p_parser, p_block);
1717
1718     p_sys->b_flushed = false;
1719
1720     return 1;
1721 }