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.
11649 Commits
32 Branches
305MB
Tree: b17270878e
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'b17270878e'
${ noResults }
FFmpeg/libavformat/mov.c

1651 lines
55KB
Raw Blame History 

  1. /*

  2.  * MOV demuxer

  3.  * Copyright (c) 2001 Fabrice Bellard.

  4.  *

  5.  * This file is part of FFmpeg.

  6.  *

  7.  * FFmpeg is free software; you can redistribute it and/or

  8.  * modify it under the terms of the GNU Lesser General Public

  9.  * License as published by the Free Software Foundation; either

  10.  * version 2.1 of the License, or (at your option) any later version.

  11.  *

  12.  * FFmpeg is distributed in the hope that it will be useful,

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

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

  15.  * Lesser General Public License for more details.

  16.  *

  17.  * You should have received a copy of the GNU Lesser General Public

  18.  * License along with FFmpeg; if not, write to the Free Software

  19.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

  20.  */

  21. 

  22. #include <limits.h>

  23. 

  24. //#define DEBUG

  25. 

  26. #include "avformat.h"

  27. #include "riff.h"

  28. #include "isom.h"

  29. #include "dv.h"

  30. 

  31. #ifdef CONFIG_ZLIB

  32. #include <zlib.h>

  33. #endif

  34. 

  35. /*

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

  37.  * Seek function by Gael Chardon gael.dev@4now.net

  38.  *

  39.  * Features and limitations:

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

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

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

  43.  *

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

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

  46.  *

  47.  * Reference documents:

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

  49.  * Apple:

  50.  *  http://developer.apple.com/documentation/QuickTime/QTFF/

  51.  *  http://developer.apple.com/documentation/QuickTime/QTFF/qtff.pdf

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

  53.  */

  54. 

  55. #include "qtpalette.h"

  56. 

  57. 

  58. #undef NDEBUG

  59. #include <assert.h>

  60. 

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

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

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

  64.  */

  65. 

  66. typedef struct {

  67.     int first;

  68.     int count;

  69.     int id;

  70. } MOV_stsc_t;

  71. 

  72. typedef struct {

  73.     uint32_t type;

  74.     int64_t offset;

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

  76. } MOV_atom_t;

  77. 

  78. typedef struct {

  79.     offset_t offset;

  80.     int64_t size;

  81. } MOV_mdat_t;

  82. 

  83. struct MOVParseTableEntry;

  84. 

  85. typedef struct MOVStreamContext {

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

  87.     int next_chunk;

  88.     unsigned int chunk_count;

  89.     int64_t *chunk_offsets;

  90.     unsigned int stts_count;

  91.     MOV_stts_t *stts_data;

  92.     unsigned int ctts_count;

  93.     MOV_stts_t *ctts_data;

  94.     unsigned int edit_count; /* number of 'edit' (elst atom) */

  95.     unsigned int sample_to_chunk_sz;

  96.     MOV_stsc_t *sample_to_chunk;

  97.     int sample_to_ctime_index;

  98.     int sample_to_ctime_sample;

  99.     unsigned int sample_size;

  100.     unsigned int sample_count;

  101.     int *sample_sizes;

  102.     unsigned int keyframe_count;

  103.     int *keyframes;

  104.     int time_scale;

  105.     int time_rate;

  106.     int current_sample;

  107.     unsigned int bytes_per_frame;

  108.     unsigned int samples_per_frame;

  109.     int dv_audio_container;

  110. } MOVStreamContext;

  111. 

  112. typedef struct MOVContext {

  113.     AVFormatContext *fc;

  114.     int time_scale;

  115.     int64_t duration; /* duration of the longest track */

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

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

  118.     int64_t mdat_offset;

  119.     int total_streams;

  120.     MOVStreamContext *streams[MAX_STREAMS];

  121. 

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

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

  124. 

  125.     AVPaletteControl palette_control;

  126.     MOV_mdat_t *mdat_list;

  127.     int mdat_count;

  128.     DVDemuxContext *dv_demux;

  129.     AVFormatContext *dv_fctx;

  130.     int isom; /* 1 if file is ISO Media (mp4/3gp) */

  131. } MOVContext;

  132. 

  133. 

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

  135. 

  136. /* those functions parse an atom */

  137. /* return code:

  138.  1: found what I wanted, exit

  139.  0: continue to parse next atom

  140.  -1: error occured, exit

  141.  */

  142. typedef int (*mov_parse_function)(MOVContext *ctx, ByteIOContext *pb, MOV_atom_t atom);

  143. 

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

  145. typedef struct MOVParseTableEntry {

  146.     uint32_t type;

  147.     mov_parse_function func;

  148. } MOVParseTableEntry;

  149. 

  150. static int mov_read_default(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  151. {

  152.     int64_t total_size = 0;

  153.     MOV_atom_t a;

  154.     int i;

  155.     int err = 0;

  156. 

  157.     a.offset = atom.offset;

  158. 

  159.     if (atom.size < 0)

  160.         atom.size = INT64_MAX;

  161.     while(((total_size + 8) < atom.size) && !url_feof(pb) && !err) {

  162.         a.size = atom.size;

  163.         a.type=0L;

  164.         if(atom.size >= 8) {

  165.             a.size = get_be32(pb);

  166.             a.type = get_le32(pb);

  167.         }

  168.         total_size += 8;

  169.         a.offset += 8;

  170.         dprintf(c->fc, "type: %08x  %.4s  sz: %"PRIx64"  %"PRIx64"   %"PRIx64"\n", a.type, (char*)&a.type, a.size, atom.size, total_size);

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

  172.             a.size = get_be64(pb) - 8;

  173.             a.offset += 8;

  174.             total_size += 8;

  175.         }

  176.         if (a.size == 0) {

  177.             a.size = atom.size - total_size;

  178.             if (a.size <= 8)

  179.                 break;

  180.         }

  181.         a.size -= 8;

  182.         if(a.size < 0)

  183.             break;

  184.         a.size = FFMIN(a.size, atom.size - total_size);

  185. 

  186.         for (i = 0; c->parse_table[i].type != 0L

  187.              && c->parse_table[i].type != a.type; i++)

  188.             /* empty */;

  189. 

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

  191.             url_fskip(pb, a.size);

  192.         } else {

  193.             offset_t start_pos = url_ftell(pb);

  194.             int64_t left;

  195.             err = (c->parse_table[i].func)(c, pb, a);

  196.             left = a.size - url_ftell(pb) + start_pos;

  197.             if (left > 0) /* skip garbage at atom end */

  198.                 url_fskip(pb, left);

  199.         }

  200. 

  201.         a.offset += a.size;

  202.         total_size += a.size;

  203.     }

  204. 

  205.     if (!err && total_size < atom.size && atom.size < 0x7ffff) {

  206.         url_fskip(pb, atom.size - total_size);

  207.     }

  208. 

  209.     return err;

  210. }

  211. 

  212. static int mov_read_hdlr(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  213. {

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

  215.     uint32_t type;

  216.     uint32_t ctype;

  217. 

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

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

  220. 

  221.     /* component type */

  222.     ctype = get_le32(pb);

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

  224. 

  225.     dprintf(c->fc, "ctype= %c%c%c%c (0x%08x)\n", *((char *)&ctype), ((char *)&ctype)[1], ((char *)&ctype)[2], ((char *)&ctype)[3], (int) ctype);

  226.     dprintf(c->fc, "stype= %c%c%c%c\n", *((char *)&type), ((char *)&type)[1], ((char *)&type)[2], ((char *)&type)[3]);

  227.     if(!ctype)

  228.         c->isom = 1;

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

  230.         st->codec->codec_type = CODEC_TYPE_VIDEO;

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

  232.         st->codec->codec_type = CODEC_TYPE_AUDIO;

  233.     else if(type == MKTAG('m', '1', 'a', ' '))

  234.         st->codec->codec_id = CODEC_ID_MP2;

  235.     else if(type == MKTAG('s', 'u', 'b', 'p')) {

  236.         st->codec->codec_type = CODEC_TYPE_SUBTITLE;

  237.         st->codec->codec_id = CODEC_ID_DVD_SUBTITLE;

  238.     }

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

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

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

  242. 

  243.     if(atom.size <= 24)

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

  245. 

  246.     url_fskip(pb, atom.size - (url_ftell(pb) - atom.offset));

  247.     return 0;

  248. }

  249. 

  250. static int mp4_read_descr_len(ByteIOContext *pb)

  251. {

  252.     int len = 0;

  253.     int count = 4;

  254.     while (count--) {

  255.         int c = get_byte(pb);

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

  257.         if (!(c & 0x80))

  258.             break;

  259.     }

  260.     return len;

  261. }

  262. 

  263. static int mp4_read_descr(MOVContext *c, ByteIOContext *pb, int *tag)

  264. {

  265.     int len;

  266.     *tag = get_byte(pb);

  267.     len = mp4_read_descr_len(pb);

  268.     dprintf(c->fc, "MPEG4 description: tag=0x%02x len=%d\n", *tag, len);

  269.     return len;

  270. }

  271. 

  272. #define MP4ESDescrTag                   0x03

  273. #define MP4DecConfigDescrTag            0x04

  274. #define MP4DecSpecificDescrTag          0x05

  275. 

  276. static int mov_read_esds(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  277. {

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

  279.     int tag, len;

  280. 

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

  282.     len = mp4_read_descr(c, pb, &tag);

  283.     if (tag == MP4ESDescrTag) {

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

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

  286.     } else

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

  288. 

  289.     len = mp4_read_descr(c, pb, &tag);

  290.     if (tag == MP4DecConfigDescrTag) {

  291.         int object_type_id = get_byte(pb);

  292.         get_byte(pb); /* stream type */

  293.         get_be24(pb); /* buffer size db */

  294.         get_be32(pb); /* max bitrate */

  295.         get_be32(pb); /* avg bitrate */

  296. 

  297.         st->codec->codec_id= codec_get_id(ff_mp4_obj_type, object_type_id);

  298.         dprintf(c->fc, "esds object type id %d\n", object_type_id);

  299.         len = mp4_read_descr(c, pb, &tag);

  300.         if (tag == MP4DecSpecificDescrTag) {

  301.             dprintf(c->fc, "Specific MPEG4 header len=%d\n", len);

  302.             st->codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);

  303.             if (st->codec->extradata) {

  304.                 get_buffer(pb, st->codec->extradata, len);

  305.                 st->codec->extradata_size = len;

  306.                 /* from mplayer */

  307.                 if ((*st->codec->extradata >> 3) == 29) {

  308.                     st->codec->codec_id = CODEC_ID_MP3ON4;

  309.                 }

  310.             }

  311.         }

  312.     }

  313.     return 0;

  314. }

  315. 

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

  317. static int mov_read_mdat(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  318. {

  319.     if(atom.size == 0) /* wrong one (MP4) */

  320.         return 0;

  321.     c->mdat_list = av_realloc(c->mdat_list, (c->mdat_count + 1) * sizeof(*c->mdat_list));

  322.     c->mdat_list[c->mdat_count].offset = atom.offset;

  323.     c->mdat_list[c->mdat_count].size = atom.size;

  324.     c->mdat_count++;

  325.     c->found_mdat=1;

  326.     c->mdat_offset = atom.offset;

  327.     if(c->found_moov)

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

  329.     url_fskip(pb, atom.size);

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

  331. }

  332. 

  333. static int mov_read_ftyp(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  334. {

  335.     uint32_t type = get_le32(pb);

  336. 

  337.     if (type != MKTAG('q','t',' ',' '))

  338.         c->isom = 1;

  339.     av_log(c->fc, AV_LOG_DEBUG, "ISO: File Type Major Brand: %.4s\n",(char *)&type);

  340.     get_be32(pb); /* minor version */

  341.     url_fskip(pb, atom.size - 8);

  342.     return 0;

  343. }

  344. 

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

  346. static int mov_read_moov(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  347. {

  348.     if (mov_read_default(c, pb, atom) < 0)

  349.         return -1;

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

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

  352.     c->found_moov=1;

  353.     if(c->found_mdat)

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

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

  356. }

  357. 

  358. 

  359. static int mov_read_mdhd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  360. {

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

  362.     MOVStreamContext *sc = st->priv_data;

  363.     int version = get_byte(pb);

  364.     int lang;

  365. 

  366.     if (version > 1)

  367.         return 1; /* unsupported */

  368. 

  369.     get_byte(pb); get_byte(pb);

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

  371. 

  372.     if (version == 1) {

  373.         get_be64(pb);

  374.         get_be64(pb);

  375.     } else {

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

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

  378.     }

  379. 

  380.     sc->time_scale = get_be32(pb);

  381.     st->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */

  382. 

  383.     lang = get_be16(pb); /* language */

  384.     ff_mov_lang_to_iso639(lang, st->language);

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

  386. 

  387.     return 0;

  388. }

  389. 

  390. static int mov_read_mvhd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  391. {

  392.     int version = get_byte(pb); /* version */

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

  394. 

  395.     if (version == 1) {

  396.         get_be64(pb);

  397.         get_be64(pb);

  398.     } else {

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

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

  401.     }

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

  403. 

  404.     dprintf(c->fc, "time scale = %i\n", c->time_scale);

  405. 

  406.     c->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */

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

  408. 

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

  410. 

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

  412. 

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

  414. 

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

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

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

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

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

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

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

  422. 

  423.     return 0;

  424. }

  425. 

  426. static int mov_read_smi(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  427. {

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

  429. 

  430.     if((uint64_t)atom.size > (1<<30))

  431.         return -1;

  432. 

  433.     // currently SVQ3 decoder expect full STSD header - so let's fake it

  434.     // this should be fixed and just SMI header should be passed

  435.     av_free(st->codec->extradata);

  436.     st->codec->extradata_size = 0x5a + atom.size;

  437.     st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);

  438. 

  439.     if (st->codec->extradata) {

  440.         memcpy(st->codec->extradata, "SVQ3", 4); // fake

  441.         get_buffer(pb, st->codec->extradata + 0x5a, atom.size);

  442.         dprintf(c->fc, "Reading SMI %"PRId64"  %s\n", atom.size, st->codec->extradata + 0x5a);

  443.     } else

  444.         url_fskip(pb, atom.size);

  445. 

  446.     return 0;

  447. }

  448. 

  449. static int mov_read_enda(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  450. {

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

  452.     int little_endian = get_be16(pb);

  453. 

  454.     if (little_endian) {

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

  456.         case CODEC_ID_PCM_S24BE:

  457.             st->codec->codec_id = CODEC_ID_PCM_S24LE;

  458.             break;

  459.         case CODEC_ID_PCM_S32BE:

  460.             st->codec->codec_id = CODEC_ID_PCM_S32LE;

  461.             break;

  462.         default:

  463.             break;

  464.         }

  465.     }

  466.     return 0;

  467. }

  468. 

  469. /* FIXME modify qdm2/svq3/h264 decoders to take full atom as extradata */

  470. static int mov_read_extradata(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  471. {

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

  473.     uint64_t size= (uint64_t)st->codec->extradata_size + atom.size + 8 + FF_INPUT_BUFFER_PADDING_SIZE;

  474.     uint8_t *buf;

  475.     if(size > INT_MAX || (uint64_t)atom.size > INT_MAX)

  476.         return -1;

  477.     buf= av_realloc(st->codec->extradata, size);

  478.     if(!buf)

  479.         return -1;

  480.     st->codec->extradata= buf;

  481.     buf+= st->codec->extradata_size;

  482.     st->codec->extradata_size= size - FF_INPUT_BUFFER_PADDING_SIZE;

  483.     AV_WB32(       buf    , atom.size + 8);

  484.     AV_WL32(       buf + 4, atom.type);

  485.     get_buffer(pb, buf + 8, atom.size);

  486.     return 0;

  487. }

  488. 

  489. static int mov_read_wave(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  490. {

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

  492. 

  493.     if((uint64_t)atom.size > (1<<30))

  494.         return -1;

  495. 

  496.     if (st->codec->codec_id == CODEC_ID_QDM2) {

  497.         // pass all frma atom to codec, needed at least for QDM2

  498.         av_free(st->codec->extradata);

  499.         st->codec->extradata_size = atom.size;

  500.         st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);

  501. 

  502.         if (st->codec->extradata) {

  503.             get_buffer(pb, st->codec->extradata, atom.size);

  504.         } else

  505.             url_fskip(pb, atom.size);

  506.     } else if (atom.size > 8) { /* to read frma, esds atoms */

  507.         if (mov_read_default(c, pb, atom) < 0)

  508.             return -1;

  509.     } else

  510.         url_fskip(pb, atom.size);

  511.     return 0;

  512. }

  513. 

  514. static int mov_read_avcC(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  515. {

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

  517. 

  518.     if((uint64_t)atom.size > (1<<30))

  519.         return -1;

  520. 

  521.     av_free(st->codec->extradata);

  522. 

  523.     st->codec->extradata_size = atom.size;

  524.     st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);

  525. 

  526.     if (st->codec->extradata) {

  527.         get_buffer(pb, st->codec->extradata, atom.size);

  528.     } else

  529.         url_fskip(pb, atom.size);

  530. 

  531.     return 0;

  532. }

  533. 

  534. static int mov_read_stco(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  535. {

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

  537.     MOVStreamContext *sc = st->priv_data;

  538.     unsigned int i, entries;

  539. 

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

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

  542. 

  543.     entries = get_be32(pb);

  544. 

  545.     if(entries >= UINT_MAX/sizeof(int64_t))

  546.         return -1;

  547. 

  548.     sc->chunk_count = entries;

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

  550.     if (!sc->chunk_offsets)

  551.         return -1;

  552.     if (atom.type == MKTAG('s', 't', 'c', 'o')) {

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

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

  555.         }

  556.     } else if (atom.type == MKTAG('c', 'o', '6', '4')) {

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

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

  559.         }

  560.     } else

  561.         return -1;

  562. 

  563.     return 0;

  564. }

  565. 

  566. static int mov_read_stsd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  567. {

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

  569.     MOVStreamContext *sc = st->priv_data;

  570.     int entries, frames_per_sample;

  571.     uint32_t format;

  572.     uint8_t codec_name[32];

  573. 

  574.     /* for palette traversal */

  575.     unsigned int color_depth;

  576.     unsigned int color_start;

  577.     unsigned int color_count;

  578.     unsigned int color_end;

  579.     int color_index;

  580.     int color_dec;

  581.     int color_greyscale;

  582.     const uint8_t *color_table;

  583.     int j;

  584.     unsigned char r, g, b;

  585. 

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

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

  588. 

  589.     entries = get_be32(pb);

  590. 

  591.     while(entries--) { //Parsing Sample description table

  592.         enum CodecID id;

  593.         MOV_atom_t a = { 0, 0, 0 };

  594.         offset_t start_pos = url_ftell(pb);

  595.         int size = get_be32(pb); /* size */

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

  597. 

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

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

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

  601. 

  602.         if (st->codec->codec_tag) {

  603.             /* multiple fourcc, just skip for now */

  604.             url_fskip(pb, size - (url_ftell(pb) - start_pos));

  605.             continue;

  606.         }

  607. 

  608.         st->codec->codec_tag = format;

  609.         id = codec_get_id(codec_movaudio_tags, format);

  610.         if (id<=0 && (format&0xFFFF) == 'm' + ('s'<<8))

  611.             id = codec_get_id(codec_wav_tags, bswap_32(format)&0xFFFF);

  612. 

  613.         if (st->codec->codec_type != CODEC_TYPE_VIDEO && id > 0) {

  614.             st->codec->codec_type = CODEC_TYPE_AUDIO;

  615.         } else if (st->codec->codec_type != CODEC_TYPE_AUDIO && /* do not overwrite codec type */

  616.                    format && format != MKTAG('m', 'p', '4', 's')) { /* skip old asf mpeg4 tag */

  617.             id = codec_get_id(codec_movvideo_tags, format);

  618.             if (id <= 0)

  619.                 id = codec_get_id(codec_bmp_tags, format);

  620.             if (id > 0)

  621.                 st->codec->codec_type = CODEC_TYPE_VIDEO;

  622.         }

  623. 

  624.         dprintf(c->fc, "size=%d 4CC= %c%c%c%c codec_type=%d\n",

  625.                 size,

  626.                 (format >> 0) & 0xff, (format >> 8) & 0xff, (format >> 16) & 0xff, (format >> 24) & 0xff,

  627.                 st->codec->codec_type);

  628. 

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

  630.             st->codec->codec_id = id;

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

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

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

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

  635.             get_be32(pb); /* spatial quality */

  636. 

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

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

  639. 

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

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

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

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

  644. 

  645.             dprintf(c->fc, "frames/samples = %d\n", frames_per_sample);

  646. 

  647.             get_buffer(pb, codec_name, 32); /* codec name, pascal string (FIXME: true for mp4?) */

  648.             if (codec_name[0] <= 31) {

  649.                 memcpy(st->codec->codec_name, &codec_name[1],codec_name[0]);

  650.                 st->codec->codec_name[codec_name[0]] = 0;

  651.             }

  652. 

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

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

  655. 

  656.             /* figure out the palette situation */

  657.             color_depth = st->codec->bits_per_sample & 0x1F;

  658.             color_greyscale = st->codec->bits_per_sample & 0x20;

  659. 

  660.             /* if the depth is 2, 4, or 8 bpp, file is palettized */

  661.             if ((color_depth == 2) || (color_depth == 4) ||

  662.                 (color_depth == 8)) {

  663. 

  664.                 if (color_greyscale) {

  665. 

  666.                     /* compute the greyscale palette */

  667.                     color_count = 1 << color_depth;

  668.                     color_index = 255;

  669.                     color_dec = 256 / (color_count - 1);

  670.                     for (j = 0; j < color_count; j++) {

  671.                         r = g = b = color_index;

  672.                         c->palette_control.palette[j] =

  673.                             (r << 16) | (g << 8) | (b);

  674.                         color_index -= color_dec;

  675.                         if (color_index < 0)

  676.                             color_index = 0;

  677.                     }

  678. 

  679.                 } else if (st->codec->color_table_id & 0x08) {

  680. 

  681.                     /* if flag bit 3 is set, use the default palette */

  682.                     color_count = 1 << color_depth;

  683.                     if (color_depth == 2)

  684.                         color_table = ff_qt_default_palette_4;

  685.                     else if (color_depth == 4)

  686.                         color_table = ff_qt_default_palette_16;

  687.                     else

  688.                         color_table = ff_qt_default_palette_256;

  689. 

  690.                     for (j = 0; j < color_count; j++) {

  691.                         r = color_table[j * 4 + 0];

  692.                         g = color_table[j * 4 + 1];

  693.                         b = color_table[j * 4 + 2];

  694.                         c->palette_control.palette[j] =

  695.                             (r << 16) | (g << 8) | (b);

  696.                     }

  697. 

  698.                 } else {

  699. 

  700.                     /* load the palette from the file */

  701.                     color_start = get_be32(pb);

  702.                     color_count = get_be16(pb);

  703.                     color_end = get_be16(pb);

  704.                     if ((color_start <= 255) &&

  705.                         (color_end <= 255)) {

  706.                         for (j = color_start; j <= color_end; j++) {

  707.                             /* each R, G, or B component is 16 bits;

  708.                              * only use the top 8 bits; skip alpha bytes

  709.                              * up front */

  710.                             get_byte(pb);

  711.                             get_byte(pb);

  712.                             r = get_byte(pb);

  713.                             get_byte(pb);

  714.                             g = get_byte(pb);

  715.                             get_byte(pb);

  716.                             b = get_byte(pb);

  717.                             get_byte(pb);

  718.                             c->palette_control.palette[j] =

  719.                                 (r << 16) | (g << 8) | (b);

  720.                         }

  721.                     }

  722.                 }

  723. 

  724.                 st->codec->palctrl = &c->palette_control;

  725.                 st->codec->palctrl->palette_changed = 1;

  726.             } else

  727.                 st->codec->palctrl = NULL;

  728.         } else if(st->codec->codec_type==CODEC_TYPE_AUDIO) {

  729.             int bits_per_sample;

  730.             uint16_t version = get_be16(pb);

  731. 

  732.             st->codec->codec_id = id;

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

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

  735. 

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

  737.             dprintf(c->fc, "audio channels %d\n", st->codec->channels);

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

  739.             /* do we need to force to 16 for AMR ? */

  740. 

  741.             /* handle specific s8 codec */

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

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

  744. 

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

  746. 

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

  748.             case CODEC_ID_PCM_S8:

  749.             case CODEC_ID_PCM_U8:

  750.                 if (st->codec->bits_per_sample == 16)

  751.                     st->codec->codec_id = CODEC_ID_PCM_S16BE;

  752.                 break;

  753.             case CODEC_ID_PCM_S16LE:

  754.             case CODEC_ID_PCM_S16BE:

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

  756.                     st->codec->codec_id = CODEC_ID_PCM_S8;

  757.                 else if (st->codec->bits_per_sample == 24)

  758.                     st->codec->codec_id = CODEC_ID_PCM_S24BE;

  759.                 break;

  760.             default:

  761.                 break;

  762.             }

  763. 

  764.             //Read QT version 1 fields. In version 0 these do not exist.

  765.             dprintf(c->fc, "version =%d, isom =%d\n",version,c->isom);

  766.             if(!c->isom) {

  767.                 if(version==1) {

  768.                     sc->samples_per_frame = get_be32(pb);

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

  770.                     sc->bytes_per_frame = get_be32(pb);

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

  772.                 } else if(version==2) {

  773.                     get_be32(pb); /* sizeof struct only */

  774.                     st->codec->sample_rate = av_int2dbl(get_be64(pb)); /* float 64 */

  775.                     st->codec->channels = get_be32(pb);

  776.                     get_be32(pb); /* always 0x7F000000 */

  777.                     get_be32(pb); /* bits per channel if sound is uncompressed */

  778.                     get_be32(pb); /* lcpm format specific flag */

  779.                     get_be32(pb); /* bytes per audio packet if constant */

  780.                     get_be32(pb); /* lpcm frames per audio packet if constant */

  781.                 }

  782.             }

  783. 

  784.             bits_per_sample = av_get_bits_per_sample(st->codec->codec_id);

  785.             if (bits_per_sample) {

  786.                 st->codec->bits_per_sample = bits_per_sample;

  787.                 sc->sample_size = (bits_per_sample >> 3) * st->codec->channels;

  788.             }

  789.         } else {

  790.             /* other codec type, just skip (rtp, mp4s, tmcd ...) */

  791.             url_fskip(pb, size - (url_ftell(pb) - start_pos));

  792.         }

  793.         /* this will read extra atoms at the end (wave, alac, damr, avcC, SMI ...) */

  794.         a.size = size - (url_ftell(pb) - start_pos);

  795.         if (a.size > 8) {

  796.             if (mov_read_default(c, pb, a) < 0)

  797.                 return -1;

  798.         } else if (a.size > 0)

  799.             url_fskip(pb, a.size);

  800.     }

  801. 

  802.     if(st->codec->codec_type==CODEC_TYPE_AUDIO && st->codec->sample_rate==0 && sc->time_scale>1) {

  803.         st->codec->sample_rate= sc->time_scale;

  804.     }

  805. 

  806.     /* special codec parameters handling */

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

  808. #ifdef CONFIG_H261_DECODER

  809.     case CODEC_ID_H261:

  810. #endif

  811. #ifdef CONFIG_H263_DECODER

  812.     case CODEC_ID_H263:

  813. #endif

  814. #ifdef CONFIG_MPEG4_DECODER

  815.     case CODEC_ID_MPEG4:

  816. #endif

  817.         st->codec->width= 0; /* let decoder init width/height */

  818.         st->codec->height= 0;

  819.         break;

  820. #ifdef CONFIG_LIBFAAD

  821.     case CODEC_ID_AAC:

  822. #endif

  823. #ifdef CONFIG_VORBIS_DECODER

  824.     case CODEC_ID_VORBIS:

  825. #endif

  826.     case CODEC_ID_MP3ON4:

  827.         st->codec->sample_rate= 0; /* let decoder init parameters properly */

  828.         break;

  829. #ifdef CONFIG_DV_DEMUXER

  830.     case CODEC_ID_DVAUDIO:

  831.         c->dv_fctx = av_alloc_format_context();

  832.         c->dv_demux = dv_init_demux(c->dv_fctx);

  833.         if (!c->dv_demux) {

  834.             av_log(c->fc, AV_LOG_ERROR, "dv demux context init error\n");

  835.             return -1;

  836.         }

  837.         sc->dv_audio_container = 1;

  838.         st->codec->codec_id = CODEC_ID_PCM_S16LE;

  839.         break;

  840. #endif

  841.     /* no ifdef since parameters are always those */

  842.     case CODEC_ID_AMR_WB:

  843.         st->codec->sample_rate= 16000;

  844.         st->codec->channels= 1; /* really needed */

  845.         break;

  846.     case CODEC_ID_AMR_NB:

  847.         st->codec->sample_rate= 8000;

  848.         st->codec->channels= 1; /* really needed */

  849.         break;

  850.     case CODEC_ID_MP2:

  851.     case CODEC_ID_MP3:

  852.         st->codec->codec_type = CODEC_TYPE_AUDIO; /* force type after stsd for m1a hdlr */

  853.         st->need_parsing = AVSTREAM_PARSE_FULL;

  854.         break;

  855.     case CODEC_ID_ADPCM_MS:

  856.     case CODEC_ID_ADPCM_IMA_WAV:

  857.         st->codec->block_align = sc->bytes_per_frame;

  858.         break;

  859.     default:

  860.         break;

  861.     }

  862. 

  863.     return 0;

  864. }

  865. 

  866. static int mov_read_stsc(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  867. {

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

  869.     MOVStreamContext *sc = st->priv_data;

  870.     unsigned int i, entries;

  871. 

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

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

  874. 

  875.     entries = get_be32(pb);

  876. 

  877.     if(entries >= UINT_MAX / sizeof(MOV_stsc_t))

  878.         return -1;

  879. 

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

  881. 

  882.     sc->sample_to_chunk_sz = entries;

  883.     sc->sample_to_chunk = av_malloc(entries * sizeof(MOV_stsc_t));

  884.     if (!sc->sample_to_chunk)

  885.         return -1;

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

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

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

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

  890.     }

  891.     return 0;

  892. }

  893. 

  894. static int mov_read_stss(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  895. {

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

  897.     MOVStreamContext *sc = st->priv_data;

  898.     unsigned int i, entries;

  899. 

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

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

  902. 

  903.     entries = get_be32(pb);

  904. 

  905.     if(entries >= UINT_MAX / sizeof(int))

  906.         return -1;

  907. 

  908.     sc->keyframe_count = entries;

  909. 

  910.     dprintf(c->fc, "keyframe_count = %d\n", sc->keyframe_count);

  911. 

  912.     sc->keyframes = av_malloc(entries * sizeof(int));

  913.     if (!sc->keyframes)

  914.         return -1;

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

  916.         sc->keyframes[i] = get_be32(pb);

  917.         //dprintf(c->fc, "keyframes[]=%d\n", sc->keyframes[i]);

  918.     }

  919.     return 0;

  920. }

  921. 

  922. static int mov_read_stsz(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  923. {

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

  925.     MOVStreamContext *sc = st->priv_data;

  926.     unsigned int i, entries, sample_size;

  927. 

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

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

  930. 

  931.     sample_size = get_be32(pb);

  932.     if (!sc->sample_size) /* do not overwrite value computed in stsd */

  933.         sc->sample_size = sample_size;

  934.     entries = get_be32(pb);

  935.     if(entries >= UINT_MAX / sizeof(int))

  936.         return -1;

  937. 

  938.     sc->sample_count = entries;

  939.     if (sample_size)

  940.         return 0;

  941. 

  942.     dprintf(c->fc, "sample_size = %d sample_count = %d\n", sc->sample_size, sc->sample_count);

  943. 

  944.     sc->sample_sizes = av_malloc(entries * sizeof(int));

  945.     if (!sc->sample_sizes)

  946.         return -1;

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

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

  949.         dprintf(c->fc, "sample_sizes[]=%d\n", sc->sample_sizes[i]);

  950.     }

  951.     return 0;

  952. }

  953. 

  954. static int mov_read_stts(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  955. {

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

  957.     MOVStreamContext *sc = st->priv_data;

  958.     unsigned int i, entries;

  959.     int64_t duration=0;

  960.     int64_t total_sample_count=0;

  961. 

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

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

  964.     entries = get_be32(pb);

  965.     if(entries >= UINT_MAX / sizeof(MOV_stts_t))

  966.         return -1;

  967. 

  968.     sc->stts_count = entries;

  969.     sc->stts_data = av_malloc(entries * sizeof(MOV_stts_t));

  970.     if (!sc->stts_data)

  971.         return -1;

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

  973. 

  974.     sc->time_rate=0;

  975. 

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

  977.         int sample_duration;

  978.         int sample_count;

  979. 

  980.         sample_count=get_be32(pb);

  981.         sample_duration = get_be32(pb);

  982.         sc->stts_data[i].count= sample_count;

  983.         sc->stts_data[i].duration= sample_duration;

  984. 

  985.         sc->time_rate= ff_gcd(sc->time_rate, sample_duration);

  986. 

  987.         dprintf(c->fc, "sample_count=%d, sample_duration=%d\n",sample_count,sample_duration);

  988. 

  989.         duration+=(int64_t)sample_duration*sample_count;

  990.         total_sample_count+=sample_count;

  991.     }

  992. 

  993.     st->nb_frames= total_sample_count;

  994.     if(duration)

  995.         st->duration= duration;

  996.     return 0;

  997. }

  998. 

  999. static int mov_read_ctts(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  1000. {

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

  1002.     MOVStreamContext *sc = st->priv_data;

  1003.     unsigned int i, entries;

  1004. 

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

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

  1007.     entries = get_be32(pb);

  1008.     if(entries >= UINT_MAX / sizeof(MOV_stts_t))

  1009.         return -1;

  1010. 

  1011.     sc->ctts_count = entries;

  1012.     sc->ctts_data = av_malloc(entries * sizeof(MOV_stts_t));

  1013.     if (!sc->ctts_data)

  1014.         return -1;

  1015.     dprintf(c->fc, "track[%i].ctts.entries = %i\n", c->fc->nb_streams-1, entries);

  1016. 

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

  1018.         int count    =get_be32(pb);

  1019.         int duration =get_be32(pb);

  1020. 

  1021.         if (duration < 0) {

  1022.             av_log(c->fc, AV_LOG_ERROR, "negative ctts, ignoring\n");

  1023.             sc->ctts_count = 0;

  1024.             url_fskip(pb, 8 * (entries - i - 1));

  1025.             break;

  1026.         }

  1027.         sc->ctts_data[i].count   = count;

  1028.         sc->ctts_data[i].duration= duration;

  1029. 

  1030.         sc->time_rate= ff_gcd(sc->time_rate, duration);

  1031.     }

  1032.     return 0;

  1033. }

  1034. 

  1035. static int mov_read_trak(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  1036. {

  1037.     AVStream *st;

  1038.     MOVStreamContext *sc;

  1039. 

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

  1041.     if (!st) return -2;

  1042.     sc = av_mallocz(sizeof(MOVStreamContext));

  1043.     if (!sc) {

  1044.         av_free(st);

  1045.         return -1;

  1046.     }

  1047. 

  1048.     st->priv_data = sc;

  1049.     st->codec->codec_type = CODEC_TYPE_DATA;

  1050.     st->start_time = 0; /* XXX: check */

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

  1052. 

  1053.     return mov_read_default(c, pb, atom);

  1054. }

  1055. 

  1056. static void mov_parse_udta_string(ByteIOContext *pb, char *str, int size)

  1057. {

  1058.     uint16_t str_size = get_be16(pb); /* string length */;

  1059. 

  1060.     get_be16(pb); /* skip language */

  1061.     get_buffer(pb, str, FFMIN(size, str_size));

  1062. }

  1063. 

  1064. static int mov_read_udta(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  1065. {

  1066.     uint64_t end = url_ftell(pb) + atom.size;

  1067. 

  1068.     while (url_ftell(pb) + 8 < end) {

  1069.         uint32_t tag_size = get_be32(pb);

  1070.         uint32_t tag      = get_le32(pb);

  1071.         uint64_t next     = url_ftell(pb) + tag_size - 8;

  1072. 

  1073.         if (next > end) // stop if tag_size is wrong

  1074.             break;

  1075. 

  1076.         switch (tag) {

  1077.         case MKTAG(0xa9,'n','a','m'):

  1078.             mov_parse_udta_string(pb, c->fc->title,     sizeof(c->fc->title));

  1079.             break;

  1080.         case MKTAG(0xa9,'w','r','t'):

  1081.             mov_parse_udta_string(pb, c->fc->author,    sizeof(c->fc->author));

  1082.             break;

  1083.         case MKTAG(0xa9,'c','p','y'):

  1084.             mov_parse_udta_string(pb, c->fc->copyright, sizeof(c->fc->copyright));

  1085.             break;

  1086.         case MKTAG(0xa9,'i','n','f'):

  1087.             mov_parse_udta_string(pb, c->fc->comment,   sizeof(c->fc->comment));

  1088.             break;

  1089.         default:

  1090.             break;

  1091.         }

  1092. 

  1093.         url_fseek(pb, next, SEEK_SET);

  1094.     }

  1095. 

  1096.     return 0;

  1097. }

  1098. 

  1099. static int mov_read_tkhd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  1100. {

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

  1102.     int version = get_byte(pb);

  1103. 

  1104.     get_byte(pb); get_byte(pb);

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

  1106.     /*

  1107.     MOV_TRACK_ENABLED 0x0001

  1108.     MOV_TRACK_IN_MOVIE 0x0002

  1109.     MOV_TRACK_IN_PREVIEW 0x0004

  1110.     MOV_TRACK_IN_POSTER 0x0008

  1111.     */

  1112. 

  1113.     if (version == 1) {

  1114.         get_be64(pb);

  1115.         get_be64(pb);

  1116.     } else {

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

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

  1119.     }

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

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

  1122.     st->start_time = 0; /* check */

  1123.     (version == 1) ? get_be64(pb) : get_be32(pb); /* highlevel (considering edits) duration in movie timebase */

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

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

  1126. 

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

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

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

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

  1131. 

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

  1133. 

  1134.     /* those are fixed-point */

  1135.     get_be32(pb); /* track width */

  1136.     get_be32(pb); /* track height */

  1137. 

  1138.     return 0;

  1139. }

  1140. 

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

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

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

  1144. static int mov_read_wide(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  1145. {

  1146.     int err;

  1147. 

  1148.     if (atom.size < 8)

  1149.         return 0; /* continue */

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

  1151.         url_fskip(pb, atom.size - 4);

  1152.         return 0;

  1153.     }

  1154.     atom.type = get_le32(pb);

  1155.     atom.offset += 8;

  1156.     atom.size -= 8;

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

  1158.         url_fskip(pb, atom.size);

  1159.         return 0;

  1160.     }

  1161.     err = mov_read_mdat(c, pb, atom);

  1162.     return err;

  1163. }

  1164. 

  1165. static int mov_read_cmov(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  1166. {

  1167. #ifdef CONFIG_ZLIB

  1168.     ByteIOContext ctx;

  1169.     uint8_t *cmov_data;

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

  1171.     long cmov_len, moov_len;

  1172.     int ret;

  1173. 

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

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

  1176.         return -1;

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

  1178.         av_log(NULL, AV_LOG_ERROR, "unknown compression for cmov atom !");

  1179.         return -1;

  1180.     }

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

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

  1183.         return -1;

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

  1185.     cmov_len = atom.size - 6 * 4;

  1186. 

  1187.     cmov_data = av_malloc(cmov_len);

  1188.     if (!cmov_data)

  1189.         return -1;

  1190.     moov_data = av_malloc(moov_len);

  1191.     if (!moov_data) {

  1192.         av_free(cmov_data);

  1193.         return -1;

  1194.     }

  1195.     get_buffer(pb, cmov_data, cmov_len);

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

  1197.         return -1;

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

  1199.         return -1;

  1200.     atom.type = MKTAG( 'm', 'o', 'o', 'v' );

  1201.     atom.offset = 0;

  1202.     atom.size = moov_len;

  1203. #ifdef DEBUG

  1204. //    { int fd = open("/tmp/uncompheader.mov", O_WRONLY | O_CREAT); write(fd, moov_data, moov_len); close(fd); }

  1205. #endif

  1206.     ret = mov_read_default(c, &ctx, atom);

  1207.     av_free(moov_data);

  1208.     av_free(cmov_data);

  1209.     return ret;

  1210. #else

  1211.     av_log(c->fc, AV_LOG_ERROR, "this file requires zlib support compiled in\n");

  1212.     return -1;

  1213. #endif

  1214. }

  1215. 

  1216. /* edit list atom */

  1217. static int mov_read_elst(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)

  1218. {

  1219.     int i, edit_count;

  1220. 

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

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

  1223.     edit_count= c->streams[c->fc->nb_streams-1]->edit_count = get_be32(pb);     /* entries */

  1224. 

  1225.     for(i=0; i<edit_count; i++){

  1226.         get_be32(pb); /* Track duration */

  1227.         get_be32(pb); /* Media time */

  1228.         get_be32(pb); /* Media rate */

  1229.     }

  1230.     dprintf(c->fc, "track[%i].edit_count = %i\n", c->fc->nb_streams-1, c->streams[c->fc->nb_streams-1]->edit_count);

  1231.     return 0;

  1232. }

  1233. 

  1234. static const MOVParseTableEntry mov_default_parse_table[] = {

  1235. /* mp4 atoms */

  1236. { MKTAG( 'c', 'o', '6', '4' ), mov_read_stco },

  1237. { MKTAG( 'c', 't', 't', 's' ), mov_read_ctts }, /* composition time to sample */

  1238. { MKTAG( 'e', 'd', 't', 's' ), mov_read_default },

  1239. { MKTAG( 'e', 'l', 's', 't' ), mov_read_elst },

  1240. { MKTAG( 'e', 'n', 'd', 'a' ), mov_read_enda },

  1241. { MKTAG( 'f', 'i', 'e', 'l' ), mov_read_extradata },

  1242. { MKTAG( 'f', 't', 'y', 'p' ), mov_read_ftyp },

  1243. { MKTAG( 'h', 'd', 'l', 'r' ), mov_read_hdlr },

  1244. { MKTAG( 'j', 'p', '2', 'h' ), mov_read_extradata },

  1245. { MKTAG( 'm', 'd', 'a', 't' ), mov_read_mdat },

  1246. { MKTAG( 'm', 'd', 'h', 'd' ), mov_read_mdhd },

  1247. { MKTAG( 'm', 'd', 'i', 'a' ), mov_read_default },

  1248. { MKTAG( 'm', 'i', 'n', 'f' ), mov_read_default },

  1249. { MKTAG( 'm', 'o', 'o', 'v' ), mov_read_moov },

  1250. { MKTAG( 'm', 'v', 'h', 'd' ), mov_read_mvhd },

  1251. { MKTAG( 'S', 'M', 'I', ' ' ), mov_read_smi }, /* Sorenson extension ??? */

  1252. { MKTAG( 'a', 'l', 'a', 'c' ), mov_read_extradata }, /* alac specific atom */

  1253. { MKTAG( 'a', 'v', 'c', 'C' ), mov_read_avcC },

  1254. { MKTAG( 's', 't', 'b', 'l' ), mov_read_default },

  1255. { MKTAG( 's', 't', 'c', 'o' ), mov_read_stco },

  1256. { MKTAG( 's', 't', 's', 'c' ), mov_read_stsc },

  1257. { MKTAG( 's', 't', 's', 'd' ), mov_read_stsd }, /* sample description */

  1258. { MKTAG( 's', 't', 's', 's' ), mov_read_stss }, /* sync sample */

  1259. { MKTAG( 's', 't', 's', 'z' ), mov_read_stsz }, /* sample size */

  1260. { MKTAG( 's', 't', 't', 's' ), mov_read_stts },

  1261. { MKTAG( 't', 'k', 'h', 'd' ), mov_read_tkhd }, /* track header */

  1262. { MKTAG( 't', 'r', 'a', 'k' ), mov_read_trak },

  1263. { MKTAG( 'u', 'd', 't', 'a' ), mov_read_udta },

  1264. { MKTAG( 'w', 'a', 'v', 'e' ), mov_read_wave },

  1265. { MKTAG( 'e', 's', 'd', 's' ), mov_read_esds },

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

  1267. { MKTAG( 'c', 'm', 'o', 'v' ), mov_read_cmov },

  1268. { 0L, NULL }

  1269. };

  1270. 

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

  1272. static int mov_probe(AVProbeData *p)

  1273. {

  1274.     unsigned int offset;

  1275.     uint32_t tag;

  1276.     int score = 0;

  1277. 

  1278.     /* check file header */

  1279.     offset = 0;

  1280.     for(;;) {

  1281.         /* ignore invalid offset */

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

  1283.             return score;

  1284.         tag = AV_RL32(p->buf + offset + 4);

  1285.         switch(tag) {

  1286.         /* check for obvious tags */

  1287.         case MKTAG( 'j', 'P', ' ', ' ' ): /* jpeg 2000 signature */

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

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

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

  1291.         case MKTAG( 'u', 'd', 't', 'a' ): /* Packet Video PVAuthor adds this and a lot of more junk */

  1292.             return AVPROBE_SCORE_MAX;

  1293.         /* those are more common words, so rate then a bit less */

  1294.         case MKTAG( 'e', 'd', 'i', 'w' ): /* xdcam files have reverted first tags */

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

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

  1297.         case MKTAG( 'j', 'u', 'n', 'k' ):

  1298.         case MKTAG( 'p', 'i', 'c', 't' ):

  1299.             return AVPROBE_SCORE_MAX - 5;

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

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

  1302.         case MKTAG( 'u', 'u', 'i', 'd' ):

  1303.             offset = AV_RB32(p->buf+offset) + offset;

  1304.             /* if we only find those cause probedata is too small at least rate them */

  1305.             score = AVPROBE_SCORE_MAX - 50;

  1306.             break;

  1307.         default:

  1308.             /* unrecognized tag */

  1309.             return score;

  1310.         }

  1311.     }

  1312.     return score;

  1313. }

  1314. 

  1315. static void mov_build_index(MOVContext *mov, AVStream *st)

  1316. {

  1317.     MOVStreamContext *sc = st->priv_data;

  1318.     offset_t current_offset;

  1319.     int64_t current_dts = 0;

  1320.     unsigned int stts_index = 0;

  1321.     unsigned int stsc_index = 0;

  1322.     unsigned int stss_index = 0;

  1323.     unsigned int i, j, k;

  1324. 

  1325.     if (sc->sample_sizes || st->codec->codec_type == CODEC_TYPE_VIDEO || sc->dv_audio_container) {

  1326.         unsigned int current_sample = 0;

  1327.         unsigned int stts_sample = 0;

  1328.         unsigned int keyframe, sample_size;

  1329.         unsigned int distance = 0;

  1330. 

  1331.         st->nb_frames = sc->sample_count;

  1332.         for (i = 0; i < sc->chunk_count; i++) {

  1333.             current_offset = sc->chunk_offsets[i];

  1334.             if (stsc_index + 1 < sc->sample_to_chunk_sz && i + 1 == sc->sample_to_chunk[stsc_index + 1].first)

  1335.                 stsc_index++;

  1336.             for (j = 0; j < sc->sample_to_chunk[stsc_index].count; j++) {

  1337.                 if (current_sample >= sc->sample_count) {

  1338.                     av_log(mov->fc, AV_LOG_ERROR, "wrong sample count\n");

  1339.                     goto out;

  1340.                 }

  1341.                 keyframe = !sc->keyframe_count || current_sample + 1 == sc->keyframes[stss_index];

  1342.                 if (keyframe) {

  1343.                     distance = 0;

  1344.                     if (stss_index + 1 < sc->keyframe_count)

  1345.                         stss_index++;

  1346.                 }

  1347.                 sample_size = sc->sample_size > 0 ? sc->sample_size : sc->sample_sizes[current_sample];

  1348.                 dprintf(mov->fc, "AVIndex stream %d, sample %d, offset %"PRIx64", dts %"PRId64", size %d, distance %d, keyframe %d\n",

  1349.                         st->index, current_sample, current_offset, current_dts, sample_size, distance, keyframe);

  1350.                 av_add_index_entry(st, current_offset, current_dts, sample_size, distance, keyframe ? AVINDEX_KEYFRAME : 0);

  1351.                 current_offset += sample_size;

  1352.                 assert(sc->stts_data[stts_index].duration % sc->time_rate == 0);

  1353.                 current_dts += sc->stts_data[stts_index].duration / sc->time_rate;

  1354.                 distance++;

  1355.                 stts_sample++;

  1356.                 current_sample++;

  1357.                 if (stts_index + 1 < sc->stts_count && stts_sample == sc->stts_data[stts_index].count) {

  1358.                     stts_sample = 0;

  1359.                     stts_index++;

  1360.                 }

  1361.             }

  1362.         }

  1363.     } else { /* read whole chunk */

  1364.         unsigned int chunk_samples, chunk_size, chunk_duration;

  1365. 

  1366.         for (i = 0; i < sc->chunk_count; i++) {

  1367.             current_offset = sc->chunk_offsets[i];

  1368.             if (stsc_index + 1 < sc->sample_to_chunk_sz && i + 1 == sc->sample_to_chunk[stsc_index + 1].first)

  1369.                 stsc_index++;

  1370.             chunk_samples = sc->sample_to_chunk[stsc_index].count;

  1371.             /* get chunk size */

  1372.             if (sc->sample_size > 1 || st->codec->codec_id == CODEC_ID_PCM_U8 || st->codec->codec_id == CODEC_ID_PCM_S8)

  1373.                 chunk_size = chunk_samples * sc->sample_size;

  1374.             else if (sc->samples_per_frame > 0 && (chunk_samples * sc->bytes_per_frame % sc->samples_per_frame == 0))

  1375.                 chunk_size = chunk_samples * sc->bytes_per_frame / sc->samples_per_frame;

  1376.             else { /* workaround to find nearest next chunk offset */

  1377.                 chunk_size = INT_MAX;

  1378.                 for (j = 0; j < mov->total_streams; j++) {

  1379.                     MOVStreamContext *msc = mov->streams[j];

  1380. 

  1381.                     for (k = msc->next_chunk; k < msc->chunk_count; k++) {

  1382.                         if (msc->chunk_offsets[k] > current_offset && msc->chunk_offsets[k] - current_offset < chunk_size) {

  1383.                             chunk_size = msc->chunk_offsets[k] - current_offset;

  1384.                             msc->next_chunk = k;

  1385.                             break;

  1386.                         }

  1387.                     }

  1388.                 }

  1389.                 /* check for last chunk */

  1390.                 if (chunk_size == INT_MAX)

  1391.                     for (j = 0; j < mov->mdat_count; j++) {

  1392.                         dprintf(mov->fc, "mdat %d, offset %"PRIx64", size %"PRId64", current offset %"PRIx64"\n",

  1393.                                 j, mov->mdat_list[j].offset, mov->mdat_list[j].size, current_offset);

  1394.                         if (mov->mdat_list[j].offset <= current_offset && mov->mdat_list[j].offset + mov->mdat_list[j].size > current_offset)

  1395.                             chunk_size = mov->mdat_list[j].offset + mov->mdat_list[j].size - current_offset;

  1396.                     }

  1397.                 assert(chunk_size != INT_MAX);

  1398.                 for (j = 0; j < mov->total_streams; j++) {

  1399.                     mov->streams[j]->next_chunk = 0;

  1400.                 }

  1401.             }

  1402.             av_add_index_entry(st, current_offset, current_dts, chunk_size, 0, AVINDEX_KEYFRAME);

  1403.             /* get chunk duration */

  1404.             chunk_duration = 0;

  1405.             while (chunk_samples > 0) {

  1406.                 if (chunk_samples < sc->stts_data[stts_index].count) {

  1407.                     chunk_duration += sc->stts_data[stts_index].duration * chunk_samples;

  1408.                     sc->stts_data[stts_index].count -= chunk_samples;

  1409.                     break;

  1410.                 } else {

  1411.                     chunk_duration += sc->stts_data[stts_index].duration * chunk_samples;

  1412.                     chunk_samples -= sc->stts_data[stts_index].count;

  1413.                     if (stts_index + 1 < sc->stts_count) {

  1414.                         stts_index++;

  1415.                     }

  1416.                 }

  1417.             }

  1418.             dprintf(mov->fc, "AVIndex stream %d, chunk %d, offset %"PRIx64", dts %"PRId64", size %d, duration %d\n",

  1419.                     st->index, i, current_offset, current_dts, chunk_size, chunk_duration);

  1420.             assert(chunk_duration % sc->time_rate == 0);

  1421.             current_dts += chunk_duration / sc->time_rate;

  1422.         }

  1423.     }

  1424.  out:

  1425.     /* adjust sample count to avindex entries */

  1426.     sc->sample_count = st->nb_index_entries;

  1427. }

  1428. 

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

  1430. {

  1431.     MOVContext *mov = s->priv_data;

  1432.     ByteIOContext *pb = s->pb;

  1433.     int i, err;

  1434.     MOV_atom_t atom = { 0, 0, 0 };

  1435. 

  1436.     mov->fc = s;

  1437.     mov->parse_table = mov_default_parse_table;

  1438. 

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

  1440.         atom.size = url_fsize(pb);

  1441.     else

  1442.         atom.size = INT64_MAX;

  1443. 

  1444.     /* check MOV header */

  1445.     err = mov_read_default(mov, pb, atom);

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

  1447.         av_log(s, AV_LOG_ERROR, "mov: header not found !!! (err:%d, moov:%d, mdat:%d) pos:%"PRId64"\n",

  1448.                 err, mov->found_moov, mov->found_mdat, url_ftell(pb));

  1449.         return -1;

  1450.     }

  1451.     dprintf(mov->fc, "on_parse_exit_offset=%d\n", (int) url_ftell(pb));

  1452. 

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

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

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

  1456. 

  1457.     mov->total_streams = s->nb_streams;

  1458. 

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

  1460.         MOVStreamContext *sc = mov->streams[i];

  1461.         AVStream *st = s->streams[i];

  1462.         /* sanity checks */

  1463.         if(!sc->stts_count || !sc->chunk_count || !sc->sample_to_chunk_sz ||

  1464.            (!sc->sample_size && !sc->sample_count)){

  1465.             av_log(s, AV_LOG_ERROR, "missing mandatory atoms, broken header\n");

  1466.             sc->sample_count = 0; //ignore track

  1467.             continue;

  1468.         }

  1469.         if(!sc->time_rate)

  1470.             sc->time_rate=1;

  1471.         if(!sc->time_scale)

  1472.             sc->time_scale= mov->time_scale;

  1473.         av_set_pts_info(st, 64, sc->time_rate, sc->time_scale);

  1474. 

  1475.         if (st->codec->codec_type == CODEC_TYPE_AUDIO && sc->stts_count == 1)

  1476.             st->codec->frame_size = sc->stts_data[0].duration;

  1477. 

  1478.         if(st->duration != AV_NOPTS_VALUE){

  1479.             assert(st->duration % sc->time_rate == 0);

  1480.             st->duration /= sc->time_rate;

  1481.         }

  1482.         sc->ffindex = i;

  1483.         mov_build_index(mov, st);

  1484.     }

  1485. 

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

  1487.         /* Do not need those anymore. */

  1488.         av_freep(&mov->streams[i]->chunk_offsets);

  1489.         av_freep(&mov->streams[i]->sample_to_chunk);

  1490.         av_freep(&mov->streams[i]->sample_sizes);

  1491.         av_freep(&mov->streams[i]->keyframes);

  1492.         av_freep(&mov->streams[i]->stts_data);

  1493.     }

  1494.     av_freep(&mov->mdat_list);

  1495.     return 0;

  1496. }

  1497. 

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

  1499. {

  1500.     MOVContext *mov = s->priv_data;

  1501.     MOVStreamContext *sc = 0;

  1502.     AVIndexEntry *sample = 0;

  1503.     int64_t best_dts = INT64_MAX;

  1504.     int i;

  1505. 

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

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

  1508. 

  1509.         if (s->streams[i]->discard != AVDISCARD_ALL && msc->current_sample < msc->sample_count) {

  1510.             AVIndexEntry *current_sample = &s->streams[i]->index_entries[msc->current_sample];

  1511.             int64_t dts = av_rescale(current_sample->timestamp * (int64_t)msc->time_rate, AV_TIME_BASE, msc->time_scale);

  1512. 

  1513.             dprintf(s, "stream %d, sample %d, dts %"PRId64"\n", i, msc->current_sample, dts);

  1514.             if (dts < best_dts) {

  1515.                 sample = current_sample;

  1516.                 best_dts = dts;

  1517.                 sc = msc;

  1518.             }

  1519.         }

  1520.     }

  1521.     if (!sample)

  1522.         return -1;

  1523.     /* must be done just before reading, to avoid infinite loop on sample */

  1524.     sc->current_sample++;

  1525.     if (sample->pos >= url_fsize(s->pb)) {

  1526.         av_log(mov->fc, AV_LOG_ERROR, "stream %d, offset 0x%"PRIx64": partial file\n", sc->ffindex, sample->pos);

  1527.         return -1;

  1528.     }

  1529. #ifdef CONFIG_DV_DEMUXER

  1530.     if (sc->dv_audio_container) {

  1531.         dv_get_packet(mov->dv_demux, pkt);

  1532.         dprintf(s, "dv audio pkt size %d\n", pkt->size);

  1533.     } else {

  1534. #endif

  1535.         url_fseek(s->pb, sample->pos, SEEK_SET);

  1536.         av_get_packet(s->pb, pkt, sample->size);

  1537. #ifdef CONFIG_DV_DEMUXER

  1538.         if (mov->dv_demux) {

  1539.             void *pkt_destruct_func = pkt->destruct;

  1540.             dv_produce_packet(mov->dv_demux, pkt, pkt->data, pkt->size);

  1541.             pkt->destruct = pkt_destruct_func;

  1542.         }

  1543.     }

  1544. #endif

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

  1546.     pkt->dts = sample->timestamp;

  1547.     if (sc->ctts_data) {

  1548.         assert(sc->ctts_data[sc->sample_to_ctime_index].duration % sc->time_rate == 0);

  1549.         pkt->pts = pkt->dts + sc->ctts_data[sc->sample_to_ctime_index].duration / sc->time_rate;

  1550.         /* update ctts context */

  1551.         sc->sample_to_ctime_sample++;

  1552.         if (sc->sample_to_ctime_index < sc->ctts_count && sc->ctts_data[sc->sample_to_ctime_index].count == sc->sample_to_ctime_sample) {

  1553.             sc->sample_to_ctime_index++;

  1554.             sc->sample_to_ctime_sample = 0;

  1555.         }

  1556.     } else {

  1557.         pkt->pts = pkt->dts;

  1558.     }

  1559.     pkt->flags |= sample->flags & AVINDEX_KEYFRAME ? PKT_FLAG_KEY : 0;

  1560.     pkt->pos = sample->pos;

  1561.     dprintf(s, "stream %d, pts %"PRId64", dts %"PRId64", pos 0x%"PRIx64", duration %d\n", pkt->stream_index, pkt->pts, pkt->dts, pkt->pos, pkt->duration);

  1562.     return 0;

  1563. }

  1564. 

  1565. static int mov_seek_stream(AVStream *st, int64_t timestamp, int flags)

  1566. {

  1567.     MOVStreamContext *sc = st->priv_data;

  1568.     int sample, time_sample;

  1569.     int i;

  1570. 

  1571.     sample = av_index_search_timestamp(st, timestamp, flags);

  1572.     dprintf(st->codec, "stream %d, timestamp %"PRId64", sample %d\n", st->index, timestamp, sample);

  1573.     if (sample < 0) /* not sure what to do */

  1574.         return -1;

  1575.     sc->current_sample = sample;

  1576.     dprintf(st->codec, "stream %d, found sample %d\n", st->index, sc->current_sample);

  1577.     /* adjust ctts index */

  1578.     if (sc->ctts_data) {

  1579.         time_sample = 0;

  1580.         for (i = 0; i < sc->ctts_count; i++) {

  1581.             int next = time_sample + sc->ctts_data[i].count;

  1582.             if (next > sc->current_sample) {

  1583.                 sc->sample_to_ctime_index = i;

  1584.                 sc->sample_to_ctime_sample = sc->current_sample - time_sample;

  1585.                 break;

  1586.             }

  1587.             time_sample = next;

  1588.         }

  1589.     }

  1590.     return sample;

  1591. }

  1592. 

  1593. static int mov_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)

  1594. {

  1595.     AVStream *st;

  1596.     int64_t seek_timestamp, timestamp;

  1597.     int sample;

  1598.     int i;

  1599. 

  1600.     if (stream_index >= s->nb_streams)

  1601.         return -1;

  1602. 

  1603.     st = s->streams[stream_index];

  1604.     sample = mov_seek_stream(st, sample_time, flags);

  1605.     if (sample < 0)

  1606.         return -1;

  1607. 

  1608.     /* adjust seek timestamp to found sample timestamp */

  1609.     seek_timestamp = st->index_entries[sample].timestamp;

  1610. 

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

  1612.         st = s->streams[i];

  1613.         if (stream_index == i || st->discard == AVDISCARD_ALL)

  1614.             continue;

  1615. 

  1616.         timestamp = av_rescale_q(seek_timestamp, s->streams[stream_index]->time_base, st->time_base);

  1617.         mov_seek_stream(st, timestamp, flags);

  1618.     }

  1619.     return 0;

  1620. }

  1621. 

  1622. static int mov_read_close(AVFormatContext *s)

  1623. {

  1624.     int i;

  1625.     MOVContext *mov = s->priv_data;

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

  1627.         av_freep(&mov->streams[i]->ctts_data);

  1628.         av_freep(&mov->streams[i]);

  1629.     }

  1630.     if(mov->dv_demux){

  1631.         for(i=0; i<mov->dv_fctx->nb_streams; i++){

  1632.             av_freep(&mov->dv_fctx->streams[i]->codec);

  1633.             av_freep(&mov->dv_fctx->streams[i]);

  1634.         }

  1635.         av_freep(&mov->dv_fctx);

  1636.         av_freep(&mov->dv_demux);

  1637.     }

  1638.     return 0;

  1639. }

  1640. 

  1641. AVInputFormat mov_demuxer = {

  1642.     "mov,mp4,m4a,3gp,3g2,mj2",

  1643.     "QuickTime/MPEG4/Motion JPEG 2000 format",

  1644.     sizeof(MOVContext),

  1645.     mov_probe,

  1646.     mov_read_header,

  1647.     mov_read_packet,

  1648.     mov_read_close,

  1649.     mov_read_seek,

  1650. };