/*

 * 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;

}