libmpdemux/demuxer.h

   1 #ifndef MPLAYER_DEMUXER_H
   2 #define MPLAYER_DEMUXER_H
   3
   4 #include <sys/types.h>
   5 #include <stdint.h>
   6 #include <stdlib.h>
   7 #include <string.h>
   8
   9 #include "stream/stream.h"
  10
  11 struct MPOpts;
  12
  13 #define MAX_PACKS 4096
  14 #ifdef CONFIG_TV_BSDBT848
  15 #define MAX_PACK_BYTES 0x2000000
  16 #else
  17 #define MAX_PACK_BYTES 0x800000
  18 #endif
  19
  20 #define DEMUXER_TYPE_UNKNOWN 0
  21 #define DEMUXER_TYPE_MPEG_ES 1
  22 #define DEMUXER_TYPE_MPEG_PS 2
  23 #define DEMUXER_TYPE_AVI 3
  24 #define DEMUXER_TYPE_AVI_NI 4
  25 #define DEMUXER_TYPE_AVI_NINI 5
  26 #define DEMUXER_TYPE_ASF 6
  27 #define DEMUXER_TYPE_MOV 7
  28 #define DEMUXER_TYPE_VIVO 8
  29 #define DEMUXER_TYPE_TV 9
  30 #define DEMUXER_TYPE_FLI 10
  31 #define DEMUXER_TYPE_REAL 11
  32 #define DEMUXER_TYPE_Y4M 12
  33 #define DEMUXER_TYPE_NUV 13
  34 #define DEMUXER_TYPE_FILM 14
  35 #define DEMUXER_TYPE_ROQ 15
  36 #define DEMUXER_TYPE_MF 16
  37 #define DEMUXER_TYPE_AUDIO 17
  38 #define DEMUXER_TYPE_OGG 18
  39 #define DEMUXER_TYPE_RAWAUDIO 20
  40 #define DEMUXER_TYPE_RTP 21
  41 #define DEMUXER_TYPE_RAWDV 22
  42 #define DEMUXER_TYPE_PVA 23
  43 #define DEMUXER_TYPE_SMJPEG 24
  44 #define DEMUXER_TYPE_XMMS 25
  45 #define DEMUXER_TYPE_RAWVIDEO 26
  46 #define DEMUXER_TYPE_MPEG4_ES 27
  47 #define DEMUXER_TYPE_GIF 28
  48 #define DEMUXER_TYPE_MPEG_TS 29
  49 #define DEMUXER_TYPE_H264_ES 30
  50 #define DEMUXER_TYPE_MATROSKA 31
  51 #define DEMUXER_TYPE_REALAUDIO 32
  52 #define DEMUXER_TYPE_MPEG_TY 33
  53 #define DEMUXER_TYPE_LMLM4 34
  54 #define DEMUXER_TYPE_LAVF 35
  55 #define DEMUXER_TYPE_NSV 36
  56 #define DEMUXER_TYPE_VQF 37
  57 #define DEMUXER_TYPE_AVS 38
  58 #define DEMUXER_TYPE_AAC 39
  59 #define DEMUXER_TYPE_MPC 40
  60 #define DEMUXER_TYPE_MPEG_PES 41
  61 #define DEMUXER_TYPE_MPEG_GXF 42
  62 #define DEMUXER_TYPE_NUT 43
  63 #define DEMUXER_TYPE_LAVF_PREFERRED 44
  64 #define DEMUXER_TYPE_RTP_NEMESI 45
  65
  66 // This should always match the higest demuxer type number.
  67 // Unless you want to disallow users to force the demuxer to some types
  68 #define DEMUXER_TYPE_MIN 0
  69 #define DEMUXER_TYPE_MAX 45
  70
  71 #define DEMUXER_TYPE_DEMUXERS (1<<16)
  72 // A virtual demuxer type for the network code
  73 #define DEMUXER_TYPE_PLAYLIST (2<<16)
  74
  75
  76 #define MP_NOPTS_VALUE (-1LL<<63) //both int64_t and double should be able to represent this exactly
  77
  78
  79 // DEMUXER control commands/answers
  80 #define DEMUXER_CTRL_NOTIMPL -1
  81 #define DEMUXER_CTRL_DONTKNOW 0
  82 #define DEMUXER_CTRL_OK 1
  83 #define DEMUXER_CTRL_GUESS 2
  84 #define DEMUXER_CTRL_GET_TIME_LENGTH 10
  85 #define DEMUXER_CTRL_GET_PERCENT_POS 11
  86 #define DEMUXER_CTRL_SWITCH_AUDIO 12
  87 #define DEMUXER_CTRL_RESYNC 13
  88 #define DEMUXER_CTRL_SWITCH_VIDEO 14
  89 #define DEMUXER_CTRL_IDENTIFY_PROGRAM 15
  90 #define DEMUXER_CTRL_CORRECT_PTS 16
  91
  92 #define SEEK_ABSOLUTE (1 << 0)
  93 #define SEEK_FACTOR   (1 << 1)
  94
  95 #define MP_INPUT_BUFFER_PADDING_SIZE 8
  96
  97 // Holds one packet/frame/whatever
  98 typedef struct demux_packet {
  99   int len;
 100   double pts;
 101   double endpts;
 102   double stream_pts;
 103   off_t pos;  // position in index (AVI) or file (MPG)
 104   unsigned char* buffer;
 105   int flags; // keyframe, etc
 106   int refcount;   //refcounter for the master packet, if 0, buffer can be free()d
 107   struct demux_packet *master; //pointer to the master packet if this one is a cloned one
 108   struct demux_packet *next;
 109 } demux_packet_t;
 110
 111 typedef struct demux_stream {
 112   int buffer_pos;          // current buffer position
 113   int buffer_size;         // current buffer size
 114   unsigned char* buffer;   // current buffer, never free() it, always use free_demux_packet(buffer_ref);
 115   double pts;              // current buffer's pts
 116   int pts_bytes;           // number of bytes read after last pts stamp
 117   int eof;                 // end of demuxed stream? (true if all buffer empty)
 118   off_t pos;                 // position in the input stream (file)
 119   off_t dpos;                // position in the demuxed stream
 120   int pack_no;             // serial number of packet
 121   int flags;               // flags of current packet (keyframe etc)
 122 //---------------
 123   int packs;              // number of packets in buffer
 124   int bytes;              // total bytes of packets in buffer
 125   demux_packet_t *first;  // read to current buffer from here
 126   demux_packet_t *last;   // append new packets from input stream to here
 127   demux_packet_t *current;// needed for refcounting of the buffer
 128   int id;                 // stream ID  (for multiple audio/video streams)
 129   struct demuxer *demuxer; // parent demuxer structure (stream handler)
 130 // ---- asf -----
 131   demux_packet_t *asf_packet;  // read asf fragments here
 132   int asf_seq;
 133 // ---- mov -----
 134   unsigned int ss_mul,ss_div;
 135 // ---- stream header ----
 136   void* sh;
 137 } demux_stream_t;
 138
 139 typedef struct demuxer_info {
 140   char *name;
 141   char *author;
 142   char *encoder;
 143   char *comments;
 144   char *copyright;
 145 } demuxer_info_t;
 146
 147 #define MAX_A_STREAMS 256
 148 #define MAX_V_STREAMS 256
 149 #define MAX_S_STREAMS 32
 150
 151 struct demuxer;
 152
 153 /**
 154  * Demuxer description structure
 155  */
 156 typedef struct demuxer_desc {
 157   const char *info; ///< What is it (long name and/or description)
 158   const char *name; ///< Demuxer name, used with -demuxer switch
 159   const char *shortdesc; ///< Description printed at demuxer detection
 160   const char *author; ///< Demuxer author(s)
 161   const char *comment; ///< Comment, printed with -demuxer help
 162
 163   int type; ///< DEMUXER_TYPE_xxx
 164   int safe_check; ///< If 1 detection is safe and fast, do it before file extension check
 165
 166   /// Check if can demux the file, return DEMUXER_TYPE_xxx on success
 167   int (*check_file)(struct demuxer *demuxer); ///< Mandatory if safe_check == 1, else optional
 168   /// Get packets from file, return 0 on eof
 169   int (*fill_buffer)(struct demuxer *demuxer, demux_stream_t *ds); ///< Mandatory
 170   /// Open the demuxer, return demuxer on success, NULL on failure
 171   struct demuxer* (*open)(struct demuxer *demuxer); ///< Optional
 172   /// Close the demuxer
 173   void (*close)(struct demuxer *demuxer); ///< Optional
 174   // Seek
 175   void (*seek)(struct demuxer *demuxer, float rel_seek_secs, float audio_delay, int flags); ///< Optional
 176   // Control
 177   int (*control)(struct demuxer *demuxer, int cmd, void *arg); ///< Optional
 178 } demuxer_desc_t;
 179
 180 typedef struct demux_chapter
 181 {
 182   uint64_t start, end;
 183   char* name;
 184 } demux_chapter_t;
 185
 186 typedef struct demux_attachment
 187 {
 188   char* name;
 189   char* type;
 190   void* data;
 191   unsigned int data_size;
 192 } demux_attachment_t;
 193
 194 typedef struct demuxer {
 195   const demuxer_desc_t *desc;  ///< Demuxer description structure
 196   off_t filepos; // input stream current pos.
 197   off_t movi_start;
 198   off_t movi_end;
 199   stream_t *stream;
 200   double stream_pts;       // current stream pts, if applicable (e.g. dvd)
 201   double reference_clock;
 202   char *filename; ///< Needed by avs_check_file
 203   int synced;  // stream synced (used by mpeg)
 204   int type;    // demuxer type: mpeg PS, mpeg ES, avi, avi-ni, avi-nini, asf
 205   int file_format;  // file format: mpeg/avi/asf
 206   int seekable;  // flag
 207   //
 208   demux_stream_t *audio; // audio buffer/demuxer
 209   demux_stream_t *video; // video buffer/demuxer
 210   demux_stream_t *sub;   // dvd subtitle buffer/demuxer
 211
 212   // stream headers:
 213   struct sh_audio *a_streams[MAX_A_STREAMS];
 214   struct sh_video *v_streams[MAX_V_STREAMS];
 215   struct sh_sub   *s_streams[MAX_S_STREAMS];
 216
 217   demux_chapter_t* chapters;
 218   int num_chapters;
 219
 220   demux_attachment_t* attachments;
 221   int num_attachments;
 222
 223   void* priv;  // fileformat-dependent data
 224   char** info;
 225   struct MPOpts *opts;
 226 } demuxer_t;
 227
 228 typedef struct {
 229   int progid;        //program id
 230   int aid, vid, sid; //audio, video and subtitle id
 231 } demux_program_t;
 232
 233 static inline demux_packet_t* new_demux_packet(int len){
 234   demux_packet_t* dp=(demux_packet_t*)malloc(sizeof(demux_packet_t));
 235   dp->len=len;
 236   dp->next=NULL;
 237   dp->pts=MP_NOPTS_VALUE;
 238   dp->endpts=MP_NOPTS_VALUE;
 239   dp->stream_pts = MP_NOPTS_VALUE;
 240   dp->pos=0;
 241   dp->flags=0;
 242   dp->refcount=1;
 243   dp->master=NULL;
 244   dp->buffer=NULL;
 245   if (len > 0 && (dp->buffer = (unsigned char *)malloc(len + MP_INPUT_BUFFER_PADDING_SIZE)))
 246     memset(dp->buffer + len, 0, 8);
 247   else
 248     dp->len = 0;
 249   return dp;
 250 }
 251
 252 static inline void resize_demux_packet(demux_packet_t* dp, int len)
 253 {
 254   if(len > 0)
 255   {
 256      dp->buffer=(unsigned char *)realloc(dp->buffer,len+8);
 257   }
 258   else
 259   {
 260      if(dp->buffer) free(dp->buffer);
 261      dp->buffer=NULL;
 262   }
 263   dp->len=len;
 264   if (dp->buffer)
 265      memset(dp->buffer + len, 0, 8);
 266   else
 267      dp->len = 0;
 268 }
 269
 270 static inline demux_packet_t* clone_demux_packet(demux_packet_t* pack){
 271   demux_packet_t* dp=(demux_packet_t*)malloc(sizeof(demux_packet_t));
 272   while(pack->master) pack=pack->master; // find the master
 273   memcpy(dp,pack,sizeof(demux_packet_t));
 274   dp->next=NULL;
 275   dp->refcount=0;
 276   dp->master=pack;
 277   pack->refcount++;
 278   return dp;
 279 }
 280
 281 static inline void free_demux_packet(demux_packet_t* dp){
 282   if (dp->master==NULL){  //dp is a master packet
 283     dp->refcount--;
 284     if (dp->refcount==0){
 285       if (dp->buffer) free(dp->buffer);
 286       free(dp);
 287     }
 288     return;
 289   }
 290   // dp is a clone:
 291   free_demux_packet(dp->master);
 292   free(dp);
 293 }
 294
 295 #ifndef SIZE_MAX
 296 #define SIZE_MAX ((size_t)-1)
 297 #endif
 298
 299 static inline void *realloc_struct(void *ptr, size_t nmemb, size_t size) {
 300   if (nmemb > SIZE_MAX / size) {
 301     free(ptr);
 302     return NULL;
 303   }
 304   return realloc(ptr, nmemb * size);
 305 }
 306
 307 demux_stream_t* new_demuxer_stream(struct demuxer *demuxer,int id);
 308 demuxer_t* new_demuxer(struct MPOpts *opts, stream_t *stream,int type,int a_id,int v_id,int s_id,char *filename);
 309 void free_demuxer_stream(demux_stream_t *ds);
 310 void free_demuxer(demuxer_t *demuxer);
 311
 312 void ds_add_packet(demux_stream_t *ds,demux_packet_t* dp);
 313 void ds_read_packet(demux_stream_t *ds, stream_t *stream, int len, double pts, off_t pos, int flags);
 314
 315 int demux_fill_buffer(demuxer_t *demux,demux_stream_t *ds);
 316 int ds_fill_buffer(demux_stream_t *ds);
 317
 318 static inline off_t ds_tell(demux_stream_t *ds){
 319   return (ds->dpos-ds->buffer_size)+ds->buffer_pos;
 320 }
 321
 322 static inline int ds_tell_pts(demux_stream_t *ds){
 323   return (ds->pts_bytes-ds->buffer_size)+ds->buffer_pos;
 324 }
 325
 326 int demux_read_data(demux_stream_t *ds,unsigned char* mem,int len);
 327 int demux_pattern_3(demux_stream_t *ds, unsigned char *mem, int maxlen,
 328                     int *read, uint32_t pattern);
 329
 330 #define demux_peekc(ds) (\
 331      (likely(ds->buffer_pos<ds->buffer_size)) ? ds->buffer[ds->buffer_pos] \
 332      :((unlikely(!ds_fill_buffer(ds)))? (-1) : ds->buffer[ds->buffer_pos] ) )
 333 #if 1
 334 #define demux_getc(ds) (\
 335      (likely(ds->buffer_pos<ds->buffer_size)) ? ds->buffer[ds->buffer_pos++] \
 336      :((unlikely(!ds_fill_buffer(ds)))? (-1) : ds->buffer[ds->buffer_pos++] ) )
 337 #else
 338 static inline int demux_getc(demux_stream_t *ds){
 339   if(ds->buffer_pos>=ds->buffer_size){
 340     if(!ds_fill_buffer(ds)){
 341 //      printf("DEMUX_GETC: EOF reached!\n");
 342       return -1; // EOF
 343     }
 344   }
 345 //  printf("[%02X]",ds->buffer[ds->buffer_pos]);
 346   return ds->buffer[ds->buffer_pos++];
 347 }
 348 #endif
 349
 350 void ds_free_packs(demux_stream_t *ds);
 351 int ds_get_packet(demux_stream_t *ds,unsigned char **start);
 352 int ds_get_packet_pts(demux_stream_t *ds, unsigned char **start, double *pts);
 353 int ds_get_packet_sub(demux_stream_t *ds,unsigned char **start);
 354 double ds_get_next_pts(demux_stream_t *ds);
 355
 356 // This is defined here because demux_stream_t ins't defined in stream.h
 357 stream_t* new_ds_stream(demux_stream_t *ds);
 358
 359 static inline int avi_stream_id(unsigned int id){
 360   unsigned char *p=(unsigned char *)&id;
 361   unsigned char a,b;
 362 #if WORDS_BIGENDIAN
 363   a=p[3]-'0'; b=p[2]-'0';
 364 #else
 365   a=p[0]-'0'; b=p[1]-'0';
 366 #endif
 367   if(a>9 || b>9) return 100; // invalid ID
 368   return a*10+b;
 369 }
 370
 371 demuxer_t* demux_open(struct MPOpts *opts, stream_t *stream,int file_format,int aid,int vid,int sid,char* filename);
 372 void demux_flush(demuxer_t *demuxer);
 373 int demux_seek(demuxer_t *demuxer,float rel_seek_secs,float audio_delay,int flags);
 374 demuxer_t*  new_demuxers_demuxer(demuxer_t* vd, demuxer_t* ad, demuxer_t* sd);
 375
 376 // AVI demuxer params:
 377 extern int index_mode;  // -1=untouched  0=don't use index  1=use (geneate) index
 378 extern char *index_file_save, *index_file_load;
 379 extern int force_ni;
 380 extern int pts_from_bps;
 381
 382 extern int extension_parsing;
 383
 384 int demux_info_add(demuxer_t *demuxer, const char *opt, const char *param);
 385 char* demux_info_get(demuxer_t *demuxer, const char *opt);
 386 int demux_info_print(demuxer_t *demuxer);
 387 int demux_control(demuxer_t *demuxer, int cmd, void *arg);
 388
 389 #ifdef CONFIG_OGGVORBIS
 390 /* Found in demux_ogg.c */
 391 int demux_ogg_num_subs(demuxer_t *demuxer);
 392 int demux_ogg_sub_id(demuxer_t *demuxer, int index);
 393 #endif
 394
 395 extern int demuxer_get_current_time(demuxer_t *demuxer);
 396 extern double demuxer_get_time_length(demuxer_t *demuxer);
 397 extern int demuxer_get_percent_pos(demuxer_t *demuxer);
 398 extern int demuxer_switch_audio(demuxer_t *demuxer, int index);
 399 extern int demuxer_switch_video(demuxer_t *demuxer, int index);
 400
 401 extern int demuxer_type_by_filename(char* filename);
 402
 403 extern void demuxer_help(void);
 404 extern int get_demuxer_type_from_name(char *demuxer_name, int *force);
 405
 406 int demuxer_add_attachment(demuxer_t* demuxer, const char* name,
 407                            const char* type, const void* data, size_t size);
 408
 409 int demuxer_add_chapter(demuxer_t* demuxer, const char* name, uint64_t start, uint64_t end);
 410 int demuxer_seek_chapter(demuxer_t *demuxer, int chapter, int mode, float *seek_pts, int *num_chapters, char **chapter_name);
 411
 412 /// Get current chapter index if available.
 413 int demuxer_get_current_chapter(demuxer_t *demuxer);
 414 /// Get chapter name by index if available.
 415 char *demuxer_chapter_name(demuxer_t *demuxer, int chapter);
 416 /// Get chapter display name by index.
 417 char *demuxer_chapter_display_name(demuxer_t *demuxer, int chapter);
 418 /// Get chapter start time and end time by index if available.
 419 float demuxer_chapter_time(demuxer_t *demuxer, int chapter, float *end);
 420 /// Get total chapter number.
 421 int demuxer_chapter_count(demuxer_t *demuxer);
 422 /// Get current angle index.
 423 int demuxer_get_current_angle(demuxer_t *demuxer);
 424 /// Set angle.
 425 int demuxer_set_angle(demuxer_t *demuxer, int angle);
 426 /// Get number of angles.
 427 int demuxer_angles_count(demuxer_t *demuxer);
 428
 429 // get the index of a track
 430 // lang is a comma-separated list
 431 int demuxer_audio_track_by_lang(demuxer_t* demuxer, char* lang);
 432 int demuxer_sub_track_by_lang(demuxer_t* demuxer, char* lang);
 433
 434 // find the default track
 435 // for subtitles, it is the first track with default attribute
 436 // for audio, additionally, the first track is selected if no track has default attribute set
 437 int demuxer_default_audio_track(demuxer_t* d);
 438 int demuxer_default_sub_track(demuxer_t* d);
 439
 440 #endif /* MPLAYER_DEMUXER_H */