libmpdemux/demuxer.h

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