OSDN Git Service

WinGui:
[handbrake-jp/handbrake-jp-git.git] / libhb / dvd.c
index faa82b2..94f020f 100644 (file)
@@ -30,9 +30,12 @@ struct hb_dvd_s
     int            title_block_count;
     int            cell_cur;
     int            cell_next;
+    int            cell_overlap;
     int            block;
     int            pack_len;
     int            next_vobu;
+    int            in_cell;
+    int            in_sync;
 };
 
 /***********************************************************************
@@ -78,14 +81,14 @@ hb_dvd_t * hb_dvd_init( char * path )
     /* Open device */
     if( !( d->reader = DVDOpen( path ) ) )
     {
-        hb_log( "dvd: DVDOpen failed (%s)", path );
+        hb_error( "dvd: DVDOpen failed (%s)", path );
         goto fail;
     }
 
     /* Open main IFO */
     if( !( d->vmg = ifoOpen( d->reader, 0 ) ) )
     {
-        hb_log( "dvd: ifoOpen failed" );
+        hb_error( "dvd: ifoOpen failed" );
         goto fail;
     }
 
@@ -143,10 +146,17 @@ hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
     /* VTS which our title is in */
     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_error("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" );
+        hb_error( "scan: ifoOpen failed" );
         goto fail;
     }
 
@@ -158,6 +168,14 @@ hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
     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: 0x%x", pgc_id, pgn, d->pgc);
+
+    if( !d->pgc )
+    {
+        hb_error( "scan: pgc not valid, skipping" );
+        goto fail;
+    }
     /* Start cell */
     title->cell_start  = d->pgc->program_map[pgn-1] - 1;
     title->block_start = d->pgc->cell_playback[title->cell_start].first_sector;
@@ -183,7 +201,7 @@ hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
             title->cell_end, title->block_start, title->block_end,
             title->block_count );
 
-    if( title->block_count < 2048 )
+    if( title->block_count < 2048  )
     {
         hb_log( "scan: title too short (%d blocks), ignoring",
                 title->block_count );
@@ -213,6 +231,7 @@ hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
         hb_audio_t * audio, * audio_tmp;
         int          audio_format, lang_code, audio_control,
                      position, j;
+        iso639_lang_t * lang;
 
         hb_log( "scan: checking audio %d", i + 1 );
 
@@ -250,6 +269,11 @@ hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
                 audio->codec = HB_ACODEC_LPCM;
                 break;
 
+            case 0x06:
+                audio->id    = ( ( 0x88 + position ) << 8 ) | 0xbd;
+                audio->codec = HB_ACODEC_DCA;
+                break;
+
             default:
                 audio->id    = 0;
                 audio->codec = 0;
@@ -280,15 +304,20 @@ hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
             continue;
         }
 
+        lang = lang_for_code( vts->vtsi_mat->vts_audio_attr[i].lang_code );
+
         snprintf( audio->lang, sizeof( audio->lang ), "%s (%s)",
-            lang_for_code( vts->vtsi_mat->vts_audio_attr[i].lang_code ),
+            strlen(lang->native_name) ? lang->native_name : lang->eng_name,
             audio->codec == HB_ACODEC_AC3 ? "AC3" : ( audio->codec ==
-                HB_ACODEC_MPGA ? "MPEG" : "LPCM" ) );
+                HB_ACODEC_DCA ? "DTS" : ( audio->codec ==
+                HB_ACODEC_MPGA ? "MPEG" : "LPCM" ) ) );
         snprintf( audio->lang_simple, sizeof( audio->lang_simple ), "%s",
-                  lang_for_code( vts->vtsi_mat->vts_audio_attr[i].lang_code ) );
+                  strlen(lang->native_name) ? lang->native_name : lang->eng_name );
+        snprintf( audio->iso639_2, sizeof( audio->iso639_2 ), "%s",
+                  lang->iso639_2);
 
-        hb_log( "scan: id=%x, lang=%s", audio->id,
-                audio->lang );
+        hb_log( "scan: id=%x, lang=%s, 3cc=%s", audio->id,
+                audio->lang, audio->iso639_2 );
 
         hb_list_add( title->list_audio, audio );
     }
