3 * Copyright (c) 2006 SmartJog S.A., Baptiste Coudurier <baptiste dot coudurier at smartjog dot com>
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
24 * SMPTE 336M KLV Data Encoding Protocol Using Key-Length-Value
25 * SMPTE 377M MXF File Format Specifications
26 * SMPTE 378M Operational Pattern 1a
27 * SMPTE 379M MXF Generic Container
28 * SMPTE 381M Mapping MPEG Streams into the MXF Generic Container
29 * SMPTE 382M Mapping AES3 and Broadcast Wave Audio into the MXF Generic Container
30 * SMPTE 383M Mapping DV-DIF Data to the MXF Generic Container
33 * Search for Track numbers which will identify essence element KLV packets.
34 * Search for SourcePackage which define tracks which contains Track numbers.
35 * Material Package contains tracks with reference to SourcePackage tracks.
36 * Search for Descriptors (Picture, Sound) which contains codec info and parameters.
37 * Assign Descriptors to correct Tracks.
39 * Metadata reading functions read Local Tags, get InstanceUID(0x3C0A) then add MetaDataSet to MXFContext.
40 * Metadata parsing resolves Strong References to objects.
42 * Simple demuxer, only OP1A supported and some files might not work at all.
43 * Only tracks with associated descriptors will be decoded. "Highly Desirable" SMPTE 377M D.1
48 #include "libavutil/aes.h"
49 #include "libavutil/avassert.h"
50 #include "libavutil/mathematics.h"
51 #include "libavcodec/bytestream.h"
52 #include "libavutil/intreadwrite.h"
53 #include "libavutil/timecode.h"
75 OPSONYOpt, /* FATE sample, violates the spec in places */
78 typedef struct MXFPartition {
81 MXFPartitionType type;
82 uint64_t previous_partition;
85 int64_t this_partition;
86 int64_t essence_offset; ///< absolute offset of essence
87 int64_t essence_length;
89 int64_t header_byte_count;
90 int64_t index_byte_count;
92 int64_t pack_ofs; ///< absolute offset of pack in file, including run-in
95 typedef struct MXFCryptoContext {
97 enum MXFMetadataSetType type;
98 UID source_container_ul;
101 typedef struct MXFStructuralComponent {
103 enum MXFMetadataSetType type;
104 UID source_package_uid;
105 UID data_definition_ul;
107 int64_t start_position;
109 } MXFStructuralComponent;
111 typedef struct MXFSequence {
113 enum MXFMetadataSetType type;
114 UID data_definition_ul;
115 UID *structural_components_refs;
116 int structural_components_count;
121 typedef struct MXFTrack {
123 enum MXFMetadataSetType type;
126 struct AVRational rate;
128 } MXFTimecodeComponent;
132 enum MXFMetadataSetType type;
133 UID input_segment_ref;
134 } MXFPulldownComponent;
138 enum MXFMetadataSetType type;
139 UID *structural_components_refs;
140 int structural_components_count;
146 enum MXFMetadataSetType type;
153 enum MXFMetadataSetType type;
154 MXFSequence *sequence; /* mandatory, and only one */
157 uint8_t track_number[4];
158 AVRational edit_rate;
160 uint64_t sample_count;
161 int64_t original_duration; /* st->duration in SampleRate/EditRate units */
164 typedef struct MXFDescriptor {
166 enum MXFMetadataSetType type;
167 UID essence_container_ul;
168 UID essence_codec_ul;
170 AVRational sample_rate;
171 AVRational aspect_ratio;
173 int height; /* Field height, not frame height */
174 int frame_layout; /* See MXFFrameLayout enum */
180 int64_t duration; /* ContainerDuration optional property */
181 unsigned int component_depth;
182 unsigned int horiz_subsampling;
183 unsigned int vert_subsampling;
184 UID *sub_descriptors_refs;
185 int sub_descriptors_count;
189 enum AVPixelFormat pix_fmt;
192 typedef struct MXFIndexTableSegment {
194 enum MXFMetadataSetType type;
195 int edit_unit_byte_count;
198 AVRational index_edit_rate;
199 uint64_t index_start_position;
200 uint64_t index_duration;
201 int8_t *temporal_offset_entries;
203 uint64_t *stream_offset_entries;
204 int nb_index_entries;
205 } MXFIndexTableSegment;
207 typedef struct MXFPackage {
209 enum MXFMetadataSetType type;
214 MXFDescriptor *descriptor; /* only one */
221 typedef struct MXFMetadataSet {
223 enum MXFMetadataSetType type;
226 /* decoded index table */
227 typedef struct MXFIndexTable {
230 int nb_ptses; /* number of PTSes or total duration of index */
231 int64_t first_dts; /* DTS = EditUnit + first_dts */
232 int64_t *ptses; /* maps EditUnit -> PTS */
234 MXFIndexTableSegment **segments; /* sorted by IndexStartPosition */
235 AVIndexEntry *fake_index; /* used for calling ff_index_search_timestamp() */
236 int8_t *offsets; /* temporal offsets for display order to stored order conversion */
239 typedef struct MXFContext {
240 MXFPartition *partitions;
241 unsigned partitions_count;
245 MXFMetadataSet **metadata_sets;
246 int metadata_sets_count;
250 int local_tags_count;
251 uint64_t footer_partition;
252 KLVPacket current_klv_data;
253 int current_klv_index;
255 MXFPartition *current_partition;
256 int parsing_backward;
257 int64_t last_forward_tell;
258 int last_forward_partition;
259 int current_edit_unit;
261 MXFIndexTable *index_tables;
262 int edit_units_per_packet; ///< how many edit units to read at a time (PCM, OPAtom)
265 enum MXFWrappingScheme {
270 /* NOTE: klv_offset is not set (-1) for local keys */
271 typedef int MXFMetadataReadFunc(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset);
273 typedef struct MXFMetadataReadTableEntry {
275 MXFMetadataReadFunc *read;
277 enum MXFMetadataSetType type;
278 } MXFMetadataReadTableEntry;
280 static int mxf_read_close(AVFormatContext *s);
282 /* partial keys to match */
283 static const uint8_t mxf_header_partition_pack_key[] = { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02 };
284 static const uint8_t mxf_essence_element_key[] = { 0x06,0x0e,0x2b,0x34,0x01,0x02,0x01,0x01,0x0d,0x01,0x03,0x01 };
285 static const uint8_t mxf_avid_essence_element_key[] = { 0x06,0x0e,0x2b,0x34,0x01,0x02,0x01,0x01,0x0e,0x04,0x03,0x01 };
286 static const uint8_t mxf_canopus_essence_element_key[] = { 0x06,0x0e,0x2b,0x34,0x01,0x02,0x01,0x0a,0x0e,0x0f,0x03,0x01 };
287 static const uint8_t mxf_system_item_key[] = { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x03,0x01,0x04 };
288 static const uint8_t mxf_klv_key[] = { 0x06,0x0e,0x2b,0x34 };
289 /* complete keys to match */
290 static const uint8_t mxf_crypto_source_container_ul[] = { 0x06,0x0e,0x2b,0x34,0x01,0x01,0x01,0x09,0x06,0x01,0x01,0x02,0x02,0x00,0x00,0x00 };
291 static const uint8_t mxf_encrypted_triplet_key[] = { 0x06,0x0e,0x2b,0x34,0x02,0x04,0x01,0x07,0x0d,0x01,0x03,0x01,0x02,0x7e,0x01,0x00 };
292 static const uint8_t mxf_encrypted_essence_container[] = { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x07,0x0d,0x01,0x03,0x01,0x02,0x0b,0x01,0x00 };
293 static const uint8_t mxf_random_index_pack_key[] = { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x11,0x01,0x00 };
294 static const uint8_t mxf_sony_mpeg4_extradata[] = { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x01,0x0e,0x06,0x06,0x02,0x02,0x01,0x00,0x00 };
295 static const uint8_t mxf_avid_project_name[] = { 0xa5,0xfb,0x7b,0x25,0xf6,0x15,0x94,0xb9,0x62,0xfc,0x37,0x17,0x49,0x2d,0x42,0xbf };
296 static const uint8_t mxf_jp2k_rsiz[] = { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02,0x01,0x00 };
297 static const uint8_t mxf_indirect_value_utf16le[] = { 0x4c,0x00,0x02,0x10,0x01,0x00,0x00,0x00,0x00,0x06,0x0e,0x2b,0x34,0x01,0x04,0x01,0x01 };
298 static const uint8_t mxf_indirect_value_utf16be[] = { 0x42,0x01,0x10,0x02,0x00,0x00,0x00,0x00,0x00,0x06,0x0e,0x2b,0x34,0x01,0x04,0x01,0x01 };
300 #define IS_KLV_KEY(x, y) (!memcmp(x, y, sizeof(y)))
302 static void mxf_free_metadataset(MXFMetadataSet **ctx, int freectx)
304 MXFIndexTableSegment *seg;
305 switch ((*ctx)->type) {
307 av_freep(&((MXFDescriptor *)*ctx)->extradata);
309 case MultipleDescriptor:
310 av_freep(&((MXFDescriptor *)*ctx)->sub_descriptors_refs);
313 av_freep(&((MXFSequence *)*ctx)->structural_components_refs);
316 av_freep(&((MXFEssenceGroup *)*ctx)->structural_components_refs);
319 case MaterialPackage:
320 av_freep(&((MXFPackage *)*ctx)->tracks_refs);
321 av_freep(&((MXFPackage *)*ctx)->name);
322 av_freep(&((MXFPackage *)*ctx)->comment_refs);
325 av_freep(&((MXFTaggedValue *)*ctx)->name);
326 av_freep(&((MXFTaggedValue *)*ctx)->value);
328 case IndexTableSegment:
329 seg = (MXFIndexTableSegment *)*ctx;
330 av_freep(&seg->temporal_offset_entries);
331 av_freep(&seg->flag_entries);
332 av_freep(&seg->stream_offset_entries);
340 static int64_t klv_decode_ber_length(AVIOContext *pb)
342 uint64_t size = avio_r8(pb);
343 if (size & 0x80) { /* long form */
344 int bytes_num = size & 0x7f;
345 /* SMPTE 379M 5.3.4 guarantee that bytes_num must not exceed 8 bytes */
347 return AVERROR_INVALIDDATA;
350 size = size << 8 | avio_r8(pb);
355 static int mxf_read_sync(AVIOContext *pb, const uint8_t *key, unsigned size)
358 for (i = 0; i < size && !avio_feof(pb); i++) {
362 else if (b != key[i])
368 static int klv_read_packet(KLVPacket *klv, AVIOContext *pb)
370 if (!mxf_read_sync(pb, mxf_klv_key, 4))
371 return AVERROR_INVALIDDATA;
372 klv->offset = avio_tell(pb) - 4;
373 memcpy(klv->key, mxf_klv_key, 4);
374 avio_read(pb, klv->key + 4, 12);
375 klv->length = klv_decode_ber_length(pb);
376 return klv->length == -1 ? -1 : 0;
379 static int mxf_get_stream_index(AVFormatContext *s, KLVPacket *klv)
383 for (i = 0; i < s->nb_streams; i++) {
384 MXFTrack *track = s->streams[i]->priv_data;
386 if (!memcmp(klv->key + sizeof(mxf_essence_element_key), track->track_number, sizeof(track->track_number)))
389 /* return 0 if only one stream, for OP Atom files with 0 as track number */
390 return s->nb_streams == 1 ? 0 : -1;
393 /* XXX: use AVBitStreamFilter */
394 static int mxf_get_d10_aes3_packet(AVIOContext *pb, AVStream *st, AVPacket *pkt, int64_t length)
396 const uint8_t *buf_ptr, *end_ptr;
400 if (length > 61444) /* worst case PAL 1920 samples 8 channels */
401 return AVERROR_INVALIDDATA;
402 length = av_get_packet(pb, pkt, length);
405 data_ptr = pkt->data;
406 end_ptr = pkt->data + length;
407 buf_ptr = pkt->data + 4; /* skip SMPTE 331M header */
408 for (; end_ptr - buf_ptr >= st->codecpar->channels * 4; ) {
409 for (i = 0; i < st->codecpar->channels; i++) {
410 uint32_t sample = bytestream_get_le32(&buf_ptr);
411 if (st->codecpar->bits_per_coded_sample == 24)
412 bytestream_put_le24(&data_ptr, (sample >> 4) & 0xffffff);
414 bytestream_put_le16(&data_ptr, (sample >> 12) & 0xffff);
416 buf_ptr += 32 - st->codecpar->channels*4; // always 8 channels stored SMPTE 331M
418 av_shrink_packet(pkt, data_ptr - pkt->data);
422 static int mxf_decrypt_triplet(AVFormatContext *s, AVPacket *pkt, KLVPacket *klv)
424 static const uint8_t checkv[16] = {0x43, 0x48, 0x55, 0x4b, 0x43, 0x48, 0x55, 0x4b, 0x43, 0x48, 0x55, 0x4b, 0x43, 0x48, 0x55, 0x4b};
425 MXFContext *mxf = s->priv_data;
426 AVIOContext *pb = s->pb;
427 int64_t end = avio_tell(pb) + klv->length;
430 uint64_t plaintext_size;
435 if (!mxf->aesc && s->key && s->keylen == 16) {
436 mxf->aesc = av_aes_alloc();
438 return AVERROR(ENOMEM);
439 av_aes_init(mxf->aesc, s->key, 128, 1);
442 avio_skip(pb, klv_decode_ber_length(pb));
444 klv_decode_ber_length(pb);
445 plaintext_size = avio_rb64(pb);
447 klv_decode_ber_length(pb);
448 avio_read(pb, klv->key, 16);
449 if (!IS_KLV_KEY(klv, mxf_essence_element_key))
450 return AVERROR_INVALIDDATA;
451 index = mxf_get_stream_index(s, klv);
453 return AVERROR_INVALIDDATA;
455 klv_decode_ber_length(pb);
456 orig_size = avio_rb64(pb);
457 if (orig_size < plaintext_size)
458 return AVERROR_INVALIDDATA;
460 size = klv_decode_ber_length(pb);
461 if (size < 32 || size - 32 < orig_size)
462 return AVERROR_INVALIDDATA;
463 avio_read(pb, ivec, 16);
464 avio_read(pb, tmpbuf, 16);
466 av_aes_crypt(mxf->aesc, tmpbuf, tmpbuf, 1, ivec, 1);
467 if (memcmp(tmpbuf, checkv, 16))
468 av_log(s, AV_LOG_ERROR, "probably incorrect decryption key\n");
470 size = av_get_packet(pb, pkt, size);
473 else if (size < plaintext_size)
474 return AVERROR_INVALIDDATA;
475 size -= plaintext_size;
477 av_aes_crypt(mxf->aesc, &pkt->data[plaintext_size],
478 &pkt->data[plaintext_size], size >> 4, ivec, 1);
479 av_shrink_packet(pkt, orig_size);
480 pkt->stream_index = index;
481 avio_skip(pb, end - avio_tell(pb));
485 static int mxf_read_primer_pack(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
487 MXFContext *mxf = arg;
488 int item_num = avio_rb32(pb);
489 int item_len = avio_rb32(pb);
491 if (item_len != 18) {
492 avpriv_request_sample(pb, "Primer pack item length %d", item_len);
493 return AVERROR_PATCHWELCOME;
495 if (item_num > 65536) {
496 av_log(mxf->fc, AV_LOG_ERROR, "item_num %d is too large\n", item_num);
497 return AVERROR_INVALIDDATA;
500 av_log(mxf->fc, AV_LOG_VERBOSE, "Multiple primer packs\n");
501 av_free(mxf->local_tags);
502 mxf->local_tags_count = 0;
503 mxf->local_tags = av_calloc(item_num, item_len);
504 if (!mxf->local_tags)
505 return AVERROR(ENOMEM);
506 mxf->local_tags_count = item_num;
507 avio_read(pb, mxf->local_tags, item_num*item_len);
511 static int mxf_read_partition_pack(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
513 MXFContext *mxf = arg;
514 MXFPartition *partition, *tmp_part;
516 uint64_t footer_partition;
517 uint32_t nb_essence_containers;
519 tmp_part = av_realloc_array(mxf->partitions, mxf->partitions_count + 1, sizeof(*mxf->partitions));
521 return AVERROR(ENOMEM);
522 mxf->partitions = tmp_part;
524 if (mxf->parsing_backward) {
525 /* insert the new partition pack in the middle
526 * this makes the entries in mxf->partitions sorted by offset */
527 memmove(&mxf->partitions[mxf->last_forward_partition+1],
528 &mxf->partitions[mxf->last_forward_partition],
529 (mxf->partitions_count - mxf->last_forward_partition)*sizeof(*mxf->partitions));
530 partition = mxf->current_partition = &mxf->partitions[mxf->last_forward_partition];
532 mxf->last_forward_partition++;
533 partition = mxf->current_partition = &mxf->partitions[mxf->partitions_count];
536 memset(partition, 0, sizeof(*partition));
537 mxf->partitions_count++;
538 partition->pack_length = avio_tell(pb) - klv_offset + size;
539 partition->pack_ofs = klv_offset;
543 partition->type = Header;
546 partition->type = BodyPartition;
549 partition->type = Footer;
552 av_log(mxf->fc, AV_LOG_ERROR, "unknown partition type %i\n", uid[13]);
553 return AVERROR_INVALIDDATA;
556 /* consider both footers to be closed (there is only Footer and CompleteFooter) */
557 partition->closed = partition->type == Footer || !(uid[14] & 1);
558 partition->complete = uid[14] > 2;
560 partition->kag_size = avio_rb32(pb);
561 partition->this_partition = avio_rb64(pb);
562 partition->previous_partition = avio_rb64(pb);
563 footer_partition = avio_rb64(pb);
564 partition->header_byte_count = avio_rb64(pb);
565 partition->index_byte_count = avio_rb64(pb);
566 partition->index_sid = avio_rb32(pb);
568 partition->body_sid = avio_rb32(pb);
569 if (avio_read(pb, op, sizeof(UID)) != sizeof(UID)) {
570 av_log(mxf->fc, AV_LOG_ERROR, "Failed reading UID\n");
571 return AVERROR_INVALIDDATA;
573 nb_essence_containers = avio_rb32(pb);
575 if (partition->this_partition &&
576 partition->previous_partition == partition->this_partition) {
577 av_log(mxf->fc, AV_LOG_ERROR,
578 "PreviousPartition equal to ThisPartition %"PRIx64"\n",
579 partition->previous_partition);
580 /* override with the actual previous partition offset */
581 if (!mxf->parsing_backward && mxf->last_forward_partition > 1) {
583 mxf->partitions + mxf->last_forward_partition - 2;
584 partition->previous_partition = prev->this_partition;
586 /* if no previous body partition are found point to the header
588 if (partition->previous_partition == partition->this_partition)
589 partition->previous_partition = 0;
590 av_log(mxf->fc, AV_LOG_ERROR,
591 "Overriding PreviousPartition with %"PRIx64"\n",
592 partition->previous_partition);
595 /* some files don't have FooterPartition set in every partition */
596 if (footer_partition) {
597 if (mxf->footer_partition && mxf->footer_partition != footer_partition) {
598 av_log(mxf->fc, AV_LOG_ERROR,
599 "inconsistent FooterPartition value: %"PRIu64" != %"PRIu64"\n",
600 mxf->footer_partition, footer_partition);
602 mxf->footer_partition = footer_partition;
606 av_log(mxf->fc, AV_LOG_TRACE,
607 "PartitionPack: ThisPartition = 0x%"PRIX64
608 ", PreviousPartition = 0x%"PRIX64", "
609 "FooterPartition = 0x%"PRIX64", IndexSID = %i, BodySID = %i\n",
610 partition->this_partition,
611 partition->previous_partition, footer_partition,
612 partition->index_sid, partition->body_sid);
614 /* sanity check PreviousPartition if set */
615 //NOTE: this isn't actually enough, see mxf_seek_to_previous_partition()
616 if (partition->previous_partition &&
617 mxf->run_in + partition->previous_partition >= klv_offset) {
618 av_log(mxf->fc, AV_LOG_ERROR,
619 "PreviousPartition points to this partition or forward\n");
620 return AVERROR_INVALIDDATA;
623 if (op[12] == 1 && op[13] == 1) mxf->op = OP1a;
624 else if (op[12] == 1 && op[13] == 2) mxf->op = OP1b;
625 else if (op[12] == 1 && op[13] == 3) mxf->op = OP1c;
626 else if (op[12] == 2 && op[13] == 1) mxf->op = OP2a;
627 else if (op[12] == 2 && op[13] == 2) mxf->op = OP2b;
628 else if (op[12] == 2 && op[13] == 3) mxf->op = OP2c;
629 else if (op[12] == 3 && op[13] == 1) mxf->op = OP3a;
630 else if (op[12] == 3 && op[13] == 2) mxf->op = OP3b;
631 else if (op[12] == 3 && op[13] == 3) mxf->op = OP3c;
632 else if (op[12] == 64&& op[13] == 1) mxf->op = OPSONYOpt;
633 else if (op[12] == 0x10) {
634 /* SMPTE 390m: "There shall be exactly one essence container"
635 * The following block deals with files that violate this, namely:
636 * 2011_DCPTEST_24FPS.V.mxf - two ECs, OP1a
637 * abcdefghiv016f56415e.mxf - zero ECs, OPAtom, output by Avid AirSpeed */
638 if (nb_essence_containers != 1) {
639 MXFOP op = nb_essence_containers ? OP1a : OPAtom;
643 av_log(mxf->fc, AV_LOG_WARNING,
644 "\"OPAtom\" with %"PRIu32" ECs - assuming %s\n",
645 nb_essence_containers,
646 op == OP1a ? "OP1a" : "OPAtom");
652 av_log(mxf->fc, AV_LOG_ERROR, "unknown operational pattern: %02xh %02xh - guessing OP1a\n", op[12], op[13]);
656 if (partition->kag_size <= 0 || partition->kag_size > (1 << 20)) {
657 av_log(mxf->fc, AV_LOG_WARNING, "invalid KAGSize %"PRId32" - guessing ",
658 partition->kag_size);
660 if (mxf->op == OPSONYOpt)
661 partition->kag_size = 512;
663 partition->kag_size = 1;
665 av_log(mxf->fc, AV_LOG_WARNING, "%"PRId32"\n", partition->kag_size);
671 static int mxf_add_metadata_set(MXFContext *mxf, void *metadata_set)
673 MXFMetadataSet **tmp;
675 tmp = av_realloc_array(mxf->metadata_sets, mxf->metadata_sets_count + 1, sizeof(*mxf->metadata_sets));
677 return AVERROR(ENOMEM);
678 mxf->metadata_sets = tmp;
679 mxf->metadata_sets[mxf->metadata_sets_count] = metadata_set;
680 mxf->metadata_sets_count++;
684 static int mxf_read_cryptographic_context(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
686 MXFCryptoContext *cryptocontext = arg;
688 return AVERROR_INVALIDDATA;
689 if (IS_KLV_KEY(uid, mxf_crypto_source_container_ul))
690 avio_read(pb, cryptocontext->source_container_ul, 16);
694 static int mxf_read_strong_ref_array(AVIOContext *pb, UID **refs, int *count)
696 *count = avio_rb32(pb);
697 *refs = av_calloc(*count, sizeof(UID));
700 return AVERROR(ENOMEM);
702 avio_skip(pb, 4); /* useless size of objects, always 16 according to specs */
703 avio_read(pb, (uint8_t *)*refs, *count * sizeof(UID));
707 static int mxf_read_content_storage(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
709 MXFContext *mxf = arg;
712 if (mxf->packages_refs)
713 av_log(mxf->fc, AV_LOG_VERBOSE, "Multiple packages_refs\n");
714 av_free(mxf->packages_refs);
715 return mxf_read_strong_ref_array(pb, &mxf->packages_refs, &mxf->packages_count);
720 static int mxf_read_source_clip(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
722 MXFStructuralComponent *source_clip = arg;
725 source_clip->duration = avio_rb64(pb);
728 source_clip->start_position = avio_rb64(pb);
731 /* UMID, only get last 16 bytes */
733 avio_read(pb, source_clip->source_package_uid, 16);
736 source_clip->source_track_id = avio_rb32(pb);
742 static int mxf_read_timecode_component(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
744 MXFTimecodeComponent *mxf_timecode = arg;
747 mxf_timecode->start_frame = avio_rb64(pb);
750 mxf_timecode->rate = (AVRational){avio_rb16(pb), 1};
753 mxf_timecode->drop_frame = avio_r8(pb);
759 static int mxf_read_pulldown_component(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
761 MXFPulldownComponent *mxf_pulldown = arg;
764 avio_read(pb, mxf_pulldown->input_segment_ref, 16);
770 static int mxf_read_track(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
772 MXFTrack *track = arg;
775 track->track_id = avio_rb32(pb);
778 avio_read(pb, track->track_number, 4);
781 track->edit_rate.num = avio_rb32(pb);
782 track->edit_rate.den = avio_rb32(pb);
785 avio_read(pb, track->sequence_ref, 16);
791 static int mxf_read_sequence(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
793 MXFSequence *sequence = arg;
796 sequence->duration = avio_rb64(pb);
799 avio_read(pb, sequence->data_definition_ul, 16);
802 sequence->origin = avio_r8(pb);
805 return mxf_read_strong_ref_array(pb, &sequence->structural_components_refs,
806 &sequence->structural_components_count);
811 static int mxf_read_essence_group(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
813 MXFEssenceGroup *essence_group = arg;
816 essence_group->duration = avio_rb64(pb);
819 return mxf_read_strong_ref_array(pb, &essence_group->structural_components_refs,
820 &essence_group->structural_components_count);
825 static inline int mxf_read_utf16_string(AVIOContext *pb, int size, char** str, int be)
831 return AVERROR(EINVAL);
833 buf_size = size + size / 2 + 1;
834 *str = av_malloc(buf_size);
836 return AVERROR(ENOMEM);
839 ret = avio_get_str16be(pb, size, *str, buf_size);
841 ret = avio_get_str16le(pb, size, *str, buf_size);
851 #define READ_STR16(type, big_endian) \
852 static int mxf_read_utf16 ## type ##_string(AVIOContext *pb, int size, char** str) \
854 return mxf_read_utf16_string(pb, size, str, big_endian); \
860 static int mxf_read_package(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
862 MXFPackage *package = arg;
865 return mxf_read_strong_ref_array(pb, &package->tracks_refs,
866 &package->tracks_count);
869 avio_read(pb, package->package_ul, 16);
870 avio_read(pb, package->package_uid, 16);
873 avio_read(pb, package->descriptor_ref, 16);
876 return mxf_read_utf16be_string(pb, size, &package->name);
878 return mxf_read_strong_ref_array(pb, &package->comment_refs,
879 &package->comment_count);
884 static int mxf_read_index_entry_array(AVIOContext *pb, MXFIndexTableSegment *segment)
888 segment->nb_index_entries = avio_rb32(pb);
890 length = avio_rb32(pb);
892 if (!(segment->temporal_offset_entries=av_calloc(segment->nb_index_entries, sizeof(*segment->temporal_offset_entries))) ||
893 !(segment->flag_entries = av_calloc(segment->nb_index_entries, sizeof(*segment->flag_entries))) ||
894 !(segment->stream_offset_entries = av_calloc(segment->nb_index_entries, sizeof(*segment->stream_offset_entries)))) {
895 av_freep(&segment->temporal_offset_entries);
896 av_freep(&segment->flag_entries);
897 return AVERROR(ENOMEM);
900 for (i = 0; i < segment->nb_index_entries; i++) {
901 segment->temporal_offset_entries[i] = avio_r8(pb);
902 avio_r8(pb); /* KeyFrameOffset */
903 segment->flag_entries[i] = avio_r8(pb);
904 segment->stream_offset_entries[i] = avio_rb64(pb);
905 avio_skip(pb, length - 11);
910 static int mxf_read_index_table_segment(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
912 MXFIndexTableSegment *segment = arg;
915 segment->edit_unit_byte_count = avio_rb32(pb);
916 av_log(NULL, AV_LOG_TRACE, "EditUnitByteCount %d\n", segment->edit_unit_byte_count);
919 segment->index_sid = avio_rb32(pb);
920 av_log(NULL, AV_LOG_TRACE, "IndexSID %d\n", segment->index_sid);
923 segment->body_sid = avio_rb32(pb);
924 av_log(NULL, AV_LOG_TRACE, "BodySID %d\n", segment->body_sid);
927 av_log(NULL, AV_LOG_TRACE, "IndexEntryArray found\n");
928 return mxf_read_index_entry_array(pb, segment);
930 segment->index_edit_rate.num = avio_rb32(pb);
931 segment->index_edit_rate.den = avio_rb32(pb);
932 av_log(NULL, AV_LOG_TRACE, "IndexEditRate %d/%d\n", segment->index_edit_rate.num,
933 segment->index_edit_rate.den);
936 segment->index_start_position = avio_rb64(pb);
937 av_log(NULL, AV_LOG_TRACE, "IndexStartPosition %"PRId64"\n", segment->index_start_position);
940 segment->index_duration = avio_rb64(pb);
941 av_log(NULL, AV_LOG_TRACE, "IndexDuration %"PRId64"\n", segment->index_duration);
947 static void mxf_read_pixel_layout(AVIOContext *pb, MXFDescriptor *descriptor)
949 int code, value, ofs = 0;
950 char layout[16] = {0}; /* not for printing, may end up not terminated on purpose */
955 av_log(NULL, AV_LOG_TRACE, "pixel layout: code %#x\n", code);
958 layout[ofs++] = code;
959 layout[ofs++] = value;
961 break; /* don't read byte by byte on sneaky files filled with lots of non-zeroes */
962 } while (code != 0); /* SMPTE 377M E.2.46 */
964 ff_mxf_decode_pixel_layout(layout, &descriptor->pix_fmt);
967 static int mxf_read_generic_descriptor(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
969 MXFDescriptor *descriptor = arg;
972 return mxf_read_strong_ref_array(pb, &descriptor->sub_descriptors_refs,
973 &descriptor->sub_descriptors_count);
974 case 0x3002: /* ContainerDuration */
975 descriptor->duration = avio_rb64(pb);
978 avio_read(pb, descriptor->essence_container_ul, 16);
981 avio_read(pb, descriptor->codec_ul, 16);
984 descriptor->linked_track_id = avio_rb32(pb);
986 case 0x3201: /* PictureEssenceCoding */
987 avio_read(pb, descriptor->essence_codec_ul, 16);
990 descriptor->width = avio_rb32(pb);
993 descriptor->height = avio_rb32(pb);
996 descriptor->frame_layout = avio_r8(pb);
999 descriptor->aspect_ratio.num = avio_rb32(pb);
1000 descriptor->aspect_ratio.den = avio_rb32(pb);
1003 descriptor->field_dominance = avio_r8(pb);
1006 descriptor->component_depth = avio_rb32(pb);
1009 descriptor->horiz_subsampling = avio_rb32(pb);
1012 descriptor->vert_subsampling = avio_rb32(pb);
1015 descriptor->sample_rate.num = avio_rb32(pb);
1016 descriptor->sample_rate.den = avio_rb32(pb);
1018 case 0x3D06: /* SoundEssenceCompression */
1019 avio_read(pb, descriptor->essence_codec_ul, 16);
1022 descriptor->channels = avio_rb32(pb);
1025 descriptor->bits_per_sample = avio_rb32(pb);
1028 mxf_read_pixel_layout(pb, descriptor);
1031 /* Private uid used by SONY C0023S01.mxf */
1032 if (IS_KLV_KEY(uid, mxf_sony_mpeg4_extradata)) {
1033 if (descriptor->extradata)
1034 av_log(NULL, AV_LOG_WARNING, "Duplicate sony_mpeg4_extradata\n");
1035 av_free(descriptor->extradata);
1036 descriptor->extradata_size = 0;
1037 descriptor->extradata = av_malloc(size);
1038 if (!descriptor->extradata)
1039 return AVERROR(ENOMEM);
1040 descriptor->extradata_size = size;
1041 avio_read(pb, descriptor->extradata, size);
1043 if (IS_KLV_KEY(uid, mxf_jp2k_rsiz)) {
1044 uint32_t rsiz = avio_rb16(pb);
1045 if (rsiz == FF_PROFILE_JPEG2000_DCINEMA_2K ||
1046 rsiz == FF_PROFILE_JPEG2000_DCINEMA_4K)
1047 descriptor->pix_fmt = AV_PIX_FMT_XYZ12;
1054 static int mxf_read_indirect_value(void *arg, AVIOContext *pb, int size)
1056 MXFTaggedValue *tagged_value = arg;
1062 avio_read(pb, key, 17);
1063 /* TODO: handle other types of of indirect values */
1064 if (memcmp(key, mxf_indirect_value_utf16le, 17) == 0) {
1065 return mxf_read_utf16le_string(pb, size - 17, &tagged_value->value);
1066 } else if (memcmp(key, mxf_indirect_value_utf16be, 17) == 0) {
1067 return mxf_read_utf16be_string(pb, size - 17, &tagged_value->value);
1072 static int mxf_read_tagged_value(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
1074 MXFTaggedValue *tagged_value = arg;
1077 return mxf_read_utf16be_string(pb, size, &tagged_value->name);
1079 return mxf_read_indirect_value(tagged_value, pb, size);
1085 * Match an uid independently of the version byte and up to len common bytes
1088 static int mxf_match_uid(const UID key, const UID uid, int len)
1091 for (i = 0; i < len; i++) {
1092 if (i != 7 && key[i] != uid[i])
1098 static const MXFCodecUL *mxf_get_codec_ul(const MXFCodecUL *uls, UID *uid)
1100 while (uls->uid[0]) {
1101 if(mxf_match_uid(uls->uid, *uid, uls->matching_len))
1108 static void *mxf_resolve_strong_ref(MXFContext *mxf, UID *strong_ref, enum MXFMetadataSetType type)
1114 for (i = 0; i < mxf->metadata_sets_count; i++) {
1115 if (!memcmp(*strong_ref, mxf->metadata_sets[i]->uid, 16) &&
1116 (type == AnyType || mxf->metadata_sets[i]->type == type)) {
1117 return mxf->metadata_sets[i];
1123 static const MXFCodecUL mxf_picture_essence_container_uls[] = {
1124 // video essence container uls
1125 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x07,0x0d,0x01,0x03,0x01,0x02,0x0c,0x01,0x00 }, 14, AV_CODEC_ID_JPEG2000 },
1126 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x02,0x0d,0x01,0x03,0x01,0x02,0x10,0x60,0x01 }, 14, AV_CODEC_ID_H264 }, /* H.264 frame wrapped */
1127 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x02,0x0d,0x01,0x03,0x01,0x02,0x12,0x01,0x00 }, 14, AV_CODEC_ID_VC1 }, /* VC-1 frame wrapped */
1128 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x02,0x0d,0x01,0x03,0x01,0x02,0x04,0x60,0x01 }, 14, AV_CODEC_ID_MPEG2VIDEO }, /* MPEG-ES frame wrapped */
1129 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x01,0x0d,0x01,0x03,0x01,0x02,0x01,0x04,0x01 }, 14, AV_CODEC_ID_MPEG2VIDEO }, /* Type D-10 mapping of 40Mbps 525/60-I */
1130 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x01,0x0d,0x01,0x03,0x01,0x02,0x02,0x41,0x01 }, 14, AV_CODEC_ID_DVVIDEO }, /* DV 625 25mbps */
1131 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x01,0x0d,0x01,0x03,0x01,0x02,0x05,0x00,0x00 }, 14, AV_CODEC_ID_RAWVIDEO }, /* uncompressed picture */
1132 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0a,0x0e,0x0f,0x03,0x01,0x02,0x20,0x01,0x01 }, 15, AV_CODEC_ID_HQ_HQA },
1133 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0a,0x0e,0x0f,0x03,0x01,0x02,0x20,0x02,0x01 }, 15, AV_CODEC_ID_HQX },
1134 { { 0x06,0x0e,0x2b,0x34,0x01,0x01,0x01,0xff,0x4b,0x46,0x41,0x41,0x00,0x0d,0x4d,0x4f }, 14, AV_CODEC_ID_RAWVIDEO }, /* Legacy ?? Uncompressed Picture */
1135 { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, AV_CODEC_ID_NONE },
1138 /* EC ULs for intra-only formats */
1139 static const MXFCodecUL mxf_intra_only_essence_container_uls[] = {
1140 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x01,0x0d,0x01,0x03,0x01,0x02,0x01,0x00,0x00 }, 14, AV_CODEC_ID_MPEG2VIDEO }, /* MXF-GC SMPTE D-10 mappings */
1141 { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, AV_CODEC_ID_NONE },
1144 /* intra-only PictureEssenceCoding ULs, where no corresponding EC UL exists */
1145 static const MXFCodecUL mxf_intra_only_picture_essence_coding_uls[] = {
1146 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0A,0x04,0x01,0x02,0x02,0x01,0x32,0x00,0x00 }, 14, AV_CODEC_ID_H264 }, /* H.264/MPEG-4 AVC Intra Profiles */
1147 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x07,0x04,0x01,0x02,0x02,0x03,0x01,0x01,0x00 }, 14, AV_CODEC_ID_JPEG2000 }, /* JPEG 2000 code stream */
1148 { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, AV_CODEC_ID_NONE },
1151 /* actual coded width for AVC-Intra to allow selecting correct SPS/PPS */
1152 static const MXFCodecUL mxf_intra_only_picture_coded_width[] = {
1153 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0A,0x04,0x01,0x02,0x02,0x01,0x32,0x21,0x01 }, 16, 1440 },
1154 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0A,0x04,0x01,0x02,0x02,0x01,0x32,0x21,0x02 }, 16, 1440 },
1155 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0A,0x04,0x01,0x02,0x02,0x01,0x32,0x21,0x03 }, 16, 1440 },
1156 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0A,0x04,0x01,0x02,0x02,0x01,0x32,0x21,0x04 }, 16, 1440 },
1157 { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, 0 },
1160 static const MXFCodecUL mxf_sound_essence_container_uls[] = {
1161 // sound essence container uls
1162 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x01,0x0d,0x01,0x03,0x01,0x02,0x06,0x01,0x00 }, 14, AV_CODEC_ID_PCM_S16LE }, /* BWF Frame wrapped */
1163 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x02,0x0d,0x01,0x03,0x01,0x02,0x04,0x40,0x01 }, 14, AV_CODEC_ID_MP2 }, /* MPEG-ES Frame wrapped, 0x40 ??? stream id */
1164 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x01,0x0d,0x01,0x03,0x01,0x02,0x01,0x01,0x01 }, 14, AV_CODEC_ID_PCM_S16LE }, /* D-10 Mapping 50Mbps PAL Extended Template */
1165 { { 0x06,0x0e,0x2b,0x34,0x01,0x01,0x01,0xff,0x4b,0x46,0x41,0x41,0x00,0x0d,0x4d,0x4F }, 14, AV_CODEC_ID_PCM_S16LE }, /* 0001GL00.MXF.A1.mxf_opatom.mxf */
1166 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x03,0x04,0x02,0x02,0x02,0x03,0x03,0x01,0x00 }, 14, AV_CODEC_ID_AAC }, /* MPEG-2 AAC ADTS (legacy) */
1167 { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, AV_CODEC_ID_NONE },
1170 static const MXFCodecUL mxf_data_essence_container_uls[] = {
1171 { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x09,0x0d,0x01,0x03,0x01,0x02,0x0e,0x00,0x00 }, 16, 0 },
1172 { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, AV_CODEC_ID_NONE },
1175 static const char* const mxf_data_essence_descriptor[] = {
1176 "vbi_vanc_smpte_436M",
1179 static int mxf_get_sorted_table_segments(MXFContext *mxf, int *nb_sorted_segments, MXFIndexTableSegment ***sorted_segments)
1181 int i, j, nb_segments = 0;
1182 MXFIndexTableSegment **unsorted_segments;
1183 int last_body_sid = -1, last_index_sid = -1, last_index_start = -1;
1185 /* count number of segments, allocate arrays and copy unsorted segments */
1186 for (i = 0; i < mxf->metadata_sets_count; i++)
1187 if (mxf->metadata_sets[i]->type == IndexTableSegment)
1191 return AVERROR_INVALIDDATA;
1193 if (!(unsorted_segments = av_calloc(nb_segments, sizeof(*unsorted_segments))) ||
1194 !(*sorted_segments = av_calloc(nb_segments, sizeof(**sorted_segments)))) {
1195 av_freep(sorted_segments);
1196 av_free(unsorted_segments);
1197 return AVERROR(ENOMEM);
1200 for (i = j = 0; i < mxf->metadata_sets_count; i++)
1201 if (mxf->metadata_sets[i]->type == IndexTableSegment)
1202 unsorted_segments[j++] = (MXFIndexTableSegment*)mxf->metadata_sets[i];
1204 *nb_sorted_segments = 0;
1206 /* sort segments by {BodySID, IndexSID, IndexStartPosition}, remove duplicates while we're at it */
1207 for (i = 0; i < nb_segments; i++) {
1208 int best = -1, best_body_sid = -1, best_index_sid = -1, best_index_start = -1;
1209 uint64_t best_index_duration = 0;
1211 for (j = 0; j < nb_segments; j++) {
1212 MXFIndexTableSegment *s = unsorted_segments[j];
1214 /* Require larger BosySID, IndexSID or IndexStartPosition then the previous entry. This removes duplicates.
1215 * We want the smallest values for the keys than what we currently have, unless this is the first such entry this time around.
1216 * If we come across an entry with the same IndexStartPosition but larger IndexDuration, then we'll prefer it over the one we currently have.
1218 if ((i == 0 || s->body_sid > last_body_sid || s->index_sid > last_index_sid || s->index_start_position > last_index_start) &&
1219 (best == -1 || s->body_sid < best_body_sid || s->index_sid < best_index_sid || s->index_start_position < best_index_start ||
1220 (s->index_start_position == best_index_start && s->index_duration > best_index_duration))) {
1222 best_body_sid = s->body_sid;
1223 best_index_sid = s->index_sid;
1224 best_index_start = s->index_start_position;
1225 best_index_duration = s->index_duration;
1229 /* no suitable entry found -> we're done */
1233 (*sorted_segments)[(*nb_sorted_segments)++] = unsorted_segments[best];
1234 last_body_sid = best_body_sid;
1235 last_index_sid = best_index_sid;
1236 last_index_start = best_index_start;
1239 av_free(unsorted_segments);
1245 * Computes the absolute file offset of the given essence container offset
1247 static int mxf_absolute_bodysid_offset(MXFContext *mxf, int body_sid, int64_t offset, int64_t *offset_out)
1250 int64_t offset_in = offset; /* for logging */
1252 for (x = 0; x < mxf->partitions_count; x++) {
1253 MXFPartition *p = &mxf->partitions[x];
1255 if (p->body_sid != body_sid)
1258 if (offset < p->essence_length || !p->essence_length) {
1259 *offset_out = p->essence_offset + offset;
1263 offset -= p->essence_length;
1266 av_log(mxf->fc, AV_LOG_ERROR,
1267 "failed to find absolute offset of %"PRIX64" in BodySID %i - partial file?\n",
1268 offset_in, body_sid);
1270 return AVERROR_INVALIDDATA;
1274 * Returns the end position of the essence container with given BodySID, or zero if unknown
1276 static int64_t mxf_essence_container_end(MXFContext *mxf, int body_sid)
1281 for (x = 0; x < mxf->partitions_count; x++) {
1282 MXFPartition *p = &mxf->partitions[x];
1284 if (p->body_sid != body_sid)
1287 if (!p->essence_length)
1290 ret = p->essence_offset + p->essence_length;
1296 /* EditUnit -> absolute offset */
1297 static int mxf_edit_unit_absolute_offset(MXFContext *mxf, MXFIndexTable *index_table, int64_t edit_unit, int64_t *edit_unit_out, int64_t *offset_out, int nag)
1300 int64_t offset_temp = 0;
1302 for (i = 0; i < index_table->nb_segments; i++) {
1303 MXFIndexTableSegment *s = index_table->segments[i];
1305 edit_unit = FFMAX(edit_unit, s->index_start_position); /* clamp if trying to seek before start */
1307 if (edit_unit < s->index_start_position + s->index_duration) {
1308 int64_t index = edit_unit - s->index_start_position;
1310 if (s->edit_unit_byte_count)
1311 offset_temp += s->edit_unit_byte_count * index;
1312 else if (s->nb_index_entries) {
1313 if (s->nb_index_entries == 2 * s->index_duration + 1)
1314 index *= 2; /* Avid index */
1316 if (index < 0 || index >= s->nb_index_entries) {
1317 av_log(mxf->fc, AV_LOG_ERROR, "IndexSID %i segment at %"PRId64" IndexEntryArray too small\n",
1318 index_table->index_sid, s->index_start_position);
1319 return AVERROR_INVALIDDATA;
1322 offset_temp = s->stream_offset_entries[index];
1324 av_log(mxf->fc, AV_LOG_ERROR, "IndexSID %i segment at %"PRId64" missing EditUnitByteCount and IndexEntryArray\n",
1325 index_table->index_sid, s->index_start_position);
1326 return AVERROR_INVALIDDATA;
1330 *edit_unit_out = edit_unit;
1332 return mxf_absolute_bodysid_offset(mxf, index_table->body_sid, offset_temp, offset_out);
1334 /* EditUnitByteCount == 0 for VBR indexes, which is fine since they use explicit StreamOffsets */
1335 offset_temp += s->edit_unit_byte_count * s->index_duration;
1340 av_log(mxf->fc, AV_LOG_ERROR, "failed to map EditUnit %"PRId64" in IndexSID %i to an offset\n", edit_unit, index_table->index_sid);
1342 return AVERROR_INVALIDDATA;
1345 static int mxf_compute_ptses_fake_index(MXFContext *mxf, MXFIndexTable *index_table)
1348 int8_t max_temporal_offset = -128;
1351 /* first compute how many entries we have */
1352 for (i = 0; i < index_table->nb_segments; i++) {
1353 MXFIndexTableSegment *s = index_table->segments[i];
1355 if (!s->nb_index_entries) {
1356 index_table->nb_ptses = 0;
1357 return 0; /* no TemporalOffsets */
1360 index_table->nb_ptses += s->index_duration;
1363 /* paranoid check */
1364 if (index_table->nb_ptses <= 0)
1367 if (!(index_table->ptses = av_calloc(index_table->nb_ptses, sizeof(int64_t))) ||
1368 !(index_table->fake_index = av_calloc(index_table->nb_ptses, sizeof(AVIndexEntry))) ||
1369 !(index_table->offsets = av_calloc(index_table->nb_ptses, sizeof(int8_t))) ||
1370 !(flags = av_calloc(index_table->nb_ptses, sizeof(uint8_t)))) {
1371 av_freep(&index_table->ptses);
1372 av_freep(&index_table->fake_index);
1373 av_freep(&index_table->offsets);
1374 return AVERROR(ENOMEM);
1377 /* we may have a few bad TemporalOffsets
1378 * make sure the corresponding PTSes don't have the bogus value 0 */
1379 for (x = 0; x < index_table->nb_ptses; x++)
1380 index_table->ptses[x] = AV_NOPTS_VALUE;
1394 * We want to transform it into this:
1405 * We do this by bucket sorting x by x+TemporalOffset[x] into mxf->ptses,
1406 * then settings mxf->first_dts = -max(TemporalOffset[x]).
1407 * The latter makes DTS <= PTS.
1409 for (i = x = 0; i < index_table->nb_segments; i++) {
1410 MXFIndexTableSegment *s = index_table->segments[i];
1411 int index_delta = 1;
1412 int n = s->nb_index_entries;
1414 if (s->nb_index_entries == 2 * s->index_duration + 1) {
1415 index_delta = 2; /* Avid index */
1416 /* ignore the last entry - it's the size of the essence container */
1420 for (j = 0; j < n; j += index_delta, x++) {
1421 int offset = s->temporal_offset_entries[j] / index_delta;
1422 int index = x + offset;
1424 if (x >= index_table->nb_ptses) {
1425 av_log(mxf->fc, AV_LOG_ERROR,
1426 "x >= nb_ptses - IndexEntryCount %i < IndexDuration %"PRId64"?\n",
1427 s->nb_index_entries, s->index_duration);
1431 flags[x] = !(s->flag_entries[j] & 0x30) ? AVINDEX_KEYFRAME : 0;
1433 if (index < 0 || index >= index_table->nb_ptses) {
1434 av_log(mxf->fc, AV_LOG_ERROR,
1435 "index entry %i + TemporalOffset %i = %i, which is out of bounds\n",
1440 index_table->offsets[x] = offset;
1441 index_table->ptses[index] = x;
1442 max_temporal_offset = FFMAX(max_temporal_offset, offset);
1446 /* calculate the fake index table in display order */
1447 for (x = 0; x < index_table->nb_ptses; x++) {
1448 index_table->fake_index[x].timestamp = x;
1449 if (index_table->ptses[x] != AV_NOPTS_VALUE)
1450 index_table->fake_index[index_table->ptses[x]].flags = flags[x];
1454 index_table->first_dts = -max_temporal_offset;
1460 * Sorts and collects index table segments into index tables.
1461 * Also computes PTSes if possible.
1463 static int mxf_compute_index_tables(MXFContext *mxf)
1465 int i, j, k, ret, nb_sorted_segments;
1466 MXFIndexTableSegment **sorted_segments = NULL;
1468 if ((ret = mxf_get_sorted_table_segments(mxf, &nb_sorted_segments, &sorted_segments)) ||
1469 nb_sorted_segments <= 0) {
1470 av_log(mxf->fc, AV_LOG_WARNING, "broken or empty index\n");
1474 /* sanity check and count unique BodySIDs/IndexSIDs */
1475 for (i = 0; i < nb_sorted_segments; i++) {
1476 if (i == 0 || sorted_segments[i-1]->index_sid != sorted_segments[i]->index_sid)
1477 mxf->nb_index_tables++;
1478 else if (sorted_segments[i-1]->body_sid != sorted_segments[i]->body_sid) {
1479 av_log(mxf->fc, AV_LOG_ERROR, "found inconsistent BodySID\n");
1480 ret = AVERROR_INVALIDDATA;
1481 goto finish_decoding_index;
1485 mxf->index_tables = av_mallocz_array(mxf->nb_index_tables,
1486 sizeof(*mxf->index_tables));
1487 if (!mxf->index_tables) {
1488 av_log(mxf->fc, AV_LOG_ERROR, "failed to allocate index tables\n");
1489 ret = AVERROR(ENOMEM);
1490 goto finish_decoding_index;
1493 /* distribute sorted segments to index tables */
1494 for (i = j = 0; i < nb_sorted_segments; i++) {
1495 if (i != 0 && sorted_segments[i-1]->index_sid != sorted_segments[i]->index_sid) {
1500 mxf->index_tables[j].nb_segments++;
1503 for (i = j = 0; j < mxf->nb_index_tables; i += mxf->index_tables[j++].nb_segments) {
1504 MXFIndexTable *t = &mxf->index_tables[j];
1506 t->segments = av_mallocz_array(t->nb_segments,
1507 sizeof(*t->segments));
1510 av_log(mxf->fc, AV_LOG_ERROR, "failed to allocate IndexTableSegment"
1511 " pointer array\n");
1512 ret = AVERROR(ENOMEM);
1513 goto finish_decoding_index;
1516 if (sorted_segments[i]->index_start_position)
1517 av_log(mxf->fc, AV_LOG_WARNING, "IndexSID %i starts at EditUnit %"PRId64" - seeking may not work as expected\n",
1518 sorted_segments[i]->index_sid, sorted_segments[i]->index_start_position);
1520 memcpy(t->segments, &sorted_segments[i], t->nb_segments * sizeof(MXFIndexTableSegment*));
1521 t->index_sid = sorted_segments[i]->index_sid;
1522 t->body_sid = sorted_segments[i]->body_sid;
1524 if ((ret = mxf_compute_ptses_fake_index(mxf, t)) < 0)
1525 goto finish_decoding_index;
1527 /* fix zero IndexDurations */
1528 for (k = 0; k < t->nb_segments; k++) {
1529 if (t->segments[k]->index_duration)
1532 if (t->nb_segments > 1)
1533 av_log(mxf->fc, AV_LOG_WARNING, "IndexSID %i segment %i has zero IndexDuration and there's more than one segment\n",
1536 if (mxf->fc->nb_streams <= 0) {
1537 av_log(mxf->fc, AV_LOG_WARNING, "no streams?\n");
1541 /* assume the first stream's duration is reasonable
1542 * leave index_duration = 0 on further segments in case we have any (unlikely)
1544 t->segments[k]->index_duration = mxf->fc->streams[0]->duration;
1550 finish_decoding_index:
1551 av_free(sorted_segments);
1555 static int mxf_is_intra_only(MXFDescriptor *descriptor)
1557 return mxf_get_codec_ul(mxf_intra_only_essence_container_uls,
1558 &descriptor->essence_container_ul)->id != AV_CODEC_ID_NONE ||
1559 mxf_get_codec_ul(mxf_intra_only_picture_essence_coding_uls,
1560 &descriptor->essence_codec_ul)->id != AV_CODEC_ID_NONE;
1563 static int mxf_uid_to_str(UID uid, char **str)
1567 p = *str = av_mallocz(sizeof(UID) * 2 + 4 + 1);
1569 return AVERROR(ENOMEM);
1570 for (i = 0; i < sizeof(UID); i++) {
1571 snprintf(p, 2 + 1, "%.2x", uid[i]);
1573 if (i == 3 || i == 5 || i == 7 || i == 9) {
1574 snprintf(p, 1 + 1, "-");
1581 static int mxf_umid_to_str(UID ul, UID uid, char **str)
1585 p = *str = av_mallocz(sizeof(UID) * 4 + 2 + 1);
1587 return AVERROR(ENOMEM);
1588 snprintf(p, 2 + 1, "0x");
1590 for (i = 0; i < sizeof(UID); i++) {
1591 snprintf(p, 2 + 1, "%.2X", ul[i]);
1595 for (i = 0; i < sizeof(UID); i++) {
1596 snprintf(p, 2 + 1, "%.2X", uid[i]);
1602 static int mxf_add_umid_metadata(AVDictionary **pm, const char *key, MXFPackage* package)
1608 if ((ret = mxf_umid_to_str(package->package_ul, package->package_uid, &str)) < 0)
1610 av_dict_set(pm, key, str, AV_DICT_DONT_STRDUP_VAL);
1614 static int mxf_add_timecode_metadata(AVDictionary **pm, const char *key, AVTimecode *tc)
1616 char buf[AV_TIMECODE_STR_SIZE];
1617 av_dict_set(pm, key, av_timecode_make_string(tc, buf, 0), 0);
1622 static MXFTimecodeComponent* mxf_resolve_timecode_component(MXFContext *mxf, UID *strong_ref)
1624 MXFStructuralComponent *component = NULL;
1625 MXFPulldownComponent *pulldown = NULL;
1627 component = mxf_resolve_strong_ref(mxf, strong_ref, AnyType);
1631 switch (component->type) {
1632 case TimecodeComponent:
1633 return (MXFTimecodeComponent*)component;
1634 case PulldownComponent: /* timcode component may be located on a pulldown component */
1635 pulldown = (MXFPulldownComponent*)component;
1636 return mxf_resolve_strong_ref(mxf, &pulldown->input_segment_ref, TimecodeComponent);
1643 static MXFPackage* mxf_resolve_source_package(MXFContext *mxf, UID package_uid)
1645 MXFPackage *package = NULL;
1648 for (i = 0; i < mxf->packages_count; i++) {
1649 package = mxf_resolve_strong_ref(mxf, &mxf->packages_refs[i], SourcePackage);
1653 if (!memcmp(package->package_uid, package_uid, 16))
1659 static MXFDescriptor* mxf_resolve_multidescriptor(MXFContext *mxf, MXFDescriptor *descriptor, int track_id)
1661 MXFDescriptor *sub_descriptor = NULL;
1667 if (descriptor->type == MultipleDescriptor) {
1668 for (i = 0; i < descriptor->sub_descriptors_count; i++) {
1669 sub_descriptor = mxf_resolve_strong_ref(mxf, &descriptor->sub_descriptors_refs[i], Descriptor);
1671 if (!sub_descriptor) {
1672 av_log(mxf->fc, AV_LOG_ERROR, "could not resolve sub descriptor strong ref\n");
1675 if (sub_descriptor->linked_track_id == track_id) {
1676 return sub_descriptor;
1679 } else if (descriptor->type == Descriptor)
1685 static MXFStructuralComponent* mxf_resolve_essence_group_choice(MXFContext *mxf, MXFEssenceGroup *essence_group)
1687 MXFStructuralComponent *component = NULL;
1688 MXFPackage *package = NULL;
1689 MXFDescriptor *descriptor = NULL;
1692 if (!essence_group || !essence_group->structural_components_count)
1695 /* essence groups contains multiple representations of the same media,
1696 this return the first components with a valid Descriptor typically index 0 */
1697 for (i =0; i < essence_group->structural_components_count; i++){
1698 component = mxf_resolve_strong_ref(mxf, &essence_group->structural_components_refs[i], SourceClip);
1702 if (!(package = mxf_resolve_source_package(mxf, component->source_package_uid)))
1705 descriptor = mxf_resolve_strong_ref(mxf, &package->descriptor_ref, Descriptor);
1712 static MXFStructuralComponent* mxf_resolve_sourceclip(MXFContext *mxf, UID *strong_ref)
1714 MXFStructuralComponent *component = NULL;
1716 component = mxf_resolve_strong_ref(mxf, strong_ref, AnyType);
1719 switch (component->type) {
1723 return mxf_resolve_essence_group_choice(mxf, (MXFEssenceGroup*) component);
1730 static int mxf_parse_package_comments(MXFContext *mxf, AVDictionary **pm, MXFPackage *package)
1732 MXFTaggedValue *tag;
1736 for (i = 0; i < package->comment_count; i++) {
1737 tag = mxf_resolve_strong_ref(mxf, &package->comment_refs[i], TaggedValue);
1738 if (!tag || !tag->name || !tag->value)
1741 size = strlen(tag->name) + 8 + 1;
1742 key = av_mallocz(size);
1744 return AVERROR(ENOMEM);
1746 snprintf(key, size, "comment_%s", tag->name);
1747 av_dict_set(pm, key, tag->value, AV_DICT_DONT_STRDUP_KEY);
1752 static int mxf_parse_physical_source_package(MXFContext *mxf, MXFTrack *source_track, AVStream *st)
1754 MXFPackage *physical_package = NULL;
1755 MXFTrack *physical_track = NULL;
1756 MXFStructuralComponent *sourceclip = NULL;
1757 MXFTimecodeComponent *mxf_tc = NULL;
1761 int64_t start_position;
1763 for (i = 0; i < source_track->sequence->structural_components_count; i++) {
1764 sourceclip = mxf_resolve_strong_ref(mxf, &source_track->sequence->structural_components_refs[i], SourceClip);
1768 if (!(physical_package = mxf_resolve_source_package(mxf, sourceclip->source_package_uid)))
1771 mxf_add_umid_metadata(&st->metadata, "reel_umid", physical_package);
1773 /* the name of physical source package is name of the reel or tape */
1774 if (physical_package->name && physical_package->name[0])
1775 av_dict_set(&st->metadata, "reel_name", physical_package->name, 0);
1777 /* the source timecode is calculated by adding the start_position of the sourceclip from the file source package track
1778 * to the start_frame of the timecode component located on one of the tracks of the physical source package.
1780 for (j = 0; j < physical_package->tracks_count; j++) {
1781 if (!(physical_track = mxf_resolve_strong_ref(mxf, &physical_package->tracks_refs[j], Track))) {
1782 av_log(mxf->fc, AV_LOG_ERROR, "could not resolve source track strong ref\n");
1786 if (!(physical_track->sequence = mxf_resolve_strong_ref(mxf, &physical_track->sequence_ref, Sequence))) {
1787 av_log(mxf->fc, AV_LOG_ERROR, "could not resolve source track sequence strong ref\n");
1791 if (physical_track->edit_rate.num <= 0 ||
1792 physical_track->edit_rate.den <= 0) {
1793 av_log(mxf->fc, AV_LOG_WARNING,
1794 "Invalid edit rate (%d/%d) found on structural"
1795 " component #%d, defaulting to 25/1\n",
1796 physical_track->edit_rate.num,
1797 physical_track->edit_rate.den, i);
1798 physical_track->edit_rate = (AVRational){25, 1};
1801 for (k = 0; k < physical_track->sequence->structural_components_count; k++) {
1802 if (!(mxf_tc = mxf_resolve_timecode_component(mxf, &physical_track->sequence->structural_components_refs[k])))
1805 flags = mxf_tc->drop_frame == 1 ? AV_TIMECODE_FLAG_DROPFRAME : 0;
1806 /* scale sourceclip start_position to match physical track edit rate */
1807 start_position = av_rescale_q(sourceclip->start_position,
1808 physical_track->edit_rate,
1809 source_track->edit_rate);
1811 if (av_timecode_init(&tc, mxf_tc->rate, flags, start_position + mxf_tc->start_frame, mxf->fc) == 0) {
1812 mxf_add_timecode_metadata(&st->metadata, "timecode", &tc);
1822 static int mxf_parse_structural_metadata(MXFContext *mxf)
1824 MXFPackage *material_package = NULL;
1827 av_log(mxf->fc, AV_LOG_TRACE, "metadata sets count %d\n", mxf->metadata_sets_count);
1828 /* TODO: handle multiple material packages (OP3x) */
1829 for (i = 0; i < mxf->packages_count; i++) {
1830 material_package = mxf_resolve_strong_ref(mxf, &mxf->packages_refs[i], MaterialPackage);
1831 if (material_package) break;
1833 if (!material_package) {
1834 av_log(mxf->fc, AV_LOG_ERROR, "no material package found\n");
1835 return AVERROR_INVALIDDATA;
1838 mxf_add_umid_metadata(&mxf->fc->metadata, "material_package_umid", material_package);
1839 if (material_package->name && material_package->name[0])
1840 av_dict_set(&mxf->fc->metadata, "material_package_name", material_package->name, 0);
1841 mxf_parse_package_comments(mxf, &mxf->fc->metadata, material_package);
1843 for (i = 0; i < material_package->tracks_count; i++) {
1844 MXFPackage *source_package = NULL;
1845 MXFTrack *material_track = NULL;
1846 MXFTrack *source_track = NULL;
1847 MXFTrack *temp_track = NULL;
1848 MXFDescriptor *descriptor = NULL;
1849 MXFStructuralComponent *component = NULL;
1850 MXFTimecodeComponent *mxf_tc = NULL;
1851 UID *essence_container_ul = NULL;
1852 const MXFCodecUL *codec_ul = NULL;
1853 const MXFCodecUL *container_ul = NULL;
1854 const MXFCodecUL *pix_fmt_ul = NULL;
1859 if (!(material_track = mxf_resolve_strong_ref(mxf, &material_package->tracks_refs[i], Track))) {
1860 av_log(mxf->fc, AV_LOG_ERROR, "could not resolve material track strong ref\n");
1864 if ((component = mxf_resolve_strong_ref(mxf, &material_track->sequence_ref, TimecodeComponent))) {
1865 mxf_tc = (MXFTimecodeComponent*)component;
1866 flags = mxf_tc->drop_frame == 1 ? AV_TIMECODE_FLAG_DROPFRAME : 0;
1867 if (av_timecode_init(&tc, mxf_tc->rate, flags, mxf_tc->start_frame, mxf->fc) == 0) {
1868 mxf_add_timecode_metadata(&mxf->fc->metadata, "timecode", &tc);
1872 if (!(material_track->sequence = mxf_resolve_strong_ref(mxf, &material_track->sequence_ref, Sequence))) {
1873 av_log(mxf->fc, AV_LOG_ERROR, "could not resolve material track sequence strong ref\n");
1877 for (j = 0; j < material_track->sequence->structural_components_count; j++) {
1878 component = mxf_resolve_strong_ref(mxf, &material_track->sequence->structural_components_refs[j], TimecodeComponent);
1882 mxf_tc = (MXFTimecodeComponent*)component;
1883 flags = mxf_tc->drop_frame == 1 ? AV_TIMECODE_FLAG_DROPFRAME : 0;
1884 if (av_timecode_init(&tc, mxf_tc->rate, flags, mxf_tc->start_frame, mxf->fc) == 0) {
1885 mxf_add_timecode_metadata(&mxf->fc->metadata, "timecode", &tc);
1890 /* TODO: handle multiple source clips */
1891 for (j = 0; j < material_track->sequence->structural_components_count; j++) {
1892 component = mxf_resolve_sourceclip(mxf, &material_track->sequence->structural_components_refs[j]);
1896 source_package = mxf_resolve_source_package(mxf, component->source_package_uid);
1897 if (!source_package) {
1898 av_log(mxf->fc, AV_LOG_TRACE, "material track %d: no corresponding source package found\n", material_track->track_id);
1901 for (k = 0; k < source_package->tracks_count; k++) {
1902 if (!(temp_track = mxf_resolve_strong_ref(mxf, &source_package->tracks_refs[k], Track))) {
1903 av_log(mxf->fc, AV_LOG_ERROR, "could not resolve source track strong ref\n");
1904 ret = AVERROR_INVALIDDATA;
1907 if (temp_track->track_id == component->source_track_id) {
1908 source_track = temp_track;
1912 if (!source_track) {
1913 av_log(mxf->fc, AV_LOG_ERROR, "material track %d: no corresponding source track found\n", material_track->track_id);
1917 if (!source_track || !component || !source_package)
1920 if (!(source_track->sequence = mxf_resolve_strong_ref(mxf, &source_track->sequence_ref, Sequence))) {
1921 av_log(mxf->fc, AV_LOG_ERROR, "could not resolve source track sequence strong ref\n");
1922 ret = AVERROR_INVALIDDATA;
1926 /* 0001GL00.MXF.A1.mxf_opatom.mxf has the same SourcePackageID as 0001GL.MXF.V1.mxf_opatom.mxf
1927 * This would result in both files appearing to have two streams. Work around this by sanity checking DataDefinition */
1928 if (memcmp(material_track->sequence->data_definition_ul, source_track->sequence->data_definition_ul, 16)) {
1929 av_log(mxf->fc, AV_LOG_ERROR, "material track %d: DataDefinition mismatch\n", material_track->track_id);
1933 st = avformat_new_stream(mxf->fc, NULL);
1935 av_log(mxf->fc, AV_LOG_ERROR, "could not allocate stream\n");
1936 ret = AVERROR(ENOMEM);
1939 st->id = source_track->track_id;
1940 st->priv_data = source_track;
1942 source_package->descriptor = mxf_resolve_strong_ref(mxf, &source_package->descriptor_ref, AnyType);
1943 descriptor = mxf_resolve_multidescriptor(mxf, source_package->descriptor, source_track->track_id);
1945 /* A SourceClip from a EssenceGroup may only be a single frame of essence data. The clips duration is then how many
1946 * frames its suppose to repeat for. Descriptor->duration, if present, contains the real duration of the essence data */
1947 if (descriptor && descriptor->duration != AV_NOPTS_VALUE)
1948 source_track->original_duration = st->duration = FFMIN(descriptor->duration, component->duration);
1950 source_track->original_duration = st->duration = component->duration;
1952 if (st->duration == -1)
1953 st->duration = AV_NOPTS_VALUE;
1954 st->start_time = component->start_position;
1955 if (material_track->edit_rate.num <= 0 ||
1956 material_track->edit_rate.den <= 0) {
1957 av_log(mxf->fc, AV_LOG_WARNING,
1958 "Invalid edit rate (%d/%d) found on stream #%d, "
1959 "defaulting to 25/1\n",
1960 material_track->edit_rate.num,
1961 material_track->edit_rate.den, st->index);
1962 material_track->edit_rate = (AVRational){25, 1};
1964 avpriv_set_pts_info(st, 64, material_track->edit_rate.den, material_track->edit_rate.num);
1966 /* ensure SourceTrack EditRate == MaterialTrack EditRate since only
1967 * the former is accessible via st->priv_data */
1968 source_track->edit_rate = material_track->edit_rate;
1970 PRINT_KEY(mxf->fc, "data definition ul", source_track->sequence->data_definition_ul);
1971 codec_ul = mxf_get_codec_ul(ff_mxf_data_definition_uls, &source_track->sequence->data_definition_ul);
1972 st->codecpar->codec_type = codec_ul->id;
1975 av_log(mxf->fc, AV_LOG_INFO, "source track %d: stream %d, no descriptor found\n", source_track->track_id, st->index);
1978 PRINT_KEY(mxf->fc, "essence codec ul", descriptor->essence_codec_ul);
1979 PRINT_KEY(mxf->fc, "essence container ul", descriptor->essence_container_ul);
1980 essence_container_ul = &descriptor->essence_container_ul;
1981 /* HACK: replacing the original key with mxf_encrypted_essence_container
1982 * is not allowed according to s429-6, try to find correct information anyway */
1983 if (IS_KLV_KEY(essence_container_ul, mxf_encrypted_essence_container)) {
1984 av_log(mxf->fc, AV_LOG_INFO, "broken encrypted mxf file\n");
1985 for (k = 0; k < mxf->metadata_sets_count; k++) {
1986 MXFMetadataSet *metadata = mxf->metadata_sets[k];
1987 if (metadata->type == CryptoContext) {
1988 essence_container_ul = &((MXFCryptoContext *)metadata)->source_container_ul;
1994 /* TODO: drop PictureEssenceCoding and SoundEssenceCompression, only check EssenceContainer */
1995 codec_ul = mxf_get_codec_ul(ff_mxf_codec_uls, &descriptor->essence_codec_ul);
1996 st->codecpar->codec_id = (enum AVCodecID)codec_ul->id;
1997 if (st->codecpar->codec_id == AV_CODEC_ID_NONE) {
1998 codec_ul = mxf_get_codec_ul(ff_mxf_codec_uls, &descriptor->codec_ul);
1999 st->codecpar->codec_id = (enum AVCodecID)codec_ul->id;
2002 av_log(mxf->fc, AV_LOG_VERBOSE, "%s: Universal Label: ",
2003 avcodec_get_name(st->codecpar->codec_id));
2004 for (k = 0; k < 16; k++) {
2005 av_log(mxf->fc, AV_LOG_VERBOSE, "%.2x",
2006 descriptor->essence_codec_ul[k]);
2007 if (!(k+1 & 19) || k == 5)
2008 av_log(mxf->fc, AV_LOG_VERBOSE, ".");
2010 av_log(mxf->fc, AV_LOG_VERBOSE, "\n");
2012 mxf_add_umid_metadata(&st->metadata, "file_package_umid", source_package);
2013 if (source_package->name && source_package->name[0])
2014 av_dict_set(&st->metadata, "file_package_name", source_package->name, 0);
2016 mxf_parse_physical_source_package(mxf, source_track, st);
2018 if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
2019 source_track->intra_only = mxf_is_intra_only(descriptor);
2020 container_ul = mxf_get_codec_ul(mxf_picture_essence_container_uls, essence_container_ul);
2021 if (st->codecpar->codec_id == AV_CODEC_ID_NONE)
2022 st->codecpar->codec_id = container_ul->id;
2023 st->codecpar->width = descriptor->width;
2024 st->codecpar->height = descriptor->height; /* Field height, not frame height */
2025 switch (descriptor->frame_layout) {
2027 st->codecpar->field_order = AV_FIELD_PROGRESSIVE;
2030 /* Every other line is stored and needs to be duplicated. */
2031 av_log(mxf->fc, AV_LOG_INFO, "OneField frame layout isn't currently supported\n");
2032 break; /* The correct thing to do here is fall through, but by breaking we might be
2033 able to decode some streams at half the vertical resolution, rather than not al all.
2034 It's also for compatibility with the old behavior. */
2037 case SegmentedFrame:
2038 st->codecpar->field_order = AV_FIELD_PROGRESSIVE;
2039 case SeparateFields:
2040 switch (descriptor->field_dominance) {
2042 st->codecpar->field_order = AV_FIELD_TT;
2045 st->codecpar->field_order = AV_FIELD_BB;
2048 avpriv_request_sample(mxf->fc,
2049 "Field dominance %d support",
2050 descriptor->field_dominance);
2051 case 0: // we already have many samples with field_dominance == unknown
2054 /* Turn field height into frame height. */
2055 st->codecpar->height *= 2;
2058 av_log(mxf->fc, AV_LOG_INFO, "Unknown frame layout type: %d\n", descriptor->frame_layout);
2060 if (st->codecpar->codec_id == AV_CODEC_ID_RAWVIDEO) {
2061 st->codecpar->format = descriptor->pix_fmt;
2062 if (st->codecpar->format == AV_PIX_FMT_NONE) {
2063 pix_fmt_ul = mxf_get_codec_ul(ff_mxf_pixel_format_uls,
2064 &descriptor->essence_codec_ul);
2065 st->codecpar->format = (enum AVPixelFormat)pix_fmt_ul->id;
2066 if (st->codecpar->format== AV_PIX_FMT_NONE) {
2067 st->codecpar->codec_tag = mxf_get_codec_ul(ff_mxf_codec_tag_uls,
2068 &descriptor->essence_codec_ul)->id;
2069 if (!st->codecpar->codec_tag) {
2070 /* support files created before RP224v10 by defaulting to UYVY422
2071 if subsampling is 4:2:2 and component depth is 8-bit */
2072 if (descriptor->horiz_subsampling == 2 &&
2073 descriptor->vert_subsampling == 1 &&
2074 descriptor->component_depth == 8) {
2075 st->codecpar->format = AV_PIX_FMT_UYVY422;
2081 st->need_parsing = AVSTREAM_PARSE_HEADERS;
2082 if (material_track->sequence->origin) {
2083 av_dict_set_int(&st->metadata, "material_track_origin", material_track->sequence->origin, 0);
2085 if (source_track->sequence->origin) {
2086 av_dict_set_int(&st->metadata, "source_track_origin", source_track->sequence->origin, 0);
2088 if (descriptor->aspect_ratio.num && descriptor->aspect_ratio.den)
2089 st->display_aspect_ratio = descriptor->aspect_ratio;
2090 } else if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
2091 container_ul = mxf_get_codec_ul(mxf_sound_essence_container_uls, essence_container_ul);
2092 /* Only overwrite existing codec ID if it is unset or A-law, which is the default according to SMPTE RP 224. */
2093 if (st->codecpar->codec_id == AV_CODEC_ID_NONE || (st->codecpar->codec_id == AV_CODEC_ID_PCM_ALAW && (enum AVCodecID)container_ul->id != AV_CODEC_ID_NONE))
2094 st->codecpar->codec_id = (enum AVCodecID)container_ul->id;
2095 st->codecpar->channels = descriptor->channels;
2096 st->codecpar->bits_per_coded_sample = descriptor->bits_per_sample;
2098 if (descriptor->sample_rate.den > 0) {
2099 st->codecpar->sample_rate = descriptor->sample_rate.num / descriptor->sample_rate.den;
2100 avpriv_set_pts_info(st, 64, descriptor->sample_rate.den, descriptor->sample_rate.num);
2102 av_log(mxf->fc, AV_LOG_WARNING, "invalid sample rate (%d/%d) "
2103 "found for stream #%d, time base forced to 1/48000\n",
2104 descriptor->sample_rate.num, descriptor->sample_rate.den,
2106 avpriv_set_pts_info(st, 64, 1, 48000);
2109 /* if duration is set, rescale it from EditRate to SampleRate */
2110 if (st->duration != AV_NOPTS_VALUE)
2111 st->duration = av_rescale_q(st->duration,
2112 av_inv_q(material_track->edit_rate),
2115 /* TODO: implement AV_CODEC_ID_RAWAUDIO */
2116 if (st->codecpar->codec_id == AV_CODEC_ID_PCM_S16LE) {
2117 if (descriptor->bits_per_sample > 16 && descriptor->bits_per_sample <= 24)
2118 st->codecpar->codec_id = AV_CODEC_ID_PCM_S24LE;
2119 else if (descriptor->bits_per_sample == 32)
2120 st->codecpar->codec_id = AV_CODEC_ID_PCM_S32LE;
2121 } else if (st->codecpar->codec_id == AV_CODEC_ID_PCM_S16BE) {
2122 if (descriptor->bits_per_sample > 16 && descriptor->bits_per_sample <= 24)
2123 st->codecpar->codec_id = AV_CODEC_ID_PCM_S24BE;
2124 else if (descriptor->bits_per_sample == 32)
2125 st->codecpar->codec_id = AV_CODEC_ID_PCM_S32BE;
2126 } else if (st->codecpar->codec_id == AV_CODEC_ID_MP2) {
2127 st->need_parsing = AVSTREAM_PARSE_FULL;
2129 } else if (st->codecpar->codec_type == AVMEDIA_TYPE_DATA) {
2130 int codec_id = mxf_get_codec_ul(mxf_data_essence_container_uls,
2131 essence_container_ul)->id;
2132 if (codec_id >= 0 &&
2133 codec_id < FF_ARRAY_ELEMS(mxf_data_essence_descriptor)) {
2134 av_dict_set(&st->metadata, "data_type",
2135 mxf_data_essence_descriptor[codec_id], 0);
2138 if (descriptor->extradata) {
2139 if (!ff_alloc_extradata(st->codecpar, descriptor->extradata_size)) {
2140 memcpy(st->codecpar->extradata, descriptor->extradata, descriptor->extradata_size);
2142 } else if (st->codecpar->codec_id == AV_CODEC_ID_H264) {
2143 int coded_width = mxf_get_codec_ul(mxf_intra_only_picture_coded_width,
2144 &descriptor->essence_codec_ul)->id;
2146 st->codecpar->width = coded_width;
2147 ret = ff_generate_avci_extradata(st);
2151 if (st->codecpar->codec_type != AVMEDIA_TYPE_DATA && (*essence_container_ul)[15] > 0x01) {
2152 /* TODO: decode timestamps */
2153 st->need_parsing = AVSTREAM_PARSE_TIMESTAMPS;
2162 static int mxf_timestamp_to_str(uint64_t timestamp, char **str)
2164 struct tm time = { 0 };
2165 time.tm_year = (timestamp >> 48) - 1900;
2166 time.tm_mon = (timestamp >> 40 & 0xFF) - 1;
2167 time.tm_mday = (timestamp >> 32 & 0xFF);
2168 time.tm_hour = (timestamp >> 24 & 0xFF);
2169 time.tm_min = (timestamp >> 16 & 0xFF);
2170 time.tm_sec = (timestamp >> 8 & 0xFF);
2172 /* msvcrt versions of strftime calls the invalid parameter handler
2173 * (aborting the process if one isn't set) if the parameters are out
2175 time.tm_mon = av_clip(time.tm_mon, 0, 11);
2176 time.tm_mday = av_clip(time.tm_mday, 1, 31);
2177 time.tm_hour = av_clip(time.tm_hour, 0, 23);
2178 time.tm_min = av_clip(time.tm_min, 0, 59);
2179 time.tm_sec = av_clip(time.tm_sec, 0, 59);
2181 *str = av_mallocz(32);
2183 return AVERROR(ENOMEM);
2184 if (!strftime(*str, 32, "%Y-%m-%d %H:%M:%S", &time))
2190 #define SET_STR_METADATA(pb, name, str) do { \
2191 if ((ret = mxf_read_utf16be_string(pb, size, &str)) < 0) \
2193 av_dict_set(&s->metadata, name, str, AV_DICT_DONT_STRDUP_VAL); \
2196 #define SET_UID_METADATA(pb, name, var, str) do { \
2197 avio_read(pb, var, 16); \
2198 if ((ret = mxf_uid_to_str(var, &str)) < 0) \
2200 av_dict_set(&s->metadata, name, str, AV_DICT_DONT_STRDUP_VAL); \
2203 #define SET_TS_METADATA(pb, name, var, str) do { \
2204 var = avio_rb64(pb); \
2205 if ((ret = mxf_timestamp_to_str(var, &str)) < 0) \
2207 av_dict_set(&s->metadata, name, str, AV_DICT_DONT_STRDUP_VAL); \
2210 static int mxf_read_identification_metadata(void *arg, AVIOContext *pb, int tag, int size, UID _uid, int64_t klv_offset)
2212 MXFContext *mxf = arg;
2213 AVFormatContext *s = mxf->fc;
2220 SET_STR_METADATA(pb, "company_name", str);
2223 SET_STR_METADATA(pb, "product_name", str);
2226 SET_STR_METADATA(pb, "product_version", str);
2229 SET_UID_METADATA(pb, "product_uid", uid, str);
2232 SET_TS_METADATA(pb, "modification_date", ts, str);
2235 SET_STR_METADATA(pb, "application_platform", str);
2238 SET_UID_METADATA(pb, "generation_uid", uid, str);
2241 SET_UID_METADATA(pb, "uid", uid, str);
2247 static int mxf_read_preface_metadata(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
2249 MXFContext *mxf = arg;
2250 AVFormatContext *s = mxf->fc;
2254 if (tag >= 0x8000 && (IS_KLV_KEY(uid, mxf_avid_project_name))) {
2255 SET_STR_METADATA(pb, "project_name", str);
2260 static const MXFMetadataReadTableEntry mxf_metadata_read_table[] = {
2261 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x05,0x01,0x00 }, mxf_read_primer_pack },
2262 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02,0x01,0x00 }, mxf_read_partition_pack },
2263 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02,0x02,0x00 }, mxf_read_partition_pack },
2264 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02,0x03,0x00 }, mxf_read_partition_pack },
2265 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02,0x04,0x00 }, mxf_read_partition_pack },
2266 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x03,0x01,0x00 }, mxf_read_partition_pack },
2267 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x03,0x02,0x00 }, mxf_read_partition_pack },
2268 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x03,0x03,0x00 }, mxf_read_partition_pack },
2269 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x03,0x04,0x00 }, mxf_read_partition_pack },
2270 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x04,0x02,0x00 }, mxf_read_partition_pack },
2271 { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x04,0x04,0x00 }, mxf_read_partition_pack },
2272 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x2f,0x00 }, mxf_read_preface_metadata },
2273 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x30,0x00 }, mxf_read_identification_metadata },
2274 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x18,0x00 }, mxf_read_content_storage, 0, AnyType },
2275 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x37,0x00 }, mxf_read_package, sizeof(MXFPackage), SourcePackage },
2276 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x36,0x00 }, mxf_read_package, sizeof(MXFPackage), MaterialPackage },
2277 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x0f,0x00 }, mxf_read_sequence, sizeof(MXFSequence), Sequence },
2278 { { 0x06,0x0E,0x2B,0x34,0x02,0x53,0x01,0x01,0x0D,0x01,0x01,0x01,0x01,0x01,0x05,0x00 }, mxf_read_essence_group, sizeof(MXFEssenceGroup), EssenceGroup},
2279 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x11,0x00 }, mxf_read_source_clip, sizeof(MXFStructuralComponent), SourceClip },
2280 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x3f,0x00 }, mxf_read_tagged_value, sizeof(MXFTaggedValue), TaggedValue },
2281 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x44,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), MultipleDescriptor },
2282 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x42,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* Generic Sound */
2283 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x28,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* CDCI */
2284 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x29,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* RGBA */
2285 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x48,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* Wave */
2286 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x47,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* AES3 */
2287 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x51,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* MPEG2VideoDescriptor */
2288 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x5c,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* VANC/VBI - SMPTE 436M */
2289 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x5e,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* MPEG2AudioDescriptor */
2290 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x3A,0x00 }, mxf_read_track, sizeof(MXFTrack), Track }, /* Static Track */
2291 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x3B,0x00 }, mxf_read_track, sizeof(MXFTrack), Track }, /* Generic Track */
2292 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x14,0x00 }, mxf_read_timecode_component, sizeof(MXFTimecodeComponent), TimecodeComponent },
2293 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x0c,0x00 }, mxf_read_pulldown_component, sizeof(MXFPulldownComponent), PulldownComponent },
2294 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x04,0x01,0x02,0x02,0x00,0x00 }, mxf_read_cryptographic_context, sizeof(MXFCryptoContext), CryptoContext },
2295 { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x10,0x01,0x00 }, mxf_read_index_table_segment, sizeof(MXFIndexTableSegment), IndexTableSegment },
2296 { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, NULL, 0, AnyType },
2299 static int mxf_metadataset_init(MXFMetadataSet *ctx, enum MXFMetadataSetType type)
2302 case MultipleDescriptor:
2304 ((MXFDescriptor*)ctx)->pix_fmt = AV_PIX_FMT_NONE;
2305 ((MXFDescriptor*)ctx)->duration = AV_NOPTS_VALUE;
2313 static int mxf_read_local_tags(MXFContext *mxf, KLVPacket *klv, MXFMetadataReadFunc *read_child, int ctx_size, enum MXFMetadataSetType type)
2315 AVIOContext *pb = mxf->fc->pb;
2316 MXFMetadataSet *ctx = ctx_size ? av_mallocz(ctx_size) : mxf;
2317 uint64_t klv_end = avio_tell(pb) + klv->length;
2320 return AVERROR(ENOMEM);
2321 mxf_metadataset_init(ctx, type);
2322 while (avio_tell(pb) + 4 < klv_end && !avio_feof(pb)) {
2324 int tag = avio_rb16(pb);
2325 int size = avio_rb16(pb); /* KLV specified by 0x53 */
2326 uint64_t next = avio_tell(pb) + size;
2329 av_log(mxf->fc, AV_LOG_TRACE, "local tag %#04x size %d\n", tag, size);
2330 if (!size) { /* ignore empty tag, needed for some files with empty UMID tag */
2331 av_log(mxf->fc, AV_LOG_ERROR, "local tag %#04x with 0 size\n", tag);
2334 if (tag > 0x7FFF) { /* dynamic tag */
2336 for (i = 0; i < mxf->local_tags_count; i++) {
2337 int local_tag = AV_RB16(mxf->local_tags+i*18);
2338 if (local_tag == tag) {
2339 memcpy(uid, mxf->local_tags+i*18+2, 16);
2340 av_log(mxf->fc, AV_LOG_TRACE, "local tag %#04x\n", local_tag);
2341 PRINT_KEY(mxf->fc, "uid", uid);
2345 if (ctx_size && tag == 0x3C0A) {
2346 avio_read(pb, ctx->uid, 16);
2347 } else if ((ret = read_child(ctx, pb, tag, size, uid, -1)) < 0) {
2348 mxf_free_metadataset(&ctx, !!ctx_size);
2352 /* Accept the 64k local set limit being exceeded (Avid). Don't accept
2353 * it extending past the end of the KLV though (zzuf5.mxf). */
2354 if (avio_tell(pb) > klv_end) {
2357 mxf_free_metadataset(&ctx, !!ctx_size);
2360 av_log(mxf->fc, AV_LOG_ERROR,
2361 "local tag %#04x extends past end of local set @ %#"PRIx64"\n",
2363 return AVERROR_INVALIDDATA;
2364 } else if (avio_tell(pb) <= next) /* only seek forward, else this can loop for a long time */
2365 avio_seek(pb, next, SEEK_SET);
2367 if (ctx_size) ctx->type = type;
2368 return ctx_size ? mxf_add_metadata_set(mxf, ctx) : 0;
2372 * Matches any partition pack key, in other words:
2376 * @return non-zero if the key is a partition pack key, zero otherwise
2378 static int mxf_is_partition_pack_key(UID key)
2380 //NOTE: this is a little lax since it doesn't constraint key[14]
2381 return !memcmp(key, mxf_header_partition_pack_key, 13) &&
2382 key[13] >= 2 && key[13] <= 4;
2386 * Parses a metadata KLV
2387 * @return <0 on error, 0 otherwise
2389 static int mxf_parse_klv(MXFContext *mxf, KLVPacket klv, MXFMetadataReadFunc *read,
2390 int ctx_size, enum MXFMetadataSetType type)
2392 AVFormatContext *s = mxf->fc;
2394 if (klv.key[5] == 0x53) {
2395 res = mxf_read_local_tags(mxf, &klv, read, ctx_size, type);
2397 uint64_t next = avio_tell(s->pb) + klv.length;
2398 res = read(mxf, s->pb, 0, klv.length, klv.key, klv.offset);
2400 /* only seek forward, else this can loop for a long time */
2401 if (avio_tell(s->pb) > next) {
2402 av_log(s, AV_LOG_ERROR, "read past end of KLV @ %#"PRIx64"\n",
2404 return AVERROR_INVALIDDATA;
2407 avio_seek(s->pb, next, SEEK_SET);
2410 av_log(s, AV_LOG_ERROR, "error reading header metadata\n");
2417 * Seeks to the previous partition and parses it, if possible
2418 * @return <= 0 if we should stop parsing, > 0 if we should keep going
2420 static int mxf_seek_to_previous_partition(MXFContext *mxf)
2422 AVIOContext *pb = mxf->fc->pb;
2424 int64_t current_partition_ofs;
2427 if (!mxf->current_partition ||
2428 mxf->run_in + mxf->current_partition->previous_partition <= mxf->last_forward_tell)
2429 return 0; /* we've parsed all partitions */
2431 /* seek to previous partition */
2432 current_partition_ofs = mxf->current_partition->pack_ofs; //includes run-in
2433 avio_seek(pb, mxf->run_in + mxf->current_partition->previous_partition, SEEK_SET);
2434 mxf->current_partition = NULL;
2436 av_log(mxf->fc, AV_LOG_TRACE, "seeking to previous partition\n");
2438 /* Make sure this is actually a PartitionPack, and if so parse it.
2441 if ((ret = klv_read_packet(&klv, pb)) < 0) {
2442 av_log(mxf->fc, AV_LOG_ERROR, "failed to read PartitionPack KLV\n");
2446 if (!mxf_is_partition_pack_key(klv.key)) {
2447 av_log(mxf->fc, AV_LOG_ERROR, "PreviousPartition @ %" PRIx64 " isn't a PartitionPack\n", klv.offset);
2448 return AVERROR_INVALIDDATA;
2451 /* We can't just check ofs >= current_partition_ofs because PreviousPartition
2452 * can point to just before the current partition, causing klv_read_packet()
2453 * to sync back up to it. See deadlock3.mxf
2455 if (klv.offset >= current_partition_ofs) {
2456 av_log(mxf->fc, AV_LOG_ERROR, "PreviousPartition for PartitionPack @ %"
2457 PRIx64 " indirectly points to itself\n", current_partition_ofs);
2458 return AVERROR_INVALIDDATA;
2461 if ((ret = mxf_parse_klv(mxf, klv, mxf_read_partition_pack, 0, 0)) < 0)
2468 * Called when essence is encountered
2469 * @return <= 0 if we should stop parsing, > 0 if we should keep going
2471 static int mxf_parse_handle_essence(MXFContext *mxf)
2473 AVIOContext *pb = mxf->fc->pb;
2476 if (mxf->parsing_backward) {
2477 return mxf_seek_to_previous_partition(mxf);
2479 if (!mxf->footer_partition) {
2480 av_log(mxf->fc, AV_LOG_TRACE, "no FooterPartition\n");
2484 av_log(mxf->fc, AV_LOG_TRACE, "seeking to FooterPartition\n");
2486 /* remember where we were so we don't end up seeking further back than this */
2487 mxf->last_forward_tell = avio_tell(pb);
2489 if (!pb->seekable) {
2490 av_log(mxf->fc, AV_LOG_INFO, "file is not seekable - not parsing FooterPartition\n");
2494 /* seek to FooterPartition and parse backward */
2495 if ((ret = avio_seek(pb, mxf->run_in + mxf->footer_partition, SEEK_SET)) < 0) {
2496 av_log(mxf->fc, AV_LOG_ERROR,
2497 "failed to seek to FooterPartition @ 0x%" PRIx64
2498 " (%"PRId64") - partial file?\n",
2499 mxf->run_in + mxf->footer_partition, ret);
2503 mxf->current_partition = NULL;
2504 mxf->parsing_backward = 1;
2511 * Called when the next partition or EOF is encountered
2512 * @return <= 0 if we should stop parsing, > 0 if we should keep going
2514 static int mxf_parse_handle_partition_or_eof(MXFContext *mxf)
2516 return mxf->parsing_backward ? mxf_seek_to_previous_partition(mxf) : 1;
2520 * Figures out the proper offset and length of the essence container in each partition
2522 static void mxf_compute_essence_containers(MXFContext *mxf)
2526 /* everything is already correct */
2527 if (mxf->op == OPAtom)
2530 for (x = 0; x < mxf->partitions_count; x++) {
2531 MXFPartition *p = &mxf->partitions[x];
2534 continue; /* BodySID == 0 -> no essence */
2536 if (x >= mxf->partitions_count - 1)
2537 break; /* FooterPartition - can't compute length (and we don't need to) */
2539 /* essence container spans to the next partition */
2540 p->essence_length = mxf->partitions[x+1].this_partition - p->essence_offset;
2542 if (p->essence_length < 0) {
2543 /* next ThisPartition < essence_offset */
2544 p->essence_length = 0;
2545 av_log(mxf->fc, AV_LOG_ERROR,
2546 "partition %i: bad ThisPartition = %"PRIX64"\n",
2547 x+1, mxf->partitions[x+1].this_partition);
2552 static int64_t round_to_kag(int64_t position, int kag_size)
2554 /* TODO: account for run-in? the spec isn't clear whether KAG should account for it */
2555 /* NOTE: kag_size may be any integer between 1 - 2^10 */
2556 int64_t ret = (position / kag_size) * kag_size;
2557 return ret == position ? ret : ret + kag_size;
2560 static int is_pcm(enum AVCodecID codec_id)
2562 /* we only care about "normal" PCM codecs until we get samples */
2563 return codec_id >= AV_CODEC_ID_PCM_S16LE && codec_id < AV_CODEC_ID_PCM_S24DAUD;
2567 * Deal with the case where for some audio atoms EditUnitByteCount is
2568 * very small (2, 4..). In those cases we should read more than one
2569 * sample per call to mxf_read_packet().
2571 static void mxf_handle_small_eubc(AVFormatContext *s)
2573 MXFContext *mxf = s->priv_data;
2575 /* assuming non-OPAtom == frame wrapped
2576 * no sane writer would wrap 2 byte PCM packets with 20 byte headers.. */
2577 if (mxf->op != OPAtom)
2580 /* expect PCM with exactly one index table segment and a small (< 32) EUBC */
2581 if (s->nb_streams != 1 ||
2582 s->streams[0]->codecpar->codec_type != AVMEDIA_TYPE_AUDIO ||
2583 !is_pcm(s->streams[0]->codecpar->codec_id) ||
2584 mxf->nb_index_tables != 1 ||
2585 mxf->index_tables[0].nb_segments != 1 ||
2586 mxf->index_tables[0].segments[0]->edit_unit_byte_count >= 32)
2589 /* arbitrarily default to 48 kHz PAL audio frame size */
2590 /* TODO: We could compute this from the ratio between the audio
2591 * and video edit rates for 48 kHz NTSC we could use the
2592 * 1802-1802-1802-1802-1801 pattern. */
2593 mxf->edit_units_per_packet = 1920;
2597 * Deal with the case where OPAtom files does not have any IndexTableSegments.
2599 static int mxf_handle_missing_index_segment(MXFContext *mxf)
2601 AVFormatContext *s = mxf->fc;
2602 AVStream *st = NULL;
2603 MXFIndexTableSegment *segment = NULL;
2604 MXFPartition *p = NULL;
2605 int essence_partition_count = 0;
2608 if (mxf->op != OPAtom)
2611 /* TODO: support raw video without an index if they exist */
2612 if (s->nb_streams != 1 || s->streams[0]->codecpar->codec_type != AVMEDIA_TYPE_AUDIO || !is_pcm(s->streams[0]->codecpar->codec_id))
2615 /* check if file already has a IndexTableSegment */
2616 for (i = 0; i < mxf->metadata_sets_count; i++) {
2617 if (mxf->metadata_sets[i]->type == IndexTableSegment)
2621 /* find the essence partition */
2622 for (i = 0; i < mxf->partitions_count; i++) {
2623 /* BodySID == 0 -> no essence */
2624 if (!mxf->partitions[i].body_sid)
2627 p = &mxf->partitions[i];
2628 essence_partition_count++;
2631 /* only handle files with a single essence partition */
2632 if (essence_partition_count != 1)
2635 if (!(segment = av_mallocz(sizeof(*segment))))
2636 return AVERROR(ENOMEM);
2638 if ((ret = mxf_add_metadata_set(mxf, segment))) {
2639 mxf_free_metadataset((MXFMetadataSet**)&segment, 1);
2644 segment->type = IndexTableSegment;
2645 /* stream will be treated as small EditUnitByteCount */
2646 segment->edit_unit_byte_count = (av_get_bits_per_sample(st->codecpar->codec_id) * st->codecpar->channels) >> 3;
2647 segment->index_start_position = 0;
2648 segment->index_duration = s->streams[0]->duration;
2649 segment->index_sid = p->index_sid;
2650 segment->body_sid = p->body_sid;
2654 static void mxf_read_random_index_pack(AVFormatContext *s)
2656 MXFContext *mxf = s->priv_data;
2658 int64_t file_size, max_rip_length, min_rip_length;
2661 if (!s->pb->seekable)
2664 file_size = avio_size(s->pb);
2666 /* S377m says to check the RIP length for "silly" values, without defining "silly".
2667 * The limit below assumes a file with nothing but partition packs and a RIP.
2668 * Before changing this, consider that a muxer may place each sample in its own partition.
2670 * 105 is the size of the smallest possible PartitionPack
2671 * 12 is the size of each RIP entry
2672 * 28 is the size of the RIP header and footer, assuming an 8-byte BER
2674 max_rip_length = ((file_size - mxf->run_in) / 105) * 12 + 28;
2675 max_rip_length = FFMIN(max_rip_length, INT_MAX); //2 GiB and up is also silly
2677 /* We're only interested in RIPs with at least two entries.. */
2678 min_rip_length = 16+1+24+4;
2680 /* See S377m section 11 */
2681 avio_seek(s->pb, file_size - 4, SEEK_SET);
2682 length = avio_rb32(s->pb);
2684 if (length < min_rip_length || length > max_rip_length)
2686 avio_seek(s->pb, file_size - length, SEEK_SET);
2687 if (klv_read_packet(&klv, s->pb) < 0 ||
2688 !IS_KLV_KEY(klv.key, mxf_random_index_pack_key) ||
2689 klv.length != length - 20)
2692 avio_skip(s->pb, klv.length - 12);
2693 mxf->footer_partition = avio_rb64(s->pb);
2696 if (mxf->run_in + mxf->footer_partition >= file_size) {
2697 av_log(s, AV_LOG_WARNING, "bad FooterPartition in RIP - ignoring\n");
2698 mxf->footer_partition = 0;
2702 avio_seek(s->pb, mxf->run_in, SEEK_SET);
2705 static int mxf_read_header(AVFormatContext *s)
2707 MXFContext *mxf = s->priv_data;
2709 int64_t essence_offset = 0;
2712 mxf->last_forward_tell = INT64_MAX;
2713 mxf->edit_units_per_packet = 1;
2715 if (!mxf_read_sync(s->pb, mxf_header_partition_pack_key, 14)) {
2716 av_log(s, AV_LOG_ERROR, "could not find header partition pack key\n");
2717 return AVERROR_INVALIDDATA;
2719 avio_seek(s->pb, -14, SEEK_CUR);
2721 mxf->run_in = avio_tell(s->pb);
2723 mxf_read_random_index_pack(s);
2725 while (!avio_feof(s->pb)) {
2726 const MXFMetadataReadTableEntry *metadata;
2728 if (klv_read_packet(&klv, s->pb) < 0) {
2729 /* EOF - seek to previous partition or stop */
2730 if(mxf_parse_handle_partition_or_eof(mxf) <= 0)
2736 PRINT_KEY(s, "read header", klv.key);
2737 av_log(s, AV_LOG_TRACE, "size %"PRIu64" offset %#"PRIx64"\n", klv.length, klv.offset);
2738 if (IS_KLV_KEY(klv.key, mxf_encrypted_triplet_key) ||
2739 IS_KLV_KEY(klv.key, mxf_essence_element_key) ||
2740 IS_KLV_KEY(klv.key, mxf_avid_essence_element_key) ||
2741 IS_KLV_KEY(klv.key, mxf_system_item_key)) {
2743 if (!mxf->current_partition) {
2744 av_log(mxf->fc, AV_LOG_ERROR, "found essence prior to first PartitionPack\n");
2745 return AVERROR_INVALIDDATA;
2748 if (!mxf->current_partition->essence_offset) {
2749 /* for OP1a we compute essence_offset
2750 * for OPAtom we point essence_offset after the KL (usually op1a_essence_offset + 20 or 25)
2751 * TODO: for OP1a we could eliminate this entire if statement, always stopping parsing at op1a_essence_offset
2752 * for OPAtom we still need the actual essence_offset though (the KL's length can vary)
2754 int64_t op1a_essence_offset =
2755 round_to_kag(mxf->current_partition->this_partition +
2756 mxf->current_partition->pack_length, mxf->current_partition->kag_size) +
2757 round_to_kag(mxf->current_partition->header_byte_count, mxf->current_partition->kag_size) +
2758 round_to_kag(mxf->current_partition->index_byte_count, mxf->current_partition->kag_size);
2760 if (mxf->op == OPAtom) {
2761 /* point essence_offset to the actual data
2762 * OPAtom has all the essence in one big KLV
2764 mxf->current_partition->essence_offset = avio_tell(s->pb);
2765 mxf->current_partition->essence_length = klv.length;
2767 /* NOTE: op1a_essence_offset may be less than to klv.offset (C0023S01.mxf) */
2768 mxf->current_partition->essence_offset = op1a_essence_offset;
2772 if (!essence_offset)
2773 essence_offset = klv.offset;
2775 /* seek to footer, previous partition or stop */
2776 if (mxf_parse_handle_essence(mxf) <= 0)
2779 } else if (mxf_is_partition_pack_key(klv.key) && mxf->current_partition) {
2780 /* next partition pack - keep going, seek to previous partition or stop */
2781 if(mxf_parse_handle_partition_or_eof(mxf) <= 0)
2783 else if (mxf->parsing_backward)
2785 /* we're still parsing forward. proceed to parsing this partition pack */
2788 for (metadata = mxf_metadata_read_table; metadata->read; metadata++) {
2789 if (IS_KLV_KEY(klv.key, metadata->key)) {
2790 if ((ret = mxf_parse_klv(mxf, klv, metadata->read, metadata->ctx_size, metadata->type)) < 0)
2795 if (!metadata->read) {
2796 av_log(s, AV_LOG_VERBOSE, "Dark key " PRIxUID "\n",
2798 avio_skip(s->pb, klv.length);
2801 /* FIXME avoid seek */
2802 if (!essence_offset) {
2803 av_log(s, AV_LOG_ERROR, "no essence\n");
2804 ret = AVERROR_INVALIDDATA;
2807 avio_seek(s->pb, essence_offset, SEEK_SET);
2809 mxf_compute_essence_containers(mxf);
2811 /* we need to do this before computing the index tables
2812 * to be able to fill in zero IndexDurations with st->duration */
2813 if ((ret = mxf_parse_structural_metadata(mxf)) < 0)
2816 mxf_handle_missing_index_segment(mxf);
2817 if ((ret = mxf_compute_index_tables(mxf)) < 0)
2820 if (mxf->nb_index_tables > 1) {
2821 /* TODO: look up which IndexSID to use via EssenceContainerData */
2822 av_log(mxf->fc, AV_LOG_INFO, "got %i index tables - only the first one (IndexSID %i) will be used\n",
2823 mxf->nb_index_tables, mxf->index_tables[0].index_sid);
2824 } else if (mxf->nb_index_tables == 0 && mxf->op == OPAtom) {
2825 av_log(mxf->fc, AV_LOG_ERROR, "cannot demux OPAtom without an index\n");
2826 ret = AVERROR_INVALIDDATA;
2830 mxf_handle_small_eubc(s);
2840 * Sets mxf->current_edit_unit based on what offset we're currently at.
2841 * @return next_ofs if OK, <0 on error
2843 static int64_t mxf_set_current_edit_unit(MXFContext *mxf, int64_t current_offset)
2845 int64_t last_ofs = -1, next_ofs = -1;
2846 MXFIndexTable *t = &mxf->index_tables[0];
2848 /* this is called from the OP1a demuxing logic, which means there
2849 * may be no index tables */
2850 if (mxf->nb_index_tables <= 0)
2853 /* find mxf->current_edit_unit so that the next edit unit starts ahead of current_offset */
2854 while (mxf->current_edit_unit >= 0) {
2855 if (mxf_edit_unit_absolute_offset(mxf, t, mxf->current_edit_unit + 1, NULL, &next_ofs, 0) < 0)
2858 if (next_ofs <= last_ofs) {
2859 /* large next_ofs didn't change or current_edit_unit wrapped
2860 * around this fixes the infinite loop on zzuf3.mxf */
2861 av_log(mxf->fc, AV_LOG_ERROR,
2862 "next_ofs didn't change. not deriving packet timestamps\n");
2866 if (next_ofs > current_offset)
2869 last_ofs = next_ofs;
2870 mxf->current_edit_unit++;
2873 /* not checking mxf->current_edit_unit >= t->nb_ptses here since CBR files may lack IndexEntryArrays */
2874 if (mxf->current_edit_unit < 0)
2880 static int mxf_compute_sample_count(MXFContext *mxf, int stream_index,
2881 uint64_t *sample_count)
2883 int i, total = 0, size = 0;
2884 AVStream *st = mxf->fc->streams[stream_index];
2885 MXFTrack *track = st->priv_data;
2886 AVRational time_base = av_inv_q(track->edit_rate);
2887 AVRational sample_rate = av_inv_q(st->time_base);
2888 const MXFSamplesPerFrame *spf = NULL;
2890 if ((sample_rate.num / sample_rate.den) == 48000)
2891 spf = ff_mxf_get_samples_per_frame(mxf->fc, time_base);
2893 int remainder = (sample_rate.num * time_base.num) %
2894 (time_base.den * sample_rate.den);
2895 *sample_count = av_q2d(av_mul_q((AVRational){mxf->current_edit_unit, 1},
2896 av_mul_q(sample_rate, time_base)));
2898 av_log(mxf->fc, AV_LOG_WARNING,
2899 "seeking detected on stream #%d with time base (%d/%d) and "
2900 "sample rate (%d/%d), audio pts won't be accurate.\n",
2901 stream_index, time_base.num, time_base.den,
2902 sample_rate.num, sample_rate.den);
2906 while (spf->samples_per_frame[size]) {
2907 total += spf->samples_per_frame[size];
2913 *sample_count = (mxf->current_edit_unit / size) * (uint64_t)total;
2914 for (i = 0; i < mxf->current_edit_unit % size; i++) {
2915 *sample_count += spf->samples_per_frame[i];
2921 static int mxf_set_audio_pts(MXFContext *mxf, AVCodecParameters *par,
2924 MXFTrack *track = mxf->fc->streams[pkt->stream_index]->priv_data;
2925 int64_t bits_per_sample = par->bits_per_coded_sample;
2927 if (!bits_per_sample)
2928 bits_per_sample = av_get_bits_per_sample(par->codec_id);
2930 pkt->pts = track->sample_count;
2932 if ( par->channels <= 0
2933 || bits_per_sample <= 0
2934 || par->channels * (int64_t)bits_per_sample < 8)
2935 return AVERROR(EINVAL);
2936 track->sample_count += pkt->size / (par->channels * (int64_t)bits_per_sample / 8);
2940 static int mxf_read_packet_old(AVFormatContext *s, AVPacket *pkt)
2943 MXFContext *mxf = s->priv_data;
2946 while ((ret = klv_read_packet(&klv, s->pb)) == 0) {
2947 PRINT_KEY(s, "read packet", klv.key);
2948 av_log(s, AV_LOG_TRACE, "size %"PRIu64" offset %#"PRIx64"\n", klv.length, klv.offset);
2949 if (IS_KLV_KEY(klv.key, mxf_encrypted_triplet_key)) {
2950 ret = mxf_decrypt_triplet(s, pkt, &klv);
2952 av_log(s, AV_LOG_ERROR, "invalid encoded triplet\n");
2957 if (IS_KLV_KEY(klv.key, mxf_essence_element_key) ||
2958 IS_KLV_KEY(klv.key, mxf_canopus_essence_element_key) ||
2959 IS_KLV_KEY(klv.key, mxf_avid_essence_element_key)) {
2960 int index = mxf_get_stream_index(s, &klv);
2961 int64_t next_ofs, next_klv;
2964 AVCodecParameters *par;
2967 av_log(s, AV_LOG_ERROR,
2968 "error getting stream index %"PRIu32"\n",
2969 AV_RB32(klv.key + 12));
2973 st = s->streams[index];
2974 track = st->priv_data;
2976 if (s->streams[index]->discard == AVDISCARD_ALL)
2979 next_klv = avio_tell(s->pb) + klv.length;
2980 next_ofs = mxf_set_current_edit_unit(mxf, klv.offset);
2982 if (next_ofs >= 0 && next_klv > next_ofs) {
2983 /* if this check is hit then it's possible OPAtom was treated as OP1a
2984 * truncate the packet since it's probably very large (>2 GiB is common) */
2985 avpriv_request_sample(s,
2986 "OPAtom misinterpreted as OP1a? "
2987 "KLV for edit unit %i extending into "
2989 mxf->current_edit_unit);
2990 klv.length = next_ofs - avio_tell(s->pb);
2993 /* check for 8 channels AES3 element */
2994 if (klv.key[12] == 0x06 && klv.key[13] == 0x01 && klv.key[14] == 0x10) {
2995 ret = mxf_get_d10_aes3_packet(s->pb, s->streams[index],
2998 av_log(s, AV_LOG_ERROR, "error reading D-10 aes3 frame\n");
3002 ret = av_get_packet(s->pb, pkt, klv.length);
3006 pkt->stream_index = index;
3007 pkt->pos = klv.offset;
3011 if (par->codec_type == AVMEDIA_TYPE_VIDEO && next_ofs >= 0) {
3012 /* mxf->current_edit_unit good - see if we have an
3013 * index table to derive timestamps from */
3014 MXFIndexTable *t = &mxf->index_tables[0];
3016 if (mxf->nb_index_tables >= 1 && mxf->current_edit_unit < t->nb_ptses) {
3017 pkt->dts = mxf->current_edit_unit + t->first_dts;
3018 pkt->pts = t->ptses[mxf->current_edit_unit];
3019 } else if (track->intra_only) {
3020 /* intra-only -> PTS = EditUnit.
3021 * let utils.c figure out DTS since it can be < PTS if low_delay = 0 (Sony IMX30) */
3022 pkt->pts = mxf->current_edit_unit;
3024 } else if (par->codec_type == AVMEDIA_TYPE_AUDIO) {
3025 ret = mxf_set_audio_pts(mxf, par, pkt);
3030 /* seek for truncated packets */
3031 avio_seek(s->pb, next_klv, SEEK_SET);
3036 avio_skip(s->pb, klv.length);
3038 return avio_feof(s->pb) ? AVERROR_EOF : ret;
3041 static int mxf_read_packet(AVFormatContext *s, AVPacket *pkt)
3043 MXFContext *mxf = s->priv_data;
3045 int64_t ret64, pos, next_pos;
3050 if (mxf->op != OPAtom)
3051 return mxf_read_packet_old(s, pkt);
3053 // If we have no streams then we basically are at EOF
3054 if (s->nb_streams < 1)
3057 /* OPAtom - clip wrapped demuxing */
3058 /* NOTE: mxf_read_header() makes sure nb_index_tables > 0 for OPAtom */
3060 t = &mxf->index_tables[0];
3062 if (mxf->current_edit_unit >= st->duration)
3065 edit_units = FFMIN(mxf->edit_units_per_packet, st->duration - mxf->current_edit_unit);
3067 if ((ret = mxf_edit_unit_absolute_offset(mxf, t, mxf->current_edit_unit, NULL, &pos, 1)) < 0)
3070 /* compute size by finding the next edit unit or the end of the essence container
3071 * not pretty, but it works */
3072 if ((ret = mxf_edit_unit_absolute_offset(mxf, t, mxf->current_edit_unit + edit_units, NULL, &next_pos, 0)) < 0 &&
3073 (next_pos = mxf_essence_container_end(mxf, t->body_sid)) <= 0) {
3074 av_log(s, AV_LOG_ERROR, "unable to compute the size of the last packet\n");
3075 return AVERROR_INVALIDDATA;
3078 if ((size = next_pos - pos) <= 0) {
3079 av_log(s, AV_LOG_ERROR, "bad size: %i\n", size);
3080 return AVERROR_INVALIDDATA;
3083 if ((ret64 = avio_seek(s->pb, pos, SEEK_SET)) < 0)
3086 if ((size = av_get_packet(s->pb, pkt, size)) < 0)
3089 pkt->stream_index = 0;
3091 if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && t->ptses &&
3092 mxf->current_edit_unit >= 0 && mxf->current_edit_unit < t->nb_ptses) {
3093 pkt->dts = mxf->current_edit_unit + t->first_dts;
3094 pkt->pts = t->ptses[mxf->current_edit_unit];
3095 } else if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
3096 int ret = mxf_set_audio_pts(mxf, st->codecpar, pkt);
3101 mxf->current_edit_unit += edit_units;
3106 static int mxf_read_close(AVFormatContext *s)
3108 MXFContext *mxf = s->priv_data;
3111 av_freep(&mxf->packages_refs);
3113 for (i = 0; i < s->nb_streams; i++)
3114 s->streams[i]->priv_data = NULL;
3116 for (i = 0; i < mxf->metadata_sets_count; i++) {
3117 mxf_free_metadataset(mxf->metadata_sets + i, 1);
3119 av_freep(&mxf->partitions);
3120 av_freep(&mxf->metadata_sets);
3121 av_freep(&mxf->aesc);
3122 av_freep(&mxf->local_tags);
3124 if (mxf->index_tables) {
3125 for (i = 0; i < mxf->nb_index_tables; i++) {
3126 av_freep(&mxf->index_tables[i].segments);
3127 av_freep(&mxf->index_tables[i].ptses);
3128 av_freep(&mxf->index_tables[i].fake_index);
3129 av_freep(&mxf->index_tables[i].offsets);
3132 av_freep(&mxf->index_tables);
3137 static int mxf_probe(AVProbeData *p) {
3138 const uint8_t *bufp = p->buf;
3139 const uint8_t *end = p->buf + p->buf_size;
3141 if (p->buf_size < sizeof(mxf_header_partition_pack_key))
3144 /* Must skip Run-In Sequence and search for MXF header partition pack key SMPTE 377M 5.5 */
3145 end -= sizeof(mxf_header_partition_pack_key);
3147 for (; bufp < end;) {
3148 if (!((bufp[13] - 1) & 0xF2)){
3149 if (AV_RN32(bufp ) == AV_RN32(mxf_header_partition_pack_key ) &&
3150 AV_RN32(bufp+ 4) == AV_RN32(mxf_header_partition_pack_key+ 4) &&
3151 AV_RN32(bufp+ 8) == AV_RN32(mxf_header_partition_pack_key+ 8) &&
3152 AV_RN16(bufp+12) == AV_RN16(mxf_header_partition_pack_key+12))
3153 return AVPROBE_SCORE_MAX;
3162 /* rudimentary byte seek */
3163 /* XXX: use MXF Index */
3164 static int mxf_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
3166 AVStream *st = s->streams[stream_index];
3168 MXFContext* mxf = s->priv_data;
3172 MXFTrack *source_track = st->priv_data;
3174 /* if audio then truncate sample_time to EditRate */
3175 if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO)
3176 sample_time = av_rescale_q(sample_time, st->time_base,
3177 av_inv_q(source_track->edit_rate));
3179 if (mxf->nb_index_tables <= 0) {
3181 return AVERROR_INVALIDDATA;
3182 if (sample_time < 0)
3184 seconds = av_rescale(sample_time, st->time_base.num, st->time_base.den);
3186 seekpos = avio_seek(s->pb, (s->bit_rate * seconds) >> 3, SEEK_SET);
3190 ff_update_cur_dts(s, st, sample_time);
3191 mxf->current_edit_unit = sample_time;
3193 t = &mxf->index_tables[0];
3195 /* clamp above zero, else ff_index_search_timestamp() returns negative
3196 * this also means we allow seeking before the start */
3197 sample_time = FFMAX(sample_time, 0);
3199 if (t->fake_index) {
3200 /* The first frames may not be keyframes in presentation order, so
3201 * we have to advance the target to be able to find the first
3202 * keyframe backwards... */
3203 if (!(flags & AVSEEK_FLAG_ANY) &&
3204 (flags & AVSEEK_FLAG_BACKWARD) &&
3205 t->ptses[0] != AV_NOPTS_VALUE &&
3206 sample_time < t->ptses[0] &&
3207 (t->fake_index[t->ptses[0]].flags & AVINDEX_KEYFRAME))
3208 sample_time = t->ptses[0];
3210 /* behave as if we have a proper index */
3211 if ((sample_time = ff_index_search_timestamp(t->fake_index, t->nb_ptses, sample_time, flags)) < 0)
3213 /* get the stored order index from the display order index */
3214 sample_time += t->offsets[sample_time];
3216 /* no IndexEntryArray (one or more CBR segments)
3217 * make sure we don't seek past the end */
3218 sample_time = FFMIN(sample_time, source_track->original_duration - 1);
3221 if ((ret = mxf_edit_unit_absolute_offset(mxf, t, sample_time, &sample_time, &seekpos, 1)) < 0)
3224 ff_update_cur_dts(s, st, sample_time);
3225 mxf->current_edit_unit = sample_time;
3226 avio_seek(s->pb, seekpos, SEEK_SET);
3229 // Update all tracks sample count
3230 for (i = 0; i < s->nb_streams; i++) {
3231 AVStream *cur_st = s->streams[i];
3232 MXFTrack *cur_track = cur_st->priv_data;
3233 uint64_t current_sample_count = 0;
3234 if (cur_st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
3235 ret = mxf_compute_sample_count(mxf, i, ¤t_sample_count);
3239 cur_track->sample_count = current_sample_count;
3245 AVInputFormat ff_mxf_demuxer = {
3247 .long_name = NULL_IF_CONFIG_SMALL("MXF (Material eXchange Format)"),
3248 .flags = AVFMT_SEEK_TO_PTS,
3249 .priv_data_size = sizeof(MXFContext),
3250 .read_probe = mxf_probe,
3251 .read_header = mxf_read_header,
3252 .read_packet = mxf_read_packet,
3253 .read_close = mxf_read_close,
3254 .read_seek = mxf_read_seek,