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.

1660 lines
49KB

  1. /*
  2. * MOV, 3GP, MP4 encoder.
  3. * Copyright (c) 2003 Thomas Raivio.
  4. * Copyright (c) 2004 Gildas Bazin <gbazin at videolan dot org>.
  5. *
  6. * This library is free software; you can redistribute it and/or
  7. * modify it under the terms of the GNU Lesser General Public
  8. * License as published by the Free Software Foundation; either
  9. * version 2 of the License, or (at your option) any later version.
  10. *
  11. * This library is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14. * Lesser General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU Lesser General Public
  17. * License along with this library; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. #include "avformat.h"
  21. #include "avi.h"
  22. #include "avio.h"
  23. #include "mov.h"
  24. #undef NDEBUG
  25. #include <assert.h>
  26. #define MOV_INDEX_CLUSTER_SIZE 16384
  27. #define globalTimescale 1000
  28. #define MODE_MP4 0
  29. #define MODE_MOV 1
  30. #define MODE_3GP 2
  31. #define MODE_PSP 3 // example working PSP command line:
  32. // ffmpeg -i testinput.avi -f psp -r 14.985 -s 320x240 -b 768 -ar 24000 -ab 32 M4V00001.MP4
  33. #define MODE_3G2 4
  34. typedef struct MOVIentry {
  35. unsigned int flags, size;
  36. uint64_t pos;
  37. unsigned int samplesInChunk;
  38. char key_frame;
  39. unsigned int entries;
  40. int64_t cts;
  41. } MOVIentry;
  42. typedef struct MOVIndex {
  43. int mode;
  44. int entry;
  45. uint64_t mdat_size;
  46. int ents_allocated;
  47. long timescale;
  48. long time;
  49. long trackDuration;
  50. long sampleCount;
  51. long sampleDuration;
  52. int hasKeyframes;
  53. int hasBframes;
  54. int language;
  55. int trackID;
  56. AVCodecContext *enc;
  57. int vosLen;
  58. uint8_t *vosData;
  59. MOVIentry** cluster;
  60. } MOVTrack;
  61. typedef struct MOVContext {
  62. int mode;
  63. long time;
  64. int nb_streams;
  65. int mdat_written;
  66. offset_t mdat_pos;
  67. long timescale;
  68. MOVTrack tracks[MAX_STREAMS];
  69. } MOVContext;
  70. static int mov_write_esds_tag(ByteIOContext *pb, MOVTrack* track);
  71. /* output language code from iso639 language name */
  72. extern int ff_mov_iso639_to_lang(const char *lang, int mp4);
  73. //FIXME supprt 64bit varaint with wide placeholders
  74. static offset_t updateSize (ByteIOContext *pb, offset_t pos)
  75. {
  76. offset_t curpos = url_ftell(pb);
  77. url_fseek(pb, pos, SEEK_SET);
  78. put_be32(pb, curpos - pos); /* rewrite size */
  79. url_fseek(pb, curpos, SEEK_SET);
  80. return curpos - pos;
  81. }
  82. /* Chunk offset atom */
  83. static int mov_write_stco_tag(ByteIOContext *pb, MOVTrack* track)
  84. {
  85. int i;
  86. int mode64 = 0; // use 32 bit size variant if possible
  87. offset_t pos = url_ftell(pb);
  88. put_be32(pb, 0); /* size */
  89. if (pos > UINT32_MAX) {
  90. mode64 = 1;
  91. put_tag(pb, "co64");
  92. } else
  93. put_tag(pb, "stco");
  94. put_be32(pb, 0); /* version & flags */
  95. put_be32(pb, track->entry); /* entry count */
  96. for (i=0; i<track->entry; i++) {
  97. int cl = i / MOV_INDEX_CLUSTER_SIZE;
  98. int id = i % MOV_INDEX_CLUSTER_SIZE;
  99. if(mode64 == 1)
  100. put_be64(pb, track->cluster[cl][id].pos);
  101. else
  102. put_be32(pb, track->cluster[cl][id].pos);
  103. }
  104. return updateSize (pb, pos);
  105. }
  106. /* Sample size atom */
  107. static int mov_write_stsz_tag(ByteIOContext *pb, MOVTrack* track)
  108. {
  109. int equalChunks = 1;
  110. int i, j, entries = 0, tst = -1, oldtst = -1;
  111. offset_t pos = url_ftell(pb);
  112. put_be32(pb, 0); /* size */
  113. put_tag(pb, "stsz");
  114. put_be32(pb, 0); /* version & flags */
  115. for (i=0; i<track->entry; i++) {
  116. int cl = i / MOV_INDEX_CLUSTER_SIZE;
  117. int id = i % MOV_INDEX_CLUSTER_SIZE;
  118. tst = track->cluster[cl][id].size/track->cluster[cl][id].entries;
  119. if(oldtst != -1 && tst != oldtst) {
  120. equalChunks = 0;
  121. }
  122. oldtst = tst;
  123. entries += track->cluster[cl][id].entries;
  124. }
  125. if (equalChunks) {
  126. int sSize = track->cluster[0][0].size/track->cluster[0][0].entries;
  127. put_be32(pb, sSize); // sample size
  128. put_be32(pb, entries); // sample count
  129. }
  130. else {
  131. put_be32(pb, 0); // sample size
  132. put_be32(pb, entries); // sample count
  133. for (i=0; i<track->entry; i++) {
  134. int cl = i / MOV_INDEX_CLUSTER_SIZE;
  135. int id = i % MOV_INDEX_CLUSTER_SIZE;
  136. for ( j=0; j<track->cluster[cl][id].entries; j++) {
  137. put_be32(pb, track->cluster[cl][id].size /
  138. track->cluster[cl][id].entries);
  139. }
  140. }
  141. }
  142. return updateSize (pb, pos);
  143. }
  144. /* Sample to chunk atom */
  145. static int mov_write_stsc_tag(ByteIOContext *pb, MOVTrack* track)
  146. {
  147. int index = 0, oldval = -1, i;
  148. offset_t entryPos, curpos;
  149. offset_t pos = url_ftell(pb);
  150. put_be32(pb, 0); /* size */
  151. put_tag(pb, "stsc");
  152. put_be32(pb, 0); // version & flags
  153. entryPos = url_ftell(pb);
  154. put_be32(pb, track->entry); // entry count
  155. for (i=0; i<track->entry; i++) {
  156. int cl = i / MOV_INDEX_CLUSTER_SIZE;
  157. int id = i % MOV_INDEX_CLUSTER_SIZE;
  158. if(oldval != track->cluster[cl][id].samplesInChunk)
  159. {
  160. put_be32(pb, i+1); // first chunk
  161. put_be32(pb, track->cluster[cl][id].samplesInChunk); // samples per chunk
  162. put_be32(pb, 0x1); // sample description index
  163. oldval = track->cluster[cl][id].samplesInChunk;
  164. index++;
  165. }
  166. }
  167. curpos = url_ftell(pb);
  168. url_fseek(pb, entryPos, SEEK_SET);
  169. put_be32(pb, index); // rewrite size
  170. url_fseek(pb, curpos, SEEK_SET);
  171. return updateSize (pb, pos);
  172. }
  173. /* Sync sample atom */
  174. static int mov_write_stss_tag(ByteIOContext *pb, MOVTrack* track)
  175. {
  176. offset_t curpos, entryPos;
  177. int i, index = 0;
  178. offset_t pos = url_ftell(pb);
  179. put_be32(pb, 0); // size
  180. put_tag(pb, "stss");
  181. put_be32(pb, 0); // version & flags
  182. entryPos = url_ftell(pb);
  183. put_be32(pb, track->entry); // entry count
  184. for (i=0; i<track->entry; i++) {
  185. int cl = i / MOV_INDEX_CLUSTER_SIZE;
  186. int id = i % MOV_INDEX_CLUSTER_SIZE;
  187. if(track->cluster[cl][id].key_frame == 1) {
  188. put_be32(pb, i+1);
  189. index++;
  190. }
  191. }
  192. curpos = url_ftell(pb);
  193. url_fseek(pb, entryPos, SEEK_SET);
  194. put_be32(pb, index); // rewrite size
  195. url_fseek(pb, curpos, SEEK_SET);
  196. return updateSize (pb, pos);
  197. }
  198. static int mov_write_damr_tag(ByteIOContext *pb)
  199. {
  200. put_be32(pb, 0x11); /* size */
  201. put_tag(pb, "damr");
  202. put_tag(pb, "FFMP");
  203. put_byte(pb, 0);
  204. put_be16(pb, 0x80); /* Mode set (all modes for AMR_NB) */
  205. put_be16(pb, 0xa); /* Mode change period (no restriction) */
  206. //put_be16(pb, 0x81ff); /* Mode set (all modes for AMR_NB) */
  207. //put_be16(pb, 1); /* Mode change period (no restriction) */
  208. return 0x11;
  209. }
  210. static int mov_write_wave_tag(ByteIOContext *pb, MOVTrack* track)
  211. {
  212. offset_t pos = url_ftell(pb);
  213. put_be32(pb, 0); /* size */
  214. put_tag(pb, "wave");
  215. put_be32(pb, 12); /* size */
  216. put_tag(pb, "frma");
  217. put_tag(pb, "mp4a");
  218. put_be32(pb, 12); /* size */
  219. put_tag(pb, "mp4a");
  220. put_be32(pb, 0);
  221. mov_write_esds_tag(pb, track);
  222. put_be32(pb, 12); /* size */
  223. put_tag(pb, "srcq");
  224. put_be32(pb, 0x40);
  225. put_be32(pb, 8); /* size */
  226. put_be32(pb, 0); /* null tag */
  227. return updateSize (pb, pos);
  228. }
  229. static const CodecTag codec_movaudio_tags[] = {
  230. { CODEC_ID_PCM_MULAW, MKTAG('u', 'l', 'a', 'w') },
  231. { CODEC_ID_PCM_ALAW, MKTAG('a', 'l', 'a', 'w') },
  232. { CODEC_ID_ADPCM_IMA_QT, MKTAG('i', 'm', 'a', '4') },
  233. { CODEC_ID_MACE3, MKTAG('M', 'A', 'C', '3') },
  234. { CODEC_ID_MACE6, MKTAG('M', 'A', 'C', '6') },
  235. { CODEC_ID_AAC, MKTAG('m', 'p', '4', 'a') },
  236. { CODEC_ID_AMR_NB, MKTAG('s', 'a', 'm', 'r') },
  237. { CODEC_ID_AMR_WB, MKTAG('s', 'a', 'w', 'b') },
  238. { CODEC_ID_PCM_S16BE, MKTAG('t', 'w', 'o', 's') },
  239. { CODEC_ID_PCM_S16LE, MKTAG('s', 'o', 'w', 't') },
  240. { CODEC_ID_MP3, MKTAG('.', 'm', 'p', '3') },
  241. { 0, 0 },
  242. };
  243. static int mov_write_audio_tag(ByteIOContext *pb, MOVTrack* track)
  244. {
  245. offset_t pos = url_ftell(pb);
  246. int tag;
  247. put_be32(pb, 0); /* size */
  248. tag = track->enc->codec_tag;
  249. if (!tag)
  250. tag = codec_get_tag(codec_movaudio_tags, track->enc->codec_id);
  251. // if no mac fcc found, try with Microsoft tags
  252. if (!tag)
  253. {
  254. int tmp = codec_get_tag(codec_wav_tags, track->enc->codec_id);
  255. tag = MKTAG('m', 's', ((tmp >> 8) & 0xff), (tmp & 0xff));
  256. }
  257. put_le32(pb, tag); // store it byteswapped
  258. put_be32(pb, 0); /* Reserved */
  259. put_be16(pb, 0); /* Reserved */
  260. put_be16(pb, 1); /* Data-reference index, XXX == 1 */
  261. /* SoundDescription */
  262. if(track->mode == MODE_MOV && track->enc->codec_id == CODEC_ID_AAC)
  263. put_be16(pb, 1); /* Version 1 */
  264. else
  265. put_be16(pb, 0); /* Version 0 */
  266. put_be16(pb, 0); /* Revision level */
  267. put_be32(pb, 0); /* Reserved */
  268. put_be16(pb, track->enc->channels); /* Number of channels */
  269. /* TODO: Currently hard-coded to 16-bit, there doesn't seem
  270. to be a good way to get number of bits of audio */
  271. put_be16(pb, 0x10); /* Reserved */
  272. if(track->enc->codec_id == CODEC_ID_AAC ||
  273. track->enc->codec_id == CODEC_ID_MP3)
  274. {
  275. put_be16(pb, 0xfffe); /* compression ID (vbr)*/
  276. }
  277. else
  278. {
  279. put_be16(pb, 0); /* compression ID (= 0) */
  280. }
  281. put_be16(pb, 0); /* packet size (= 0) */
  282. put_be16(pb, track->timescale); /* Time scale */
  283. put_be16(pb, 0); /* Reserved */
  284. if(track->mode == MODE_MOV && track->enc->codec_id == CODEC_ID_AAC)
  285. {
  286. /* SoundDescription V1 extended info */
  287. put_be32(pb, track->enc->frame_size); /* Samples per packet */
  288. put_be32(pb, 1536); /* Bytes per packet */
  289. put_be32(pb, 2); /* Bytes per frame */
  290. put_be32(pb, 2); /* Bytes per sample */
  291. }
  292. if(track->enc->codec_id == CODEC_ID_AAC) {
  293. if( track->mode == MODE_MOV ) mov_write_wave_tag(pb, track);
  294. else mov_write_esds_tag(pb, track);
  295. }
  296. if(track->enc->codec_id == CODEC_ID_AMR_NB)
  297. mov_write_damr_tag(pb);
  298. return updateSize (pb, pos);
  299. }
  300. static int mov_write_d263_tag(ByteIOContext *pb)
  301. {
  302. put_be32(pb, 0xf); /* size */
  303. put_tag(pb, "d263");
  304. put_tag(pb, "FFMP");
  305. put_be16(pb, 0x0a);
  306. put_byte(pb, 0);
  307. return 0xf;
  308. }
  309. /* TODO: No idea about these values */
  310. static int mov_write_svq3_tag(ByteIOContext *pb)
  311. {
  312. put_be32(pb, 0x15);
  313. put_tag(pb, "SMI ");
  314. put_tag(pb, "SEQH");
  315. put_be32(pb, 0x5);
  316. put_be32(pb, 0xe2c0211d);
  317. put_be32(pb, 0xc0000000);
  318. put_byte(pb, 0);
  319. return 0x15;
  320. }
  321. static unsigned int descrLength(unsigned int len)
  322. {
  323. if (len < 0x00000080)
  324. return 2 + len;
  325. else if (len < 0x00004000)
  326. return 3 + len;
  327. else if(len < 0x00200000)
  328. return 4 + len;
  329. else
  330. return 5 + len;
  331. }
  332. static void putDescr(ByteIOContext *pb, int tag, int size)
  333. {
  334. uint32_t len;
  335. uint8_t vals[4];
  336. len = size;
  337. vals[3] = (uint8_t)(len & 0x7f);
  338. len >>= 7;
  339. vals[2] = (uint8_t)((len & 0x7f) | 0x80);
  340. len >>= 7;
  341. vals[1] = (uint8_t)((len & 0x7f) | 0x80);
  342. len >>= 7;
  343. vals[0] = (uint8_t)((len & 0x7f) | 0x80);
  344. put_byte(pb, tag); // DescriptorTag
  345. if (size < 0x00000080)
  346. {
  347. put_byte(pb, vals[3]);
  348. }
  349. else if (size < 0x00004000)
  350. {
  351. put_byte(pb, vals[2]);
  352. put_byte(pb, vals[3]);
  353. }
  354. else if (size < 0x00200000)
  355. {
  356. put_byte(pb, vals[1]);
  357. put_byte(pb, vals[2]);
  358. put_byte(pb, vals[3]);
  359. }
  360. else if (size < 0x10000000)
  361. {
  362. put_byte(pb, vals[0]);
  363. put_byte(pb, vals[1]);
  364. put_byte(pb, vals[2]);
  365. put_byte(pb, vals[3]);
  366. }
  367. }
  368. static int mov_write_esds_tag(ByteIOContext *pb, MOVTrack* track) // Basic
  369. {
  370. int decoderSpecificInfoLen;
  371. offset_t pos = url_ftell(pb);
  372. decoderSpecificInfoLen = track->vosLen ? descrLength(track->vosLen):0;
  373. put_be32(pb, 0); // size
  374. put_tag(pb, "esds");
  375. put_be32(pb, 0); // Version
  376. // ES descriptor
  377. putDescr(pb, 0x03, 3 + descrLength(13 + decoderSpecificInfoLen) +
  378. descrLength(1));
  379. put_be16(pb, track->trackID);
  380. put_byte(pb, 0x00); // flags (= no flags)
  381. // DecoderConfig descriptor
  382. putDescr(pb, 0x04, 13 + decoderSpecificInfoLen);
  383. // Object type indication
  384. put_byte(pb, codec_get_tag(ff_mov_obj_type, track->enc->codec_id));
  385. // the following fields is made of 6 bits to identify the streamtype (4 for video, 5 for audio)
  386. // plus 1 bit to indicate upstream and 1 bit set to 1 (reserved)
  387. if(track->enc->codec_type == CODEC_TYPE_AUDIO)
  388. put_byte(pb, 0x15); // flags (= Audiostream)
  389. else
  390. put_byte(pb, 0x11); // flags (= Visualstream)
  391. put_byte(pb, track->enc->rc_buffer_size>>(3+16)); // Buffersize DB (24 bits)
  392. put_be16(pb, (track->enc->rc_buffer_size>>3)&0xFFFF); // Buffersize DB
  393. put_be32(pb, FFMAX(track->enc->bit_rate, track->enc->rc_max_rate)); // maxbitrate (FIXME should be max rate in any 1 sec window)
  394. if(track->enc->rc_max_rate != track->enc->rc_min_rate || track->enc->rc_min_rate==0)
  395. put_be32(pb, 0); // vbr
  396. else
  397. put_be32(pb, track->enc->rc_max_rate); // avg bitrate
  398. if (track->vosLen)
  399. {
  400. // DecoderSpecific info descriptor
  401. putDescr(pb, 0x05, track->vosLen);
  402. put_buffer(pb, track->vosData, track->vosLen);
  403. }
  404. // SL descriptor
  405. putDescr(pb, 0x06, 1);
  406. put_byte(pb, 0x02);
  407. return updateSize (pb, pos);
  408. }
  409. static const CodecTag codec_movvideo_tags[] = {
  410. { CODEC_ID_SVQ1, MKTAG('S', 'V', 'Q', '1') },
  411. { CODEC_ID_SVQ3, MKTAG('S', 'V', 'Q', '3') },
  412. { CODEC_ID_MPEG4, MKTAG('m', 'p', '4', 'v') },
  413. { CODEC_ID_H263, MKTAG('s', '2', '6', '3') },
  414. { CODEC_ID_H264, MKTAG('a', 'v', 'c', '1') },
  415. { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'c', ' ') },
  416. { 0, 0 },
  417. };
  418. static int mov_write_video_tag(ByteIOContext *pb, MOVTrack* track)
  419. {
  420. offset_t pos = url_ftell(pb);
  421. char compressor_name[32];
  422. int tag;
  423. put_be32(pb, 0); /* size */
  424. tag = track->enc->codec_tag;
  425. if (!tag)
  426. tag = codec_get_tag(codec_movvideo_tags, track->enc->codec_id);
  427. // if no mac fcc found, try with Microsoft tags
  428. if (!tag)
  429. tag = codec_get_tag(codec_bmp_tags, track->enc->codec_id);
  430. put_le32(pb, tag); // store it byteswapped
  431. put_be32(pb, 0); /* Reserved */
  432. put_be16(pb, 0); /* Reserved */
  433. put_be16(pb, 1); /* Data-reference index */
  434. put_be16(pb, 0); /* Codec stream version */
  435. put_be16(pb, 0); /* Codec stream revision (=0) */
  436. put_tag(pb, "FFMP"); /* Vendor */
  437. if(track->enc->codec_id == CODEC_ID_RAWVIDEO) {
  438. put_be32(pb, 0); /* Temporal Quality */
  439. put_be32(pb, 0x400); /* Spatial Quality = lossless*/
  440. } else {
  441. put_be32(pb, 0x200); /* Temporal Quality = normal */
  442. put_be32(pb, 0x200); /* Spatial Quality = normal */
  443. }
  444. put_be16(pb, track->enc->width); /* Video width */
  445. put_be16(pb, track->enc->height); /* Video height */
  446. put_be32(pb, 0x00480000); /* Horizontal resolution 72dpi */
  447. put_be32(pb, 0x00480000); /* Vertical resolution 72dpi */
  448. put_be32(pb, 0); /* Data size (= 0) */
  449. put_be16(pb, 1); /* Frame count (= 1) */
  450. memset(compressor_name,0,32);
  451. if (track->enc->codec && track->enc->codec->name)
  452. strncpy(compressor_name,track->enc->codec->name,31);
  453. put_byte(pb, strlen(compressor_name));
  454. put_buffer(pb, compressor_name, 31);
  455. put_be16(pb, 0x18); /* Reserved */
  456. put_be16(pb, 0xffff); /* Reserved */
  457. if(track->enc->codec_id == CODEC_ID_MPEG4)
  458. mov_write_esds_tag(pb, track);
  459. else if(track->enc->codec_id == CODEC_ID_H263)
  460. mov_write_d263_tag(pb);
  461. else if(track->enc->codec_id == CODEC_ID_SVQ3)
  462. mov_write_svq3_tag(pb);
  463. return updateSize (pb, pos);
  464. }
  465. static int mov_write_stsd_tag(ByteIOContext *pb, MOVTrack* track)
  466. {
  467. offset_t pos = url_ftell(pb);
  468. put_be32(pb, 0); /* size */
  469. put_tag(pb, "stsd");
  470. put_be32(pb, 0); /* version & flags */
  471. put_be32(pb, 1); /* entry count */
  472. if (track->enc->codec_type == CODEC_TYPE_VIDEO)
  473. mov_write_video_tag(pb, track);
  474. else if (track->enc->codec_type == CODEC_TYPE_AUDIO)
  475. mov_write_audio_tag(pb, track);
  476. return updateSize(pb, pos);
  477. }
  478. static int mov_write_ctts_tag(ByteIOContext *pb, MOVTrack* track)
  479. {
  480. Time2Sample *ctts_entries;
  481. uint32_t entries = 0;
  482. uint32_t atom_size;
  483. int i;
  484. ctts_entries = av_malloc((track->entry + 1) * sizeof(*ctts_entries)); /* worst case */
  485. ctts_entries[0].count = 1;
  486. ctts_entries[0].duration = track->cluster[0][0].cts;
  487. for (i=1; i<track->entry; i++) {
  488. int cl = i / MOV_INDEX_CLUSTER_SIZE;
  489. int id = i % MOV_INDEX_CLUSTER_SIZE;
  490. if (track->cluster[cl][id].cts == ctts_entries[entries].duration) {
  491. ctts_entries[entries].count++; /* compress */
  492. } else {
  493. entries++;
  494. ctts_entries[entries].duration = track->cluster[cl][id].cts;
  495. ctts_entries[entries].count = 1;
  496. }
  497. }
  498. entries++; /* last one */
  499. atom_size = 16 + (entries * 8);
  500. put_be32(pb, atom_size); /* size */
  501. put_tag(pb, "ctts");
  502. put_be32(pb, 0); /* version & flags */
  503. put_be32(pb, entries); /* entry count */
  504. for (i=0; i<entries; i++) {
  505. put_be32(pb, ctts_entries[i].count);
  506. put_be32(pb, ctts_entries[i].duration);
  507. }
  508. av_free(ctts_entries);
  509. return atom_size;
  510. }
  511. /* TODO: */
  512. /* Time to sample atom */
  513. static int mov_write_stts_tag(ByteIOContext *pb, MOVTrack* track)
  514. {
  515. put_be32(pb, 0x18); /* size */
  516. put_tag(pb, "stts");
  517. put_be32(pb, 0); /* version & flags */
  518. put_be32(pb, 1); /* entry count */
  519. put_be32(pb, track->sampleCount); /* sample count */
  520. put_be32(pb, track->sampleDuration); /* sample duration */
  521. return 0x18;
  522. }
  523. static int mov_write_dref_tag(ByteIOContext *pb)
  524. {
  525. put_be32(pb, 28); /* size */
  526. put_tag(pb, "dref");
  527. put_be32(pb, 0); /* version & flags */
  528. put_be32(pb, 1); /* entry count */
  529. put_be32(pb, 0xc); /* size */
  530. put_tag(pb, "url ");
  531. put_be32(pb, 1); /* version & flags */
  532. return 28;
  533. }
  534. static int mov_write_stbl_tag(ByteIOContext *pb, MOVTrack* track)
  535. {
  536. offset_t pos = url_ftell(pb);
  537. put_be32(pb, 0); /* size */
  538. put_tag(pb, "stbl");
  539. mov_write_stsd_tag(pb, track);
  540. mov_write_stts_tag(pb, track);
  541. if (track->enc->codec_type == CODEC_TYPE_VIDEO &&
  542. track->hasKeyframes)
  543. mov_write_stss_tag(pb, track);
  544. if (track->enc->codec_type == CODEC_TYPE_VIDEO &&
  545. track->hasBframes)
  546. mov_write_ctts_tag(pb, track);
  547. mov_write_stsc_tag(pb, track);
  548. mov_write_stsz_tag(pb, track);
  549. mov_write_stco_tag(pb, track);
  550. return updateSize(pb, pos);
  551. }
  552. static int mov_write_dinf_tag(ByteIOContext *pb)
  553. {
  554. offset_t pos = url_ftell(pb);
  555. put_be32(pb, 0); /* size */
  556. put_tag(pb, "dinf");
  557. mov_write_dref_tag(pb);
  558. return updateSize(pb, pos);
  559. }
  560. static int mov_write_smhd_tag(ByteIOContext *pb)
  561. {
  562. put_be32(pb, 16); /* size */
  563. put_tag(pb, "smhd");
  564. put_be32(pb, 0); /* version & flags */
  565. put_be16(pb, 0); /* reserved (balance, normally = 0) */
  566. put_be16(pb, 0); /* reserved */
  567. return 16;
  568. }
  569. static int mov_write_vmhd_tag(ByteIOContext *pb)
  570. {
  571. put_be32(pb, 0x14); /* size (always 0x14) */
  572. put_tag(pb, "vmhd");
  573. put_be32(pb, 0x01); /* version & flags */
  574. put_be64(pb, 0); /* reserved (graphics mode = copy) */
  575. return 0x14;
  576. }
  577. static int mov_write_hdlr_tag(ByteIOContext *pb, MOVTrack* track)
  578. {
  579. char *descr, *hdlr, *hdlr_type;
  580. offset_t pos = url_ftell(pb);
  581. if (!track) { /* no media --> data handler */
  582. hdlr = "dhlr";
  583. hdlr_type = "url ";
  584. descr = "DataHandler";
  585. } else {
  586. hdlr = (track->mode == MODE_MOV) ? "mhlr" : "\0\0\0\0";
  587. if (track->enc->codec_type == CODEC_TYPE_VIDEO) {
  588. hdlr_type = "vide";
  589. descr = "VideoHandler";
  590. } else {
  591. hdlr_type = "soun";
  592. descr = "SoundHandler";
  593. }
  594. }
  595. put_be32(pb, 0); /* size */
  596. put_tag(pb, "hdlr");
  597. put_be32(pb, 0); /* Version & flags */
  598. put_buffer(pb, hdlr, 4); /* handler */
  599. put_tag(pb, hdlr_type); /* handler type */
  600. put_be32(pb ,0); /* reserved */
  601. put_be32(pb ,0); /* reserved */
  602. put_be32(pb ,0); /* reserved */
  603. put_byte(pb, strlen(descr)); /* string counter */
  604. put_buffer(pb, descr, strlen(descr)); /* handler description */
  605. return updateSize(pb, pos);
  606. }
  607. static int mov_write_minf_tag(ByteIOContext *pb, MOVTrack* track)
  608. {
  609. offset_t pos = url_ftell(pb);
  610. put_be32(pb, 0); /* size */
  611. put_tag(pb, "minf");
  612. if(track->enc->codec_type == CODEC_TYPE_VIDEO)
  613. mov_write_vmhd_tag(pb);
  614. else
  615. mov_write_smhd_tag(pb);
  616. if (track->mode == MODE_MOV) /* FIXME: Why do it for MODE_MOV only ? */
  617. mov_write_hdlr_tag(pb, NULL);
  618. mov_write_dinf_tag(pb);
  619. mov_write_stbl_tag(pb, track);
  620. return updateSize(pb, pos);
  621. }
  622. static int mov_write_mdhd_tag(ByteIOContext *pb, MOVTrack* track)
  623. {
  624. put_be32(pb, 32); /* size */
  625. put_tag(pb, "mdhd");
  626. put_be32(pb, 0); /* Version & flags */
  627. put_be32(pb, track->time); /* creation time */
  628. put_be32(pb, track->time); /* modification time */
  629. put_be32(pb, track->timescale); /* time scale (sample rate for audio) */
  630. put_be32(pb, track->trackDuration); /* duration */
  631. put_be16(pb, track->language); /* language */
  632. put_be16(pb, 0); /* reserved (quality) */
  633. return 32;
  634. }
  635. static int mov_write_mdia_tag(ByteIOContext *pb, MOVTrack* track)
  636. {
  637. offset_t pos = url_ftell(pb);
  638. put_be32(pb, 0); /* size */
  639. put_tag(pb, "mdia");
  640. mov_write_mdhd_tag(pb, track);
  641. mov_write_hdlr_tag(pb, track);
  642. mov_write_minf_tag(pb, track);
  643. return updateSize(pb, pos);
  644. }
  645. static int mov_write_tkhd_tag(ByteIOContext *pb, MOVTrack* track)
  646. {
  647. put_be32(pb, 0x5c); /* size (always 0x5c) */
  648. put_tag(pb, "tkhd");
  649. put_be32(pb, 0xf); /* version & flags (track enabled) */
  650. put_be32(pb, track->time); /* creation time */
  651. put_be32(pb, track->time); /* modification time */
  652. put_be32(pb, track->trackID); /* track-id */
  653. put_be32(pb, 0); /* reserved */
  654. put_be32(pb, av_rescale_rnd(track->trackDuration, globalTimescale, track->timescale, AV_ROUND_UP)); /* duration */
  655. put_be32(pb, 0); /* reserved */
  656. put_be32(pb, 0); /* reserved */
  657. put_be32(pb, 0x0); /* reserved (Layer & Alternate group) */
  658. /* Volume, only for audio */
  659. if(track->enc->codec_type == CODEC_TYPE_AUDIO)
  660. put_be16(pb, 0x0100);
  661. else
  662. put_be16(pb, 0);
  663. put_be16(pb, 0); /* reserved */
  664. /* Matrix structure */
  665. put_be32(pb, 0x00010000); /* reserved */
  666. put_be32(pb, 0x0); /* reserved */
  667. put_be32(pb, 0x0); /* reserved */
  668. put_be32(pb, 0x0); /* reserved */
  669. put_be32(pb, 0x00010000); /* reserved */
  670. put_be32(pb, 0x0); /* reserved */
  671. put_be32(pb, 0x0); /* reserved */
  672. put_be32(pb, 0x0); /* reserved */
  673. put_be32(pb, 0x40000000); /* reserved */
  674. /* Track width and height, for visual only */
  675. if(track->enc->codec_type == CODEC_TYPE_VIDEO) {
  676. double sample_aspect_ratio = av_q2d(track->enc->sample_aspect_ratio);
  677. if( !sample_aspect_ratio ) sample_aspect_ratio = 1;
  678. put_be32(pb, sample_aspect_ratio * track->enc->width*0x10000);
  679. put_be32(pb, track->enc->height*0x10000);
  680. }
  681. else {
  682. put_be32(pb, 0);
  683. put_be32(pb, 0);
  684. }
  685. return 0x5c;
  686. }
  687. // This box seems important for the psp playback ... without it the movie seems to hang
  688. static int mov_write_edts_tag(ByteIOContext *pb, MOVTrack *track)
  689. {
  690. put_be32(pb, 0x24); /* size */
  691. put_tag(pb, "edts");
  692. put_be32(pb, 0x1c); /* size */
  693. put_tag(pb, "elst");
  694. put_be32(pb, 0x0);
  695. put_be32(pb, 0x1);
  696. put_be32(pb, av_rescale_rnd(track->trackDuration, globalTimescale, track->timescale, AV_ROUND_UP)); /* duration ... doesn't seem to effect psp */
  697. put_be32(pb, 0x0);
  698. put_be32(pb, 0x00010000);
  699. return 0x24;
  700. }
  701. // goes at the end of each track! ... Critical for PSP playback ("Incompatible data" without it)
  702. static int mov_write_uuid_tag_psp(ByteIOContext *pb, MOVTrack *mov)
  703. {
  704. put_be32(pb, 0x34); /* size ... reports as 28 in mp4box! */
  705. put_tag(pb, "uuid");
  706. put_tag(pb, "USMT");
  707. put_be32(pb, 0x21d24fce);
  708. put_be32(pb, 0xbb88695c);
  709. put_be32(pb, 0xfac9c740);
  710. put_be32(pb, 0x1c); // another size here!
  711. put_tag(pb, "MTDT");
  712. put_be32(pb, 0x00010012);
  713. put_be32(pb, 0x0a);
  714. put_be32(pb, 0x55c40000);
  715. put_be32(pb, 0x1);
  716. put_be32(pb, 0x0);
  717. return 0x34;
  718. }
  719. static int mov_write_trak_tag(ByteIOContext *pb, MOVTrack* track)
  720. {
  721. offset_t pos = url_ftell(pb);
  722. put_be32(pb, 0); /* size */
  723. put_tag(pb, "trak");
  724. mov_write_tkhd_tag(pb, track);
  725. if (track->mode == MODE_PSP)
  726. mov_write_edts_tag(pb, track); // PSP Movies require edts box
  727. mov_write_mdia_tag(pb, track);
  728. if (track->mode == MODE_PSP)
  729. mov_write_uuid_tag_psp(pb,track); // PSP Movies require this uuid box
  730. return updateSize(pb, pos);
  731. }
  732. #if 0
  733. /* TODO: Not sorted out, but not necessary either */
  734. static int mov_write_iods_tag(ByteIOContext *pb, MOVContext *mov)
  735. {
  736. put_be32(pb, 0x15); /* size */
  737. put_tag(pb, "iods");
  738. put_be32(pb, 0); /* version & flags */
  739. put_be16(pb, 0x1007);
  740. put_byte(pb, 0);
  741. put_be16(pb, 0x4fff);
  742. put_be16(pb, 0xfffe);
  743. put_be16(pb, 0x01ff);
  744. return 0x15;
  745. }
  746. #endif
  747. static int mov_write_mvhd_tag(ByteIOContext *pb, MOVContext *mov)
  748. {
  749. int maxTrackID = 1, i;
  750. int64_t maxTrackLenTemp, maxTrackLen = 0;
  751. put_be32(pb, 0x6c); /* size (always 0x6c) */
  752. put_tag(pb, "mvhd");
  753. put_be32(pb, 0); /* version & flags */
  754. put_be32(pb, mov->time); /* creation time */
  755. put_be32(pb, mov->time); /* modification time */
  756. put_be32(pb, mov->timescale); /* timescale */
  757. for (i=0; i<MAX_STREAMS; i++) {
  758. if(mov->tracks[i].entry > 0) {
  759. maxTrackLenTemp = av_rescale_rnd(mov->tracks[i].trackDuration, globalTimescale, mov->tracks[i].timescale, AV_ROUND_UP);
  760. if(maxTrackLen < maxTrackLenTemp)
  761. maxTrackLen = maxTrackLenTemp;
  762. if(maxTrackID < mov->tracks[i].trackID)
  763. maxTrackID = mov->tracks[i].trackID;
  764. }
  765. }
  766. put_be32(pb, maxTrackLen); /* duration of longest track */
  767. put_be32(pb, 0x00010000); /* reserved (preferred rate) 1.0 = normal */
  768. put_be16(pb, 0x0100); /* reserved (preferred volume) 1.0 = normal */
  769. put_be16(pb, 0); /* reserved */
  770. put_be32(pb, 0); /* reserved */
  771. put_be32(pb, 0); /* reserved */
  772. /* Matrix structure */
  773. put_be32(pb, 0x00010000); /* reserved */
  774. put_be32(pb, 0x0); /* reserved */
  775. put_be32(pb, 0x0); /* reserved */
  776. put_be32(pb, 0x0); /* reserved */
  777. put_be32(pb, 0x00010000); /* reserved */
  778. put_be32(pb, 0x0); /* reserved */
  779. put_be32(pb, 0x0); /* reserved */
  780. put_be32(pb, 0x0); /* reserved */
  781. put_be32(pb, 0x40000000); /* reserved */
  782. put_be32(pb, 0); /* reserved (preview time) */
  783. put_be32(pb, 0); /* reserved (preview duration) */
  784. put_be32(pb, 0); /* reserved (poster time) */
  785. put_be32(pb, 0); /* reserved (selection time) */
  786. put_be32(pb, 0); /* reserved (selection duration) */
  787. put_be32(pb, 0); /* reserved (current time) */
  788. put_be32(pb, maxTrackID+1); /* Next track id */
  789. return 0x6c;
  790. }
  791. static int mov_write_itunes_hdlr_tag(ByteIOContext *pb, MOVContext* mov,
  792. AVFormatContext *s)
  793. {
  794. offset_t pos = url_ftell(pb);
  795. put_be32(pb, 0); /* size */
  796. put_tag(pb, "hdlr");
  797. put_be32(pb, 0);
  798. put_be32(pb, 0);
  799. put_tag(pb, "mdir");
  800. put_tag(pb, "appl");
  801. put_be32(pb, 0);
  802. put_be32(pb, 0);
  803. put_be16(pb, 0);
  804. return updateSize(pb, pos);
  805. }
  806. /* helper function to write a data tag with the specified string as data */
  807. static int mov_write_string_data_tag(ByteIOContext *pb, MOVContext* mov,
  808. AVFormatContext *s, const char *data)
  809. {
  810. offset_t pos = url_ftell(pb);
  811. put_be32(pb, 0); /* size */
  812. put_tag(pb, "data");
  813. put_be32(pb, 1);
  814. put_be32(pb, 0);
  815. put_buffer(pb, data, strlen(data));
  816. return updateSize(pb, pos);
  817. }
  818. /* iTunes name of the song/movie */
  819. static int mov_write_nam_tag(ByteIOContext *pb, MOVContext* mov,
  820. AVFormatContext *s)
  821. {
  822. int size = 0;
  823. if ( s->title[0] ) {
  824. offset_t pos = url_ftell(pb);
  825. put_be32(pb, 0); /* size */
  826. put_tag(pb, "\251nam");
  827. mov_write_string_data_tag(pb, mov, s, s->title);
  828. size = updateSize(pb, pos);
  829. }
  830. return size;
  831. }
  832. /* iTunes name of the artist/performer */
  833. static int mov_write_ART_tag(ByteIOContext *pb, MOVContext* mov,
  834. AVFormatContext *s)
  835. {
  836. int size = 0;
  837. if ( s->author[0] ) {
  838. offset_t pos = url_ftell(pb);
  839. put_be32(pb, 0); /* size */
  840. put_tag(pb, "\251ART");
  841. // we use the author here as this is the only thing that we have...
  842. mov_write_string_data_tag(pb, mov, s, s->author);
  843. size = updateSize(pb, pos);
  844. }
  845. return size;
  846. }
  847. /* iTunes name of the writer */
  848. static int mov_write_wrt_tag(ByteIOContext *pb, MOVContext* mov,
  849. AVFormatContext *s)
  850. {
  851. int size = 0;
  852. if ( s->author[0] ) {
  853. offset_t pos = url_ftell(pb);
  854. put_be32(pb, 0); /* size */
  855. put_tag(pb, "\251wrt");
  856. mov_write_string_data_tag(pb, mov, s, s->author);
  857. size = updateSize(pb, pos);
  858. }
  859. return size;
  860. }
  861. /* iTunes name of the album */
  862. static int mov_write_alb_tag(ByteIOContext *pb, MOVContext* mov,
  863. AVFormatContext *s)
  864. {
  865. int size = 0;
  866. if ( s->album[0] ) {
  867. offset_t pos = url_ftell(pb);
  868. put_be32(pb, 0); /* size */
  869. put_tag(pb, "\251alb");
  870. mov_write_string_data_tag(pb, mov, s, s->album);
  871. size = updateSize(pb, pos);
  872. }
  873. return size;
  874. }
  875. /* iTunes year */
  876. static int mov_write_day_tag(ByteIOContext *pb, MOVContext* mov,
  877. AVFormatContext *s)
  878. {
  879. char year[5];
  880. int size = 0;
  881. if ( s->year ) {
  882. offset_t pos = url_ftell(pb);
  883. put_be32(pb, 0); /* size */
  884. put_tag(pb, "\251day");
  885. snprintf(year, 5, "%04d", s->year);
  886. mov_write_string_data_tag(pb, mov, s, year);
  887. size = updateSize(pb, pos);
  888. }
  889. return size;
  890. }
  891. /* iTunes tool used to create the file */
  892. static int mov_write_too_tag(ByteIOContext *pb, MOVContext* mov,
  893. AVFormatContext *s)
  894. {
  895. offset_t pos = url_ftell(pb);
  896. put_be32(pb, 0); /* size */
  897. put_tag(pb, "\251too");
  898. mov_write_string_data_tag(pb, mov, s, LIBAVFORMAT_IDENT);
  899. return updateSize(pb, pos);
  900. }
  901. /* iTunes comment */
  902. static int mov_write_cmt_tag(ByteIOContext *pb, MOVContext* mov,
  903. AVFormatContext *s)
  904. {
  905. int size = 0;
  906. if ( s->comment[0] ) {
  907. offset_t pos = url_ftell(pb);
  908. put_be32(pb, 0); /* size */
  909. put_tag(pb, "\251cmt");
  910. mov_write_string_data_tag(pb, mov, s, s->comment);
  911. size = updateSize(pb, pos);
  912. }
  913. return size;
  914. }
  915. /* iTunes custom genre */
  916. static int mov_write_gen_tag(ByteIOContext *pb, MOVContext* mov,
  917. AVFormatContext *s)
  918. {
  919. int size = 0;
  920. if ( s->genre[0] ) {
  921. offset_t pos = url_ftell(pb);
  922. put_be32(pb, 0); /* size */
  923. put_tag(pb, "\251gen");
  924. mov_write_string_data_tag(pb, mov, s, s->genre);
  925. size = updateSize(pb, pos);
  926. }
  927. return size;
  928. }
  929. /* iTunes track number */
  930. static int mov_write_trkn_tag(ByteIOContext *pb, MOVContext* mov,
  931. AVFormatContext *s)
  932. {
  933. int size = 0;
  934. if ( s->track ) {
  935. offset_t pos = url_ftell(pb);
  936. put_be32(pb, 0); /* size */
  937. put_tag(pb, "trkn");
  938. {
  939. offset_t pos = url_ftell(pb);
  940. put_be32(pb, 0); /* size */
  941. put_tag(pb, "data");
  942. put_be32(pb, 0); // 8 bytes empty
  943. put_be32(pb, 0);
  944. put_be16(pb, 0); // empty
  945. put_be16(pb, s->track); // track number
  946. put_be16(pb, 0); // total track number
  947. put_be16(pb, 0); // empty
  948. updateSize(pb, pos);
  949. }
  950. size = updateSize(pb, pos);
  951. }
  952. return size;
  953. }
  954. /* iTunes meta data list */
  955. static int mov_write_ilst_tag(ByteIOContext *pb, MOVContext* mov,
  956. AVFormatContext *s)
  957. {
  958. offset_t pos = url_ftell(pb);
  959. put_be32(pb, 0); /* size */
  960. put_tag(pb, "ilst");
  961. mov_write_nam_tag(pb, mov, s);
  962. mov_write_ART_tag(pb, mov, s);
  963. mov_write_wrt_tag(pb, mov, s);
  964. mov_write_alb_tag(pb, mov, s);
  965. mov_write_day_tag(pb, mov, s);
  966. mov_write_too_tag(pb, mov, s);
  967. mov_write_cmt_tag(pb, mov, s);
  968. mov_write_gen_tag(pb, mov, s);
  969. mov_write_trkn_tag(pb, mov, s);
  970. return updateSize(pb, pos);
  971. }
  972. /* iTunes meta data tag */
  973. static int mov_write_meta_tag(ByteIOContext *pb, MOVContext* mov,
  974. AVFormatContext *s)
  975. {
  976. int size = 0;
  977. // only save meta tag if required
  978. if ( s->title[0] || s->author[0] || s->album[0] || s->year ||
  979. s->comment[0] || s->genre[0] || s->track ) {
  980. offset_t pos = url_ftell(pb);
  981. put_be32(pb, 0); /* size */
  982. put_tag(pb, "meta");
  983. put_be32(pb, 0);
  984. mov_write_itunes_hdlr_tag(pb, mov, s);
  985. mov_write_ilst_tag(pb, mov, s);
  986. size = updateSize(pb, pos);
  987. }
  988. return size;
  989. }
  990. static int mov_write_udta_tag(ByteIOContext *pb, MOVContext* mov,
  991. AVFormatContext *s)
  992. {
  993. offset_t pos = url_ftell(pb);
  994. int i;
  995. put_be32(pb, 0); /* size */
  996. put_tag(pb, "udta");
  997. /* iTunes meta data */
  998. mov_write_meta_tag(pb, mov, s);
  999. /* Requirements */
  1000. for (i=0; i<MAX_STREAMS; i++) {
  1001. if(mov->tracks[i].entry <= 0) continue;
  1002. if (mov->tracks[i].enc->codec_id == CODEC_ID_AAC ||
  1003. mov->tracks[i].enc->codec_id == CODEC_ID_MPEG4) {
  1004. offset_t pos = url_ftell(pb);
  1005. put_be32(pb, 0); /* size */
  1006. put_tag(pb, "\251req");
  1007. put_be16(pb, sizeof("QuickTime 6.0 or greater") - 1);
  1008. put_be16(pb, 0);
  1009. put_buffer(pb, "QuickTime 6.0 or greater",
  1010. sizeof("QuickTime 6.0 or greater") - 1);
  1011. updateSize(pb, pos);
  1012. break;
  1013. }
  1014. }
  1015. /* Encoder */
  1016. if(mov->tracks[0].enc && !(mov->tracks[0].enc->flags & CODEC_FLAG_BITEXACT))
  1017. {
  1018. offset_t pos = url_ftell(pb);
  1019. put_be32(pb, 0); /* size */
  1020. put_tag(pb, "\251enc");
  1021. put_be16(pb, sizeof(LIBAVFORMAT_IDENT) - 1); /* string length */
  1022. put_be16(pb, 0);
  1023. put_buffer(pb, LIBAVFORMAT_IDENT, sizeof(LIBAVFORMAT_IDENT) - 1);
  1024. updateSize(pb, pos);
  1025. }
  1026. if( s->title[0] )
  1027. {
  1028. offset_t pos = url_ftell(pb);
  1029. put_be32(pb, 0); /* size */
  1030. put_tag(pb, "\251nam");
  1031. put_be16(pb, strlen(s->title)); /* string length */
  1032. put_be16(pb, 0);
  1033. put_buffer(pb, s->title, strlen(s->title));
  1034. updateSize(pb, pos);
  1035. }
  1036. if( s->author[0] )
  1037. {
  1038. offset_t pos = url_ftell(pb);
  1039. put_be32(pb, 0); /* size */
  1040. put_tag(pb, /*"\251aut"*/ "\251day" );
  1041. put_be16(pb, strlen(s->author)); /* string length */
  1042. put_be16(pb, 0);
  1043. put_buffer(pb, s->author, strlen(s->author));
  1044. updateSize(pb, pos);
  1045. }
  1046. if( s->comment[0] )
  1047. {
  1048. offset_t pos = url_ftell(pb);
  1049. put_be32(pb, 0); /* size */
  1050. put_tag(pb, "\251des");
  1051. put_be16(pb, strlen(s->comment)); /* string length */
  1052. put_be16(pb, 0);
  1053. put_buffer(pb, s->comment, strlen(s->comment));
  1054. updateSize(pb, pos);
  1055. }
  1056. return updateSize(pb, pos);
  1057. }
  1058. static size_t ascii_to_wc (ByteIOContext *pb, char *b, size_t n)
  1059. {
  1060. size_t i;
  1061. unsigned char c;
  1062. for (i = 0; i < n - 1; i++) {
  1063. c = b[i];
  1064. if (! (0x20 <= c && c <= 0x7f ))
  1065. c = 0x3f; /* '?' */
  1066. put_be16(pb, c);
  1067. }
  1068. put_be16(pb, 0x00);
  1069. return 2*n;
  1070. }
  1071. static uint16_t language_code (char *str)
  1072. {
  1073. return ((((str[0]-'a') & 0x1F)<<10) + (((str[1]-'a') & 0x1F)<<5) + ((str[2]-'a') & 0x1F));
  1074. }
  1075. static int mov_write_uuidusmt_tag (ByteIOContext *pb, AVFormatContext *s)
  1076. {
  1077. size_t len, size;
  1078. offset_t pos, curpos;
  1079. size = 0;
  1080. if (s->title[0]) {
  1081. pos = url_ftell(pb);
  1082. put_be32(pb, 0); /* size placeholder*/
  1083. put_tag(pb, "uuid");
  1084. put_tag(pb, "USMT");
  1085. put_be32(pb, 0x21d24fce ); /* 96 bit UUID */
  1086. put_be32(pb, 0xbb88695c );
  1087. put_be32(pb, 0xfac9c740 );
  1088. size += 24;
  1089. put_be32(pb, 0); /* size placeholder*/
  1090. put_tag(pb, "MTDT");
  1091. put_be16(pb, 1);
  1092. size += 10;
  1093. // Title
  1094. len = strlen(s->title)+1;
  1095. put_be16(pb, len*2+10); /* size */
  1096. put_be32(pb, 0x01); /* type */
  1097. put_be16(pb, language_code("und")); /* language */
  1098. put_be16(pb, 0x01); /* ? */
  1099. ascii_to_wc (pb, s->title, len);
  1100. size += len*2+10;
  1101. // size
  1102. curpos = url_ftell(pb);
  1103. url_fseek(pb, pos, SEEK_SET);
  1104. put_be32(pb, size);
  1105. url_fseek(pb, pos+24, SEEK_SET);
  1106. put_be32(pb, size-24);
  1107. url_fseek(pb, curpos, SEEK_SET);
  1108. }
  1109. return size;
  1110. }
  1111. static int mov_write_moov_tag(ByteIOContext *pb, MOVContext *mov,
  1112. AVFormatContext *s)
  1113. {
  1114. int i;
  1115. offset_t pos = url_ftell(pb);
  1116. put_be32(pb, 0); /* size placeholder*/
  1117. put_tag(pb, "moov");
  1118. mov->timescale = globalTimescale;
  1119. for (i=0; i<MAX_STREAMS; i++) {
  1120. if(mov->tracks[i].entry <= 0) continue;
  1121. if(mov->tracks[i].enc->codec_type == CODEC_TYPE_VIDEO) {
  1122. mov->tracks[i].timescale = mov->tracks[i].enc->time_base.den;
  1123. mov->tracks[i].sampleDuration = mov->tracks[i].enc->time_base.num;
  1124. }
  1125. else if(mov->tracks[i].enc->codec_type == CODEC_TYPE_AUDIO) {
  1126. /* If AMR, track timescale = 8000, AMR_WB = 16000 */
  1127. if(mov->tracks[i].enc->codec_id == CODEC_ID_AMR_NB) {
  1128. mov->tracks[i].sampleDuration = 160; // Bytes per chunk
  1129. mov->tracks[i].timescale = 8000;
  1130. }
  1131. else {
  1132. mov->tracks[i].timescale = mov->tracks[i].enc->sample_rate;
  1133. mov->tracks[i].sampleDuration = mov->tracks[i].enc->frame_size;
  1134. }
  1135. }
  1136. mov->tracks[i].trackDuration =
  1137. mov->tracks[i].sampleCount * mov->tracks[i].sampleDuration;
  1138. mov->tracks[i].time = mov->time;
  1139. mov->tracks[i].trackID = i+1;
  1140. }
  1141. mov_write_mvhd_tag(pb, mov);
  1142. //mov_write_iods_tag(pb, mov);
  1143. for (i=0; i<MAX_STREAMS; i++) {
  1144. if(mov->tracks[i].entry > 0) {
  1145. mov_write_trak_tag(pb, &(mov->tracks[i]));
  1146. }
  1147. }
  1148. if (mov->mode == MODE_PSP)
  1149. mov_write_uuidusmt_tag(pb, s);
  1150. else
  1151. mov_write_udta_tag(pb, mov, s);
  1152. return updateSize(pb, pos);
  1153. }
  1154. int mov_write_mdat_tag(ByteIOContext *pb, MOVContext* mov)
  1155. {
  1156. put_be32(pb, 8); // placeholder for extended size field (64 bit)
  1157. put_tag(pb, "wide");
  1158. mov->mdat_pos = url_ftell(pb);
  1159. put_be32(pb, 0); /* size placeholder*/
  1160. put_tag(pb, "mdat");
  1161. return 0;
  1162. }
  1163. /* TODO: This needs to be more general */
  1164. static void mov_write_ftyp_tag (ByteIOContext *pb, AVFormatContext *s)
  1165. {
  1166. MOVContext *mov = s->priv_data;
  1167. put_be32(pb, 0x14 ); /* size */
  1168. put_tag(pb, "ftyp");
  1169. if ( mov->mode == MODE_3GP )
  1170. put_tag(pb, "3gp4");
  1171. else if ( mov->mode == MODE_3G2 )
  1172. put_tag(pb, "3g2a");
  1173. else if ( mov->mode == MODE_PSP )
  1174. put_tag(pb, "MSNV");
  1175. else
  1176. put_tag(pb, "isom");
  1177. put_be32(pb, 0x200 );
  1178. if ( mov->mode == MODE_3GP )
  1179. put_tag(pb, "3gp4");
  1180. else if ( mov->mode == MODE_3G2 )
  1181. put_tag(pb, "3g2a");
  1182. else if ( mov->mode == MODE_PSP )
  1183. put_tag(pb, "MSNV");
  1184. else
  1185. put_tag(pb, "mp41");
  1186. }
  1187. static void mov_write_uuidprof_tag(ByteIOContext *pb, AVFormatContext *s)
  1188. {
  1189. AVCodecContext *VideoCodec = s->streams[0]->codec;
  1190. AVCodecContext *AudioCodec = s->streams[1]->codec;
  1191. int AudioRate = AudioCodec->sample_rate;
  1192. int FrameRate = ((VideoCodec->time_base.den) * (0x10000))/ (VideoCodec->time_base.num);
  1193. put_be32(pb, 0x94 ); /* size */
  1194. put_tag(pb, "uuid");
  1195. put_tag(pb, "PROF");
  1196. put_be32(pb, 0x21d24fce ); /* 96 bit UUID */
  1197. put_be32(pb, 0xbb88695c );
  1198. put_be32(pb, 0xfac9c740 );
  1199. put_be32(pb, 0x0 ); /* ? */
  1200. put_be32(pb, 0x3 ); /* 3 sections ? */
  1201. put_be32(pb, 0x14 ); /* size */
  1202. put_tag(pb, "FPRF");
  1203. put_be32(pb, 0x0 ); /* ? */
  1204. put_be32(pb, 0x0 ); /* ? */
  1205. put_be32(pb, 0x0 ); /* ? */
  1206. put_be32(pb, 0x2c ); /* size */
  1207. put_tag(pb, "APRF"); /* audio */
  1208. put_be32(pb, 0x0 );
  1209. put_be32(pb, 0x2 ); /* TrackID */
  1210. put_tag(pb, "mp4a");
  1211. put_be32(pb, 0x20f );
  1212. put_be32(pb, 0x0 );
  1213. put_be32(pb, AudioCodec->bit_rate / 1000);
  1214. put_be32(pb, AudioCodec->bit_rate / 1000);
  1215. put_be32(pb, AudioRate );
  1216. put_be32(pb, AudioCodec->channels );
  1217. put_be32(pb, 0x34 ); /* size */
  1218. put_tag(pb, "VPRF"); /* video */
  1219. put_be32(pb, 0x0 );
  1220. put_be32(pb, 0x1 ); /* TrackID */
  1221. put_tag(pb, "mp4v");
  1222. put_be32(pb, 0x103 );
  1223. put_be32(pb, 0x0 );
  1224. put_be32(pb, VideoCodec->bit_rate / 1000);
  1225. put_be32(pb, VideoCodec->bit_rate / 1000);
  1226. put_be32(pb, FrameRate);
  1227. put_be32(pb, FrameRate);
  1228. put_be16(pb, VideoCodec->width);
  1229. put_be16(pb, VideoCodec->height);
  1230. put_be32(pb, 0x010001); /* ? */
  1231. }
  1232. static int mov_write_header(AVFormatContext *s)
  1233. {
  1234. ByteIOContext *pb = &s->pb;
  1235. MOVContext *mov = s->priv_data;
  1236. int i;
  1237. for(i=0; i<s->nb_streams; i++){
  1238. AVCodecContext *c= s->streams[i]->codec;
  1239. if(c->codec_type == CODEC_TYPE_VIDEO){
  1240. av_set_pts_info(s->streams[i], 64, 1, c->time_base.den);
  1241. if (!codec_get_tag(codec_movvideo_tags, c->codec_id)){
  1242. if(!codec_get_tag(codec_bmp_tags, c->codec_id))
  1243. return -1;
  1244. else
  1245. av_log(s, AV_LOG_INFO, "Warning, using MS style video codec tag, the file may be unplayable!\n");
  1246. }
  1247. }else if(c->codec_type == CODEC_TYPE_AUDIO){
  1248. av_set_pts_info(s->streams[i], 64, 1, c->sample_rate);
  1249. if (!codec_get_tag(codec_movaudio_tags, c->codec_id)){
  1250. if(!codec_get_tag(codec_wav_tags, c->codec_id))
  1251. return -1;
  1252. else
  1253. av_log(s, AV_LOG_INFO, "Warning, using MS style audio codec tag, the file may be unplayable!\n");
  1254. }
  1255. }
  1256. /* don't know yet if mp4 or not */
  1257. mov->tracks[i].language = ff_mov_iso639_to_lang(s->streams[i]->language, 1);
  1258. }
  1259. /* Default mode == MP4 */
  1260. mov->mode = MODE_MP4;
  1261. if (s->oformat != NULL) {
  1262. if (!strcmp("3gp", s->oformat->name)) mov->mode = MODE_3GP;
  1263. else if (!strcmp("3g2", s->oformat->name)) mov->mode = MODE_3G2;
  1264. else if (!strcmp("mov", s->oformat->name)) mov->mode = MODE_MOV;
  1265. else if (!strcmp("psp", s->oformat->name)) mov->mode = MODE_PSP;
  1266. if ( mov->mode == MODE_3GP || mov->mode == MODE_3G2 ||
  1267. mov->mode == MODE_MP4 || mov->mode == MODE_PSP )
  1268. mov_write_ftyp_tag(pb,s);
  1269. if ( mov->mode == MODE_PSP ) {
  1270. if ( s->nb_streams != 2 ) {
  1271. av_log(s, AV_LOG_ERROR, "PSP mode need one video and one audio stream\n");
  1272. return -1;
  1273. }
  1274. mov_write_uuidprof_tag(pb,s);
  1275. }
  1276. }
  1277. for (i=0; i<MAX_STREAMS; i++) {
  1278. mov->tracks[i].mode = mov->mode;
  1279. }
  1280. put_flush_packet(pb);
  1281. return 0;
  1282. }
  1283. static int mov_write_packet(AVFormatContext *s, AVPacket *pkt)
  1284. {
  1285. MOVContext *mov = s->priv_data;
  1286. ByteIOContext *pb = &s->pb;
  1287. AVCodecContext *enc = s->streams[pkt->stream_index]->codec;
  1288. MOVTrack* trk = &mov->tracks[pkt->stream_index];
  1289. int cl, id;
  1290. unsigned int samplesInChunk = 0;
  1291. int size= pkt->size;
  1292. if (url_is_streamed(&s->pb)) return 0; /* Can't handle that */
  1293. if (!size) return 0; /* Discard 0 sized packets */
  1294. if (enc->codec_type == CODEC_TYPE_VIDEO ) {
  1295. samplesInChunk = 1;
  1296. }
  1297. else if (enc->codec_type == CODEC_TYPE_AUDIO ) {
  1298. if( enc->codec_id == CODEC_ID_AMR_NB) {
  1299. /* We must find out how many AMR blocks there are in one packet */
  1300. static uint16_t packed_size[16] =
  1301. {13, 14, 16, 18, 20, 21, 27, 32, 6, 0, 0, 0, 0, 0, 0, 0};
  1302. int len = 0;
  1303. while (len < size && samplesInChunk < 100) {
  1304. len += packed_size[(pkt->data[len] >> 3) & 0x0F];
  1305. samplesInChunk++;
  1306. }
  1307. }
  1308. else if(enc->codec_id == CODEC_ID_PCM_ALAW) {
  1309. samplesInChunk = size/enc->channels;
  1310. }
  1311. else if(enc->codec_id == CODEC_ID_PCM_S16BE || enc->codec_id == CODEC_ID_PCM_S16LE) {
  1312. samplesInChunk = size/(2*enc->channels);
  1313. }
  1314. else {
  1315. samplesInChunk = 1;
  1316. }
  1317. }
  1318. if ((enc->codec_id == CODEC_ID_MPEG4 || enc->codec_id == CODEC_ID_AAC)
  1319. && trk->vosLen == 0) {
  1320. // assert(enc->extradata_size);
  1321. trk->vosLen = enc->extradata_size;
  1322. trk->vosData = av_malloc(trk->vosLen);
  1323. memcpy(trk->vosData, enc->extradata, trk->vosLen);
  1324. }
  1325. cl = trk->entry / MOV_INDEX_CLUSTER_SIZE;
  1326. id = trk->entry % MOV_INDEX_CLUSTER_SIZE;
  1327. if (trk->ents_allocated <= trk->entry) {
  1328. trk->cluster = av_realloc(trk->cluster, (cl+1)*sizeof(void*));
  1329. if (!trk->cluster)
  1330. return -1;
  1331. trk->cluster[cl] = av_malloc(MOV_INDEX_CLUSTER_SIZE*sizeof(MOVIentry));
  1332. if (!trk->cluster[cl])
  1333. return -1;
  1334. trk->ents_allocated += MOV_INDEX_CLUSTER_SIZE;
  1335. }
  1336. if (mov->mdat_written == 0) {
  1337. mov_write_mdat_tag(pb, mov);
  1338. mov->mdat_written = 1;
  1339. mov->time = s->timestamp + 0x7C25B080; //1970 based -> 1904 based
  1340. }
  1341. trk->cluster[cl][id].pos = url_ftell(pb);
  1342. trk->cluster[cl][id].samplesInChunk = samplesInChunk;
  1343. trk->cluster[cl][id].size = size;
  1344. trk->cluster[cl][id].entries = samplesInChunk;
  1345. if(enc->codec_type == CODEC_TYPE_VIDEO) {
  1346. if (pkt->dts != pkt->pts)
  1347. trk->hasBframes = 1;
  1348. trk->cluster[cl][id].cts = pkt->pts - pkt->dts;
  1349. trk->cluster[cl][id].key_frame = !!(pkt->flags & PKT_FLAG_KEY);
  1350. if(trk->cluster[cl][id].key_frame)
  1351. trk->hasKeyframes = 1;
  1352. }
  1353. trk->enc = enc;
  1354. trk->entry++;
  1355. trk->sampleCount += samplesInChunk;
  1356. trk->mdat_size += size;
  1357. put_buffer(pb, pkt->data, size);
  1358. put_flush_packet(pb);
  1359. return 0;
  1360. }
  1361. static int mov_write_trailer(AVFormatContext *s)
  1362. {
  1363. MOVContext *mov = s->priv_data;
  1364. ByteIOContext *pb = &s->pb;
  1365. int res = 0;
  1366. int i;
  1367. uint64_t j;
  1368. offset_t moov_pos = url_ftell(pb);
  1369. /* Write size of mdat tag */
  1370. for (i=0, j=0; i<MAX_STREAMS; i++) {
  1371. if(mov->tracks[i].ents_allocated > 0) {
  1372. j += mov->tracks[i].mdat_size;
  1373. }
  1374. }
  1375. if (j+8 <= UINT32_MAX) {
  1376. url_fseek(pb, mov->mdat_pos, SEEK_SET);
  1377. put_be32(pb, j+8);
  1378. } else {
  1379. /* overwrite 'wide' placeholder atom */
  1380. url_fseek(pb, mov->mdat_pos - 8, SEEK_SET);
  1381. put_be32(pb, 1); /* special value: real atom size will be 64 bit value after tag field */
  1382. put_tag(pb, "mdat");
  1383. put_be64(pb, j+16);
  1384. }
  1385. url_fseek(pb, moov_pos, SEEK_SET);
  1386. mov_write_moov_tag(pb, mov, s);
  1387. for (i=0; i<MAX_STREAMS; i++) {
  1388. for (j=0; j<mov->tracks[i].ents_allocated/MOV_INDEX_CLUSTER_SIZE; j++) {
  1389. av_free(mov->tracks[i].cluster[j]);
  1390. }
  1391. av_free(mov->tracks[i].cluster);
  1392. if( mov->tracks[i].vosLen ) av_free( mov->tracks[i].vosData );
  1393. mov->tracks[i].cluster = NULL;
  1394. mov->tracks[i].ents_allocated = mov->tracks[i].entry = 0;
  1395. }
  1396. put_flush_packet(pb);
  1397. return res;
  1398. }
  1399. static AVOutputFormat mov_oformat = {
  1400. "mov",
  1401. "mov format",
  1402. NULL,
  1403. "mov",
  1404. sizeof(MOVContext),
  1405. CODEC_ID_AAC,
  1406. CODEC_ID_MPEG4,
  1407. mov_write_header,
  1408. mov_write_packet,
  1409. mov_write_trailer,
  1410. .flags = AVFMT_GLOBALHEADER,
  1411. };
  1412. static AVOutputFormat _3gp_oformat = {
  1413. "3gp",
  1414. "3gp format",
  1415. NULL,
  1416. "3gp",
  1417. sizeof(MOVContext),
  1418. CODEC_ID_AMR_NB,
  1419. CODEC_ID_H263,
  1420. mov_write_header,
  1421. mov_write_packet,
  1422. mov_write_trailer,
  1423. .flags = AVFMT_GLOBALHEADER,
  1424. };
  1425. static AVOutputFormat mp4_oformat = {
  1426. "mp4",
  1427. "mp4 format",
  1428. "application/mp4",
  1429. "mp4,m4a",
  1430. sizeof(MOVContext),
  1431. CODEC_ID_AAC,
  1432. CODEC_ID_MPEG4,
  1433. mov_write_header,
  1434. mov_write_packet,
  1435. mov_write_trailer,
  1436. .flags = AVFMT_GLOBALHEADER,
  1437. };
  1438. static AVOutputFormat psp_oformat = {
  1439. "psp",
  1440. "psp mp4 format",
  1441. NULL,
  1442. "mp4,psp",
  1443. sizeof(MOVContext),
  1444. CODEC_ID_AAC,
  1445. CODEC_ID_MPEG4,
  1446. mov_write_header,
  1447. mov_write_packet,
  1448. mov_write_trailer,
  1449. .flags = AVFMT_GLOBALHEADER,
  1450. };
  1451. static AVOutputFormat _3g2_oformat = {
  1452. "3g2",
  1453. "3gp2 format",
  1454. NULL,
  1455. "3g2",
  1456. sizeof(MOVContext),
  1457. CODEC_ID_AMR_NB,
  1458. CODEC_ID_H263,
  1459. mov_write_header,
  1460. mov_write_packet,
  1461. mov_write_trailer,
  1462. .flags = AVFMT_GLOBALHEADER,
  1463. };
  1464. int movenc_init(void)
  1465. {
  1466. av_register_output_format(&mov_oformat);
  1467. av_register_output_format(&_3gp_oformat);
  1468. av_register_output_format(&mp4_oformat);
  1469. av_register_output_format(&psp_oformat);
  1470. av_register_output_format(&_3g2_oformat);
  1471. return 0;
  1472. }