@@ -309,6 +338,7 @@ hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
         hb_subtitle_t * subtitle;
         int spu_control;
         int position;
+        iso639_lang_t * lang;
 
         hb_log( "scan: checking subtitle %d", i + 1 );
 
@@ -340,13 +370,17 @@ hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
             position = ( spu_control >> 24 ) & 0x7F;
         }
 
+        lang = lang_for_code( vts->vtsi_mat->vts_subp_attr[i].lang_code );
+
         subtitle = calloc( sizeof( hb_subtitle_t ), 1 );
         subtitle->id = ( ( 0x20 + position ) << 8 ) | 0xbd;
         snprintf( subtitle->lang, sizeof( subtitle->lang ), "%s",
-            lang_for_code( vts->vtsi_mat->vts_subp_attr[i].lang_code ) );
+             strlen(lang->native_name) ? lang->native_name : lang->eng_name);
+       snprintf( subtitle->iso639_2, sizeof( subtitle->iso639_2 ), "%s",
+                 lang->iso639_2);
 
-        hb_log( "scan: id=%x, lang=%s", subtitle->id,
-                subtitle->lang );
+        hb_log( "scan: id=%x, lang=%s, 3cc=%s", subtitle->id,
+                subtitle->lang, subtitle->iso639_2 );
 
         hb_list_add( title->list_subtitle, subtitle );
     }
@@ -361,7 +395,8 @@ hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
         pgc_t * pgc_next;
 
         chapter = calloc( sizeof( hb_chapter_t ), 1 );
-        chapter->index = c;
+        /* remember the on-disc chapter number */
+        chapter->index = i + 1;
 
         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;
@@ -414,17 +449,17 @@ hb_title_t * hb_dvd_title_scan( hb_dvd_t * d, int t )
             d->cell_cur = d->cell_next;
         }
 
-        if( chapter->block_count < 2048 && chapter->index > 1 )
+        if( chapter->block_count < 2048 && c > 1 )
         {
-            hb_log( "scan: chapter %d too short (%d blocks, "
-                    "cells=%d->%d), merging", chapter->index,
+            hb_log( "scan: chapter %d(%d) too short (%d blocks, "
+                    "cells=%d->%d), merging", c, chapter->index,
                     chapter->block_count, chapter->cell_start,
                     chapter->cell_end );
-            chapter_old = hb_list_item( title->list_chapter,
-                                        chapter->index - 2 );
+            chapter_old = hb_list_item( title->list_chapter, c - 2 );
             chapter_old->cell_end    = chapter->cell_end;
             chapter_old->block_end   = chapter->block_end;
             chapter_old->block_count += chapter->block_count;
+            chapter_old->duration += chapter->duration;
             free( chapter );
             chapter = chapter_old;
         }
@@ -506,13 +541,13 @@ int hb_dvd_start( hb_dvd_t * d, int title, int chapter )
     d->ttn = d->vmg->tt_srpt->title[title-1].vts_ttn;
     if( !( d->ifo = ifoOpen( d->reader, d->vts ) ) )
     {
-        hb_log( "dvd: ifoOpen failed for VTS %d", 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_log( "dvd: DVDOpenFile failed for VTS %d", d->vts );
+        hb_error( "dvd: DVDOpenFile failed for VTS %d", d->vts );
         return 0;
     }
 
@@ -545,6 +580,9 @@ int hb_dvd_start( hb_dvd_t * d, int title, int chapter )
     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;
 }
@@ -604,9 +642,59 @@ int hb_dvd_seek( hb_dvd_t * d, float f )
         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_dvd_read
  ***********************************************************************
@@ -624,14 +712,38 @@ int hb_dvd_read( hb_dvd_t * d, hb_buffer_t * b )
         
         for( ;; )
         {
-            int block, pack_len, next_vobu;
+            int block, pack_len, next_vobu, read_retry;
 
-            if( DVDReadBlocks( d->file, d->next_vobu, 1, b->data ) != 1 )
+            for( read_retry = 0; read_retry < 3; read_retry++ )
             {
-                hb_log( "dvd: DVDReadBlocks failed (%d)", d->next_vobu );
+                if( DVDReadBlocks( d->file, d->next_vobu, 1, b->data ) == 1 )
+                {
+                    /*
+                     * Successful read.
+                     */
+                    break;
+                } else {
+                    hb_log( "dvd: Read Error on blk %d, attempt %d",
+                            d->next_vobu, read_retry );
+                }
+            }
+
+            if( read_retry == 3 )
+            {
+                hb_log( "dvd: Unrecoverable Read Error from DVD, potential HD or DVD Failure (blk: %d)", d->next_vobu );
                 return 0;
             }
 
+            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] );
             
             block     = dsi_pack.dsi_gi.nv_pck_lbn;
