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