libhb/dvd.c

   1 /* $Id: dvd.c,v 1.12 2005/11/25 15:05:25 titer Exp $
   2
   3    This file is part of the HandBrake source code.
   4    Homepage: <http://handbrake.fr/>.
   5    It may be used under the terms of the GNU General Public License. */
   6
   7 #include "hb.h"
   8 #include "lang.h"
   9 #include "dvd.h"
  10
  11 #include "dvdread/ifo_read.h"
  12 #include "dvdread/nav_read.h"
  13
  14 static hb_dvd_t    * hb_dvdread_init( char * path );
  15 static void          hb_dvdread_close( hb_dvd_t ** _d );
  16 static char        * hb_dvdread_name( char * path );
  17 static int           hb_dvdread_title_count( hb_dvd_t * d );
  18 static hb_title_t  * hb_dvdread_title_scan( hb_dvd_t * d, int t );
  19 static int           hb_dvdread_start( hb_dvd_t * d, hb_title_t *title, int chapter );
  20 static void          hb_dvdread_stop( hb_dvd_t * d );
  21 static int           hb_dvdread_seek( hb_dvd_t * d, float f );
  22 static int           hb_dvdread_read( hb_dvd_t * d, hb_buffer_t * b );
  23 static int           hb_dvdread_chapter( hb_dvd_t * d );
  24 static int           hb_dvdread_angle_count( hb_dvd_t * d );
  25 static void          hb_dvdread_set_angle( hb_dvd_t * d, int angle );
  26
  27 hb_dvd_func_t hb_dvdread_func =
  28 {
  29     hb_dvdread_init,
  30     hb_dvdread_close,
  31     hb_dvdread_name,
  32     hb_dvdread_title_count,
  33     hb_dvdread_title_scan,
  34     hb_dvdread_start,
  35     hb_dvdread_stop,
  36     hb_dvdread_seek,
  37     hb_dvdread_read,
  38     hb_dvdread_chapter,
  39     hb_dvdread_angle_count,
  40     hb_dvdread_set_angle
  41 };
  42
  43 static hb_dvd_func_t *dvd_methods = &hb_dvdread_func;
  44
  45 /***********************************************************************
  46  * Local prototypes
  47  **********************************************************************/
  48 static void FindNextCell( hb_dvdread_t * );
  49 static int  dvdtime2msec( dvd_time_t * );
  50 static int hb_dvdread_is_break( hb_dvdread_t * d );
  51
  52 hb_dvd_func_t * hb_dvdread_methods( void )
  53 {
  54     return &hb_dvdread_func;
  55 }
  56
  57 static char * hb_dvdread_name( char * path )
  58 {
  59     static char name[1024];
  60     unsigned char unused[1024];
  61     dvd_reader_t * reader;
  62
  63     reader = DVDOpen( path );
  64     if( !reader )
  65     {
  66         return NULL;
  67     }
  68
  69     if( DVDUDFVolumeInfo( reader, name, sizeof( name ),
  70                           unused, sizeof( unused ) ) )
  71     {
  72         DVDClose( reader );
  73         return NULL;
  74     }
  75
  76     DVDClose( reader );
  77     return name;
  78 }
  79
  80 /***********************************************************************
  81  * hb_dvdread_init
  82  ***********************************************************************
  83  *
  84  **********************************************************************/
  85 hb_dvd_t * hb_dvdread_init( char * path )
  86 {
  87     hb_dvd_t * e;
  88     hb_dvdread_t * d;
  89
  90     e = calloc( sizeof( hb_dvd_t ), 1 );
  91     d = &(e->dvdread);
  92
  93     /* Open device */
  94     if( !( d->reader = DVDOpen( path ) ) )
  95     {
  96         /*
  97          * Not an error, may be a stream - which we'll try in a moment.
  98          */
  99         hb_log( "dvd: not a dvd - trying as a stream/file instead" );
 100         goto fail;
 101     }
 102
 103     /* Open main IFO */
 104     if( !( d->vmg = ifoOpen( d->reader, 0 ) ) )
 105     {
 106         hb_error( "dvd: ifoOpen failed" );
 107         goto fail;
 108     }
 109
 110     d->path = strdup( path );
 111
 112     return e;
 113
 114 fail:
 115     if( d->vmg )    ifoClose( d->vmg );
 116     if( d->reader ) DVDClose( d->reader );
 117     free( d );
 118     return NULL;
 119 }
 120
 121 /***********************************************************************
 122  * hb_dvdread_title_count
 123  **********************************************************************/
 124 static int hb_dvdread_title_count( hb_dvd_t * e )
 125 {
 126     hb_dvdread_t *d = &(e->dvdread);
 127     return d->vmg->tt_srpt->nr_of_srpts;
 128 }
 129
 130 /***********************************************************************
 131  * hb_dvdread_title_scan
 132  **********************************************************************/
 133 static hb_title_t * hb_dvdread_title_scan( hb_dvd_t * e, int t )
 134 {
 135
 136     hb_dvdread_t *d = &(e->dvdread);
 137     hb_title_t   * title;
 138     ifo_handle_t * vts = NULL;
 139     int            pgc_id, pgn, i;
 140     hb_chapter_t * chapter;
 141     int            c;
 142     uint64_t       duration;
 143     float          duration_correction;
 144     unsigned char  unused[1024];
 145
 146     hb_log( "scan: scanning title %d", t );
 147
 148     title = hb_title_init( d->path, t );
 149     if( DVDUDFVolumeInfo( d->reader, title->name, sizeof( title->name ),
 150                           unused, sizeof( unused ) ) )
 151     {
 152         char * p_cur, * p_last = d->path;
 153         for( p_cur = d->path; *p_cur; p_cur++ )
 154         {
 155             if( p_cur[0] == '/' && p_cur[1] )
 156             {
 157                 p_last = &p_cur[1];
 158             }
 159         }
 160         snprintf( title->name, sizeof( title->name ), "%s", p_last );
 161     }
 162
 163     /* VTS which our title is in */
 164     title->vts = d->vmg->tt_srpt->title[t-1].title_set_nr;
 165
 166     if ( !title->vts )
 167     {
 168         /* A VTS of 0 means the title wasn't found in the title set */
 169         hb_error("Invalid VTS (title set) number: %i", title->vts);
 170         goto fail;
 171     }
 172
 173     hb_log( "scan: opening IFO for VTS %d", title->vts );
 174     if( !( vts = ifoOpen( d->reader, title->vts ) ) )
 175     {
 176         hb_error( "scan: ifoOpen failed" );
 177         goto fail;
 178     }
 179
 180     /* ignore titles with bogus cell addresses so we don't abort later
 181      * in libdvdread. */
 182     for ( i = 0; i < vts->vts_c_adt->nr_of_vobs; ++i)
 183     {
 184         if( (vts->vts_c_adt->cell_adr_table[i].start_sector & 0xffffff ) ==
 185             0xffffff )
 186         {
 187             hb_error( "scan: cell_adr_table[%d].start_sector invalid (0x%x) "
 188                       "- skipping title", i,
 189                       vts->vts_c_adt->cell_adr_table[i].start_sector );
 190             goto fail;
 191         }
 192         if( (vts->vts_c_adt->cell_adr_table[i].last_sector & 0xffffff ) ==
 193             0xffffff )
 194         {
 195             hb_error( "scan: cell_adr_table[%d].last_sector invalid (0x%x) "
 196                       "- skipping title", i,
 197                       vts->vts_c_adt->cell_adr_table[i].last_sector );
 198             goto fail;
 199         }
 200         if( vts->vts_c_adt->cell_adr_table[i].start_sector >=
 201             vts->vts_c_adt->cell_adr_table[i].last_sector )
 202         {
 203             hb_error( "scan: cell_adr_table[%d].start_sector (0x%x) "
 204                       "is not before last_sector (0x%x) - skipping title", i,
 205                       vts->vts_c_adt->cell_adr_table[i].start_sector,
 206                       vts->vts_c_adt->cell_adr_table[i].last_sector );
 207             goto fail;
 208         }
 209     }
 210
 211     if( global_verbosity_level == 3 )
 212     {
 213         ifo_print( d->reader, title->vts );
 214     }
 215
 216     /* Position of the title in the VTS */
 217     title->ttn = d->vmg->tt_srpt->title[t-1].vts_ttn;
 218     if ( title->ttn < 1 || title->ttn > vts->vts_ptt_srpt->nr_of_srpts )
 219     {
 220         hb_error( "invalid VTS PTT offset %d for title %d, skipping", title->ttn, t );
 221         goto fail;
 222     }
 223
 224     /* Get pgc */
 225     pgc_id = vts->vts_ptt_srpt->title[title->ttn-1].ptt[0].pgcn;
 226     if ( pgc_id < 1 || pgc_id > vts->vts_pgcit->nr_of_pgci_srp )
 227     {
 228         hb_error( "invalid PGC ID %d for title %d, skipping", pgc_id, t );
 229         goto fail;
 230     }
 231     pgn    = vts->vts_ptt_srpt->title[title->ttn-1].ptt[0].pgn;
 232     d->pgc = vts->vts_pgcit->pgci_srp[pgc_id-1].pgc;
 233
 234     hb_log("pgc_id: %d, pgn: %d: pgc: 0x%x", pgc_id, pgn, d->pgc);
 235
 236     if( !d->pgc )
 237     {
 238         hb_error( "scan: pgc not valid, skipping" );
 239         goto fail;
 240     }
 241
 242     if( pgn <= 0 || pgn > 99 )
 243     {
 244         hb_error( "scan: pgn %d not valid, skipping", pgn );
 245         goto fail;
 246     }
 247
 248     /* Start cell */
 249     title->cell_start  = d->pgc->program_map[pgn-1] - 1;
 250     title->block_start = d->pgc->cell_playback[title->cell_start].first_sector;
 251
 252     /* End cell */
 253     title->cell_end  = d->pgc->nr_of_cells - 1;
 254     title->block_end = d->pgc->cell_playback[title->cell_end].last_sector;
 255
 256     /* Block count */
 257     title->block_count = 0;
 258     d->cell_cur = title->cell_start;
 259     while( d->cell_cur <= title->cell_end )
 260     {
 261 #define cp d->pgc->cell_playback[d->cell_cur]
 262         title->block_count += cp.last_sector + 1 - cp.first_sector;
 263 #undef cp
 264         FindNextCell( d );
 265         d->cell_cur = d->cell_next;
 266     }
 267
 268     hb_log( "scan: vts=%d, ttn=%d, cells=%d->%d, blocks=%d->%d, "
 269             "%d blocks", title->vts, title->ttn, title->cell_start,
 270             title->cell_end, title->block_start, title->block_end,
 271             title->block_count );
 272
 273     /* Get duration */
 274     title->duration = 90LL * dvdtime2msec( &d->pgc->playback_time );
 275     title->hours    = title->duration / 90000 / 3600;
 276     title->minutes  = ( ( title->duration / 90000 ) % 3600 ) / 60;
 277     title->seconds  = ( title->duration / 90000 ) % 60;
 278     hb_log( "scan: duration is %02d:%02d:%02d (%lld ms)",
 279             title->hours, title->minutes, title->seconds,
 280             title->duration / 90 );
 281
 282     /* ignore titles under 10 seconds because they're often stills or
 283      * clips with no audio & our preview code doesn't currently handle
 284      * either of these. */
 285     if( title->duration < 900000LL )
 286     {
 287         hb_log( "scan: ignoring title (too short)" );
 288         goto fail;
 289     }
 290
 291     /* Detect languages */
 292     for( i = 0; i < vts->vtsi_mat->nr_of_vts_audio_streams; i++ )
 293     {
 294         hb_audio_t * audio, * audio_tmp;
 295         int          audio_format, lang_code, audio_control,
 296                      position, j;
 297         iso639_lang_t * lang;
 298         int lang_extension = 0;
 299
 300         hb_log( "scan: checking audio %d", i + 1 );
 301
 302         audio = calloc( sizeof( hb_audio_t ), 1 );
 303
 304         audio_format  = vts->vtsi_mat->vts_audio_attr[i].audio_format;
 305         lang_code     = vts->vtsi_mat->vts_audio_attr[i].lang_code;
 306         lang_extension = vts->vtsi_mat->vts_audio_attr[i].code_extension;
 307         audio_control =
 308             vts->vts_pgcit->pgci_srp[pgc_id-1].pgc->audio_control[i];
 309
 310         if( !( audio_control & 0x8000 ) )
 311         {
 312             hb_log( "scan: audio channel is not active" );
 313             free( audio );
 314             continue;
 315         }
 316
 317         position = ( audio_control & 0x7F00 ) >> 8;
 318
 319         switch( audio_format )
 320         {
 321             case 0x00:
 322                 audio->id    = ( ( 0x80 + position ) << 8 ) | 0xbd;
 323                 audio->config.in.codec = HB_ACODEC_AC3;
 324                 break;
 325
 326             case 0x02:
 327             case 0x03:
 328                 audio->id    = 0xc0 + position;
 329                 audio->config.in.codec = HB_ACODEC_MPGA;
 330                 break;
 331
 332             case 0x04:
 333                 audio->id    = ( ( 0xa0 + position ) << 8 ) | 0xbd;
 334                 audio->config.in.codec = HB_ACODEC_LPCM;
 335                 break;
 336
 337             case 0x06:
 338                 audio->id    = ( ( 0x88 + position ) << 8 ) | 0xbd;
 339                 audio->config.in.codec = HB_ACODEC_DCA;
 340                 break;
 341
 342             default:
 343                 audio->id    = 0;
 344                 audio->config.in.codec = 0;
 345                 hb_log( "scan: unknown audio codec (%x)",
 346                         audio_format );
 347                 break;
 348         }
 349         if( !audio->id )
 350         {
 351             continue;
 352         }
 353
 354         /* Check for duplicate tracks */
 355         audio_tmp = NULL;
 356         for( j = 0; j < hb_list_count( title->list_audio ); j++ )
 357         {
 358             audio_tmp = hb_list_item( title->list_audio, j );
 359             if( audio->id == audio_tmp->id )
 360             {
 361                 break;
 362             }
 363             audio_tmp = NULL;
 364         }
 365         if( audio_tmp )
 366         {
 367             hb_log( "scan: duplicate audio track" );
 368             free( audio );
 369             continue;
 370         }
 371
 372         audio->config.lang.type = lang_extension;
 373
 374         lang = lang_for_code( vts->vtsi_mat->vts_audio_attr[i].lang_code );
 375
 376         snprintf( audio->config.lang.description, sizeof( audio->config.lang.description ), "%s (%s)",
 377             strlen(lang->native_name) ? lang->native_name : lang->eng_name,
 378             audio->config.in.codec == HB_ACODEC_AC3 ? "AC3" : ( audio->config.in.codec ==
 379                 HB_ACODEC_DCA ? "DTS" : ( audio->config.in.codec ==
 380                 HB_ACODEC_MPGA ? "MPEG" : "LPCM" ) ) );
 381         snprintf( audio->config.lang.simple, sizeof( audio->config.lang.simple ), "%s",
 382                   strlen(lang->native_name) ? lang->native_name : lang->eng_name );
 383         snprintf( audio->config.lang.iso639_2, sizeof( audio->config.lang.iso639_2 ), "%s",
 384                   lang->iso639_2);
 385
 386         switch( lang_extension )
 387         {
 388         case 0:
 389         case 1:
 390             break;
 391         case 2:
 392             strcat( audio->config.lang.description, " (Visually Impaired)" );
 393             break;
 394         case 3:
 395             strcat( audio->config.lang.description, " (Director's Commentary 1)" );
 396             break;
 397         case 4:
 398             strcat( audio->config.lang.description, " (Director's Commentary 2)" );
 399             break;
 400         default:
 401             break;
 402         }
 403
 404         hb_log( "scan: id=%x, lang=%s, 3cc=%s ext=%i", audio->id,
 405                 audio->config.lang.description, audio->config.lang.iso639_2,
 406                 lang_extension );
 407
 408         audio->config.in.track = i;
 409         hb_list_add( title->list_audio, audio );
 410     }
 411
 412     if( !hb_list_count( title->list_audio ) )
 413     {
 414         hb_log( "scan: ignoring title (no audio track)" );
 415         goto fail;
 416     }
 417
 418     memcpy( title->palette,
 419             vts->vts_pgcit->pgci_srp[pgc_id-1].pgc->palette,
 420             16 * sizeof( uint32_t ) );
 421
 422     /* Check for subtitles */
 423     for( i = 0; i < vts->vtsi_mat->nr_of_vts_subp_streams; i++ )
 424     {
 425         hb_subtitle_t * subtitle;
 426         int spu_control;
 427         int position;
 428         iso639_lang_t * lang;
 429         int lang_extension = 0;
 430
 431         hb_log( "scan: checking subtitle %d", i + 1 );
 432
 433         spu_control =
 434             vts->vts_pgcit->pgci_srp[pgc_id-1].pgc->subp_control[i];
 435
 436         if( !( spu_control & 0x80000000 ) )
 437         {
 438             hb_log( "scan: subtitle channel is not active" );
 439             continue;
 440         }
 441
 442         if( vts->vtsi_mat->vts_video_attr.display_aspect_ratio )
 443         {
 444             switch( vts->vtsi_mat->vts_video_attr.permitted_df )
 445             {
 446                 case 1:
 447                     position = spu_control & 0xFF;
 448                     break;
 449                 case 2:
 450                     position = ( spu_control >> 8 ) & 0xFF;
 451                     break;
 452                 default:
 453                     position = ( spu_control >> 16 ) & 0xFF;
 454             }
 455         }
 456         else
 457         {
 458             position = ( spu_control >> 24 ) & 0x7F;
 459         }
 460
 461         lang_extension = vts->vtsi_mat->vts_subp_attr[i].code_extension;
 462
 463         lang = lang_for_code( vts->vtsi_mat->vts_subp_attr[i].lang_code );
 464
 465         subtitle = calloc( sizeof( hb_subtitle_t ), 1 );
 466         subtitle->track = i+1;
 467         subtitle->id = ( ( 0x20 + position ) << 8 ) | 0xbd;
 468         snprintf( subtitle->lang, sizeof( subtitle->lang ), "%s",
 469              strlen(lang->native_name) ? lang->native_name : lang->eng_name);
 470         snprintf( subtitle->iso639_2, sizeof( subtitle->iso639_2 ), "%s",
 471                   lang->iso639_2);
 472         subtitle->format = PICTURESUB;
 473         subtitle->source = VOBSUB;
 474         subtitle->config.dest   = RENDERSUB;  // By default render (burn-in) the VOBSUB.
 475
 476         subtitle->type = lang_extension;
 477
 478         switch( lang_extension )
 479         {
 480         case 0:
 481             break;
 482         case 1:
 483             break;
 484         case 2:
 485             strcat( subtitle->lang, " (Caption with bigger size character)");
 486             break;
 487         case 3:
 488             strcat( subtitle->lang, " (Caption for Children)");
 489             break;
 490         case 4:
 491             break;
 492         case 5:
 493             strcat( subtitle->lang, " (Closed Caption)");
 494             break;
 495         case 6:
 496             strcat( subtitle->lang, " (Closed Caption with bigger size character)");
 497             break;
 498         case 7:
 499             strcat( subtitle->lang, " (Closed Caption for Children)");
 500             break;
 501         case 8:
 502             break;
 503         case 9:
 504             strcat( subtitle->lang, " (Forced Caption)");
 505             break;
 506         case 10:
 507             break;
 508         case 11:
 509             break;
 510         case 12:
 511             break;
 512         case 13:
 513             strcat( subtitle->lang, " (Director's Commentary)");
 514             break;
 515         case 14:
 516             strcat( subtitle->lang, " (Director's Commentary with bigger size character)");
 517             break;
 518         case 15:
 519             strcat( subtitle->lang, " (Director's Commentary for Children)");
 520         default:
 521             break;
 522         }
 523
 524         hb_log( "scan: id=%x, lang=%s, 3cc=%s", subtitle->id,
 525                 subtitle->lang, subtitle->iso639_2 );
 526
 527         hb_list_add( title->list_subtitle, subtitle );
 528     }
 529
 530     /* Chapters */
 531     hb_log( "scan: title %d has %d chapters", t,
 532             vts->vts_ptt_srpt->title[title->ttn-1].nr_of_ptts );
 533     for( i = 0, c = 1;
 534          i < vts->vts_ptt_srpt->title[title->ttn-1].nr_of_ptts; i++ )
 535     {
 536         chapter = calloc( sizeof( hb_chapter_t ), 1 );
 537         /* remember the on-disc chapter number */
 538         chapter->index = i + 1;
 539
 540         pgc_id = vts->vts_ptt_srpt->title[title->ttn-1].ptt[i].pgcn;
 541         pgn    = vts->vts_ptt_srpt->title[title->ttn-1].ptt[i].pgn;
 542         d->pgc = vts->vts_pgcit->pgci_srp[pgc_id-1].pgc;
 543
 544         /* Start cell */
 545         chapter->cell_start  = d->pgc->program_map[pgn-1] - 1;
 546         chapter->block_start =
 547             d->pgc->cell_playback[chapter->cell_start].first_sector;
 548
 549         // if there are no more programs in this pgc, the end cell is the
 550         // last cell. Otherwise it's the cell before the start cell of the
 551         // next program.
 552         if ( pgn == d->pgc->nr_of_programs )
 553         {
 554             chapter->cell_end = d->pgc->nr_of_cells - 1;
 555         }
 556         else
 557         {
 558             chapter->cell_end = d->pgc->program_map[pgn] - 2;;
 559         }
 560         chapter->block_end = d->pgc->cell_playback[chapter->cell_end].last_sector;
 561
 562         /* Block count, duration */
 563         chapter->block_count = 0;
 564         chapter->duration = 0;
 565
 566         d->cell_cur = chapter->cell_start;
 567         while( d->cell_cur <= chapter->cell_end )
 568         {
 569 #define cp d->pgc->cell_playback[d->cell_cur]
 570             chapter->block_count += cp.last_sector + 1 - cp.first_sector;
 571             chapter->duration += 90LL * dvdtime2msec( &cp.playback_time );
 572 #undef cp
 573             FindNextCell( d );
 574             d->cell_cur = d->cell_next;
 575         }
 576         hb_list_add( title->list_chapter, chapter );
 577         c++;
 578     }
 579
 580     /* The durations we get for chapters aren't precise. Scale them so
 581        the total matches the title duration */
 582     duration = 0;
 583     for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
 584     {
 585         chapter = hb_list_item( title->list_chapter, i );
 586         duration += chapter->duration;
 587     }
 588     duration_correction = (float) title->duration / (float) duration;
 589     for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
 590     {
 591         int seconds;
 592         chapter            = hb_list_item( title->list_chapter, i );
 593         chapter->duration  = duration_correction * chapter->duration;
 594         seconds            = ( chapter->duration + 45000 ) / 90000;
 595         chapter->hours     = seconds / 3600;
 596         chapter->minutes   = ( seconds % 3600 ) / 60;
 597         chapter->seconds   = seconds % 60;
 598
 599         hb_log( "scan: chap %d c=%d->%d, b=%d->%d (%d), %lld ms",
 600                 chapter->index, chapter->cell_start, chapter->cell_end,
 601                 chapter->block_start, chapter->block_end,
 602                 chapter->block_count, chapter->duration / 90 );
 603     }
 604
 605     /* Get aspect. We don't get width/height/rate infos here as
 606        they tend to be wrong */
 607     switch( vts->vtsi_mat->vts_video_attr.display_aspect_ratio )
 608     {
 609         case 0:
 610             title->container_aspect = 4. / 3.;
 611             break;
 612         case 3:
 613             title->container_aspect = 16. / 9.;
 614             break;
 615         default:
 616             hb_log( "scan: unknown aspect" );
 617             goto fail;
 618     }
 619
 620     hb_log( "scan: aspect = %g", title->aspect );
 621
 622     /* This title is ok so far */
 623     goto cleanup;
 624
 625 fail:
 626     hb_list_close( &title->list_audio );
 627     free( title );
 628     title = NULL;
 629
 630 cleanup:
 631     if( vts ) ifoClose( vts );
 632
 633     return title;
 634 }
 635
 636 /***********************************************************************
 637  * hb_dvdread_start
 638  ***********************************************************************
 639  * Title and chapter start at 1
 640  **********************************************************************/
 641 static int hb_dvdread_start( hb_dvd_t * e, hb_title_t *title, int chapter )
 642 {
 643     hb_dvdread_t *d = &(e->dvdread);
 644     int pgc_id, pgn;
 645     int i;
 646     int t = title->index;
 647
 648     /* Open the IFO and the VOBs for this title */
 649     d->vts = d->vmg->tt_srpt->title[t-1].title_set_nr;
 650     d->ttn = d->vmg->tt_srpt->title[t-1].vts_ttn;
 651     if( !( d->ifo = ifoOpen( d->reader, d->vts ) ) )
 652     {
 653         hb_error( "dvd: ifoOpen failed for VTS %d", d->vts );
 654         return 0;
 655     }
 656     if( !( d->file = DVDOpenFile( d->reader, d->vts,
 657                                   DVD_READ_TITLE_VOBS ) ) )
 658     {
 659         hb_error( "dvd: DVDOpenFile failed for VTS %d", d->vts );
 660         return 0;
 661     }
 662
 663     /* Get title first and last blocks */
 664     pgc_id         = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[0].pgcn;
 665     pgn            = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[0].pgn;
 666     d->pgc         = d->ifo->vts_pgcit->pgci_srp[pgc_id-1].pgc;
 667     d->cell_start  = d->pgc->program_map[pgn - 1] - 1;
 668     d->cell_end    = d->pgc->nr_of_cells - 1;
 669     d->title_start = d->pgc->cell_playback[d->cell_start].first_sector;
 670     d->title_end   = d->pgc->cell_playback[d->cell_end].last_sector;
 671
 672     /* Block count */
 673     d->title_block_count = 0;
 674     for( i = d->cell_start; i <= d->cell_end; i++ )
 675     {
 676         d->title_block_count += d->pgc->cell_playback[i].last_sector + 1 -
 677             d->pgc->cell_playback[i].first_sector;
 678     }
 679
 680     /* Get pgc for the current chapter */
 681     pgc_id = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[chapter-1].pgcn;
 682     pgn    = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[chapter-1].pgn;
 683     d->pgc = d->ifo->vts_pgcit->pgci_srp[pgc_id-1].pgc;
 684
 685     /* Get the two first cells */
 686     d->cell_cur = d->pgc->program_map[pgn-1] - 1;
 687     FindNextCell( d );
 688
 689     d->block     = d->pgc->cell_playback[d->cell_cur].first_sector;
 690     d->next_vobu = d->block;
 691     d->pack_len  = 0;
 692     d->cell_overlap = 0;
 693     d->in_cell = 0;
 694     d->in_sync = 2;
 695
 696     return 1;
 697 }
 698
 699 /***********************************************************************
 700  * hb_dvdread_stop
 701  ***********************************************************************
 702  *
 703  **********************************************************************/
 704 static void hb_dvdread_stop( hb_dvd_t * e )
 705 {
 706     hb_dvdread_t *d = &(e->dvdread);
 707     if( d->ifo )
 708     {
 709         ifoClose( d->ifo );
 710         d->ifo = NULL;
 711     }
 712     if( d->file )
 713     {
 714         DVDCloseFile( d->file );
 715         d->file = NULL;
 716     }
 717 }
 718
 719 /***********************************************************************
 720  * hb_dvdread_seek
 721  ***********************************************************************
 722  *
 723  **********************************************************************/
 724 static int hb_dvdread_seek( hb_dvd_t * e, float f )
 725 {
 726     hb_dvdread_t *d = &(e->dvdread);
 727     int count, sizeCell;
 728     int i;
 729
 730     count = f * d->title_block_count;
 731
 732     for( i = d->cell_start; i <= d->cell_end; i++ )
 733     {
 734         sizeCell = d->pgc->cell_playback[i].last_sector + 1 -
 735             d->pgc->cell_playback[i].first_sector;
 736
 737         if( count < sizeCell )
 738         {
 739             d->cell_cur = i;
 740             d->cur_cell_id = 0;
 741             FindNextCell( d );
 742
 743             /* Now let hb_dvdread_read find the next VOBU */
 744             d->next_vobu = d->pgc->cell_playback[i].first_sector + count;
 745             d->pack_len  = 0;
 746             break;
 747         }
 748
 749         count -= sizeCell;
 750     }
 751
 752     if( i > d->cell_end )
 753     {
 754         return 0;
 755     }
 756
 757     /*
 758      * Assume that we are in sync, even if we are not given that it is obvious
 759      * that we are out of sync if we have just done a seek.
 760      */
 761     d->in_sync = 2;
 762
 763     return 1;
 764 }
 765
 766
 767 /***********************************************************************
 768  * is_nav_pack
 769  ***********************************************************************/
 770 int is_nav_pack( unsigned char *buf )
 771 {
 772     /*
 773      * The NAV Pack is comprised of the PCI Packet and DSI Packet, both
 774      * of these start at known offsets and start with a special identifier.
 775      *
 776      * NAV = {
 777      *  PCI = { 00 00 01 bf  # private stream header
 778      *          ?? ??        # length
 779      *          00           # substream
 780      *          ...
 781      *        }
 782      *  DSI = { 00 00 01 bf  # private stream header
 783      *          ?? ??        # length
 784      *          01           # substream
 785      *          ...
 786      *        }
 787      *
 788      * The PCI starts at offset 0x26 into the sector, and the DSI starts at 0x400
 789      *
 790      * This information from: http://dvd.sourceforge.net/dvdinfo/
 791      */
 792     if( ( buf[0x26] == 0x00 &&      // PCI
 793           buf[0x27] == 0x00 &&
 794           buf[0x28] == 0x01 &&
 795           buf[0x29] == 0xbf &&
 796           buf[0x2c] == 0x00 ) &&
 797         ( buf[0x400] == 0x00 &&     // DSI
 798           buf[0x401] == 0x00 &&
 799           buf[0x402] == 0x01 &&
 800           buf[0x403] == 0xbf &&
 801           buf[0x406] == 0x01 ) )
 802     {
 803         return ( 1 );
 804     } else {
 805         return ( 0 );
 806     }
 807 }
 808
 809
 810 /***********************************************************************
 811  * hb_dvdread_read
 812  ***********************************************************************
 813  *
 814  **********************************************************************/
 815 static int hb_dvdread_read( hb_dvd_t * e, hb_buffer_t * b )
 816 {
 817     hb_dvdread_t *d = &(e->dvdread);
 818  top:
 819     if( !d->pack_len )
 820     {
 821         /* New pack */
 822         dsi_t dsi_pack;
 823         int   error = 0;
 824
 825         // if we've just finished the last cell of the title we don't
 826         // want to read another block because our next_vobu pointer
 827         // is probably invalid. Just return 'no data' & our caller
 828         // should check and discover we're at eof.
 829         if ( d->cell_cur > d->cell_end )
 830             return 0;
 831
 832         for( ;; )
 833         {
 834             int block, pack_len, next_vobu, read_retry;
 835
 836             for( read_retry = 0; read_retry < 3; read_retry++ )
 837             {
 838                 if( DVDReadBlocks( d->file, d->next_vobu, 1, b->data ) == 1 )
 839                 {
 840                     /*
 841                      * Successful read.
 842                      */
 843                     break;
 844                 } else {
 845                     hb_log( "dvd: Read Error on blk %d, attempt %d",
 846                             d->next_vobu, read_retry );
 847                 }
 848             }
 849
 850             if( read_retry == 3 )
 851             {
 852                 hb_log( "dvd: vobu read error blk %d - skipping to cell %d",
 853                         d->next_vobu, d->cell_next );
 854                 d->cell_cur  = d->cell_next;
 855                 if ( d->cell_cur > d->cell_end )
 856                     return 0;
 857                 d->in_cell = 0;
 858                 d->next_vobu = d->pgc->cell_playback[d->cell_cur].first_sector;
 859                 FindNextCell( d );
 860                 d->cell_overlap = 1;
 861                 continue;
 862             }
 863
 864             if ( !is_nav_pack( b->data ) ) {
 865                 (d->next_vobu)++;
 866                 if( d->in_sync == 1 ) {
 867                     hb_log("dvd: Lost sync, searching for NAV pack at blk %d",
 868                            d->next_vobu);
 869                     d->in_sync = 0;
 870                 }
 871                 continue;
 872             }
 873
 874             navRead_DSI( &dsi_pack, &b->data[DSI_START_BYTE] );
 875
 876             if ( d->in_sync == 0 && d->cur_cell_id &&
 877                  (d->cur_vob_id != dsi_pack.dsi_gi.vobu_vob_idn ||
 878                   d->cur_cell_id != dsi_pack.dsi_gi.vobu_c_idn ) )
 879             {
 880                 // We walked out of the cell we're supposed to be in.
 881                 // If we're not at the start of our next cell go there.
 882                 hb_log("dvd: left cell %d (%u,%u) for (%u,%u) at block %u",
 883                        d->cell_cur, d->cur_vob_id, d->cur_cell_id,
 884                        dsi_pack.dsi_gi.vobu_vob_idn, dsi_pack.dsi_gi.vobu_c_idn,
 885                        d->next_vobu );
 886                 if ( d->next_vobu != d->pgc->cell_playback[d->cell_next].first_sector )
 887                 {
 888                     d->next_vobu = d->pgc->cell_playback[d->cell_next].first_sector;
 889                     d->cur_cell_id = 0;
 890                     continue;
 891                 }
 892             }
 893
 894             block     = dsi_pack.dsi_gi.nv_pck_lbn;
 895             pack_len  = dsi_pack.dsi_gi.vobu_ea;
 896
 897             // There are a total of 21 next vobu offsets (and 21 prev_vobu
 898             // offsets) in the navpack SRI structure. The primary one is
 899             // 'next_vobu' which is the offset (in dvd blocks) from the current
 900             // block to the start of the next vobu. If the block at 'next_vobu'
 901             // can't be read, 'next_video' is the offset to the vobu closest to it.
 902             // The other 19 offsets are vobus at successively longer distances from
 903             // the current block (this is so that a hardware player can do
 904             // adaptive error recovery to skip over a bad spot on the disk). In all
 905             // these offsets the high bit is set to indicate when it contains a
 906             // valid offset. The next bit (2^30) is set to indicate that there's
 907             // another valid offset in the SRI that's closer to the current block.
 908             // A hardware player uses these bits to chose the closest valid offset
 909             // and uses that as its next vobu. (Some mastering schemes appear to
 910             // put a bogus offset in next_vobu with the 2^30 bit set & the
 911             // correct offset in next_video. This works fine in hardware players
 912             // but will mess up software that doesn't implement the full next
 913             // vobu decision logic.) In addition to the flag bits there's a
 914             // reserved value of the offset that indicates 'no next vobu' (which
 915             // indicates the end of a cell). But having no next vobu pointer with a
 916             // 'valid' bit set also indicates end of cell. Different mastering
 917             // schemes seem to use all possible combinations of the flag bits
 918             // and reserved values to indicate end of cell so we have to check
 919             // them all or we'll get a disk read error from following an
 920             // invalid offset.
 921             uint32_t next_ptr = dsi_pack.vobu_sri.next_vobu;
 922             if ( ( next_ptr & ( 1 << 31 ) ) == 0  ||
 923                  ( next_ptr & ( 1 << 30 ) ) != 0 ||
 924                  ( next_ptr & 0x3fffffff ) == 0x3fffffff )
 925             {
 926                 next_ptr = dsi_pack.vobu_sri.next_video;
 927                 if ( ( next_ptr & ( 1 << 31 ) ) == 0 ||
 928                      ( next_ptr & 0x3fffffff ) == 0x3fffffff )
 929                 {
 930                     // don't have a valid forward pointer - assume end-of-cell
 931                     d->block     = block;
 932                     d->pack_len  = pack_len;
 933                     break;
 934                 }
 935             }
 936             next_vobu = block + ( next_ptr & 0x3fffffff );
 937
 938             if( pack_len >  0    &&
 939                 pack_len <  1024 &&
 940                 block    >= d->next_vobu &&
 941                 ( block <= d->title_start + d->title_block_count ||
 942                   block <= d->title_end ) )
 943             {
 944                 d->block     = block;
 945                 d->pack_len  = pack_len;
 946                 d->next_vobu = next_vobu;
 947                 break;
 948             }
 949
 950             /* Wasn't a valid VOBU, try next block */
 951             if( ++error > 1024 )
 952             {
 953                 hb_log( "dvd: couldn't find a VOBU after 1024 blocks" );
 954                 return 0;
 955             }
 956
 957             (d->next_vobu)++;
 958         }
 959
 960         if( d->in_sync == 0 || d->in_sync == 2 )
 961         {
 962             if( d->in_sync == 0 )
 963             {
 964                 hb_log( "dvd: In sync with DVD at block %d", d->block );
 965             }
 966             d->in_sync = 1;
 967         }
 968
 969         // Revert the cell overlap, and check for a chapter break
 970         // If this cell is zero length (prev_vobu & next_vobu both
 971         // set to END_OF_CELL) we need to check for beginning of
 972         // cell before checking for end or we'll advance to the next
 973         // cell too early and fail to generate a chapter mark when this
 974         // cell starts a chapter.
 975         if( ( dsi_pack.vobu_sri.prev_vobu & (1 << 31 ) ) == 0 ||
 976             ( dsi_pack.vobu_sri.prev_vobu & 0x3fffffff ) == 0x3fffffff )
 977         {
 978             // A vobu that's not at the start of a cell can have an
 979             // EOC prev pointer (this seems to be associated with some
 980             // sort of drm). The rest of the content in the cell may be
 981             // booby-trapped so treat this like an end-of-cell rather
 982             // than a beginning of cell.
 983             if ( d->pgc->cell_playback[d->cell_cur].first_sector < dsi_pack.dsi_gi.nv_pck_lbn &&
 984                  d->pgc->cell_playback[d->cell_cur].last_sector >= dsi_pack.dsi_gi.nv_pck_lbn )
 985             {
 986                 hb_log( "dvd: null prev_vobu in cell %d at block %d", d->cell_cur,
 987                         d->block );
 988                 // treat like end-of-cell then go directly to start of next cell.
 989                 d->cell_cur  = d->cell_next;
 990                 d->in_cell = 0;
 991                 d->next_vobu = d->pgc->cell_playback[d->cell_cur].first_sector;
 992                 FindNextCell( d );
 993                 d->cell_overlap = 1;
 994                 goto top;
 995             }
 996             else
 997             {
 998                 if ( d->block != d->pgc->cell_playback[d->cell_cur].first_sector )
 999                 {
1000                     hb_log( "dvd: beginning of cell %d at block %d", d->cell_cur,
1001                            d->block );
1002                 }
1003                 if( d->in_cell )
1004                 {
1005                     hb_error( "dvd: assuming missed end of cell %d", d->cell_cur );
1006                     d->cell_cur  = d->cell_next;
1007                     d->next_vobu = d->pgc->cell_playback[d->cell_cur].first_sector;
1008                     FindNextCell( d );
1009                     d->cell_overlap = 1;
1010                     d->in_cell = 0;
1011                 } else {
1012                     d->in_cell = 1;
1013                 }
1014                 d->cur_vob_id = dsi_pack.dsi_gi.vobu_vob_idn;
1015                 d->cur_cell_id = dsi_pack.dsi_gi.vobu_c_idn;
1016
1017                 if( d->cell_overlap )
1018                 {
1019                     b->new_chap = hb_dvdread_is_break( d );
1020                     d->cell_overlap = 0;
1021                 }
1022             }
1023         }
1024
1025         if( ( dsi_pack.vobu_sri.next_vobu & (1 << 31 ) ) == 0 ||
1026             ( dsi_pack.vobu_sri.next_vobu & 0x3fffffff ) == 0x3fffffff )
1027         {
1028             if ( d->block <= d->pgc->cell_playback[d->cell_cur].first_sector ||
1029                  d->block > d->pgc->cell_playback[d->cell_cur].last_sector )
1030             {
1031                 hb_log( "dvd: end of cell %d at block %d", d->cell_cur,
1032                         d->block );
1033             }
1034             d->cell_cur  = d->cell_next;
1035             d->in_cell = 0;
1036             d->next_vobu = d->pgc->cell_playback[d->cell_cur].first_sector;
1037             FindNextCell( d );
1038             d->cell_overlap = 1;
1039
1040         }
1041     }
1042     else
1043     {
1044         if( DVDReadBlocks( d->file, d->block, 1, b->data ) != 1 )
1045         {
1046             // this may be a real DVD error or may be DRM. Either way
1047             // we don't want to quit because of one bad block so set
1048             // things up so we'll advance to the next vobu and recurse.
1049             hb_error( "dvd: DVDReadBlocks failed (%d), skipping to vobu %u",
1050                       d->block, d->next_vobu );
1051             d->pack_len = 0;
1052             goto top;  /* XXX need to restructure this routine & avoid goto */
1053         }
1054         d->pack_len--;
1055     }
1056
1057     d->block++;
1058
1059     return 1;
1060 }
1061
1062 /***********************************************************************
1063  * hb_dvdread_chapter
1064  ***********************************************************************
1065  * Returns in which chapter the next block to be read is.
1066  * Chapter numbers start at 1.
1067  **********************************************************************/
1068 static int hb_dvdread_chapter( hb_dvd_t * e )
1069 {
1070     hb_dvdread_t *d = &(e->dvdread);
1071     int     i;
1072     int     pgc_id, pgn;
1073     int     nr_of_ptts = d->ifo->vts_ptt_srpt->title[d->ttn-1].nr_of_ptts;
1074     pgc_t * pgc;
1075
1076     for( i = nr_of_ptts - 1;
1077          i >= 0;
1078          i-- )
1079     {
1080         /* Get pgc for chapter (i+1) */
1081         pgc_id = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[i].pgcn;
1082         pgn    = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[i].pgn;
1083         pgc    = d->ifo->vts_pgcit->pgci_srp[pgc_id-1].pgc;
1084
1085         if( d->cell_cur - d->cell_overlap >= pgc->program_map[pgn-1] - 1 &&
1086             d->cell_cur - d->cell_overlap <= pgc->nr_of_cells - 1 )
1087         {
1088             /* We are in this chapter */
1089             return i + 1;
1090         }
1091     }
1092
1093     /* End of title */
1094     return -1;
1095 }
1096
1097 /***********************************************************************
1098  * hb_dvdread_is_break
1099  ***********************************************************************
1100  * Returns chapter number if the current block is a new chapter start
1101  **********************************************************************/
1102 static int hb_dvdread_is_break( hb_dvdread_t * d )
1103 {
1104     int     i;
1105     int     pgc_id, pgn;
1106     int     nr_of_ptts = d->ifo->vts_ptt_srpt->title[d->ttn-1].nr_of_ptts;
1107     pgc_t * pgc;
1108     int     cell;
1109
1110     for( i = nr_of_ptts - 1;
1111          i > 0;
1112          i-- )
1113     {
1114         /* Get pgc for chapter (i+1) */
1115         pgc_id = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[i].pgcn;
1116         pgn    = d->ifo->vts_ptt_srpt->title[d->ttn-1].ptt[i].pgn;
1117         pgc    = d->ifo->vts_pgcit->pgci_srp[pgc_id-1].pgc;
1118         cell   = pgc->program_map[pgn-1] - 1;
1119
1120         if( cell <= d->cell_start )
1121             break;
1122
1123         // This must not match against the start cell.
1124         if( pgc->cell_playback[cell].first_sector == d->block && cell != d->cell_start )
1125         {
1126             return i + 1;
1127         }
1128     }
1129
1130     return 0;
1131 }
1132
1133 /***********************************************************************
1134  * hb_dvdread_close
1135  ***********************************************************************
1136  * Closes and frees everything
1137  **********************************************************************/
1138 static void hb_dvdread_close( hb_dvd_t ** _d )
1139 {
1140     hb_dvdread_t * d = &((*_d)->dvdread);
1141
1142     if( d->vmg )
1143     {
1144         ifoClose( d->vmg );
1145     }
1146     if( d->reader )
1147     {
1148         DVDClose( d->reader );
1149     }
1150
1151     free( d );
1152     *_d = NULL;
1153 }
1154
1155 /***********************************************************************
1156  * hb_dvdread_angle_count
1157  ***********************************************************************
1158  * Returns the number of angles supported.  We do not support angles
1159  * with dvdread
1160  **********************************************************************/
1161 static int hb_dvdread_angle_count( hb_dvd_t * d )
1162 {
1163     return 1;
1164 }
1165
1166 /***********************************************************************
1167  * hb_dvdread_set_angle
1168  ***********************************************************************
1169  * Sets the angle to read.  Not supported with dvdread
1170  **********************************************************************/
1171 static void hb_dvdread_set_angle( hb_dvd_t * d, int angle )
1172 {
1173 }
1174
1175 /***********************************************************************
1176  * FindNextCell
1177  ***********************************************************************
1178  * Assumes pgc and cell_cur are correctly set, and sets cell_next to the
1179  * cell to be read when we will be done with cell_cur.
1180  **********************************************************************/
1181 static void FindNextCell( hb_dvdread_t * d )
1182 {
1183     int i = 0;
1184
1185     if( d->pgc->cell_playback[d->cell_cur].block_type ==
1186             BLOCK_TYPE_ANGLE_BLOCK )
1187     {
1188
1189         while( d->pgc->cell_playback[d->cell_cur+i].block_mode !=
1190                    BLOCK_MODE_LAST_CELL )
1191         {
1192              i++;
1193         }
1194         d->cell_next = d->cell_cur + i + 1;
1195         hb_log( "dvd: Skipping multi-angle cells %d-%d",
1196                 d->cell_cur,
1197                 d->cell_next - 1 );
1198     }
1199     else
1200     {
1201         d->cell_next = d->cell_cur + 1;
1202     }
1203 }
1204
1205 /***********************************************************************
1206  * dvdtime2msec
1207  ***********************************************************************
1208  * From lsdvd
1209  **********************************************************************/
1210 static int dvdtime2msec(dvd_time_t * dt)
1211 {
1212     double frames_per_s[4] = {-1.0, 25.00, -1.0, 29.97};
1213     double fps = frames_per_s[(dt->frame_u & 0xc0) >> 6];
1214     long   ms;
1215     ms  = (((dt->hour &   0xf0) >> 3) * 5 + (dt->hour   & 0x0f)) * 3600000;
1216     ms += (((dt->minute & 0xf0) >> 3) * 5 + (dt->minute & 0x0f)) * 60000;
1217     ms += (((dt->second & 0xf0) >> 3) * 5 + (dt->second & 0x0f)) * 1000;
1218
1219     if( fps > 0 )
1220     {
1221         ms += ((dt->frame_u & 0x30) >> 3) * 5 +
1222               (dt->frame_u & 0x0f) * 1000.0 / fps;
1223     }
1224
1225     return ms;
1226 }
1227
1228 char * hb_dvd_name( char * path )
1229 {
1230     return dvd_methods->name(path);
1231 }
1232
1233 hb_dvd_t * hb_dvd_init( char * path )
1234 {
1235     return dvd_methods->init(path);
1236 }
1237
1238 int hb_dvd_title_count( hb_dvd_t * d )
1239 {
1240     return dvd_methods->title_count(d);
1241 }
1242
1243 hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
1244 {
1245     return dvd_methods->title_scan(d, t);
1246 }
1247
1248 int hb_dvd_start( hb_dvd_t * d, hb_title_t *title, int chapter )
1249 {
1250     return dvd_methods->start(d, title, chapter);
1251 }
1252
1253 void hb_dvd_stop( hb_dvd_t * d )
1254 {
1255     dvd_methods->stop(d);
1256 }
1257
1258 int hb_dvd_seek( hb_dvd_t * d, float f )
1259 {
1260     return dvd_methods->seek(d, f);
1261 }
1262
1263 int hb_dvd_read( hb_dvd_t * d, hb_buffer_t * b )
1264 {
1265     return dvd_methods->read(d, b);
1266 }
1267
1268 int hb_dvd_chapter( hb_dvd_t * d )
1269 {
1270     return dvd_methods->chapter(d);
1271 }
1272
1273 void hb_dvd_close( hb_dvd_t ** _d )
1274 {
1275     dvd_methods->close(_d);
1276 }
1277
1278 int hb_dvd_angle_count( hb_dvd_t * d )
1279 {
1280     return dvd_methods->angle_count(d);
1281 }
1282
1283 void hb_dvd_set_angle( hb_dvd_t * d, int angle )
1284 {
1285     dvd_methods->set_angle(d, angle);
1286 }
1287
1288 // hb_dvd_set_dvdnav must only be called when no dvd source is open
1289 // it rips the rug out from under things so be careful
1290 void hb_dvd_set_dvdnav( int enable )
1291 {
1292     if (enable)
1293         dvd_methods = hb_dvdnav_methods();
1294     else
1295         dvd_methods = hb_dvdread_methods();
1296 }
1297