projects / libav.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
QDM2 compatible decoder
[libav.git] / libavformat / mov.c
CommitLineData
6cea494e
ZK		 1/*
2 * MOV decoder.
19720f15 3 * Copyright (c) 2001 Fabrice Bellard.
6cea494e 4 *
19720f15
FB		 5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2 of the License, or (at your option) any later version.
6cea494e 9 *
19720f15 10 * This library is distributed in the hope that it will be useful,
6cea494e 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19720f15
FB		 12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
6cea494e 14 *
19720f15
FB		 15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
6cea494e 18 */
d957696f
MN		 19
20#include <limits.h>
21
6cea494e
ZK		 22#include "avformat.h"
23#include "avi.h"
24
0147f198
FR		 25#ifdef CONFIG_ZLIB
26#include <zlib.h>
27#endif
28
6cea494e
ZK		 29/*
30 * First version by Francois Revol revol@free.fr
baf25c9d 31 * Seek function by Gael Chardon gael.dev@4now.net
5ca1d879 32 *
6cea494e 33 * Features and limitations:
5ca1d879 34 * - reads most of the QT files I have (at least the structure),
6cea494e 35 * the exceptions are .mov with zlib compressed headers ('cmov' section). It shouldn't be hard to implement.
0147f198 36 * FIXED, Francois Revol, 07/17/2002
6cea494e
ZK		 37 * - ffmpeg has nearly none of the usual QuickTime codecs,
38 * although I succesfully dumped raw and mp3 audio tracks off .mov files.
39 * Sample QuickTime files with mp3 audio can be found at: http://www.3ivx.com/showcase.html
40 * - .mp4 parsing is still hazardous, although the format really is QuickTime with some minor changes
41 * (to make .mov parser crash maybe ?), despite what they say in the MPEG FAQ at
42 * http://mpeg.telecomitalialab.com/faq.htm
43 * - the code is quite ugly... maybe I won't do it recursive next time :-)
baf25c9d 44 * - seek is not supported with files that contain edit list
5ca1d879 45 *
6cea494e
ZK		 46 * Funny I didn't know about http://sourceforge.net/projects/qt-ffmpeg/
47 * when coding this :) (it's a writer anyway)
5ca1d879 48 *
6cea494e
ZK		 49 * Reference documents:
50 * http://www.geocities.com/xhelmboyx/quicktime/formats/qtm-layout.txt
51 * Apple:
baf25c9d 52 * http://developer.apple.com/documentation/QuickTime/QTFF/
15c8dbe7 53 * http://developer.apple.com/documentation/QuickTime/QTFF/qtff.pdf
6cea494e
ZK		 54 * QuickTime is a trademark of Apple (AFAIK :))
55 */
56
c9a65ca8 57//#define DEBUG
9ed83b0a
ZK		 58#ifdef DEBUG
59#include <stdio.h>
60#include <fcntl.h>
61#endif
6cea494e 62
b595afaa
MM		 63#include "qtpalette.h"
64
baf25c9d
GC		 65
66/* Allows seeking (MOV_SPLIT_CHUNKS should also be defined) */
67#define MOV_SEEK
68
3ffe3793
FR		 69/* allows chunk splitting - should work now... */
70/* in case you can't read a file, try commenting */
71#define MOV_SPLIT_CHUNKS
72
14342fd5
JC		 73/* Special handling for movies created with Minolta Dimaxe Xi*/
74/* this fix should not interfere with other .mov files, but just in case*/
75#define MOV_MINOLTA_FIX
76
6cea494e
ZK		 77/* some streams in QT (and in MP4 mostly) aren't either video nor audio */
78/* so we first list them as this, then clean up the list of streams we give back, */
79/* getting rid of these */
b6a17df4 80#define CODEC_TYPE_MOV_OTHER (enum CodecType) 2
6cea494e 81
a266644f 82static const CodecTag mov_video_tags[] = {
6cea494e 83/* { CODEC_ID_, MKTAG('c', 'v', 'i', 'd') }, *//* Cinepak */
0147f198
FR		 84/* { CODEC_ID_H263, MKTAG('r', 'a', 'w', ' ') }, *//* Uncompressed RGB */
85/* { CODEC_ID_H263, MKTAG('Y', 'u', 'v', '2') }, *//* Uncompressed YUV422 */
53e27dd5 86/* { CODEC_ID_RAWVIDEO, MKTAG('A', 'V', 'U', 'I') }, *//* YUV with alpha-channel (AVID Uncompressed) */
6cea494e
ZK		 87/* Graphics */
88/* Animation */
89/* Apple video */
90/* Kodak Photo CD */
3ffe3793 91 { CODEC_ID_MJPEG, MKTAG('j', 'p', 'e', 'g') }, /* PhotoJPEG */
6cea494e 92 { CODEC_ID_MPEG1VIDEO, MKTAG('m', 'p', 'e', 'g') }, /* MPEG */
3ffe3793 93 { CODEC_ID_MJPEG, MKTAG('m', 'j', 'p', 'a') }, /* Motion-JPEG (format A) */
6cea494e 94 { CODEC_ID_MJPEG, MKTAG('m', 'j', 'p', 'b') }, /* Motion-JPEG (format B) */
53e27dd5
SB		 95 { CODEC_ID_MJPEG, MKTAG('A', 'V', 'D', 'J') }, /* MJPEG with alpha-channel (AVID JFIF meridien compressed) */
96/* { CODEC_ID_MJPEG, MKTAG('A', 'V', 'R', 'n') }, *//* MJPEG with alpha-channel (AVID ABVB/Truevision NuVista) */
6cea494e
ZK		 97/* { CODEC_ID_GIF, MKTAG('g', 'i', 'f', ' ') }, *//* embedded gif files as frames (usually one "click to play movie" frame) */
98/* Sorenson video */
0147f198
FR		 99 { CODEC_ID_SVQ1, MKTAG('S', 'V', 'Q', '1') }, /* Sorenson Video v1 */
100 { CODEC_ID_SVQ1, MKTAG('s', 'v', 'q', '1') }, /* Sorenson Video v1 */
3ffe3793 101 { CODEC_ID_SVQ1, MKTAG('s', 'v', 'q', 'i') }, /* Sorenson Video v1 (from QT specs)*/
8b82a956 102 { CODEC_ID_SVQ3, MKTAG('S', 'V', 'Q', '3') }, /* Sorenson Video v3 */
0e7eed09 103 { CODEC_ID_MPEG4, MKTAG('m', 'p', '4', 'v') },
6cea494e 104 { CODEC_ID_MPEG4, MKTAG('D', 'I', 'V', 'X') }, /* OpenDiVX *//* sample files at http://heroinewarrior.com/xmovie.php3 use this tag */
8dafdb88 105 { CODEC_ID_MPEG4, MKTAG('X', 'V', 'I', 'D') },
fcc87242 106 { CODEC_ID_MPEG4, MKTAG('3', 'I', 'V', '2') }, /* experimental: 3IVX files before ivx D4 4.5.1 */
6cea494e 107/* { CODEC_ID_, MKTAG('I', 'V', '5', '0') }, *//* Indeo 5.0 */
0147f198 108 { CODEC_ID_H263, MKTAG('h', '2', '6', '3') }, /* H263 */
25fa62e1 109 { CODEC_ID_H263, MKTAG('s', '2', '6', '3') }, /* H263 ?? works */
e1687cc0
FB		 110 { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'c', ' ') }, /* DV NTSC */
111 { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'c', 'p') }, /* DV PAL */
53e27dd5 112/* { CODEC_ID_DVVIDEO, MKTAG('A', 'V', 'd', 'v') }, *//* AVID dv */
d86053a4 113 { CODEC_ID_VP3, MKTAG('V', 'P', '3', '1') }, /* On2 VP3 */
2fdf638b
MM		 114 { CODEC_ID_RPZA, MKTAG('r', 'p', 'z', 'a') }, /* Apple Video (RPZA) */
115 { CODEC_ID_CINEPAK, MKTAG('c', 'v', 'i', 'd') }, /* Cinepak */
1dc1ed99 116 { CODEC_ID_8BPS, MKTAG('8', 'B', 'P', 'S') }, /* Planar RGB (8BPS) */
b595afaa 117 { CODEC_ID_SMC, MKTAG('s', 'm', 'c', ' ') }, /* Apple Graphics (SMC) */
070ed1bc 118 { CODEC_ID_QTRLE, MKTAG('r', 'l', 'e', ' ') }, /* Apple Animation (RLE) */
d08d7142 119 { CODEC_ID_QDRAW, MKTAG('q', 'd', 'r', 'w') }, /* QuickDraw */
169eb021 120 { CODEC_ID_H264, MKTAG('a', 'v', 'c', '1') }, /* AVC-1/H.264 */
a3075830 121 { CODEC_ID_MPEG2VIDEO, MKTAG('h', 'd', 'v', '2') }, /* MPEG2 produced by Sony HD camera */
5ca1d879 122 { CODEC_ID_NONE, 0 },
6cea494e
ZK		 123};
124
a266644f 125static const CodecTag mov_audio_tags[] = {
6cea494e
ZK		 126/* { CODEC_ID_PCM_S16BE, MKTAG('N', 'O', 'N', 'E') }, *//* uncompressed */
127 { CODEC_ID_PCM_S16BE, MKTAG('t', 'w', 'o', 's') }, /* 16 bits */
5ca1d879 128 /* { CODEC_ID_PCM_S8, MKTAG('t', 'w', 'o', 's') },*/ /* 8 bits */
5cd62665 129 { CODEC_ID_PCM_U8, MKTAG('r', 'a', 'w', ' ') }, /* 8 bits unsigned */
6cea494e
ZK		 130 { CODEC_ID_PCM_S16LE, MKTAG('s', 'o', 'w', 't') }, /* */
131 { CODEC_ID_PCM_MULAW, MKTAG('u', 'l', 'a', 'w') }, /* */
132 { CODEC_ID_PCM_ALAW, MKTAG('a', 'l', 'a', 'w') }, /* */
0147f198 133 { CODEC_ID_ADPCM_IMA_QT, MKTAG('i', 'm', 'a', '4') }, /* IMA-4 ADPCM */
3f95e843
FR		 134 { CODEC_ID_MACE3, MKTAG('M', 'A', 'C', '3') }, /* Macintosh Audio Compression and Expansion 3:1 */
135 { CODEC_ID_MACE6, MKTAG('M', 'A', 'C', '6') }, /* Macintosh Audio Compression and Expansion 6:1 */
6cea494e
ZK		 136
137 { CODEC_ID_MP2, MKTAG('.', 'm', 'p', '3') }, /* MPEG layer 3 */ /* sample files at http://www.3ivx.com/showcase.html use this tag */
138 { CODEC_ID_MP2, 0x6D730055 }, /* MPEG layer 3 */
139 { CODEC_ID_MP2, 0x5500736D }, /* MPEG layer 3 *//* XXX: check endianness */
140/* { CODEC_ID_OGG_VORBIS, MKTAG('O', 'g', 'g', 'S') }, *//* sample files at http://heroinewarrior.com/xmovie.php3 use this tag */
141/* MP4 tags */
f2196640 142 { CODEC_ID_AAC, MKTAG('m', 'p', '4', 'a') }, /* MPEG-4 AAC */
5ca1d879 143 /* The standard for mpeg4 audio is still not normalised AFAIK anyway */
891f64b3 144 { CODEC_ID_AMR_NB, MKTAG('s', 'a', 'm', 'r') }, /* AMR-NB 3gp */
d663a1fd 145 { CODEC_ID_AMR_WB, MKTAG('s', 'a', 'w', 'b') }, /* AMR-WB 3gp */
8dafdb88 146 { CODEC_ID_AC3, MKTAG('m', 's', 0x20, 0x00) }, /* Dolby AC-3 */
6d6d7970 147 { CODEC_ID_ALAC,MKTAG('a', 'l', 'a', 'c') }, /* Apple Lossless */
d9b1c197 148 { CODEC_ID_QDM2,MKTAG('Q', 'D', 'M', '2') }, /* QDM2 */
5ca1d879 149 { CODEC_ID_NONE, 0 },
6cea494e
ZK		 150};
151
152/* the QuickTime file format is quite convoluted...
153 * it has lots of index tables, each indexing something in another one...
154 * Here we just use what is needed to read the chunks
155 */
156
157typedef struct MOV_sample_to_chunk_tbl {
158 long first;
159 long count;
160 long id;
161} MOV_sample_to_chunk_tbl;
162
5cd62665 163typedef struct {
5ca1d879
ZK		 164 uint32_t type;
165 int64_t offset;
166 int64_t size; /* total size (excluding the size and type fields) */
167} MOV_atom_t;
168
169typedef struct {
170 int seed;
171 int flags;
172 int size;
173 void* clrs;
174} MOV_ctab_t;
175
176typedef struct {
5cd62665
ZK		 177 uint8_t version;
178 uint32_t flags; // 24bit
179
180 /* 0x03 ESDescrTag */
181 uint16_t es_id;
182#define MP4ODescrTag 0x01
183#define MP4IODescrTag 0x02
184#define MP4ESDescrTag 0x03
185#define MP4DecConfigDescrTag 0x04
186#define MP4DecSpecificDescrTag 0x05
187#define MP4SLConfigDescrTag 0x06
188#define MP4ContentIdDescrTag 0x07
189#define MP4SupplContentIdDescrTag 0x08
190#define MP4IPIPtrDescrTag 0x09
191#define MP4IPMPPtrDescrTag 0x0A
192#define MP4IPMPDescrTag 0x0B
193#define MP4RegistrationDescrTag 0x0D
194#define MP4ESIDIncDescrTag 0x0E
195#define MP4ESIDRefDescrTag 0x0F
196#define MP4FileIODescrTag 0x10
197#define MP4FileODescrTag 0x11
198#define MP4ExtProfileLevelDescrTag 0x13
199#define MP4ExtDescrTagsStart 0x80
200#define MP4ExtDescrTagsEnd 0xFE
201 uint8_t stream_priority;
202
203 /* 0x04 DecConfigDescrTag */
204 uint8_t object_type_id;
205 uint8_t stream_type;
206 /* XXX: really streamType is
207 * only 6bit, followed by:
208 * 1bit upStream
209 * 1bit reserved
210 */
211 uint32_t buffer_size_db; // 24
212 uint32_t max_bitrate;
213 uint32_t avg_bitrate;
214
215 /* 0x05 DecSpecificDescrTag */
216 uint8_t decoder_cfg_len;
217 uint8_t *decoder_cfg;
218
219 /* 0x06 SLConfigDescrTag */
220 uint8_t sl_config_len;
221 uint8_t *sl_config;
222} MOV_esds_t;
223
b6a17df4
ZK		 224struct MOVParseTableEntry;
225
cd461d48
MN		 226typedef struct Time2Sample{
227 int count;
228 int duration;
229}Time2Sample;
230
6cea494e
ZK		 231typedef struct MOVStreamContext {
232 int ffindex; /* the ffmpeg stream id */
233 int is_ff_stream; /* Is this stream presented to ffmpeg ? i.e. is this an audio or video stream ? */
234 long next_chunk;
235 long chunk_count;
0c1a9eda 236 int64_t *chunk_offsets;
cd461d48
MN		 237 int stts_count;
238 Time2Sample *stts_data;
c4ac052b
MN		 239 int ctts_count;
240 Time2Sample *ctts_data;
cd461d48 241 int edit_count; /* number of 'edit' (elst atom) */
6cea494e
ZK		 242 long sample_to_chunk_sz;
243 MOV_sample_to_chunk_tbl *sample_to_chunk;
3ffe3793 244 long sample_to_chunk_index;
7fe5a3c0
GC		 245 int sample_to_time_index;
246 long sample_to_time_sample;
247 uint64_t sample_to_time_time;
c4ac052b
MN		 248 int sample_to_ctime_index;
249 int sample_to_ctime_sample;
6cea494e
ZK		 250 long sample_size;
251 long sample_count;
252 long *sample_sizes;
247d56f5
BB		 253 long keyframe_count;
254 long *keyframes;
5cd62665 255 int time_scale;
3ffe3793
FR		 256 long current_sample;
257 long left_in_chunk; /* how many samples before next chunk */
5ca1d879 258 MOV_esds_t esds;
6cea494e
ZK		 259} MOVStreamContext;
260
261typedef struct MOVContext {
262 int mp4; /* set to 1 as soon as we are sure that the file is an .mp4 file (even some header parsing depends on this) */
263 AVFormatContext *fc;
5cd62665
ZK		 264 int time_scale;
265 int duration; /* duration of the longest track */
6cea494e
ZK		 266 int found_moov; /* when both 'moov' and 'mdat' sections has been found */
267 int found_mdat; /* we suppose we have enough data to read the file */
0c1a9eda
ZK		 268 int64_t mdat_size;
269 int64_t mdat_offset;
86d8602f 270 int ni; ///< non interleaved mode
6cea494e
ZK		 271 int total_streams;
272 /* some streams listed here aren't presented to the ffmpeg API, since they aren't either video nor audio
273 * but we need the info to be able to skip data from those streams in the 'mdat' section
274 */
275 MOVStreamContext *streams[MAX_STREAMS];
5cd62665 276
0c1a9eda 277 int64_t next_chunk_offset;
5cd62665 278 MOVStreamContext *partial; /* != 0 : there is still to read in the current chunk */
5ca1d879
ZK		 279 int ctab_size;
280 MOV_ctab_t **ctab; /* color tables */
b6a17df4
ZK		 281 const struct MOVParseTableEntry *parse_table; /* could be eventually used to change the table */
282 /* NOTE: for recursion save to/ restore from local variable! */
b595afaa
MM		 283
284 AVPaletteControl palette_control;
6cea494e
ZK		 285} MOVContext;
286
287
6cea494e
ZK		 288/* XXX: it's the first time I make a recursive parser I think... sorry if it's ugly :P */
289
290/* those functions parse an atom */
291/* return code:
292 1: found what I wanted, exit
293 0: continue to parse next atom
294 -1: error occured, exit
295 */
5ca1d879 296typedef int (*mov_parse_function)(MOVContext *ctx, ByteIOContext *pb, MOV_atom_t atom);
6cea494e
ZK		 297
298/* links atom IDs to parse functions */
299typedef struct MOVParseTableEntry {
0c1a9eda 300 uint32_t type;
6cea494e
ZK		 301 mov_parse_function func;
302} MOVParseTableEntry;
303
5ca1d879
ZK		 304#ifdef DEBUG
305/*
306 * XXX: static sux, even more in a multithreaded environment...
307 * Avoid them. This is here just to help debugging.
308 */
309static int debug_indent = 0;
310void print_atom(const char *str, MOV_atom_t atom)
311{
312 unsigned int tag, i;
313 tag = (unsigned int) atom.type;
314 i=debug_indent;
315 if(tag == 0) tag = MKTAG('N', 'U', 'L', 'L');
316 while(i--)
baf25c9d
GC		 317 av_log(NULL, AV_LOG_DEBUG, "|");
318 av_log(NULL, AV_LOG_DEBUG, "parse:");
319 av_log(NULL, AV_LOG_DEBUG, " %s: tag=%c%c%c%c offset=0x%x size=0x%x\n",
5ca1d879
ZK		 320 str, tag & 0xff,
321 (tag >> 8) & 0xff,
322 (tag >> 16) & 0xff,
323 (tag >> 24) & 0xff,
324 (unsigned int)atom.offset,
325 (unsigned int)atom.size);
326 assert((unsigned int)atom.size < 0x7fffffff);// catching errors
327}
328#else
329#define print_atom(a,b)
330#endif
331
332
333static int mov_read_leaf(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
6cea494e 334{
5ca1d879 335 print_atom("leaf", atom);
b6a17df4 336
5ca1d879
ZK		 337 if (atom.size>1)
338 url_fskip(pb, atom.size);
339/* url_seek(pb, atom_offset+atom.size, SEEK_SET); */
6cea494e
ZK		 340 return 0;
341}
342
5ca1d879 343static int mov_read_default(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
6cea494e 344{
b6a17df4 345 int64_t total_size = 0;
5ca1d879 346 MOV_atom_t a;
6cea494e
ZK		 347 int i;
348 int err = 0;
5ca1d879 349
6cea494e 350#ifdef DEBUG
5ca1d879 351 print_atom("default", atom);
6cea494e
ZK		 352 debug_indent++;
353#endif
5cd62665 354
5ca1d879 355 a.offset = atom.offset;
6cea494e 356
9ed83b0a
ZK		 357 if (atom.size < 0)
358 atom.size = 0x7fffffffffffffffLL;
5ca1d879
ZK		 359 while(((total_size + 8) < atom.size) && !url_feof(pb) && !err) {
360 a.size = atom.size;
361 a.type=0L;
362 if(atom.size >= 8) {
363 a.size = get_be32(pb);
364 a.type = get_le32(pb);
6cea494e 365 }
5cd62665 366 total_size += 8;
5ca1d879 367 a.offset += 8;
baf25c9d 368 //av_log(NULL, AV_LOG_DEBUG, "type: %08x %.4s sz: %Lx %Lx %Lx\n", type, (char*)&type, size, atom.size, total_size);
5ca1d879
ZK		 369 if (a.size == 1) { /* 64 bit extended size */
370 a.size = get_be64(pb) - 8;
371 a.offset += 8;
372 total_size += 8;
6cea494e 373 }
5ca1d879
ZK		 374 if (a.size == 0) {
375 a.size = atom.size - total_size;
376 if (a.size <= 8)
5cd62665
ZK		 377 break;
378 }
5ca1d879
ZK		 379 for (i = 0; c->parse_table[i].type != 0L
380 && c->parse_table[i].type != a.type; i++)
5cd62665
ZK		 381 /* empty */;
382
5ca1d879 383 a.size -= 8;
568e18b1
MN		 384
385 if(a.size < 0)
386 break;
387
baf25c9d 388// av_log(NULL, AV_LOG_DEBUG, " i=%ld\n", i);
b6a17df4 389 if (c->parse_table[i].type == 0) { /* skip leaf atoms data */
5ca1d879 390// url_seek(pb, atom.offset+atom.size, SEEK_SET);
6cea494e 391#ifdef DEBUG
5ca1d879 392 print_atom("unknown", a);
6cea494e 393#endif
5ca1d879 394 url_fskip(pb, a.size);
b6a17df4
ZK		 395 } else {
396#ifdef DEBUG
397 //char b[5] = { type & 0xff, (type >> 8) & 0xff, (type >> 16) & 0xff, (type >> 24) & 0xff, 0 };
398 //print_atom(b, type, offset, size);
399#endif
5ca1d879 400 err = (c->parse_table[i].func)(c, pb, a);
b6a17df4 401 }
6cea494e 402
5ca1d879
ZK		 403 a.offset += a.size;
404 total_size += a.size;
6cea494e
ZK		 405 }
406
5ca1d879 407 if (!err && total_size < atom.size && atom.size < 0x7ffff) {
baf25c9d 408 //av_log(NULL, AV_LOG_DEBUG, "RESET %Ld %Ld err:%d\n", atom.size, total_size, err);
5ca1d879 409 url_fskip(pb, atom.size - total_size);
5cd62665
ZK		 410 }
411
6cea494e
ZK		 412#ifdef DEBUG
413 debug_indent--;
414#endif
415 return err;
416}
417
5ca1d879 418static int mov_read_ctab(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
6cea494e 419{
568e18b1
MN		 420#if 1
421 url_fskip(pb, atom.size); // for now
422#else
423 VERY VERY BROKEN, NEVER execute this, needs rewrite
88730be6
MR		 424 unsigned int len;
425 MOV_ctab_t *t;
5ca1d879
ZK		 426 c->ctab = av_realloc(c->ctab, ++c->ctab_size);
427 t = c->ctab[c->ctab_size];
428 t->seed = get_be32(pb);
429 t->flags = get_be16(pb);
430 t->size = get_be16(pb) + 1;
431 len = 2 * t->size * 4;
432 if (len > 0) {
433 t->clrs = av_malloc(len); // 16bit A R G B
434 if (t->clrs)
435 get_buffer(pb, t->clrs, len);
3ffe3793 436 }
568e18b1 437#endif
5cd62665 438
0147f198
FR		 439 return 0;
440}
441
5ca1d879 442static int mov_read_hdlr(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
6cea494e 443{
5ca1d879 444 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
3ffe3793 445 int len = 0;
0c1a9eda 446 uint32_t type;
0c1a9eda 447 uint32_t ctype;
b6a17df4 448
5ca1d879 449 print_atom("hdlr", atom);
6cea494e
ZK		 450
451 get_byte(pb); /* version */
452 get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
453
454 /* component type */
455 ctype = get_le32(pb);
456 type = get_le32(pb); /* component subtype */
457
458#ifdef DEBUG
baf25c9d
GC		 459 av_log(NULL, AV_LOG_DEBUG, "ctype= %c%c%c%c (0x%08lx)\n", *((char *)&ctype), ((char *)&ctype)[1], ((char *)&ctype)[2], ((char *)&ctype)[3], (long) ctype);
460 av_log(NULL, AV_LOG_DEBUG, "stype= %c%c%c%c\n", *((char *)&type), ((char *)&type)[1], ((char *)&type)[2], ((char *)&type)[3]);
6cea494e
ZK		 461#endif
462#ifdef DEBUG
463/* XXX: yeah this is ugly... */
464 if(ctype == MKTAG('m', 'h', 'l', 'r')) { /* MOV */
465 if(type == MKTAG('v', 'i', 'd', 'e'))
466 puts("hdlr: vide");
467 else if(type == MKTAG('s', 'o', 'u', 'n'))
468 puts("hdlr: soun");
469 } else if(ctype == 0) { /* MP4 */
470 if(type == MKTAG('v', 'i', 'd', 'e'))
471 puts("hdlr: vide");
472 else if(type == MKTAG('s', 'o', 'u', 'n'))
473 puts("hdlr: soun");
474 else if(type == MKTAG('o', 'd', 's', 'm'))
475 puts("hdlr: odsm");
476 else if(type == MKTAG('s', 'd', 's', 'm'))
477 puts("hdlr: sdsm");
478 } else puts("hdlr: meta");
479#endif
480
481 if(ctype == MKTAG('m', 'h', 'l', 'r')) { /* MOV */
0147f198
FR		 482 /* helps parsing the string hereafter... */
483 c->mp4 = 0;
6cea494e 484 if(type == MKTAG('v', 'i', 'd', 'e'))
01f4895c 485 st->codec->codec_type = CODEC_TYPE_VIDEO;
6cea494e 486 else if(type == MKTAG('s', 'o', 'u', 'n'))
01f4895c 487 st->codec->codec_type = CODEC_TYPE_AUDIO;
6cea494e 488 } else if(ctype == 0) { /* MP4 */
0147f198
FR		 489 /* helps parsing the string hereafter... */
490 c->mp4 = 1;
6cea494e 491 if(type == MKTAG('v', 'i', 'd', 'e'))
01f4895c 492 st->codec->codec_type = CODEC_TYPE_VIDEO;
6cea494e 493 else if(type == MKTAG('s', 'o', 'u', 'n'))
01f4895c 494 st->codec->codec_type = CODEC_TYPE_AUDIO;
6cea494e
ZK		 495 }
496 get_be32(pb); /* component manufacture */
497 get_be32(pb); /* component flags */
498 get_be32(pb); /* component flags mask */
499
5ca1d879 500 if(atom.size <= 24)
6cea494e
ZK		 501 return 0; /* nothing left to read */
502 /* XXX: MP4 uses a C string, not a pascal one */
503 /* component name */
0147f198
FR		 504
505 if(c->mp4) {
506 /* .mp4: C string */
5ca1d879 507 while(get_byte(pb) && (++len < (atom.size - 24)));
0147f198
FR		 508 } else {
509 /* .mov: PASCAL string */
9ed83b0a
ZK		 510#ifdef DEBUG
511 char* buf;
512#endif
0147f198 513 len = get_byte(pb);
5ca1d879 514#ifdef DEBUG
b6a17df4
ZK		 515 buf = (uint8_t*) av_malloc(len+1);
516 if (buf) {
517 get_buffer(pb, buf, len);
b6a17df4 518 buf[len] = '\0';
baf25c9d 519 av_log(NULL, AV_LOG_DEBUG, "**buf='%s'\n", buf);
b6a17df4
ZK		 520 av_free(buf);
521 } else
6cea494e 522#endif
5ca1d879 523 url_fskip(pb, len);
6cea494e 524 }
5cd62665 525
3c13647a 526 url_fskip(pb, atom.size - (url_ftell(pb) - atom.offset));
6cea494e
ZK		 527 return 0;
528}
529
5ca1d879 530static int mov_mp4_read_descr_len(ByteIOContext *pb)
0e7eed09 531{
5cd62665 532 int len = 0;
5ca1d879
ZK		 533 int count = 4;
534 while (count--) {
5cd62665
ZK		 535 int c = get_byte(pb);
536 len = (len << 7) | (c & 0x7f);
5ca1d879
ZK		 537 if (!(c & 0x80))
538 break;
0e7eed09
FB		 539 }
540 return len;
541}
542
5ca1d879 543static int mov_mp4_read_descr(ByteIOContext *pb, int *tag)
0e7eed09
FB		 544{
545 int len;
546 *tag = get_byte(pb);
5ca1d879 547 len = mov_mp4_read_descr_len(pb);
0e7eed09 548#ifdef DEBUG
baf25c9d 549 av_log(NULL, AV_LOG_DEBUG, "MPEG4 description: tag=0x%02x len=%d\n", *tag, len);
0e7eed09
FB		 550#endif
551 return len;
552}
553
5ca1d879 554static int mov_read_esds(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
5cd62665 555{
5cd62665
ZK		 556 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
557 MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
5ca1d879 558 int64_t start_pos = url_ftell(pb);
5cd62665 559 int tag, len;
b6a17df4 560
5ca1d879 561 print_atom("esds", atom);
5cd62665
ZK		 562
563 /* Well, broken but suffisant for some MP4 streams */
564 get_be32(pb); /* version + flags */
5ca1d879 565 len = mov_mp4_read_descr(pb, &tag);
5cd62665
ZK		 566 if (tag == MP4ESDescrTag) {
567 get_be16(pb); /* ID */
568 get_byte(pb); /* priority */
569 } else
570 get_be16(pb); /* ID */
571
5ca1d879 572 len = mov_mp4_read_descr(pb, &tag);
5cd62665
ZK		 573 if (tag == MP4DecConfigDescrTag) {
574 sc->esds.object_type_id = get_byte(pb);
5ca1d879
ZK		 575 sc->esds.stream_type = get_byte(pb);
576 sc->esds.buffer_size_db = get_be24(pb);
5cd62665
ZK		 577 sc->esds.max_bitrate = get_be32(pb);
578 sc->esds.avg_bitrate = get_be32(pb);
579
5ca1d879 580 len = mov_mp4_read_descr(pb, &tag);
baf25c9d 581 //av_log(NULL, AV_LOG_DEBUG, "LEN %d TAG %d m:%d a:%d\n", len, tag, sc->esds.max_bitrate, sc->esds.avg_bitrate);
5cd62665
ZK		 582 if (tag == MP4DecSpecificDescrTag) {
583#ifdef DEBUG
baf25c9d 584 av_log(NULL, AV_LOG_DEBUG, "Specific MPEG4 header len=%d\n", len);
5cd62665 585#endif
01f4895c
MN		 586 st->codec->extradata = (uint8_t*) av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);
587 if (st->codec->extradata) {
588 get_buffer(pb, st->codec->extradata, len);
589 st->codec->extradata_size = len;
5cd62665
ZK		 590 }
591 }
592 }
593 /* in any case, skip garbage */
5ca1d879 594 url_fskip(pb, atom.size - ((url_ftell(pb) - start_pos)));
5cd62665
ZK		 595 return 0;
596}
597
5ca1d879
ZK		 598/* this atom contains actual media data */
599static int mov_read_mdat(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
6cea494e 600{
5ca1d879
ZK		 601 print_atom("mdat", atom);
602
603 if(atom.size == 0) /* wrong one (MP4) */
604 return 0;
605 c->found_mdat=1;
606 c->mdat_offset = atom.offset;
607 c->mdat_size = atom.size;
608 if(c->found_moov)
609 return 1; /* found both, just go */
610 url_fskip(pb, atom.size);
611 return 0; /* now go for moov */
612}
613
614/* this atom should contain all header atoms */
615static int mov_read_moov(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
616{
617 int err;
b6a17df4 618
5ca1d879 619 print_atom("moov", atom);
b6a17df4 620
5ca1d879
ZK		 621 err = mov_read_default(c, pb, atom);
622 /* we parsed the 'moov' atom, we can terminate the parsing as soon as we find the 'mdat' */
623 /* so we don't parse the whole file if over a network */
624 c->found_moov=1;
625 if(c->found_mdat)
626 return 1; /* found both, just go */
627 return 0; /* now go for mdat */
628}
629
630
631static int mov_read_mdhd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
632{
5ca1d879
ZK		 633 print_atom("mdhd", atom);
634
635 get_byte(pb); /* version */
636
637 get_byte(pb); get_byte(pb);
638 get_byte(pb); /* flags */
639
640 get_be32(pb); /* creation time */
641 get_be32(pb); /* modification time */
642
baf25c9d 643 c->streams[c->fc->nb_streams-1]->time_scale = get_be32(pb);
da7272b3 644 av_set_pts_info(c->fc->streams[c->fc->nb_streams-1], 64, 1, c->streams[c->fc->nb_streams-1]->time_scale);
5ca1d879
ZK		 645
646#ifdef DEBUG
baf25c9d 647 av_log(NULL, AV_LOG_DEBUG, "track[%i].time_scale = %i\n", c->fc->nb_streams-1, c->streams[c->fc->nb_streams-1]->time_scale); /* time scale */
5ca1d879 648#endif
961e0ccd 649 c->fc->streams[c->fc->nb_streams-1]->duration= get_be32(pb); /* duration */
5ca1d879
ZK		 650
651 get_be16(pb); /* language */
652 get_be16(pb); /* quality */
653
654 return 0;
655}
656
657static int mov_read_mvhd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
658{
659 print_atom("mvhd", atom);
660
661 get_byte(pb); /* version */
662 get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
663
664 get_be32(pb); /* creation time */
665 get_be32(pb); /* modification time */
666 c->time_scale = get_be32(pb); /* time scale */
667#ifdef DEBUG
baf25c9d 668 av_log(NULL, AV_LOG_DEBUG, "time scale = %i\n", c->time_scale);
5ca1d879
ZK		 669#endif
670 c->duration = get_be32(pb); /* duration */
671 get_be32(pb); /* preferred scale */
672
673 get_be16(pb); /* preferred volume */
674
675 url_fskip(pb, 10); /* reserved */
676
677 url_fskip(pb, 36); /* display matrix */
678
679 get_be32(pb); /* preview time */
680 get_be32(pb); /* preview duration */
681 get_be32(pb); /* poster time */
682 get_be32(pb); /* selection time */
683 get_be32(pb); /* selection duration */
684 get_be32(pb); /* current time */
685 get_be32(pb); /* next track ID */
686
687 return 0;
688}
689
9ed83b0a
ZK		 690static int mov_read_smi(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
691{
692 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
693
568e18b1
MN		 694 if((uint64_t)atom.size > (1<<30))
695 return -1;
696
9ed83b0a
ZK		 697 // currently SVQ3 decoder expect full STSD header - so let's fake it
698 // this should be fixed and just SMI header should be passed
01f4895c
MN		 699 av_free(st->codec->extradata);
700 st->codec->extradata_size = 0x5a + atom.size;
701 st->codec->extradata = (uint8_t*) av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
702
703 if (st->codec->extradata) {
704 strcpy(st->codec->extradata, "SVQ3"); // fake
705 get_buffer(pb, st->codec->extradata + 0x5a, atom.size);
706 //av_log(NULL, AV_LOG_DEBUG, "Reading SMI %Ld %s\n", atom.size, (char*)st->codec->extradata + 0x5a);
9ed83b0a
ZK		 707 } else
708 url_fskip(pb, atom.size);
709
710 return 0;
711}
5ca1d879 712
d9b1c197
RT		 713static int mov_read_wave(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
714{
715 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
716
717 if((uint64_t)atom.size > (1<<30))
718 return -1;
719
720 // pass all frma atom to codec, needed at least for QDM2
721 av_free(st->codec->extradata);
722 st->codec->extradata_size = atom.size;
723 st->codec->extradata = (uint8_t*) av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
724
725 if (st->codec->extradata) {
726 get_buffer(pb, st->codec->extradata, atom.size);
727 //av_log(NULL, AV_LOG_DEBUG, "Reading frma %Ld %s\n", atom.size, (char*)st->codec->extradata);
728 } else
729 url_fskip(pb, atom.size);
730
731 return 0;
732}
733
169eb021
MM		 734static int mov_read_avcC(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
735{
736 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
737
568e18b1
MN		 738 if((uint64_t)atom.size > (1<<30))
739 return -1;
740
01f4895c 741 av_free(st->codec->extradata);
169eb021 742
01f4895c
MN		 743 st->codec->extradata_size = atom.size;
744 st->codec->extradata = (uint8_t*) av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
169eb021 745
01f4895c
MN		 746 if (st->codec->extradata) {
747 get_buffer(pb, st->codec->extradata, atom.size);
169eb021
MM		 748 } else
749 url_fskip(pb, atom.size);
750
751 return 0;
752}
753
5ca1d879
ZK		 754static int mov_read_stco(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
755{
756 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
757 MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
568e18b1 758 unsigned int i, entries;
5ca1d879
ZK		 759
760 print_atom("stco", atom);
761
762 get_byte(pb); /* version */
763 get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
764
765 entries = get_be32(pb);
568e18b1
MN		 766
767 if(entries >= UINT_MAX/sizeof(int64_t))
768 return -1;
769
5ca1d879
ZK		 770 sc->chunk_count = entries;
771 sc->chunk_offsets = (int64_t*) av_malloc(entries * sizeof(int64_t));
772 if (!sc->chunk_offsets)
773 return -1;
774 if (atom.type == MKTAG('s', 't', 'c', 'o')) {
775 for(i=0; i<entries; i++) {
776 sc->chunk_offsets[i] = get_be32(pb);
777 }
778 } else if (atom.type == MKTAG('c', 'o', '6', '4')) {
779 for(i=0; i<entries; i++) {
780 sc->chunk_offsets[i] = get_be64(pb);
781 }
782 } else
783 return -1;
86d8602f
MN		 784
785 for(i=0; i<c->fc->nb_streams; i++){
786 MOVStreamContext *sc2 = (MOVStreamContext *)c->fc->streams[i]->priv_data;
45139adf
MN		 787 if(sc2 && sc2->chunk_offsets){
788 int64_t first= sc2->chunk_offsets[0];
789 int64_t last= sc2->chunk_offsets[sc2->chunk_count-1];
790 if(first >= sc->chunk_offsets[entries-1] || last <= sc->chunk_offsets[0])
791 c->ni=1;
792 }
86d8602f 793 }
5ca1d879
ZK		 794#ifdef DEBUG
795/*
796 for(i=0; i<entries; i++) {
baf25c9d 797 av_log(NULL, AV_LOG_DEBUG, "chunk offset=0x%Lx\n", sc->chunk_offsets[i]);
5ca1d879
ZK		 798 }
799*/
800#endif
801 return 0;
802}
803
804static int mov_read_stsd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
805{
806 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
302c389e 807 MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
5ca1d879
ZK		 808 int entries, frames_per_sample;
809 uint32_t format;
53ffdd14 810 uint8_t codec_name[32];
5ca1d879 811
b595afaa
MM		 812 /* for palette traversal */
813 int color_depth;
814 int color_start;
815 int color_count;
816 int color_end;
817 int color_index;
818 int color_dec;
819 int color_greyscale;
820 unsigned char *color_table;
821 int j;
822 unsigned char r, g, b;
823
5ca1d879 824 print_atom("stsd", atom);
6cea494e
ZK		 825
826 get_byte(pb); /* version */
827 get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
828
6cea494e
ZK		 829 entries = get_be32(pb);
830
891f64b3 831 while(entries--) { //Parsing Sample description table
b6a17df4 832 enum CodecID id;
5ca1d879 833 int size = get_be32(pb); /* size */
6cea494e 834 format = get_le32(pb); /* data format */
5cd62665 835
6cea494e
ZK		 836 get_be32(pb); /* reserved */
837 get_be16(pb); /* reserved */
838 get_be16(pb); /* index */
0e7eed09
FB		 839
840 /* for MPEG4: set codec type by looking for it */
841 id = codec_get_id(mov_video_tags, format);
842 if (id >= 0) {
843 AVCodec *codec;
b6a17df4 844 codec = avcodec_find_decoder(id);
0e7eed09 845 if (codec)
01f4895c 846 st->codec->codec_type = codec->type;
0e7eed09
FB		 847 }
848#ifdef DEBUG
baf25c9d 849 av_log(NULL, AV_LOG_DEBUG, "size=%d 4CC= %c%c%c%c codec_type=%d\n",
5cd62665 850 size,
0e7eed09
FB		 851 (format >> 0) & 0xff,
852 (format >> 8) & 0xff,
853 (format >> 16) & 0xff,
854 (format >> 24) & 0xff,
01f4895c 855 st->codec->codec_type);
0e7eed09 856#endif
01f4895c
MN		 857 st->codec->codec_tag = format;
858 if(st->codec->codec_type==CODEC_TYPE_VIDEO) {
5ca1d879 859 MOV_atom_t a = { 0, 0, 0 };
01f4895c 860 st->codec->codec_id = id;
6cea494e
ZK		 861 get_be16(pb); /* version */
862 get_be16(pb); /* revision level */
863 get_be32(pb); /* vendor */
864 get_be32(pb); /* temporal quality */
865 get_be32(pb); /* spacial quality */
01f4895c 866 if(st->codec->codec_id == CODEC_ID_MPEG4){ //FIXME this is silly
d57b7316
MN		 867 get_be16(pb);
868 get_be16(pb);
869 }else{
01f4895c
MN		 870 st->codec->width = get_be16(pb); /* width */
871 st->codec->height = get_be16(pb); /* height */
0e7eed09 872 }
6cea494e
ZK		 873 get_be32(pb); /* horiz resolution */
874 get_be32(pb); /* vert resolution */
875 get_be32(pb); /* data size, always 0 */
5cd62665 876 frames_per_sample = get_be16(pb); /* frames per samples */
6cea494e 877#ifdef DEBUG
baf25c9d 878 av_log(NULL, AV_LOG_DEBUG, "frames/samples = %d\n", frames_per_sample);
6cea494e 879#endif
53ffdd14
RG		 880 get_buffer(pb, codec_name, 32); /* codec name, pascal string (FIXME: true for mp4?) */
881 if (codec_name[0] <= 31) {
01f4895c
MN		 882 memcpy(st->codec->codec_name, &codec_name[1],codec_name[0]);
883 st->codec->codec_name[codec_name[0]] = 0;
53ffdd14 884 }
5cd62665 885
01f4895c
MN		 886 st->codec->bits_per_sample = get_be16(pb); /* depth */
887 st->codec->color_table_id = get_be16(pb); /* colortable id */
6cea494e 888
14342fd5 889/* These are set in mov_read_stts and might already be set!
01f4895c
MN		 890 st->codec->time_base.den = 25;
891 st->codec->time_base.num = 1;
14342fd5 892*/
5ca1d879 893 size -= (16+8*4+2+32+2*2);
5cd62665
ZK		 894#if 0
895 while (size >= 8) {
5ca1d879 896 MOV_atom_t a;
0c1a9eda 897 int64_t start_pos;
5cd62665 898
5ca1d879
ZK		 899 a.size = get_be32(pb);
900 a.type = get_le32(pb);
5cd62665 901 size -= 8;
0e7eed09 902#ifdef DEBUG
baf25c9d 903 av_log(NULL, AV_LOG_DEBUG, "VIDEO: atom_type=%c%c%c%c atom.size=%Ld size_left=%d\n",
5ca1d879
ZK		 904 (a.type >> 0) & 0xff,
905 (a.type >> 8) & 0xff,
906 (a.type >> 16) & 0xff,
907 (a.type >> 24) & 0xff,
908 a.size, size);
0e7eed09
FB		 909#endif
910 start_pos = url_ftell(pb);
911
5ca1d879 912 switch(a.type) {
0e7eed09
FB		 913 case MKTAG('e', 's', 'd', 's'):
914 {
915 int tag, len;
916 /* Well, broken but suffisant for some MP4 streams */
917 get_be32(pb); /* version + flags */
5ca1d879 918 len = mov_mp4_read_descr(pb, &tag);
0e7eed09
FB		 919 if (tag == 0x03) {
920 /* MP4ESDescrTag */
921 get_be16(pb); /* ID */
922 get_byte(pb); /* priority */
5ca1d879 923 len = mov_mp4_read_descr(pb, &tag);
0e7eed09
FB		 924 if (tag != 0x04)
925 goto fail;
926 /* MP4DecConfigDescrTag */
927 get_byte(pb); /* objectTypeId */
928 get_be32(pb); /* streamType + buffer size */
5cd62665 929 get_be32(pb); /* max bit rate */
0e7eed09 930 get_be32(pb); /* avg bit rate */
891f64b3 931 len = mov_mp4_read_descr(pb, &tag);
0e7eed09
FB		 932 if (tag != 0x05)
933 goto fail;
934 /* MP4DecSpecificDescrTag */
935#ifdef DEBUG
baf25c9d 936 av_log(NULL, AV_LOG_DEBUG, "Specific MPEG4 header len=%d\n", len);
0e7eed09
FB		 937#endif
938 sc->header_data = av_mallocz(len);
939 if (sc->header_data) {
940 get_buffer(pb, sc->header_data, len);
5cd62665 941 sc->header_len = len;
0e7eed09
FB		 942 }
943 }
944 /* in any case, skip garbage */
945 }
946 break;
947 default:
948 break;
949 }
5cd62665 950 fail:
baf25c9d 951 av_log(NULL, AV_LOG_DEBUG, "ATOMENEWSIZE %Ld %d\n", atom.size, url_ftell(pb) - start_pos);
5ca1d879
ZK		 952 if (atom.size > 8) {
953 url_fskip(pb, (atom.size - 8) -
5cd62665 954 ((url_ftell(pb) - start_pos)));
5ca1d879 955 size -= atom.size - 8;
5cd62665
ZK		 956 }
957 }
0e7eed09
FB		 958 if (size > 0) {
959 /* unknown extension */
960 url_fskip(pb, size);
961 }
5cd62665 962#else
b595afaa
MM		 963
964 /* figure out the palette situation */
01f4895c
MN		 965 color_depth = st->codec->bits_per_sample & 0x1F;
966 color_greyscale = st->codec->bits_per_sample & 0x20;
b595afaa
MM		 967
968 /* if the depth is 2, 4, or 8 bpp, file is palettized */
969 if ((color_depth == 2) || (color_depth == 4) ||
970 (color_depth == 8)) {
971
972 if (color_greyscale) {
973
974 /* compute the greyscale palette */
975 color_count = 1 << color_depth;
976 color_index = 255;
977 color_dec = 256 / (color_count - 1);
978 for (j = 0; j < color_count; j++) {
979 r = g = b = color_index;
980 c->palette_control.palette[j] =
981 (r << 16) | (g << 8) | (b);
982 color_index -= color_dec;
983 if (color_index < 0)
984 color_index = 0;
985 }
986
01f4895c 987 } else if (st->codec->color_table_id & 0x08) {
b595afaa
MM		 988
989 /* if flag bit 3 is set, use the default palette */
990 color_count = 1 << color_depth;
991 if (color_depth == 2)
a90466f7 992 color_table = ff_qt_default_palette_4;
b595afaa 993 else if (color_depth == 4)
a90466f7 994 color_table = ff_qt_default_palette_16;
b595afaa 995 else
a90466f7 996 color_table = ff_qt_default_palette_256;
b595afaa
MM		 997
998 for (j = 0; j < color_count; j++) {
999 r = color_table[j * 4 + 0];
1000 g = color_table[j * 4 + 1];
1001 b = color_table[j * 4 + 2];
1002 c->palette_control.palette[j] =
1003 (r << 16) | (g << 8) | (b);
1004 }
1005
1006 } else {
1007
1008 /* load the palette from the file */
1009 color_start = get_be32(pb);
1010 color_count = get_be16(pb);
1011 color_end = get_be16(pb);
1012 for (j = color_start; j <= color_end; j++) {
1013 /* each R, G, or B component is 16 bits;
1014 * only use the top 8 bits; skip alpha bytes
1015 * up front */
1016 get_byte(pb);
1017 get_byte(pb);
1018 r = get_byte(pb);
1019 get_byte(pb);
1020 g = get_byte(pb);
1021 get_byte(pb);
1022 b = get_byte(pb);
1023 get_byte(pb);
1024 c->palette_control.palette[j] =
1025 (r << 16) | (g << 8) | (b);
1026 }
1027 }
1028
01f4895c
MN		 1029 st->codec->palctrl = &c->palette_control;
1030 st->codec->palctrl->palette_changed = 1;
b595afaa 1031 } else
01f4895c 1032 st->codec->palctrl = NULL;
b595afaa 1033
5ca1d879
ZK		 1034 a.size = size;
1035 mov_read_default(c, pb, a);
5cd62665
ZK		 1036#endif
1037 } else {
01f4895c
MN		 1038 st->codec->codec_id = codec_get_id(mov_audio_tags, format);
1039 if(st->codec->codec_id==CODEC_ID_AMR_NB || st->codec->codec_id==CODEC_ID_AMR_WB) //from TS26.244
891f64b3 1040 {
1041#ifdef DEBUG
baf25c9d 1042 av_log(NULL, AV_LOG_DEBUG, "AMR-NB or AMR-WB audio identified!!\n");
891f64b3 1043#endif
1044 get_be32(pb);get_be32(pb); //Reserved_8
1045 get_be16(pb);//Reserved_2
1046 get_be16(pb);//Reserved_2
1047 get_be32(pb);//Reserved_4
1048 get_be16(pb);//TimeScale
1049 get_be16(pb);//Reserved_2
1050
1051 //AMRSpecificBox.(10 bytes)
d663a1fd 1052
891f64b3 1053 get_be32(pb); //size
891f64b3 1054 get_be32(pb); //type=='damr'
891f64b3 1055 get_be32(pb); //vendor
1056 get_byte(pb); //decoder version
1057 get_be16(pb); //mode_set
1058 get_byte(pb); //mode_change_period
1059 get_byte(pb); //frames_per_sample
6cea494e 1060
25c4950e 1061 st->duration = AV_NOPTS_VALUE;//Not possible to get from this info, must count number of AMR frames
01f4895c 1062 if(st->codec->codec_id==CODEC_ID_AMR_NB)
d663a1fd 1063 {
01f4895c
MN		 1064 st->codec->sample_rate=8000;
1065 st->codec->channels=1;
d663a1fd
MN		 1066 }
1067 else //AMR-WB
1068 {
01f4895c
MN		 1069 st->codec->sample_rate=16000;
1070 st->codec->channels=1;
d663a1fd 1071 }
01f4895c
MN		 1072 st->codec->bits_per_sample=16;
1073 st->codec->bit_rate=0; /*It is not possible to tell this before we have
891f64b3 1074 an audio frame and even then every frame can be different*/
5cd62665 1075 }
01f4895c 1076 else if( st->codec->codec_tag == MKTAG( 'm', 'p', '4', 's' ))
14342fd5
JC		 1077 {
1078 //This is some stuff for the hint track, lets ignore it!
1079 //Do some mp4 auto detect.
1080 c->mp4=1;
1081 size-=(16);
1082 url_fskip(pb, size); /* The mp4s atom also contians a esds atom that we can skip*/
1083 }
01f4895c 1084 else if( st->codec->codec_tag == MKTAG( 'm', 'p', '4', 'a' ))
2768b0d9 1085 {
2f1e1ed3
MN		 1086 MOV_atom_t a;
1087 int mp4_version;
1088
2768b0d9 1089 /* Handle mp4 audio tag */
2f1e1ed3
MN		 1090 mp4_version=get_be16(pb);/*version*/
1091 get_be16(pb); /*revesion*/
2768b0d9 1092 get_be32(pb);
01f4895c
MN		 1093 st->codec->channels = get_be16(pb); /* channels */
1094 st->codec->bits_per_sample = get_be16(pb); /* bits per sample */
2768b0d9 1095 get_be32(pb);
01f4895c 1096 st->codec->sample_rate = get_be16(pb); /* sample rate, not always correct */
302c389e
MN		 1097 if(st->codec->sample_rate == 1) //nonsese rate? -> ignore
1098 st->codec->sample_rate= 0;
1099
2768b0d9
TR		 1100 get_be16(pb);
1101 c->mp4=1;
2f1e1ed3
MN		 1102
1103 if(mp4_version==1)
1104 {
1105 url_fskip(pb,16);
1106 a.size=size-(16+20+16);
1107 }
1108 else
1109 a.size=size-(16+20);
1110
1111 a.offset=url_ftell(pb);
1112
2768b0d9 1113 mov_read_default(c, pb, a);
2f1e1ed3 1114
2768b0d9 1115 /* Get correct sample rate from extradata */
01f4895c 1116 if(st->codec->extradata_size) {
2768b0d9
TR		 1117 const int samplerate_table[] = {
1118 96000, 88200, 64000, 48000, 44100, 32000,
1119 24000, 22050, 16000, 12000, 11025, 8000,
1120 7350, 0, 0, 0
1121 };
01f4895c 1122 unsigned char *px = st->codec->extradata;
2768b0d9
TR		 1123 // 5 bits objectTypeIndex, 4 bits sampleRateIndex, 4 bits channels
1124 int samplerate_index = ((px[0] & 7) << 1) + ((px[1] >> 7) & 1);
01f4895c
MN		 1125 st->codec->sample_rate = samplerate_table[samplerate_index];
1126 st->codec->channels = (px[1] >> 3) & 15;
2768b0d9
TR		 1127 }
1128 }
01f4895c 1129 else if( st->codec->codec_tag == MKTAG( 'a', 'l', 'a', 'c' ))
6d6d7970
MM		 1130 {
1131 /* Handle alac audio tag + special extradata */
1132 get_be32(pb); /* version */
1133 get_be32(pb);
01f4895c
MN		 1134 st->codec->channels = get_be16(pb); /* channels */
1135 st->codec->bits_per_sample = get_be16(pb); /* bits per sample */
6d6d7970 1136 get_be32(pb);
01f4895c 1137 st->codec->sample_rate = get_be16(pb);
6d6d7970
MM		 1138 get_be16(pb);
1139
1140 /* fetch the 36-byte extradata needed for alac decoding */
01f4895c
MN		 1141 st->codec->extradata_size = 36;
1142 st->codec->extradata = (uint8_t*)
1143 av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
1144 get_buffer(pb, st->codec->extradata, st->codec->extradata_size);
6d6d7970 1145 }
14342fd5
JC		 1146 else if(size>=(16+20))
1147 {//16 bytes read, reading atleast 20 more
c3ea71a0 1148 uint16_t version;
14342fd5 1149#ifdef DEBUG
baf25c9d 1150 av_log(NULL, AV_LOG_DEBUG, "audio size=0x%X\n",size);
14342fd5 1151#endif
c3ea71a0 1152 version = get_be16(pb); /* version */
891f64b3 1153 get_be16(pb); /* revision level */
1154 get_be32(pb); /* vendor */
1155
01f4895c
MN		 1156 st->codec->channels = get_be16(pb); /* channel count */
1157 st->codec->bits_per_sample = get_be16(pb); /* sample size */
891f64b3 1158
1159 /* handle specific s8 codec */
1160 get_be16(pb); /* compression id = 0*/
1161 get_be16(pb); /* packet size = 0 */
1162
01f4895c
MN		 1163 st->codec->sample_rate = ((get_be32(pb) >> 16));
1164 //av_log(NULL, AV_LOG_DEBUG, "CODECID %d %d %.4s\n", st->codec->codec_id, CODEC_ID_PCM_S16BE, (char*)&format);
891f64b3 1165
01f4895c 1166 switch (st->codec->codec_id) {
891f64b3 1167 case CODEC_ID_PCM_S16BE:
01f4895c
MN		 1168 if (st->codec->bits_per_sample == 8)
1169 st->codec->codec_id = CODEC_ID_PCM_S8;
891f64b3 1170 /* fall */
1171 case CODEC_ID_PCM_U8:
01f4895c 1172 st->codec->bit_rate = st->codec->sample_rate * 8;
891f64b3 1173 break;
1174 default:
1175 ;
1176 }
14342fd5
JC		 1177
1178 //Read QT version 1 fields. In version 0 theese dont exist
1179#ifdef DEBUG
baf25c9d
GC		 1180 av_log(NULL, AV_LOG_DEBUG, "version =%d mp4=%d\n",version,c->mp4);
1181 av_log(NULL, AV_LOG_DEBUG, "size-(16+20+16)=%d\n",size-(16+20+16));
14342fd5
JC		 1182#endif
1183 if((version==1) && size>=(16+20+16))
1184 {
1185 get_be32(pb); /* samples per packet */
1186 get_be32(pb); /* bytes per packet */
1187 get_be32(pb); /* bytes per frame */
1188 get_be32(pb); /* bytes per sample */
1189 if(size>(16+20+16))
1190 {
1191 //Optional, additional atom-based fields
c3ea71a0 1192 MOV_atom_t a = { format, url_ftell(pb), size - (16 + 20 + 16 + 8) };
14342fd5 1193#ifdef DEBUG
baf25c9d 1194 av_log(NULL, AV_LOG_DEBUG, "offest=0x%X, sizeleft=%d=0x%x,format=%c%c%c%c\n",(int)url_ftell(pb),size - (16 + 20 + 16 ),size - (16 + 20 + 16 ),
14342fd5
JC		 1195 (format >> 0) & 0xff,
1196 (format >> 8) & 0xff,
1197 (format >> 16) & 0xff,
1198 (format >> 24) & 0xff);
1199#endif
14342fd5
JC		 1200 mov_read_default(c, pb, a);
1201 }
1202 }
1203 else
1204 {
1205 //We should be down to 0 bytes here, but lets make sure.
1206 size-=(16+20);
1207#ifdef DEBUG
1208 if(size>0)
baf25c9d 1209 av_log(NULL, AV_LOG_DEBUG, "skipping 0x%X bytes\n",size-(16+20));
14342fd5
JC		 1210#endif
1211 url_fskip(pb, size);
1212 }
1213 }
1214 else
1215 {
1216 size-=16;
1217 //Unknown size, but lets do our best and skip the rest.
1218#ifdef DEBUG
baf25c9d 1219 av_log(NULL, AV_LOG_DEBUG, "Strange size, skipping 0x%X bytes\n",size);
14342fd5
JC		 1220#endif
1221 url_fskip(pb, size);
891f64b3 1222 }
6cea494e 1223 }
6cea494e 1224 }
302c389e
MN		 1225
1226 if(st->codec->codec_type==CODEC_TYPE_AUDIO && st->codec->sample_rate==0 && sc->time_scale>1) {
1227 st->codec->sample_rate= sc->time_scale;
1228 }
5cd62665 1229
6cea494e
ZK		 1230 return 0;
1231}
1232
5ca1d879 1233static int mov_read_stsc(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
6cea494e 1234{
5ca1d879
ZK		 1235 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1236 MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
568e18b1 1237 unsigned int i, entries;
b6a17df4 1238
5ca1d879 1239 print_atom("stsc", atom);
5cd62665 1240
6cea494e
ZK		 1241 get_byte(pb); /* version */
1242 get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
1243
1244 entries = get_be32(pb);
568e18b1
MN		 1245
1246 if(entries >= UINT_MAX / sizeof(MOV_sample_to_chunk_tbl))
1247 return -1;
1248
3ffe3793 1249#ifdef DEBUG
baf25c9d 1250av_log(NULL, AV_LOG_DEBUG, "track[%i].stsc.entries = %i\n", c->fc->nb_streams-1, entries);
3ffe3793 1251#endif
6cea494e 1252 sc->sample_to_chunk_sz = entries;
b6a17df4
ZK		 1253 sc->sample_to_chunk = (MOV_sample_to_chunk_tbl*) av_malloc(entries * sizeof(MOV_sample_to_chunk_tbl));
1254 if (!sc->sample_to_chunk)
1255 return -1;
6cea494e
ZK		 1256 for(i=0; i<entries; i++) {
1257 sc->sample_to_chunk[i].first = get_be32(pb);
1258 sc->sample_to_chunk[i].count = get_be32(pb);
1259 sc->sample_to_chunk[i].id = get_be32(pb);
1260#ifdef DEBUG
baf25c9d 1261/* av_log(NULL, AV_LOG_DEBUG, "sample_to_chunk first=%ld count=%ld, id=%ld\n", sc->sample_to_chunk[i].first, sc->sample_to_chunk[i].count, sc->sample_to_chunk[i].id); */
6cea494e
ZK		 1262#endif
1263 }
1264 return 0;
1265}
1266
247d56f5
BB		 1267static int mov_read_stss(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1268{
1269 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1270 MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
568e18b1 1271 unsigned int i, entries;
247d56f5
BB		 1272
1273 print_atom("stss", atom);
1274
1275 get_byte(pb); /* version */
1276 get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
1277
1278 entries = get_be32(pb);
568e18b1
MN		 1279
1280 if(entries >= UINT_MAX / sizeof(long))
1281 return -1;
1282
247d56f5
BB		 1283 sc->keyframe_count = entries;
1284#ifdef DEBUG
1285 av_log(NULL, AV_LOG_DEBUG, "keyframe_count = %ld\n", sc->keyframe_count);
1286#endif
1287 sc->keyframes = (long*) av_malloc(entries * sizeof(long));
1288 if (!sc->keyframes)
1289 return -1;
1290 for(i=0; i<entries; i++) {
1291 sc->keyframes[i] = get_be32(pb);
1292#ifdef DEBUG
1293/* av_log(NULL, AV_LOG_DEBUG, "keyframes[]=%ld\n", sc->keyframes[i]); */
1294#endif
1295 }
1296 return 0;
1297}
1298
5ca1d879 1299static int mov_read_stsz(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
6cea494e 1300{
5ca1d879
ZK		 1301 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1302 MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
568e18b1 1303 unsigned int i, entries;
b6a17df4 1304
5ca1d879 1305 print_atom("stsz", atom);
5cd62665 1306
6cea494e
ZK		 1307 get_byte(pb); /* version */
1308 get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
5cd62665 1309
6cea494e
ZK		 1310 sc->sample_size = get_be32(pb);
1311 entries = get_be32(pb);
568e18b1
MN		 1312 if(entries >= UINT_MAX / sizeof(long))
1313 return -1;
1314
6cea494e 1315 sc->sample_count = entries;
0147f198 1316#ifdef DEBUG
baf25c9d 1317 av_log(NULL, AV_LOG_DEBUG, "sample_size = %ld sample_count = %ld\n", sc->sample_size, sc->sample_count);
0147f198 1318#endif
6cea494e
ZK		 1319 if(sc->sample_size)
1320 return 0; /* there isn't any table following */
b6a17df4
ZK		 1321 sc->sample_sizes = (long*) av_malloc(entries * sizeof(long));
1322 if (!sc->sample_sizes)
1323 return -1;
6cea494e
ZK		 1324 for(i=0; i<entries; i++) {
1325 sc->sample_sizes[i] = get_be32(pb);
1326#ifdef DEBUG
7dbab4a9 1327 av_log(NULL, AV_LOG_DEBUG, "sample_sizes[]=%ld\n", sc->sample_sizes[i]);
6cea494e
ZK		 1328#endif
1329 }
1330 return 0;
1331}
1332
5ca1d879 1333static int mov_read_stts(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
0147f198 1334{
5ca1d879 1335 AVStream *st = c->fc->streams[c->fc->nb_streams-1];
fd6e513e 1336 //MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
568e18b1 1337 unsigned int i, entries;
d957696f
MN		 1338 int64_t duration=0;
1339 int64_t total_sample_count=0;
b6a17df4 1340
5ca1d879 1341 print_atom("stts", atom);
0147f198
FR		 1342
1343 get_byte(pb); /* version */
1344 get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
1345 entries = get_be32(pb);
cd461d48 1346 if(entries >= UINT_MAX / sizeof(Time2Sample))
568e18b1 1347 return -1;
891f64b3 1348
baf25c9d 1349 c->streams[c->fc->nb_streams-1]->stts_count = entries;
cd461d48 1350 c->streams[c->fc->nb_streams-1]->stts_data = av_malloc(entries * sizeof(Time2Sample));
891f64b3 1351
3ffe3793 1352#ifdef DEBUG
baf25c9d 1353av_log(NULL, AV_LOG_DEBUG, "track[%i].stts.entries = %i\n", c->fc->nb_streams-1, entries);
3ffe3793 1354#endif
0147f198 1355 for(i=0; i<entries; i++) {
cd461d48
MN		 1356 int sample_duration;
1357 int sample_count;
0147f198 1358
891f64b3 1359 sample_count=get_be32(pb);
0147f198 1360 sample_duration = get_be32(pb);
cd461d48
MN		 1361 c->streams[c->fc->nb_streams - 1]->stts_data[i].count= sample_count;
1362 c->streams[c->fc->nb_streams - 1]->stts_data[i].duration= sample_duration;
961e0ccd 1363
891f64b3 1364#ifdef DEBUG
baf25c9d 1365 av_log(NULL, AV_LOG_DEBUG, "sample_count=%d, sample_duration=%d\n",sample_count,sample_duration);
891f64b3 1366#endif
1367 duration+=sample_duration*sample_count;
1368 total_sample_count+=sample_count;
891f64b3 1369 }
1370
961e0ccd
MN		 1371 st->nb_frames= total_sample_count;
1372 if(duration)
1373 st->duration= duration;
0147f198
FR		 1374 return 0;
1375}
1376
c4ac052b
MN		 1377static int mov_read_ctts(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1378{
1379// AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1380 //MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
1381 unsigned int i, entries;
1382
1383 print_atom("ctts", atom);
1384
1385 get_byte(pb); /* version */
1386 get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
1387 entries = get_be32(pb);
1388 if(entries >= UINT_MAX / sizeof(Time2Sample))
1389 return -1;
1390
1391 c->streams[c->fc->nb_streams-1]->ctts_count = entries;
1392 c->streams[c->fc->nb_streams-1]->ctts_data = av_malloc(entries * sizeof(Time2Sample));
1393
1394// #ifdef DEBUG
1395av_log(NULL, AV_LOG_DEBUG, "track[%i].ctts.entries = %i\n", c->fc->nb_streams-1, entries);
1396// #endif
1397 for(i=0; i<entries; i++) {
1398 c->streams[c->fc->nb_streams - 1]->ctts_data[i].count= get_be32(pb);
1399 c->streams[c->fc->nb_streams - 1]->ctts_data[i].duration= get_be32(pb);
1400 }
1401 return 0;
1402}
1403
5ca1d879
ZK		 1404static int mov_read_trak(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1405{
1406 AVStream *st;
1407 MOVStreamContext *sc;
1408
1409 print_atom("trak", atom);
1410
1411 st = av_new_stream(c->fc, c->fc->nb_streams);
1412 if (!st) return -2;
1413 sc = (MOVStreamContext*) av_mallocz(sizeof(MOVStreamContext));
1414 if (!sc) {
1415 av_free(st);
1416 return -1;
1417 }
1418
1419 sc->sample_to_chunk_index = -1;
1420 st->priv_data = sc;
01f4895c 1421 st->codec->codec_type = CODEC_TYPE_MOV_OTHER;
25c4950e 1422 st->start_time = 0; /* XXX: check */
5ca1d879
ZK		 1423 c->streams[c->fc->nb_streams-1] = sc;
1424
1425 return mov_read_default(c, pb, atom);
1426}
1427
1428static int mov_read_tkhd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1429{
1430 AVStream *st;
1431
1432 print_atom("tkhd", atom);
1433
1434 st = c->fc->streams[c->fc->nb_streams-1];
1435
1436 get_byte(pb); /* version */
1437
1438 get_byte(pb); get_byte(pb);
1439 get_byte(pb); /* flags */
1440 /*
1441 MOV_TRACK_ENABLED 0x0001
1442 MOV_TRACK_IN_MOVIE 0x0002
1443 MOV_TRACK_IN_PREVIEW 0x0004
1444 MOV_TRACK_IN_POSTER 0x0008
1445 */
1446
1447 get_be32(pb); /* creation time */
1448 get_be32(pb); /* modification time */
1449 st->id = (int)get_be32(pb); /* track id (NOT 0 !)*/
1450 get_be32(pb); /* reserved */
25c4950e 1451 st->start_time = 0; /* check */
961e0ccd 1452 get_be32(pb); /* highlevel (considering edits) duration in movie timebase */
5ca1d879
ZK		 1453 get_be32(pb); /* reserved */
1454 get_be32(pb); /* reserved */
1455
1456 get_be16(pb); /* layer */
1457 get_be16(pb); /* alternate group */
1458 get_be16(pb); /* volume */
1459 get_be16(pb); /* reserved */
1460
1461 url_fskip(pb, 36); /* display matrix */
1462
1463 /* those are fixed-point */
01f4895c
MN		 1464 /*st->codec->width =*/ get_be32(pb) >> 16; /* track width */
1465 /*st->codec->height =*/ get_be32(pb) >> 16; /* track height */
5ca1d879
ZK		 1466
1467 return 0;
1468}
1469
1470/* this atom should be null (from specs), but some buggy files put the 'moov' atom inside it... */
1471/* like the files created with Adobe Premiere 5.0, for samples see */
1472/* http://graphics.tudelft.nl/~wouter/publications/soundtests/ */
1473static int mov_read_wide(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1474{
1475 int err;
5ca1d879
ZK		 1476
1477#ifdef DEBUG
1478 print_atom("wide", atom);
1479 debug_indent++;
1480#endif
1481 if (atom.size < 8)
1482 return 0; /* continue */
1483 if (get_be32(pb) != 0) { /* 0 sized mdat atom... use the 'wide' atom size */
1484 url_fskip(pb, atom.size - 4);
1485 return 0;
1486 }
1487 atom.type = get_le32(pb);
1488 atom.offset += 8;
1489 atom.size -= 8;
fd6e513e 1490 if (atom.type != MKTAG('m', 'd', 'a', 't')) {
5ca1d879
ZK		 1491 url_fskip(pb, atom.size);
1492 return 0;
1493 }
1494 err = mov_read_mdat(c, pb, atom);
1495#ifdef DEBUG
1496 debug_indent--;
1497#endif
1498 return err;
1499}
1500
1501
0147f198 1502#ifdef CONFIG_ZLIB
0c1a9eda 1503static int null_read_packet(void *opaque, uint8_t *buf, int buf_size)
0147f198
FR		 1504{
1505 return -1;
1506}
1507
5ca1d879 1508static int mov_read_cmov(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
0147f198 1509{
0147f198 1510 ByteIOContext ctx;
b6a17df4
ZK		 1511 uint8_t *cmov_data;
1512 uint8_t *moov_data; /* uncompressed data */
0147f198
FR		 1513 long cmov_len, moov_len;
1514 int ret;
b6a17df4 1515
5ca1d879 1516 print_atom("cmov", atom);
0147f198
FR		 1517
1518 get_be32(pb); /* dcom atom */
1519 if (get_le32(pb) != MKTAG( 'd', 'c', 'o', 'm' ))
1520 return -1;
1521 if (get_le32(pb) != MKTAG( 'z', 'l', 'i', 'b' )) {
baf25c9d 1522 av_log(NULL, AV_LOG_DEBUG, "unknown compression for cmov atom !");
0147f198
FR		 1523 return -1;
1524 }
1525 get_be32(pb); /* cmvd atom */
1526 if (get_le32(pb) != MKTAG( 'c', 'm', 'v', 'd' ))
1527 return -1;
1528 moov_len = get_be32(pb); /* uncompressed size */
5ca1d879 1529 cmov_len = atom.size - 6 * 4;
5cd62665 1530
b6a17df4 1531 cmov_data = (uint8_t *) av_malloc(cmov_len);
0147f198
FR		 1532 if (!cmov_data)
1533 return -1;
b6a17df4 1534 moov_data = (uint8_t *) av_malloc(moov_len);
0147f198
FR		 1535 if (!moov_data) {
1536 av_free(cmov_data);
1537 return -1;
1538 }
1539 get_buffer(pb, cmov_data, cmov_len);
b6a17df4 1540 if(uncompress (moov_data, (uLongf *) &moov_len, (const Bytef *)cmov_data, cmov_len) != Z_OK)
0147f198
FR		 1541 return -1;
1542 if(init_put_byte(&ctx, moov_data, moov_len, 0, NULL, null_read_packet, NULL, NULL) != 0)
1543 return -1;
1544 ctx.buf_end = ctx.buffer + moov_len;
5ca1d879
ZK		 1545 atom.type = MKTAG( 'm', 'o', 'o', 'v' );
1546 atom.offset = 0;
1547 atom.size = moov_len;
9ed83b0a
ZK		 1548#ifdef DEBUG
1549 { int fd = open("/tmp/uncompheader.mov", O_WRONLY | O_CREAT); write(fd, moov_data, moov_len); close(fd); }
1550#endif
5ca1d879 1551 ret = mov_read_default(c, &ctx, atom);
0147f198
FR		 1552 av_free(moov_data);
1553 av_free(cmov_data);
9ed83b0a 1554
0147f198
FR		 1555 return ret;
1556}
1557#endif
1558
baf25c9d
GC		 1559/* edit list atom */
1560static int mov_read_elst(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1561{
58e555d4 1562 int i, edit_count;
baf25c9d
GC		 1563 print_atom("elst", atom);
1564
1565 get_byte(pb); /* version */
1566 get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
58e555d4
MN		 1567 edit_count= c->streams[c->fc->nb_streams-1]->edit_count = get_be32(pb); /* entries */
1568
1569 for(i=0; i<edit_count; i++){
1570 get_be32(pb); /* Track duration */
1571 get_be32(pb); /* Media time */
1572 get_be32(pb); /* Media rate */
1573 }
baf25c9d
GC		 1574#ifdef DEBUG
1575 av_log(NULL, AV_LOG_DEBUG, "track[%i].edit_count = %i\n", c->fc->nb_streams-1, c->streams[c->fc->nb_streams-1]->edit_count);
1576#endif
1577 return 0;
1578}
1579
6cea494e
ZK		 1580static const MOVParseTableEntry mov_default_parse_table[] = {
1581/* mp4 atoms */
5ca1d879
ZK		 1582{ MKTAG( 'c', 'o', '6', '4' ), mov_read_stco },
1583{ MKTAG( 'c', 'p', 'r', 't' ), mov_read_default },
1584{ MKTAG( 'c', 'r', 'h', 'd' ), mov_read_default },
c4ac052b 1585{ MKTAG( 'c', 't', 't', 's' ), mov_read_ctts }, /* composition time to sample */
5ca1d879
ZK		 1586{ MKTAG( 'd', 'i', 'n', 'f' ), mov_read_default }, /* data information */
1587{ MKTAG( 'd', 'p', 'n', 'd' ), mov_read_leaf },
1588{ MKTAG( 'd', 'r', 'e', 'f' ), mov_read_leaf },
1589{ MKTAG( 'e', 'd', 't', 's' ), mov_read_default },
58e555d4 1590{ MKTAG( 'e', 'l', 's', 't' ), mov_read_elst },
5ca1d879
ZK		 1591{ MKTAG( 'f', 'r', 'e', 'e' ), mov_read_leaf },
1592{ MKTAG( 'h', 'd', 'l', 'r' ), mov_read_hdlr },
1593{ MKTAG( 'h', 'i', 'n', 't' ), mov_read_leaf },
1594{ MKTAG( 'h', 'm', 'h', 'd' ), mov_read_leaf },
1595{ MKTAG( 'i', 'o', 'd', 's' ), mov_read_leaf },
1596{ MKTAG( 'm', 'd', 'a', 't' ), mov_read_mdat },
1597{ MKTAG( 'm', 'd', 'h', 'd' ), mov_read_mdhd },
1598{ MKTAG( 'm', 'd', 'i', 'a' ), mov_read_default },
1599{ MKTAG( 'm', 'i', 'n', 'f' ), mov_read_default },
1600{ MKTAG( 'm', 'o', 'o', 'v' ), mov_read_moov },
1601{ MKTAG( 'm', 'p', '4', 'a' ), mov_read_default },
1602{ MKTAG( 'm', 'p', '4', 's' ), mov_read_default },
1603{ MKTAG( 'm', 'p', '4', 'v' ), mov_read_default },
1604{ MKTAG( 'm', 'p', 'o', 'd' ), mov_read_leaf },
1605{ MKTAG( 'm', 'v', 'h', 'd' ), mov_read_mvhd },
1606{ MKTAG( 'n', 'm', 'h', 'd' ), mov_read_leaf },
1607{ MKTAG( 'o', 'd', 'h', 'd' ), mov_read_default },
1608{ MKTAG( 's', 'd', 'h', 'd' ), mov_read_default },
3c13647a 1609{ MKTAG( 's', 'k', 'i', 'p' ), mov_read_leaf },
5ca1d879 1610{ MKTAG( 's', 'm', 'h', 'd' ), mov_read_leaf }, /* sound media info header */
169eb021
MM		 1611{ MKTAG( 'S', 'M', 'I', ' ' ), mov_read_smi }, /* Sorenson extension ??? */
1612{ MKTAG( 'a', 'v', 'c', 'C' ), mov_read_avcC },
5ca1d879
ZK		 1613{ MKTAG( 's', 't', 'b', 'l' ), mov_read_default },
1614{ MKTAG( 's', 't', 'c', 'o' ), mov_read_stco },
1615{ MKTAG( 's', 't', 'd', 'p' ), mov_read_default },
1616{ MKTAG( 's', 't', 's', 'c' ), mov_read_stsc },
1617{ MKTAG( 's', 't', 's', 'd' ), mov_read_stsd }, /* sample description */
1618{ MKTAG( 's', 't', 's', 'h' ), mov_read_default },
247d56f5 1619{ MKTAG( 's', 't', 's', 's' ), mov_read_stss }, /* sync sample */
5ca1d879
ZK		 1620{ MKTAG( 's', 't', 's', 'z' ), mov_read_stsz }, /* sample size */
1621{ MKTAG( 's', 't', 't', 's' ), mov_read_stts },
1622{ MKTAG( 't', 'k', 'h', 'd' ), mov_read_tkhd }, /* track header */
1623{ MKTAG( 't', 'r', 'a', 'k' ), mov_read_trak },
1624{ MKTAG( 't', 'r', 'e', 'f' ), mov_read_default }, /* not really */
1625{ MKTAG( 'u', 'd', 't', 'a' ), mov_read_leaf },
1626{ MKTAG( 'u', 'r', 'l', ' ' ), mov_read_leaf },
1627{ MKTAG( 'u', 'r', 'n', ' ' ), mov_read_leaf },
a0f1f165 1628{ MKTAG( 'u', 'u', 'i', 'd' ), mov_read_leaf },
5ca1d879 1629{ MKTAG( 'v', 'm', 'h', 'd' ), mov_read_leaf }, /* video media info header */
d9b1c197 1630{ MKTAG( 'w', 'a', 'v', 'e' ), mov_read_wave },
6cea494e 1631/* extra mp4 */
5ca1d879 1632{ MKTAG( 'M', 'D', 'E', 'S' ), mov_read_leaf },
6cea494e 1633/* QT atoms */
5ca1d879
ZK		 1634{ MKTAG( 'c', 'h', 'a', 'p' ), mov_read_leaf },
1635{ MKTAG( 'c', 'l', 'i', 'p' ), mov_read_default },
1636{ MKTAG( 'c', 'r', 'g', 'n' ), mov_read_leaf },
1637{ MKTAG( 'c', 't', 'a', 'b' ), mov_read_ctab },
1638{ MKTAG( 'e', 's', 'd', 's' ), mov_read_esds },
1639{ MKTAG( 'k', 'm', 'a', 't' ), mov_read_leaf },
1640{ MKTAG( 'm', 'a', 't', 't' ), mov_read_default },
1641{ MKTAG( 'r', 'd', 'r', 'f' ), mov_read_leaf },
1642{ MKTAG( 'r', 'm', 'd', 'a' ), mov_read_default },
1643{ MKTAG( 'r', 'm', 'd', 'r' ), mov_read_leaf },
1644{ MKTAG( 'r', 'm', 'r', 'a' ), mov_read_default },
1645{ MKTAG( 's', 'c', 'p', 't' ), mov_read_leaf },
1646{ MKTAG( 's', 's', 'r', 'c' ), mov_read_leaf },
1647{ MKTAG( 's', 'y', 'n', 'c' ), mov_read_leaf },
1648{ MKTAG( 't', 'c', 'm', 'd' ), mov_read_leaf },
1649{ MKTAG( 'w', 'i', 'd', 'e' ), mov_read_wide }, /* place holder */
1650//{ MKTAG( 'r', 'm', 'q', 'u' ), mov_read_leaf },
0147f198 1651#ifdef CONFIG_ZLIB
5ca1d879 1652{ MKTAG( 'c', 'm', 'o', 'v' ), mov_read_cmov },
0147f198 1653#else
5ca1d879 1654{ MKTAG( 'c', 'm', 'o', 'v' ), mov_read_leaf },
0147f198 1655#endif
5ca1d879 1656{ 0L, mov_read_leaf }
6cea494e
ZK		 1657};
1658
1659static void mov_free_stream_context(MOVStreamContext *sc)
1660{
1661 if(sc) {
c4ac052b
MN		 1662 av_freep(&sc->chunk_offsets);
1663 av_freep(&sc->sample_to_chunk);
1664 av_freep(&sc->sample_sizes);
1665 av_freep(&sc->keyframes);
c4ac052b
MN		 1666 av_freep(&sc->stts_data);
1667 av_freep(&sc->ctts_data);
1668 av_freep(&sc);
6cea494e
ZK		 1669 }
1670}
1671
5ca1d879 1672static inline uint32_t mov_to_tag(uint8_t *buf)
0e7eed09 1673{
5ca1d879 1674 return MKTAG(buf[0], buf[1], buf[2], buf[3]);
0e7eed09
FB		 1675}
1676
5cd62665 1677static inline uint32_t to_be32(uint8_t *buf)
0e7eed09
FB		 1678{
1679 return (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];
1680}
1681
b6a17df4 1682/* XXX: is it sufficient ? */
c9a65ca8
FB		 1683static int mov_probe(AVProbeData *p)
1684{
0e7eed09
FB		 1685 unsigned int offset;
1686 uint32_t tag;
2497479f 1687 int score = 0;
3ffe3793 1688
c9a65ca8
FB		 1689 /* check file header */
1690 if (p->buf_size <= 12)
1691 return 0;
0e7eed09
FB		 1692 offset = 0;
1693 for(;;) {
1694 /* ignore invalid offset */
1695 if ((offset + 8) > (unsigned int)p->buf_size)
2497479f 1696 return score;
5ca1d879 1697 tag = mov_to_tag(p->buf + offset + 4);
0e7eed09 1698 switch(tag) {
2497479f 1699 /* check for obvious tags */
0e7eed09 1700 case MKTAG( 'm', 'o', 'o', 'v' ):
8b879f18
FR		 1701 case MKTAG( 'm', 'd', 'a', 't' ):
1702 case MKTAG( 'p', 'n', 'o', 't' ): /* detect movs with preview pics like ew.mov and april.mov */
bc634f6f 1703 case MKTAG( 'u', 'd', 't', 'a' ): /* Packet Video PVAuthor adds this and a lot of more junk */
3ffe3793 1704 return AVPROBE_SCORE_MAX;
2497479f
FR		 1705 /* those are more common words, so rate then a bit less */
1706 case MKTAG( 'w', 'i', 'd', 'e' ):
1707 case MKTAG( 'f', 'r', 'e', 'e' ):
1708 case MKTAG( 'j', 'u', 'n', 'k' ):
1709 case MKTAG( 'p', 'i', 'c', 't' ):
1710 return AVPROBE_SCORE_MAX - 5;
0e7eed09 1711 case MKTAG( 'f', 't', 'y', 'p' ):
5ca1d879 1712 case MKTAG( 's', 'k', 'i', 'p' ):
0d23cb84 1713 case MKTAG( 'u', 'u', 'i', 'd' ):
14342fd5 1714 offset = to_be32(p->buf+offset) + offset;
2497479f
FR		 1715 /* if we only find those cause probedata is too small at least rate them */
1716 score = AVPROBE_SCORE_MAX - 50;
0e7eed09
FB		 1717 break;
1718 default:
1719 /* unrecognized tag */
2497479f 1720 return score;
0e7eed09 1721 }
3ffe3793 1722 }
2497479f 1723 return score;
c9a65ca8
FB		 1724}
1725
a266644f 1726static int mov_read_header(AVFormatContext *s, AVFormatParameters *ap)
6cea494e 1727{
b6a17df4 1728 MOVContext *mov = (MOVContext *) s->priv_data;
6cea494e 1729 ByteIOContext *pb = &s->pb;
684f44d9 1730 int i, j, nb, err;
5ca1d879 1731 MOV_atom_t atom = { 0, 0, 0 };
6cea494e 1732
6cea494e 1733 mov->fc = s;
b6a17df4 1734 mov->parse_table = mov_default_parse_table;
c9a65ca8
FB		 1735#if 0
1736 /* XXX: I think we should auto detect */
1737 if(s->iformat->name[1] == 'p')
6cea494e 1738 mov->mp4 = 1;
c9a65ca8 1739#endif
6cea494e 1740 if(!url_is_streamed(pb)) /* .mov and .mp4 aren't streamable anyway (only progressive download if moov is before mdat) */
a965c478 1741 atom.size = url_fsize(pb);
6cea494e 1742 else
9ed83b0a 1743 atom.size = 0x7FFFFFFFFFFFFFFFLL;
6cea494e
ZK		 1744
1745#ifdef DEBUG
baf25c9d 1746 av_log(NULL, AV_LOG_DEBUG, "filesz=%Ld\n", atom.size);
6cea494e
ZK		 1747#endif
1748
1749 /* check MOV header */
5ca1d879 1750 err = mov_read_default(mov, pb, atom);
25fa62e1 1751 if (err<0 || (!mov->found_moov && !mov->found_mdat)) {
bc874dae 1752 av_log(s, AV_LOG_ERROR, "mov: header not found !!! (err:%d, moov:%d, mdat:%d) pos:%lld\n",
25fa62e1
ZK		 1753 err, mov->found_moov, mov->found_mdat, url_ftell(pb));
1754 return -1;
6cea494e
ZK		 1755 }
1756#ifdef DEBUG
baf25c9d 1757 av_log(NULL, AV_LOG_DEBUG, "on_parse_exit_offset=%d\n", (int) url_ftell(pb));
6cea494e
ZK		 1758#endif
1759 /* some cleanup : make sure we are on the mdat atom */
1760 if(!url_is_streamed(pb) && (url_ftell(pb) != mov->mdat_offset))
1761 url_fseek(pb, mov->mdat_offset, SEEK_SET);
1762
1763 mov->next_chunk_offset = mov->mdat_offset; /* initialise reading */
1764
1765#ifdef DEBUG
baf25c9d 1766 av_log(NULL, AV_LOG_DEBUG, "mdat_reset_offset=%d\n", (int) url_ftell(pb));
6cea494e
ZK		 1767#endif
1768
1769#ifdef DEBUG
baf25c9d 1770 av_log(NULL, AV_LOG_DEBUG, "streams= %d\n", s->nb_streams);
6cea494e
ZK		 1771#endif
1772 mov->total_streams = nb = s->nb_streams;
5ca1d879 1773
6cea494e
ZK		 1774#if 1
1775 for(i=0; i<s->nb_streams;) {
01f4895c 1776 if(s->streams[i]->codec->codec_type == CODEC_TYPE_MOV_OTHER) {/* not audio, not video, delete */
1ea4f593 1777 av_free(s->streams[i]);
6cea494e
ZK		 1778 for(j=i+1; j<s->nb_streams; j++)
1779 s->streams[j-1] = s->streams[j];
1780 s->nb_streams--;
1781 } else
1782 i++;
1783 }
1784 for(i=0; i<s->nb_streams;i++) {
1785 MOVStreamContext *sc;
1786 sc = (MOVStreamContext *)s->streams[i]->priv_data;
1787 sc->ffindex = i;
1788 sc->is_ff_stream = 1;
1789 }
1790#endif
1791#ifdef DEBUG
baf25c9d 1792 av_log(NULL, AV_LOG_DEBUG, "real streams= %d\n", s->nb_streams);
6cea494e
ZK		 1793#endif
1794 return 0;
1795}
1796
1797/* Yes, this is ugly... I didn't write the specs of QT :p */
1798/* XXX:remove useless commented code sometime */
a266644f 1799static int mov_read_packet(AVFormatContext *s, AVPacket *pkt)
6cea494e 1800{
b6a17df4 1801 MOVContext *mov = (MOVContext *) s->priv_data;
0e7eed09 1802 MOVStreamContext *sc;
86d8602f
MN		 1803 int64_t offset = INT64_MAX;
1804 int64_t best_dts = INT64_MAX;
247d56f5 1805 int i, a, b, m;
5cd62665 1806 int size;
7619ed2b 1807 int idx;
6cea494e 1808 size = 0x0FFFFFFF;
5cd62665 1809
3ffe3793
FR		 1810#ifdef MOV_SPLIT_CHUNKS
1811 if (mov->partial) {
5cd62665
ZK		 1812 sc = mov->partial;
1813 idx = sc->sample_to_chunk_index;
1814
3ffe3793 1815 if (idx < 0) return 0;
7dbab4a9
FR		 1816#ifdef DEBUG
1817 fprintf(stderr, "sc[ffid %d]->sample_size = %d\n", sc->ffindex, sc->sample_size);
1818#endif
de39cdf5
MN		 1819 //size = sc->sample_sizes[sc->current_sample];
1820 // that ain't working...
1821 //size = (sc->sample_size)?sc->sample_size:sc->sample_sizes[sc->current_sample];
1822 size = (sc->sample_size > 1)?sc->sample_size:sc->sample_sizes[sc->current_sample];
1823
5cd62665
ZK		 1824 sc->current_sample++;
1825 sc->left_in_chunk--;
3ffe3793 1826
5cd62665 1827 if (sc->left_in_chunk <= 0)
3ffe3793
FR		 1828 mov->partial = 0;
1829 offset = mov->next_chunk_offset;
1830 /* extract the sample */
1831
1832 goto readchunk;
1833 }
1834#endif
1835
6cea494e 1836again:
5cd62665 1837 sc = 0;
86d8602f
MN		 1838 if(offset == INT64_MAX)
1839 best_dts= INT64_MAX;
6cea494e 1840 for(i=0; i<mov->total_streams; i++) {
5cd62665 1841 MOVStreamContext *msc = mov->streams[i];
86d8602f
MN		 1842
1843 if ((msc->next_chunk < msc->chunk_count) && msc->next_chunk >= 0){
1844 if (msc->sample_to_time_index < msc->stts_count && mov->ni) {
1845 int64_t dts;
1846 int index= msc->sample_to_time_index;
1847 int sample= msc->sample_to_time_sample;
1848 int time= msc->sample_to_time_time;
1849 int duration = msc->stts_data[index].duration;
1850 int count = msc->stts_data[index].count;
1851 if (sample + count < msc->current_sample) {
1852 sample += count;
1853 time += count*duration;
1854 index ++;
1855 duration = msc->stts_data[index].duration;
1856 }
1857 dts = time + (msc->current_sample-1 - sample) * (int64_t)duration;
1858 dts = av_rescale(dts, AV_TIME_BASE, msc->time_scale);
1859// av_log(NULL, AV_LOG_DEBUG, "%d %Ld %Ld %Ld \n", i, dts, best_dts, offset);
1860 if(dts < best_dts){
1861 best_dts= dts;
1862 sc = msc;
1863 offset = msc->chunk_offsets[msc->next_chunk];
1864 }
1865 }else{
1866 //av_log(NULL, AV_LOG_DEBUG, "MOCHUNK %ld %d %p pos:%Ld\n", mov->streams[i]->next_chunk, mov->total_streams, mov->streams[i], url_ftell(&s->pb));
1867 if ((msc->chunk_offsets[msc->next_chunk] < offset)) {
1868 sc = msc;
1869 offset = msc->chunk_offsets[msc->next_chunk];
1870 //av_log(NULL, AV_LOG_DEBUG, "SELETED %Ld i:%d\n", offset, i);
1871 }
1872 }
6cea494e 1873 }
6cea494e 1874 }
86d8602f 1875 if (!sc || offset==INT64_MAX)
5cd62665
ZK		 1876 return -1;
1877
1878 sc->next_chunk++;
1879
3ffe3793 1880 if(mov->next_chunk_offset < offset) { /* some meta data */
6cea494e 1881 url_fskip(&s->pb, (offset - mov->next_chunk_offset));
3ffe3793
FR		 1882 mov->next_chunk_offset = offset;
1883 }
1884
baf25c9d 1885//av_log(NULL, AV_LOG_DEBUG, "chunk: [%i] %lli -> %lli\n", st_id, mov->next_chunk_offset, offset);
f3356e9c 1886 if(!sc->is_ff_stream || (s->streams[sc->ffindex]->discard >= AVDISCARD_ALL)) {
6cea494e 1887 url_fskip(&s->pb, (offset - mov->next_chunk_offset));
3ffe3793 1888 mov->next_chunk_offset = offset;
86d8602f 1889 offset = INT64_MAX;
6cea494e
ZK		 1890 goto again;
1891 }
6cea494e
ZK		 1892
1893 /* now get the chunk size... */
1894
1895 for(i=0; i<mov->total_streams; i++) {
5cd62665
ZK		 1896 MOVStreamContext *msc = mov->streams[i];
1897 if ((msc->next_chunk < msc->chunk_count)
86d8602f
MN		 1898 && msc->chunk_offsets[msc->next_chunk] - offset < size
1899 && msc->chunk_offsets[msc->next_chunk] > offset)
5cd62665 1900 size = msc->chunk_offsets[msc->next_chunk] - offset;
6cea494e 1901 }
bc634f6f
ZK		 1902
1903#ifdef MOV_MINOLTA_FIX
1904 //Make sure that size is according to sample_size (Needed by .mov files
1905 //created on a Minolta Dimage Xi where audio chunks contains waste data in the end)
1906 //Maybe we should really not only check sc->sample_size, but also sc->sample_sizes
1907 //but I have no such movies
1908 if (sc->sample_size > 0) {
1909 int foundsize=0;
1910 for(i=0; i<(sc->sample_to_chunk_sz); i++) {
92e51b66 1911 if( (sc->sample_to_chunk[i].first)<=(sc->next_chunk) )
bc634f6f 1912 {
cac0a56c
RS		 1913 // I can't figure out why for PCM audio sample_size is always 1
1914 // (it should actually be channels*bits_per_second/8) but it is.
01f4895c 1915 AVCodecContext* cod = s->streams[sc->ffindex]->codec;
cac0a56c
RS		 1916 if (sc->sample_size == 1 && (cod->codec_id == CODEC_ID_PCM_S16BE || cod->codec_id == CODEC_ID_PCM_S16LE))
1917 foundsize=(sc->sample_to_chunk[i].count*cod->channels*cod->bits_per_sample)/8;
1918 else
1919 foundsize=sc->sample_to_chunk[i].count*sc->sample_size;
bc634f6f
ZK		 1920 }
1921#ifdef DEBUG
c3ea71a0 1922 av_log(NULL, AV_LOG_DEBUG, "sample_to_chunk first=%ld count=%ld, id=%ld\n", sc->sample_to_chunk[i].first, sc->sample_to_chunk[i].count, sc->sample_to_chunk[i].id);
bc634f6f
ZK		 1923#endif
1924 }
1925 if( (foundsize>0) && (foundsize<size) )
1926 {
1927#ifdef DEBUG
baf25c9d 1928 /*av_log(NULL, AV_LOG_DEBUG, "this size should actually be %d\n",foundsize);*/
bc634f6f
ZK		 1929#endif
1930 size=foundsize;
1931 }
1932 }
1933#endif //MOV_MINOLTA_FIX
1934
e2b9cf4e
MN		 1935 idx = sc->sample_to_chunk_index;
1936 if (idx + 1 < sc->sample_to_chunk_sz && sc->next_chunk >= sc->sample_to_chunk[idx + 1].first)
1937 idx++;
1938 sc->sample_to_chunk_index = idx;
3ffe3793
FR		 1939#ifdef MOV_SPLIT_CHUNKS
1940 /* split chunks into samples */
e2b9cf4e 1941 if (sc->sample_size == 0 || sc->sample_size > 100) {
7dbab4a9
FR		 1942 if (idx >= 0 && sc->sample_to_chunk[idx].count != 1) {
1943 mov->partial = sc;
1944 /* we'll have to get those samples before next chunk */
1945 sc->left_in_chunk = sc->sample_to_chunk[idx].count - 1;
de39cdf5 1946 size = (sc->sample_size > 1)?sc->sample_size:sc->sample_sizes[sc->current_sample];
7dbab4a9 1947 }
7619ed2b 1948
7dbab4a9 1949 sc->current_sample++;
e2b9cf4e
MN		 1950 }else if(idx + 1 < sc->sample_to_chunk_sz){
1951 sc->current_sample += sc->sample_size * sc->sample_to_chunk[idx].count;
7dbab4a9 1952 }
3ffe3793
FR		 1953#endif
1954
1955readchunk:
c3ea71a0
FR		 1956#ifdef DEBUG
1957 av_log(NULL, AV_LOG_DEBUG, "chunk: %lli -> %lli (%i)\n", offset, offset + size, size);
1958#endif
6cea494e
ZK		 1959 if(size == 0x0FFFFFFF)
1960 size = mov->mdat_size + mov->mdat_offset - offset;
1961 if(size < 0)
1962 return -1;
1963 if(size == 0)
1964 return -1;
0e7eed09 1965 url_fseek(&s->pb, offset, SEEK_SET);
5cd62665 1966
2692067a 1967 av_get_packet(&s->pb, pkt, size);
0e7eed09 1968 pkt->stream_index = sc->ffindex;
247d56f5
BB		 1969
1970 // If the keyframes table exists, mark any samples that are in the table as key frames.
1971 // If no table exists, treat very sample as a key frame.
1972 if (sc->keyframes) {
1973 a = 0;
1974 b = sc->keyframe_count - 1;
1975
1976 while (a < b) {
1977 m = (a + b + 1) >> 1;
1978 if (sc->keyframes[m] > sc->current_sample) {
1979 b = m - 1;
1980 } else {
1981 a = m;
1982 }
1983 }
1984
1985 if (sc->keyframes[a] == sc->current_sample)
1986 pkt->flags |= PKT_FLAG_KEY;
1987 }
1988 else
1989 pkt->flags |= PKT_FLAG_KEY;
6cea494e
ZK		 1990
1991#ifdef DEBUG
7dbab4a9 1992/*
c3ea71a0 1993 av_log(NULL, AV_LOG_DEBUG, "Packet (%d, %ld) ", pkt->stream_index, pkt->size);
6cea494e 1994 for(i=0; i<8; i++)
baf25c9d 1995 av_log(NULL, AV_LOG_DEBUG, "%02x ", pkt->data[i]);
6cea494e 1996 for(i=0; i<8; i++)
baf25c9d 1997 av_log(NULL, AV_LOG_DEBUG, "%c ", (pkt->data[i]) & 0x7F);
c3ea71a0 1998 av_log(NULL, AV_LOG_DEBUG, "\n");
7dbab4a9 1999*/
6cea494e
ZK		 2000#endif
2001
2002 mov->next_chunk_offset = offset + size;
7fe5a3c0
GC		 2003
2004 /* find the corresponding dts */
2005 if (sc && sc->sample_to_time_index < sc->stts_count && pkt) {
cd461d48 2006 unsigned int count;
c4ac052b 2007 uint64_t dts, pts;
cd461d48
MN		 2008 unsigned int duration = sc->stts_data[sc->sample_to_time_index].duration;
2009 count = sc->stts_data[sc->sample_to_time_index].count;
7fe5a3c0
GC		 2010 if ((sc->sample_to_time_sample + count) < sc->current_sample) {
2011 sc->sample_to_time_sample += count;
2012 sc->sample_to_time_time += count*duration;
2013 sc->sample_to_time_index ++;
cd461d48 2014 duration = sc->stts_data[sc->sample_to_time_index].duration;
7fe5a3c0 2015 }
cd461d48 2016 dts = sc->sample_to_time_time + (sc->current_sample-1 - sc->sample_to_time_sample) * (int64_t)duration;
c4ac052b
MN		 2017 /* find the corresponding pts */
2018 if (sc->sample_to_ctime_index < sc->ctts_count) {
2019 int duration = sc->ctts_data[sc->sample_to_ctime_index].duration;
2020 int count = sc->ctts_data[sc->sample_to_ctime_index].count;
2021
2022 if ((sc->sample_to_ctime_sample + count) < sc->current_sample) {
2023 sc->sample_to_ctime_sample += count;
2024 sc->sample_to_ctime_index ++;
2025 duration = sc->ctts_data[sc->sample_to_ctime_index].duration;
2026 }
2027 pts = dts + duration;
2028 }else
2029 pts = dts;
961e0ccd
MN		 2030 pkt->pts = pts;
2031 pkt->dts = dts;
c4ac052b
MN		 2032#ifdef DEBUG
2033 av_log(NULL, AV_LOG_DEBUG, "stream #%d smp #%ld dts = %lld pts = %lld (smp:%ld time:%lld idx:%d ent:%d count:%d dur:%d)\n"
2034 , pkt->stream_index, sc->current_sample-1, pkt->dts, pkt->pts
2035 , sc->sample_to_time_sample
2036 , sc->sample_to_time_time
2037 , sc->sample_to_time_index
2038 , sc->stts_count
2039 , count
2040 , duration);
2041#endif
2042 }
6cea494e
ZK		 2043
2044 return 0;
2045}
2046
baf25c9d
GC		 2047#if defined(MOV_SPLIT_CHUNKS) && defined(MOV_SEEK)
2048/**
2049 * Seek method based on the one described in the Appendix C of QTFileFormat.pdf
2050 */
961e0ccd 2051static int mov_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
baf25c9d
GC		 2052{
2053 MOVContext* mov = (MOVContext *) s->priv_data;
2054 MOVStreamContext* sc;
2055 int32_t i, a, b, m;
baf25c9d
GC		 2056 int64_t start_time;
2057 int32_t seek_sample, sample;
2058 int32_t duration;
2059 int32_t count;
2060 int32_t chunk;
2061 int32_t left_in_chunk;
2062 int64_t chunk_file_offset;
2063 int64_t sample_file_offset;
2064 int32_t first_chunk_sample;
2065 int32_t sample_to_chunk_idx;
7fe5a3c0
GC		 2066 int sample_to_time_index;
2067 long sample_to_time_sample = 0;
2068 uint64_t sample_to_time_time = 0;
baf25c9d
GC		 2069 int mov_idx;
2070
2071 // Find the corresponding mov stream
2072 for (mov_idx = 0; mov_idx < mov->total_streams; mov_idx++)
2073 if (mov->streams[mov_idx]->ffindex == stream_index)
2074 break;
2075 if (mov_idx == mov->total_streams) {
2076 av_log(s, AV_LOG_ERROR, "mov: requested stream was not found in mov streams (idx=%i)\n", stream_index);
2077 return -1;
2078 }
2079 sc = mov->streams[mov_idx];
2080
2081 // Step 1. Find the edit that contains the requested time (elst)
2082 if (sc->edit_count) {
2083 // FIXME should handle edit list
2084 av_log(s, AV_LOG_ERROR, "mov: does not handle seeking in files that contain edit list (c:%d)\n", sc->edit_count);
2085 return -1;
2086 }
2087
2088 // Step 2. Find the corresponding sample using the Time-to-sample atom (stts) */
2089#ifdef DEBUG
2090 av_log(s, AV_LOG_DEBUG, "Searching for time %li in stream #%i (time_scale=%i)\n", (long)timestamp, mov_idx, sc->time_scale);
2091#endif
baf25c9d
GC		 2092 start_time = 0; // FIXME use elst atom
2093 sample = 1; // sample are 0 based in table
2094#ifdef DEBUG
2095 av_log(s, AV_LOG_DEBUG, "Searching for sample_time %li \n", (long)sample_time);
2096#endif
2097 for (i = 0; i < sc->stts_count; i++) {
cd461d48
MN		 2098 count = sc->stts_data[i].count;
2099 duration = sc->stts_data[i].duration;
baf25c9d
GC		 2100//av_log(s, AV_LOG_DEBUG, "> sample_time %lli \n", (long)sample_time);
2101//av_log(s, AV_LOG_DEBUG, "> count=%i duration=%i\n", count, duration);
2102 if ((start_time + count*duration) > sample_time) {
7fe5a3c0
GC		 2103 sample_to_time_time = start_time;
2104 sample_to_time_index = i;
2105 sample_to_time_sample = sample;
baf25c9d
GC		 2106 sample += (sample_time - start_time) / duration;
2107 break;
2108 }
2109 sample += count;
2110 start_time += count * duration;
2111 }
7fe5a3c0
GC		 2112 sample_to_time_time = start_time;
2113 sample_to_time_index = i;
baf25c9d
GC		 2114 /* NOTE: despite what qt doc say, the dt value (Display Time in qt vocabulary) computed with the stts atom
2115 is a decoding time stamp (dts) not a presentation time stamp. And as usual dts != pts for stream with b frames */
2116
2117#ifdef DEBUG
2118 av_log(s, AV_LOG_DEBUG, "Found time %li at sample #%u\n", (long)sample_time, sample);
2119#endif
2120 if (sample > sc->sample_count) {
2121 av_log(s, AV_LOG_ERROR, "mov: sample pos is too high, unable to seek (req. sample=%i, sample count=%ld)\n", sample, sc->sample_count);
2122 return -1;
2123 }
2124
2125 // Step 3. Find the prior sync. sample using the Sync sample atom (stss)
2126 if (sc->keyframes) {
2127 a = 0;
2128 b = sc->keyframe_count - 1;
2129 while (a < b) {
2130 m = (a + b + 1) >> 1;
2131 if (sc->keyframes[m] > sample) {
2132 b = m - 1;
2133 } else {
2134 a = m;
2135 }
2136#ifdef DEBUG
2137 // av_log(s, AV_LOG_DEBUG, "a=%i (%i) b=%i (%i) m=%i (%i) stream #%i\n", a, sc->keyframes[a], b, sc->keyframes[b], m, sc->keyframes[m], mov_idx);
2138#endif
2139 }
7fe5a3c0
GC		 2140 // for low latency prob: always use the previous keyframe, just uncomment the next line
2141 // if (a) a--;
baf25c9d
GC		 2142 seek_sample = sc->keyframes[a];
2143 }
2144 else
2145 seek_sample = sample; // else all samples are key frames
2146#ifdef DEBUG
2147 av_log(s, AV_LOG_DEBUG, "Found nearest keyframe at sample #%i \n", seek_sample);
2148#endif
2149
2150 // Step 4. Find the chunk of the sample using the Sample-to-chunk-atom (stsc)
2151 for (first_chunk_sample = 1, i = 0; i < (sc->sample_to_chunk_sz - 1); i++) {
2152 b = (sc->sample_to_chunk[i + 1].first - sc->sample_to_chunk[i].first) * sc->sample_to_chunk[i].count;
2153 if (seek_sample >= first_chunk_sample && seek_sample < (first_chunk_sample + b))
2154 break;
2155 first_chunk_sample += b;
2156 }
2157 chunk = sc->sample_to_chunk[i].first + (seek_sample - first_chunk_sample) / sc->sample_to_chunk[i].count;
2158 left_in_chunk = sc->sample_to_chunk[i].count - (seek_sample - first_chunk_sample) % sc->sample_to_chunk[i].count;
2159 first_chunk_sample += ((seek_sample - first_chunk_sample) / sc->sample_to_chunk[i].count) * sc->sample_to_chunk[i].count;
2160 sample_to_chunk_idx = i;
2161#ifdef DEBUG
2162 av_log(s, AV_LOG_DEBUG, "Sample was found in chunk #%i at sample offset %i (idx %i)\n", chunk, seek_sample - first_chunk_sample, sample_to_chunk_idx);
2163#endif
2164
2165 // Step 5. Find the offset of the chunk using the chunk offset atom
2166 if (!sc->chunk_offsets) {
2167 av_log(s, AV_LOG_ERROR, "mov: no chunk offset atom, unable to seek\n");
2168 return -1;
2169 }
2170 if (chunk > sc->chunk_count) {
2171 av_log(s, AV_LOG_ERROR, "mov: chunk offset atom too short, unable to seek (req. chunk=%i, chunk count=%li)\n", chunk, sc->chunk_count);
2172 return -1;
2173 }
2174 chunk_file_offset = sc->chunk_offsets[chunk - 1];
2175#ifdef DEBUG
2176 av_log(s, AV_LOG_DEBUG, "Chunk file offset is #%llu \n", chunk_file_offset);
2177#endif
2178
2179 // Step 6. Find the byte offset within the chunk using the sample size atom
2180 sample_file_offset = chunk_file_offset;
2181 if (sc->sample_size)
2182 sample_file_offset += (seek_sample - first_chunk_sample) * sc->sample_size;
2183 else {
2184 for (i = 0; i < (seek_sample - first_chunk_sample); i++) {
2185 sample_file_offset += sc->sample_sizes[first_chunk_sample + i - 1];
2186 }
2187 }
2188#ifdef DEBUG
2189 av_log(s, AV_LOG_DEBUG, "Sample file offset is #%llu \n", sample_file_offset);
2190#endif
2191
2192 // Step 6. Update the parser
2193 mov->partial = sc;
2194 mov->next_chunk_offset = sample_file_offset;
2195 // Update current stream state
2196 sc->current_sample = seek_sample - 1; // zero based
2197 sc->left_in_chunk = left_in_chunk;
2198 sc->next_chunk = chunk; // +1 -1 (zero based)
2199 sc->sample_to_chunk_index = sample_to_chunk_idx;
2200
2201 // Update other streams
2202 for (i = 0; i<mov->total_streams; i++) {
5c030d3e 2203 MOVStreamContext *msc;
baf25c9d
GC		 2204 if (i == mov_idx) continue;
2205 // Find the nearest 'next' chunk
5c030d3e 2206 msc = mov->streams[i];
baf25c9d
GC		 2207 a = 0;
2208 b = msc->chunk_count - 1;
2209 while (a < b) {
2210 m = (a + b + 1) >> 1;
2211 if (msc->chunk_offsets[m] > chunk_file_offset) {
2212 b = m - 1;
2213 } else {
2214 a = m;
2215 }
2216#ifdef DEBUG
2217/* av_log(s, AV_LOG_DEBUG, "a=%i (%li) b=%i (%li) m=%i (%li) stream #%i\n"
2218 , a, (long)msc->chunk_offsets[a], b, (long)msc->chunk_offsets[b], m, (long)msc->chunk_offsets[m], i); */
2219#endif
2220 }
2221 msc->next_chunk = a;
2222 if (msc->chunk_offsets[a] < chunk_file_offset && a < (msc->chunk_count-1))
2223 msc->next_chunk ++;
2224#ifdef DEBUG
2225 av_log(s, AV_LOG_DEBUG, "Nearest next chunk for stream #%i is #%i @%lli\n", i, msc->next_chunk+1, msc->chunk_offsets[msc->next_chunk]);
2226#endif
2227 // Compute sample count and index in the sample_to_chunk table (what a pity)
2228 msc->sample_to_chunk_index = 0;
2229 msc->current_sample = 0;
2230 for(; msc->sample_to_chunk_index < (msc->sample_to_chunk_sz - 1)
2231 && msc->sample_to_chunk[msc->sample_to_chunk_index + 1].first <= (1 + msc->next_chunk); msc->sample_to_chunk_index++) {
2232 msc->current_sample += (msc->sample_to_chunk[msc->sample_to_chunk_index + 1].first - msc->sample_to_chunk[msc->sample_to_chunk_index].first) \
2233 * msc->sample_to_chunk[msc->sample_to_chunk_index].count;
2234 }
2235 msc->current_sample += (msc->next_chunk - (msc->sample_to_chunk[msc->sample_to_chunk_index].first - 1)) * sc->sample_to_chunk[msc->sample_to_chunk_index].count;
2236 msc->left_in_chunk = msc->sample_to_chunk[msc->sample_to_chunk_index].count - 1;
7fe5a3c0
GC		 2237 // Find corresponding position in stts (used later to compute dts)
2238 sample = 0;
2239 start_time = 0;
2240 for (msc->sample_to_time_index = 0; msc->sample_to_time_index < msc->stts_count; msc->sample_to_time_index++) {
cd461d48
MN		 2241 count = msc->stts_data[msc->sample_to_time_index].count;
2242 duration = msc->stts_data[msc->sample_to_time_index].duration;
7fe5a3c0
GC		 2243 if ((sample + count - 1) > msc->current_sample) {
2244 msc->sample_to_time_time = start_time;
2245 msc->sample_to_time_sample = sample;
2246 break;
2247 }
2248 sample += count;
2249 start_time += count * duration;
c4ac052b
MN		 2250 }
2251 sample = 0;
2252 for (msc->sample_to_ctime_index = 0; msc->sample_to_ctime_index < msc->ctts_count; msc->sample_to_ctime_index++) {
2253 count = msc->ctts_data[msc->sample_to_ctime_index].count;
2254 duration = msc->ctts_data[msc->sample_to_ctime_index].duration;
2255 if ((sample + count - 1) > msc->current_sample) {
2256 msc->sample_to_ctime_sample = sample;
2257 break;
2258 }
2259 sample += count;
2260 }
baf25c9d
GC		 2261#ifdef DEBUG
2262 av_log(s, AV_LOG_DEBUG, "Next Sample for stream #%i is #%i @%i\n", i, msc->current_sample + 1, msc->sample_to_chunk_index + 1);
2263#endif
2264 }
2265 return 0;
2266}
2267#endif
2268
a266644f 2269static int mov_read_close(AVFormatContext *s)
6cea494e
ZK		 2270{
2271 int i;
b6a17df4 2272 MOVContext *mov = (MOVContext *) s->priv_data;
6cea494e
ZK		 2273 for(i=0; i<mov->total_streams; i++)
2274 mov_free_stream_context(mov->streams[i]);
5ca1d879
ZK		 2275 /* free color tabs */
2276 for(i=0; i<mov->ctab_size; i++)
2277 av_freep(&mov->ctab[i]);
2278 av_freep(&mov->ctab);
6cea494e
ZK		 2279 return 0;
2280}
2281
c9a65ca8 2282static AVInputFormat mov_iformat = {
8536ab89 2283 "mov,mp4,m4a,3gp,3g2",
c9a65ca8
FB		 2284 "QuickTime/MPEG4 format",
2285 sizeof(MOVContext),
2286 mov_probe,
6cea494e
ZK		 2287 mov_read_header,
2288 mov_read_packet,
2289 mov_read_close,
baf25c9d
GC		 2290#if defined(MOV_SPLIT_CHUNKS) && defined(MOV_SEEK)
2291 mov_read_seek,
2292#endif
6cea494e
ZK		 2293};
2294
c9a65ca8
FB		 2295int mov_init(void)
2296{
2297 av_register_input_format(&mov_iformat);
2298 return 0;
2299}
Libav master