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
;
57 m4af_sample_entry_t
*sample_table
;
59 uint32_t sample_table_capacity
;
61 m4af_chunk_entry_t
*chunk_table
;
63 uint32_t chunk_table_capacity
;
65 uint8_t *chunk_buffer
;
67 uint32_t chunk_capacity
;
69 /* temporary, to help parsing */
79 int64_t creation_time
;
80 int64_t modification_time
;
86 m4af_itmf_entry_t
*itmf_table
;
88 uint32_t itmf_table_capacity
;
90 m4af_io_callbacks_t io
;
94 m4af_track_t track
[2];
96 m4af_itmf_entry_t current_tag
;
99 typedef struct m4af_box_parser_t
{
101 int (*handler
)(m4af_ctx_t
*ctx
, uint32_t name
, uint64_t size
);
105 int64_t m4af_timestamp(void)
107 return (int64_t)(time(0)) + (((1970 - 1904) * 365) + 17) * 24 * 60 * 60;
111 * http://graphics.stanford.edu/~seander/bithacks.html#RoundUpPowerOf2
114 uint32_t m4af_roundup(uint32_t n
)
127 int64_t m4af_tell(m4af_ctx_t
*ctx
)
130 if ((pos
= ctx
->io
.tell(ctx
->io_cookie
)) < 0)
131 ctx
->last_error
= M4AF_IO_ERROR
;
136 int m4af_set_pos(m4af_ctx_t
*ctx
, int64_t pos
)
139 if ((rc
= ctx
->io
.seek(ctx
->io_cookie
, pos
, SEEK_SET
)) < 0)
140 ctx
->last_error
= M4AF_IO_ERROR
;
145 int m4af_write(m4af_ctx_t
*ctx
, const void *data
, uint32_t size
)
148 if ((rc
= ctx
->io
.write(ctx
->io_cookie
, data
, size
)) < 0)
149 ctx
->last_error
= M4AF_IO_ERROR
;
154 int m4af_write16(m4af_ctx_t
*ctx
, uint32_t data
)
156 data
= m4af_htob16(data
);
157 return m4af_write(ctx
, &data
, 2);
161 int m4af_write32(m4af_ctx_t
*ctx
, uint32_t data
)
163 data
= m4af_htob32(data
);
164 return m4af_write(ctx
, &data
, 4);
168 int m4af_write64(m4af_ctx_t
*ctx
, uint64_t data
)
170 data
= m4af_htob64(data
);
171 return m4af_write(ctx
, &data
, 8);
175 int m4af_write24(m4af_ctx_t
*ctx
, uint32_t data
)
177 data
= m4af_htob32(data
<< 8);
178 return m4af_write(ctx
, &data
, 3);
182 void m4af_write32_at(m4af_ctx_t
*ctx
, int64_t pos
, uint32_t value
)
184 int64_t current_pos
= m4af_tell(ctx
);
185 m4af_set_pos(ctx
, pos
);
186 m4af_write32(ctx
, value
);
187 m4af_set_pos(ctx
, current_pos
);
190 m4af_ctx_t
*m4af_create(uint32_t codec
, uint32_t timescale
,
191 m4af_io_callbacks_t
*io
, void *io_cookie
)
196 if (codec
!= M4AF_FOURCC('m','p','4','a') &&
197 codec
!= M4AF_FOURCC('a','l','a','c'))
199 if ((ctx
= m4af_realloc(0, sizeof(m4af_ctx_t
))) == 0)
201 memset(ctx
, 0, sizeof(m4af_ctx_t
));
202 memcpy(&ctx
->io
, io
, sizeof(m4af_io_callbacks_t
));
203 ctx
->io_cookie
= io_cookie
;
204 ctx
->timescale
= timescale
;
205 timestamp
= m4af_timestamp();
206 ctx
->creation_time
= timestamp
;
207 ctx
->modification_time
= timestamp
;
209 ctx
->track
[0].codec
= codec
;
210 ctx
->track
[0].timescale
= timescale
;
211 ctx
->track
[0].creation_time
= timestamp
;
212 ctx
->track
[0].modification_time
= timestamp
;
217 void m4af_free_itmf_table(m4af_ctx_t
*ctx
)
220 m4af_itmf_entry_t
*entry
= ctx
->itmf_table
;
221 for (i
= 0; i
< ctx
->num_tags
; ++i
, ++entry
) {
222 if (entry
->fcc
== M4AF_FOURCC('-','-','-','-'))
223 m4af_free(entry
->name
);
224 m4af_free(entry
->data
);
226 m4af_free(ctx
->itmf_table
);
230 void m4af_clear_track(m4af_ctx_t
*ctx
, int track_idx
)
232 m4af_track_t
*track
= ctx
->track
+ track_idx
;
233 if (track
->decSpecificInfo
)
234 m4af_free(track
->decSpecificInfo
);
235 if (track
->sample_table
)
236 m4af_free(track
->sample_table
);
237 if (track
->chunk_table
)
238 m4af_free(track
->chunk_table
);
239 if (track
->chunk_buffer
)
240 m4af_free(track
->chunk_buffer
);
241 memset(track
, 0, sizeof(m4af_track_t
));
244 void m4af_teardown(m4af_ctx_t
**ctxp
)
247 m4af_ctx_t
*ctx
= *ctxp
;
248 for (i
= 0; i
< ctx
->num_tracks
; ++i
)
249 m4af_clear_track(ctx
, i
);
251 m4af_free_itmf_table(ctx
);
256 void m4af_set_fixed_frame_duration(m4af_ctx_t
*ctx
, uint32_t track_idx
,
259 ctx
->track
[track_idx
].frame_duration
= length
;
262 int m4af_set_decoder_specific_info(m4af_ctx_t
*ctx
, uint32_t track_idx
,
263 uint8_t *data
, uint32_t size
)
265 m4af_track_t
*track
= &ctx
->track
[track_idx
];
266 if (size
> track
->decSpecificInfoSize
) {
267 uint8_t *memory
= m4af_realloc(track
->decSpecificInfo
, size
);
269 ctx
->last_error
= M4AF_NO_MEMORY
;
272 track
->decSpecificInfo
= memory
;
275 memcpy(track
->decSpecificInfo
, data
, size
);
276 track
->decSpecificInfoSize
= size
;
278 return ctx
->last_error
;
281 void m4af_set_priming(m4af_ctx_t
*ctx
, uint32_t track_idx
,
282 uint32_t encoder_delay
, uint32_t padding
)
284 m4af_track_t
*track
= &ctx
->track
[track_idx
];
285 track
->encoder_delay
= encoder_delay
;
286 track
->padding
= padding
;
289 void m4af_set_priming_mode(m4af_ctx_t
*ctx
, int mode
)
291 ctx
->priming_mode
= mode
;
295 int m4af_add_sample_entry(m4af_ctx_t
*ctx
, uint32_t track_idx
,
296 uint32_t size
, uint32_t delta
)
298 m4af_track_t
*track
= &ctx
->track
[track_idx
];
299 m4af_sample_entry_t
*entry
;
303 if (track
->num_samples
== track
->sample_table_capacity
) {
304 uint32_t new_size
= track
->sample_table_capacity
;
305 new_size
= new_size
? new_size
* 2 : 1;
306 entry
= m4af_realloc(track
->sample_table
, new_size
* sizeof(*entry
));
308 ctx
->last_error
= M4AF_NO_MEMORY
;
311 track
->sample_table
= entry
;
312 track
->sample_table_capacity
= new_size
;
314 entry
= track
->sample_table
+ track
->num_samples
;
316 entry
->delta
= delta
;
317 ++track
->num_samples
;
322 int m4af_flush_chunk(m4af_ctx_t
*ctx
, uint32_t track_idx
)
324 m4af_track_t
*track
= &ctx
->track
[track_idx
];
325 m4af_chunk_entry_t
*entry
;
326 if (!track
->num_chunks
|| !track
->chunk_size
)
328 entry
= &track
->chunk_table
[track
->num_chunks
- 1];
329 entry
->offset
= m4af_tell(ctx
);
330 m4af_write(ctx
, track
->chunk_buffer
, track
->chunk_size
);
331 ctx
->mdat_size
+= track
->chunk_size
;
332 track
->chunk_size
= 0;
333 return ctx
->last_error
? -1 : 0;
337 int m4af_add_chunk_entry(m4af_ctx_t
*ctx
, uint32_t track_idx
)
339 m4af_track_t
*track
= &ctx
->track
[track_idx
];
340 m4af_chunk_entry_t
*entry
;
341 if (track
->num_chunks
== track
->chunk_table_capacity
) {
342 uint32_t new_size
= track
->chunk_table_capacity
;
343 new_size
= new_size
? new_size
* 2 : 1;
344 entry
= m4af_realloc(track
->chunk_table
, new_size
* sizeof(*entry
));
346 ctx
->last_error
= M4AF_NO_MEMORY
;
349 track
->chunk_table
= entry
;
350 track
->chunk_table_capacity
= new_size
;
352 memset(&track
->chunk_table
[track
->num_chunks
++], 0,
353 sizeof(m4af_chunk_entry_t
));
358 int m4af_update_chunk_table(m4af_ctx_t
*ctx
, uint32_t track_idx
,
359 uint32_t size
, uint32_t delta
)
361 m4af_track_t
*track
= &ctx
->track
[track_idx
];
362 m4af_chunk_entry_t
*entry
;
363 int add_new_chunk
= 0;
367 if (track
->num_chunks
== 0)
370 entry
= &track
->chunk_table
[track
->num_chunks
- 1];
371 if (entry
->duration
+ delta
> track
->timescale
/ 2)
375 m4af_flush_chunk(ctx
, track_idx
);
376 if (m4af_add_chunk_entry(ctx
, track_idx
) < 0)
379 entry
= &track
->chunk_table
[track
->num_chunks
- 1];
381 ++entry
->samples_per_chunk
;
382 entry
->duration
+= delta
;
387 void m4af_update_max_bitrate(m4af_ctx_t
*ctx
, uint32_t track_idx
)
389 m4af_track_t
*track
= &ctx
->track
[track_idx
];
390 uint32_t duration
= 0, size
= 0, bitrate
;
391 m4af_sample_entry_t
*ent
= track
->sample_table
+ track
->num_samples
- 1;
393 for (; ent
>= track
->sample_table
&& duration
< track
->timescale
; --ent
) {
394 duration
+= ent
->delta
;
397 bitrate
= (uint32_t)(size
* 8.0 * track
->timescale
/ duration
+ .5);
398 if (bitrate
> track
->maxBitrate
)
399 track
->maxBitrate
= bitrate
;
403 int m4af_append_sample_to_chunk(m4af_ctx_t
*ctx
, uint32_t track_idx
,
404 const void *data
, uint32_t size
)
406 m4af_track_t
*track
= &ctx
->track
[track_idx
];
407 uint32_t newsize
= track
->chunk_size
+ size
;
411 if (track
->chunk_capacity
< newsize
) {
412 uint32_t capacity
= m4af_roundup(newsize
);
413 uint8_t *memory
= realloc(track
->chunk_buffer
, capacity
);
415 ctx
->last_error
= M4AF_NO_MEMORY
;
418 track
->chunk_buffer
= memory
;
419 track
->chunk_capacity
= capacity
;
421 memcpy(track
->chunk_buffer
+ track
->chunk_size
, data
, size
);
422 track
->chunk_size
= newsize
;
426 int m4af_write_sample(m4af_ctx_t
*ctx
, uint32_t track_idx
, const void *data
,
427 uint32_t size
, uint32_t duration
)
429 m4af_track_t
*track
= &ctx
->track
[track_idx
];
430 if (track
->frame_duration
)
431 duration
= track
->frame_duration
;
432 if (size
> track
->bufferSizeDB
)
433 track
->bufferSizeDB
= size
;
434 track
->duration
+= duration
;
435 m4af_add_sample_entry(ctx
, track_idx
, size
, duration
);
436 m4af_update_chunk_table(ctx
, track_idx
, size
, duration
);
437 m4af_update_max_bitrate(ctx
, track_idx
);
438 m4af_append_sample_to_chunk(ctx
, track_idx
, data
, size
);
439 return ctx
->last_error
;
443 m4af_itmf_entry_t
*m4af_find_itmf_slot(m4af_ctx_t
*ctx
, uint32_t fcc
,
446 m4af_itmf_entry_t
*entry
= ctx
->itmf_table
;
449 fcc
= M4AF_FOURCC('-','-','-','-');
451 if (fcc
!= M4AF_TAG_ARTWORK
)
452 for (; entry
!= ctx
->itmf_table
+ ctx
->num_tags
; ++entry
)
453 if (fcc
== entry
->fcc
&& (!name
|| !strcmp(name
, entry
->name
)))
456 if (ctx
->num_tags
== ctx
->itmf_table_capacity
) {
457 uint32_t new_size
= ctx
->itmf_table_capacity
;
458 new_size
= new_size
? new_size
* 2 : 1;
459 entry
= m4af_realloc(ctx
->itmf_table
, new_size
* sizeof(*entry
));
461 ctx
->last_error
= M4AF_NO_MEMORY
;
464 ctx
->itmf_table
= entry
;
465 ctx
->itmf_table_capacity
= new_size
;
467 entry
= &ctx
->itmf_table
[ctx
->num_tags
++];
468 memset(entry
, 0, sizeof(m4af_itmf_entry_t
));
471 char *name_copy
= m4af_realloc(0, strlen(name
) + 1);
473 ctx
->last_error
= M4AF_NO_MEMORY
;
477 strcpy(name_copy
, name
);
478 entry
->name
= name_copy
;
483 int m4af_add_itmf_long_tag(m4af_ctx_t
*ctx
, const char *name
,
486 m4af_itmf_entry_t
*entry
;
488 size_t name_len
= strlen(name
);
489 size_t data_len
= strlen(data
);
490 if (!name_len
|| !data_len
)
493 if ((entry
= m4af_find_itmf_slot(ctx
, 0, name
)) == 0)
495 entry
->type_code
= M4AF_UTF8
;
496 if ((data_copy
= m4af_realloc(entry
->data
, data_len
)) == 0) {
497 ctx
->last_error
= M4AF_NO_MEMORY
;
500 memcpy(data_copy
, data
, data_len
);
501 entry
->data
= data_copy
;
502 entry
->data_size
= data_len
;
505 return ctx
->last_error
;
508 int m4af_add_itmf_short_tag(m4af_ctx_t
*ctx
, uint32_t fcc
,
509 uint32_t type_code
, const void *data
,
512 m4af_itmf_entry_t
*entry
;
517 if ((entry
= m4af_find_itmf_slot(ctx
, fcc
, 0)) == 0)
519 entry
->type_code
= type_code
;
520 if ((data_copy
= m4af_realloc(entry
->data
, data_size
)) == 0) {
521 ctx
->last_error
= M4AF_NO_MEMORY
;
524 memcpy(data_copy
, data
, data_size
);
525 entry
->data
= data_copy
;
526 entry
->data_size
= data_size
;
529 return ctx
->last_error
;
532 int m4af_add_itmf_string_tag(m4af_ctx_t
*ctx
, uint32_t fcc
, const char *data
)
534 return m4af_add_itmf_short_tag(ctx
, fcc
, M4AF_UTF8
, data
, strlen(data
));
537 int m4af_add_itmf_int8_tag(m4af_ctx_t
*ctx
, uint32_t fcc
, int value
)
539 uint8_t data
= value
;
540 return m4af_add_itmf_short_tag(ctx
, fcc
, M4AF_INTEGER
, &data
, 1);
543 int m4af_add_itmf_int16_tag(m4af_ctx_t
*ctx
, uint32_t fcc
, int value
)
545 uint16_t data
= m4af_htob16(value
);
546 return m4af_add_itmf_short_tag(ctx
, fcc
, M4AF_INTEGER
, &data
, 2);
549 int m4af_add_itmf_int32_tag(m4af_ctx_t
*ctx
, uint32_t fcc
, uint32_t value
)
551 uint32_t data
= m4af_htob32(value
);
552 return m4af_add_itmf_short_tag(ctx
, fcc
, M4AF_INTEGER
, &data
, 4);
555 int m4af_add_itmf_int64_tag(m4af_ctx_t
*ctx
, uint32_t fcc
, uint64_t value
)
557 uint64_t data
= m4af_htob64(value
);
558 return m4af_add_itmf_short_tag(ctx
, fcc
, M4AF_INTEGER
, &data
, 8);
561 int m4af_add_itmf_track_tag(m4af_ctx_t
*ctx
, int track
, int total
)
563 uint16_t data
[4] = { 0 };
564 data
[1] = m4af_htob16(track
);
565 data
[2] = m4af_htob16(total
);
566 return m4af_add_itmf_short_tag(ctx
, M4AF_FOURCC('t','r','k','n'),
567 M4AF_IMPLICIT
, &data
, 8);
570 int m4af_add_itmf_disk_tag(m4af_ctx_t
*ctx
, int disk
, int total
)
572 uint16_t data
[3] = { 0 };
573 data
[1] = m4af_htob16(disk
);
574 data
[2] = m4af_htob16(total
);
575 return m4af_add_itmf_short_tag(ctx
, M4AF_FOURCC('d','i','s','k'),
576 M4AF_IMPLICIT
, &data
, 6);
579 int m4af_add_itmf_genre_tag(m4af_ctx_t
*ctx
, int genre
)
581 uint16_t data
= m4af_htob16(genre
);
582 return m4af_add_itmf_short_tag(ctx
, M4AF_FOURCC('g','n','r','e'),
583 M4AF_IMPLICIT
, &data
, 2);
587 int m4af_set_iTunSMPB(m4af_ctx_t
*ctx
)
589 const char *fmt
= " 00000000 %08X %08X %08X%08X 00000000 00000000 "
590 "00000000 00000000 00000000 00000000 00000000 00000000";
591 m4af_track_t
*track
= &ctx
->track
[0];
593 uint64_t length
= track
->duration
- track
->encoder_delay
- track
->padding
;
594 sprintf(buf
, fmt
, track
->encoder_delay
, track
->padding
,
595 (uint32_t)(length
>> 32), (uint32_t)length
);
596 return m4af_add_itmf_long_tag(ctx
, "iTunSMPB", buf
);
600 uint32_t m4af_update_box_size(m4af_ctx_t
*ctx
, int64_t pos
)
602 int64_t current_pos
= m4af_tell(ctx
);
603 m4af_set_pos(ctx
, pos
);
604 m4af_write32(ctx
, current_pos
- pos
);
605 m4af_set_pos(ctx
, current_pos
);
606 return current_pos
- pos
;
610 void m4af_write_descriptor(m4af_ctx_t
*ctx
, uint32_t tag
, uint32_t size
)
614 buf
[1] = ((size
>> 21) | 0x80);
615 buf
[2] = ((size
>> 14) | 0x80);
616 buf
[3] = ((size
>> 7) | 0x80);
617 buf
[4] = (size
& 0x7f);
618 m4af_write(ctx
, buf
, 5);
622 void m4af_write_ftyp_box(m4af_ctx_t
*ctx
)
624 m4af_write(ctx
, "\0\0\0\040""ftypM4A \0\0\0\0M4A mp42isom\0\0\0\0", 32);
628 void m4af_write_free_box(m4af_ctx_t
*ctx
, uint32_t size
)
630 int64_t pos
= m4af_tell(ctx
);
631 m4af_write32(ctx
, size
+ 8);
632 m4af_write(ctx
, "free", 4);
634 m4af_set_pos(ctx
, pos
+ size
+ 8);
637 int m4af_begin_write(m4af_ctx_t
*ctx
)
639 m4af_write_ftyp_box(ctx
);
640 m4af_write_free_box(ctx
, 0);
641 m4af_write(ctx
, "\0\0\0\0mdat", 8);
642 ctx
->mdat_pos
= m4af_tell(ctx
);
643 return ctx
->last_error
;
647 void m4af_write_stco_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
649 m4af_track_t
*track
= &ctx
->track
[track_idx
];
651 m4af_chunk_entry_t
*index
= track
->chunk_table
;
652 int is_co64
= (ctx
->mdat_pos
+ ctx
->mdat_size
> UINT32_MAX
);
653 int64_t pos
= m4af_tell(ctx
);
655 m4af_write32(ctx
, 0); /* size */
656 m4af_write(ctx
, is_co64
? "co64" : "stco", 4);
657 m4af_write32(ctx
, 0); /* version and flags */
658 m4af_write32(ctx
, track
->num_chunks
);
659 for (i
= 0; i
< track
->num_chunks
; ++i
, ++index
) {
661 m4af_write64(ctx
, index
->offset
);
663 m4af_write32(ctx
, index
->offset
);
665 m4af_update_box_size(ctx
, pos
);
669 void m4af_write_stsz_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
671 m4af_track_t
*track
= &ctx
->track
[track_idx
];
672 m4af_sample_entry_t
*index
= track
->sample_table
;
674 int64_t pos
= m4af_tell(ctx
);
676 "\0\0\0\0" /* size */
680 "\0\0\0\0" /* sample_size: 0(variable) */
682 m4af_write32(ctx
, track
->num_samples
);
683 for (i
= 0; i
< track
->num_samples
; ++i
, ++index
)
684 m4af_write32(ctx
, index
->size
);
685 m4af_update_box_size(ctx
, pos
);
689 void m4af_write_stsc_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
691 m4af_track_t
*track
= &ctx
->track
[track_idx
];
692 m4af_chunk_entry_t
*index
= track
->chunk_table
;
693 uint32_t i
, prev_samples_per_chunk
= 0, entry_count
= 0;
694 int64_t pos
= m4af_tell(ctx
);
696 "\0\0\0\0" /* size */
700 "\0\0\0\0" /* entry_count */
703 for (i
= 0; i
< track
->num_chunks
; ++i
, ++index
) {
704 if (index
->samples_per_chunk
!= prev_samples_per_chunk
) {
706 m4af_write32(ctx
, i
+ 1);
707 m4af_write32(ctx
, index
->samples_per_chunk
);
708 m4af_write32(ctx
, 1); /* sample_description_index */
709 prev_samples_per_chunk
= index
->samples_per_chunk
;
712 m4af_write32_at(ctx
, pos
+ 12, entry_count
);
713 m4af_update_box_size(ctx
, pos
);
717 void m4af_write_stts_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
719 m4af_track_t
*track
= &ctx
->track
[track_idx
];
720 m4af_sample_entry_t
*index
= track
->sample_table
;
721 uint32_t i
, prev_delta
= 0, entry_count
= 0, sample_count
= 0;
722 int64_t pos
= m4af_tell(ctx
);
724 "\0\0\0\0" /* size */
728 "\0\0\0\0" /* entry_count */
731 for (i
= 0; i
< track
->num_samples
; ++i
, ++index
) {
732 if (index
->delta
== prev_delta
)
737 m4af_write32(ctx
, sample_count
);
738 m4af_write32(ctx
, prev_delta
);
740 prev_delta
= index
->delta
;
745 m4af_write32(ctx
, sample_count
);
746 m4af_write32(ctx
, prev_delta
);
748 m4af_write32_at(ctx
, pos
+ 12, entry_count
);
749 m4af_update_box_size(ctx
, pos
);
753 void m4af_write_esds_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
755 m4af_track_t
*track
= &ctx
->track
[track_idx
];
756 int64_t pos
= m4af_tell(ctx
);
757 m4af_write(ctx
, "\0\0\0\0esds", 8);
758 m4af_write32(ctx
, 0); /* version + flags */
761 m4af_write_descriptor(ctx
, 3, 32 + track
->decSpecificInfoSize
);
762 m4af_write(ctx
, "\0\0\0", 3);
763 /* DecoderConfigDescriptor */
764 m4af_write_descriptor(ctx
, 4, 18 + track
->decSpecificInfoSize
);
766 "\x40" /* objectTypeIndication: 0x40(Audio ISO/IEC 14496-3)*/
767 "\x15" /* streamType(6): 0x05(AudioStream)
772 m4af_write24(ctx
, track
->bufferSizeDB
);
773 m4af_write32(ctx
, track
->maxBitrate
);
774 m4af_write32(ctx
, track
->avgBitrate
);
775 /* DecoderSpecificInfo */
776 m4af_write_descriptor(ctx
, 5, track
->decSpecificInfoSize
);
777 m4af_write(ctx
, track
->decSpecificInfo
, track
->decSpecificInfoSize
);
778 /* SLConfigDescriptor */
779 m4af_write_descriptor(ctx
, 6, 1);
780 m4af_write(ctx
, "\002", 1); /* predefined */
782 m4af_update_box_size(ctx
, pos
);
786 void m4af_write_alac_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
788 m4af_track_t
*track
= &ctx
->track
[track_idx
];
789 int64_t pos
= m4af_tell(ctx
);
791 "\0\0\0\0" /* size */
796 m4af_write(ctx
, track
->decSpecificInfo
, track
->decSpecificInfoSize
);
797 m4af_update_box_size(ctx
, pos
);
801 void m4af_write_mp4a_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
803 m4af_track_t
*track
= &ctx
->track
[track_idx
];
804 int64_t pos
= m4af_tell(ctx
);
805 m4af_write32(ctx
, 0); /* size */
806 m4af_write32(ctx
, track
->codec
); /* mp4a or alac */
808 "\0\0\0\0\0\0" /* reserved */
809 "\0\001" /* data_reference_index: 1 */
810 "\0\0\0\0" /* reserved[0] */
811 "\0\0\0\0" /* reserved[1] */
812 "\0\002" /* channelcount: 2 */
813 "\0\020" /* samplesize: 16 */
814 "\0\0" /* pre_defined */
815 "\0\0" /* reserved */
817 if (track
->codec
== M4AF_FOURCC('m','p','4','a')) {
818 m4af_write32(ctx
, track
->timescale
<< 16);
819 m4af_write_esds_box(ctx
, track_idx
);
821 m4af_write32(ctx
, 44100 << 16);
822 m4af_write_alac_box(ctx
, track_idx
);
824 m4af_update_box_size(ctx
, pos
);
828 void m4af_write_stsd_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
830 int64_t pos
= m4af_tell(ctx
);
831 m4af_write(ctx
, "\0\0\0\0stsd", 8);
835 "\0\0\0\001" /* entry_count: 1 */
837 m4af_write_mp4a_box(ctx
, track_idx
);
838 m4af_update_box_size(ctx
, pos
);
842 void m4af_write_sbgp_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
844 m4af_track_t
*track
= &ctx
->track
[track_idx
];
846 "\0\0\0\034" /* size: 28 */
850 "roll" /* grouping_type */
851 "\0\0\0\001" /* entry_count: 1 */
853 m4af_write32(ctx
, track
->num_samples
);
854 m4af_write32(ctx
, 1); /* group_description_index */
858 void m4af_write_sgpd_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
860 m4af_track_t
*track
= &ctx
->track
[track_idx
];
862 "\0\0\0\032" /* size */
866 "roll" /* grouping_type */
867 "\0\0\0\002" /* default_length: 2 */
868 "\0\0\0\001" /* entry_count: 1 */
869 "\377\377" /* payload_data: -1 */
874 void m4af_write_stbl_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
876 m4af_track_t
*track
= &ctx
->track
[track_idx
];
877 int64_t pos
= m4af_tell(ctx
);
878 m4af_write(ctx
, "\0\0\0\0stbl", 8);
879 m4af_write_stsd_box(ctx
, track_idx
);
880 if ((ctx
->priming_mode
& M4AF_PRIMING_MODE_EDTS
) &&
881 (track
->encoder_delay
|| track
->padding
)) {
882 m4af_write_sbgp_box(ctx
, track_idx
);
883 m4af_write_sgpd_box(ctx
, track_idx
);
885 m4af_write_stts_box(ctx
, track_idx
);
886 m4af_write_stsc_box(ctx
, track_idx
);
887 m4af_write_stsz_box(ctx
, track_idx
);
888 m4af_write_stco_box(ctx
, track_idx
);
889 m4af_update_box_size(ctx
, pos
);
893 void m4af_write_url_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
896 "\0\0\0\014" /* size */
899 "\0\0\001" /* flags: 1(in the same file) */
904 void m4af_write_dref_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
906 int64_t pos
= m4af_tell(ctx
);
907 m4af_write(ctx
, "\0\0\0\0dref", 8);
911 "\0\0\0\001" /* entry_count: 1 */
913 m4af_write_url_box(ctx
, track_idx
);
914 m4af_update_box_size(ctx
, pos
);
918 void m4af_write_dinf_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
920 int64_t pos
= m4af_tell(ctx
);
921 m4af_write(ctx
, "\0\0\0\0dinf", 8);
922 m4af_write_dref_box(ctx
, track_idx
);
923 m4af_update_box_size(ctx
, pos
);
927 void m4af_write_smhd_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
930 "\0\0\0\020" /* size */
935 "\0\0" /* reserved */
940 void m4af_write_minf_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
942 m4af_track_t
*track
= &ctx
->track
[track_idx
];
943 int64_t pos
= m4af_tell(ctx
);
944 m4af_write(ctx
, "\0\0\0\0minf", 8);
945 /* TODO: add TEXT support */
946 if (track
->codec
!= M4AF_CODEC_TEXT
)
947 m4af_write_smhd_box(ctx
, track_idx
);
948 m4af_write_dinf_box(ctx
, track_idx
);
949 m4af_write_stbl_box(ctx
, track_idx
);
950 m4af_update_box_size(ctx
, pos
);
954 void m4af_write_mdhd_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
956 m4af_track_t
*track
= &ctx
->track
[track_idx
];
957 int64_t pos
= m4af_tell(ctx
);
958 uint8_t version
= (track
->creation_time
> UINT32_MAX
||
959 track
->modification_time
> UINT32_MAX
||
960 track
->duration
> UINT32_MAX
);
962 m4af_write(ctx
, "\0\0\0\0mdhd", 8);
963 m4af_write(ctx
, &version
, 1);
964 m4af_write(ctx
, "\0\0\0", 3); /* flags */
966 m4af_write64(ctx
, track
->creation_time
);
967 m4af_write64(ctx
, track
->modification_time
);
968 m4af_write32(ctx
, track
->timescale
);
969 m4af_write64(ctx
, track
->duration
);
971 m4af_write32(ctx
, track
->creation_time
);
972 m4af_write32(ctx
, track
->modification_time
);
973 m4af_write32(ctx
, track
->timescale
);
974 m4af_write32(ctx
, track
->duration
);
977 "\x55\xc4" /* language: und */
978 "\0\0" /* pre_defined */
980 m4af_update_box_size(ctx
, pos
);
984 void m4af_write_hdlr_box(m4af_ctx_t
*ctx
, uint32_t track_idx
, const char *type
)
986 int64_t pos
= m4af_tell(ctx
);
987 static const char reserved_and_name
[10] = { 0 };
990 "\0\0\0\0" /* size */
994 "\0\0\0\0" /* pre_defined */
996 m4af_write(ctx
, type
, 4); /* handler_type */
998 m4af_write(ctx
, !strcmp(type
, "mdir") ? "appl" : "\0\0\0\0", 4);
999 /* reserved[1], reserved[2], name */
1000 m4af_write(ctx
, reserved_and_name
, (pos
& 1) ? 9 : 10);
1001 m4af_update_box_size(ctx
, pos
);
1005 void m4af_write_mdia_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
1007 m4af_track_t
*track
= &ctx
->track
[track_idx
];
1009 (track
->codec
== M4AF_CODEC_TEXT
) ? "text" : "soun";
1010 int64_t pos
= m4af_tell(ctx
);
1011 m4af_write(ctx
, "\0\0\0\0mdia", 8);
1012 m4af_write_mdhd_box(ctx
, track_idx
);
1013 m4af_write_hdlr_box(ctx
, track_idx
, hdlr
);
1014 m4af_write_minf_box(ctx
, track_idx
);
1015 m4af_update_box_size(ctx
, pos
);
1019 void m4af_write_elst_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
1021 m4af_track_t
*track
= &ctx
->track
[track_idx
];
1023 int64_t duration
= track
->duration
- track
->encoder_delay
- track
->padding
;
1024 int64_t pos
= m4af_tell(ctx
);
1025 duration
= (double)duration
/ track
->timescale
* ctx
->timescale
+ .5;
1026 version
= (duration
> UINT32_MAX
);
1028 m4af_write(ctx
, "\0\0\0\0elst", 8);
1029 m4af_write(ctx
, &version
, 1);
1030 m4af_write(ctx
, "\0\0\0", 3); /* flags */
1031 m4af_write32(ctx
, 1); /* entry_count: 1 */
1033 m4af_write64(ctx
, duration
);
1034 m4af_write64(ctx
, track
->encoder_delay
);
1036 m4af_write32(ctx
, duration
);
1037 m4af_write32(ctx
, track
->encoder_delay
);
1039 m4af_write16(ctx
, 1); /* media_rate_integer */
1040 m4af_write16(ctx
, 0); /* media_rate_fraction */
1041 m4af_update_box_size(ctx
, pos
);
1045 void m4af_write_edts_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
1047 m4af_track_t
*track
= &ctx
->track
[track_idx
];
1048 int64_t pos
= m4af_tell(ctx
);
1049 m4af_write(ctx
, "\0\0\0\0edts", 8);
1050 m4af_write_elst_box(ctx
, track_idx
);
1051 m4af_update_box_size(ctx
, pos
);
1055 void m4af_write_tkhd_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
1057 m4af_track_t
*track
= &ctx
->track
[track_idx
];
1058 int64_t pos
= m4af_tell(ctx
);
1060 (double)track
->duration
/ track
->timescale
* ctx
->timescale
+ .5;
1061 uint8_t version
= (track
->creation_time
> UINT32_MAX
||
1062 track
->modification_time
> UINT32_MAX
||
1063 duration
> UINT32_MAX
);
1064 m4af_write(ctx
, "\0\0\0\0tkhd", 8);
1065 m4af_write(ctx
, &version
, 1);
1066 m4af_write(ctx
, "\0\0\007", 3); /* flags */
1068 m4af_write64(ctx
, track
->creation_time
);
1069 m4af_write64(ctx
, track
->modification_time
);
1070 m4af_write32(ctx
, track_idx
+ 1);
1071 m4af_write(ctx
, "\0\0\0\0" /* reserved */
1073 m4af_write64(ctx
, duration
);
1075 m4af_write32(ctx
, track
->creation_time
);
1076 m4af_write32(ctx
, track
->modification_time
);
1077 m4af_write32(ctx
, track_idx
+ 1);
1078 m4af_write(ctx
, "\0\0\0\0" /* reserved */
1080 m4af_write32(ctx
, duration
);
1083 "\0\0\0\0" /* reserved[0] */
1084 "\0\0\0\0" /* reserved[1] */
1086 "\0\0" /* alternate_group */
1087 "\001\0" /* volume: 1.0 */
1088 "\0\0" /* reserved */
1089 "\0\001\0\0" /* matrix[0] */
1090 "\0\0\0\0" /* matrix[1] */
1091 "\0\0\0\0" /* matrix[2] */
1092 "\0\0\0\0" /* matrix[3] */
1093 "\0\001\0\0" /* matrix[4] */
1094 "\0\0\0\0" /* matrix[5] */
1095 "\0\0\0\0" /* matrix[6] */
1096 "\0\0\0\0" /* matrix[7] */
1097 "\100\0\0\0" /* matrix[8] */
1098 "\0\0\0\0" /* width */
1099 "\0\0\0\0" /* height */
1101 m4af_update_box_size(ctx
, pos
);
1105 void m4af_write_trak_box(m4af_ctx_t
*ctx
, uint32_t track_idx
)
1107 m4af_track_t
*track
= &ctx
->track
[track_idx
];
1108 int64_t pos
= m4af_tell(ctx
);
1109 m4af_write(ctx
, "\0\0\0\0trak", 8);
1110 m4af_write_tkhd_box(ctx
, track_idx
);
1111 if ((ctx
->priming_mode
& M4AF_PRIMING_MODE_EDTS
) &&
1112 (track
->encoder_delay
|| track
->padding
))
1113 m4af_write_edts_box(ctx
, track_idx
);
1114 m4af_write_mdia_box(ctx
, track_idx
);
1115 m4af_update_box_size(ctx
, pos
);
1119 int64_t m4af_movie_duration(m4af_ctx_t
*ctx
)
1121 int64_t movie_duration
= 0;
1123 for (i
= 0; i
< ctx
->num_tracks
; ++i
) {
1124 double x
= ctx
->track
[i
].duration
;
1125 int64_t duration
= x
/ ctx
->track
[i
].timescale
* ctx
->timescale
+ .5;
1126 if (duration
> movie_duration
)
1127 movie_duration
= duration
;
1129 return movie_duration
;
1133 void m4af_write_mvhd_box(m4af_ctx_t
*ctx
)
1135 int64_t pos
= m4af_tell(ctx
);
1136 int64_t movie_duration
= m4af_movie_duration(ctx
);
1137 uint8_t version
= (ctx
->creation_time
> UINT32_MAX
||
1138 ctx
->modification_time
> UINT32_MAX
||
1139 movie_duration
> UINT32_MAX
);
1141 m4af_write(ctx
, "\0\0\0\0mvhd", 8);
1142 m4af_write(ctx
, &version
, 1);
1143 m4af_write(ctx
, "\0\0\0", 3); /* flags */
1145 m4af_write64(ctx
, ctx
->creation_time
);
1146 m4af_write64(ctx
, ctx
->modification_time
);
1147 m4af_write32(ctx
, ctx
->timescale
);
1148 m4af_write64(ctx
, movie_duration
);
1150 m4af_write32(ctx
, ctx
->creation_time
);
1151 m4af_write32(ctx
, ctx
->modification_time
);
1152 m4af_write32(ctx
, ctx
->timescale
);
1153 m4af_write32(ctx
, movie_duration
);
1156 "\0\001\0\0" /* rate: 1.0 */
1157 "\001\0" /* volume: 1.0 */
1158 "\0\0" /* reserved */
1159 "\0\0\0\0" /* reserved[0] */
1160 "\0\0\0\0" /* reserved[1] */
1161 "\0\001\0\0" /* matrix[0] */
1162 "\0\0\0\0" /* matrix[1] */
1163 "\0\0\0\0" /* matrix[2] */
1164 "\0\0\0\0" /* matrix[3] */
1165 "\0\001\0\0" /* matrix[4] */
1166 "\0\0\0\0" /* matrix[5] */
1167 "\0\0\0\0" /* matrix[6] */
1168 "\0\0\0\0" /* matrix[7] */
1169 "\100\0\0\0" /* matrix[8] */
1170 "\0\0\0\0" /* pre_defined[0] */
1171 "\0\0\0\0" /* pre_defined[1] */
1172 "\0\0\0\0" /* pre_defined[2] */
1173 "\0\0\0\0" /* pre_defined[3] */
1174 "\0\0\0\0" /* pre_defined[4] */
1175 "\0\0\0\0" /* pre_defined[5] */
1177 m4af_write32(ctx
, ctx
->num_tracks
+ 1);
1178 m4af_update_box_size(ctx
, pos
);
1182 void m4af_write_mean_box(m4af_ctx_t
*ctx
)
1185 "\0\0\0\034" /* size */
1188 "\0\0\0" /* flags */
1189 "com.apple.iTunes" /* meaning-string */
1194 void m4af_write_name_box(m4af_ctx_t
*ctx
, const char *name
)
1196 int64_t pos
= m4af_tell(ctx
);
1198 "\0\0\0\0" /* size */
1201 "\0\0\0" /* flags */
1203 m4af_write(ctx
, name
, strlen(name
));
1204 m4af_update_box_size(ctx
, pos
);
1208 void m4af_write_data_box(m4af_ctx_t
*ctx
, uint32_t type_code
,
1209 const char *data
, uint32_t data_size
)
1211 int64_t pos
= m4af_tell(ctx
);
1212 uint8_t code
= type_code
;
1214 "\0\0\0\0" /* size */
1216 "\0\0" /* reserved */
1217 "\0" /* type_set_indifier */
1219 m4af_write(ctx
, &code
, 1);
1220 m4af_write(ctx
, "\0\0\0\0", 4); /* locale */
1221 m4af_write(ctx
, data
, data_size
);
1222 m4af_update_box_size(ctx
, pos
);
1226 void m4af_write_metadata(m4af_ctx_t
*ctx
, m4af_itmf_entry_t
*entry
)
1228 int64_t pos
= m4af_tell(ctx
);
1229 m4af_write(ctx
, "\0\0\0\0", 4);
1230 m4af_write32(ctx
, entry
->fcc
);
1231 if (entry
->fcc
!= M4AF_FOURCC('-','-','-','-'))
1232 m4af_write_data_box(ctx
, entry
->type_code
,
1233 entry
->data
, entry
->data_size
);
1235 m4af_write_mean_box(ctx
);
1236 m4af_write_name_box(ctx
, entry
->name
);
1237 m4af_write_data_box(ctx
, 1, entry
->data
, entry
->data_size
);
1239 m4af_update_box_size(ctx
, pos
);
1243 void m4af_write_ilst_box(m4af_ctx_t
*ctx
)
1246 int64_t pos
= m4af_tell(ctx
);
1247 m4af_write(ctx
, "\0\0\0\0ilst", 8);
1248 for (i
= 0; i
< ctx
->num_tags
; ++i
)
1249 m4af_write_metadata(ctx
, &ctx
->itmf_table
[i
]);
1250 m4af_update_box_size(ctx
, pos
);
1254 void m4af_write_meta_box(m4af_ctx_t
*ctx
)
1256 int64_t pos
= m4af_tell(ctx
);
1258 "\0\0\0\0" /* size */
1261 "\0\0\0" /* flags */
1263 m4af_write_hdlr_box(ctx
, 0, "mdir");
1264 m4af_write_ilst_box(ctx
);
1265 m4af_update_box_size(ctx
, pos
);
1269 void m4af_write_udta_box(m4af_ctx_t
*ctx
)
1271 int64_t pos
= m4af_tell(ctx
);
1272 m4af_write(ctx
, "\0\0\0\0udta", 8);
1273 m4af_write_meta_box(ctx
);
1274 m4af_update_box_size(ctx
, pos
);
1278 uint32_t m4af_write_moov_box(m4af_ctx_t
*ctx
)
1281 int64_t pos
= m4af_tell(ctx
);
1282 m4af_write(ctx
, "\0\0\0\0moov", 8);
1283 m4af_write_mvhd_box(ctx
);
1284 for (i
= 0; i
< ctx
->num_tracks
; ++i
)
1285 m4af_write_trak_box(ctx
, i
);
1287 m4af_write_udta_box(ctx
);
1288 return m4af_update_box_size(ctx
, pos
);
1292 void m4af_finalize_mdat(m4af_ctx_t
*ctx
)
1294 if (ctx
->mdat_size
+ 8 > UINT32_MAX
) {
1295 m4af_set_pos(ctx
, ctx
->mdat_pos
- 16);
1296 m4af_write32(ctx
, 1);
1297 m4af_write(ctx
, "mdat", 4);
1298 m4af_write64(ctx
, ctx
->mdat_size
+ 16);
1300 m4af_set_pos(ctx
, ctx
->mdat_pos
- 8);
1301 m4af_write32(ctx
, ctx
->mdat_size
+ 8);
1303 m4af_set_pos(ctx
, ctx
->mdat_pos
+ ctx
->mdat_size
);
1307 void m4af_shift_mdat_pos(m4af_ctx_t
*ctx
, uint32_t offset
)
1313 end
= ctx
->mdat_pos
+ ctx
->mdat_size
;
1314 for (; (begin
= m4af_max(ctx
->mdat_pos
, end
- 8192)) < end
; end
= begin
) {
1315 m4af_set_pos(ctx
, begin
);
1316 ctx
->io
.read(ctx
->io_cookie
, buf
, end
- begin
);
1317 m4af_set_pos(ctx
, begin
+ offset
);
1318 m4af_write(ctx
, buf
, end
- begin
);
1320 for (i
= 0; i
< ctx
->num_tracks
; ++i
)
1321 for (j
= 0; j
< ctx
->track
[i
].num_chunks
; ++j
)
1322 ctx
->track
[i
].chunk_table
[j
].offset
+= offset
;
1323 ctx
->mdat_pos
+= offset
;
1324 m4af_set_pos(ctx
, ctx
->mdat_pos
- 16);
1325 m4af_write_free_box(ctx
, 0);
1326 m4af_write(ctx
, "\0\0\0\0mdat", 8);
1327 m4af_finalize_mdat(ctx
);
1330 int m4af_finalize(m4af_ctx_t
*ctx
, int optimize
)
1333 m4af_track_t
*track
;
1336 for (i
= 0; i
< ctx
->num_tracks
; ++i
) {
1337 track
= ctx
->track
+ i
;
1338 if (track
->duration
) {
1339 int64_t track_size
= 0;
1341 for (j
= 0; j
< track
->num_chunks
; ++j
)
1342 track_size
+= track
->chunk_table
[j
].size
;
1344 8.0 * track_size
* track
->timescale
/ track
->duration
+ .5;
1346 m4af_flush_chunk(ctx
, i
);
1349 if ((ctx
->priming_mode
& M4AF_PRIMING_MODE_ITUNSMPB
) &&
1350 (track
->encoder_delay
|| track
->padding
))
1351 m4af_set_iTunSMPB(ctx
);
1352 m4af_finalize_mdat(ctx
);
1353 moov_size
= m4af_write_moov_box(ctx
);
1356 m4af_shift_mdat_pos(ctx
, moov_size
+ 1024);
1357 m4af_set_pos(ctx
, 32);
1358 m4af_write_moov_box(ctx
);
1359 pos
= m4af_tell(ctx
);
1360 m4af_write_free_box(ctx
, ctx
->mdat_pos
- pos
- 24);
1362 return ctx
->last_error
;