2 * Copyright (C) 2013 nu774
3 * For conditions of distribution and use, see copyright notice in COPYING
17 # include <inttypes.h>
18 #elif defined _MSC_VER
19 # define PRId64 "I64d"
22 #include "m4af_endian.h"
24 #define m4af_realloc(memory,size) realloc(memory, size)
25 #define m4af_free(memory) free(memory)
26 #define m4af_max(a,b) ((a)<(b)?(b):(a))
28 #define M4AF_ATOM_WILD 0xffffffff
30 typedef struct m4af_sample_entry_t
{
33 } m4af_sample_entry_t
;
35 typedef struct m4af_chunk_entry_t
{
38 uint32_t samples_per_chunk
;
42 typedef struct m4af_track_t
{
45 int64_t creation_time
;
46 int64_t modification_time
;
48 uint32_t frame_duration
;
49 uint32_t encoder_delay
;
51 uint8_t *decSpecificInfo
;
52 uint32_t decSpecificInfoSize
;
53 uint32_t bufferSizeDB
;
58 m4af_sample_entry_t
*sample_table
;
60 uint32_t sample_table_capacity
;
62 m4af_chunk_entry_t
*chunk_table
;
64 uint32_t chunk_table_capacity
;
66 uint8_t *chunk_buffer
;
68 uint32_t chunk_capacity
;
70 /* temporary, to help parsing */
80 int64_t creation_time
;
81 int64_t modification_time
;
87 m4af_itmf_entry_t
*itmf_table
;
89 uint32_t itmf_table_capacity
;
91 m4af_io_callbacks_t io
;
95 m4af_track_t track
[2];
97 m4af_itmf_entry_t current_tag
;
100 typedef struct m4af_box_parser_t
{
102 int (*handler
)(m4af_ctx_t
*ctx
, uint32_t name
, uint64_t size
);
106 int64_t m4af_timestamp(void)
108 return (int64_t)(time(0)) + (((1970 - 1904) * 365) + 17) * 24 * 60 * 60;
112 * http://graphics.stanford.edu/~seander/bithacks.html#RoundUpPowerOf2
115 uint32_t m4af_roundup(uint32_t n
)
128 int64_t m4af_tell(m4af_ctx_t
*ctx
)
131 if ((pos
= ctx
->io
.tell(ctx
->io_cookie
)) < 0)
132 ctx
->last_error
= M4AF_IO_ERROR
;
137 int m4af_set_pos(m4af_ctx_t
*ctx
, int64_t pos
)
140 if ((rc
= ctx
->io
.seek(ctx
->io_cookie
, pos
, SEEK_SET
)) < 0)
141 ctx
->last_error
= M4AF_IO_ERROR
;
146 int m4af_write(m4af_ctx_t
*ctx
, const void *data
, uint32_t size
)
149 if ((rc
= ctx
->io
.write(ctx
->io_cookie
, data
, size
)) < 0)
150 ctx
->last_error
= M4AF_IO_ERROR
;
155 int m4af_write16(m4af_ctx_t
*ctx
, uint32_t data
)
157 data
= m4af_htob16(data
);
158 return m4af_write(ctx
, &data
, 2);
162 int m4af_write32(m4af_ctx_t
*ctx
, uint32_t data
)
164 data
= m4af_htob32(data
);
165 return m4af_write(ctx
, &data
, 4);
169 int m4af_write64(m4af_ctx_t
*ctx
, uint64_t data
)
171 data
= m4af_htob64(data
);
172 return m4af_write(ctx
, &data
, 8);
176 int m4af_write24(m4af_ctx_t
*ctx
, uint32_t data
)
178 data
= m4af_htob32(data
<< 8);
179 return m4af_write(ctx
, &data
, 3);
183 void m4af_write32_at(m4af_ctx_t
*ctx
, int64_t pos
, uint32_t value
)
185 int64_t current_pos
= m4af_tell(ctx
);
186 m4af_set_pos(ctx
, pos
);
187 m4af_write32(ctx
, value
);
188 m4af_set_pos(ctx
, current_pos
);
191 m4af_ctx_t
*m4af_create(uint32_t codec
, uint32_t timescale
,
192 m4af_io_callbacks_t
*io
, void *io_cookie
)
197 if (codec
!= M4AF_FOURCC('m','p','4','a') &&
198 codec
!= M4AF_FOURCC('a','l','a','c'))
200 if ((ctx
= m4af_realloc(0, sizeof(m4af_ctx_t
))) == 0)
202 memset(ctx
, 0, sizeof(m4af_ctx_t
));
203 memcpy(&ctx
->io
, io
, sizeof(m4af_io_callbacks_t
));
204 ctx
->io_cookie
= io_cookie
;
205 ctx
->timescale
= timescale
;
206 timestamp
= m4af_timestamp();
207 ctx
->creation_time
= timestamp
;
208 ctx
->modification_time
= timestamp
;
210 ctx
->track
[0].codec
= codec
;
211 ctx
->track
[0].timescale
= timescale
;
212 ctx
->track
[0].creation_time
= timestamp
;
213 ctx
->track
[0].modification_time
= timestamp
;
218 void m4af_free_itmf_table(m4af_ctx_t
*ctx
)
221 m4af_itmf_entry_t
*entry
= ctx
->itmf_table
;
222 for (i
= 0; i
< ctx
->num_tags
; ++i
, ++entry
) {
223 if (entry
->fcc
== M4AF_FOURCC('-','-','-','-'))
224 m4af_free(entry
->name
);
225 m4af_free(entry
->data
);
227 m4af_free(ctx
->itmf_table
);
231 void m4af_clear_track(m4af_ctx_t
*ctx
, int track_idx
)
233 m4af_track_t
*track
= ctx
->track
+ track_idx
;
234 if (track
->decSpecificInfo
)
235 m4af_free(track
->decSpecificInfo
);
236 if (track
->sample_table
)
237 m4af_free(track
->sample_table
);
238 if (track
->chunk_table
)
239 m4af_free(track
->chunk_table
);
240 if (track
->chunk_buffer
)
241 m4af_free(track
->chunk_buffer
);
242 memset(track
, 0, sizeof(m4af_track_t
));
245 void m4af_teardown(m4af_ctx_t
**ctxp
)
248 m4af_ctx_t
*ctx
= *ctxp
;
249 for (i
= 0; i
< ctx
->num_tracks
; ++i
)
250 m4af_clear_track(ctx
, i
);
252 m4af_free_itmf_table(ctx
);
257 void m4af_set_fixed_frame_duration(m4af_ctx_t
*ctx
, uint32_t track_idx
,
260 ctx
->track
[track_idx
].frame_duration
= length
;
263 int m4af_set_decoder_specific_info(m4af_ctx_t
*ctx
, uint32_t track_idx
,
264 uint8_t *data
, uint32_t size
)
266 m4af_track_t
*track
= &ctx
->track
[track_idx
];
267 if (size
> track
->decSpecificInfoSize
) {
268 uint8_t *memory
= m4af_realloc(track
->decSpecificInfo
, size
);
270 ctx
->last_error
= M4AF_NO_MEMORY
;
273 track
->decSpecificInfo
= memory
;
276 memcpy(track
->decSpecificInfo
, data
, size
);
277 track
->decSpecificInfoSize
= size
;
279 return ctx
->last_error
;
282 void m4af_set_vbr_mode(m4af_ctx_t
*ctx
, uint32_t track_idx
, int is_vbr
)
284 m4af_track_t
*track
= &ctx
->track
[track_idx
];
285 track
->is_vbr
= is_vbr
;
288 void m4af_set_priming(m4af_ctx_t
*ctx
, uint32_t track_idx
,
289 uint32_t encoder_delay
, uint32_t padding
)
291 m4af_track_t
*track
= &ctx
->track
[track_idx
];
292 track
->encoder_delay
= encoder_delay
;
293 track
->padding
= padding
;
296 void m4af_set_priming_mode(m4af_ctx_t
*ctx
, int mode
)
298 ctx
->priming_mode
= mode
;
302 int m4af_add_sample_entry(m4af_ctx_t
*ctx
, uint32_t track_idx
,
303 uint32_t size
, uint32_t delta
)
305 m4af_track_t
*track
= &ctx
->track
[track_idx
];
306 m4af_sample_entry_t
*entry
;
310 if (track
->num_samples
== track
->sample_table_capacity
) {
311 uint32_t new_size
= track
->sample_table_capacity
;
312 new_size
= new_size
? new_size
* 2 : 1;
313 entry
= m4af_realloc(track
->sample_table
, new_size
* sizeof(*entry
));
315 ctx
->last_error
= M4AF_NO_MEMORY
;
318 track
->sample_table
= entry
;
319 track
->sample_table_capacity
= new_size
;
321 entry
= track
->sample_table
+ track
->num_samples
;
323 entry
->delta
= delta
;
324 ++track
->num_samples
;
329 int m4af_flush_chunk(m4af_ctx_t
*ctx
, uint32_t track_idx
)
331 m4af_track_t
*track
= &ctx
->track
[track_idx
];
332 m4af_chunk_entry_t
*entry
;
333 if (!track
->num_chunks
|| !track
->chunk_size
)
335 entry
= &track
->chunk_table
[track
->num_chunks
- 1];
336 entry
->offset
= m4af_tell(ctx
);
337 m4af_write(ctx
, track
->chunk_buffer
, track
->chunk_size
);
338 ctx
->mdat_size
+= track
->chunk_size
;
339 track
->chunk_size
= 0;
340 return ctx
->last_error
? -1 : 0;
344 int m4af_add_chunk_entry(m4af_ctx_t
*ctx
, uint32_t track_idx
)
346 m4af_track_t
*track
= &ctx
->track
[track_idx
];
347 m4af_chunk_entry_t
*entry
;
348 if (track
->num_chunks
== track
->chunk_table_capacity
) {
349 uint32_t new_size
= track
->chunk_table_capacity
;
350 new_size
= new_size
? new_size
* 2 : 1;
351 entry
= m4af_realloc(track
->chunk_table
, new_size
* sizeof(*entry
));
353 ctx
->last_error
= M4AF_NO_MEMORY
;
356 track
->chunk_table
= entry
;
357 track
->chunk_table_capacity
= new_size
;
359 memset(&track
->chunk_table
[track
->num_chunks
++], 0,
360 sizeof(m4af_chunk_entry_t
));
365 int m4af_update_chunk_table(m4af_ctx_t
*ctx
, uint32_t track_idx
,
366 uint32_t size
, uint32_t delta
)
368 m4af_track_t
*track
= &ctx
->track
[track_idx
];
369 m4af_chunk_entry_t
*entry
;
370 int add_new_chunk
= 0;
374 if (track
->num_chunks
== 0)
377 entry
= &track
->chunk_table
[track
->num_chunks
- 1];
378 if (entry
->duration
+ delta
> track
->timescale
/ 2)
382 m4af_flush_chunk(ctx
, track_idx
);
383 if (m4af_add_chunk_entry(ctx
, track_idx
) < 0)
386 entry
= &track
->chunk_table
[track
->num_chunks
- 1];
388 ++entry
->samples_per_chunk
;
389 entry
->duration
+= delta
;
394 void m4af_update_max_bitrate(m4af_ctx_t
*ctx
, uint32_t track_idx
)
396 m4af_track_t
*track
= &ctx
->track
[track_idx
];
397 uint32_t duration
= 0, size
= 0, bitrate
;
398 m4af_sample_entry_t
*ent
= track
->sample_table
+ track
->num_samples
- 1;
400 for (; ent
>= track
->sample_table
&& duration
< track
->timescale
; --ent
) {
401 duration
+= ent
->delta
;
404 bitrate
= (uint32_t)(size
* 8.0 * track
->timescale
/ duration
+ .5);
405 if (bitrate
> track
->maxBitrate
)
406 track
->maxBitrate
= bitrate
;
410 int m4af_append_sample_to_chunk(m4af_ctx_t
*ctx
, uint32_t track_idx
,
411 const void *data
, uint32_t size
)
413 m4af_track_t
*track
= &ctx
->track
[track_idx
];
414 uint32_t newsize
= track
->chunk_size
+ size
;
418 if (track
->chunk_capacity
< newsize
) {
419 uint32_t capacity
= m4af_roundup(newsize
);
420 uint8_t *memory
= realloc(track
->chunk_buffer
, capacity
);
422 ctx
->last_error
= M4AF_NO_MEMORY
;
425 track
->chunk_buffer
= memory
;
426 track
->chunk_capacity
= capacity
;
428 memcpy(track
->chunk_buffer
+ track
->chunk_size
, data
, size
);
429 track
->chunk_size
= newsize
;
433 int m4af_write_sample(m4af_ctx_t
*ctx
, uint32_t track_idx
, const void *data
,
434 uint32_t size
, uint32_t duration
)
436 m4af_track_t
*track
= &ctx
->track
[track_idx
];
437 if (track
->frame_duration
)
438 duration
= track
->frame_duration
;
439 if (size
> track
->bufferSizeDB
)
440 track
->bufferSizeDB
= size
;
441 track
->duration
+= duration
;
442 m4af_add_sample_entry(ctx
, track_idx
, size
, duration
);
443 m4af_update_chunk_table(ctx
, track_idx
, size
, duration
);
444 m4af_update_max_bitrate(ctx
, track_idx
);
445 m4af_append_sample_to_chunk(ctx
, track_idx
, data
, size
);
446 return ctx
->last_error
;
450 m4af_itmf_entry_t
*m4af_find_itmf_slot(m4af_ctx_t
*ctx
, uint32_t fcc
,
453 m4af_itmf_entry_t
*entry
= ctx
->itmf_table
;
456 fcc
= M4AF_FOURCC('-','-','-','-');
458 if (fcc
!= M4AF_TAG_ARTWORK
)
459 for (; entry
!= ctx
->itmf_table
+ ctx
->num_tags
; ++entry
)
460 if (fcc
== entry
->fcc
&& (!name
|| !strcmp(name
, entry
->name
)))
463 if (ctx
->num_tags
== ctx
->itmf_table_capacity
) {
464 uint32_t new_size
= ctx
->itmf_table_capacity
;
465 new_size
= new_size
? new_size
* 2 : 1;
466 entry
= m4af_realloc(ctx
->itmf_table
, new_size
* sizeof(*entry
));
468 ctx
->last_error
= M4AF_NO_MEMORY
;
471 ctx
->itmf_table
= entry
;
472 ctx
->itmf_table_capacity
= new_size
;
474 entry
= &ctx
->itmf_table
[ctx
->num_tags
++];
475 memset(entry
, 0, sizeof(m4af_itmf_entry_t
));
478 char *name_copy
= m4af_realloc(0, strlen(name
) + 1);
480 ctx
->last_error
= M4AF_NO_MEMORY
;
484 strcpy(name_copy
, name
);
485 entry
->name
= name_copy
;
490 int m4af_add_itmf_long_tag(m4af_ctx_t
*ctx
, const char *name
,
493 m4af_itmf_entry_t
*entry
;
495 size_t name_len
= strlen(name
);
496 size_t data_len
= strlen(data
);
497 if (!name_len
|| !data_len
)
500 if ((entry
= m4af_find_itmf_slot(ctx
, 0, name
)) == 0)
502 entry
->type_code
= M4AF_UTF8
;
503 if ((data_copy
= m4af_realloc(entry
->data
, data_len
)) == 0) {
504 ctx
->last_error
= M4AF_NO_MEMORY
;
507 memcpy(data_copy
, data
, data_len
);
508 entry
->data
= data_copy
;
509 entry
->data_size
= data_len
;
512 return ctx
->last_error
;
515 int m4af_add_itmf_short_tag(m4af_ctx_t
*ctx
, uint32_t fcc
,
516 uint32_t type_code
, const void *data
,
519 m4af_itmf_entry_t
*entry
;
524 if ((entry
= m4af_find_itmf_slot(ctx
, fcc
, 0)) == 0)
526 entry
->type_code
= type_code
;
527 if ((data_copy
= m4af_realloc(entry
->data
, data_size
)) == 0) {
528 ctx
->last_error
= M4AF_NO_MEMORY
;
531 memcpy(data_copy
, data
, data_size
);
532 entry
->data
= data_copy
;
533 entry
->data_size
= data_size
;
536 return ctx
->last_error
;
539 int m4af_add_itmf_string_tag(m4af_ctx_t
*ctx
, uint32_t fcc
, const char *data
)
541 return m4af_add_itmf_short_tag(ctx
, fcc
, M4AF_UTF8
, data
, strlen(data
));
544 int m4af_add_itmf_int8_tag(m4af_ctx_t
*ctx
, uint32_t fcc
, int value
)
546 uint8_t data
= value
;
547 return m4af_add_itmf_short_tag(ctx
, fcc
, M4AF_INTEGER
, &data
, 1);
550 int m4af_add_itmf_int16_tag(m4af_ctx_t
*ctx
, uint32_t fcc
, int value
)
552 uint16_t data
= m4af_htob16(value
);
553 return m4af_add_itmf_short_tag(ctx
, fcc
, M4AF_INTEGER
, &data
, 2);
556 int m4af_add_itmf_int32_tag(m4af_ctx_t
*ctx
, uint32_t fcc
, uint32_t value
)
558 uint32_t data
= m4af_htob32(value
);
559 return m4af_add_itmf_short_tag(ctx
, fcc
, M4AF_INTEGER
, &data
, 4);
562 int m4af_add_itmf_int64_tag(m4af_ctx_t
*ctx
, uint32_t fcc
, uint64_t value
)
564 uint64_t data
= m4af_htob64(value
);
565 return m4af_add_itmf_short_tag(ctx
, fcc
, M4AF_INTEGER
, &data
, 8);
568 int m4af_add_itmf_track_tag(m4af_ctx_t
*ctx
, int track
, int total
)
570 uint16_t data
[4] = { 0 };
571 data
[1] = m4af_htob16(track
);
572 data
[2] = m4af_htob16(total
);
573 return m4af_add_itmf_short_tag(ctx
, M4AF_FOURCC('t','r','k','n'),
574 M4AF_IMPLICIT
, &data
, 8);
577 int m4af_add_itmf_disk_tag(m4af_ctx_t
*ctx
, int disk
, int total
)
579 uint16_t data
[3] = { 0 };
580 data
[1] = m4af_htob16(disk
);
581 data
[2] = m4af_htob16(total
);
582 return m4af_add_itmf_short_tag(ctx
, M4AF_FOURCC('d','i','s','k'),
583 M4AF_IMPLICIT
, &data
, 6);
586 int m4af_add_itmf_genre_tag(m4af_ctx_t
*ctx
, int genre
)
588 uint16_t data
= m4af_htob16(genre
);
589 return m4af_add_itmf_short_tag(ctx
, M4AF_FOURCC('g','n','r','e'),
590 M4AF_IMPLICIT
, &data
, 2);
594 int m4af_set_iTunSMPB(m4af_ctx_t
*ctx
)
596 const char *fmt
= " 00000000 %08X %08X %08X%08X 00000000 00000000 "
597 "00000000 00000000 00000000 00000000 00000000 00000000";
598 m4af_track_t
*track
= &ctx
->track
[0];
600 uint64_t length
= track
->duration
- track
->encoder_delay
- track
->padding
;
601 sprintf(buf
, fmt
, track
->encoder_delay
, track
->padding
,
602 (uint32_t)(length
>> 32), (uint32_t)length
);
603 return m4af_add_itmf_long_tag(ctx
, "iTunSMPB", buf
);
607 uint32_t m4af_update_box_size(m4af_ctx_t
*ctx
, int64_t pos
)
609 int64_t current_pos
= m4af_tell(ctx
);
610 m4af_set_pos(ctx
, pos
);
611 m4af_write32(ctx
, current_pos
- pos
);
612 m4af_set_pos(ctx
, current_pos
);
613 return current_pos
- pos
;
617 void m4af_write_descriptor(m4af_ctx_t
*ctx
, uint32_t tag
, uint32_t size
)
621 buf
[1] = ((size
>> 21) | 0x80);
622 buf
[2] = ((size
>> 14) | 0x80);
623 buf
[3] = ((size
>> 7) | 0x80);
624 buf
[4] = (size
& 0x7f);
625 m4af_write(ctx
, buf
, 5);
629 void m4af_write_ftyp_box(m4af_ctx_t
*ctx
)
631 m4af_write(ctx
, "\0\0\0\040""ftypM4A \0\0\0\0M4A mp42isom\0\0\0\0", 32);
635 void m4af_write_free_box(m4af_ctx_t
*ctx
, uint32_t size
)
637 int64_t pos
= m4af_tell(ctx
);
638 m4af_write32(ctx
, size
+ 8);
639 m4af_write(ctx
, "free", 4);
641 m4af_set_pos(ctx
, pos
+ size
+ 8);
644 int m4af_begin_write(m4af_ctx_t
*ctx
)
646 m4af_write_ftyp_box(ctx
);
647 m4af_write_free_box(ctx
, 0);
648 m4af_write(ctx
, "\0\0\0\0mdat", 8);
649 ctx
->mdat_pos
= m4af_tell(ctx
);
650 return ctx
->last_error
;
654 void m4af_write_stco_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
656 m4af_track_t
*track
= &ctx
->track
[track_idx
];
658 m4af_chunk_entry_t
*index
= track
->chunk_table
;
659 int is_co64
= (ctx
->mdat_pos
+ ctx
->mdat_size
> UINT32_MAX
);
660 int64_t pos
= m4af_tell(ctx
);
662 m4af_write32(ctx
, 0); /* size */
663 m4af_write(ctx
, is_co64
? "co64" : "stco", 4);
664 m4af_write32(ctx
, 0); /* version and flags */
665 m4af_write32(ctx
, track
->num_chunks
);
666 for (i
= 0; i
< track
->num_chunks
; ++i
, ++index
) {
668 m4af_write64(ctx
, index
->offset
);
670 m4af_write32(ctx
, index
->offset
);
672 m4af_update_box_size(ctx
, pos
);
676 void m4af_write_stsz_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
678 m4af_track_t
*track
= &ctx
->track
[track_idx
];
679 m4af_sample_entry_t
*index
= track
->sample_table
;
681 int64_t pos
= m4af_tell(ctx
);
683 "\0\0\0\0" /* size */
687 "\0\0\0\0" /* sample_size: 0(variable) */
689 m4af_write32(ctx
, track
->num_samples
);
690 for (i
= 0; i
< track
->num_samples
; ++i
, ++index
)
691 m4af_write32(ctx
, index
->size
);
692 m4af_update_box_size(ctx
, pos
);
696 void m4af_write_stsc_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
698 m4af_track_t
*track
= &ctx
->track
[track_idx
];
699 m4af_chunk_entry_t
*index
= track
->chunk_table
;
700 uint32_t i
, prev_samples_per_chunk
= 0, entry_count
= 0;
701 int64_t pos
= m4af_tell(ctx
);
703 "\0\0\0\0" /* size */
707 "\0\0\0\0" /* entry_count */
710 for (i
= 0; i
< track
->num_chunks
; ++i
, ++index
) {
711 if (index
->samples_per_chunk
!= prev_samples_per_chunk
) {
713 m4af_write32(ctx
, i
+ 1);
714 m4af_write32(ctx
, index
->samples_per_chunk
);
715 m4af_write32(ctx
, 1); /* sample_description_index */
716 prev_samples_per_chunk
= index
->samples_per_chunk
;
719 m4af_write32_at(ctx
, pos
+ 12, entry_count
);
720 m4af_update_box_size(ctx
, pos
);
724 void m4af_write_stts_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
726 m4af_track_t
*track
= &ctx
->track
[track_idx
];
727 m4af_sample_entry_t
*index
= track
->sample_table
;
728 uint32_t i
, prev_delta
= 0, entry_count
= 0, sample_count
= 0;
729 int64_t pos
= m4af_tell(ctx
);
731 "\0\0\0\0" /* size */
735 "\0\0\0\0" /* entry_count */
738 for (i
= 0; i
< track
->num_samples
; ++i
, ++index
) {
739 if (index
->delta
== prev_delta
)
744 m4af_write32(ctx
, sample_count
);
745 m4af_write32(ctx
, prev_delta
);
747 prev_delta
= index
->delta
;
752 m4af_write32(ctx
, sample_count
);
753 m4af_write32(ctx
, prev_delta
);
755 m4af_write32_at(ctx
, pos
+ 12, entry_count
);
756 m4af_update_box_size(ctx
, pos
);
760 void m4af_write_esds_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
762 m4af_track_t
*track
= &ctx
->track
[track_idx
];
763 int64_t pos
= m4af_tell(ctx
);
764 m4af_write(ctx
, "\0\0\0\0esds", 8);
765 m4af_write32(ctx
, 0); /* version + flags */
768 m4af_write_descriptor(ctx
, 3, 32 + track
->decSpecificInfoSize
);
769 m4af_write(ctx
, "\0\0\0", 3);
770 /* DecoderConfigDescriptor */
771 m4af_write_descriptor(ctx
, 4, 18 + track
->decSpecificInfoSize
);
773 "\x40" /* objectTypeIndication: 0x40(Audio ISO/IEC 14496-3)*/
774 "\x15" /* streamType(6): 0x05(AudioStream)
779 m4af_write24(ctx
, track
->bufferSizeDB
);
780 m4af_write32(ctx
, track
->maxBitrate
);
781 m4af_write32(ctx
, track
->is_vbr
? 0: track
->avgBitrate
);
782 /* DecoderSpecificInfo */
783 m4af_write_descriptor(ctx
, 5, track
->decSpecificInfoSize
);
784 m4af_write(ctx
, track
->decSpecificInfo
, track
->decSpecificInfoSize
);
785 /* SLConfigDescriptor */
786 m4af_write_descriptor(ctx
, 6, 1);
787 m4af_write(ctx
, "\002", 1); /* predefined */
789 m4af_update_box_size(ctx
, pos
);
793 void m4af_write_alac_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
795 m4af_track_t
*track
= &ctx
->track
[track_idx
];
796 int64_t pos
= m4af_tell(ctx
);
798 "\0\0\0\0" /* size */
803 m4af_write(ctx
, track
->decSpecificInfo
, track
->decSpecificInfoSize
);
804 m4af_update_box_size(ctx
, pos
);
808 void m4af_write_mp4a_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
810 m4af_track_t
*track
= &ctx
->track
[track_idx
];
811 int64_t pos
= m4af_tell(ctx
);
812 m4af_write32(ctx
, 0); /* size */
813 m4af_write32(ctx
, track
->codec
); /* mp4a or alac */
815 "\0\0\0\0\0\0" /* reserved */
816 "\0\001" /* data_reference_index: 1 */
817 "\0\0\0\0" /* reserved[0] */
818 "\0\0\0\0" /* reserved[1] */
819 "\0\002" /* channelcount: 2 */
820 "\0\020" /* samplesize: 16 */
821 "\0\0" /* pre_defined */
822 "\0\0" /* reserved */
824 if (track
->codec
== M4AF_FOURCC('m','p','4','a')) {
825 m4af_write32(ctx
, track
->timescale
<< 16);
826 m4af_write_esds_box(ctx
, track_idx
);
828 m4af_write32(ctx
, 44100 << 16);
829 m4af_write_alac_box(ctx
, track_idx
);
831 m4af_update_box_size(ctx
, pos
);
835 void m4af_write_stsd_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
837 int64_t pos
= m4af_tell(ctx
);
838 m4af_write(ctx
, "\0\0\0\0stsd", 8);
842 "\0\0\0\001" /* entry_count: 1 */
844 m4af_write_mp4a_box(ctx
, track_idx
);
845 m4af_update_box_size(ctx
, pos
);
849 void m4af_write_sbgp_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
851 m4af_track_t
*track
= &ctx
->track
[track_idx
];
853 "\0\0\0\034" /* size: 28 */
857 "roll" /* grouping_type */
858 "\0\0\0\001" /* entry_count: 1 */
860 m4af_write32(ctx
, track
->num_samples
);
861 m4af_write32(ctx
, 1); /* group_description_index */
865 void m4af_write_sgpd_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
868 "\0\0\0\026" /* size: 22 */
872 "roll" /* grouping_type */
873 "\0\0\0\001" /* entry_count: 1 */
874 "\377\377" /* payload_data: -1 */
879 void m4af_write_stbl_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
881 m4af_track_t
*track
= &ctx
->track
[track_idx
];
882 int64_t pos
= m4af_tell(ctx
);
883 m4af_write(ctx
, "\0\0\0\0stbl", 8);
884 m4af_write_stsd_box(ctx
, track_idx
);
885 if ((ctx
->priming_mode
& M4AF_PRIMING_MODE_EDTS
) &&
886 (track
->encoder_delay
|| track
->padding
)) {
887 m4af_write_sbgp_box(ctx
, track_idx
);
888 m4af_write_sgpd_box(ctx
, track_idx
);
890 m4af_write_stts_box(ctx
, track_idx
);
891 m4af_write_stsc_box(ctx
, track_idx
);
892 m4af_write_stsz_box(ctx
, track_idx
);
893 m4af_write_stco_box(ctx
, track_idx
);
894 m4af_update_box_size(ctx
, pos
);
898 void m4af_write_url_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
901 "\0\0\0\014" /* size */
904 "\0\0\001" /* flags: 1(in the same file) */
909 void m4af_write_dref_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
911 int64_t pos
= m4af_tell(ctx
);
912 m4af_write(ctx
, "\0\0\0\0dref", 8);
916 "\0\0\0\001" /* entry_count: 1 */
918 m4af_write_url_box(ctx
, track_idx
);
919 m4af_update_box_size(ctx
, pos
);
923 void m4af_write_dinf_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
925 int64_t pos
= m4af_tell(ctx
);
926 m4af_write(ctx
, "\0\0\0\0dinf", 8);
927 m4af_write_dref_box(ctx
, track_idx
);
928 m4af_update_box_size(ctx
, pos
);
932 void m4af_write_smhd_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
935 "\0\0\0\020" /* size */
940 "\0\0" /* reserved */
945 void m4af_write_minf_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
947 m4af_track_t
*track
= &ctx
->track
[track_idx
];
948 int64_t pos
= m4af_tell(ctx
);
949 m4af_write(ctx
, "\0\0\0\0minf", 8);
950 /* TODO: add TEXT support */
951 if (track
->codec
!= M4AF_CODEC_TEXT
)
952 m4af_write_smhd_box(ctx
, track_idx
);
953 m4af_write_dinf_box(ctx
, track_idx
);
954 m4af_write_stbl_box(ctx
, track_idx
);
955 m4af_update_box_size(ctx
, pos
);
959 void m4af_write_mdhd_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
961 m4af_track_t
*track
= &ctx
->track
[track_idx
];
962 int64_t pos
= m4af_tell(ctx
);
963 uint8_t version
= (track
->creation_time
> UINT32_MAX
||
964 track
->modification_time
> UINT32_MAX
||
965 track
->duration
> UINT32_MAX
);
967 m4af_write(ctx
, "\0\0\0\0mdhd", 8);
968 m4af_write(ctx
, &version
, 1);
969 m4af_write(ctx
, "\0\0\0", 3); /* flags */
971 m4af_write64(ctx
, track
->creation_time
);
972 m4af_write64(ctx
, track
->modification_time
);
973 m4af_write32(ctx
, track
->timescale
);
974 m4af_write64(ctx
, track
->duration
);
976 m4af_write32(ctx
, track
->creation_time
);
977 m4af_write32(ctx
, track
->modification_time
);
978 m4af_write32(ctx
, track
->timescale
);
979 m4af_write32(ctx
, track
->duration
);
982 "\x55\xc4" /* language: und */
983 "\0\0" /* pre_defined */
985 m4af_update_box_size(ctx
, pos
);
989 void m4af_write_hdlr_box(m4af_ctx_t
*ctx
, uint32_t track_idx
, const char *type
)
991 int64_t pos
= m4af_tell(ctx
);
992 static const char reserved_and_name
[10] = { 0 };
995 "\0\0\0\0" /* size */
999 "\0\0\0\0" /* pre_defined */
1001 m4af_write(ctx
, type
, 4); /* handler_type */
1003 m4af_write(ctx
, !strcmp(type
, "mdir") ? "appl" : "\0\0\0\0", 4);
1004 /* reserved[1], reserved[2], name */
1005 m4af_write(ctx
, reserved_and_name
, (pos
& 1) ? 9 : 10);
1006 m4af_update_box_size(ctx
, pos
);
1010 void m4af_write_mdia_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
1012 m4af_track_t
*track
= &ctx
->track
[track_idx
];
1014 (track
->codec
== M4AF_CODEC_TEXT
) ? "text" : "soun";
1015 int64_t pos
= m4af_tell(ctx
);
1016 m4af_write(ctx
, "\0\0\0\0mdia", 8);
1017 m4af_write_mdhd_box(ctx
, track_idx
);
1018 m4af_write_hdlr_box(ctx
, track_idx
, hdlr
);
1019 m4af_write_minf_box(ctx
, track_idx
);
1020 m4af_update_box_size(ctx
, pos
);
1024 void m4af_write_elst_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
1026 m4af_track_t
*track
= &ctx
->track
[track_idx
];
1028 int64_t duration
= track
->duration
- track
->encoder_delay
- track
->padding
;
1029 int64_t pos
= m4af_tell(ctx
);
1030 duration
= (double)duration
/ track
->timescale
* ctx
->timescale
+ .5;
1031 version
= (duration
> UINT32_MAX
);
1033 m4af_write(ctx
, "\0\0\0\0elst", 8);
1034 m4af_write(ctx
, &version
, 1);
1035 m4af_write(ctx
, "\0\0\0", 3); /* flags */
1036 m4af_write32(ctx
, 1); /* entry_count: 1 */
1038 m4af_write64(ctx
, duration
);
1039 m4af_write64(ctx
, track
->encoder_delay
);
1041 m4af_write32(ctx
, duration
);
1042 m4af_write32(ctx
, track
->encoder_delay
);
1044 m4af_write16(ctx
, 1); /* media_rate_integer */
1045 m4af_write16(ctx
, 0); /* media_rate_fraction */
1046 m4af_update_box_size(ctx
, pos
);
1050 void m4af_write_edts_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
1052 int64_t pos
= m4af_tell(ctx
);
1053 m4af_write(ctx
, "\0\0\0\0edts", 8);
1054 m4af_write_elst_box(ctx
, track_idx
);
1055 m4af_update_box_size(ctx
, pos
);
1059 void m4af_write_tkhd_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
1061 m4af_track_t
*track
= &ctx
->track
[track_idx
];
1062 int64_t pos
= m4af_tell(ctx
);
1064 (double)track
->duration
/ track
->timescale
* ctx
->timescale
+ .5;
1065 uint8_t version
= (track
->creation_time
> UINT32_MAX
||
1066 track
->modification_time
> UINT32_MAX
||
1067 duration
> UINT32_MAX
);
1068 m4af_write(ctx
, "\0\0\0\0tkhd", 8);
1069 m4af_write(ctx
, &version
, 1);
1070 m4af_write(ctx
, "\0\0\007", 3); /* flags */
1072 m4af_write64(ctx
, track
->creation_time
);
1073 m4af_write64(ctx
, track
->modification_time
);
1074 m4af_write32(ctx
, track_idx
+ 1);
1075 m4af_write(ctx
, "\0\0\0\0" /* reserved */
1077 m4af_write64(ctx
, duration
);
1079 m4af_write32(ctx
, track
->creation_time
);
1080 m4af_write32(ctx
, track
->modification_time
);
1081 m4af_write32(ctx
, track_idx
+ 1);
1082 m4af_write(ctx
, "\0\0\0\0" /* reserved */
1084 m4af_write32(ctx
, duration
);
1087 "\0\0\0\0" /* reserved[0] */
1088 "\0\0\0\0" /* reserved[1] */
1090 "\0\0" /* alternate_group */
1091 "\001\0" /* volume: 1.0 */
1092 "\0\0" /* reserved */
1093 "\0\001\0\0" /* matrix[0] */
1094 "\0\0\0\0" /* matrix[1] */
1095 "\0\0\0\0" /* matrix[2] */
1096 "\0\0\0\0" /* matrix[3] */
1097 "\0\001\0\0" /* matrix[4] */
1098 "\0\0\0\0" /* matrix[5] */
1099 "\0\0\0\0" /* matrix[6] */
1100 "\0\0\0\0" /* matrix[7] */
1101 "\100\0\0\0" /* matrix[8] */
1102 "\0\0\0\0" /* width */
1103 "\0\0\0\0" /* height */
1105 m4af_update_box_size(ctx
, pos
);
1109 void m4af_write_trak_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
1111 m4af_track_t
*track
= &ctx
->track
[track_idx
];
1112 int64_t pos
= m4af_tell(ctx
);
1113 m4af_write(ctx
, "\0\0\0\0trak", 8);
1114 m4af_write_tkhd_box(ctx
, track_idx
);
1115 if ((ctx
->priming_mode
& M4AF_PRIMING_MODE_EDTS
) &&
1116 (track
->encoder_delay
|| track
->padding
))
1117 m4af_write_edts_box(ctx
, track_idx
);
1118 m4af_write_mdia_box(ctx
, track_idx
);
1119 m4af_update_box_size(ctx
, pos
);
1123 int64_t m4af_movie_duration(m4af_ctx_t
*ctx
)
1125 int64_t movie_duration
= 0;
1127 for (i
= 0; i
< ctx
->num_tracks
; ++i
) {
1128 double x
= ctx
->track
[i
].duration
;
1129 int64_t duration
= x
/ ctx
->track
[i
].timescale
* ctx
->timescale
+ .5;
1130 if (duration
> movie_duration
)
1131 movie_duration
= duration
;
1133 return movie_duration
;
1137 void m4af_write_mvhd_box(m4af_ctx_t
*ctx
)
1139 int64_t pos
= m4af_tell(ctx
);
1140 int64_t movie_duration
= m4af_movie_duration(ctx
);
1141 uint8_t version
= (ctx
->creation_time
> UINT32_MAX
||
1142 ctx
->modification_time
> UINT32_MAX
||
1143 movie_duration
> UINT32_MAX
);
1145 m4af_write(ctx
, "\0\0\0\0mvhd", 8);
1146 m4af_write(ctx
, &version
, 1);
1147 m4af_write(ctx
, "\0\0\0", 3); /* flags */
1149 m4af_write64(ctx
, ctx
->creation_time
);
1150 m4af_write64(ctx
, ctx
->modification_time
);
1151 m4af_write32(ctx
, ctx
->timescale
);
1152 m4af_write64(ctx
, movie_duration
);
1154 m4af_write32(ctx
, ctx
->creation_time
);
1155 m4af_write32(ctx
, ctx
->modification_time
);
1156 m4af_write32(ctx
, ctx
->timescale
);
1157 m4af_write32(ctx
, movie_duration
);
1160 "\0\001\0\0" /* rate: 1.0 */
1161 "\001\0" /* volume: 1.0 */
1162 "\0\0" /* reserved */
1163 "\0\0\0\0" /* reserved[0] */
1164 "\0\0\0\0" /* reserved[1] */
1165 "\0\001\0\0" /* matrix[0] */
1166 "\0\0\0\0" /* matrix[1] */
1167 "\0\0\0\0" /* matrix[2] */
1168 "\0\0\0\0" /* matrix[3] */
1169 "\0\001\0\0" /* matrix[4] */
1170 "\0\0\0\0" /* matrix[5] */
1171 "\0\0\0\0" /* matrix[6] */
1172 "\0\0\0\0" /* matrix[7] */
1173 "\100\0\0\0" /* matrix[8] */
1174 "\0\0\0\0" /* pre_defined[0] */
1175 "\0\0\0\0" /* pre_defined[1] */
1176 "\0\0\0\0" /* pre_defined[2] */
1177 "\0\0\0\0" /* pre_defined[3] */
1178 "\0\0\0\0" /* pre_defined[4] */
1179 "\0\0\0\0" /* pre_defined[5] */
1181 m4af_write32(ctx
, ctx
->num_tracks
+ 1);
1182 m4af_update_box_size(ctx
, pos
);
1186 void m4af_write_mean_box(m4af_ctx_t
*ctx
)
1189 "\0\0\0\034" /* size */
1192 "\0\0\0" /* flags */
1193 "com.apple.iTunes" /* meaning-string */
1198 void m4af_write_name_box(m4af_ctx_t
*ctx
, const char *name
)
1200 int64_t pos
= m4af_tell(ctx
);
1202 "\0\0\0\0" /* size */
1205 "\0\0\0" /* flags */
1207 m4af_write(ctx
, name
, strlen(name
));
1208 m4af_update_box_size(ctx
, pos
);
1212 void m4af_write_data_box(m4af_ctx_t
*ctx
, uint32_t type_code
,
1213 const char *data
, uint32_t data_size
)
1215 int64_t pos
= m4af_tell(ctx
);
1216 uint8_t code
= type_code
;
1218 "\0\0\0\0" /* size */
1220 "\0\0" /* reserved */
1221 "\0" /* type_set_indifier */
1223 m4af_write(ctx
, &code
, 1);
1224 m4af_write(ctx
, "\0\0\0\0", 4); /* locale */
1225 m4af_write(ctx
, data
, data_size
);
1226 m4af_update_box_size(ctx
, pos
);
1230 void m4af_write_metadata(m4af_ctx_t
*ctx
, m4af_itmf_entry_t
*entry
)
1232 int64_t pos
= m4af_tell(ctx
);
1233 m4af_write(ctx
, "\0\0\0\0", 4);
1234 m4af_write32(ctx
, entry
->fcc
);
1235 if (entry
->fcc
!= M4AF_FOURCC('-','-','-','-'))
1236 m4af_write_data_box(ctx
, entry
->type_code
,
1237 entry
->data
, entry
->data_size
);
1239 m4af_write_mean_box(ctx
);
1240 m4af_write_name_box(ctx
, entry
->name
);
1241 m4af_write_data_box(ctx
, 1, entry
->data
, entry
->data_size
);
1243 m4af_update_box_size(ctx
, pos
);
1247 void m4af_write_ilst_box(m4af_ctx_t
*ctx
)
1250 int64_t pos
= m4af_tell(ctx
);
1251 m4af_write(ctx
, "\0\0\0\0ilst", 8);
1252 for (i
= 0; i
< ctx
->num_tags
; ++i
)
1253 m4af_write_metadata(ctx
, &ctx
->itmf_table
[i
]);
1254 m4af_update_box_size(ctx
, pos
);
1258 void m4af_write_meta_box(m4af_ctx_t
*ctx
)
1260 int64_t pos
= m4af_tell(ctx
);
1262 "\0\0\0\0" /* size */
1265 "\0\0\0" /* flags */
1267 m4af_write_hdlr_box(ctx
, 0, "mdir");
1268 m4af_write_ilst_box(ctx
);
1269 m4af_update_box_size(ctx
, pos
);
1273 void m4af_write_udta_box(m4af_ctx_t
*ctx
)
1275 int64_t pos
= m4af_tell(ctx
);
1276 m4af_write(ctx
, "\0\0\0\0udta", 8);
1277 m4af_write_meta_box(ctx
);
1278 m4af_update_box_size(ctx
, pos
);
1282 uint32_t m4af_write_moov_box(m4af_ctx_t
*ctx
)
1285 int64_t pos
= m4af_tell(ctx
);
1286 m4af_write(ctx
, "\0\0\0\0moov", 8);
1287 m4af_write_mvhd_box(ctx
);
1288 for (i
= 0; i
< ctx
->num_tracks
; ++i
)
1289 m4af_write_trak_box(ctx
, i
);
1291 m4af_write_udta_box(ctx
);
1292 return m4af_update_box_size(ctx
, pos
);
1296 void m4af_finalize_mdat(m4af_ctx_t
*ctx
)
1298 if (ctx
->mdat_size
+ 8 > UINT32_MAX
) {
1299 m4af_set_pos(ctx
, ctx
->mdat_pos
- 16);
1300 m4af_write32(ctx
, 1);
1301 m4af_write(ctx
, "mdat", 4);
1302 m4af_write64(ctx
, ctx
->mdat_size
+ 16);
1304 m4af_set_pos(ctx
, ctx
->mdat_pos
- 8);
1305 m4af_write32(ctx
, ctx
->mdat_size
+ 8);
1307 m4af_set_pos(ctx
, ctx
->mdat_pos
+ ctx
->mdat_size
);
1311 void m4af_shift_mdat_pos(m4af_ctx_t
*ctx
, uint32_t offset
)
1317 end
= ctx
->mdat_pos
+ ctx
->mdat_size
;
1318 for (; (begin
= m4af_max(ctx
->mdat_pos
, end
- 8192)) < end
; end
= begin
) {
1319 m4af_set_pos(ctx
, begin
);
1320 ctx
->io
.read(ctx
->io_cookie
, buf
, end
- begin
);
1321 m4af_set_pos(ctx
, begin
+ offset
);
1322 m4af_write(ctx
, buf
, end
- begin
);
1324 for (i
= 0; i
< ctx
->num_tracks
; ++i
)
1325 for (j
= 0; j
< ctx
->track
[i
].num_chunks
; ++j
)
1326 ctx
->track
[i
].chunk_table
[j
].offset
+= offset
;
1327 ctx
->mdat_pos
+= offset
;
1328 m4af_set_pos(ctx
, ctx
->mdat_pos
- 16);
1329 m4af_write_free_box(ctx
, 0);
1330 m4af_write(ctx
, "\0\0\0\0mdat", 8);
1331 m4af_finalize_mdat(ctx
);
1334 int m4af_finalize(m4af_ctx_t
*ctx
, int optimize
)
1337 m4af_track_t
*track
;
1340 for (i
= 0; i
< ctx
->num_tracks
; ++i
) {
1341 track
= ctx
->track
+ i
;
1342 if (track
->duration
) {
1343 int64_t track_size
= 0;
1345 for (j
= 0; j
< track
->num_chunks
; ++j
)
1346 track_size
+= track
->chunk_table
[j
].size
;
1348 8.0 * track_size
* track
->timescale
/ track
->duration
+ .5;
1350 m4af_flush_chunk(ctx
, i
);
1353 if ((ctx
->priming_mode
& M4AF_PRIMING_MODE_ITUNSMPB
) &&
1354 (track
->encoder_delay
|| track
->padding
))
1355 m4af_set_iTunSMPB(ctx
);
1356 m4af_finalize_mdat(ctx
);
1357 moov_size
= m4af_write_moov_box(ctx
);
1360 m4af_shift_mdat_pos(ctx
, moov_size
+ 1024);
1361 m4af_set_pos(ctx
, 32);
1362 m4af_write_moov_box(ctx
);
1363 pos
= m4af_tell(ctx
);
1364 m4af_write_free_box(ctx
, ctx
->mdat_pos
- pos
- 24);
1366 return ctx
->last_error
;