falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2073 Commits
32 Branches
303MB
Tree: a466e345e4
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'a466e345e4'
${ noResults }
FFmpeg/libavformat/mov.c

1494 lines
47KB
Raw Blame History 

  1. /*

  2.  * MOV decoder.

  3.  * Copyright (c) 2001 Fabrice Bellard.

  4.  *

  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.

  9.  *

  10.  * This library is distributed in the hope that it will be useful,

  11.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  12.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

  13.  * Lesser General Public License for more details.

  14.  *

  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

  18.  */

  19. #include "avformat.h"

  20. #include "avi.h"

  21. 

  22. #ifdef CONFIG_ZLIB

  23. #include <zlib.h>

  24. #endif

  25. 

  26. /*

  27.  * First version by Francois Revol revol@free.fr

  28.  * 

  29.  * Features and limitations:

  30.  * - reads most of the QT files I have (at least the structure), 

  31.  *   the exceptions are .mov with zlib compressed headers ('cmov' section). It shouldn't be hard to implement.

  32.  *   FIXED, Francois Revol, 07/17/2002

  33.  * - ffmpeg has nearly none of the usual QuickTime codecs,

  34.  *   although I succesfully dumped raw and mp3 audio tracks off .mov files.

  35.  *   Sample QuickTime files with mp3 audio can be found at: http://www.3ivx.com/showcase.html

  36.  * - .mp4 parsing is still hazardous, although the format really is QuickTime with some minor changes

  37.  *   (to make .mov parser crash maybe ?), despite what they say in the MPEG FAQ at

  38.  *   http://mpeg.telecomitalialab.com/faq.htm

  39.  * - the code is quite ugly... maybe I won't do it recursive next time :-)

  40.  * 

  41.  * Funny I didn't know about http://sourceforge.net/projects/qt-ffmpeg/

  42.  * when coding this :) (it's a writer anyway)

  43.  * 

  44.  * Reference documents:

  45.  * http://www.geocities.com/xhelmboyx/quicktime/formats/qtm-layout.txt

  46.  * Apple:

  47.  *  http://developer.apple.com/techpubs/quicktime/qtdevdocs/QTFF/qtff.html

  48.  *  http://developer.apple.com/techpubs/quicktime/qtdevdocs/PDF/QTFileFormat.pdf

  49.  * QuickTime is a trademark of Apple (AFAIK :))

  50.  */

  51. 

  52. //#define DEBUG

  53. 

  54. /* allows chunk splitting - should work now... */

  55. /* in case you can't read a file, try commenting */

  56. #define MOV_SPLIT_CHUNKS

  57. 

  58. #ifdef DEBUG

  59. /*

  60.  * XXX: static sux, even more in a multithreaded environment...

  61.  * Avoid them. This is here just to help debugging.

  62.  */

  63. static int debug_indent = 0;

  64. void print_atom(const char *str, uint32_t type, uint64_t offset, uint64_t size)

  65. {

  66.     unsigned int tag, i;

  67.     tag = (unsigned int) type;

  68.     i=debug_indent;

  69.     if(tag == 0) tag = MKTAG('N', 'U', 'L', 'L');

  70.     while(i--)

  71.         printf("|");

  72.     printf("parse:");

  73.     printf(" %s: tag=%c%c%c%c offset=0x%x size=0x%x\n",

  74.            str, tag & 0xff,

  75.            (tag >> 8) & 0xff,

  76.            (tag >> 16) & 0xff,

  77.            (tag >> 24) & 0xff,

  78.            (unsigned int)offset,

  79. 	   (unsigned int)size);

  80.     assert((unsigned int)size < 0x7fffffff);// catching errors

  81. }

  82. #else

  83. #define print_atom(a,b,c,d)

  84. #endif

  85. 

  86. /* some streams in QT (and in MP4 mostly) aren't either video nor audio */

  87. /* so we first list them as this, then clean up the list of streams we give back, */

  88. /* getting rid of these */

  89. #define CODEC_TYPE_MOV_OTHER	(enum CodecType) 2

  90. 

  91. static const CodecTag mov_video_tags[] = {

  92. /*  { CODEC_ID_, MKTAG('c', 'v', 'i', 'd') }, *//* Cinepak */

  93. /*  { CODEC_ID_H263, MKTAG('r', 'a', 'w', ' ') }, *//* Uncompressed RGB */

  94. /*  { CODEC_ID_H263, MKTAG('Y', 'u', 'v', '2') }, *//* Uncompressed YUV422 */

  95. /*    { CODEC_ID_RAWVIDEO, MKTAG('A', 'V', 'U', 'I') }, *//* YUV with alpha-channel (AVID Uncompressed) */

  96. /* Graphics */

  97. /* Animation */

  98. /* Apple video */

  99. /* Kodak Photo CD */

  100.     { CODEC_ID_MJPEG, MKTAG('j', 'p', 'e', 'g') }, /* PhotoJPEG */

  101.     { CODEC_ID_MPEG1VIDEO, MKTAG('m', 'p', 'e', 'g') }, /* MPEG */

  102.     { CODEC_ID_MJPEG, MKTAG('m', 'j', 'p', 'a') }, /* Motion-JPEG (format A) */

  103.     { CODEC_ID_MJPEG, MKTAG('m', 'j', 'p', 'b') }, /* Motion-JPEG (format B) */

  104.     { CODEC_ID_MJPEG, MKTAG('A', 'V', 'D', 'J') }, /* MJPEG with alpha-channel (AVID JFIF meridien compressed) */

  105. /*    { CODEC_ID_MJPEG, MKTAG('A', 'V', 'R', 'n') }, *//* MJPEG with alpha-channel (AVID ABVB/Truevision NuVista) */

  106. /*    { CODEC_ID_GIF, MKTAG('g', 'i', 'f', ' ') }, *//* embedded gif files as frames (usually one "click to play movie" frame) */

  107. /* Sorenson video */

  108.     { CODEC_ID_SVQ1, MKTAG('S', 'V', 'Q', '1') }, /* Sorenson Video v1 */

  109.     { CODEC_ID_SVQ1, MKTAG('s', 'v', 'q', '1') }, /* Sorenson Video v1 */

  110.     { CODEC_ID_SVQ1, MKTAG('s', 'v', 'q', 'i') }, /* Sorenson Video v1 (from QT specs)*/

  111.     { CODEC_ID_MPEG4, MKTAG('m', 'p', '4', 'v') },

  112.     { CODEC_ID_MPEG4, MKTAG('D', 'I', 'V', 'X') }, /* OpenDiVX *//* sample files at http://heroinewarrior.com/xmovie.php3 use this tag */

  113. /*    { CODEC_ID_, MKTAG('I', 'V', '5', '0') }, *//* Indeo 5.0 */

  114.     { CODEC_ID_H263, MKTAG('h', '2', '6', '3') }, /* H263 */

  115.     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'c', ' ') }, /* DV NTSC */

  116.     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'c', 'p') }, /* DV PAL */

  117. /*    { CODEC_ID_DVVIDEO, MKTAG('A', 'V', 'd', 'v') }, *//* AVID dv */

  118.     { CODEC_ID_VP3, MKTAG('V', 'P', '3', '1') }, /* On2 VP3 */

  119.     { 0, 0 },

  120. };

  121. 

  122. static const CodecTag mov_audio_tags[] = {

  123. /*    { CODEC_ID_PCM_S16BE, MKTAG('N', 'O', 'N', 'E') }, *//* uncompressed */

  124.     { CODEC_ID_PCM_S16BE, MKTAG('t', 'w', 'o', 's') }, /* 16 bits */

  125.     { CODEC_ID_PCM_S8, MKTAG('t', 'w', 'o', 's') }, /* 8 bits */

  126.     { CODEC_ID_PCM_U8, MKTAG('r', 'a', 'w', ' ') }, /* 8 bits unsigned */

  127.     { CODEC_ID_PCM_S16LE, MKTAG('s', 'o', 'w', 't') }, /*  */

  128.     { CODEC_ID_PCM_MULAW, MKTAG('u', 'l', 'a', 'w') }, /*  */

  129.     { CODEC_ID_PCM_ALAW, MKTAG('a', 'l', 'a', 'w') }, /*  */

  130.     { CODEC_ID_ADPCM_IMA_QT, MKTAG('i', 'm', 'a', '4') }, /* IMA-4 ADPCM */

  131.     { CODEC_ID_MACE3, MKTAG('M', 'A', 'C', '3') }, /* Macintosh Audio Compression and Expansion 3:1 */

  132.     { CODEC_ID_MACE6, MKTAG('M', 'A', 'C', '6') }, /* Macintosh Audio Compression and Expansion 6:1 */

  133. 

  134.     { CODEC_ID_MP2, MKTAG('.', 'm', 'p', '3') }, /* MPEG layer 3 */ /* sample files at http://www.3ivx.com/showcase.html use this tag */

  135.     { CODEC_ID_MP2, 0x6D730055 }, /* MPEG layer 3 */

  136.     { CODEC_ID_MP2, 0x5500736D }, /* MPEG layer 3 *//* XXX: check endianness */

  137. /*    { CODEC_ID_OGG_VORBIS, MKTAG('O', 'g', 'g', 'S') }, *//* sample files at http://heroinewarrior.com/xmovie.php3 use this tag */

  138. /* MP4 tags */

  139. /*    { CODEC_ID_AAC, MKTAG('m', 'p', '4', 'a') }, *//* MPEG 4 AAC or audio ? */

  140.                                                      /* The standard for mpeg4 audio is still not normalised AFAIK anyway */

  141.     { 0, 0 }, 

  142. };

  143. 

  144. /* the QuickTime file format is quite convoluted...

  145.  * it has lots of index tables, each indexing something in another one...

  146.  * Here we just use what is needed to read the chunks

  147.  */

  148. 

  149. typedef struct MOV_sample_to_chunk_tbl {

  150.     long first;

  151.     long count;

  152.     long id;

  153. } MOV_sample_to_chunk_tbl;

  154. 

  155. typedef struct {

  156.     uint8_t  version;

  157.     uint32_t flags; // 24bit

  158. 

  159.     /* 0x03 ESDescrTag */

  160.     uint16_t es_id;

  161. #define MP4ODescrTag			0x01

  162. #define MP4IODescrTag			0x02

  163. #define MP4ESDescrTag			0x03

  164. #define MP4DecConfigDescrTag		0x04

  165. #define MP4DecSpecificDescrTag		0x05

  166. #define MP4SLConfigDescrTag		0x06

  167. #define MP4ContentIdDescrTag		0x07

  168. #define MP4SupplContentIdDescrTag	0x08

  169. #define MP4IPIPtrDescrTag		0x09

  170. #define MP4IPMPPtrDescrTag		0x0A

  171. #define MP4IPMPDescrTag			0x0B

  172. #define MP4RegistrationDescrTag		0x0D

  173. #define MP4ESIDIncDescrTag		0x0E

  174. #define MP4ESIDRefDescrTag		0x0F

  175. #define MP4FileIODescrTag		0x10

  176. #define MP4FileODescrTag		0x11

  177. #define MP4ExtProfileLevelDescrTag	0x13

  178. #define MP4ExtDescrTagsStart		0x80

  179. #define MP4ExtDescrTagsEnd		0xFE

  180.     uint8_t  stream_priority;

  181. 

  182.     /* 0x04 DecConfigDescrTag */

  183.     uint8_t  object_type_id;

  184.     uint8_t  stream_type;

  185.     /* XXX: really streamType is

  186.      * only 6bit, followed by:

  187.      * 1bit  upStream

  188.      * 1bit  reserved

  189.      */

  190.     uint32_t buffer_size_db; // 24

  191.     uint32_t max_bitrate;

  192.     uint32_t avg_bitrate;

  193. 

  194.     /* 0x05 DecSpecificDescrTag */

  195.     uint8_t  decoder_cfg_len;

  196.     uint8_t *decoder_cfg;

  197. 

  198.     /* 0x06 SLConfigDescrTag */

  199.     uint8_t  sl_config_len;

  200.     uint8_t *sl_config;

  201. } MOV_esds_t;

  202. 

  203. struct MOVParseTableEntry;

  204. 

  205. typedef struct MOVStreamContext {

  206.     int ffindex; /* the ffmpeg stream id */

  207.     int is_ff_stream; /* Is this stream presented to ffmpeg ? i.e. is this an audio or video stream ? */

  208.     long next_chunk;

  209.     long chunk_count;

  210.     int64_t *chunk_offsets;

  211.     long sample_to_chunk_sz;

  212.     MOV_sample_to_chunk_tbl *sample_to_chunk;

  213.     long sample_to_chunk_index;

  214.     long sample_size;

  215.     long sample_count;

  216.     long *sample_sizes;

  217.     int time_scale;

  218.     long current_sample;

  219.     long left_in_chunk; /* how many samples before next chunk */

  220.     /* specific MPEG4 header which is added at the beginning of the stream */

  221.     int header_len;

  222.     MOV_esds_t esds;

  223.     uint8_t *header_data;

  224. } MOVStreamContext;

  225. 

  226. typedef struct MOVContext {

  227.     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) */

  228.     AVFormatContext *fc;

  229.     int time_scale;

  230.     int duration; /* duration of the longest track */

  231.     int found_moov; /* when both 'moov' and 'mdat' sections has been found */

  232.     int found_mdat; /* we suppose we have enough data to read the file */

  233.     int64_t mdat_size;

  234.     int64_t mdat_offset;

  235.     int total_streams;

  236.     /* some streams listed here aren't presented to the ffmpeg API, since they aren't either video nor audio

  237.      * but we need the info to be able to skip data from those streams in the 'mdat' section

  238.      */

  239.     MOVStreamContext *streams[MAX_STREAMS];

  240. 

  241.     int64_t next_chunk_offset;

  242.     MOVStreamContext *partial; /* != 0 : there is still to read in the current chunk */

  243.     const struct MOVParseTableEntry *parse_table; /* could be eventually used to change the table */

  244.     /* NOTE: for recursion save to/ restore from local variable! */

  245. } MOVContext;

  246. 

  247. 

  248. /* XXX: it's the first time I make a recursive parser I think... sorry if it's ugly :P */

  249. 

  250. /* those functions parse an atom */

  251. /* return code:

  252.  1: found what I wanted, exit

  253.  0: continue to parse next atom

  254.  -1: error occured, exit

  255.  */

  256. typedef int (*mov_parse_function)(MOVContext *ctx,

  257.                                   ByteIOContext *pb,

  258.                                   uint32_t atom_type,

  259.                                   int64_t atom_offset, /* after the size and type field (and eventually the extended size) */

  260.                                   int64_t atom_size); /* total size (excluding the size and type fields) */

  261. 

  262. /* links atom IDs to parse functions */

  263. typedef struct MOVParseTableEntry {

  264.     uint32_t type;

  265.     mov_parse_function func;

  266. } MOVParseTableEntry;

  267. 

  268. static int parse_leaf(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  269. {

  270.     print_atom("leaf", atom_type, atom_offset, atom_size);

  271. 

  272.     if(atom_size>1)

  273.         url_fskip(pb, atom_size);

  274. /*        url_seek(pb, atom_offset+atom_size, SEEK_SET); */

  275.     return 0;

  276. }

  277. 

  278. static int parse_default(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  279. {

  280.     uint32_t type, foo=0;

  281.     uint64_t offset, size;

  282.     int64_t total_size = 0;

  283.     int i;

  284.     int err = 0;

  285.     foo=0;

  286. #ifdef DEBUG

  287.     print_atom("default", atom_type, atom_offset, atom_size);

  288.     debug_indent++;

  289. #endif

  290. 

  291.     offset = atom_offset;

  292. 

  293.     if(atom_size < 0)

  294.         atom_size = 0x0FFFFFFFFFFFFFFF;

  295.     while(((total_size + 8) < atom_size) && !url_feof(pb) && !err) {

  296.         size=atom_size;

  297.         type=0L;

  298.         if(atom_size >= 8) {

  299. 	    size = get_be32(pb);

  300.             type = get_le32(pb);

  301.         }

  302. 	total_size += 8;

  303.         offset += 8;

  304. 	//printf("type: %08x  %.4s  sz: %Lx  %Lx   %Lx\n", type, (char*)&type, size, atom_size, total_size);

  305.         if(size == 1) { /* 64 bit extended size */

  306.             size = get_be64(pb) - 8;

  307.             offset+=8;

  308.             total_size+=8;

  309.         }

  310. 	if(size == 0) {

  311. 	    size = atom_size - total_size;

  312. 	    if (size <= 8)

  313.                 break;

  314. 	}

  315. 	for (i=0; c->parse_table[i].type != 0L && c->parse_table[i].type != type; i++)

  316. 	    /* empty */;

  317. 

  318.         size -= 8;

  319. //        printf(" i=%ld\n", i);

  320. 	if (c->parse_table[i].type == 0) { /* skip leaf atoms data */

  321. //            url_seek(pb, atom_offset+atom_size, SEEK_SET);

  322. #ifdef DEBUG

  323.             print_atom("unknown", type, offset, size);

  324. #endif

  325.             url_fskip(pb, size);

  326. 	} else {

  327. #ifdef DEBUG

  328. 	    //char b[5] = { type & 0xff, (type >> 8) & 0xff, (type >> 16) & 0xff, (type >> 24) & 0xff, 0 };

  329. 	    //print_atom(b, type, offset, size);

  330. #endif

  331. 	    err = (c->parse_table[i].func)(c, pb, type, offset, size);

  332. 	}

  333. 

  334.         offset+=size;

  335.         total_size+=size;

  336.     }

  337. 

  338.     if (!err && total_size < atom_size && atom_size < 0x7ffff) {

  339. 	printf("RESET  %Ld  %Ld  err:%d\n", atom_size, total_size, err);

  340.         url_fskip(pb, atom_size - total_size);

  341.     }

  342. 

  343. #ifdef DEBUG

  344.     debug_indent--;

  345. #endif

  346.     return err;

  347. }

  348. 

  349. static int parse_ctab(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  350. {

  351.     url_fskip(pb, atom_size); // for now

  352.     return 0;

  353. }

  354. 

  355. static int parse_mvhd(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  356. {

  357.     print_atom("mvhd", atom_type, atom_offset, atom_size);

  358. 

  359.     get_byte(pb); /* version */

  360.     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */

  361. 

  362.     get_be32(pb); /* creation time */

  363.     get_be32(pb); /* modification time */

  364.     c->time_scale = get_be32(pb); /* time scale */

  365. #ifdef DEBUG

  366.     printf("time scale = %i\n", c->time_scale);

  367. #endif

  368.     c->duration = get_be32(pb); /* duration */

  369.     get_be32(pb); /* preferred scale */

  370. 

  371.     get_be16(pb); /* preferred volume */

  372. 

  373.     url_fskip(pb, 10); /* reserved */

  374. 

  375.     url_fskip(pb, 36); /* display matrix */

  376. 

  377.     get_be32(pb); /* preview time */

  378.     get_be32(pb); /* preview duration */

  379.     get_be32(pb); /* poster time */

  380.     get_be32(pb); /* selection time */

  381.     get_be32(pb); /* selection duration */

  382.     get_be32(pb); /* current time */

  383.     get_be32(pb); /* next track ID */

  384.     

  385.     return 0;

  386. }

  387. 

  388. /* this atom should contain all header atoms */

  389. static int parse_moov(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  390. {

  391.     int err;

  392. 

  393.     print_atom("moov", atom_type, atom_offset, atom_size);

  394. 

  395.     err = parse_default(c, pb, atom_type, atom_offset, atom_size);

  396.     /* we parsed the 'moov' atom, we can terminate the parsing as soon as we find the 'mdat' */

  397.     /* so we don't parse the whole file if over a network */

  398.     c->found_moov=1;

  399.     if(c->found_mdat)

  400.         return 1; /* found both, just go */

  401.     return 0; /* now go for mdat */

  402. }

  403. 

  404. /* this atom contains actual media data */

  405. static int parse_mdat(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  406. {

  407.     print_atom("mdat", atom_type, atom_offset, atom_size);

  408. 

  409.     if(atom_size == 0) /* wrong one (MP4) */

  410.         return 0;

  411.     c->found_mdat=1;

  412.     c->mdat_offset = atom_offset;

  413.     c->mdat_size = atom_size;

  414.     if(c->found_moov)

  415.         return 1; /* found both, just go */

  416.     url_fskip(pb, atom_size);

  417.     return 0; /* now go for moov */

  418. }

  419. 

  420. /* this atom should be null (from specs), but some buggy files put the 'moov' atom inside it... */

  421. /* like the files created with Adobe Premiere 5.0, for samples see */

  422. /* http://graphics.tudelft.nl/~wouter/publications/soundtests/ */

  423. static int parse_wide(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  424. {

  425.     int err;

  426.     uint32_t type;

  427. 

  428. #ifdef DEBUG

  429.     print_atom("wide", atom_type, atom_offset, atom_size);

  430.     debug_indent++;

  431. #endif

  432.     if (atom_size < 8)

  433.         return 0; /* continue */

  434.     if (get_be32(pb) != 0) { /* 0 sized mdat atom... use the 'wide' atom size */

  435.         url_fskip(pb, atom_size - 4);

  436.         return 0;

  437.     }

  438.     type = get_le32(pb);

  439.     if (type != MKTAG('m', 'd', 'a', 't')) {

  440.         url_fskip(pb, atom_size - 8);

  441.         return 0;

  442.     }

  443.     err = parse_mdat(c, pb, type, atom_offset + 8, atom_size - 8);

  444. #ifdef DEBUG

  445.     debug_indent--;

  446. #endif

  447.     return err;

  448. }

  449. 

  450. static int parse_trak(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  451. {

  452.     AVStream *st;

  453.     MOVStreamContext *sc;

  454. 

  455.     print_atom("trak", atom_type, atom_offset, atom_size);

  456. 

  457.     st = av_new_stream(c->fc, c->fc->nb_streams);

  458.     if (!st) return -2;

  459.     sc = (MOVStreamContext*) av_mallocz(sizeof(MOVStreamContext));

  460.     if (!sc) {

  461. 	av_free(st);

  462.         return -1;

  463.     }

  464. 

  465.     sc->sample_to_chunk_index = -1;

  466.     st->priv_data = sc;

  467.     st->codec.codec_type = CODEC_TYPE_MOV_OTHER;

  468.     st->time_length = (c->duration * 1000) / c->time_scale; // time in miliseconds

  469.     c->streams[c->fc->nb_streams-1] = sc;

  470.     return parse_default(c, pb, atom_type, atom_offset, atom_size);

  471. }

  472. 

  473. static int parse_tkhd(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  474. {

  475.     AVStream *st;

  476. 

  477.     print_atom("tkhd", atom_type, atom_offset, atom_size);

  478. 

  479.     st = c->fc->streams[c->fc->nb_streams-1];

  480. 

  481.     get_byte(pb); /* version */

  482. 

  483.     get_byte(pb); get_byte(pb);

  484.     get_byte(pb); /* flags */

  485.     /*

  486.     MOV_TRACK_ENABLED 0x0001

  487.     MOV_TRACK_IN_MOVIE 0x0002

  488.     MOV_TRACK_IN_PREVIEW 0x0004

  489.     MOV_TRACK_IN_POSTER 0x0008

  490.     */

  491. 

  492.     get_be32(pb); /* creation time */

  493.     get_be32(pb); /* modification time */

  494.     st->id = (int)get_be32(pb); /* track id (NOT 0 !)*/

  495.     get_be32(pb); /* reserved */

  496.     st->time_length = get_be32(pb) * 1000 / c->time_scale; /* duration */

  497.     get_be32(pb); /* reserved */

  498.     get_be32(pb); /* reserved */

  499. 

  500.     get_be16(pb); /* layer */

  501.     get_be16(pb); /* alternate group */

  502.     get_be16(pb); /* volume */

  503.     get_be16(pb); /* reserved */

  504. 

  505.     url_fskip(pb, 36); /* display matrix */

  506. 

  507.     /* those are fixed-point */

  508.     st->codec.width = get_be32(pb) >> 16; /* track width */

  509.     st->codec.height = get_be32(pb) >> 16; /* track height */

  510.     

  511.     return 0;

  512. }

  513. 

  514. static int parse_mdhd(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  515. {

  516.     AVStream *st;

  517. 

  518.     print_atom("mdhd", atom_type, atom_offset, atom_size);

  519. 

  520.     st = c->fc->streams[c->fc->nb_streams-1];

  521. 

  522.     get_byte(pb); /* version */

  523. 

  524.     get_byte(pb); get_byte(pb);

  525.     get_byte(pb); /* flags */

  526. 

  527.     get_be32(pb); /* creation time */

  528.     get_be32(pb); /* modification time */

  529. 

  530.     c->streams[c->total_streams]->time_scale = get_be32(pb);

  531. 

  532. #ifdef DEBUG

  533.     printf("track[%i].time_scale = %i\n", c->fc->nb_streams-1, c->streams[c->total_streams]->time_scale); /* time scale */

  534. #endif

  535.     get_be32(pb); /* duration */

  536. 

  537.     get_be16(pb); /* language */

  538.     get_be16(pb); /* quality */

  539. 

  540.     return 0;

  541. }

  542. 

  543. static int parse_hdlr(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  544. {

  545.     int len = 0;

  546.     uint8_t *buf;

  547.     uint32_t type;

  548.     AVStream *st;

  549.     uint32_t ctype;

  550. 

  551.     print_atom("hdlr", atom_type, atom_offset, atom_size);

  552. 

  553.     st = c->fc->streams[c->fc->nb_streams-1];

  554. 

  555.     get_byte(pb); /* version */

  556.     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */

  557. 

  558.     /* component type */

  559.     ctype = get_le32(pb);

  560.     type = get_le32(pb); /* component subtype */

  561. 

  562. #ifdef DEBUG

  563.     printf("ctype= %c%c%c%c (0x%08lx)\n", *((char *)&ctype), ((char *)&ctype)[1], ((char *)&ctype)[2], ((char *)&ctype)[3], (long) ctype);

  564.     printf("stype= %c%c%c%c\n", *((char *)&type), ((char *)&type)[1], ((char *)&type)[2], ((char *)&type)[3]);

  565. #endif

  566. #ifdef DEBUG

  567. /* XXX: yeah this is ugly... */

  568.     if(ctype == MKTAG('m', 'h', 'l', 'r')) { /* MOV */

  569.         if(type == MKTAG('v', 'i', 'd', 'e'))

  570.             puts("hdlr: vide");

  571.         else if(type == MKTAG('s', 'o', 'u', 'n'))

  572.             puts("hdlr: soun");

  573.     } else if(ctype == 0) { /* MP4 */

  574.         if(type == MKTAG('v', 'i', 'd', 'e'))

  575.             puts("hdlr: vide");

  576.         else if(type == MKTAG('s', 'o', 'u', 'n'))

  577.             puts("hdlr: soun");

  578.         else if(type == MKTAG('o', 'd', 's', 'm'))

  579.             puts("hdlr: odsm");

  580.         else if(type == MKTAG('s', 'd', 's', 'm'))

  581.             puts("hdlr: sdsm");

  582.     } else puts("hdlr: meta");

  583. #endif

  584. 

  585.     if(ctype == MKTAG('m', 'h', 'l', 'r')) { /* MOV */

  586.         /* helps parsing the string hereafter... */

  587.         c->mp4 = 0;

  588.         if(type == MKTAG('v', 'i', 'd', 'e'))

  589.             st->codec.codec_type = CODEC_TYPE_VIDEO;

  590.         else if(type == MKTAG('s', 'o', 'u', 'n'))

  591.             st->codec.codec_type = CODEC_TYPE_AUDIO;

  592.     } else if(ctype == 0) { /* MP4 */

  593.         /* helps parsing the string hereafter... */

  594.         c->mp4 = 1;

  595.         if(type == MKTAG('v', 'i', 'd', 'e'))

  596.             st->codec.codec_type = CODEC_TYPE_VIDEO;

  597.         else if(type == MKTAG('s', 'o', 'u', 'n'))

  598.             st->codec.codec_type = CODEC_TYPE_AUDIO;

  599.     }

  600.     get_be32(pb); /* component  manufacture */

  601.     get_be32(pb); /* component flags */

  602.     get_be32(pb); /* component flags mask */

  603. 

  604.     if(atom_size <= 24)

  605.         return 0; /* nothing left to read */

  606.     /* XXX: MP4 uses a C string, not a pascal one */

  607.     /* component name */

  608. 

  609.     if(c->mp4) {

  610.         /* .mp4: C string */

  611.         while(get_byte(pb) && (++len < (atom_size - 24)));

  612.     } else {

  613.         /* .mov: PASCAL string */

  614.         len = get_byte(pb);

  615. 	buf = (uint8_t*) av_malloc(len+1);

  616. 	if (buf) {

  617. 	    get_buffer(pb, buf, len);

  618. #ifdef DEBUG

  619. 	    buf[len] = '\0';

  620. 	    printf("**buf='%s'\n", buf);

  621. #endif

  622. 	    av_free(buf);

  623. 	} else

  624. 	    url_fskip(pb, len);

  625.     }

  626. #if 0

  627.     len = get_byte(pb);

  628.     /* XXX: use a better heuristic */

  629.     if(len < 32) {

  630.         /* assume that it is a Pascal like string */

  631. 	buf = av_malloc(len+1);

  632. 	if (buf) {

  633. 	    get_buffer(pb, buf, len);

  634. 	    buf[len] = '\0';

  635. #ifdef DEBUG

  636. 	    printf("**buf='%s'\n", buf);

  637. #endif

  638. 	    av_free(buf);

  639. 	} else

  640.             url_fskip(pb, len)l

  641.     } else {

  642.         /* MP4 string */

  643.         for(;;) {

  644.             if (len == 0)

  645.                 break;

  646.             len = get_byte(pb);

  647.         }

  648.     }

  649. #endif

  650. 

  651.     return 0;

  652. }

  653. 

  654. static int mp4_read_descr_len(ByteIOContext *pb)

  655. {

  656.     int len = 0;

  657.     int count = 0;

  658.     for(;;) {

  659.         int c = get_byte(pb);

  660. 	len = (len << 7) | (c & 0x7f);

  661.         if ((c & 0x80) == 0)

  662.             break;

  663.         if (++count == 4)

  664.             break;

  665.     }

  666.     return len;

  667. }

  668. 

  669. static int mp4_read_descr(ByteIOContext *pb, int *tag)

  670. {

  671.     int len;

  672.     *tag = get_byte(pb);

  673.     len = mp4_read_descr_len(pb);

  674. #ifdef DEBUG

  675.     printf("MPEG4 description: tag=0x%02x len=%d\n", *tag, len);

  676. #endif

  677.     return len;

  678. }

  679. 

  680. static inline unsigned int get_be24(ByteIOContext *s)

  681. {

  682.     unsigned int val;

  683.     val = get_byte(s) << 16;

  684.     val |= get_byte(s) << 8;

  685.     val |= get_byte(s);

  686.     return val;

  687. }

  688. 

  689. static int parse_esds(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  690. {

  691. 

  692.     int64_t start_pos = url_ftell(pb);

  693.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  694.     MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;

  695.     int tag, len;

  696. 

  697.     print_atom("esds", atom_type, atom_offset, atom_size);

  698. 

  699.     /* Well, broken but suffisant for some MP4 streams */

  700.     get_be32(pb); /* version + flags */

  701.     len = mp4_read_descr(pb, &tag);

  702.     if (tag == MP4ESDescrTag) {

  703. 	get_be16(pb); /* ID */

  704. 	get_byte(pb); /* priority */

  705.     } else

  706. 	get_be16(pb); /* ID */

  707. 

  708.     len = mp4_read_descr(pb, &tag);

  709.     if (tag == MP4DecConfigDescrTag) {

  710. 	sc->esds.object_type_id = get_byte(pb);

  711. 	sc->esds.stream_type = get_be24(pb);

  712. 	sc->esds.max_bitrate = get_be32(pb);

  713. 	sc->esds.avg_bitrate = get_be32(pb);

  714. 

  715. 	len = mp4_read_descr(pb, &tag);

  716. 	printf("LEN %d  TAG %d  m:%d a:%d\n", len, tag, sc->esds.max_bitrate, sc->esds.avg_bitrate);

  717. 	if (tag == MP4DecSpecificDescrTag) {

  718. #ifdef DEBUG

  719. 	    printf("Specific MPEG4 header len=%d\n", len);

  720. #endif

  721. 	    sc->header_data = (uint8_t*) av_mallocz(len);

  722. 	    if (sc->header_data) {

  723. 		get_buffer(pb, sc->header_data, len);

  724. 		sc->header_len = len;

  725. 	    }

  726. 	}

  727.     }

  728.     /* in any case, skip garbage */

  729.     url_fskip(pb, (atom_size - 8) - ((url_ftell(pb) - start_pos)));

  730.     return 0;

  731. }

  732. 

  733. static int parse_stsd(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  734. {

  735.     int entries, size, frames_per_sample;

  736.     uint32_t format;

  737.     AVStream *st;

  738.     MOVStreamContext *sc;

  739. 

  740.     print_atom("stsd", atom_type, atom_offset, atom_size);

  741. 

  742.     st = c->fc->streams[c->fc->nb_streams-1];

  743.     sc = (MOVStreamContext *)st->priv_data;

  744. 

  745.     get_byte(pb); /* version */

  746.     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */

  747. 

  748.     entries = get_be32(pb);

  749. 

  750.     while(entries--) {

  751.         enum CodecID id;

  752. 	size = get_be32(pb); /* size */

  753.         format = get_le32(pb); /* data format */

  754. 

  755.         get_be32(pb); /* reserved */

  756.         get_be16(pb); /* reserved */

  757.         get_be16(pb); /* index */

  758. 

  759.         /* for MPEG4: set codec type by looking for it */

  760.         id = codec_get_id(mov_video_tags, format);

  761.         if (id >= 0) {

  762.             AVCodec *codec;

  763. 	    codec = avcodec_find_decoder(id);

  764.             if (codec)

  765. 		st->codec.codec_type = codec->type;

  766.         }

  767. #ifdef DEBUG

  768.         printf("size=%d 4CC= %c%c%c%c codec_type=%d\n",

  769.                size,

  770.                (format >> 0) & 0xff,

  771.                (format >> 8) & 0xff,

  772.                (format >> 16) & 0xff,

  773.                (format >> 24) & 0xff,

  774.                st->codec.codec_type);

  775. #endif

  776. 	st->codec.codec_tag = format;

  777.         if(st->codec.codec_type==CODEC_TYPE_VIDEO) {

  778.             st->codec.codec_id = codec_get_id(mov_video_tags, format);

  779.             get_be16(pb); /* version */

  780.             get_be16(pb); /* revision level */

  781.             get_be32(pb); /* vendor */

  782.             get_be32(pb); /* temporal quality */

  783.             get_be32(pb); /* spacial quality */

  784.             st->codec.width = get_be16(pb); /* width */

  785.             st->codec.height = get_be16(pb); /* height */

  786. #if 1

  787.             if (st->codec.codec_id == CODEC_ID_MPEG4) {

  788.                 /* in some MPEG4 the width/height are not correct, so

  789.                    we ignore this info */

  790.                 st->codec.width = 0;

  791.                 st->codec.height = 0;

  792.             }

  793. #endif

  794.             get_be32(pb); /* horiz resolution */

  795.             get_be32(pb); /* vert resolution */

  796.             get_be32(pb); /* data size, always 0 */

  797.             frames_per_sample = get_be16(pb); /* frames per samples */

  798. #ifdef DEBUG

  799. 	    printf("frames/samples = %d\n", frames_per_sample);

  800. #endif

  801. 	    get_buffer(pb, (uint8_t *)st->codec.codec_name, 32); /* codec name */

  802. 

  803. 	    st->codec.bits_per_sample = get_be16(pb); /* depth */

  804.             st->codec.color_table_id = get_be16(pb); /* colortable id */

  805. 

  806.             st->codec.frame_rate      = 25;

  807.             st->codec.frame_rate_base = 1;

  808. 

  809.             size -= (16+8*4+2+32+2*2);

  810. #if 0

  811. 	    while (size >= 8) {

  812.                 int atom_size, atom_type;

  813.                 int64_t start_pos;

  814. 

  815.                 atom_size = get_be32(pb);

  816.                 atom_type = get_le32(pb);

  817. 		size -= 8;

  818.                 printf("NEWSIZE %d\n", size);

  819. #ifdef DEBUG

  820.                 printf("VIDEO: atom_type=%c%c%c%c atom_size=%d size_left=%d\n",

  821.                        (atom_type >> 0) & 0xff,

  822.                        (atom_type >> 8) & 0xff,

  823.                        (atom_type >> 16) & 0xff,

  824.                        (atom_type >> 24) & 0xff,

  825.                        atom_size, size);

  826. #endif

  827.                 start_pos = url_ftell(pb);

  828. 

  829.                 switch(atom_type) {

  830.                 case MKTAG('e', 's', 'd', 's'):

  831.                     {

  832.                         int tag, len;

  833.                         /* Well, broken but suffisant for some MP4 streams */

  834.                         get_be32(pb); /* version + flags */

  835.                         len = mp4_read_descr(pb, &tag);

  836.                         if (tag == 0x03) {

  837.                             /* MP4ESDescrTag */

  838.                             get_be16(pb); /* ID */

  839.                             get_byte(pb); /* priority */

  840.                             len = mp4_read_descr(pb, &tag);

  841.                             if (tag != 0x04)

  842.                                 goto fail;

  843.                             /* MP4DecConfigDescrTag */

  844.                             get_byte(pb); /* objectTypeId */

  845.                             get_be32(pb); /* streamType + buffer size */

  846. 			    get_be32(pb); /* max bit rate */

  847.                             get_be32(pb); /* avg bit rate */

  848.                             len = mp4_read_descr(pb, &tag);

  849.                             if (tag != 0x05)

  850.                                 goto fail;

  851.                             /* MP4DecSpecificDescrTag */

  852. #ifdef DEBUG

  853.                             printf("Specific MPEG4 header len=%d\n", len);

  854. #endif

  855.                             sc->header_data = av_mallocz(len);

  856.                             if (sc->header_data) {

  857.                                 get_buffer(pb, sc->header_data, len);

  858. 				sc->header_len = len;

  859.                             }

  860.                         }

  861.                         /* in any case, skip garbage */

  862.                     }

  863.                     break;

  864.                 default:

  865.                     break;

  866.                 }

  867. 	    fail:

  868.                 printf("ATOMENEWSIZE %d   %d\n", atom_size, url_ftell(pb) - start_pos);

  869. 		if (atom_size > 8) {

  870. 		    url_fskip(pb, (atom_size - 8) -

  871. 			      ((url_ftell(pb) - start_pos)));

  872. 		    size -= atom_size - 8;

  873. 		    printf("NEWSIZE %d\n", size);

  874. 		}

  875. 	    }

  876.             if (size > 0) {

  877.                 /* unknown extension */

  878.                 url_fskip(pb, size);

  879.             }

  880. #else

  881. 	    parse_default(c, pb, 0L, 0LL, size);

  882. #endif

  883. 	} else {

  884.             get_be16(pb); /* version */

  885.             get_be16(pb); /* revision level */

  886.             get_be32(pb); /* vendor */

  887. 

  888.             st->codec.channels = get_be16(pb);/* channel count */

  889. 	    st->codec.bits_per_sample = get_be16(pb); /* sample size */

  890. 

  891. 	    st->codec.codec_id = codec_get_id(mov_audio_tags, format);

  892.             /* handle specific s8 codec */

  893.             get_be16(pb); /* compression id = 0*/

  894.             get_be16(pb); /* packet size = 0 */

  895. 

  896.             st->codec.sample_rate = ((get_be32(pb) >> 16));

  897. 	    printf("CODECID %d  %d  %.4s\n", st->codec.codec_id, CODEC_ID_PCM_S16BE, (char*)&format);

  898. 

  899. 	    switch (st->codec.codec_id) {

  900. 	    case CODEC_ID_PCM_S16BE:

  901. 		if (st->codec.bits_per_sample == 8)

  902. 		    st->codec.codec_id = CODEC_ID_PCM_S8;

  903.                 /* fall */

  904. 	    case CODEC_ID_PCM_U8:

  905. 		st->codec.bit_rate = st->codec.sample_rate * 8;

  906. 		break;

  907. 	    default:

  908.                 ;

  909. 	    }

  910. 	    get_be32(pb); /* samples per packet */

  911. 	    get_be32(pb); /* bytes per packet */

  912.             get_be32(pb); /* bytes per frame */

  913.             get_be32(pb); /* bytes per sample */

  914. 

  915. 	    //if (size > 16) url_fskip(pb, size-(16+20));

  916. #if 1

  917. 	    if (size >= 44 + 8) {

  918.                 int fcc;

  919. 		st->codec.extradata_size = get_be32(pb) - 8;

  920. 		fcc = get_le32(pb); // evaw

  921. 		//printf("%x  %.4s  %d\n", fcc, (char*)&fcc, st->codec.extradata_size);

  922. 		st->codec.extradata = av_mallocz(st->codec.extradata_size);

  923.                 if (st->codec.extradata)

  924. 		    get_buffer(pb, st->codec.extradata, st->codec.extradata_size); // FIXME url_fskip

  925. 		url_fskip(pb, size-(16 + 20 + 16 + 8 + st->codec.extradata_size));

  926. 	    }

  927.             else

  928. 		url_fskip(pb, size-(16 + 20 + 16));

  929. #else

  930. 	    parse_default(c, pb, 0L, 0LL, size - (16 + 20 + 16 + 8));

  931. #endif

  932.         }

  933.     }

  934. /*

  935.     if(len) {

  936.     buf = av_malloc(len+1);

  937.         get_buffer(pb, buf, len);

  938.         buf[len] = '\0';

  939.         puts(buf);

  940.         av_free(buf);

  941.     }

  942. */

  943.     return 0;

  944. }

  945. 

  946. static int parse_stco(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  947. {

  948.     int entries, i;

  949.     AVStream *st;

  950.     MOVStreamContext *sc;

  951. 

  952.     print_atom("stco", atom_type, atom_offset, atom_size);

  953. 

  954.     st = c->fc->streams[c->fc->nb_streams-1];

  955.     sc = (MOVStreamContext *)st->priv_data;

  956. 

  957.     get_byte(pb); /* version */

  958.     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */

  959. 

  960.     entries = get_be32(pb);

  961.     sc->chunk_count = entries;

  962.     sc->chunk_offsets = (int64_t*) av_malloc(entries * sizeof(int64_t));

  963.     if (!sc->chunk_offsets)

  964.         return -1;

  965.     if(atom_type == MKTAG('s', 't', 'c', 'o')) {

  966.         for(i=0; i<entries; i++) {

  967.             sc->chunk_offsets[i] = get_be32(pb);

  968.         }

  969.     } else if(atom_type == MKTAG('c', 'o', '6', '4')) {

  970.         for(i=0; i<entries; i++) {

  971.             sc->chunk_offsets[i] = get_be64(pb);

  972.         }

  973.     } else

  974.         return -1;

  975. #ifdef DEBUG

  976. /*

  977.     for(i=0; i<entries; i++) {

  978.         printf("chunk offset=0x%Lx\n", sc->chunk_offsets[i]);

  979.     }

  980. */

  981. #endif

  982.     return 0;

  983. }

  984. 

  985. static int parse_stsc(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  986. {

  987.     int entries, i;

  988.     AVStream *st;

  989.     MOVStreamContext *sc;

  990. 

  991.     print_atom("stsc", atom_type, atom_offset, atom_size);

  992. 

  993.     st = c->fc->streams[c->fc->nb_streams-1];

  994.     sc = (MOVStreamContext *)st->priv_data;

  995. 

  996.     get_byte(pb); /* version */

  997.     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */

  998. 

  999.     entries = get_be32(pb);

  1000. #ifdef DEBUG

  1001. printf("track[%i].stsc.entries = %i\n", c->fc->nb_streams-1, entries);

  1002. #endif

  1003.     sc->sample_to_chunk_sz = entries;

  1004.     sc->sample_to_chunk = (MOV_sample_to_chunk_tbl*) av_malloc(entries * sizeof(MOV_sample_to_chunk_tbl));

  1005.     if (!sc->sample_to_chunk)

  1006.         return -1;

  1007.     for(i=0; i<entries; i++) {

  1008.         sc->sample_to_chunk[i].first = get_be32(pb);

  1009.         sc->sample_to_chunk[i].count = get_be32(pb);

  1010.         sc->sample_to_chunk[i].id = get_be32(pb);

  1011. #ifdef DEBUG

  1012. /*        printf("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); */

  1013. #endif

  1014.     }

  1015.     return 0;

  1016. }

  1017. 

  1018. static int parse_stsz(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  1019. {

  1020.     int entries, i;

  1021.     AVStream *st;

  1022.     MOVStreamContext *sc;

  1023. 

  1024.     print_atom("stsz", atom_type, atom_offset, atom_size);

  1025. 

  1026.     st = c->fc->streams[c->fc->nb_streams-1];

  1027.     sc = (MOVStreamContext *)st->priv_data;

  1028. 

  1029.     get_byte(pb); /* version */

  1030.     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */

  1031. 

  1032.     sc->sample_size = get_be32(pb);

  1033.     entries = get_be32(pb);

  1034.     sc->sample_count = entries;

  1035. #ifdef DEBUG

  1036.     printf("sample_size = %ld sample_count = %ld\n", sc->sample_size, sc->sample_count);

  1037. #endif

  1038.     if(sc->sample_size)

  1039.         return 0; /* there isn't any table following */

  1040.     sc->sample_sizes = (long*) av_malloc(entries * sizeof(long));

  1041.     if (!sc->sample_sizes)

  1042.         return -1;

  1043.     for(i=0; i<entries; i++) {

  1044.         sc->sample_sizes[i] = get_be32(pb);

  1045. #ifdef DEBUG

  1046. /*        printf("sample_sizes[]=%ld\n", sc->sample_sizes[i]); */

  1047. #endif

  1048.     }

  1049.     return 0;

  1050. }

  1051. 

  1052. static int parse_stts(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  1053. {

  1054.     int entries, i;

  1055.     AVStream *st;

  1056.     MOVStreamContext *sc;

  1057. 

  1058.     print_atom("stts", atom_type, atom_offset, atom_size);

  1059. 

  1060.     st = c->fc->streams[c->fc->nb_streams-1];

  1061.     sc = (MOVStreamContext *)st->priv_data;

  1062. 

  1063.     get_byte(pb); /* version */

  1064.     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */

  1065.     entries = get_be32(pb);

  1066. #ifdef DEBUG

  1067. printf("track[%i].stts.entries = %i\n", c->fc->nb_streams-1, entries);

  1068. #endif

  1069.     for(i=0; i<entries; i++) {

  1070.         int sample_duration;

  1071. 

  1072.         get_be32(pb);

  1073.         sample_duration = get_be32(pb);

  1074. 

  1075.         if (!i && st->codec.codec_type==CODEC_TYPE_VIDEO) {

  1076.             st->codec.frame_rate_base = sample_duration ? sample_duration : 1;

  1077.             st->codec.frame_rate = c->streams[c->total_streams]->time_scale;

  1078. #ifdef DEBUG

  1079.             printf("VIDEO FRAME RATE= %i (sd= %i)\n", st->codec.frame_rate, sample_duration);

  1080. #endif

  1081.         }

  1082.     }

  1083.     return 0;

  1084. }

  1085. 

  1086. #ifdef CONFIG_ZLIB

  1087. static int null_read_packet(void *opaque, uint8_t *buf, int buf_size)

  1088. {

  1089.     return -1;

  1090. }

  1091. 

  1092. static int parse_cmov(MOVContext *c, ByteIOContext *pb, uint32_t atom_type, int64_t atom_offset, int64_t atom_size)

  1093. {

  1094.     ByteIOContext ctx;

  1095.     uint8_t *cmov_data;

  1096.     uint8_t *moov_data; /* uncompressed data */

  1097.     long cmov_len, moov_len;

  1098.     int ret;

  1099. 

  1100.     print_atom("cmov", atom_type, atom_offset, atom_size);

  1101. 

  1102.     get_be32(pb); /* dcom atom */

  1103.     if (get_le32(pb) != MKTAG( 'd', 'c', 'o', 'm' ))

  1104.         return -1;

  1105.     if (get_le32(pb) != MKTAG( 'z', 'l', 'i', 'b' )) {

  1106.         puts("unknown compression for cmov atom !");

  1107.         return -1;

  1108.     }

  1109.     get_be32(pb); /* cmvd atom */

  1110.     if (get_le32(pb) != MKTAG( 'c', 'm', 'v', 'd' ))

  1111.         return -1;

  1112.     moov_len = get_be32(pb); /* uncompressed size */

  1113.     cmov_len = atom_size - 6 * 4;

  1114. 

  1115.     cmov_data = (uint8_t *) av_malloc(cmov_len);

  1116.     if (!cmov_data)

  1117.         return -1;

  1118.     moov_data = (uint8_t *) av_malloc(moov_len);

  1119.     if (!moov_data) {

  1120.         av_free(cmov_data);

  1121.         return -1;

  1122.     }

  1123.     get_buffer(pb, cmov_data, cmov_len);

  1124.     if(uncompress (moov_data, (uLongf *) &moov_len, (const Bytef *)cmov_data, cmov_len) != Z_OK)

  1125.         return -1;

  1126.     if(init_put_byte(&ctx, moov_data, moov_len, 0, NULL, null_read_packet, NULL, NULL) != 0)

  1127.         return -1;

  1128.     ctx.buf_end = ctx.buffer + moov_len;

  1129.     ret = parse_default(c, &ctx, MKTAG( 'm', 'o', 'o', 'v' ), 0, moov_len);

  1130.     av_free(moov_data);

  1131.     av_free(cmov_data);

  1132.     return ret;

  1133. }

  1134. #endif

  1135. 

  1136. static const MOVParseTableEntry mov_default_parse_table[] = {

  1137. /* mp4 atoms */

  1138. { MKTAG( 'm', 'p', '4', 'a' ), parse_default },

  1139. { MKTAG( 'c', 'o', '6', '4' ), parse_stco },

  1140. { MKTAG( 's', 't', 'c', 'o' ), parse_stco },

  1141. { MKTAG( 'c', 'r', 'h', 'd' ), parse_default },

  1142. { MKTAG( 'c', 't', 't', 's' ), parse_leaf },

  1143. { MKTAG( 'c', 'p', 'r', 't' ), parse_default },

  1144. { MKTAG( 'u', 'r', 'l', ' ' ), parse_leaf },

  1145. { MKTAG( 'u', 'r', 'n', ' ' ), parse_leaf },

  1146. { MKTAG( 'd', 'i', 'n', 'f' ), parse_default },

  1147. { MKTAG( 'd', 'r', 'e', 'f' ), parse_leaf },

  1148. { MKTAG( 's', 't', 'd', 'p' ), parse_default },

  1149. { MKTAG( 'e', 'd', 't', 's' ), parse_default },

  1150. { MKTAG( 'e', 'l', 's', 't' ), parse_leaf },

  1151. { MKTAG( 'u', 'u', 'i', 'd' ), parse_default },

  1152. { MKTAG( 'f', 'r', 'e', 'e' ), parse_leaf },

  1153. { MKTAG( 'h', 'd', 'l', 'r' ), parse_hdlr },

  1154. { MKTAG( 'h', 'm', 'h', 'd' ), parse_leaf },

  1155. { MKTAG( 'h', 'i', 'n', 't' ), parse_leaf },

  1156. { MKTAG( 'n', 'm', 'h', 'd' ), parse_leaf },

  1157. { MKTAG( 'm', 'p', '4', 's' ), parse_default },

  1158. { MKTAG( 'm', 'd', 'i', 'a' ), parse_default },

  1159. { MKTAG( 'm', 'd', 'a', 't' ), parse_mdat },

  1160. { MKTAG( 'm', 'd', 'h', 'd' ), parse_mdhd },

  1161. { MKTAG( 'm', 'i', 'n', 'f' ), parse_default },

  1162. { MKTAG( 'm', 'o', 'o', 'v' ), parse_moov },

  1163. { MKTAG( 'm', 'v', 'h', 'd' ), parse_mvhd },

  1164. { MKTAG( 'i', 'o', 'd', 's' ), parse_leaf },

  1165. { MKTAG( 'o', 'd', 'h', 'd' ), parse_default },

  1166. { MKTAG( 'm', 'p', 'o', 'd' ), parse_leaf },

  1167. { MKTAG( 's', 't', 's', 'd' ), parse_stsd },

  1168. { MKTAG( 's', 't', 's', 'z' ), parse_stsz },

  1169. { MKTAG( 's', 't', 'b', 'l' ), parse_default },

  1170. { MKTAG( 's', 't', 's', 'c' ), parse_stsc },

  1171. { MKTAG( 's', 'd', 'h', 'd' ), parse_default },

  1172. { MKTAG( 's', 't', 's', 'h' ), parse_default },

  1173. { MKTAG( 's', 'k', 'i', 'p' ), parse_default },

  1174. { MKTAG( 's', 'm', 'h', 'd' ), parse_leaf },

  1175. { MKTAG( 'd', 'p', 'n', 'd' ), parse_leaf },

  1176. { MKTAG( 's', 't', 's', 's' ), parse_leaf },

  1177. { MKTAG( 's', 't', 't', 's' ), parse_stts },

  1178. { MKTAG( 't', 'r', 'a', 'k' ), parse_trak },

  1179. { MKTAG( 't', 'k', 'h', 'd' ), parse_tkhd },

  1180. { MKTAG( 't', 'r', 'e', 'f' ), parse_default }, /* not really */

  1181. { MKTAG( 'u', 'd', 't', 'a' ), parse_leaf },

  1182. { MKTAG( 'v', 'm', 'h', 'd' ), parse_leaf },

  1183. { MKTAG( 'm', 'p', '4', 'v' ), parse_default },

  1184. /* extra mp4 */

  1185. { MKTAG( 'M', 'D', 'E', 'S' ), parse_leaf },

  1186. /* QT atoms */

  1187. { MKTAG( 'c', 'h', 'a', 'p' ), parse_leaf },

  1188. { MKTAG( 'c', 'l', 'i', 'p' ), parse_default },

  1189. { MKTAG( 'c', 'r', 'g', 'n' ), parse_leaf },

  1190. { MKTAG( 'k', 'm', 'a', 't' ), parse_leaf },

  1191. { MKTAG( 'm', 'a', 't', 't' ), parse_default },

  1192. { MKTAG( 'r', 'd', 'r', 'f' ), parse_leaf },

  1193. { MKTAG( 'r', 'm', 'd', 'a' ), parse_default },

  1194. { MKTAG( 'r', 'm', 'd', 'r' ), parse_leaf },

  1195. //{ MKTAG( 'r', 'm', 'q', 'u' ), parse_leaf },

  1196. { MKTAG( 'r', 'm', 'r', 'a' ), parse_default },

  1197. { MKTAG( 's', 'c', 'p', 't' ), parse_leaf },

  1198. { MKTAG( 's', 'y', 'n', 'c' ), parse_leaf },

  1199. { MKTAG( 's', 's', 'r', 'c' ), parse_leaf },

  1200. { MKTAG( 't', 'c', 'm', 'd' ), parse_leaf },

  1201. { MKTAG( 'w', 'i', 'd', 'e' ), parse_wide }, /* place holder */

  1202. { MKTAG( 'c', 't', 'a', 'b' ), parse_ctab },

  1203. { MKTAG( 'e', 's', 'd', 's' ), parse_esds },

  1204. #ifdef CONFIG_ZLIB

  1205. { MKTAG( 'c', 'm', 'o', 'v' ), parse_cmov },

  1206. #else

  1207. { MKTAG( 'c', 'm', 'o', 'v' ), parse_leaf },

  1208. #endif

  1209. { 0L, parse_leaf }

  1210. };

  1211. 

  1212. static void mov_free_stream_context(MOVStreamContext *sc)

  1213. {

  1214.     if(sc) {

  1215.         av_free(sc->chunk_offsets);

  1216.         av_free(sc->sample_to_chunk);

  1217.         av_free(sc->sample_sizes);

  1218.         av_free(sc->header_data);

  1219.         av_free(sc);

  1220.     }

  1221. }

  1222. 

  1223. static inline uint32_t to_tag(uint8_t *buf)

  1224. {

  1225.     return buf[0] | (buf[1] << 8) | (buf[2] << 16) | (buf[3] << 24);

  1226. }

  1227. 

  1228. static inline uint32_t to_be32(uint8_t *buf)

  1229. {

  1230.     return (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];

  1231. }

  1232. 

  1233. /* XXX: is it sufficient ? */

  1234. static int mov_probe(AVProbeData *p)

  1235. {

  1236.     unsigned int offset;

  1237.     uint32_t tag;

  1238. 

  1239.     /* check file header */

  1240.     if (p->buf_size <= 12)

  1241.         return 0;

  1242.     offset = 0;

  1243.     for(;;) {

  1244.         /* ignore invalid offset */

  1245.         if ((offset + 8) > (unsigned int)p->buf_size)

  1246.             return 0;

  1247.         tag = to_tag(p->buf + offset + 4);

  1248.         switch(tag) {

  1249.         case MKTAG( 'm', 'o', 'o', 'v' ):

  1250.         case MKTAG( 'w', 'i', 'd', 'e' ):

  1251.         case MKTAG( 'f', 'r', 'e', 'e' ):

  1252.         case MKTAG( 'm', 'd', 'a', 't' ):

  1253.         case MKTAG( 'p', 'n', 'o', 't' ): /* detect movs with preview pics like ew.mov and april.mov */

  1254.             return AVPROBE_SCORE_MAX;

  1255.         case MKTAG( 'f', 't', 'y', 'p' ):

  1256.         case MKTAG( 's', 'k', 'i', 'p' ):            

  1257.             offset = to_be32(p->buf) + offset;

  1258.             break;

  1259.         default:

  1260.             /* unrecognized tag */

  1261.             return 0;

  1262.         }

  1263.     }

  1264.     return 0;

  1265. }

  1266. 

  1267. static int mov_read_header(AVFormatContext *s, AVFormatParameters *ap)

  1268. {

  1269.     MOVContext *mov = (MOVContext *) s->priv_data;

  1270.     ByteIOContext *pb = &s->pb;

  1271.     int i, j, nb, err;

  1272.     int64_t size;

  1273. 

  1274.     mov->fc = s;

  1275.     mov->parse_table = mov_default_parse_table;

  1276. #if 0

  1277.     /* XXX: I think we should auto detect */

  1278.     if(s->iformat->name[1] == 'p')

  1279.         mov->mp4 = 1;

  1280. #endif

  1281.     if(!url_is_streamed(pb)) /* .mov and .mp4 aren't streamable anyway (only progressive download if moov is before mdat) */

  1282.         size = url_filesize(url_fileno(pb));

  1283.     else

  1284.         size = 0x7FFFFFFFFFFFFFFF;

  1285. 

  1286. #ifdef DEBUG

  1287.     printf("filesz=%Ld\n", size);

  1288. #endif

  1289. 

  1290.     /* check MOV header */

  1291.     err = parse_default(mov, pb, 0L, 0LL, size);

  1292.     if(err<0 || (!mov->found_moov || !mov->found_mdat)) {

  1293.         puts("header not found !!!");

  1294.         exit(1);

  1295.     }

  1296. #ifdef DEBUG

  1297.     printf("on_parse_exit_offset=%d\n", (int) url_ftell(pb));

  1298. #endif

  1299.     /* some cleanup : make sure we are on the mdat atom */

  1300.     if(!url_is_streamed(pb) && (url_ftell(pb) != mov->mdat_offset))

  1301.         url_fseek(pb, mov->mdat_offset, SEEK_SET);

  1302. 

  1303.     mov->next_chunk_offset = mov->mdat_offset; /* initialise reading */

  1304. 

  1305. #ifdef DEBUG

  1306.     printf("mdat_reset_offset=%d\n", (int) url_ftell(pb));

  1307. #endif

  1308. 

  1309. #ifdef DEBUG

  1310.     printf("streams= %d\n", s->nb_streams);

  1311. #endif

  1312.     mov->total_streams = nb = s->nb_streams;

  1313.     

  1314. #if 1

  1315.     for(i=0; i<s->nb_streams;) {

  1316.         if(s->streams[i]->codec.codec_type == CODEC_TYPE_MOV_OTHER) {/* not audio, not video, delete */

  1317.             av_free(s->streams[i]);

  1318.             for(j=i+1; j<s->nb_streams; j++)

  1319.                 s->streams[j-1] = s->streams[j];

  1320.             s->nb_streams--;

  1321.         } else

  1322.             i++;

  1323.     }

  1324.     for(i=0; i<s->nb_streams;i++) {

  1325.         MOVStreamContext *sc;

  1326.         sc = (MOVStreamContext *)s->streams[i]->priv_data;

  1327.         sc->ffindex = i;

  1328.         sc->is_ff_stream = 1;

  1329.     }

  1330. #endif

  1331. #ifdef DEBUG

  1332.     printf("real streams= %d\n", s->nb_streams);

  1333. #endif

  1334.     return 0;

  1335. }

  1336. 

  1337. /* Yes, this is ugly... I didn't write the specs of QT :p */

  1338. /* XXX:remove useless commented code sometime */

  1339. static int mov_read_packet(AVFormatContext *s, AVPacket *pkt)

  1340. {

  1341.     MOVContext *mov = (MOVContext *) s->priv_data;

  1342.     MOVStreamContext *sc;

  1343.     int64_t offset = 0x0FFFFFFFFFFFFFFF;

  1344.     int i;

  1345.     int size;

  1346.     size = 0x0FFFFFFF;

  1347. 

  1348. #ifdef MOV_SPLIT_CHUNKS

  1349.     if (mov->partial) {

  1350. 

  1351.         int idx;

  1352. 

  1353. 	sc = mov->partial;

  1354. 	idx = sc->sample_to_chunk_index;

  1355. 

  1356.         if (idx < 0) return 0;

  1357. 	size = sc->sample_sizes[sc->current_sample];

  1358. 

  1359.         sc->current_sample++;

  1360.         sc->left_in_chunk--;

  1361. 

  1362.         if (sc->left_in_chunk <= 0)

  1363.             mov->partial = 0;

  1364.         offset = mov->next_chunk_offset;

  1365.         /* extract the sample */

  1366. 

  1367.         goto readchunk;

  1368.     }

  1369. #endif

  1370. 

  1371. again:

  1372.     sc = 0;

  1373.     for(i=0; i<mov->total_streams; i++) {

  1374. 	MOVStreamContext *msc = mov->streams[i];

  1375. 	//printf("MOCHUNK %ld  %d   %p  pos:%Ld\n", mov->streams[i]->next_chunk, mov->total_streams, mov->streams[i], url_ftell(&s->pb));

  1376.         if ((msc->next_chunk < msc->chunk_count) && msc->next_chunk >= 0

  1377. 	   && (msc->chunk_offsets[msc->next_chunk] < offset)) {

  1378. 	    sc = msc;

  1379. 	    offset = msc->chunk_offsets[msc->next_chunk];

  1380. 	    //printf("SELETED  %Ld  i:%d\n", offset, i);

  1381.         }

  1382.     }

  1383.     if (!sc || offset==0x0FFFFFFFFFFFFFFF)

  1384. 	return -1;

  1385. 

  1386.     sc->next_chunk++;

  1387. 

  1388.     if(mov->next_chunk_offset < offset) { /* some meta data */

  1389.         url_fskip(&s->pb, (offset - mov->next_chunk_offset));

  1390.         mov->next_chunk_offset = offset;

  1391.     }

  1392. 

  1393. //printf("chunk: [%i] %lli -> %lli\n", st_id, mov->next_chunk_offset, offset);

  1394.     if(!sc->is_ff_stream) {

  1395.         url_fskip(&s->pb, (offset - mov->next_chunk_offset));

  1396.         mov->next_chunk_offset = offset;

  1397.         offset = 0x0FFFFFFFFFFFFFFF;

  1398.         goto again;

  1399.     }

  1400. 

  1401.     /* now get the chunk size... */

  1402. 

  1403.     for(i=0; i<mov->total_streams; i++) {

  1404. 	MOVStreamContext *msc = mov->streams[i];

  1405. 	if ((msc->next_chunk < msc->chunk_count)

  1406. 	    && ((msc->chunk_offsets[msc->next_chunk] - offset) < size))

  1407. 	    size = msc->chunk_offsets[msc->next_chunk] - offset;

  1408.     }

  1409. #ifdef MOV_SPLIT_CHUNKS

  1410.     /* split chunks into samples */

  1411.     if (sc->sample_size == 0) {

  1412.         int idx = sc->sample_to_chunk_index;

  1413.         if ((idx + 1 < sc->sample_to_chunk_sz)

  1414. 	    && (sc->next_chunk >= sc->sample_to_chunk[idx + 1].first))

  1415.            idx++;

  1416.         sc->sample_to_chunk_index = idx;

  1417.         if (idx >= 0 && sc->sample_to_chunk[idx].count != 1) {

  1418. 	    mov->partial = sc;

  1419.             /* we'll have to get those samples before next chunk */

  1420.             sc->left_in_chunk = sc->sample_to_chunk[idx].count - 1;

  1421.             size = sc->sample_sizes[sc->current_sample];

  1422.         }

  1423. 

  1424.         sc->current_sample++;

  1425.     }

  1426. #endif

  1427. 

  1428. readchunk:

  1429. //printf("chunk: [%i] %lli -> %lli (%i)\n", st_id, offset, offset + size, size);

  1430.     if(size == 0x0FFFFFFF)

  1431.         size = mov->mdat_size + mov->mdat_offset - offset;

  1432.     if(size < 0)

  1433.         return -1;

  1434.     if(size == 0)

  1435.         return -1;

  1436.     url_fseek(&s->pb, offset, SEEK_SET);

  1437. 

  1438.     //printf("READCHUNK hlen: %d  %d off: %Ld   pos:%Ld\n", size, sc->header_len, offset, url_ftell(&s->pb));

  1439.     if (sc->header_len > 0) {

  1440.         av_new_packet(pkt, size + sc->header_len);

  1441.         memcpy(pkt->data, sc->header_data, sc->header_len);

  1442.         get_buffer(&s->pb, pkt->data + sc->header_len, size);

  1443.         /* free header */

  1444.         av_freep(&sc->header_data);

  1445.         sc->header_len = 0;

  1446.     } else {

  1447.         av_new_packet(pkt, size);

  1448.         get_buffer(&s->pb, pkt->data, pkt->size);

  1449.     }

  1450.     pkt->stream_index = sc->ffindex;

  1451. 

  1452. #ifdef DEBUG

  1453. /*

  1454.     printf("Packet (%d, %d, %ld) ", pkt->stream_index, st_id, pkt->size);

  1455.     for(i=0; i<8; i++)

  1456.         printf("%02x ", pkt->data[i]);

  1457.     for(i=0; i<8; i++)

  1458.         printf("%c ", (pkt->data[i]) & 0x7F);

  1459.     puts("");

  1460. */

  1461. #endif

  1462. 

  1463.     mov->next_chunk_offset = offset + size;

  1464. 

  1465.     return 0;

  1466. }

  1467. 

  1468. static int mov_read_close(AVFormatContext *s)

  1469. {

  1470.     int i;

  1471.     MOVContext *mov = (MOVContext *) s->priv_data;

  1472.     for(i=0; i<mov->total_streams; i++)

  1473.         mov_free_stream_context(mov->streams[i]);

  1474.     for(i=0; i<s->nb_streams; i++)

  1475. 	av_freep(&s->streams[i]);

  1476.     return 0;

  1477. }

  1478. 

  1479. static AVInputFormat mov_iformat = {

  1480.     "mov",

  1481.     "QuickTime/MPEG4 format",

  1482.     sizeof(MOVContext),

  1483.     mov_probe,

  1484.     mov_read_header,

  1485.     mov_read_packet,

  1486.     mov_read_close,

  1487. };

  1488. 

  1489. int mov_init(void)

  1490. {

  1491.     av_register_input_format(&mov_iformat);

  1492.     return 0;

  1493. }