HandBrake/libhb/dvd.c

1461 lines
48 KiB
C

/* dvd.c
Copyright (c) 2003-2025 HandBrake Team
This file is part of the HandBrake source code
Homepage: <http://handbrake.fr/>.
It may be used under the terms of the GNU General Public License v2.
For full terms see the file COPYING file or visit http://www.gnu.org/licenses/gpl-2.0.html
*/
#include "libavcodec/avcodec.h"
#include "handbrake/handbrake.h"
#include "handbrake/lang.h"
#include "handbrake/dvd.h"
#include "dvdread/ifo_read.h"
#include "dvdread/ifo_print.h"
#include "dvdread/nav_read.h"
static hb_dvd_t * hb_dvdread_init( hb_handle_t * h, const char * path );
static void hb_dvdread_close( hb_dvd_t ** _d );
static char * hb_dvdread_name( char * path );
static int hb_dvdread_title_count( hb_dvd_t * d );
static hb_title_t * hb_dvdread_title_scan( hb_dvd_t * d, int t, uint64_t min_duration, uint64_t max_duration );
static int hb_dvdread_start( hb_dvd_t * d, hb_title_t *title, int chapter );
static void hb_dvdread_stop( hb_dvd_t * d );
static int hb_dvdread_seek( hb_dvd_t * d, float f );
static hb_buffer_t * hb_dvdread_read( hb_dvd_t * d );
static int hb_dvdread_chapter( hb_dvd_t * d );
static int hb_dvdread_angle_count( hb_dvd_t * d );
static void hb_dvdread_set_angle( hb_dvd_t * d, int angle );
static int hb_dvdread_main_feature( hb_dvd_t * d, hb_list_t * list_title );
hb_dvd_func_t hb_dvdread_func =
{
hb_dvdread_init,
hb_dvdread_close,
hb_dvdread_name,
hb_dvdread_title_count,
hb_dvdread_title_scan,
hb_dvdread_start,
hb_dvdread_stop,
hb_dvdread_seek,
hb_dvdread_read,
hb_dvdread_chapter,
hb_dvdread_angle_count,
hb_dvdread_set_angle,
hb_dvdread_main_feature
};
static hb_dvd_func_t *dvd_methods = &hb_dvdread_func;
/***********************************************************************
* Local prototypes
**********************************************************************/
static void FindNextCell( hb_dvdread_t * );
static int dvdtime2msec( dvd_time_t * );
static int hb_dvdread_is_break( hb_dvdread_t * d );
hb_dvd_func_t * hb_dvdread_methods( void )
{
return &hb_dvdread_func;
}
static int hb_dvdread_main_feature( hb_dvd_t * e, hb_list_t * list_title )
{
int ii;
uint64_t longest_duration = 0;
int longest = -1;
for ( ii = 0; ii < hb_list_count( list_title ); ii++ )
{
hb_title_t * title = hb_list_item( list_title, ii );
if ( title->duration > longest_duration )
{
longest_duration = title->duration;
longest = title->index;
}
}
return longest;
}
static char * hb_dvdread_name( char * path )
{
static char name[1024];
unsigned char unused[1024];
dvd_reader_t * reader;
reader = DVDOpen( path );
if( !reader )
{
return NULL;
}
if( DVDUDFVolumeInfo( reader, name, sizeof( name ),
unused, sizeof( unused ) ) )
{
DVDClose( reader );
return NULL;
}
DVDClose( reader );
return name;
}
/***********************************************************************
* hb_dvdread_init
***********************************************************************
*
**********************************************************************/
hb_dvd_t * hb_dvdread_init( hb_handle_t * h, const char * path )
{
hb_dvd_t * e;
hb_dvdread_t * d;
int region_mask;
e = calloc( sizeof( hb_dvd_t ), 1 );
d = &(e->dvdread);
d->h = h;
/* Log DVD drive region code */
if ( hb_dvd_region( path, &region_mask ) == 0 )
{
hb_log( "dvd: Region mask 0x%02x", region_mask );
if ( region_mask == 0xFF )
{
hb_log( "dvd: Warning, DVD device has no region set" );
}
}
/* Open device */
if( !( d->reader = DVDOpen( path ) ) )
{
/*
* Not an error, may be a stream - which we'll try in a moment.
*/
hb_log( "dvd: not a dvd - trying as a stream/file instead" );
goto fail;
}
/* Open main IFO */
if( !( d->vmg = ifoOpen( d->reader, 0 ) ) )
{
hb_log( "dvd: not a dvd - trying as a stream/file instead" );
goto fail;
}
d->path = strdup( path );
return e;
fail:
if( d->vmg ) ifoClose( d->vmg );
if( d->reader ) DVDClose( d->reader );
free( e );
return NULL;
}
/***********************************************************************
* hb_dvdread_title_count
**********************************************************************/
static int hb_dvdread_title_count( hb_dvd_t * e )
{
hb_dvdread_t *d = &(e->dvdread);
return d->vmg->tt_srpt->nr_of_srpts;
}
static void add_subtitle( hb_list_t * list_subtitle, int position,
iso639_lang_t * lang, int lang_extension,
uint8_t * palette, int style )
{
hb_subtitle_t * subtitle;
int ii, count;
count = hb_list_count(list_subtitle);
for (ii = 0; ii < count; ii++)
{
subtitle = hb_list_item(list_subtitle, ii);
if (((subtitle->id >> 8) & 0x1f) == position)
{
// The subtitle is already in the list
return;
}
}
subtitle = calloc(sizeof(hb_subtitle_t), 1);
subtitle->track = count;
subtitle->id = ((0x20 + position) << 8) | 0xbd;
snprintf(subtitle->lang, sizeof( subtitle->lang ), "%s",
strlen(lang->native_name) ? lang->native_name : lang->eng_name);
snprintf(subtitle->iso639_2, sizeof( subtitle->iso639_2 ), "%s",
lang->iso639_2);
subtitle->format = PICTURESUB;
subtitle->source = VOBSUB;
subtitle->config.dest = RENDERSUB;
subtitle->stream_type = 0xbd;
subtitle->substream_type = 0x20 + position;
subtitle->codec = WORK_DECAVSUB;
subtitle->codec_param = AV_CODEC_ID_DVD_SUBTITLE;
subtitle->timebase.num = 1;
subtitle->timebase.den = 90000;
memcpy(subtitle->palette, palette, 16 * sizeof(uint32_t));
subtitle->palette_set = 1;
const char * name = NULL;
switch (lang_extension)
{
case 1:
subtitle->attributes = HB_SUBTITLE_ATTR_NORMAL;
break;
case 2:
subtitle->attributes = HB_SUBTITLE_ATTR_LARGE;
strcat(subtitle->lang, " Large Type");
name = "Large Type";
break;
case 3:
subtitle->attributes = HB_SUBTITLE_ATTR_CHILDREN;
strcat(subtitle->lang, " Children");
name = "Children";
break;
case 5:
subtitle->attributes = HB_SUBTITLE_ATTR_CC;
strcat(subtitle->lang, " Closed Caption");
name = "Closed Caption";
break;
case 6:
subtitle->attributes = HB_SUBTITLE_ATTR_CC | HB_SUBTITLE_ATTR_LARGE;
strcat(subtitle->lang, " Closed Caption, Large Type");
name = "Closed Caption, Large Type";
break;
case 7:
subtitle->attributes = HB_SUBTITLE_ATTR_CC |
HB_SUBTITLE_ATTR_CHILDREN;
strcat(subtitle->lang, " Closed Caption, Children");
name = "Closed Caption, Children";
break;
case 9:
subtitle->attributes = HB_SUBTITLE_ATTR_FORCED;
strcat(subtitle->lang, " Forced");
break;
case 13:
subtitle->attributes = HB_SUBTITLE_ATTR_COMMENTARY;
strcat(subtitle->lang, " Director's Commentary");
name = "Commentary";
break;
case 14:
subtitle->attributes = HB_SUBTITLE_ATTR_COMMENTARY |
HB_SUBTITLE_ATTR_LARGE;
strcat(subtitle->lang, " Director's Commentary, Large Type");
name = "Commentary, Large Type";
break;
case 15:
subtitle->attributes = HB_SUBTITLE_ATTR_COMMENTARY |
HB_SUBTITLE_ATTR_CHILDREN;
strcat(subtitle->lang, " Director's Commentary, Children");
name = "Commentary, Children";
default:
subtitle->attributes = HB_SUBTITLE_ATTR_UNKNOWN;
break;
}
if (name != NULL)
{
subtitle->name = strdup(name);
}
switch (style)
{
case HB_VOBSUB_STYLE_4_3:
subtitle->attributes |= HB_SUBTITLE_ATTR_4_3;
strcat(subtitle->lang, " (4:3)");
break;
case HB_VOBSUB_STYLE_WIDE:
subtitle->attributes |= HB_SUBTITLE_ATTR_WIDE;
strcat(subtitle->lang, " (Wide Screen)");
break;
case HB_VOBSUB_STYLE_LETTERBOX:
subtitle->attributes |= HB_SUBTITLE_ATTR_LETTERBOX;
strcat(subtitle->lang, " (Letterbox)");
break;
case HB_VOBSUB_STYLE_PANSCAN:
subtitle->attributes |= HB_SUBTITLE_ATTR_PANSCAN;
strcat(subtitle->lang, " (Pan & Scan)");
break;
}
strcat(subtitle->lang, " [");
strcat(subtitle->lang, hb_subsource_name(subtitle->source));
strcat(subtitle->lang, "]");
hb_log("scan: id=0x%x, lang=%s, 3cc=%s ext=%i", subtitle->id,
subtitle->lang, subtitle->iso639_2, lang_extension);
hb_list_add(list_subtitle, subtitle);
}
/***********************************************************************
* hb_dvdread_title_scan
**********************************************************************/
static hb_title_t * hb_dvdread_title_scan( hb_dvd_t * e, int t, uint64_t min_duration, uint64_t max_duration)
{
hb_dvdread_t *d = &(e->dvdread);
hb_title_t * title;
ifo_handle_t * vts = NULL;
int pgc_id, pgn, i;
hb_chapter_t * chapter;
char name[1024];
unsigned char unused[1024];
const char * codec_name;
hb_log( "scan: scanning title %d", t );
title = hb_title_init( d->path, t );
title->type = HB_DVD_TYPE;
if( DVDUDFVolumeInfo(d->reader, name, sizeof(name), unused, sizeof(unused)))
{
char * p_cur, * p_last = d->path;
for( p_cur = d->path; *p_cur; p_cur++ )
{
if( IS_DIR_SEP(p_cur[0]) && p_cur[1] )
{
p_last = &p_cur[1];
}
}
title->name = strdup(p_last);
char *dot_term = strrchr(title->name, '.');
if (dot_term)
*dot_term = '\0';
}
else
{
title->name = strdup(name);
}
/* VTS which our title is in */
int title_vts = d->vmg->tt_srpt->title[t-1].title_set_nr;
if ( !title_vts )
{
/* A VTS of 0 means the title wasn't found in the title set */
hb_log("Invalid VTS (title set) number: %i", title_vts);
goto fail;
}
hb_log( "scan: opening IFO for VTS %d", title_vts );
if( !( vts = ifoOpen( d->reader, title_vts ) ) )
{
hb_log( "scan: ifoOpen failed" );
goto fail;
}
/* ignore titles with bogus cell addresses so we don't abort later
* in libdvdread. */
for ( i = 0; i < vts->vts_c_adt->nr_of_vobs; ++i)
{
if( (vts->vts_c_adt->cell_adr_table[i].start_sector & 0xffffff ) ==
0xffffff )
{
hb_log( "scan: cell_adr_table[%d].start_sector invalid (0x%x) "
"- skipping title", i,
vts->vts_c_adt->cell_adr_table[i].start_sector );
goto fail;
}
if( (vts->vts_c_adt->cell_adr_table[i].last_sector & 0xffffff ) ==
0xffffff )
{
hb_log( "scan: cell_adr_table[%d].last_sector invalid (0x%x) "
"- skipping title", i,
vts->vts_c_adt->cell_adr_table[i].last_sector );
goto fail;
}
if( vts->vts_c_adt->cell_adr_table[i].start_sector >=
vts->vts_c_adt->cell_adr_table[i].last_sector )
{
hb_log( "scan: cell_adr_table[%d].start_sector (0x%x) "
"is not before last_sector (0x%x) - skipping title", i,
vts->vts_c_adt->cell_adr_table[i].start_sector,
vts->vts_c_adt->cell_adr_table[i].last_sector );
goto fail;
}
}
if( global_verbosity_level == 3 )
{
ifo_print( d->reader, title_vts );
}
/* Position of the title in the VTS */
int title_ttn = d->vmg->tt_srpt->title[t-1].vts_ttn;
if ( title_ttn < 1 || title_ttn > vts->vts_ptt_srpt->nr_of_srpts )
{
hb_log( "invalid VTS PTT offset %d for title %d, skipping", title_ttn, t );
goto fail;
}
/* Get pgc */
pgc_id = vts->vts_ptt_srpt->title[title_ttn-1].ptt[0].pgcn;
if ( pgc_id < 1 || pgc_id > vts->vts_pgcit->nr_of_pgci_srp )
{
hb_log( "invalid PGC ID %d for title %d, skipping", pgc_id, t );
goto fail;
}
pgn = vts->vts_ptt_srpt->title[title_ttn-1].ptt[0].pgn;
d->pgc = vts->vts_pgcit->pgci_srp[pgc_id-1].pgc;
hb_log("pgc_id: %d, pgn: %d: pgc: %p", pgc_id, pgn, d->pgc);
if( !d->pgc || !d->pgc->program_map )
{
hb_log( "scan: pgc not valid, skipping" );
goto fail;
}
if (d->pgc->cell_playback == NULL)
{
hb_log( "invalid PGC cell_playback table for title %d, skipping", t );
goto fail;
}
if( pgn <= 0 || pgn > 99 )
{
hb_log( "scan: pgn %d not valid, skipping", pgn );
goto fail;
}
/* Get duration */
title->duration = 90LL * dvdtime2msec( &d->pgc->playback_time );
title->hours = title->duration / 90000 / 3600;
title->minutes = ( ( title->duration / 90000 ) % 3600 ) / 60;
title->seconds = ( title->duration / 90000 ) % 60;
hb_log( "scan: duration is %02d:%02d:%02d (%"PRId64" ms)",
title->hours, title->minutes, title->seconds,
title->duration / 90 );
/* ignore titles under 10 seconds because they're often stills or
* clips with no audio & our preview code doesn't currently handle
* either of these. */
if( title->duration < min_duration )
{
hb_log( "scan: ignoring title (too short)" );
goto fail;
}
if( max_duration > 0 && title->duration > max_duration )
{
hb_log( "scan: ignoring title (too long)" );
goto fail;
}
/* Detect languages */
for( i = 0; i < vts->vtsi_mat->nr_of_vts_audio_streams; i++ )
{
int audio_format, lang_code, lang_extension, audio_control, position, j;
hb_audio_t * audio, * audio_tmp;
iso639_lang_t * lang;
hb_log( "scan: checking audio %d", i + 1 );
audio = calloc( sizeof( hb_audio_t ), 1 );
audio_format = vts->vtsi_mat->vts_audio_attr[i].audio_format;
lang_code = vts->vtsi_mat->vts_audio_attr[i].lang_code;
lang_extension = vts->vtsi_mat->vts_audio_attr[i].code_extension;
audio_control =
vts->vts_pgcit->pgci_srp[pgc_id-1].pgc->audio_control[i];
if( !( audio_control & 0x8000 ) )
{
hb_log( "scan: audio channel is not active" );
free( audio );
continue;
}
position = ( audio_control & 0x7F00 ) >> 8;
switch( audio_format )
{
case 0x00:
audio->id = ( ( 0x80 + position ) << 8 ) | 0xbd;
audio->config.in.codec = HB_ACODEC_AC3;
audio->config.in.codec_param = AV_CODEC_ID_AC3;
codec_name = "AC3";
break;
case 0x02:
case 0x03:
audio->id = 0xc0 + position;
audio->config.in.codec = HB_ACODEC_FFMPEG;
audio->config.in.codec_param = AV_CODEC_ID_MP2;
codec_name = "MPEG";
break;
case 0x04:
audio->id = ( ( 0xa0 + position ) << 8 ) | 0xbd;
audio->config.in.codec = HB_ACODEC_LPCM;
codec_name = "LPCM";
break;
case 0x06:
audio->id = ( ( 0x88 + position ) << 8 ) | 0xbd;
audio->config.in.codec = HB_ACODEC_DCA;
audio->config.in.codec_param = AV_CODEC_ID_DTS;
codec_name = "DTS";
break;
default:
audio->id = 0;
audio->config.in.codec = 0;
codec_name = "Unknown";
hb_log( "scan: unknown audio codec (%x)",
audio_format );
break;
}
if( !audio->id )
{
free(audio);
continue;
}
/* Check for duplicate tracks */
audio_tmp = NULL;
for( j = 0; j < hb_list_count( title->list_audio ); j++ )
{
audio_tmp = hb_list_item( title->list_audio, j );
if( audio->id == audio_tmp->id )
{
break;
}
audio_tmp = NULL;
}
if( audio_tmp )
{
hb_log( "scan: duplicate audio track" );
free( audio );
continue;
}
lang = lang_for_code( lang_code );
const char * name = NULL;
switch ( lang_extension )
{
case 1:
audio->config.lang.attributes = HB_AUDIO_ATTR_NORMAL;
break;
case 2:
audio->config.lang.attributes = HB_AUDIO_ATTR_VISUALLY_IMPAIRED;
name = "Visually Impaired";
break;
case 3:
audio->config.lang.attributes = HB_AUDIO_ATTR_COMMENTARY;
name = "Commentary";
break;
case 4:
audio->config.lang.attributes = HB_AUDIO_ATTR_ALT_COMMENTARY;
name = "Commentary";
break;
default:
audio->config.lang.attributes = HB_AUDIO_ATTR_NONE;
break;
}
if (name != NULL)
{
audio->config.in.name = strdup(name);
}
snprintf( audio->config.lang.simple,
sizeof( audio->config.lang.simple ), "%s",
strlen( lang->native_name ) ? lang->native_name : lang->eng_name );
snprintf( audio->config.lang.iso639_2,
sizeof( audio->config.lang.iso639_2 ), "%s", lang->iso639_2 );
hb_log("scan: id=0x%x, lang=%s (%s), 3cc=%s ext=%i", audio->id,
audio->config.lang.simple, codec_name,
audio->config.lang.iso639_2, lang_extension);
audio->config.index = hb_list_count(title->list_audio);
audio->config.in.track = i;
audio->config.in.timebase.num = 1;
audio->config.in.timebase.den = 90000;
hb_list_add( title->list_audio, audio );
}
/* Check for subtitles */
for( i = 0; i < vts->vtsi_mat->nr_of_vts_subp_streams; i++ )
{
int spu_control, pos, lang_ext = 0;
iso639_lang_t * lang;
hb_log( "scan: checking subtitle %d", i + 1 );
// spu_control
// 0x80000000 - Subtitle enabled
// 0x1f000000 - Position mask for 4:3 aspect subtitle track
// 0x001f0000 - Position mask for Wide Screen subtitle track
// 0x00001f00 - Position mask for Letterbox subtitle track
// 0x0000001f - Position mask for Pan&Scan subtitle track
spu_control =
vts->vts_pgcit->pgci_srp[pgc_id-1].pgc->subp_control[i];
if( !( spu_control & 0x80000000 ) )
{
hb_log( "scan: subtitle channel is not active" );
continue;
}
lang_ext = vts->vtsi_mat->vts_subp_attr[i].code_extension;
lang = lang_for_code(vts->vtsi_mat->vts_subp_attr[i].lang_code);
// display_aspect_ratio
// 0 = 4:3
// 3 = 16:9
// other = invalid
if (vts->vtsi_mat->vts_video_attr.display_aspect_ratio)
{
// Add Wide Screen subtitle.
pos = (spu_control >> 16) & 0x1F;
add_subtitle(title->list_subtitle, pos, lang, lang_ext,
(uint8_t*)vts->vts_pgcit->pgci_srp[pgc_id-1].pgc->palette,
HB_VOBSUB_STYLE_WIDE);
// permitted_df
// 1 - Letterbox not permitted
// 2 - Pan&Scan not permitted
// 3 - Letterbox and Pan&Scan not permitted
if (!(vts->vtsi_mat->vts_video_attr.permitted_df & 1))
{
// Letterbox permitted. Add Letterbox subtitle.
pos = (spu_control >> 8) & 0x1F;
add_subtitle(title->list_subtitle, pos, lang, lang_ext,
(uint8_t*)vts->vts_pgcit->pgci_srp[pgc_id-1].pgc->palette,
HB_VOBSUB_STYLE_LETTERBOX);
}
if (!(vts->vtsi_mat->vts_video_attr.permitted_df & 2))
{
// Pan&Scan permitted. Add Pan&Scan subtitle.
pos = spu_control & 0x1F;
add_subtitle(title->list_subtitle, pos, lang, lang_ext,
(uint8_t*)vts->vts_pgcit->pgci_srp[pgc_id-1].pgc->palette,
HB_VOBSUB_STYLE_PANSCAN);
}
}
else
{
pos = (spu_control >> 24) & 0x1F;
add_subtitle(title->list_subtitle, pos, lang, lang_ext,
(uint8_t*)vts->vts_pgcit->pgci_srp[pgc_id-1].pgc->palette,
HB_VOBSUB_STYLE_4_3);
}
}
/* Chapters */
hb_log( "scan: title %d has %d chapters", t,
vts->vts_ptt_srpt->title[title_ttn-1].nr_of_ptts );
for( i = 0;
i < vts->vts_ptt_srpt->title[title_ttn-1].nr_of_ptts; i++ )
{
char chapter_title[80];
chapter = calloc( sizeof( hb_chapter_t ), 1 );
/* remember the on-disc chapter number */
chapter->index = i + 1;
snprintf( chapter_title, sizeof(chapter_title), "Chapter %d", chapter->index );
hb_chapter_set_title( chapter, chapter_title );
pgc_id = vts->vts_ptt_srpt->title[title_ttn-1].ptt[i].pgcn;
pgn = vts->vts_ptt_srpt->title[title_ttn-1].ptt[i].pgn;
d->pgc = vts->vts_pgcit->pgci_srp[pgc_id-1].pgc;
/* Start cell */
int cell_start, cell_end;
cell_start = d->pgc->program_map[pgn-1] - 1;
// if there are no more programs in this pgc, the end cell is the
// last cell. Otherwise it's the cell before the start cell of the
// next program.
if ( pgn == d->pgc->nr_of_programs )
{
cell_end = d->pgc->nr_of_cells - 1;
}
else
{
cell_end = d->pgc->program_map[pgn] - 2;;
}
/* Block count, duration */
chapter->duration = 0;
d->cell_cur = cell_start;
while( d->cell_cur <= cell_end )
{
#define cp d->pgc->cell_playback[d->cell_cur]
chapter->duration += 90LL * dvdtime2msec( &cp.playback_time );
#undef cp
FindNextCell( d );
d->cell_cur = d->cell_next;
}
hb_list_add( title->list_chapter, chapter );
}
for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
{
chapter = hb_list_item( title->list_chapter, i );
int seconds = ( chapter->duration + 45000 ) / 90000;
chapter->hours = ( seconds / 3600 );
chapter->minutes = ( seconds % 3600 ) / 60;
chapter->seconds = ( seconds % 60 );
hb_log( "scan: chap %d, %"PRId64" ms",
chapter->index, chapter->duration / 90 );
}
/* Get aspect. We don't get width/height/rate infos here as
they tend to be wrong */
switch( vts->vtsi_mat->vts_video_attr.display_aspect_ratio )
{
case 0:
title->container_dar.num = 4;
title->container_dar.den = 3;
break;
case 3:
title->container_dar.num = 16;
title->container_dar.den = 9;
break;
default:
hb_log( "scan: unknown aspect" );
goto fail;
}
switch( vts->vtsi_mat->vts_video_attr.mpeg_version )
{
case 0:
title->video_codec = WORK_DECAVCODECV;
title->video_codec_param = AV_CODEC_ID_MPEG1VIDEO;
break;
case 1:
title->video_codec = WORK_DECAVCODECV;
title->video_codec_param = AV_CODEC_ID_MPEG2VIDEO;
break;
default:
hb_log("scan: unknown/reserved MPEG version %d",
vts->vtsi_mat->vts_video_attr.mpeg_version);
title->video_codec = WORK_DECAVCODECV;
title->video_codec_param = AV_CODEC_ID_MPEG2VIDEO;
break;
}
hb_log("scan: aspect = %d:%d",
title->container_dar.num, title->container_dar.den);
/* This title is ok so far */
goto cleanup;
fail:
hb_title_close( &title );
cleanup:
if( vts ) ifoClose( vts );
return title;
}
/***********************************************************************
* hb_dvdread_start
***********************************************************************
* Title and chapter start at 1
**********************************************************************/
static int hb_dvdread_start( hb_dvd_t * e, hb_title_t *title, int chapter )
{
hb_dvdread_t *d = &(e->dvdread);
int pgc_id, pgn;
int i;
int t = title->index;
/* Open the IFO and the VOBs for this title */
d->vts = d->vmg->tt_srpt->title[t-1].title_set_nr;
d->ttn = d->vmg->tt_srpt->title[t-1].vts_ttn;
if( !( d->ifo = ifoOpen( d->reader, d->vts ) ) )
{
hb_error( "dvd: ifoOpen failed for VTS %d", d->vts );
return 0;
}
if( !( d->file = DVDOpenFile( d->reader, d->vts,
DVD_READ_TITLE_VOBS ) ) )
{
hb_error( "dvd: DVDOpenFile failed for VTS %d", d->vts );
return 0;
}
/* Get title first and last blocks */
pgc_id = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[0].pgcn;
pgn = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[0].pgn;
d->pgc = d->ifo->vts_pgcit->pgci_srp[pgc_id-1].pgc;
d->cell_start = d->pgc->program_map[pgn - 1] - 1;
d->cell_end = d->pgc->nr_of_cells - 1;
d->title_start = d->pgc->cell_playback[d->cell_start].first_sector;
d->title_end = d->pgc->cell_playback[d->cell_end].last_sector;
/* Block count */
d->title_block_count = 0;
for( i = d->cell_start; i <= d->cell_end; i++ )
{
d->title_block_count += d->pgc->cell_playback[i].last_sector + 1 -
d->pgc->cell_playback[i].first_sector;
}
/* Get pgc for the current chapter */
pgc_id = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[chapter-1].pgcn;
pgn = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[chapter-1].pgn;
d->pgc = d->ifo->vts_pgcit->pgci_srp[pgc_id-1].pgc;
/* Get the two first cells */
d->cell_cur = d->pgc->program_map[pgn-1] - 1;
FindNextCell( d );
d->block = d->pgc->cell_playback[d->cell_cur].first_sector;
d->next_vobu = d->block;
d->pack_len = 0;
d->cell_overlap = 0;
d->in_cell = 0;
d->in_sync = 2;
return 1;
}
/***********************************************************************
* hb_dvdread_stop
***********************************************************************
*
**********************************************************************/
static void hb_dvdread_stop( hb_dvd_t * e )
{
hb_dvdread_t *d = &(e->dvdread);
if( d->ifo )
{
ifoClose( d->ifo );
d->ifo = NULL;
}
if( d->file )
{
DVDCloseFile( d->file );
d->file = NULL;
}
}
/***********************************************************************
* hb_dvdread_seek
***********************************************************************
*
**********************************************************************/
static int hb_dvdread_seek( hb_dvd_t * e, float f )
{
hb_dvdread_t *d = &(e->dvdread);
int count, sizeCell;
int i;
count = f * d->title_block_count;
if (d->file == NULL)
{
return 1;
}
for( i = d->cell_start; i <= d->cell_end; i++ )
{
sizeCell = d->pgc->cell_playback[i].last_sector + 1 -
d->pgc->cell_playback[i].first_sector;
if( count < sizeCell )
{
d->cell_cur = i;
d->cur_cell_id = 0;
FindNextCell( d );
/* Now let hb_dvdread_read find the next VOBU */
d->next_vobu = d->pgc->cell_playback[i].first_sector + count;
d->pack_len = 0;
break;
}
count -= sizeCell;
}
if( i > d->cell_end )
{
return 0;
}
/*
* Assume that we are in sync, even if we are not given that it is obvious
* that we are out of sync if we have just done a seek.
*/
d->in_sync = 2;
return 1;
}
/***********************************************************************
* is_nav_pack
***********************************************************************/
int is_nav_pack( unsigned char *buf )
{
/*
* The NAV Pack is comprised of the PCI Packet and DSI Packet, both
* of these start at known offsets and start with a special identifier.
*
* NAV = {
* PCI = { 00 00 01 bf # private stream header
* ?? ?? # length
* 00 # substream
* ...
* }
* DSI = { 00 00 01 bf # private stream header
* ?? ?? # length
* 01 # substream
* ...
* }
*
* The PCI starts at offset 0x26 into the sector, and the DSI starts at 0x400
*
* This information from: http://dvd.sourceforge.net/dvdinfo/
*/
if( ( buf[0x26] == 0x00 && // PCI
buf[0x27] == 0x00 &&
buf[0x28] == 0x01 &&
buf[0x29] == 0xbf &&
buf[0x2c] == 0x00 ) &&
( buf[0x400] == 0x00 && // DSI
buf[0x401] == 0x00 &&
buf[0x402] == 0x01 &&
buf[0x403] == 0xbf &&
buf[0x406] == 0x01 ) )
{
return ( 1 );
} else {
return ( 0 );
}
}
/***********************************************************************
* hb_dvdread_read
***********************************************************************
*
**********************************************************************/
static hb_buffer_t * hb_dvdread_read( hb_dvd_t * e )
{
hb_dvdread_t *d = &(e->dvdread);
hb_buffer_t *b = hb_buffer_init( HB_DVD_READ_BUFFER_SIZE );
top:
if( !d->pack_len )
{
/* New pack */
dsi_t dsi_pack;
int error = 0;
// if we've just finished the last cell of the title we don't
// want to read another block because our next_vobu pointer
// is probably invalid. Just return 'no data' & our caller
// should check and discover we're at eof.
if ( d->cell_cur > d->cell_end )
{
hb_buffer_close( &b );
return NULL;
}
for( ;; )
{
int block, pack_len, next_vobu, read_retry;
for( read_retry = 1; read_retry < 1024; read_retry++ )
{
if( DVDReadBlocks( d->file, d->next_vobu, 1, b->data ) == 1 )
{
/*
* Successful read.
*/
if( read_retry > 1 && !is_nav_pack( b->data) )
{
// But wasn't a nav pack, so carry on looking
read_retry = 1;
d->next_vobu++;
continue;
}
break;
} else {
// First retry the same block, then try the next one,
// adjust the skip increment upwards so that we can skip
// large sections of bad blocks more efficiently (at the
// cost of some missed good blocks at the end).
hb_log( "dvd: vobu read error blk %d - skipping to next blk incr %d",
d->next_vobu, (read_retry * 10));
d->next_vobu += (read_retry * 10);
}
}
if( read_retry == 1024 )
{
// That's too many bad blocks, jump to the start of the
// next cell.
hb_log( "dvd: vobu read error blk %d - skipping to cell %d",
d->next_vobu, d->cell_next );
d->cell_cur = d->cell_next;
if ( d->cell_cur > d->cell_end )
{
hb_buffer_close( &b );
hb_set_work_error(d->h, HB_ERROR_READ);
return NULL;
}
d->in_cell = 0;
d->next_vobu = d->pgc->cell_playback[d->cell_cur].first_sector;
FindNextCell( d );
d->cell_overlap = 1;
continue;
}
if ( !is_nav_pack( b->data ) ) {
(d->next_vobu)++;
if( d->in_sync == 1 ) {
hb_log("dvd: Lost sync, searching for NAV pack at blk %d",
d->next_vobu);
d->in_sync = 0;
}
continue;
}
navRead_DSI( &dsi_pack, &b->data[DSI_START_BYTE] );
if ( d->in_sync == 0 && d->cur_cell_id &&
(d->cur_vob_id != dsi_pack.dsi_gi.vobu_vob_idn ||
d->cur_cell_id != dsi_pack.dsi_gi.vobu_c_idn ) )
{
// We walked out of the cell we're supposed to be in.
// If we're not at the start of our next cell go there.
hb_log("dvd: left cell %d (%u,%u) for (%u,%u) at block %u",
d->cell_cur, d->cur_vob_id, d->cur_cell_id,
dsi_pack.dsi_gi.vobu_vob_idn, dsi_pack.dsi_gi.vobu_c_idn,
d->next_vobu );
if ( d->next_vobu != d->pgc->cell_playback[d->cell_next].first_sector )
{
d->next_vobu = d->pgc->cell_playback[d->cell_next].first_sector;
d->cur_cell_id = 0;
continue;
}
}
block = dsi_pack.dsi_gi.nv_pck_lbn;
pack_len = dsi_pack.dsi_gi.vobu_ea;
// There are a total of 21 next vobu offsets (and 21 prev_vobu
// offsets) in the navpack SRI structure. The primary one is
// 'next_vobu' which is the offset (in dvd blocks) from the current
// block to the start of the next vobu. If the block at 'next_vobu'
// can't be read, 'next_video' is the offset to the vobu closest to it.
// The other 19 offsets are vobus at successively longer distances from
// the current block (this is so that a hardware player can do
// adaptive error recovery to skip over a bad spot on the disk). In all
// these offsets the high bit is set to indicate when it contains a
// valid offset. The next bit (2^30) is set to indicate that there's
// another valid offset in the SRI that's closer to the current block.
// A hardware player uses these bits to chose the closest valid offset
// and uses that as its next vobu. (Some mastering schemes appear to
// put a bogus offset in next_vobu with the 2^30 bit set & the
// correct offset in next_video. This works fine in hardware players
// but will mess up software that doesn't implement the full next
// vobu decision logic.) In addition to the flag bits there's a
// reserved value of the offset that indicates 'no next vobu' (which
// indicates the end of a cell). But having no next vobu pointer with a
// 'valid' bit set also indicates end of cell. Different mastering
// schemes seem to use all possible combinations of the flag bits
// and reserved values to indicate end of cell so we have to check
// them all or we'll get a disk read error from following an
// invalid offset.
uint32_t next_ptr = dsi_pack.vobu_sri.next_vobu;
if ( ( next_ptr & ( 1 << 31 ) ) == 0 ||
( next_ptr & ( 1 << 30 ) ) != 0 ||
( next_ptr & 0x3fffffff ) == 0x3fffffff )
{
next_ptr = dsi_pack.vobu_sri.next_video;
if ( ( next_ptr & ( 1 << 31 ) ) == 0 ||
( next_ptr & 0x3fffffff ) == 0x3fffffff )
{
// don't have a valid forward pointer - assume end-of-cell
d->block = block;
d->pack_len = pack_len;
break;
}
}
next_vobu = block + ( next_ptr & 0x3fffffff );
if( pack_len > 0 &&
pack_len < 1024 &&
block >= d->next_vobu &&
( block <= d->title_start + d->title_block_count ||
block <= d->title_end ) )
{
d->block = block;
d->pack_len = pack_len;
d->next_vobu = next_vobu;
break;
}
/* Wasn't a valid VOBU, try next block */
if( ++error > 1024 )
{
hb_error( "dvd: couldn't find a VOBU after 1024 blocks" );
hb_buffer_close( &b );
hb_set_work_error(d->h, HB_ERROR_READ);
return NULL;
}
(d->next_vobu)++;
}
if( d->in_sync == 0 || d->in_sync == 2 )
{
if( d->in_sync == 0 )
{
hb_log( "dvd: In sync with DVD at block %d", d->block );
}
d->in_sync = 1;
}
// Revert the cell overlap, and check for a chapter break
// If this cell is zero length (prev_vobu & next_vobu both
// set to END_OF_CELL) we need to check for beginning of
// cell before checking for end or we'll advance to the next
// cell too early and fail to generate a chapter mark when this
// cell starts a chapter.
if( ( dsi_pack.vobu_sri.prev_vobu & (1 << 31 ) ) == 0 ||
( dsi_pack.vobu_sri.prev_vobu & 0x3fffffff ) == 0x3fffffff )
{
// A vobu that's not at the start of a cell can have an
// EOC prev pointer (this seems to be associated with some
// sort of drm). The rest of the content in the cell may be
// booby-trapped so treat this like an end-of-cell rather
// than a beginning of cell.
if ( d->pgc->cell_playback[d->cell_cur].first_sector < dsi_pack.dsi_gi.nv_pck_lbn &&
d->pgc->cell_playback[d->cell_cur].last_sector >= dsi_pack.dsi_gi.nv_pck_lbn )
{
hb_log( "dvd: null prev_vobu in cell %d at block %d", d->cell_cur,
d->block );
// treat like end-of-cell then go directly to start of next cell.
d->cell_cur = d->cell_next;
d->in_cell = 0;
d->next_vobu = d->pgc->cell_playback[d->cell_cur].first_sector;
FindNextCell( d );
d->cell_overlap = 1;
goto top;
}
else
{
if ( d->block != d->pgc->cell_playback[d->cell_cur].first_sector )
{
hb_deep_log(3, "dvd: beginning of cell %d at block %d", d->cell_cur,
d->block);
}
if( d->in_cell )
{
hb_error( "dvd: assuming missed end of cell %d at block %d", d->cell_cur, d->block );
d->cell_cur = d->cell_next;
d->next_vobu = d->pgc->cell_playback[d->cell_cur].first_sector;
FindNextCell( d );
d->cell_overlap = 1;
d->in_cell = 0;
} else {
d->in_cell = 1;
}
d->cur_vob_id = dsi_pack.dsi_gi.vobu_vob_idn;
d->cur_cell_id = dsi_pack.dsi_gi.vobu_c_idn;
d->cell_overlap = 0;
}
}
if( ( dsi_pack.vobu_sri.next_vobu & (1 << 31 ) ) == 0 ||
( dsi_pack.vobu_sri.next_vobu & 0x3fffffff ) == 0x3fffffff )
{
if ( d->block <= d->pgc->cell_playback[d->cell_cur].first_sector ||
d->block > d->pgc->cell_playback[d->cell_cur].last_sector )
{
hb_deep_log(3, "dvd: end of cell %d at block %d", d->cell_cur,
d->block);
}
d->cell_cur = d->cell_next;
d->in_cell = 0;
d->next_vobu = d->pgc->cell_playback[d->cell_cur].first_sector;
FindNextCell( d );
d->cell_overlap = 1;
}
}
else
{
if( DVDReadBlocks( d->file, d->block, 1, b->data ) != 1 )
{
// this may be a real DVD error or may be DRM. Either way
// we don't want to quit because of one bad block so set
// things up so we'll advance to the next vobu and recurse.
hb_error( "dvd: DVDReadBlocks failed (%d), skipping to vobu %u",
d->block, d->next_vobu );
d->pack_len = 0;
goto top; /* XXX need to restructure this routine & avoid goto */
}
d->pack_len--;
}
if (b != NULL)
{
b->s.new_chap = hb_dvdread_is_break( d );
}
d->block++;
return b;
}
/***********************************************************************
* hb_dvdread_chapter
***********************************************************************
* Returns in which chapter the next block to be read is.
* Chapter numbers start at 1.
**********************************************************************/
static int hb_dvdread_chapter( hb_dvd_t * e )
{
hb_dvdread_t *d = &(e->dvdread);
int i;
int pgc_id, pgn;
int nr_of_ptts = d->ifo->vts_ptt_srpt->title[d->ttn-1].nr_of_ptts;
pgc_t * pgc;
for( i = nr_of_ptts - 1;
i >= 0;
i-- )
{
/* Get pgc for chapter (i+1) */
pgc_id = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[i].pgcn;
pgn = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[i].pgn;
pgc = d->ifo->vts_pgcit->pgci_srp[pgc_id-1].pgc;
if( d->cell_cur - d->cell_overlap >= pgc->program_map[pgn-1] - 1 &&
d->cell_cur - d->cell_overlap <= pgc->nr_of_cells - 1 )
{
/* We are in this chapter */
return i + 1;
}
}
/* End of title */
return -1;
}
/***********************************************************************
* hb_dvdread_is_break
***********************************************************************
* Returns chapter number if the current block is a new chapter start
**********************************************************************/
static int hb_dvdread_is_break( hb_dvdread_t * d )
{
int i;
int pgc_id, pgn;
int nr_of_ptts = d->ifo->vts_ptt_srpt->title[d->ttn-1].nr_of_ptts;
pgc_t * pgc;
int cell;
for (i = nr_of_ptts - 1; i >= 0; i--)
{
/* Get pgc for chapter (i+1) */
pgc_id = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[i].pgcn;
pgn = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[i].pgn;
pgc = d->ifo->vts_pgcit->pgci_srp[pgc_id-1].pgc;
cell = pgc->program_map[pgn-1] - 1;
if( cell < d->cell_start )
break;
// This must not match against the start cell.
if (pgc->cell_playback[cell].first_sector == d->block)
{
return i + 1;
}
}
return 0;
}
/***********************************************************************
* hb_dvdread_close
***********************************************************************
* Closes and frees everything
**********************************************************************/
static void hb_dvdread_close( hb_dvd_t ** _d )
{
hb_dvdread_t * d = &((*_d)->dvdread);
if( d->vmg )
{
ifoClose( d->vmg );
}
if( d->reader )
{
DVDClose( d->reader );
}
free( d->path );
free( d );
*_d = NULL;
}
/***********************************************************************
* hb_dvdread_angle_count
***********************************************************************
* Returns the number of angles supported. We do not support angles
* with dvdread
**********************************************************************/
static int hb_dvdread_angle_count( hb_dvd_t * d )
{
return 1;
}
/***********************************************************************
* hb_dvdread_set_angle
***********************************************************************
* Sets the angle to read. Not supported with dvdread
**********************************************************************/
static void hb_dvdread_set_angle( hb_dvd_t * d, int angle )
{
}
/***********************************************************************
* FindNextCell
***********************************************************************
* Assumes pgc and cell_cur are correctly set, and sets cell_next to the
* cell to be read when we will be done with cell_cur.
**********************************************************************/
static void FindNextCell( hb_dvdread_t * d )
{
int i = 0;
if( d->pgc->cell_playback[d->cell_cur].block_type ==
BLOCK_TYPE_ANGLE_BLOCK )
{
while( d->pgc->cell_playback[d->cell_cur+i].block_mode !=
BLOCK_MODE_LAST_CELL )
{
i++;
}
d->cell_next = d->cell_cur + i + 1;
hb_log( "dvd: Skipping multi-angle cells %d-%d",
d->cell_cur,
d->cell_next - 1 );
}
else
{
d->cell_next = d->cell_cur + 1;
}
}
/***********************************************************************
* dvdtime2msec
***********************************************************************
* From lsdvd
**********************************************************************/
static int dvdtime2msec(dvd_time_t * dt)
{
double frames_per_s[4] = {-1.0, 25.00, -1.0, 29.97};
double fps = frames_per_s[(dt->frame_u & 0xc0) >> 6];
long ms;
ms = (((dt->hour & 0xf0) >> 3) * 5 + (dt->hour & 0x0f)) * 3600000;
ms += (((dt->minute & 0xf0) >> 3) * 5 + (dt->minute & 0x0f)) * 60000;
ms += (((dt->second & 0xf0) >> 3) * 5 + (dt->second & 0x0f)) * 1000;
if( fps > 0 )
{
ms += (((dt->frame_u & 0x30) >> 3) * 5 +
(dt->frame_u & 0x0f)) * 1000.0 / fps;
}
return ms;
}
char * hb_dvd_name( char * path )
{
return dvd_methods->name(path);
}
hb_dvd_t * hb_dvd_init( hb_handle_t * h, const char * path )
{
return dvd_methods->init(h, path);
}
int hb_dvd_title_count( hb_dvd_t * d )
{
return dvd_methods->title_count(d);
}
hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t, uint64_t min_duration, uint64_t max_duration )
{
return dvd_methods->title_scan(d, t, min_duration, max_duration);
}
int hb_dvd_start( hb_dvd_t * d, hb_title_t *title, int chapter )
{
return dvd_methods->start(d, title, chapter);
}
void hb_dvd_stop( hb_dvd_t * d )
{
dvd_methods->stop(d);
}
int hb_dvd_seek( hb_dvd_t * d, float f )
{
return dvd_methods->seek(d, f);
}
hb_buffer_t * hb_dvd_read( hb_dvd_t * d )
{
return dvd_methods->read(d);
}
int hb_dvd_chapter( hb_dvd_t * d )
{
return dvd_methods->chapter(d);
}
void hb_dvd_close( hb_dvd_t ** _d )
{
dvd_methods->close(_d);
}
int hb_dvd_angle_count( hb_dvd_t * d )
{
return dvd_methods->angle_count(d);
}
void hb_dvd_set_angle( hb_dvd_t * d, int angle )
{
dvd_methods->set_angle(d, angle);
}
int hb_dvd_main_feature( hb_dvd_t * d, hb_list_t * list_title )
{
return dvd_methods->main_feature(d, list_title);
}
// hb_dvd_set_dvdnav must only be called when no dvd source is open
// it rips the rug out from under things so be careful
void hb_dvd_set_dvdnav( int enable )
{
if (enable)
dvd_methods = hb_dvdnav_methods();
else
dvd_methods = hb_dvdread_methods();
}