@@ -654,6 +766,7 @@ int hb_dvd_read( hb_dvd_t * d, hb_buffer_t * b )
                             d->next_vobu, block, pack_len, next_vobu );
 #endif
                 }
+
                 d->block     = block;
                 d->pack_len  = pack_len;
                 d->next_vobu = next_vobu;
@@ -677,18 +790,56 @@ int hb_dvd_read( hb_dvd_t * d, hb_buffer_t * b )
             (d->next_vobu)++;
         }
 
-        if( dsi_pack.vobu_sri.next_vobu == SRI_END_OF_CELL )
+        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;
+        }
+
+        if( dsi_pack.vobu_sri.next_vobu == SRI_END_OF_CELL )
+        { 
+            hb_log( "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;
+            
+        }
+
+        // Revert the cell overlap, and check for a chapter break
+        if( dsi_pack.vobu_sri.prev_vobu == SRI_END_OF_CELL )
+        {
+            hb_log( "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)", d->cell_cur );
+                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;
+            }
+
+            if( d->cell_overlap )
+            {
+                b->new_chap = hb_dvd_is_break( d );
+                d->cell_overlap = 0;
+            }
         }
     }
     else
     {
         if( DVDReadBlocks( d->file, d->block, 1, b->data ) != 1 )
         {
-            hb_log( "reader: DVDReadBlocks failed (%d)", d->block );
+            hb_error( "reader: DVDReadBlocks failed (%d)", d->block );
             return 0;
         }
         d->pack_len--;
@@ -709,19 +860,20 @@ int hb_dvd_chapter( hb_dvd_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;
 
-    for( i = 0;
-         i < d->ifo->vts_ptt_srpt->title[d->ttn-1].nr_of_ptts;
-         i++ )
+    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 >= pgc->program_map[pgn-1] - 1 &&
-            d->cell_cur <= pgc->nr_of_cells - 1 )
+        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;
@@ -733,6 +885,70 @@ int hb_dvd_chapter( hb_dvd_t * d )
 }
 
 /***********************************************************************
+ * hb_dvd_is_break
+ ***********************************************************************
+ * Returns 1 if the current block is a new chapter start
+ **********************************************************************/
+int hb_dvd_is_break( hb_dvd_t * d )
+{
+    int     i, j;
+    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, chapter_length, cell_end;
+    
+    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 && cell != d->cell_start )
+        {
+            /* Check to see if we merged this chapter into the previous one... */
+            /* As a note, merging chapters is probably bad practice for this very reason */
+            chapter_length = 0;
+            
+            if( i == nr_of_ptts - 1 )
+            {
+                cell_end = d->pgc->nr_of_cells;
+            }
+            else
+            {
+                cell_end = pgc->program_map[pgn] - 1;
+            }
+            
+            for( j = cell; j < cell_end; j++ )
+            {
+                chapter_length += pgc->cell_playback[j].last_sector + 1 - 
+                                  pgc->cell_playback[j].first_sector;
+            }
+            
+            if( chapter_length >= 2048 )
+            {
+                hb_log("DVD: Chapter Break Cell Found");
+                /* We have a chapter break */
+                return 1;
+            }
+            else
+            {
+                hb_log("DVD: Cell Found (len=%d)", chapter_length);
+            }
+        }
+    }
+    
+    return 0;
+}
+
+/***********************************************************************
  * hb_dvd_close
  ***********************************************************************
  * Closes and frees everything
@@ -774,6 +990,9 @@ static void FindNextCell( hb_dvd_t * d )
              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
     {