/*
* This file is part of libbluray
* Copyright (C) 2009-2010 John Stebbins
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library. If not, see
* <http://www.gnu.org/licenses/>.
*/
#include <sys/stat.h>
#include <dirent.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <libgen.h>
#include "bdnav/mpls_data.h"
#include "bluray.h"
#include "util.h"
#include "strings.h"
#ifdef _WIN32
# define DIR_SEP "\\"
# define PLAYLIST_DIR "\\BDMV\\PLAYLIST"
#else
# define DIR_SEP "/"
# define PLAYLIST_DIR "/BDMV/PLAYLIST"
#endif
static int verbose;
const VALUE_MAP subpath_type_map[] = {
{2, "Primary audio of the Browsable slideshow"},
{3, "Interactive Graphics presentation menu"},
{4, "Text Subtitle"},
{5, "Out-of-mux Synchronous elementary streams"},
{6, "Out-of-mux Asynchronous Picture-in-Picture presentation"},
{7, "In-mux Synchronous Picture-in-Picture presentation"},
{8, "SS Video"},
{0,NULL}
};
const VALUE_MAP playback_type_map[] = {
{1, "Sequential"},
{2, "Random"},
{3, "Shuffle"},
{0, NULL}
};
const VALUE_MAP connection_type_map[] = {
{1, "Non-seamless"},
{5, "Seamless"},
{6, "Seamless"},
{0, NULL}
};
static char *
_mk_path(const char *base, const char *sub)
{
size_t n1 = strlen(base);
size_t n2 = strlen(sub);
char *result = (char*)malloc(n1 + n2 + strlen(DIR_SEP) + 1);
if (result) {
strcpy(result, base);
strcat(result, DIR_SEP);
strcat(result, sub);
}
return result;
}
static void
_show_stream(MPLS_STREAM *ss, int level)
{
indent_printf(level, "Codec (%04x): %s", ss->coding_type,
_lookup_str(codec_map, ss->coding_type));
switch (ss->stream_type) {
case 1:
indent_printf(level, "PID: %04x", ss->pid);
break;
case 2:
case 4:
indent_printf(level, "SubPath Id: %02x", ss->subpath_id);
indent_printf(level, "SubClip Id: %02x", ss->subclip_id);
indent_printf(level, "PID: %04x", ss->pid);
break;
case 3:
indent_printf(level, "SubPath Id: %02x", ss->subpath_id);
indent_printf(level, "PID: %04x", ss->pid);
break;
default:
fprintf(stderr, "unrecognized stream type %02x\n", ss->stream_type);
break;
};
switch (ss->coding_type) {
case 0x01:
case 0x02:
case 0xea:
case 0x1b:
case 0x24:
indent_printf(level, "Format %02x: %s", ss->format,
_lookup_str(video_format_map, ss->format));
indent_printf(level, "Rate %02x: %s", ss->rate,
_lookup_str(video_rate_map, ss->rate));
break;
case 0x03:
case 0x04:
case 0x80:
case 0x81:
case 0x82:
case 0x83:
case 0x84:
case 0x85:
case 0x86:
case 0xa1:
case 0xa2:
indent_printf(level, "Format %02x: %s", ss->format,
_lookup_str(audio_format_map, ss->format));
indent_printf(level, "Rate %02x: %s", ss->rate,
_lookup_str(audio_rate_map, ss->rate));
indent_printf(level, "Language: %s", ss->lang);
break;
case 0x90:
case 0x91:
indent_printf(level, "Language: %s", ss->lang);
break;
case 0x92:
indent_printf(level, "Char Code: %02x", ss->char_code);
indent_printf(level, "Language: %s", ss->lang);
break;
default:
fprintf(stderr, "unrecognized coding type %02x\n", ss->coding_type);
break;
};
}
static void
_show_details(MPLS_PL *pl, int level)
{
int ii, jj, kk;
for (ii = 0; ii < pl->list_count; ii++) {
MPLS_PI *pi;
pi = &pl->play_item[ii];
indent_printf(level, "Clip Id %s", pi->clip[0].clip_id);
indent_printf(level+1, "Stc Id: %02x", pi->clip[0].stc_id);
indent_printf(level+1, "Connection Condition: %s (%02x)",
_lookup_str(connection_type_map, pi->connection_condition),
pi->connection_condition);
indent_printf(level+1, "In-Time: %d", pi->in_time);
indent_printf(level+1, "Out-Time: %d", pi->out_time);
if (pi->still_mode == 1) {
indent_printf(level+1, "Still time: %ds\n", pi->still_time);
}
if (pi->still_mode == 2) {
indent_printf(level+1, "Still time: infinite\n");
}
if (pi->angle_count > 1) {
for (jj = 1; jj < pi->angle_count; jj++) {
indent_printf(level+1, "Angle %d:", jj);
indent_printf(level+2, "Clip Id %s", pi->clip[jj].clip_id);
indent_printf(level+2, "Stc Id: %02x", pi->clip[jj].stc_id);
}
}
for (jj = 0; jj < pi->stn.num_video; jj++) {
indent_printf(level+1, "Video Stream %d:", jj);
_show_stream(&pi->stn.video[jj], level + 2);
}
for (jj = 0; jj < pi->stn.num_audio; jj++) {
indent_printf(level+1, "Audio Stream %d:", jj);
_show_stream(&pi->stn.audio[jj], level + 2);
}
for (jj = 0; jj < pi->stn.num_ig; jj++) {
indent_printf(level+1, "Interactive Graphics Stream %d:", jj);
_show_stream(&pi->stn.ig[jj], level + 2);
}
for (jj = 0; jj < (pi->stn.num_pg + pi->stn.num_pip_pg); jj++) {
if (jj < pi->stn.num_pg) {
indent_printf(level+1, "Presentation Graphics Stream %d:", jj);
} else {
indent_printf(level+1, "PIP Presentation Graphics Stream %d:", jj);
}
_show_stream(&pi->stn.pg[jj], level + 2);
}
for (jj = 0; jj < pi->stn.num_secondary_video; jj++) {
indent_printf(level+1, "Secondary Video Stream %d:", jj);
_show_stream(&pi->stn.secondary_video[jj], level + 2);
for (kk = 0; kk < pi->stn.secondary_video[jj].sv_num_secondary_audio_ref; kk++) {
indent_printf(level+2, "Secondary Audio Ref %d: %d", kk,pi->stn.secondary_video[jj].sv_secondary_audio_ref[kk]);
}
for (kk = 0; kk < pi->stn.secondary_video[jj].sv_num_pip_pg_ref; kk++) {
indent_printf(level+2, "PIP Presentation Graphic Ref %d: %d", kk,pi->stn.secondary_video[jj].sv_pip_pg_ref[kk]);
}
}
for (jj = 0; jj < pi->stn.num_secondary_audio; jj++) {
indent_printf(level+1, "Secondary Audio Stream %d:", jj);
_show_stream(&pi->stn.secondary_audio[jj], level + 2);
for (kk = 0; kk < pi->stn.secondary_audio[jj].sa_num_primary_audio_ref; kk++) {
indent_printf(level+2, "Primary Audio Ref %d: %d", kk,pi->stn.secondary_audio[jj].sa_primary_audio_ref[kk]);
}
}
printf("\n");
}
}
static void
_show_ai(MPLS_PL *pl, int level)
{
indent_printf(level, "Playback type: %s (%d)",
_lookup_str(playback_type_map, pl->app_info.playback_type),
pl->app_info.playback_type);
if (pl->app_info.playback_type == 2 || pl->app_info.playback_type == 3) {
indent_printf(level+1, "Playback count: %d", pl->app_info.playback_count);
}
}
static void
_show_marks(MPLS_PL *pl, int level)
{
int ii;
indent_printf(level, "PlayMark Count %d", pl->mark_count);
for (ii = 0; ii < pl->mark_count; ii++) {
MPLS_PI *pi;
MPLS_PLM *plm;
int min;
double sec;
plm = &pl->play_mark[ii];
indent_printf(level, "PlayMark %d", ii);
indent_printf(level+1, "Type: %02x", plm->mark_type);
if (plm->play_item_ref < pl->list_count) {
pi = &pl->play_item[plm->play_item_ref];
indent_printf(level+1, "PlayItem: %s", pi->clip[0].clip_id);
} else {
indent_printf(level+1, "PlayItem: Invalid reference");
}
indent_printf(level+1, "Time (ticks): %u", plm->time);
min = plm->duration / (45000*60);
sec = (double)(plm->duration - min * 45000 * 60) / 45000;
indent_printf(level+1, "Duration (mm:ss.ms, ticks): %d:%.2f, %u",
min, sec, plm->duration);
printf("\n");
}
}
static void
_show_clip_list(MPLS_PL *pl, int level)
{
int ii, jj;
for (ii = 0; ii < pl->list_count; ii++) {
MPLS_PI *pi;
pi = &pl->play_item[ii];
if (verbose) {
uint32_t duration;
duration = pi->out_time - pi->in_time;
indent_printf(level, "%s.m2ts -- Duration: %3d:%02d",
pi->clip[0].clip_id,
duration / (45000 * 60), (duration / 45000) % 60);
} else {
indent_printf(level, "%s.m2ts", pi->clip[0].clip_id);
}
if (pi->angle_count > 1) {
for (jj = 1; jj < pi->angle_count; jj++) {
indent_printf(level+1, "Angle %d: %s.m2ts", jj+1, pi->clip[jj].clip_id);
}
}
}
printf("\n");
}
static void
_show_sub_path(MPLS_SUB *sub, int level)
{
int ii;
indent_printf(level+1, "Type: %d (%s)", sub->type, _lookup_str(subpath_type_map, sub->type));
indent_printf(level+1, "Repeat: %d", sub->is_repeat);
indent_printf(level+1, "Sub playitem count: %d", sub->sub_playitem_count);
for (ii = 0; ii < sub->sub_playitem_count; ii++) {
MPLS_SUB_PI *pi;
pi = &sub->sub_play_item[ii];
if (verbose) {
indent_printf(level+1, "Sub playitem %d", ii);
indent_printf(level+2, "Clip Id %s", pi->clip[0].clip_id);
indent_printf(level+2, "Multi clip: %d", pi->is_multi_clip);
indent_printf(level+2, "Clip count: %d", pi->clip_count);
indent_printf(level+2, "Connection Condition: %s (%02x)",
_lookup_str(connection_type_map, pi->connection_condition),
pi->connection_condition);
indent_printf(level+2, "In-Time: %d", pi->in_time);
indent_printf(level+2, "Out-Time: %d", pi->out_time);
indent_printf(level+2, "Sync playitem Id: %d", pi->sync_play_item_id);
indent_printf(level+2, "Sync PTS: %d", pi->sync_pts);
} else {
indent_printf(level+1, "%s.m2ts", pi->clip[0].clip_id);
}
}
}
static void
_show_pip_metadata_block(MPLS_PIP_METADATA *block, int level)
{
int ii;
indent_printf(level, "Clip ref: %d", block->clip_ref);
indent_printf(level, "Secondary video ref: %d", block->secondary_video_ref);
indent_printf(level, "Timeline type: %d", block->timeline_type);
indent_printf(level, "Luma key flag: %d", block->luma_key_flag);
if (block->luma_key_flag) {
indent_printf(level, "Upper limit luma key: %d", block->upper_limit_luma_key);
}
indent_printf(level, "Trick play flag: %d", block->trick_play_flag);
for (ii = 0; ii < block->data_count; ii++) {
indent_printf(level, "data block %d:", ii);
indent_printf(level+1, "Timestamp: %d", block->data[ii].time);
indent_printf(level+1, "Horizontal position %d", block->data[ii].xpos);
indent_printf(level+1, "Vertical position: %d", block->data[ii].ypos);
indent_printf(level+1, "Scaling factor: %d", block->data[ii].scale_factor);
}
}
static void
_show_pip_metadata(MPLS_PL *pl, int level)
{
int ii;
for (ii = 0; ii < pl->ext_pip_data_count; ii++) {
MPLS_PIP_METADATA *data;
data = &pl->ext_pip_data[ii];
indent_printf(level, "PiP metadata block %d:", ii);
_show_pip_metadata_block(data, level+1);
}
}
static void
_show_sub_paths(MPLS_PL *pl, int level)
{
int ss;
for (ss = 0; ss < pl->sub_count; ss++) {
MPLS_SUB *sub;
sub = &pl->sub_path[ss];
indent_printf(level, "Sub Path %d:", ss);
_show_sub_path(sub, level+1);
}
}
static void
_show_sub_paths_ss(MPLS_PL *pl, int level)
{
int ss;
for (ss = 0; ss < pl->ext_sub_count; ss++) {
MPLS_SUB *sub;
sub = &pl->ext_sub_path[ss];
indent_printf(level, "Extension Sub Path %d:", ss);
_show_sub_path(sub, level+1);
}
}
static uint32_t
_pl_duration(MPLS_PL *pl)
{
int ii;
uint32_t duration = 0;
MPLS_PI *pi;
for (ii = 0; ii < pl->list_count; ii++) {
pi = &pl->play_item[ii];
duration += pi->out_time - pi->in_time;
}
return duration;
}
static int
_filter_dup(MPLS_PL *pl_list[], int count, MPLS_PL *pl)
{
int ii, jj;
for (ii = 0; ii < count; ii++) {
if (pl->list_count != pl_list[ii]->list_count ||
_pl_duration(pl) != _pl_duration(pl_list[ii])) {
continue;
}
for (jj = 0; jj < pl->list_count; jj++) {
MPLS_PI *pi1, *pi2;
pi1 = &pl->play_item[jj];
pi2 = &pl_list[ii]->play_item[jj];
if (memcmp(pi1->clip[0].clip_id, pi2->clip[0].clip_id, 5) != 0 ||
pi1->in_time != pi2->in_time ||
pi1->out_time != pi2->out_time) {
break;
}
}
if (jj != pl->list_count) {
continue;
}
return 0;
}
return 1;
}
static int
_find_repeats(MPLS_PL *pl, const char *m2ts)
{
int ii, count = 0;
for (ii = 0; ii < pl->list_count; ii++) {
MPLS_PI *pi;
pi = &pl->play_item[ii];
// Ignore titles with repeated segments
if (strcmp(pi->clip[0].clip_id, m2ts) == 0) {
count++;
}
}
return count;
}
static int
_filter_short(MPLS_PL *pl, unsigned int seconds)
{
// Ignore short playlists
if (_pl_duration(pl) / 45000 <= seconds) {
return 0;
}
return 1;
}
static int
_filter_repeats(MPLS_PL *pl, int repeats)
{
int ii;
for (ii = 0; ii < pl->list_count; ii++) {
MPLS_PI *pi;
pi = &pl->play_item[ii];
// Ignore titles with repeated segments
if (_find_repeats(pl, pi->clip[0].clip_id) > repeats) {
return 0;
}
}
return 1;
}
static int clip_list = 0, playlist_info = 0, chapter_marks = 0, sub_paths = 0, pip_metadata = 0;
static int repeats = 0, seconds = 0, dups = 0;
static MPLS_PL*
_process_file(char *name, MPLS_PL *pl_list[], int pl_count)
{
MPLS_PL *pl;
pl = bd_read_mpls(name);
if (pl == NULL) {
fprintf(stderr, "Parse failed: %s\n", name);
return NULL;
}
if (seconds) {
if (!_filter_short(pl, seconds)) {
bd_free_mpls(pl);
return NULL;
}
}
if (repeats) {
if (!_filter_repeats(pl, repeats)) {
bd_free_mpls(pl);
return NULL;
}
}
if (dups) {
if (!_filter_dup(pl_list, pl_count, pl)) {
bd_free_mpls(pl);
return NULL;
}
}
if (verbose) {
indent_printf(0,
"%s -- Num Clips: %3d , Duration: minutes %4u:%02u",
basename(name),
pl->list_count,
_pl_duration(pl) / (45000 * 60),
(_pl_duration(pl) / 45000) % 60);
_show_ai(pl, 1);
} else {
indent_printf(0, "%s -- Duration: minutes %4u:%02u",
basename(name),
_pl_duration(pl) / (45000 * 60),
(_pl_duration(pl) / 45000) % 60);
}
if (playlist_info) {
_show_details(pl, 1);
}
if (chapter_marks) {
_show_marks(pl, 1);
}
if (pip_metadata) {
_show_pip_metadata(pl, 1);
}
if (clip_list) {
_show_clip_list(pl, 1);
}
if (sub_paths) {
_show_sub_paths(pl, 1);
_show_sub_paths_ss(pl, 1);
}
return pl;
}
static void
_usage(char *cmd)
{
fprintf(stderr,
"Usage: %s -vli <mpls file> [<mpls file> ...]\n"
"With no options, produces a list of the playlist(s) with durations\n"
"Options:\n"
" v - Verbose output.\n"
" l - Produces a list of the m2ts clips\n"
" i - Dumps detailed information about each clip\n"
" c - Show chapter marks\n"
" p - Show sub paths\n"
" P - Show picture-in-picture metadata\n"
" r <N> - Filter out titles that have >N repeating clips\n"
" d - Filter out duplicate titles\n"
" s <seconds> - Filter out short titles\n"
" f - Filter combination -r2 -d -s900\n"
, cmd);
exit(EXIT_FAILURE);
}
#define OPTS "vlicpPfr:ds:"
static int
_qsort_str_cmp(const void *a, const void *b)
{
const char *stra = *(char * const *)a;
const char *strb = *(char * const *)b;
return strcmp(stra, strb);
}
int
main(int argc, char *argv[])
{
MPLS_PL *pl;
int opt;
int ii, pl_ii;
MPLS_PL *pl_list[1000];
struct stat st;
char *path = NULL;
DIR *dir = NULL;
do {
opt = getopt(argc, argv, OPTS);
switch (opt) {
case -1:
break;
case 'v':
verbose = 1;
break;
case 'l':
clip_list = 1;
break;
case 'i':
playlist_info = 1;
break;
case 'c':
chapter_marks = 1;
break;
case 'p':
sub_paths = 1;
break;
case 'P':
pip_metadata = 1;
break;
case 'd':
dups = 1;
break;
case 'r':
repeats = atoi(optarg);
break;
case 'f':
repeats = 2;
dups = 1;
seconds = 900;
break;
case 's':
seconds = atoi(optarg);
break;
default:
_usage(argv[0]);
break;
}
} while (opt != -1);
if (optind >= argc) {
_usage(argv[0]);
}
for (pl_ii = 0, ii = optind; pl_ii < 1000 && ii < argc; ii++) {
if (stat(argv[ii], &st)) {
continue;
}
dir = NULL;
if (S_ISDIR(st.st_mode)) {
printf("Directory: %s:\n", argv[ii]);
/* drop old ones (do not check for duplicates across directories) */
for (int jj = 0; jj < pl_ii; jj++) {
bd_free_mpls(pl_list[jj]);
}
pl_ii = 0;
path = _mk_path(argv[ii], PLAYLIST_DIR);
if (path == NULL) {
fprintf(stderr, "Failed to find playlist path: %s\n", argv[ii]);
continue;
}
dir = opendir(path);
if (dir == NULL) {
fprintf(stderr, "Failed to open dir: %s\n", path);
free(path);
continue;
}
}
if (dir != NULL) {
char **dirlist = (char**)calloc(10001, sizeof(char*));
if (!dirlist) {
continue;
}
struct dirent *ent;
int jj = 0;
for (ent = readdir(dir); ent != NULL && jj < 1000; ent = readdir(dir)) {
char *s = (char*)malloc(strlen(ent->d_name) + 1);
if (s) {
dirlist[jj++] = strcpy(s, ent->d_name);
}
}
qsort(dirlist, jj, sizeof(char*), _qsort_str_cmp);
for (jj = 0; dirlist[jj] != NULL && pl_ii < 1000; jj++) {
char *name = NULL;
name = _mk_path(path, dirlist[jj]);
if (name == NULL) {
continue;
}
free(dirlist[jj]);
if (stat(name, &st)) {
free(name);
continue;
}
if (!S_ISREG(st.st_mode)) {
free(name);
continue;
}
pl = _process_file(name, pl_list, pl_ii);
free(name);
if (pl != NULL) {
pl_list[pl_ii++] = pl;
}
}
free(dirlist);
free(path);
closedir(dir);
dir = NULL;
} else {
pl = _process_file(argv[ii], pl_list, pl_ii);
if (pl != NULL) {
pl_list[pl_ii++] = pl;
}
}
if (pl_ii >= 999) {
fprintf(stderr, "Error: too many play lists given. Output is truncated.\n");
}
}
// Cleanup
for (ii = 0; ii < pl_ii; ii++) {
bd_free_mpls(pl_list[ii]);
}
return 0;
}