libmpdemux/demux_ty.c

   1 /*
   2  * tivo@wingert.org, February 2003
   3  *
   4  * Copyright (C) 2003 Christopher R. Wingert
   5  *
   6  * The license covers the portions of this file regarding TiVo additions.
   7  *
   8  * Olaf Beck and Tridge (indirectly) were essential at providing
   9  * information regarding the format of the TiVo streams.
  10  *
  11  * However, no code in the following subsection is directly copied from
  12  * either author.
  13  *
  14  * This file is part of MPlayer.
  15  *
  16  * MPlayer is free software; you can redistribute it and/or modify
  17  * it under the terms of the GNU General Public License as published by
  18  * the Free Software Foundation; either version 2 of the License, or
  19  * (at your option) any later version.
  20  *
  21  * MPlayer is distributed in the hope that it will be useful,
  22  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  23  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  24  * GNU General Public License for more details.
  25  *
  26  * You should have received a copy of the GNU General Public License along
  27  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  28  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  29  */
  30
  31
  32 #include <stdio.h>
  33 #include <stdlib.h>
  34 #include <unistd.h>
  35 #include <time.h>
  36 #include <stdarg.h>
  37
  38 #include "config.h"
  39 #include "mp_msg.h"
  40
  41 #include "libmpcodecs/dec_audio.h"
  42 #include "stream/stream.h"
  43 #include "demuxer.h"
  44 #include "demux_ty_osd.h"
  45 #include "parse_es.h"
  46 #include "stheader.h"
  47 #include "sub_cc.h"
  48 #include "libavutil/avstring.h"
  49 #include "ffmpeg_files/intreadwrite.h"
  50
  51 extern int sub_justify;
  52
  53 // 2/c0: audio data
  54 // 3/c0: audio packet header (PES header)
  55 // 4/c0: audio data (S/A only?)
  56 // 9/c0: audio packet header, AC-3 audio
  57 // 2/e0: video data
  58 // 6/e0: video packet header (PES header)
  59 // 7/e0: video sequence header start
  60 // 8/e0: video I-frame header start
  61 // a/e0: video P-frame header start
  62 // b/e0: video B-frame header start
  63 // c/e0: video GOP header start
  64 // e/01: closed-caption data
  65 // e/02: Extended data services data
  66
  67
  68 #define TIVO_PES_FILEID   0xf5467abd
  69 #define TIVO_PART_LENGTH  0x20000000
  70
  71 #define CHUNKSIZE        ( 128 * 1024 )
  72 #define MAX_AUDIO_BUFFER ( 16 * 1024 )
  73
  74 #define TY_V             1
  75 #define TY_A             2
  76
  77 typedef struct
  78 {
  79    off_t startOffset;
  80    off_t fileSize;
  81    int   chunks;
  82 } tmf_fileParts;
  83
  84 #define MAX_TMF_PARTS 16
  85
  86 typedef struct
  87 {
  88    int             whichChunk;
  89    unsigned char   chunk[ CHUNKSIZE ];
  90
  91    unsigned char   lastAudio[ MAX_AUDIO_BUFFER ];
  92    int             lastAudioEnd;
  93
  94    int             tivoType;           // 1 = SA, 2 = DTiVo
  95
  96    int64_t        lastAudioPTS;
  97    int64_t        lastVideoPTS;
  98
  99    off_t           size;
 100    int             readHeader;
 101
 102    int             tmf;
 103    tmf_fileParts   tmfparts[ MAX_TMF_PARTS ];
 104    int             tmf_totalparts;
 105 } TiVoInfo;
 106
 107 // ===========================================================================
 108 #define TMF_SIG "showing.xml"
 109
 110 // ===========================================================================
 111 static int ty_tmf_filetoparts( demuxer_t *demux, TiVoInfo *tivo )
 112 {
 113    int     parts = 0;
 114
 115    stream_seek(demux->stream, 0);
 116
 117    mp_msg( MSGT_DEMUX, MSGL_DBG3, "Dumping tar contents\n" );
 118    while (!demux->stream->eof)
 119    {
 120       char    header[ 512 ];
 121       char    *name;
 122       char    *extension;
 123       char    *sizestr;
 124       int     size;
 125       off_t   skip;
 126       if (stream_read(demux->stream, header, 512) < 512)
 127       {
 128          mp_msg( MSGT_DEMUX, MSGL_DBG3, "Read bad\n" );
 129          break;
 130       }
 131       name = header;
 132       name[99] = 0;
 133       sizestr = &header[124];
 134       sizestr[11] = 0;
 135       size = strtol(sizestr, NULL, 8);
 136
 137       mp_msg( MSGT_DEMUX, MSGL_DBG3, "name %-20.20s size %-12.12s %d\n",
 138          name, sizestr, size );
 139
 140       extension = strrchr(name, '.');
 141       if (extension && strcmp(extension, ".ty") == 0)
 142       {
 143          if ( parts >= MAX_TMF_PARTS ) {
 144             mp_msg( MSGT_DEMUX, MSGL_ERR, "ty:tmf too big\n" );
 145             break;
 146          }
 147          tivo->tmfparts[ parts ].fileSize = size;
 148          tivo->tmfparts[ parts ].startOffset = stream_tell(demux->stream);
 149          tivo->tmfparts[ parts ].chunks = size / CHUNKSIZE;
 150          mp_msg(MSGT_DEMUX, MSGL_DBG3,
 151            "tmf_filetoparts(): index %d, chunks %d\n"
 152            "tmf_filetoparts(): size %"PRId64"\n"
 153            "tmf_filetoparts(): startOffset %"PRId64"\n",
 154            parts, tivo->tmfparts[ parts ].chunks,
 155            tivo->tmfparts[ parts ].fileSize, tivo->tmfparts[ parts ].startOffset
 156          );
 157          parts++;
 158       }
 159
 160       // size rounded up to blocks
 161       skip = (size + 511) & ~511;
 162       stream_skip(demux->stream, skip);
 163    }
 164    stream_reset(demux->stream);
 165    tivo->tmf_totalparts = parts;
 166    mp_msg( MSGT_DEMUX, MSGL_DBG3,
 167       "tmf_filetoparts(): No More Part Files %d\n", parts );
 168
 169    return 1;
 170 }
 171
 172
 173 // ===========================================================================
 174 static off_t tmf_filetooffset(TiVoInfo *tivo, int chunk)
 175 {
 176   int i;
 177   for (i = 0; i < tivo->tmf_totalparts; i++) {
 178     if (chunk < tivo->tmfparts[i].chunks)
 179       return tivo->tmfparts[i].startOffset + chunk * CHUNKSIZE;
 180     chunk -= tivo->tmfparts[i].chunks;
 181   }
 182   return -1;
 183 }
 184
 185
 186 // ===========================================================================
 187 static int tmf_load_chunk( demuxer_t *demux, TiVoInfo *tivo,
 188    unsigned char *buff, int readChunk )
 189 {
 190    off_t fileoffset;
 191    int    count;
 192
 193    mp_msg( MSGT_DEMUX, MSGL_DBG3, "\ntmf_load_chunk() begin %d\n",
 194       readChunk );
 195
 196    fileoffset = tmf_filetooffset(tivo, readChunk);
 197
 198    if (fileoffset == -1 || !stream_seek(demux->stream, fileoffset)) {
 199       mp_msg( MSGT_DEMUX, MSGL_ERR, "Read past EOF()\n" );
 200       return 0;
 201    }
 202    count = stream_read( demux->stream, buff, CHUNKSIZE );
 203    demux->filepos = stream_tell( demux->stream );
 204
 205    mp_msg( MSGT_DEMUX, MSGL_DBG3, "tmf_load_chunk() count %x\n",
 206            count );
 207
 208    mp_msg( MSGT_DEMUX, MSGL_DBG3,
 209            "tmf_load_chunk() bytes %x %x %x %x %x %x %x %x\n",
 210            buff[ 0 ], buff[ 1 ], buff[ 2 ], buff[ 3 ],
 211            buff[ 4 ], buff[ 5 ], buff[ 6 ], buff[ 7 ] );
 212
 213    mp_msg( MSGT_DEMUX, MSGL_DBG3, "tmf_load_chunk() end\n" );
 214
 215    return count;
 216 }
 217
 218 // ===========================================================================
 219
 220 // DTiVo MPEG 336, 480, 576, 768
 221 // SA TiVo 864
 222 // DTiVo AC-3 1550
 223 //
 224 #define SERIES1_PTS_LENGTH           11
 225 #define SERIES1_PTS_OFFSET            6
 226 #define SERIES2_PTS_LENGTH           16
 227 #define SERIES2_PTS_OFFSET            9
 228 #define AC3_PTS_LENGTH               16
 229 #define AC3_PTS_OFFSET                9
 230
 231 static int IsValidAudioPacket( int size, int *ptsOffset, int *ptsLen )
 232 {
 233    // AC-3
 234    if ( size == 1550 || size == 1552 )
 235    {
 236       *ptsOffset = AC3_PTS_OFFSET;
 237       *ptsLen = AC3_PTS_LENGTH;
 238       return 1;
 239    }
 240
 241    // MPEG
 242       if ( (size & 15) == (SERIES1_PTS_LENGTH & 15) )
 243       {
 244          *ptsOffset = SERIES1_PTS_OFFSET;
 245          *ptsLen = SERIES1_PTS_LENGTH;
 246          return 1;
 247       }
 248          if ( (size & 15) == (SERIES2_PTS_LENGTH & 15) )
 249          {
 250             *ptsOffset = SERIES2_PTS_OFFSET;
 251             *ptsLen = SERIES2_PTS_LENGTH;
 252             return 1;
 253          }
 254       mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:Tossing Audio Packet Size %d\n",
 255          size );
 256       return 0;
 257 }
 258
 259
 260 static int64_t get_ty_pts( unsigned char *buf )
 261 {
 262   int a = buf[0] & 0xe;
 263   int b = AV_RB16(buf + 1);
 264   int c = AV_RB16(buf + 3);
 265
 266   if (!(1 & a & b & c)) // invalid MPEG timestamp
 267     return MP_NOPTS_VALUE;
 268   a >>= 1; b >>= 1; c >>= 1;
 269   return (((uint64_t)a) << 30) | (b << 15) | c;
 270 }
 271
 272 static void demux_ty_AddToAudioBuffer( TiVoInfo *tivo, unsigned char *buffer,
 273    int size )
 274 {
 275    if ( tivo->lastAudioEnd + size < MAX_AUDIO_BUFFER )
 276    {
 277       memcpy( &tivo->lastAudio[ tivo->lastAudioEnd ],
 278          buffer, size );
 279       tivo->lastAudioEnd += size;
 280    }
 281    else
 282       mp_msg( MSGT_DEMUX, MSGL_ERR,
 283          "ty:WARNING - Would have blown my audio buffer\n" );
 284 }
 285
 286 static void demux_ty_CopyToDemuxPacket( demux_stream_t *ds,
 287        unsigned char *buffer, int size, off_t pos, int64_t pts )
 288 {
 289    demux_packet_t *dp = new_demux_packet( size );
 290    memcpy( dp->buffer, buffer, size );
 291    if (pts != MP_NOPTS_VALUE)
 292    dp->pts = pts / 90000.0;
 293    dp->pos = pos;
 294    dp->flags = 0;
 295    ds_add_packet( ds, dp );
 296 }
 297
 298 static int demux_ty_FindESHeader( uint8_t nal,
 299    unsigned char *buffer, int bufferSize )
 300 {
 301    uint32_t search = 0x00000100 | nal;
 302    uint32_t found = -1;
 303    uint8_t *p = buffer;
 304    uint8_t *end = p + bufferSize;
 305    while (p < end) {
 306       found <<= 8;
 307       found |= *p++;
 308       if (found == search)
 309          return p - buffer - 4;
 310    }
 311    return -1;
 312 }
 313
 314 static void demux_ty_FindESPacket( uint8_t nal,
 315    unsigned char *buffer, int bufferSize, int *esOffset1, int *esOffset2 )
 316 {
 317   *esOffset1 = demux_ty_FindESHeader(nal, buffer, bufferSize);
 318   if (*esOffset1 == -1) {
 319     *esOffset2 = -1;
 320     return;
 321   }
 322   buffer += *esOffset1 + 1;
 323   bufferSize -= *esOffset1 + 1;
 324   *esOffset2 = demux_ty_FindESHeader(nal, buffer, bufferSize);
 325   if (*esOffset2 != -1)
 326     *esOffset2 += *esOffset1 + 1;
 327 }
 328
 329 #define VIDEO_NAL 0xe0
 330 #define AUDIO_NAL 0xc0
 331 #define AC3_NAL 0xbd
 332
 333 static int demux_ty_fill_buffer( demuxer_t *demux, demux_stream_t *dsds )
 334 {
 335    int              invalidType = 0;
 336    int              errorHeader = 0;
 337    int              recordsDecoded = 0;
 338
 339    int              readSize;
 340
 341    int              numberRecs;
 342    unsigned char    *recPtr;
 343    int              offset;
 344
 345    int              counter;
 346
 347    int              aid;
 348
 349    TiVoInfo         *tivo = demux->priv;
 350    unsigned char    *chunk = tivo->chunk;
 351
 352    if ( demux->stream->type == STREAMTYPE_DVD )
 353       return 0;
 354
 355    mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:ty processing\n" );
 356
 357    if( demux->stream->eof ) return 0;
 358
 359    // ======================================================================
 360    // If we haven't figured out the size of the stream, let's do so
 361    // ======================================================================
 362    if ( demux->stream->type == STREAMTYPE_VSTREAM )
 363    {
 364       // The vstream code figures out the exact size of the stream
 365       demux->movi_start = 0;
 366       demux->movi_end = demux->stream->end_pos;
 367       tivo->size = demux->stream->end_pos;
 368    }
 369    else
 370    {
 371       // If its a local file, try to find the Part Headers, so we can
 372       // calculate the ACTUAL stream size
 373       // If we can't find it, go off with the file size and hope the
 374       // extract program did the "right thing"
 375       if ( tivo->readHeader == 0 )
 376       {
 377          off_t filePos;
 378          tivo->readHeader = 1;
 379
 380          filePos = demux->filepos;
 381          stream_seek( demux->stream, 0 );
 382
 383          readSize = stream_read( demux->stream, chunk, CHUNKSIZE );
 384
 385          if ( memcmp( chunk, TMF_SIG, sizeof( TMF_SIG ) ) == 0 )
 386          {
 387             mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:Detected a tmf\n" );
 388             tivo->tmf = 1;
 389             ty_tmf_filetoparts( demux, tivo );
 390             readSize = tmf_load_chunk( demux, tivo, chunk, 0 );
 391          }
 392
 393          if ( readSize == CHUNKSIZE && AV_RB32(chunk) == TIVO_PES_FILEID )
 394          {
 395                off_t numberParts;
 396
 397                readSize = 0;
 398
 399                if ( tivo->tmf != 1 )
 400                {
 401                   off_t offset;
 402
 403                   numberParts = demux->stream->end_pos / TIVO_PART_LENGTH;
 404                   offset = numberParts * TIVO_PART_LENGTH;
 405
 406                   mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:ty/ty+Number Parts %"PRId64"\n",
 407                     (int64_t)numberParts );
 408
 409                   if ( offset + CHUNKSIZE < demux->stream->end_pos )
 410                   {
 411                      stream_seek( demux->stream, offset );
 412                      readSize = stream_read( demux->stream, chunk, CHUNKSIZE );
 413                   }
 414                }
 415                else
 416                {
 417                   numberParts = tivo->tmf_totalparts;
 418                   offset = numberParts * TIVO_PART_LENGTH;
 419                   readSize = tmf_load_chunk( demux, tivo, chunk,
 420                      numberParts * ( TIVO_PART_LENGTH - CHUNKSIZE ) /
 421                      CHUNKSIZE );
 422                }
 423
 424                if ( readSize == CHUNKSIZE && AV_RB32(chunk) == TIVO_PES_FILEID )
 425                {
 426                      int size = AV_RB24(chunk + 12);
 427                      size -= 4;
 428                      size *= CHUNKSIZE;
 429                      tivo->size = numberParts * TIVO_PART_LENGTH;
 430                      tivo->size += size;
 431                      mp_msg( MSGT_DEMUX, MSGL_DBG3,
 432                         "ty:Header Calc Stream Size %"PRId64"\n", tivo->size );
 433                }
 434          }
 435
 436          if ( demux->stream->start_pos > 0 )
 437             filePos = demux->stream->start_pos;
 438          stream_seek( demux->stream, filePos );
 439          demux->filepos = stream_tell( demux->stream );
 440          tivo->whichChunk = filePos / CHUNKSIZE;
 441       }
 442       demux->movi_start = 0;
 443       demux->movi_end = tivo->size;
 444    }
 445
 446    // ======================================================================
 447    // Give a clue as to where we are in the stream
 448    // ======================================================================
 449    mp_msg( MSGT_DEMUX, MSGL_DBG3,
 450       "ty:ty header size %"PRIx64"\n", (int64_t)tivo->size );
 451    mp_msg( MSGT_DEMUX, MSGL_DBG3,
 452       "ty:ty which Chunk %d\n", tivo->whichChunk );
 453    mp_msg( MSGT_DEMUX, MSGL_DBG3,
 454       "ty:file end_pos   %"PRIx64"\n", (int64_t)demux->stream->end_pos );
 455    mp_msg( MSGT_DEMUX, MSGL_DBG3,
 456       "\nty:wanted current offset %"PRIx64"\n", (int64_t)stream_tell( demux->stream ) );
 457
 458    if ( tivo->size > 0 && stream_tell( demux->stream ) > tivo->size )
 459    {
 460          demux->stream->eof = 1;
 461          return 0;
 462    }
 463
 464    do {
 465    if ( tivo->tmf != 1 )
 466    {
 467       // Make sure we are on a 128k boundary
 468       if ( demux->filepos % CHUNKSIZE != 0 )
 469       {
 470          int whichChunk = demux->filepos / CHUNKSIZE;
 471          if ( demux->filepos % CHUNKSIZE > CHUNKSIZE / 2 )
 472             whichChunk++;
 473          stream_seek( demux->stream, whichChunk * CHUNKSIZE );
 474       }
 475
 476       demux->filepos = stream_tell( demux->stream );
 477       tivo->whichChunk = demux->filepos / CHUNKSIZE;
 478       readSize = stream_read( demux->stream, chunk, CHUNKSIZE );
 479       if ( readSize != CHUNKSIZE )
 480          return 0;
 481    }
 482    else
 483    {
 484       readSize = tmf_load_chunk( demux, tivo, chunk, tivo->whichChunk );
 485       if ( readSize != CHUNKSIZE )
 486          return 0;
 487       tivo->whichChunk++;
 488    }
 489    if (AV_RB32(chunk) == TIVO_PES_FILEID)
 490       mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:Skipping PART Header\n" );
 491    } while (AV_RB32(chunk) == TIVO_PES_FILEID);
 492
 493    mp_msg( MSGT_DEMUX, MSGL_DBG3,
 494       "\nty:actual current offset %"PRIx64"\n", stream_tell( demux->stream ) -
 495       CHUNKSIZE );
 496
 497
 498    // Let's make a Video Demux Stream for MPlayer
 499    aid = 0x0;
 500    if( !demux->v_streams[ aid ] ) new_sh_video( demux, aid );
 501    if( demux->video->id == -1 ) demux->video->id = aid;
 502    if( demux->video->id == aid )
 503    {
 504       demux_stream_t *ds = demux->video;
 505       if( !ds->sh ) ds->sh = demux->v_streams[ aid ];
 506    }
 507
 508    // ======================================================================
 509    // Finally, we get to actually parse the chunk
 510    // ======================================================================
 511    mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:ty parsing a chunk\n" );
 512    numberRecs = chunk[ 0 ];
 513    recPtr = &chunk[ 4 ];
 514    offset = numberRecs * 16 + 4;
 515    for ( counter = 0 ; counter < numberRecs ; counter++ )
 516    {
 517       int size = AV_RB24(recPtr) >> 4;
 518       int type = recPtr[ 3 ];
 519       int nybbleType = recPtr[ 2 ] & 0x0f;
 520       recordsDecoded++;
 521
 522       mp_msg( MSGT_DEMUX, MSGL_DBG3,
 523          "ty:Record Type %x/%x %d\n", nybbleType, type, size );
 524
 525       // ================================================================
 526       // Video Parsing
 527       // ================================================================
 528       if ( type == 0xe0 )
 529       {
 530          if ( size > 0 && size + offset <= CHUNKSIZE )
 531          {
 532             int esOffset1 = demux_ty_FindESHeader( VIDEO_NAL, &chunk[ offset ],
 533                size);
 534             if ( esOffset1 != -1 )
 535                tivo->lastVideoPTS = get_ty_pts(
 536                   &chunk[ offset + esOffset1 + 9 ] );
 537
 538             // Do NOT Pass the PES Header onto the MPEG2 Decode
 539             if( nybbleType != 0x06 )
 540                demux_ty_CopyToDemuxPacket( demux->video,
 541                   &chunk[ offset ], size, demux->filepos + offset,
 542                   tivo->lastVideoPTS );
 543             offset += size;
 544          }
 545          else
 546             errorHeader++;
 547       }
 548       // ================================================================
 549       // Audio Parsing
 550       // ================================================================
 551       else if ( type == 0xc0 )
 552       {
 553          if ( size > 0 && size + offset <= CHUNKSIZE )
 554          {
 555             if( demux->audio->id == -1 )
 556             {
 557                if ( nybbleType == 0x02 )
 558                   continue;    // DTiVo inconclusive, wait for more
 559                else if ( nybbleType == 0x09 )
 560                {
 561                   mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:Setting AC-3 Audio\n" );
 562                   aid = 0x80;  // AC-3
 563                }
 564                else
 565                {
 566                   mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:Setting MPEG Audio\n" );
 567                   aid = 0x0;   // MPEG Audio
 568                }
 569
 570                demux->audio->id = aid;
 571                if( !demux->a_streams[ aid ] ) new_sh_audio( demux, aid );
 572                if( demux->audio->id == aid )
 573                {
 574                   demux_stream_t *ds = demux->audio;
 575                   if( !ds->sh ) {
 576                     sh_audio_t* sh_a;
 577                     ds->sh = demux->a_streams[ aid ];
 578                     sh_a = (sh_audio_t*)ds->sh;
 579                     switch(aid & 0xE0){  // 1110 0000 b  (high 3 bit: type  low 5: id)
 580                       case 0x00: sh_a->format=0x50;break; // mpeg
 581                       case 0xA0: sh_a->format=0x10001;break;  // dvd pcm
 582                       case 0x80: if((aid & 0xF8) == 0x88) sh_a->format=0x2001;//dts
 583                                   else sh_a->format=0x2000;break; // ac3
 584                     }
 585                  }
 586                }
 587             }
 588
 589             aid = demux->audio->id;
 590
 591
 592             // SA DTiVo Audio Data, no PES
 593             // ================================================
 594             if ( nybbleType == 0x02 || nybbleType == 0x04 )
 595             {
 596                if ( nybbleType == 0x02 && tivo->tivoType == 2 )
 597                   demux_ty_AddToAudioBuffer( tivo, &chunk[ offset ], size );
 598                else
 599                {
 600
 601                   mp_msg( MSGT_DEMUX, MSGL_DBG3,
 602                      "ty:Adding Audio Packet Size %d\n", size );
 603                   demux_ty_CopyToDemuxPacket( demux->audio,
 604                      &chunk[ offset ], size, ( demux->filepos + offset ),
 605                      tivo->lastAudioPTS );
 606                }
 607             }
 608
 609             // 3 - MPEG Audio with PES Header, either SA or DTiVo
 610             // 9 - DTiVo AC3 Audio Data with PES Header
 611             // ================================================
 612             if ( nybbleType == 0x03 || nybbleType == 0x09 )
 613             {
 614                int esOffset1, esOffset2;
 615                if ( nybbleType == 0x03 )
 616                esOffset1 = demux_ty_FindESHeader( AUDIO_NAL, &chunk[ offset ],
 617                   size);
 618
 619                // SA PES Header, No Audio Data
 620                // ================================================
 621                if ( nybbleType == 0x03 && esOffset1 == 0 && size == 16 )
 622                {
 623                   tivo->tivoType = 1;
 624                   tivo->lastAudioPTS = get_ty_pts( &chunk[ offset +
 625                      SERIES2_PTS_OFFSET ] );
 626                }
 627                else
 628                // DTiVo Audio with PES Header
 629                // ================================================
 630                {
 631                   tivo->tivoType = 2;
 632
 633                   demux_ty_AddToAudioBuffer( tivo, &chunk[ offset ], size );
 634                   demux_ty_FindESPacket( nybbleType == 9 ? AC3_NAL : AUDIO_NAL,
 635                      tivo->lastAudio, tivo->lastAudioEnd, &esOffset1,
 636                      &esOffset2 );
 637
 638                   if ( esOffset1 != -1 && esOffset2 != -1 )
 639                   {
 640                      int packetSize = esOffset2 - esOffset1;
 641                      int headerSize;
 642                      int ptsOffset;
 643
 644                      if ( IsValidAudioPacket( packetSize, &ptsOffset,
 645                            &headerSize ) )
 646                      {
 647                         mp_msg( MSGT_DEMUX, MSGL_DBG3,
 648                            "ty:Adding DTiVo Audio Packet Size %d\n",
 649                            packetSize );
 650
 651                         tivo->lastAudioPTS = get_ty_pts(
 652                            &tivo->lastAudio[ esOffset1 + ptsOffset ] );
 653
 654                         if (nybbleType == 9) headerSize = 0;
 655                         demux_ty_CopyToDemuxPacket
 656                         (
 657                            demux->audio,
 658                            &tivo->lastAudio[ esOffset1 + headerSize ],
 659                            packetSize - headerSize,
 660                            demux->filepos + offset,
 661                            tivo->lastAudioPTS
 662                         );
 663
 664                      }
 665
 666                      // Collapse the Audio Buffer
 667                      tivo->lastAudioEnd -= esOffset2;
 668                      memmove( &tivo->lastAudio[ 0 ],
 669                         &tivo->lastAudio[ esOffset2 ],
 670                         tivo->lastAudioEnd );
 671                   }
 672                }
 673             }
 674
 675             offset += size;
 676          }
 677          else
 678             errorHeader++;
 679       }
 680       // ================================================================
 681       // 1 = Closed Caption
 682       // 2 = Extended Data Services
 683       // ================================================================
 684       else if ( type == 0x01 || type == 0x02 )
 685       {
 686                         unsigned char lastXDS[ 16 ];
 687          int b = AV_RB24(recPtr) >> 4;
 688          b &= 0x7f7f;
 689
 690          mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:%s %04x\n", type == 1 ? "CC" : "XDS", b);
 691
 692          lastXDS[ 0x00 ] = 0x00;
 693          lastXDS[ 0x01 ] = 0x00;
 694          lastXDS[ 0x02 ] = 0x01;
 695          lastXDS[ 0x03 ] = 0xb2;
 696          lastXDS[ 0x04 ] = 'T';
 697          lastXDS[ 0x05 ] = 'Y';
 698          lastXDS[ 0x06 ] = type;
 699          lastXDS[ 0x07 ] = b >> 8;
 700          lastXDS[ 0x08 ] = b;
 701          if ( subcc_enabled )
 702             demux_ty_CopyToDemuxPacket( demux->video, lastXDS, 0x09,
 703                demux->filepos + offset, tivo->lastVideoPTS );
 704       }
 705       // ================================================================
 706       // Unknown
 707       // ================================================================
 708       else
 709       {
 710          if ( size > 0 && size + offset <= CHUNKSIZE )
 711             offset += size;
 712          if (type != 3 && type != 5 && (type != 0 || size > 0)) {
 713          mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:Invalid Type %x\n", type );
 714          invalidType++;
 715          }
 716       }
 717      recPtr += 16;
 718    }
 719
 720    if ( errorHeader > 0 || invalidType > 0 )
 721    {
 722       mp_msg( MSGT_DEMUX, MSGL_DBG3,
 723          "ty:Error Check - Records %d, Parsed %d, Errors %d + %d\n",
 724          numberRecs, recordsDecoded, errorHeader, invalidType );
 725
 726       // Invalid MPEG ES Size Check
 727       if ( errorHeader > numberRecs / 2 )
 728          return 0;
 729
 730       // Invalid MPEG Stream Type Check
 731       if ( invalidType > numberRecs / 2 )
 732          return 0;
 733    }
 734
 735    demux->filepos = stream_tell( demux->stream );
 736
 737    return 1;
 738 }
 739
 740 static void demux_seek_ty( demuxer_t *demuxer, float rel_seek_secs, float audio_delay, int flags )
 741 {
 742    demux_stream_t *d_audio = demuxer->audio;
 743    demux_stream_t *d_video = demuxer->video;
 744    sh_audio_t     *sh_audio = d_audio->sh;
 745    sh_video_t     *sh_video = d_video->sh;
 746    off_t          newpos;
 747    off_t          res;
 748    TiVoInfo       *tivo = demuxer->priv;
 749
 750    mp_msg( MSGT_DEMUX, MSGL_DBG3, "ty:Seeking to %7.1f\n", rel_seek_secs );
 751
 752       tivo->lastAudioEnd = 0;
 753       tivo->lastAudioPTS = MP_NOPTS_VALUE;
 754       tivo->lastVideoPTS = MP_NOPTS_VALUE;
 755    //
 756    //================= seek in MPEG ==========================
 757    demuxer->filepos = stream_tell( demuxer->stream );
 758
 759    newpos = ( flags & SEEK_ABSOLUTE ) ? demuxer->movi_start : demuxer->filepos;
 760
 761    if( flags & SEEK_FACTOR )
 762       // float seek 0..1
 763       newpos += ( demuxer->movi_end - demuxer->movi_start ) * rel_seek_secs;
 764    else
 765    {
 766       // time seek (secs)
 767       if( ! sh_video->i_bps ) // unspecified or VBR
 768          newpos += 2324 * 75 * rel_seek_secs; // 174.3 kbyte/sec
 769       else
 770          newpos += sh_video->i_bps * rel_seek_secs;
 771    }
 772
 773    if ( newpos < demuxer->movi_start )
 774    {
 775       if( demuxer->stream->type != STREAMTYPE_VCD ) demuxer->movi_start = 0;
 776       if( newpos < demuxer->movi_start ) newpos = demuxer->movi_start;
 777    }
 778
 779    res = newpos / CHUNKSIZE;
 780    if ( rel_seek_secs >= 0 )
 781       newpos = ( res + 1 ) * CHUNKSIZE;
 782    else
 783       newpos = res * CHUNKSIZE;
 784
 785    if ( newpos < 0 )
 786       newpos = 0;
 787
 788    tivo->whichChunk = newpos / CHUNKSIZE;
 789
 790    stream_seek( demuxer->stream, newpos );
 791
 792    // re-sync video:
 793    videobuf_code_len = 0; // reset ES stream buffer
 794
 795    ds_fill_buffer( d_video );
 796    if( sh_audio )
 797      ds_fill_buffer( d_audio );
 798
 799    while( 1 )
 800    {
 801      int i;
 802      if( sh_audio && !d_audio->eof && d_video->pts && d_audio->pts )
 803      {
 804         float a_pts = d_audio->pts;
 805         a_pts += ( ds_tell_pts( d_audio ) - sh_audio->a_in_buffer_len ) /
 806            (float)sh_audio->i_bps;
 807         if( d_video->pts > a_pts )
 808         {
 809            skip_audio_frame( sh_audio );  // sync audio
 810            continue;
 811         }
 812      }
 813      i = sync_video_packet( d_video );
 814      if( i == 0x1B3 || i == 0x1B8 ) break; // found it!
 815      if( !i || !skip_video_packet( d_video ) ) break; // EOF?
 816    }
 817    if ( subcc_enabled )
 818       ty_ClearOSD( 0 );
 819 }
 820
 821 static int demux_ty_control( demuxer_t *demuxer,int cmd, void *arg )
 822 {
 823    demux_stream_t *d_video = demuxer->video;
 824    sh_video_t     *sh_video = d_video->sh;
 825
 826    switch(cmd)
 827    {
 828       case DEMUXER_CTRL_GET_TIME_LENGTH:
 829          if(!sh_video->i_bps)  // unspecified or VBR
 830              return DEMUXER_CTRL_DONTKNOW;
 831          *(double *)arg=
 832             (double)demuxer->movi_end-demuxer->movi_start/sh_video->i_bps;
 833          return DEMUXER_CTRL_GUESS;
 834
 835       case DEMUXER_CTRL_GET_PERCENT_POS:
 836           return DEMUXER_CTRL_DONTKNOW;
 837        default:
 838           return DEMUXER_CTRL_NOTIMPL;
 839     }
 840 }
 841
 842
 843 static void demux_close_ty( demuxer_t *demux )
 844 {
 845    TiVoInfo         *tivo = demux->priv;
 846
 847       free( tivo );
 848       sub_justify = 0;
 849 }
 850
 851
 852 static int ty_check_file(demuxer_t* demuxer)
 853 {
 854   TiVoInfo *tivo = calloc(1, sizeof(TiVoInfo));
 855   demuxer->priv = tivo;
 856   return ds_fill_buffer(demuxer->video) ? DEMUXER_TYPE_MPEG_TY : 0;
 857 }
 858
 859
 860 static demuxer_t* demux_open_ty(demuxer_t* demuxer)
 861 {
 862     sh_audio_t *sh_audio=NULL;
 863     sh_video_t *sh_video=NULL;
 864
 865     sh_video=demuxer->video->sh;sh_video->ds=demuxer->video;
 866
 867     if(demuxer->audio->id!=-2) {
 868         if(!ds_fill_buffer(demuxer->audio)){
 869             mp_msg(MSGT_DEMUXER, MSGL_INFO, "MPEG: %s",
 870                    mp_gtext("No audio stream found -> no sound.\n"));
 871             demuxer->audio->sh=NULL;
 872         } else {
 873             sh_audio=demuxer->audio->sh;sh_audio->ds=demuxer->audio;
 874         }
 875     }
 876
 877     return demuxer;
 878 }
 879
 880
 881 const demuxer_desc_t demuxer_desc_mpeg_ty = {
 882   "TiVo demuxer",
 883   "tivo",
 884   "TiVo",
 885   "Christopher R. Wingert",
 886   "Demux streams from TiVo",
 887   DEMUXER_TYPE_MPEG_TY,
 888   0, // unsafe autodetect
 889   ty_check_file,
 890   demux_ty_fill_buffer,
 891   demux_open_ty,
 892   demux_close_ty,
 893   demux_seek_ty,
 894   demux_ty_control
 895 };