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