libmpdemux/demux_pva.c

   1 /*
   2  * demuxer for PVA files, such as the ones produced by software to manage DVB boards
   3  * like the Hauppauge WinTV DVBs, for MPlayer.
   4  *
   5  * Uses info from the PVA file specifications found at
   6  *
   7  * http://www.technotrend.de/download/av_format_v1.pdf
   8  *
   9  * WARNING: Quite a hack was required in order to get files by MultiDec played back correctly.
  10  * If it breaks anything else, just comment out the "#define DEMUX_PVA_MULTIDEC_HACK" below
  11  * and it will not be compiled in.
  12  *
  13  * Feedback is appreciated.
  14  *
  15  * written by Matteo Giani
  16  */
  17
  18 #define DEMUX_PVA_MULTIDEC_HACK
  19 #define PVA_NEW_PREBYTES_CODE
  20
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24
  25 #include "config.h"
  26 #include "mp_msg.h"
  27 #include "help_mp.h"
  28
  29 #include "stream/stream.h"
  30 #include "demuxer.h"
  31 #include "stheader.h"
  32
  33 /*
  34  * #defines below taken from PVA spec (see URL above)
  35  */
  36
  37 #define PVA_MAX_VIDEO_PACK_LEN 6*1024
  38
  39 #define VIDEOSTREAM 0x01
  40 #define MAINAUDIOSTREAM 0x02
  41
  42 typedef struct {
  43         off_t offset;
  44         long size;
  45         uint8_t type;
  46         uint8_t is_packet_start;
  47         float pts;
  48 } pva_payload_t;
  49
  50
  51 typedef struct {
  52         float last_audio_pts;
  53         float last_video_pts;
  54 #ifdef PVA_NEW_PREBYTES_CODE
  55         float video_pts_after_prebytes;
  56         long video_size_after_prebytes;
  57         uint8_t prebytes_delivered;
  58 #endif
  59         uint8_t just_synced;
  60         uint8_t synced_stream_id;
  61 } pva_priv_t;
  62
  63
  64
  65 int pva_sync(demuxer_t * demuxer)
  66 {
  67         uint8_t buffer[5]={0,0,0,0,0};
  68         int count;
  69         pva_priv_t * priv = (pva_priv_t *) demuxer->priv;
  70
  71
  72          /* This function is used to find the next nearest PVA packet start after a seek, since a PVA file
  73           * is not indexed.
  74           * The just_synced field is in the priv structure so that pva_get_payload knows pva_sync
  75           * has already read (part of) the PVA header. This way we can avoid to seek back and (hopefully)
  76           * be able to read from pipes and such.
  77           */
  78
  79
  80         for(count=0 ; count<PVA_MAX_VIDEO_PACK_LEN && !demuxer->stream->eof && !priv->just_synced ; count++)
  81         {
  82                 buffer[0]=buffer[1];
  83                 buffer[1]=buffer[2];
  84                 buffer[2]=buffer[3];
  85                 buffer[3]=buffer[4];
  86                 buffer[4]=stream_read_char(demuxer->stream);
  87                 /*
  88                  * Check for a PVA packet beginning sequence: we check both the "AV" word at the
  89                  * very beginning and the "0x55" reserved byte (which is unused and set to 0x55 by spec)
  90                  */
  91                 if(buffer[0]=='A' && buffer[1] == 'V' && buffer[4] == 0x55) priv->just_synced=1;
  92                 //printf("demux_pva: pva_sync(): current offset= %ld\n",stream_tell(demuxer->stream));
  93         }
  94         if(priv->just_synced)
  95         {
  96                 priv->synced_stream_id=buffer[2];
  97                 return 1;
  98         }
  99         else
 100         {
 101                 return 0;
 102         }
 103 }
 104
 105 static int pva_check_file(demuxer_t * demuxer)
 106 {
 107         uint8_t buffer[5]={0,0,0,0,0};
 108         mp_msg(MSGT_DEMUX, MSGL_V, "Checking for PVA\n");
 109         stream_read(demuxer->stream,buffer,5);
 110         if(buffer[0]=='A' && buffer[1] == 'V' && buffer[4] == 0x55)
 111         {
 112                 mp_msg(MSGT_DEMUX,MSGL_DBG2, "Success: PVA\n");
 113                 return DEMUXER_TYPE_PVA;
 114         }
 115         else
 116         {
 117                 mp_msg(MSGT_DEMUX,MSGL_DBG2, "Failed: PVA\n");
 118                 return 0;
 119         }
 120 }
 121
 122 static demuxer_t * demux_open_pva (demuxer_t * demuxer)
 123 {
 124         sh_video_t *sh_video = new_sh_video(demuxer,0);
 125         sh_audio_t *sh_audio = new_sh_audio(demuxer,0);
 126
 127
 128         pva_priv_t * priv;
 129
 130         stream_reset(demuxer->stream);
 131         stream_seek(demuxer->stream,0);
 132
 133
 134
 135         priv=malloc(sizeof(pva_priv_t));
 136
 137         if(demuxer->stream->type!=STREAMTYPE_FILE) demuxer->seekable=0;
 138         else demuxer->seekable=1;
 139
 140         demuxer->priv=priv;
 141         memset(demuxer->priv,0,sizeof(pva_priv_t));
 142
 143         if(!pva_sync(demuxer))
 144         {
 145                 mp_msg(MSGT_DEMUX,MSGL_ERR,"Not a PVA file.\n");
 146                 return NULL;
 147         }
 148
 149         //printf("priv->just_synced %s after initial sync!\n",priv->just_synced?"set":"UNSET");
 150
 151         demuxer->video->sh=sh_video;
 152
 153         //printf("demuxer->stream->end_pos= %d\n",demuxer->stream->end_pos);
 154
 155
 156         mp_msg(MSGT_DEMUXER,MSGL_INFO,"Opened PVA demuxer...\n");
 157
 158         /*
 159          * Audio and Video codecs:
 160          * the PVA spec only allows MPEG2 video and MPEG layer II audio. No need to check the formats then.
 161          * Moreover, there would be no way to do that since the PVA stream format has no fields to describe
 162          * the used codecs.
 163          */
 164
 165         sh_video->format=0x10000002;
 166         sh_video->ds=demuxer->video;
 167
 168         /*
 169         printf("demuxer->video->id==%d\n",demuxer->video->id);
 170         printf("demuxer->audio->id==%d\n",demuxer->audio->id);
 171         */
 172
 173         demuxer->audio->id = 0;
 174         demuxer->audio->sh=sh_audio;
 175         sh_audio->format=0x50;
 176         sh_audio->ds=demuxer->audio;
 177
 178         demuxer->movi_start=0;
 179         demuxer->movi_end=demuxer->stream->end_pos;
 180
 181         priv->last_video_pts=-1;
 182         priv->last_audio_pts=-1;
 183
 184         return demuxer;
 185 }
 186
 187 int pva_get_payload(demuxer_t * d,pva_payload_t * payload);
 188
 189 // 0 = EOF or no stream found
 190 // 1 = successfully read a packet
 191 static int demux_pva_fill_buffer (demuxer_t * demux, demux_stream_t *ds)
 192 {
 193         uint8_t done=0;
 194         demux_packet_t * dp;
 195         pva_priv_t * priv=demux->priv;
 196         pva_payload_t current_payload;
 197
 198         while(!done)
 199         {
 200                 if(!pva_get_payload(demux,&current_payload)) return 0;
 201                 switch(current_payload.type)
 202                 {
 203                         case VIDEOSTREAM:
 204                                 if(demux->video->id==-1) demux->video->id=0;
 205                                 if(!current_payload.is_packet_start && priv->last_video_pts==-1)
 206                                 {
 207                                         /* We should only be here at the beginning of a stream, when we have
 208                                          * not yet encountered a valid Video PTS, or after a seek.
 209                                          * So, skip these starting packets in order not to deliver the
 210                                          * player a bogus PTS.
 211                                          */
 212                                         done=0;
 213                                 }
 214                                 else
 215                                 {
 216                                         /*
 217                                          * In every other condition, we are delivering the payload. Set this
 218                                          * so that the following code knows whether to skip it or read it.
 219                                          */
 220                                         done=1;
 221                                 }
 222                                 if(demux->video->id!=0) done=0;
 223                                 if(current_payload.is_packet_start)
 224                                 {
 225                                         priv->last_video_pts=current_payload.pts;
 226                                         //mp_msg(MSGT_DEMUXER,MSGL_DBG2,"demux_pva: Video PTS=%llu , delivered %f\n",current_payload.pts,priv->last_video_pts);
 227                                 }
 228                                 if(done)
 229                                 {
 230                                         dp=new_demux_packet(current_payload.size);
 231                                         dp->pts=priv->last_video_pts;
 232                                         stream_read(demux->stream,dp->buffer,current_payload.size);
 233                                         ds_add_packet(demux->video,dp);
 234                                 }
 235                                 else
 236                                 {
 237                                         //printf("Skipping %u video bytes\n",current_payload.size);
 238                                         stream_skip(demux->stream,current_payload.size);
 239                                 }
 240                                 break;
 241                         case MAINAUDIOSTREAM:
 242                                 if(demux->audio->id==-1) demux->audio->id=0;
 243                                 if(!current_payload.is_packet_start && priv->last_audio_pts==-1)
 244                                 {
 245                                         /* Same as above for invalid video PTS, just for audio. */
 246                                         done=0;
 247                                 }
 248                                 else
 249                                 {
 250                                         done=1;
 251                                 }
 252                                 if(current_payload.is_packet_start)
 253                                 {
 254                                         priv->last_audio_pts=current_payload.pts;
 255                                 }
 256                                 if(demux->audio->id!=0) done=0;
 257                                 if(done)
 258                                 {
 259                                         dp=new_demux_packet(current_payload.size);
 260                                         dp->pts=priv->last_audio_pts;
 261                                         if(current_payload.offset != stream_tell(demux->stream))
 262                                                 stream_seek(demux->stream,current_payload.offset);
 263                                         stream_read(demux->stream,dp->buffer,current_payload.size);
 264                                         ds_add_packet(demux->audio,dp);
 265                                 }
 266                                 else
 267                                 {
 268                                         stream_skip(demux->stream,current_payload.size);
 269                                 }
 270                                 break;
 271                 }
 272         }
 273         return 1;
 274 }
 275
 276 int pva_get_payload(demuxer_t * d,pva_payload_t * payload)
 277 {
 278         uint8_t flags,pes_head_len;
 279         uint16_t pack_size;
 280         off_t next_offset,pva_payload_start;
 281         unsigned char buffer[256];
 282 #ifndef PVA_NEW_PREBYTES_CODE
 283         demux_packet_t * dp;    //hack to deliver the preBytes (see PVA doc)
 284 #endif
 285         pva_priv_t * priv;
 286
 287
 288         if(d==NULL)
 289         {
 290                 mp_msg(MSGT_DEMUX,MSGL_ERR,"demux_pva: pva_get_payload got passed a NULL pointer!\n");
 291                 return 0;
 292         }
 293
 294         priv = (pva_priv_t *)d->priv;
 295         d->filepos=stream_tell(d->stream);
 296
 297
 298
 299
 300         if(d->stream->eof)
 301         {
 302                 mp_msg(MSGT_DEMUX,MSGL_V,"demux_pva: pva_get_payload() detected stream->eof!!!\n");
 303                 return 0;
 304         }
 305
 306         //printf("priv->just_synced %s\n",priv->just_synced?"SET":"UNSET");
 307
 308 #ifdef PVA_NEW_PREBYTES_CODE
 309         if(priv->prebytes_delivered)
 310                 /* The previous call to this fn has delivered the preBytes. Then we are already inside
 311                  * the payload. Let's just deliver the video along with its right PTS, the one we stored
 312                  * in the priv structure and was in the PVA header before the PreBytes.
 313                  */
 314         {
 315                 //printf("prebytes_delivered=1. Resetting.\n");
 316                 payload->size = priv->video_size_after_prebytes;
 317                 payload->pts = priv->video_pts_after_prebytes;
 318                 payload->is_packet_start = 1;
 319                 payload->offset = stream_tell(d->stream);
 320                 payload->type = VIDEOSTREAM;
 321                 priv->prebytes_delivered = 0;
 322                 return 1;
 323         }
 324 #endif
 325         if(!priv->just_synced)
 326         {
 327                 if(stream_read_word(d->stream) != (('A'<<8)|'V'))
 328                 {
 329                         mp_msg(MSGT_DEMUX,MSGL_V,"demux_pva: pva_get_payload() missed a SyncWord at %"PRId64"!! Trying to sync...\n",(int64_t)stream_tell(d->stream));
 330                         if(!pva_sync(d))
 331                         {
 332                                 if (!d->stream->eof)
 333                                 {
 334                                         mp_msg(MSGT_DEMUX,MSGL_ERR,"demux_pva: couldn't sync! (broken file?)");
 335                                 }
 336                                 return 0;
 337                         }
 338                 }
 339         }
 340         if(priv->just_synced)
 341         {
 342                 payload->type=priv->synced_stream_id;
 343                 priv->just_synced=0;
 344         }
 345         else
 346         {
 347                 payload->type=stream_read_char(d->stream);
 348                 stream_skip(d->stream,2); //counter and reserved
 349         }
 350         flags=stream_read_char(d->stream);
 351         payload->is_packet_start=flags & 0x10;
 352         pack_size=le2me_16(stream_read_word(d->stream));
 353         mp_msg(MSGT_DEMUX,MSGL_DBG2,"demux_pva::pva_get_payload(): pack_size=%u field read at offset %"PRIu64"\n",pack_size,(int64_t)stream_tell(d->stream)-2);
 354         pva_payload_start=stream_tell(d->stream);
 355         next_offset=pva_payload_start+pack_size;
 356
 357
 358         /*
 359          * The code in the #ifdef directive below is a hack needed to get badly formatted PVA files
 360          * such as the ones written by MultiDec played back correctly.
 361          * Basically, it works like this: if the PVA packet does not signal a PES header, but the
 362          * payload looks like one, let's assume it IS one. It has worked for me up to now.
 363          * It can be disabled since it's quite an ugly hack and could potentially break things up
 364          * if the PVA audio payload happens to start with 0x000001 even without being a non signalled
 365          * PES header start.
 366          * Though it's quite unlikely, it potentially could (AFAIK).
 367          */
 368 #ifdef DEMUX_PVA_MULTIDEC_HACK
 369         if(payload->type==MAINAUDIOSTREAM)
 370         {
 371                 stream_read(d->stream,buffer,3);
 372                 if(buffer[0]==0x00 && buffer[1]==0x00 && buffer[2]==0x01 && !payload->is_packet_start)
 373                 {
 374                         mp_msg(MSGT_DEMUX,MSGL_V,"demux_pva: suspecting non signaled audio PES packet start. Maybe file by MultiDec?\n");
 375                         payload->is_packet_start=1;
 376                 }
 377                 stream_seek(d->stream,stream_tell(d->stream)-3);
 378         }
 379 #endif
 380
 381
 382         if(!payload->is_packet_start)
 383         {
 384                 payload->offset=stream_tell(d->stream);
 385                 payload->size=pack_size;
 386         }
 387         else
 388         {       //here comes the good part...
 389                 switch(payload->type)
 390                 {
 391                         case VIDEOSTREAM:
 392                                 payload->pts=(float)(le2me_32(stream_read_dword(d->stream)))/90000;
 393                                 //printf("Video PTS: %f\n",payload->pts);
 394                                 if((flags&0x03)
 395 #ifdef PVA_NEW_PREBYTES_CODE
 396                                                 && !priv->prebytes_delivered
 397 #endif
 398                                                 )
 399                                 {
 400 #ifndef PVA_NEW_PREBYTES_CODE
 401                                         dp=new_demux_packet(flags&0x03);
 402                                         stream_read(d->stream,dp->buffer,flags & 0x03); //read PreBytes
 403                                         ds_add_packet(d->video,dp);
 404 #else
 405                                         //printf("Delivering prebytes. Setting prebytes_delivered.");
 406                                         payload->offset=stream_tell(d->stream);
 407                                         payload->size = flags & 0x03;
 408                                         priv->video_pts_after_prebytes = payload->pts;
 409                                         priv->video_size_after_prebytes = pack_size - 4 - (flags & 0x03);
 410                                         payload->pts=priv->last_video_pts;
 411                                         payload->is_packet_start=0;
 412                                         priv->prebytes_delivered=1;
 413                                         return 1;
 414 #endif
 415                                 }
 416
 417
 418                                 //now we are at real beginning of payload.
 419                                 payload->offset=stream_tell(d->stream);
 420                                 //size is pack_size minus PTS size minus PreBytes size.
 421                                 payload->size=pack_size - 4 - (flags & 0x03);
 422                                 break;
 423                         case MAINAUDIOSTREAM:
 424                                 stream_skip(d->stream,3); //FIXME properly parse PES header.
 425                                 //printf("StreamID in audio PES header: 0x%2X\n",stream_read_char(d->stream));
 426                                 stream_skip(d->stream,4);
 427
 428                                 buffer[255]=stream_read_char(d->stream);
 429                                 pes_head_len=stream_read_char(d->stream);
 430                                 stream_read(d->stream,buffer,pes_head_len);
 431                                 if(!buffer[255]&0x80) //PES header does not contain PTS.
 432                                 {
 433                                         mp_msg(MSGT_DEMUX,MSGL_V,"Audio PES packet does not contain PTS. (pes_head_len=%d)\n",pes_head_len);
 434                                         payload->pts=priv->last_audio_pts;
 435                                         break;
 436                                 }
 437                                 else            //PES header DOES contain PTS
 438                                 {
 439                                         if((buffer[0] & 0xf0)!=0x20) // PTS badly formatted
 440                                         {
 441                                                 mp_msg(MSGT_DEMUX,MSGL_V,"demux_pva: expected audio PTS but badly formatted... (read 0x%02X). Falling back to previous PTS (hack).\n",buffer[0]);
 442                                                 payload->pts=priv->last_audio_pts;
 443                                         //      return 0;
 444                                         }
 445                                         else
 446                                         {
 447                                                 uint64_t temp_pts;
 448
 449                                                 temp_pts=0LL;
 450                                                 temp_pts|=((uint64_t)(buffer[0] & 0x0e) << 29);
 451                                                 temp_pts|=buffer[1]<<22;
 452                                                 temp_pts|=(buffer[2] & 0xfe) << 14;
 453                                                 temp_pts|=buffer[3]<<7;
 454                                                 temp_pts|=(buffer[4] & 0xfe) >> 1;
 455                                                 /*
 456                                                 * PTS parsing is hopefully finished.
 457                                                 */
 458                                                 payload->pts=(float)le2me_64(temp_pts)/90000;
 459                                         }
 460                                 }
 461                                 payload->offset=stream_tell(d->stream);
 462                                 payload->size=pack_size-stream_tell(d->stream)+pva_payload_start;
 463                                 break;
 464                 }
 465         }
 466         return 1;
 467 }
 468
 469 static void demux_seek_pva(demuxer_t * demuxer,float rel_seek_secs,float audio_delay,int flags)
 470 {
 471         int total_bitrate=0;
 472         off_t dest_offset;
 473         pva_priv_t * priv=demuxer->priv;
 474
 475         total_bitrate=((sh_audio_t *)demuxer->audio->sh)->i_bps + ((sh_video_t *)demuxer->video->sh)->i_bps;
 476
 477         /*
 478          * Compute absolute offset inside the stream. Approximate total bitrate with sum of bitrates
 479          * reported by the audio and video codecs. The seek is not accurate because, just like
 480          * with MPEG streams, the bitrate is not constant. Moreover, we do not take into account
 481          * the overhead caused by PVA and PES headers.
 482          * If the calculated absolute offset is negative, seek to the beginning of the file.
 483          */
 484
 485         dest_offset=stream_tell(demuxer->stream)+rel_seek_secs*total_bitrate;
 486         if(dest_offset<0) dest_offset=0;
 487
 488         stream_seek(demuxer->stream,dest_offset);
 489
 490         if(!pva_sync(demuxer))
 491         {
 492                 mp_msg(MSGT_DEMUX,MSGL_V,"demux_pva: Couldn't seek!\n");
 493                 return;
 494         }
 495
 496         /*
 497          * Reset the PTS info inside the pva_priv_t structure. This way we don't deliver
 498          * data with the wrong PTSs (the ones we had before seeking).
 499          *
 500          */
 501
 502         priv->last_video_pts=-1;
 503         priv->last_audio_pts=-1;
 504 }
 505
 506
 507
 508 static void demux_close_pva(demuxer_t * demuxer)
 509 {
 510         if(demuxer->priv)
 511         {
 512                 free(demuxer->priv);
 513                 demuxer->priv=NULL;
 514         }
 515 }
 516
 517
 518 const demuxer_desc_t demuxer_desc_pva = {
 519   "PVA demuxer",
 520   "pva",
 521   "PVA",
 522   "Matteo Giani",
 523   "streams from DVB cards",
 524   DEMUXER_TYPE_PVA,
 525   0, // unsafe autodetect
 526   pva_check_file,
 527   demux_pva_fill_buffer,
 528   demux_open_pva,
 529   demux_close_pva,
 530   demux_seek_pva,
 531   NULL
 532 };