FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
mxfdec.c
Go to the documentation of this file.
1 /*
2  * MXF demuxer.
3  * Copyright (c) 2006 SmartJog S.A., Baptiste Coudurier <baptiste dot coudurier at smartjog dot com>
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 /*
23  * References
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
31  *
32  * Principle
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.
38  *
39  * Metadata reading functions read Local Tags, get InstanceUID(0x3C0A) then add MetaDataSet to MXFContext.
40  * Metadata parsing resolves Strong References to objects.
41  *
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
44  */
45 
46 #include <inttypes.h>
47 
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/parseutils.h"
54 #include "libavutil/timecode.h"
55 #include "avformat.h"
56 #include "internal.h"
57 #include "mxf.h"
58 
59 typedef enum {
64 
65 typedef enum {
66  OP1a = 1,
76  OPSONYOpt, /* FATE sample, violates the spec in places */
77 } MXFOP;
78 
79 typedef struct MXFPartition {
80  int closed;
81  int complete;
84  int index_sid;
85  int body_sid;
86  int64_t this_partition;
87  int64_t essence_offset; ///< absolute offset of essence
88  int64_t essence_length;
93  int64_t pack_ofs; ///< absolute offset of pack in file, including run-in
94  int64_t body_offset;
95 } MXFPartition;
96 
97 typedef struct MXFCryptoContext {
102 
103 typedef struct MXFStructuralComponent {
109  int64_t duration;
110  int64_t start_position;
113 
114 typedef struct MXFSequence {
120  int64_t duration;
122 } MXFSequence;
123 
124 typedef struct MXFTrack {
129  struct AVRational rate;
132 
133 typedef struct {
138 
139 typedef struct {
144  int64_t duration;
146 
147 typedef struct {
150  char *name;
151  char *value;
153 
154 typedef struct {
157  MXFSequence *sequence; /* mandatory, and only one */
159  int track_id;
160  char *name;
161  uint8_t track_number[4];
164  uint64_t sample_count;
165  int64_t original_duration; /* st->duration in SampleRate/EditRate units */
167  int body_sid;
168 } MXFTrack;
169 
170 typedef struct MXFDescriptor {
178  int width;
179  int height; /* Field height, not frame height */
180  int frame_layout; /* See MXFFrameLayout enum */
182 #define MXF_FIELD_DOMINANCE_DEFAULT 0
183 #define MXF_FIELD_DOMINANCE_FF 1 /* coded first, displayed first */
184 #define MXF_FIELD_DOMINANCE_FL 2 /* coded first, displayed last */
186  int channels;
188  int64_t duration; /* ContainerDuration optional property */
189  unsigned int component_depth;
190  unsigned int horiz_subsampling;
191  unsigned int vert_subsampling;
198 } MXFDescriptor;
199 
200 typedef struct MXFIndexTableSegment {
205  int body_sid;
208  uint64_t index_duration;
214 
215 typedef struct MXFPackage {
222  MXFDescriptor *descriptor; /* only one */
224  char *name;
227 } MXFPackage;
228 
229 typedef struct MXFEssenceContainerData {
235  int body_sid;
237 
238 typedef struct MXFMetadataSet {
242 
243 /* decoded index table */
244 typedef struct MXFIndexTable {
246  int body_sid;
247  int nb_ptses; /* number of PTSes or total duration of index */
248  int64_t first_dts; /* DTS = EditUnit + first_dts */
249  int64_t *ptses; /* maps EditUnit -> PTS */
251  MXFIndexTableSegment **segments; /* sorted by IndexStartPosition */
252  AVIndexEntry *fake_index; /* used for calling ff_index_search_timestamp() */
253  int8_t *offsets; /* temporal offsets for display order to stored order conversion */
254 } MXFIndexTable;
255 
256 typedef struct MXFContext {
267  struct AVAES *aesc;
273  int run_in;
281  int edit_units_per_packet; ///< how many edit units to read at a time (PCM, OPAtom)
282 } MXFContext;
283 
287 };
288 
289 /* NOTE: klv_offset is not set (-1) for local keys */
290 typedef int MXFMetadataReadFunc(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset);
291 
293  const UID key;
295  int ctx_size;
298 
299 static int mxf_read_close(AVFormatContext *s);
300 
301 /* partial keys to match */
302 static const uint8_t mxf_header_partition_pack_key[] = { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02 };
303 static const uint8_t mxf_essence_element_key[] = { 0x06,0x0e,0x2b,0x34,0x01,0x02,0x01,0x01,0x0d,0x01,0x03,0x01 };
304 static const uint8_t mxf_avid_essence_element_key[] = { 0x06,0x0e,0x2b,0x34,0x01,0x02,0x01,0x01,0x0e,0x04,0x03,0x01 };
305 static const uint8_t mxf_canopus_essence_element_key[] = { 0x06,0x0e,0x2b,0x34,0x01,0x02,0x01,0x0a,0x0e,0x0f,0x03,0x01 };
306 static const uint8_t mxf_system_item_key_cp[] = { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x03,0x01,0x04 };
307 static const uint8_t mxf_system_item_key_gc[] = { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x03,0x01,0x14 };
308 static const uint8_t mxf_klv_key[] = { 0x06,0x0e,0x2b,0x34 };
309 /* complete keys to match */
310 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 };
311 static const uint8_t mxf_encrypted_triplet_key[] = { 0x06,0x0e,0x2b,0x34,0x02,0x04,0x01,0x07,0x0d,0x01,0x03,0x01,0x02,0x7e,0x01,0x00 };
312 static const uint8_t mxf_encrypted_essence_container[] = { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x07,0x0d,0x01,0x03,0x01,0x02,0x0b,0x01,0x00 };
313 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 };
314 static const uint8_t mxf_sony_mpeg4_extradata[] = { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x01,0x0e,0x06,0x06,0x02,0x02,0x01,0x00,0x00 };
315 static const uint8_t mxf_avid_project_name[] = { 0xa5,0xfb,0x7b,0x25,0xf6,0x15,0x94,0xb9,0x62,0xfc,0x37,0x17,0x49,0x2d,0x42,0xbf };
316 static const uint8_t mxf_jp2k_rsiz[] = { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02,0x01,0x00 };
317 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 };
318 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 };
319 
320 #define IS_KLV_KEY(x, y) (!memcmp(x, y, sizeof(y)))
321 
322 static void mxf_free_metadataset(MXFMetadataSet **ctx, int freectx)
323 {
325  switch ((*ctx)->type) {
326  case Descriptor:
327  av_freep(&((MXFDescriptor *)*ctx)->extradata);
328  break;
329  case MultipleDescriptor:
330  av_freep(&((MXFDescriptor *)*ctx)->sub_descriptors_refs);
331  break;
332  case Sequence:
333  av_freep(&((MXFSequence *)*ctx)->structural_components_refs);
334  break;
335  case EssenceGroup:
336  av_freep(&((MXFEssenceGroup *)*ctx)->structural_components_refs);
337  break;
338  case SourcePackage:
339  case MaterialPackage:
340  av_freep(&((MXFPackage *)*ctx)->tracks_refs);
341  av_freep(&((MXFPackage *)*ctx)->name);
342  av_freep(&((MXFPackage *)*ctx)->comment_refs);
343  break;
344  case TaggedValue:
345  av_freep(&((MXFTaggedValue *)*ctx)->name);
346  av_freep(&((MXFTaggedValue *)*ctx)->value);
347  break;
348  case Track:
349  av_freep(&((MXFTrack *)*ctx)->name);
350  break;
351  case IndexTableSegment:
352  seg = (MXFIndexTableSegment *)*ctx;
354  av_freep(&seg->flag_entries);
356  default:
357  break;
358  }
359  if (freectx)
360  av_freep(ctx);
361 }
362 
364 {
365  uint64_t size = avio_r8(pb);
366  if (size & 0x80) { /* long form */
367  int bytes_num = size & 0x7f;
368  /* SMPTE 379M 5.3.4 guarantee that bytes_num must not exceed 8 bytes */
369  if (bytes_num > 8)
370  return AVERROR_INVALIDDATA;
371  size = 0;
372  while (bytes_num--)
373  size = size << 8 | avio_r8(pb);
374  }
375  if (size > INT64_MAX)
376  return AVERROR_INVALIDDATA;
377  return size;
378 }
379 
380 static int mxf_read_sync(AVIOContext *pb, const uint8_t *key, unsigned size)
381 {
382  int i, b;
383  for (i = 0; i < size && !avio_feof(pb); i++) {
384  b = avio_r8(pb);
385  if (b == key[0])
386  i = 0;
387  else if (b != key[i])
388  i = -1;
389  }
390  return i == size;
391 }
392 
393 static int klv_read_packet(KLVPacket *klv, AVIOContext *pb)
394 {
395  int64_t length;
396  if (!mxf_read_sync(pb, mxf_klv_key, 4))
397  return AVERROR_INVALIDDATA;
398  klv->offset = avio_tell(pb) - 4;
399  memcpy(klv->key, mxf_klv_key, 4);
400  avio_read(pb, klv->key + 4, 12);
401  length = klv_decode_ber_length(pb);
402  if (length < 0)
403  return length;
404  klv->length = length;
405  return 0;
406 }
407 
408 static int mxf_get_stream_index(AVFormatContext *s, KLVPacket *klv, int body_sid)
409 {
410  int i;
411 
412  for (i = 0; i < s->nb_streams; i++) {
413  MXFTrack *track = s->streams[i]->priv_data;
414  /* SMPTE 379M 7.3 */
415  if (track && (!body_sid || !track->body_sid || track->body_sid == body_sid) && !memcmp(klv->key + sizeof(mxf_essence_element_key), track->track_number, sizeof(track->track_number)))
416  return i;
417  }
418  /* return 0 if only one stream, for OP Atom files with 0 as track number */
419  return s->nb_streams == 1 ? 0 : -1;
420 }
421 
422 static int find_body_sid_by_offset(MXFContext *mxf, int64_t offset)
423 {
424  // we look for partition where the offset is placed
425  int a, b, m;
426  int64_t this_partition;
427 
428  a = -1;
429  b = mxf->partitions_count;
430 
431  while (b - a > 1) {
432  m = (a + b) >> 1;
433  this_partition = mxf->partitions[m].this_partition;
434  if (this_partition <= offset)
435  a = m;
436  else
437  b = m;
438  }
439 
440  if (a == -1)
441  return 0;
442  return mxf->partitions[a].body_sid;
443 }
444 
445 /* XXX: use AVBitStreamFilter */
447 {
448  const uint8_t *buf_ptr, *end_ptr;
449  uint8_t *data_ptr;
450  int i;
451 
452  if (length > 61444) /* worst case PAL 1920 samples 8 channels */
453  return AVERROR_INVALIDDATA;
454  length = av_get_packet(pb, pkt, length);
455  if (length < 0)
456  return length;
457  data_ptr = pkt->data;
458  end_ptr = pkt->data + length;
459  buf_ptr = pkt->data + 4; /* skip SMPTE 331M header */
460  for (; end_ptr - buf_ptr >= st->codecpar->channels * 4; ) {
461  for (i = 0; i < st->codecpar->channels; i++) {
462  uint32_t sample = bytestream_get_le32(&buf_ptr);
463  if (st->codecpar->bits_per_coded_sample == 24)
464  bytestream_put_le24(&data_ptr, (sample >> 4) & 0xffffff);
465  else
466  bytestream_put_le16(&data_ptr, (sample >> 12) & 0xffff);
467  }
468  buf_ptr += 32 - st->codecpar->channels*4; // always 8 channels stored SMPTE 331M
469  }
470  av_shrink_packet(pkt, data_ptr - pkt->data);
471  return 0;
472 }
473 
475 {
476  static const uint8_t checkv[16] = {0x43, 0x48, 0x55, 0x4b, 0x43, 0x48, 0x55, 0x4b, 0x43, 0x48, 0x55, 0x4b, 0x43, 0x48, 0x55, 0x4b};
477  MXFContext *mxf = s->priv_data;
478  AVIOContext *pb = s->pb;
479  int64_t end = avio_tell(pb) + klv->length;
480  int64_t size;
481  uint64_t orig_size;
482  uint64_t plaintext_size;
483  uint8_t ivec[16];
484  uint8_t tmpbuf[16];
485  int index;
486  int body_sid;
487 
488  if (!mxf->aesc && s->key && s->keylen == 16) {
489  mxf->aesc = av_aes_alloc();
490  if (!mxf->aesc)
491  return AVERROR(ENOMEM);
492  av_aes_init(mxf->aesc, s->key, 128, 1);
493  }
494  // crypto context
495  size = klv_decode_ber_length(pb);
496  if (size < 0)
497  return size;
498  avio_skip(pb, size);
499  // plaintext offset
501  plaintext_size = avio_rb64(pb);
502  // source klv key
504  avio_read(pb, klv->key, 16);
506  return AVERROR_INVALIDDATA;
507 
508  body_sid = find_body_sid_by_offset(mxf, klv->offset);
509  index = mxf_get_stream_index(s, klv, body_sid);
510  if (index < 0)
511  return AVERROR_INVALIDDATA;
512  // source size
514  orig_size = avio_rb64(pb);
515  if (orig_size < plaintext_size)
516  return AVERROR_INVALIDDATA;
517  // enc. code
518  size = klv_decode_ber_length(pb);
519  if (size < 32 || size - 32 < orig_size)
520  return AVERROR_INVALIDDATA;
521  avio_read(pb, ivec, 16);
522  avio_read(pb, tmpbuf, 16);
523  if (mxf->aesc)
524  av_aes_crypt(mxf->aesc, tmpbuf, tmpbuf, 1, ivec, 1);
525  if (memcmp(tmpbuf, checkv, 16))
526  av_log(s, AV_LOG_ERROR, "probably incorrect decryption key\n");
527  size -= 32;
528  size = av_get_packet(pb, pkt, size);
529  if (size < 0)
530  return size;
531  else if (size < plaintext_size)
532  return AVERROR_INVALIDDATA;
533  size -= plaintext_size;
534  if (mxf->aesc)
535  av_aes_crypt(mxf->aesc, &pkt->data[plaintext_size],
536  &pkt->data[plaintext_size], size >> 4, ivec, 1);
537  av_shrink_packet(pkt, orig_size);
538  pkt->stream_index = index;
539  avio_skip(pb, end - avio_tell(pb));
540  return 0;
541 }
542 
543 static int mxf_read_primer_pack(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
544 {
545  MXFContext *mxf = arg;
546  int item_num = avio_rb32(pb);
547  int item_len = avio_rb32(pb);
548 
549  if (item_len != 18) {
550  avpriv_request_sample(pb, "Primer pack item length %d", item_len);
551  return AVERROR_PATCHWELCOME;
552  }
553  if (item_num > 65536 || item_num < 0) {
554  av_log(mxf->fc, AV_LOG_ERROR, "item_num %d is too large\n", item_num);
555  return AVERROR_INVALIDDATA;
556  }
557  if (mxf->local_tags)
558  av_log(mxf->fc, AV_LOG_VERBOSE, "Multiple primer packs\n");
559  av_free(mxf->local_tags);
560  mxf->local_tags_count = 0;
561  mxf->local_tags = av_calloc(item_num, item_len);
562  if (!mxf->local_tags)
563  return AVERROR(ENOMEM);
564  mxf->local_tags_count = item_num;
565  avio_read(pb, mxf->local_tags, item_num*item_len);
566  return 0;
567 }
568 
569 static int mxf_read_partition_pack(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
570 {
571  MXFContext *mxf = arg;
572  MXFPartition *partition, *tmp_part;
573  UID op;
574  uint64_t footer_partition;
575  uint32_t nb_essence_containers;
576 
577  if (mxf->partitions_count >= INT_MAX / 2)
578  return AVERROR_INVALIDDATA;
579 
580  tmp_part = av_realloc_array(mxf->partitions, mxf->partitions_count + 1, sizeof(*mxf->partitions));
581  if (!tmp_part)
582  return AVERROR(ENOMEM);
583  mxf->partitions = tmp_part;
584 
585  if (mxf->parsing_backward) {
586  /* insert the new partition pack in the middle
587  * this makes the entries in mxf->partitions sorted by offset */
588  memmove(&mxf->partitions[mxf->last_forward_partition+1],
590  (mxf->partitions_count - mxf->last_forward_partition)*sizeof(*mxf->partitions));
591  partition = mxf->current_partition = &mxf->partitions[mxf->last_forward_partition];
592  } else {
593  mxf->last_forward_partition++;
594  partition = mxf->current_partition = &mxf->partitions[mxf->partitions_count];
595  }
596 
597  memset(partition, 0, sizeof(*partition));
598  mxf->partitions_count++;
599  partition->pack_length = avio_tell(pb) - klv_offset + size;
600  partition->pack_ofs = klv_offset;
601 
602  switch(uid[13]) {
603  case 2:
604  partition->type = Header;
605  break;
606  case 3:
607  partition->type = BodyPartition;
608  break;
609  case 4:
610  partition->type = Footer;
611  break;
612  default:
613  av_log(mxf->fc, AV_LOG_ERROR, "unknown partition type %i\n", uid[13]);
614  return AVERROR_INVALIDDATA;
615  }
616 
617  /* consider both footers to be closed (there is only Footer and CompleteFooter) */
618  partition->closed = partition->type == Footer || !(uid[14] & 1);
619  partition->complete = uid[14] > 2;
620  avio_skip(pb, 4);
621  partition->kag_size = avio_rb32(pb);
622  partition->this_partition = avio_rb64(pb);
623  partition->previous_partition = avio_rb64(pb);
624  footer_partition = avio_rb64(pb);
625  partition->header_byte_count = avio_rb64(pb);
626  partition->index_byte_count = avio_rb64(pb);
627  partition->index_sid = avio_rb32(pb);
628  partition->body_offset = avio_rb64(pb);
629  partition->body_sid = avio_rb32(pb);
630  if (avio_read(pb, op, sizeof(UID)) != sizeof(UID)) {
631  av_log(mxf->fc, AV_LOG_ERROR, "Failed reading UID\n");
632  return AVERROR_INVALIDDATA;
633  }
634  nb_essence_containers = avio_rb32(pb);
635 
636  if (partition->this_partition &&
637  partition->previous_partition == partition->this_partition) {
638  av_log(mxf->fc, AV_LOG_ERROR,
639  "PreviousPartition equal to ThisPartition %"PRIx64"\n",
640  partition->previous_partition);
641  /* override with the actual previous partition offset */
642  if (!mxf->parsing_backward && mxf->last_forward_partition > 1) {
643  MXFPartition *prev =
644  mxf->partitions + mxf->last_forward_partition - 2;
645  partition->previous_partition = prev->this_partition;
646  }
647  /* if no previous body partition are found point to the header
648  * partition */
649  if (partition->previous_partition == partition->this_partition)
650  partition->previous_partition = 0;
651  av_log(mxf->fc, AV_LOG_ERROR,
652  "Overriding PreviousPartition with %"PRIx64"\n",
653  partition->previous_partition);
654  }
655 
656  /* some files don't have FooterPartition set in every partition */
657  if (footer_partition) {
658  if (mxf->footer_partition && mxf->footer_partition != footer_partition) {
659  av_log(mxf->fc, AV_LOG_ERROR,
660  "inconsistent FooterPartition value: %"PRIu64" != %"PRIu64"\n",
661  mxf->footer_partition, footer_partition);
662  } else {
663  mxf->footer_partition = footer_partition;
664  }
665  }
666 
667  av_log(mxf->fc, AV_LOG_TRACE,
668  "PartitionPack: ThisPartition = 0x%"PRIX64
669  ", PreviousPartition = 0x%"PRIX64", "
670  "FooterPartition = 0x%"PRIX64", IndexSID = %i, BodySID = %i\n",
671  partition->this_partition,
672  partition->previous_partition, footer_partition,
673  partition->index_sid, partition->body_sid);
674 
675  /* sanity check PreviousPartition if set */
676  //NOTE: this isn't actually enough, see mxf_seek_to_previous_partition()
677  if (partition->previous_partition &&
678  mxf->run_in + partition->previous_partition >= klv_offset) {
679  av_log(mxf->fc, AV_LOG_ERROR,
680  "PreviousPartition points to this partition or forward\n");
681  return AVERROR_INVALIDDATA;
682  }
683 
684  if (op[12] == 1 && op[13] == 1) mxf->op = OP1a;
685  else if (op[12] == 1 && op[13] == 2) mxf->op = OP1b;
686  else if (op[12] == 1 && op[13] == 3) mxf->op = OP1c;
687  else if (op[12] == 2 && op[13] == 1) mxf->op = OP2a;
688  else if (op[12] == 2 && op[13] == 2) mxf->op = OP2b;
689  else if (op[12] == 2 && op[13] == 3) mxf->op = OP2c;
690  else if (op[12] == 3 && op[13] == 1) mxf->op = OP3a;
691  else if (op[12] == 3 && op[13] == 2) mxf->op = OP3b;
692  else if (op[12] == 3 && op[13] == 3) mxf->op = OP3c;
693  else if (op[12] == 64&& op[13] == 1) mxf->op = OPSONYOpt;
694  else if (op[12] == 0x10) {
695  /* SMPTE 390m: "There shall be exactly one essence container"
696  * The following block deals with files that violate this, namely:
697  * 2011_DCPTEST_24FPS.V.mxf - two ECs, OP1a
698  * abcdefghiv016f56415e.mxf - zero ECs, OPAtom, output by Avid AirSpeed */
699  if (nb_essence_containers != 1) {
700  MXFOP op = nb_essence_containers ? OP1a : OPAtom;
701 
702  /* only nag once */
703  if (!mxf->op)
704  av_log(mxf->fc, AV_LOG_WARNING,
705  "\"OPAtom\" with %"PRIu32" ECs - assuming %s\n",
706  nb_essence_containers,
707  op == OP1a ? "OP1a" : "OPAtom");
708 
709  mxf->op = op;
710  } else
711  mxf->op = OPAtom;
712  } else {
713  av_log(mxf->fc, AV_LOG_ERROR, "unknown operational pattern: %02xh %02xh - guessing OP1a\n", op[12], op[13]);
714  mxf->op = OP1a;
715  }
716 
717  if (partition->kag_size <= 0 || partition->kag_size > (1 << 20)) {
718  av_log(mxf->fc, AV_LOG_WARNING, "invalid KAGSize %"PRId32" - guessing ",
719  partition->kag_size);
720 
721  if (mxf->op == OPSONYOpt)
722  partition->kag_size = 512;
723  else
724  partition->kag_size = 1;
725 
726  av_log(mxf->fc, AV_LOG_WARNING, "%"PRId32"\n", partition->kag_size);
727  }
728 
729  return 0;
730 }
731 
732 static int mxf_add_metadata_set(MXFContext *mxf, void *metadata_set)
733 {
735 
736  tmp = av_realloc_array(mxf->metadata_sets, mxf->metadata_sets_count + 1, sizeof(*mxf->metadata_sets));
737  if (!tmp)
738  return AVERROR(ENOMEM);
739  mxf->metadata_sets = tmp;
740  mxf->metadata_sets[mxf->metadata_sets_count] = metadata_set;
741  mxf->metadata_sets_count++;
742  return 0;
743 }
744 
745 static int mxf_read_cryptographic_context(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
746 {
747  MXFCryptoContext *cryptocontext = arg;
748  if (size != 16)
749  return AVERROR_INVALIDDATA;
751  avio_read(pb, cryptocontext->source_container_ul, 16);
752  return 0;
753 }
754 
755 static int mxf_read_strong_ref_array(AVIOContext *pb, UID **refs, int *count)
756 {
757  *count = avio_rb32(pb);
758  *refs = av_calloc(*count, sizeof(UID));
759  if (!*refs) {
760  *count = 0;
761  return AVERROR(ENOMEM);
762  }
763  avio_skip(pb, 4); /* useless size of objects, always 16 according to specs */
764  avio_read(pb, (uint8_t *)*refs, *count * sizeof(UID));
765  return 0;
766 }
767 
768 static inline int mxf_read_utf16_string(AVIOContext *pb, int size, char** str, int be)
769 {
770  int ret;
771  size_t buf_size;
772 
773  if (size < 0 || size > INT_MAX/2)
774  return AVERROR(EINVAL);
775 
776  buf_size = size + size / 2 + 1;
777  *str = av_malloc(buf_size);
778  if (!*str)
779  return AVERROR(ENOMEM);
780 
781  if (be)
782  ret = avio_get_str16be(pb, size, *str, buf_size);
783  else
784  ret = avio_get_str16le(pb, size, *str, buf_size);
785 
786  if (ret < 0) {
787  av_freep(str);
788  return ret;
789  }
790 
791  return ret;
792 }
793 
794 #define READ_STR16(type, big_endian) \
795 static int mxf_read_utf16 ## type ##_string(AVIOContext *pb, int size, char** str) \
796 { \
797 return mxf_read_utf16_string(pb, size, str, big_endian); \
798 }
799 READ_STR16(be, 1)
800 READ_STR16(le, 0)
801 #undef READ_STR16
802 
803 static int mxf_read_content_storage(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
804 {
805  MXFContext *mxf = arg;
806  switch (tag) {
807  case 0x1901:
808  if (mxf->packages_refs)
809  av_log(mxf->fc, AV_LOG_VERBOSE, "Multiple packages_refs\n");
810  av_free(mxf->packages_refs);
811  return mxf_read_strong_ref_array(pb, &mxf->packages_refs, &mxf->packages_count);
812  case 0x1902:
815  }
816  return 0;
817 }
818 
819 static int mxf_read_source_clip(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
820 {
821  MXFStructuralComponent *source_clip = arg;
822  switch(tag) {
823  case 0x0202:
824  source_clip->duration = avio_rb64(pb);
825  break;
826  case 0x1201:
827  source_clip->start_position = avio_rb64(pb);
828  break;
829  case 0x1101:
830  /* UMID, only get last 16 bytes */
831  avio_read(pb, source_clip->source_package_ul, 16);
832  avio_read(pb, source_clip->source_package_uid, 16);
833  break;
834  case 0x1102:
835  source_clip->source_track_id = avio_rb32(pb);
836  break;
837  }
838  return 0;
839 }
840 
841 static int mxf_read_timecode_component(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
842 {
843  MXFTimecodeComponent *mxf_timecode = arg;
844  switch(tag) {
845  case 0x1501:
846  mxf_timecode->start_frame = avio_rb64(pb);
847  break;
848  case 0x1502:
849  mxf_timecode->rate = (AVRational){avio_rb16(pb), 1};
850  break;
851  case 0x1503:
852  mxf_timecode->drop_frame = avio_r8(pb);
853  break;
854  }
855  return 0;
856 }
857 
858 static int mxf_read_pulldown_component(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
859 {
860  MXFPulldownComponent *mxf_pulldown = arg;
861  switch(tag) {
862  case 0x0d01:
863  avio_read(pb, mxf_pulldown->input_segment_ref, 16);
864  break;
865  }
866  return 0;
867 }
868 
869 static int mxf_read_track(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
870 {
871  MXFTrack *track = arg;
872  switch(tag) {
873  case 0x4801:
874  track->track_id = avio_rb32(pb);
875  break;
876  case 0x4804:
877  avio_read(pb, track->track_number, 4);
878  break;
879  case 0x4802:
880  mxf_read_utf16be_string(pb, size, &track->name);
881  break;
882  case 0x4b01:
883  track->edit_rate.num = avio_rb32(pb);
884  track->edit_rate.den = avio_rb32(pb);
885  break;
886  case 0x4803:
887  avio_read(pb, track->sequence_ref, 16);
888  break;
889  }
890  return 0;
891 }
892 
893 static int mxf_read_sequence(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
894 {
895  MXFSequence *sequence = arg;
896  switch(tag) {
897  case 0x0202:
898  sequence->duration = avio_rb64(pb);
899  break;
900  case 0x0201:
901  avio_read(pb, sequence->data_definition_ul, 16);
902  break;
903  case 0x4b02:
904  sequence->origin = avio_r8(pb);
905  break;
906  case 0x1001:
908  &sequence->structural_components_count);
909  }
910  return 0;
911 }
912 
913 static int mxf_read_essence_group(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
914 {
915  MXFEssenceGroup *essence_group = arg;
916  switch (tag) {
917  case 0x0202:
918  essence_group->duration = avio_rb64(pb);
919  break;
920  case 0x0501:
921  return mxf_read_strong_ref_array(pb, &essence_group->structural_components_refs,
922  &essence_group->structural_components_count);
923  }
924  return 0;
925 }
926 
927 static int mxf_read_package(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
928 {
929  MXFPackage *package = arg;
930  switch(tag) {
931  case 0x4403:
932  return mxf_read_strong_ref_array(pb, &package->tracks_refs,
933  &package->tracks_count);
934  case 0x4401:
935  /* UMID */
936  avio_read(pb, package->package_ul, 16);
937  avio_read(pb, package->package_uid, 16);
938  break;
939  case 0x4701:
940  avio_read(pb, package->descriptor_ref, 16);
941  break;
942  case 0x4402:
943  return mxf_read_utf16be_string(pb, size, &package->name);
944  case 0x4406:
945  return mxf_read_strong_ref_array(pb, &package->comment_refs,
946  &package->comment_count);
947  }
948  return 0;
949 }
950 
951 static int mxf_read_essence_container_data(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
952 {
953  MXFEssenceContainerData *essence_data = arg;
954  switch(tag) {
955  case 0x2701:
956  /* linked package umid UMID */
957  avio_read(pb, essence_data->package_ul, 16);
958  avio_read(pb, essence_data->package_uid, 16);
959  break;
960  case 0x3f06:
961  essence_data->index_sid = avio_rb32(pb);
962  break;
963  case 0x3f07:
964  essence_data->body_sid = avio_rb32(pb);
965  break;
966  }
967  return 0;
968 }
969 
971 {
972  int i, length;
973 
974  segment->nb_index_entries = avio_rb32(pb);
975 
976  length = avio_rb32(pb);
977  if(segment->nb_index_entries && length < 11)
978  return AVERROR_INVALIDDATA;
979 
980  if (!(segment->temporal_offset_entries=av_calloc(segment->nb_index_entries, sizeof(*segment->temporal_offset_entries))) ||
981  !(segment->flag_entries = av_calloc(segment->nb_index_entries, sizeof(*segment->flag_entries))) ||
982  !(segment->stream_offset_entries = av_calloc(segment->nb_index_entries, sizeof(*segment->stream_offset_entries)))) {
984  av_freep(&segment->flag_entries);
985  return AVERROR(ENOMEM);
986  }
987 
988  for (i = 0; i < segment->nb_index_entries; i++) {
989  if(avio_feof(pb))
990  return AVERROR_INVALIDDATA;
991  segment->temporal_offset_entries[i] = avio_r8(pb);
992  avio_r8(pb); /* KeyFrameOffset */
993  segment->flag_entries[i] = avio_r8(pb);
994  segment->stream_offset_entries[i] = avio_rb64(pb);
995  avio_skip(pb, length - 11);
996  }
997  return 0;
998 }
999 
1000 static int mxf_read_index_table_segment(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
1001 {
1003  switch(tag) {
1004  case 0x3F05:
1005  segment->edit_unit_byte_count = avio_rb32(pb);
1006  av_log(NULL, AV_LOG_TRACE, "EditUnitByteCount %d\n", segment->edit_unit_byte_count);
1007  break;
1008  case 0x3F06:
1009  segment->index_sid = avio_rb32(pb);
1010  av_log(NULL, AV_LOG_TRACE, "IndexSID %d\n", segment->index_sid);
1011  break;
1012  case 0x3F07:
1013  segment->body_sid = avio_rb32(pb);
1014  av_log(NULL, AV_LOG_TRACE, "BodySID %d\n", segment->body_sid);
1015  break;
1016  case 0x3F0A:
1017  av_log(NULL, AV_LOG_TRACE, "IndexEntryArray found\n");
1018  return mxf_read_index_entry_array(pb, segment);
1019  case 0x3F0B:
1020  segment->index_edit_rate.num = avio_rb32(pb);
1021  segment->index_edit_rate.den = avio_rb32(pb);
1022  av_log(NULL, AV_LOG_TRACE, "IndexEditRate %d/%d\n", segment->index_edit_rate.num,
1023  segment->index_edit_rate.den);
1024  break;
1025  case 0x3F0C:
1026  segment->index_start_position = avio_rb64(pb);
1027  av_log(NULL, AV_LOG_TRACE, "IndexStartPosition %"PRId64"\n", segment->index_start_position);
1028  break;
1029  case 0x3F0D:
1030  segment->index_duration = avio_rb64(pb);
1031  av_log(NULL, AV_LOG_TRACE, "IndexDuration %"PRId64"\n", segment->index_duration);
1032  break;
1033  }
1034  return 0;
1035 }
1036 
1037 static void mxf_read_pixel_layout(AVIOContext *pb, MXFDescriptor *descriptor)
1038 {
1039  int code, value, ofs = 0;
1040  char layout[16] = {0}; /* not for printing, may end up not terminated on purpose */
1041 
1042  do {
1043  code = avio_r8(pb);
1044  value = avio_r8(pb);
1045  av_log(NULL, AV_LOG_TRACE, "pixel layout: code %#x\n", code);
1046 
1047  if (ofs <= 14) {
1048  layout[ofs++] = code;
1049  layout[ofs++] = value;
1050  } else
1051  break; /* don't read byte by byte on sneaky files filled with lots of non-zeroes */
1052  } while (code != 0); /* SMPTE 377M E.2.46 */
1053 
1054  ff_mxf_decode_pixel_layout(layout, &descriptor->pix_fmt);
1055 }
1056 
1057 static int mxf_read_generic_descriptor(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
1058 {
1059  MXFDescriptor *descriptor = arg;
1060  int entry_count, entry_size;
1061 
1062  switch(tag) {
1063  case 0x3F01:
1064  return mxf_read_strong_ref_array(pb, &descriptor->sub_descriptors_refs,
1065  &descriptor->sub_descriptors_count);
1066  case 0x3002: /* ContainerDuration */
1067  descriptor->duration = avio_rb64(pb);
1068  break;
1069  case 0x3004:
1070  avio_read(pb, descriptor->essence_container_ul, 16);
1071  break;
1072  case 0x3005:
1073  avio_read(pb, descriptor->codec_ul, 16);
1074  break;
1075  case 0x3006:
1076  descriptor->linked_track_id = avio_rb32(pb);
1077  break;
1078  case 0x3201: /* PictureEssenceCoding */
1079  avio_read(pb, descriptor->essence_codec_ul, 16);
1080  break;
1081  case 0x3203:
1082  descriptor->width = avio_rb32(pb);
1083  break;
1084  case 0x3202:
1085  descriptor->height = avio_rb32(pb);
1086  break;
1087  case 0x320C:
1088  descriptor->frame_layout = avio_r8(pb);
1089  break;
1090  case 0x320D:
1091  entry_count = avio_rb32(pb);
1092  entry_size = avio_rb32(pb);
1093  if (entry_size == 4) {
1094  if (entry_count > 0)
1095  descriptor->video_line_map[0] = avio_rb32(pb);
1096  else
1097  descriptor->video_line_map[0] = 0;
1098  if (entry_count > 1)
1099  descriptor->video_line_map[1] = avio_rb32(pb);
1100  else
1101  descriptor->video_line_map[1] = 0;
1102  } else
1103  av_log(NULL, AV_LOG_WARNING, "VideoLineMap element size %d currently not supported\n", entry_size);
1104  break;
1105  case 0x320E:
1106  descriptor->aspect_ratio.num = avio_rb32(pb);
1107  descriptor->aspect_ratio.den = avio_rb32(pb);
1108  break;
1109  case 0x3212:
1110  descriptor->field_dominance = avio_r8(pb);
1111  break;
1112  case 0x3301:
1113  descriptor->component_depth = avio_rb32(pb);
1114  break;
1115  case 0x3302:
1116  descriptor->horiz_subsampling = avio_rb32(pb);
1117  break;
1118  case 0x3308:
1119  descriptor->vert_subsampling = avio_rb32(pb);
1120  break;
1121  case 0x3D03:
1122  descriptor->sample_rate.num = avio_rb32(pb);
1123  descriptor->sample_rate.den = avio_rb32(pb);
1124  break;
1125  case 0x3D06: /* SoundEssenceCompression */
1126  avio_read(pb, descriptor->essence_codec_ul, 16);
1127  break;
1128  case 0x3D07:
1129  descriptor->channels = avio_rb32(pb);
1130  break;
1131  case 0x3D01:
1132  descriptor->bits_per_sample = avio_rb32(pb);
1133  break;
1134  case 0x3401:
1135  mxf_read_pixel_layout(pb, descriptor);
1136  break;
1137  default:
1138  /* Private uid used by SONY C0023S01.mxf */
1140  if (descriptor->extradata)
1141  av_log(NULL, AV_LOG_WARNING, "Duplicate sony_mpeg4_extradata\n");
1142  av_free(descriptor->extradata);
1143  descriptor->extradata_size = 0;
1144  descriptor->extradata = av_malloc(size);
1145  if (!descriptor->extradata)
1146  return AVERROR(ENOMEM);
1147  descriptor->extradata_size = size;
1148  avio_read(pb, descriptor->extradata, size);
1149  }
1150  if (IS_KLV_KEY(uid, mxf_jp2k_rsiz)) {
1151  uint32_t rsiz = avio_rb16(pb);
1152  if (rsiz == FF_PROFILE_JPEG2000_DCINEMA_2K ||
1154  descriptor->pix_fmt = AV_PIX_FMT_XYZ12;
1155  }
1156  break;
1157  }
1158  return 0;
1159 }
1160 
1161 static int mxf_read_indirect_value(void *arg, AVIOContext *pb, int size)
1162 {
1163  MXFTaggedValue *tagged_value = arg;
1164  uint8_t key[17];
1165 
1166  if (size <= 17)
1167  return 0;
1168 
1169  avio_read(pb, key, 17);
1170  /* TODO: handle other types of of indirect values */
1171  if (memcmp(key, mxf_indirect_value_utf16le, 17) == 0) {
1172  return mxf_read_utf16le_string(pb, size - 17, &tagged_value->value);
1173  } else if (memcmp(key, mxf_indirect_value_utf16be, 17) == 0) {
1174  return mxf_read_utf16be_string(pb, size - 17, &tagged_value->value);
1175  }
1176  return 0;
1177 }
1178 
1179 static int mxf_read_tagged_value(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
1180 {
1181  MXFTaggedValue *tagged_value = arg;
1182  switch (tag){
1183  case 0x5001:
1184  return mxf_read_utf16be_string(pb, size, &tagged_value->name);
1185  case 0x5003:
1186  return mxf_read_indirect_value(tagged_value, pb, size);
1187  }
1188  return 0;
1189 }
1190 
1191 /*
1192  * Match an uid independently of the version byte and up to len common bytes
1193  * Returns: boolean
1194  */
1195 static int mxf_match_uid(const UID key, const UID uid, int len)
1196 {
1197  int i;
1198  for (i = 0; i < len; i++) {
1199  if (i != 7 && key[i] != uid[i])
1200  return 0;
1201  }
1202  return 1;
1203 }
1204 
1205 static const MXFCodecUL *mxf_get_codec_ul(const MXFCodecUL *uls, UID *uid)
1206 {
1207  while (uls->uid[0]) {
1208  if(mxf_match_uid(uls->uid, *uid, uls->matching_len))
1209  break;
1210  uls++;
1211  }
1212  return uls;
1213 }
1214 
1215 static void *mxf_resolve_strong_ref(MXFContext *mxf, UID *strong_ref, enum MXFMetadataSetType type)
1216 {
1217  int i;
1218 
1219  if (!strong_ref)
1220  return NULL;
1221  for (i = 0; i < mxf->metadata_sets_count; i++) {
1222  if (!memcmp(*strong_ref, mxf->metadata_sets[i]->uid, 16) &&
1223  (type == AnyType || mxf->metadata_sets[i]->type == type)) {
1224  return mxf->metadata_sets[i];
1225  }
1226  }
1227  return NULL;
1228 }
1229 
1231  // video essence container uls
1232  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x07,0x0d,0x01,0x03,0x01,0x02,0x0c,0x01,0x00 }, 14, AV_CODEC_ID_JPEG2000 },
1233  { { 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 */
1234  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x02,0x0d,0x01,0x03,0x01,0x02,0x11,0x01,0x00 }, 14, AV_CODEC_ID_DNXHD }, /* VC-3 */
1235  { { 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 */
1236  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x02,0x0d,0x01,0x03,0x01,0x02,0x14,0x01,0x00 }, 14, AV_CODEC_ID_TIFF }, /* TIFF */
1237  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x02,0x0d,0x01,0x03,0x01,0x02,0x15,0x01,0x00 }, 14, AV_CODEC_ID_DIRAC }, /* VC-2 */
1238  { { 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 */
1239  { { 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 */
1240  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x01,0x0d,0x01,0x03,0x01,0x02,0x02,0x41,0x01 }, 14, AV_CODEC_ID_DVVIDEO }, /* DV 625 25mbps */
1241  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x01,0x0d,0x01,0x03,0x01,0x02,0x05,0x00,0x00 }, 14, AV_CODEC_ID_RAWVIDEO }, /* uncompressed picture */
1242  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0a,0x0e,0x0f,0x03,0x01,0x02,0x20,0x01,0x01 }, 15, AV_CODEC_ID_HQ_HQA },
1243  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0a,0x0e,0x0f,0x03,0x01,0x02,0x20,0x02,0x01 }, 15, AV_CODEC_ID_HQX },
1244  { { 0x06,0x0e,0x2b,0x34,0x01,0x01,0x01,0xff,0x4b,0x46,0x41,0x41,0x00,0x0d,0x4d,0x4f }, 14, AV_CODEC_ID_RAWVIDEO }, /* Legacy ?? Uncompressed Picture */
1245  { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, AV_CODEC_ID_NONE },
1246 };
1247 
1248 /* EC ULs for intra-only formats */
1250  { { 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 */
1251  { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, AV_CODEC_ID_NONE },
1252 };
1253 
1254 /* intra-only PictureEssenceCoding ULs, where no corresponding EC UL exists */
1256  { { 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 */
1257  { { 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 */
1258  { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, AV_CODEC_ID_NONE },
1259 };
1260 
1261 /* actual coded width for AVC-Intra to allow selecting correct SPS/PPS */
1263  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0A,0x04,0x01,0x02,0x02,0x01,0x32,0x21,0x01 }, 16, 1440 },
1264  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0A,0x04,0x01,0x02,0x02,0x01,0x32,0x21,0x02 }, 16, 1440 },
1265  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0A,0x04,0x01,0x02,0x02,0x01,0x32,0x21,0x03 }, 16, 1440 },
1266  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x0A,0x04,0x01,0x02,0x02,0x01,0x32,0x21,0x04 }, 16, 1440 },
1267  { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, 0 },
1268 };
1269 
1271  // sound essence container uls
1272  { { 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 */
1273  { { 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 */
1274  { { 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 */
1275  { { 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 */
1276  { { 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) */
1277  { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, AV_CODEC_ID_NONE },
1278 };
1279 
1281  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x09,0x0d,0x01,0x03,0x01,0x02,0x0d,0x00,0x00 }, 16, AV_CODEC_ID_NONE, "vbi_smpte_436M" },
1282  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x09,0x0d,0x01,0x03,0x01,0x02,0x0e,0x00,0x00 }, 16, AV_CODEC_ID_NONE, "vbi_vanc_smpte_436M" },
1283  { { 0x06,0x0e,0x2b,0x34,0x04,0x01,0x01,0x09,0x0d,0x01,0x03,0x01,0x02,0x13,0x01,0x01 }, 16, AV_CODEC_ID_TTML },
1284  { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 0, AV_CODEC_ID_NONE },
1285 };
1286 
1287 static int mxf_get_sorted_table_segments(MXFContext *mxf, int *nb_sorted_segments, MXFIndexTableSegment ***sorted_segments)
1288 {
1289  int i, j, nb_segments = 0;
1290  MXFIndexTableSegment **unsorted_segments;
1291  int last_body_sid = -1, last_index_sid = -1, last_index_start = -1;
1292 
1293  /* count number of segments, allocate arrays and copy unsorted segments */
1294  for (i = 0; i < mxf->metadata_sets_count; i++)
1295  if (mxf->metadata_sets[i]->type == IndexTableSegment)
1296  nb_segments++;
1297 
1298  if (!nb_segments)
1299  return AVERROR_INVALIDDATA;
1300 
1301  if (!(unsorted_segments = av_calloc(nb_segments, sizeof(*unsorted_segments))) ||
1302  !(*sorted_segments = av_calloc(nb_segments, sizeof(**sorted_segments)))) {
1303  av_freep(sorted_segments);
1304  av_free(unsorted_segments);
1305  return AVERROR(ENOMEM);
1306  }
1307 
1308  for (i = j = 0; i < mxf->metadata_sets_count; i++)
1309  if (mxf->metadata_sets[i]->type == IndexTableSegment)
1310  unsorted_segments[j++] = (MXFIndexTableSegment*)mxf->metadata_sets[i];
1311 
1312  *nb_sorted_segments = 0;
1313 
1314  /* sort segments by {BodySID, IndexSID, IndexStartPosition}, remove duplicates while we're at it */
1315  for (i = 0; i < nb_segments; i++) {
1316  int best = -1, best_body_sid = -1, best_index_sid = -1, best_index_start = -1;
1317  uint64_t best_index_duration = 0;
1318 
1319  for (j = 0; j < nb_segments; j++) {
1320  MXFIndexTableSegment *s = unsorted_segments[j];
1321 
1322  /* Require larger BosySID, IndexSID or IndexStartPosition then the previous entry. This removes duplicates.
1323  * We want the smallest values for the keys than what we currently have, unless this is the first such entry this time around.
1324  * If we come across an entry with the same IndexStartPosition but larger IndexDuration, then we'll prefer it over the one we currently have.
1325  */
1326  if ((i == 0 ||
1327  s->body_sid > last_body_sid ||
1328  s->body_sid == last_body_sid && s->index_sid > last_index_sid ||
1329  s->body_sid == last_body_sid && s->index_sid == last_index_sid && s->index_start_position > last_index_start) &&
1330  (best == -1 ||
1331  s->body_sid < best_body_sid ||
1332  s->body_sid == best_body_sid && s->index_sid < best_index_sid ||
1333  s->body_sid == best_body_sid && s->index_sid == best_index_sid && s->index_start_position < best_index_start ||
1334  s->body_sid == best_body_sid && s->index_sid == best_index_sid && s->index_start_position == best_index_start && s->index_duration > best_index_duration)) {
1335  best = j;
1336  best_body_sid = s->body_sid;
1337  best_index_sid = s->index_sid;
1338  best_index_start = s->index_start_position;
1339  best_index_duration = s->index_duration;
1340  }
1341  }
1342 
1343  /* no suitable entry found -> we're done */
1344  if (best == -1)
1345  break;
1346 
1347  (*sorted_segments)[(*nb_sorted_segments)++] = unsorted_segments[best];
1348  last_body_sid = best_body_sid;
1349  last_index_sid = best_index_sid;
1350  last_index_start = best_index_start;
1351  }
1352 
1353  av_free(unsorted_segments);
1354 
1355  return 0;
1356 }
1357 
1358 /**
1359  * Computes the absolute file offset of the given essence container offset
1360  */
1361 static int mxf_absolute_bodysid_offset(MXFContext *mxf, int body_sid, int64_t offset, int64_t *offset_out)
1362 {
1363  MXFPartition *last_p = NULL;
1364  int a, b, m, m0;
1365 
1366  if (offset < 0)
1367  return AVERROR(EINVAL);
1368 
1369  a = -1;
1370  b = mxf->partitions_count;
1371 
1372  while (b - a > 1) {
1373  m0 = m = (a + b) >> 1;
1374 
1375  while (m < b && mxf->partitions[m].body_sid != body_sid)
1376  m++;
1377 
1378  if (m < b && mxf->partitions[m].body_offset <= offset)
1379  a = m;
1380  else
1381  b = m0;
1382  }
1383 
1384  if (a >= 0)
1385  last_p = &mxf->partitions[a];
1386 
1387  if (last_p && (!last_p->essence_length || last_p->essence_length > (offset - last_p->body_offset))) {
1388  *offset_out = last_p->essence_offset + (offset - last_p->body_offset);
1389  return 0;
1390  }
1391 
1392  av_log(mxf->fc, AV_LOG_ERROR,
1393  "failed to find absolute offset of %"PRIX64" in BodySID %i - partial file?\n",
1394  offset, body_sid);
1395 
1396  return AVERROR_INVALIDDATA;
1397 }
1398 
1399 /**
1400  * Returns the end position of the essence container with given BodySID, or zero if unknown
1401  */
1402 static int64_t mxf_essence_container_end(MXFContext *mxf, int body_sid)
1403 {
1404  int x;
1405  int64_t ret = 0;
1406 
1407  for (x = 0; x < mxf->partitions_count; x++) {
1408  MXFPartition *p = &mxf->partitions[x];
1409 
1410  if (p->body_sid != body_sid)
1411  continue;
1412 
1413  if (!p->essence_length)
1414  return 0;
1415 
1416  ret = p->essence_offset + p->essence_length;
1417  }
1418 
1419  return ret;
1420 }
1421 
1422 /* EditUnit -> absolute offset */
1423 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)
1424 {
1425  int i;
1426  int64_t offset_temp = 0;
1427 
1428  for (i = 0; i < index_table->nb_segments; i++) {
1429  MXFIndexTableSegment *s = index_table->segments[i];
1430 
1431  edit_unit = FFMAX(edit_unit, s->index_start_position); /* clamp if trying to seek before start */
1432 
1433  if (edit_unit < s->index_start_position + s->index_duration) {
1434  int64_t index = edit_unit - s->index_start_position;
1435 
1436  if (s->edit_unit_byte_count)
1437  offset_temp += s->edit_unit_byte_count * index;
1438  else if (s->nb_index_entries) {
1439  if (s->nb_index_entries == 2 * s->index_duration + 1)
1440  index *= 2; /* Avid index */
1441 
1442  if (index < 0 || index >= s->nb_index_entries) {
1443  av_log(mxf->fc, AV_LOG_ERROR, "IndexSID %i segment at %"PRId64" IndexEntryArray too small\n",
1444  index_table->index_sid, s->index_start_position);
1445  return AVERROR_INVALIDDATA;
1446  }
1447 
1448  offset_temp = s->stream_offset_entries[index];
1449  } else {
1450  av_log(mxf->fc, AV_LOG_ERROR, "IndexSID %i segment at %"PRId64" missing EditUnitByteCount and IndexEntryArray\n",
1451  index_table->index_sid, s->index_start_position);
1452  return AVERROR_INVALIDDATA;
1453  }
1454 
1455  if (edit_unit_out)
1456  *edit_unit_out = edit_unit;
1457 
1458  return mxf_absolute_bodysid_offset(mxf, index_table->body_sid, offset_temp, offset_out);
1459  } else {
1460  /* EditUnitByteCount == 0 for VBR indexes, which is fine since they use explicit StreamOffsets */
1461  offset_temp += s->edit_unit_byte_count * s->index_duration;
1462  }
1463  }
1464 
1465  if (nag)
1466  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);
1467 
1468  return AVERROR_INVALIDDATA;
1469 }
1470 
1472 {
1473  int i, j, x;
1474  int8_t max_temporal_offset = -128;
1475  uint8_t *flags;
1476 
1477  /* first compute how many entries we have */
1478  for (i = 0; i < index_table->nb_segments; i++) {
1479  MXFIndexTableSegment *s = index_table->segments[i];
1480 
1481  if (!s->nb_index_entries) {
1482  index_table->nb_ptses = 0;
1483  return 0; /* no TemporalOffsets */
1484  }
1485 
1486  index_table->nb_ptses += s->index_duration;
1487  }
1488 
1489  /* paranoid check */
1490  if (index_table->nb_ptses <= 0)
1491  return 0;
1492 
1493  if (!(index_table->ptses = av_calloc(index_table->nb_ptses, sizeof(int64_t))) ||
1494  !(index_table->fake_index = av_calloc(index_table->nb_ptses, sizeof(AVIndexEntry))) ||
1495  !(index_table->offsets = av_calloc(index_table->nb_ptses, sizeof(int8_t))) ||
1496  !(flags = av_calloc(index_table->nb_ptses, sizeof(uint8_t)))) {
1497  av_freep(&index_table->ptses);
1498  av_freep(&index_table->fake_index);
1499  av_freep(&index_table->offsets);
1500  return AVERROR(ENOMEM);
1501  }
1502 
1503  /* we may have a few bad TemporalOffsets
1504  * make sure the corresponding PTSes don't have the bogus value 0 */
1505  for (x = 0; x < index_table->nb_ptses; x++)
1506  index_table->ptses[x] = AV_NOPTS_VALUE;
1507 
1508  /**
1509  * We have this:
1510  *
1511  * x TemporalOffset
1512  * 0: 0
1513  * 1: 1
1514  * 2: 1
1515  * 3: -2
1516  * 4: 1
1517  * 5: 1
1518  * 6: -2
1519  *
1520  * We want to transform it into this:
1521  *
1522  * x DTS PTS
1523  * 0: -1 0
1524  * 1: 0 3
1525  * 2: 1 1
1526  * 3: 2 2
1527  * 4: 3 6
1528  * 5: 4 4
1529  * 6: 5 5
1530  *
1531  * We do this by bucket sorting x by x+TemporalOffset[x] into mxf->ptses,
1532  * then settings mxf->first_dts = -max(TemporalOffset[x]).
1533  * The latter makes DTS <= PTS.
1534  */
1535  for (i = x = 0; i < index_table->nb_segments; i++) {
1536  MXFIndexTableSegment *s = index_table->segments[i];
1537  int index_delta = 1;
1538  int n = s->nb_index_entries;
1539 
1540  if (s->nb_index_entries == 2 * s->index_duration + 1) {
1541  index_delta = 2; /* Avid index */
1542  /* ignore the last entry - it's the size of the essence container */
1543  n--;
1544  }
1545 
1546  for (j = 0; j < n; j += index_delta, x++) {
1547  int offset = s->temporal_offset_entries[j] / index_delta;
1548  int index = x + offset;
1549 
1550  if (x >= index_table->nb_ptses) {
1551  av_log(mxf->fc, AV_LOG_ERROR,
1552  "x >= nb_ptses - IndexEntryCount %i < IndexDuration %"PRId64"?\n",
1554  break;
1555  }
1556 
1557  flags[x] = !(s->flag_entries[j] & 0x30) ? AVINDEX_KEYFRAME : 0;
1558 
1559  if (index < 0 || index >= index_table->nb_ptses) {
1560  av_log(mxf->fc, AV_LOG_ERROR,
1561  "index entry %i + TemporalOffset %i = %i, which is out of bounds\n",
1562  x, offset, index);
1563  continue;
1564  }
1565 
1566  index_table->offsets[x] = offset;
1567  index_table->ptses[index] = x;
1568  max_temporal_offset = FFMAX(max_temporal_offset, offset);
1569  }
1570  }
1571 
1572  /* calculate the fake index table in display order */
1573  for (x = 0; x < index_table->nb_ptses; x++) {
1574  index_table->fake_index[x].timestamp = x;
1575  if (index_table->ptses[x] != AV_NOPTS_VALUE)
1576  index_table->fake_index[index_table->ptses[x]].flags = flags[x];
1577  }
1578  av_freep(&flags);
1579 
1580  index_table->first_dts = -max_temporal_offset;
1581 
1582  return 0;
1583 }
1584 
1585 /**
1586  * Sorts and collects index table segments into index tables.
1587  * Also computes PTSes if possible.
1588  */
1590 {
1591  int i, j, k, ret, nb_sorted_segments;
1592  MXFIndexTableSegment **sorted_segments = NULL;
1593 
1594  if ((ret = mxf_get_sorted_table_segments(mxf, &nb_sorted_segments, &sorted_segments)) ||
1595  nb_sorted_segments <= 0) {
1596  av_log(mxf->fc, AV_LOG_WARNING, "broken or empty index\n");
1597  return 0;
1598  }
1599 
1600  /* sanity check and count unique BodySIDs/IndexSIDs */
1601  for (i = 0; i < nb_sorted_segments; i++) {
1602  if (i == 0 || sorted_segments[i-1]->index_sid != sorted_segments[i]->index_sid)
1603  mxf->nb_index_tables++;
1604  else if (sorted_segments[i-1]->body_sid != sorted_segments[i]->body_sid) {
1605  av_log(mxf->fc, AV_LOG_ERROR, "found inconsistent BodySID\n");
1606  ret = AVERROR_INVALIDDATA;
1607  goto finish_decoding_index;
1608  }
1609  }
1610 
1612  sizeof(*mxf->index_tables));
1613  if (!mxf->index_tables) {
1614  av_log(mxf->fc, AV_LOG_ERROR, "failed to allocate index tables\n");
1615  ret = AVERROR(ENOMEM);
1616  goto finish_decoding_index;
1617  }
1618 
1619  /* distribute sorted segments to index tables */
1620  for (i = j = 0; i < nb_sorted_segments; i++) {
1621  if (i != 0 && sorted_segments[i-1]->index_sid != sorted_segments[i]->index_sid) {
1622  /* next IndexSID */
1623  j++;
1624  }
1625 
1626  mxf->index_tables[j].nb_segments++;
1627  }
1628 
1629  for (i = j = 0; j < mxf->nb_index_tables; i += mxf->index_tables[j++].nb_segments) {
1630  MXFIndexTable *t = &mxf->index_tables[j];
1631  MXFTrack *mxf_track = NULL;
1632 
1634  sizeof(*t->segments));
1635 
1636  if (!t->segments) {
1637  av_log(mxf->fc, AV_LOG_ERROR, "failed to allocate IndexTableSegment"
1638  " pointer array\n");
1639  ret = AVERROR(ENOMEM);
1640  goto finish_decoding_index;
1641  }
1642 
1643  if (sorted_segments[i]->index_start_position)
1644  av_log(mxf->fc, AV_LOG_WARNING, "IndexSID %i starts at EditUnit %"PRId64" - seeking may not work as expected\n",
1645  sorted_segments[i]->index_sid, sorted_segments[i]->index_start_position);
1646 
1647  memcpy(t->segments, &sorted_segments[i], t->nb_segments * sizeof(MXFIndexTableSegment*));
1648  t->index_sid = sorted_segments[i]->index_sid;
1649  t->body_sid = sorted_segments[i]->body_sid;
1650 
1651  if ((ret = mxf_compute_ptses_fake_index(mxf, t)) < 0)
1652  goto finish_decoding_index;
1653 
1654  for (k = 0; k < mxf->fc->nb_streams; k++) {
1655  MXFTrack *track = mxf->fc->streams[k]->priv_data;
1656  if (track && track->index_sid == t->index_sid) {
1657  mxf_track = track;
1658  break;
1659  }
1660  }
1661 
1662  /* fix zero IndexDurations */
1663  for (k = 0; k < t->nb_segments; k++) {
1664  if (t->segments[k]->index_duration)
1665  continue;
1666 
1667  if (t->nb_segments > 1)
1668  av_log(mxf->fc, AV_LOG_WARNING, "IndexSID %i segment %i has zero IndexDuration and there's more than one segment\n",
1669  t->index_sid, k);
1670 
1671  if (!mxf_track) {
1672  av_log(mxf->fc, AV_LOG_WARNING, "no streams?\n");
1673  break;
1674  }
1675 
1676  /* assume the first stream's duration is reasonable
1677  * leave index_duration = 0 on further segments in case we have any (unlikely)
1678  */
1679  t->segments[k]->index_duration = mxf_track->original_duration;
1680  break;
1681  }
1682  }
1683 
1684  ret = 0;
1685 finish_decoding_index:
1686  av_free(sorted_segments);
1687  return ret;
1688 }
1689 
1690 static int mxf_is_intra_only(MXFDescriptor *descriptor)
1691 {
1692  return mxf_get_codec_ul(mxf_intra_only_essence_container_uls,
1693  &descriptor->essence_container_ul)->id != AV_CODEC_ID_NONE ||
1694  mxf_get_codec_ul(mxf_intra_only_picture_essence_coding_uls,
1695  &descriptor->essence_codec_ul)->id != AV_CODEC_ID_NONE;
1696 }
1697 
1698 static int mxf_uid_to_str(UID uid, char **str)
1699 {
1700  int i;
1701  char *p;
1702  p = *str = av_mallocz(sizeof(UID) * 2 + 4 + 1);
1703  if (!p)
1704  return AVERROR(ENOMEM);
1705  for (i = 0; i < sizeof(UID); i++) {
1706  snprintf(p, 2 + 1, "%.2x", uid[i]);
1707  p += 2;
1708  if (i == 3 || i == 5 || i == 7 || i == 9) {
1709  snprintf(p, 1 + 1, "-");
1710  p++;
1711  }
1712  }
1713  return 0;
1714 }
1715 
1716 static int mxf_umid_to_str(UID ul, UID uid, char **str)
1717 {
1718  int i;
1719  char *p;
1720  p = *str = av_mallocz(sizeof(UID) * 4 + 2 + 1);
1721  if (!p)
1722  return AVERROR(ENOMEM);
1723  snprintf(p, 2 + 1, "0x");
1724  p += 2;
1725  for (i = 0; i < sizeof(UID); i++) {
1726  snprintf(p, 2 + 1, "%.2X", ul[i]);
1727  p += 2;
1728 
1729  }
1730  for (i = 0; i < sizeof(UID); i++) {
1731  snprintf(p, 2 + 1, "%.2X", uid[i]);
1732  p += 2;
1733  }
1734  return 0;
1735 }
1736 
1737 static int mxf_add_umid_metadata(AVDictionary **pm, const char *key, MXFPackage* package)
1738 {
1739  char *str;
1740  int ret;
1741  if (!package)
1742  return 0;
1743  if ((ret = mxf_umid_to_str(package->package_ul, package->package_uid, &str)) < 0)
1744  return ret;
1745  av_dict_set(pm, key, str, AV_DICT_DONT_STRDUP_VAL);
1746  return 0;
1747 }
1748 
1749 static int mxf_add_timecode_metadata(AVDictionary **pm, const char *key, AVTimecode *tc)
1750 {
1751  char buf[AV_TIMECODE_STR_SIZE];
1752  av_dict_set(pm, key, av_timecode_make_string(tc, buf, 0), 0);
1753 
1754  return 0;
1755 }
1756 
1758 {
1759  MXFStructuralComponent *component = NULL;
1760  MXFPulldownComponent *pulldown = NULL;
1761 
1762  component = mxf_resolve_strong_ref(mxf, strong_ref, AnyType);
1763  if (!component)
1764  return NULL;
1765 
1766  switch (component->type) {
1767  case TimecodeComponent:
1768  return (MXFTimecodeComponent*)component;
1769  case PulldownComponent: /* timcode component may be located on a pulldown component */
1770  pulldown = (MXFPulldownComponent*)component;
1772  default:
1773  break;
1774  }
1775  return NULL;
1776 }
1777 
1778 static MXFPackage* mxf_resolve_source_package(MXFContext *mxf, UID package_ul, UID package_uid)
1779 {
1780  MXFPackage *package = NULL;
1781  int i;
1782 
1783  for (i = 0; i < mxf->packages_count; i++) {
1784  package = mxf_resolve_strong_ref(mxf, &mxf->packages_refs[i], SourcePackage);
1785  if (!package)
1786  continue;
1787 
1788  if (!memcmp(package->package_ul, package_ul, 16) && !memcmp(package->package_uid, package_uid, 16))
1789  return package;
1790  }
1791  return NULL;
1792 }
1793 
1794 static MXFDescriptor* mxf_resolve_multidescriptor(MXFContext *mxf, MXFDescriptor *descriptor, int track_id)
1795 {
1796  MXFDescriptor *sub_descriptor = NULL;
1797  int i;
1798 
1799  if (!descriptor)
1800  return NULL;
1801 
1802  if (descriptor->type == MultipleDescriptor) {
1803  for (i = 0; i < descriptor->sub_descriptors_count; i++) {
1804  sub_descriptor = mxf_resolve_strong_ref(mxf, &descriptor->sub_descriptors_refs[i], Descriptor);
1805 
1806  if (!sub_descriptor) {
1807  av_log(mxf->fc, AV_LOG_ERROR, "could not resolve sub descriptor strong ref\n");
1808  continue;
1809  }
1810  if (sub_descriptor->linked_track_id == track_id) {
1811  return sub_descriptor;
1812  }
1813  }
1814  } else if (descriptor->type == Descriptor)
1815  return descriptor;
1816 
1817  return NULL;
1818 }
1819 
1821 {
1822  MXFStructuralComponent *component = NULL;
1823  MXFPackage *package = NULL;
1824  MXFDescriptor *descriptor = NULL;
1825  int i;
1826 
1827  if (!essence_group || !essence_group->structural_components_count)
1828  return NULL;
1829 
1830  /* essence groups contains multiple representations of the same media,
1831  this return the first components with a valid Descriptor typically index 0 */
1832  for (i =0; i < essence_group->structural_components_count; i++){
1833  component = mxf_resolve_strong_ref(mxf, &essence_group->structural_components_refs[i], SourceClip);
1834  if (!component)
1835  continue;
1836 
1837  if (!(package = mxf_resolve_source_package(mxf, component->source_package_ul, component->source_package_uid)))
1838  continue;
1839 
1840  descriptor = mxf_resolve_strong_ref(mxf, &package->descriptor_ref, Descriptor);
1841  if (descriptor)
1842  return component;
1843  }
1844  return NULL;
1845 }
1846 
1848 {
1849  MXFStructuralComponent *component = NULL;
1850 
1851  component = mxf_resolve_strong_ref(mxf, strong_ref, AnyType);
1852  if (!component)
1853  return NULL;
1854  switch (component->type) {
1855  case SourceClip:
1856  return component;
1857  case EssenceGroup:
1858  return mxf_resolve_essence_group_choice(mxf, (MXFEssenceGroup*) component);
1859  default:
1860  break;
1861  }
1862  return NULL;
1863 }
1864 
1866 {
1868  int size, i;
1869  char *key = NULL;
1870 
1871  for (i = 0; i < package->comment_count; i++) {
1872  tag = mxf_resolve_strong_ref(mxf, &package->comment_refs[i], TaggedValue);
1873  if (!tag || !tag->name || !tag->value)
1874  continue;
1875 
1876  size = strlen(tag->name) + 8 + 1;
1877  key = av_mallocz(size);
1878  if (!key)
1879  return AVERROR(ENOMEM);
1880 
1881  snprintf(key, size, "comment_%s", tag->name);
1882  av_dict_set(pm, key, tag->value, AV_DICT_DONT_STRDUP_KEY);
1883  }
1884  return 0;
1885 }
1886 
1888 {
1889  MXFPackage *physical_package = NULL;
1890  MXFTrack *physical_track = NULL;
1891  MXFStructuralComponent *sourceclip = NULL;
1892  MXFTimecodeComponent *mxf_tc = NULL;
1893  int i, j, k;
1894  AVTimecode tc;
1895  int flags;
1896  int64_t start_position;
1897 
1898  for (i = 0; i < source_track->sequence->structural_components_count; i++) {
1899  sourceclip = mxf_resolve_strong_ref(mxf, &source_track->sequence->structural_components_refs[i], SourceClip);
1900  if (!sourceclip)
1901  continue;
1902 
1903  if (!(physical_package = mxf_resolve_source_package(mxf, sourceclip->source_package_ul, sourceclip->source_package_uid)))
1904  break;
1905 
1906  mxf_add_umid_metadata(&st->metadata, "reel_umid", physical_package);
1907 
1908  /* the name of physical source package is name of the reel or tape */
1909  if (physical_package->name && physical_package->name[0])
1910  av_dict_set(&st->metadata, "reel_name", physical_package->name, 0);
1911 
1912  /* the source timecode is calculated by adding the start_position of the sourceclip from the file source package track
1913  * to the start_frame of the timecode component located on one of the tracks of the physical source package.
1914  */
1915  for (j = 0; j < physical_package->tracks_count; j++) {
1916  if (!(physical_track = mxf_resolve_strong_ref(mxf, &physical_package->tracks_refs[j], Track))) {
1917  av_log(mxf->fc, AV_LOG_ERROR, "could not resolve source track strong ref\n");
1918  continue;
1919  }
1920 
1921  if (!(physical_track->sequence = mxf_resolve_strong_ref(mxf, &physical_track->sequence_ref, Sequence))) {
1922  av_log(mxf->fc, AV_LOG_ERROR, "could not resolve source track sequence strong ref\n");
1923  continue;
1924  }
1925 
1926  if (physical_track->edit_rate.num <= 0 ||
1927  physical_track->edit_rate.den <= 0) {
1928  av_log(mxf->fc, AV_LOG_WARNING,
1929  "Invalid edit rate (%d/%d) found on structural"
1930  " component #%d, defaulting to 25/1\n",
1931  physical_track->edit_rate.num,
1932  physical_track->edit_rate.den, i);
1933  physical_track->edit_rate = (AVRational){25, 1};
1934  }
1935 
1936  for (k = 0; k < physical_track->sequence->structural_components_count; k++) {
1937  if (!(mxf_tc = mxf_resolve_timecode_component(mxf, &physical_track->sequence->structural_components_refs[k])))
1938  continue;
1939 
1940  flags = mxf_tc->drop_frame == 1 ? AV_TIMECODE_FLAG_DROPFRAME : 0;
1941  /* scale sourceclip start_position to match physical track edit rate */
1942  start_position = av_rescale_q(sourceclip->start_position,
1943  physical_track->edit_rate,
1944  source_track->edit_rate);
1945 
1946  if (av_timecode_init(&tc, mxf_tc->rate, flags, start_position + mxf_tc->start_frame, mxf->fc) == 0) {
1947  mxf_add_timecode_metadata(&st->metadata, "timecode", &tc);
1948  return 0;
1949  }
1950  }
1951  }
1952  }
1953 
1954  return 0;
1955 }
1956 
1958 {
1959  MXFStructuralComponent *component = NULL;
1960  const MXFCodecUL *codec_ul = NULL;
1961  MXFPackage tmp_package;
1962  AVStream *st;
1963  int j;
1964 
1965  for (j = 0; j < track->sequence->structural_components_count; j++) {
1966  component = mxf_resolve_sourceclip(mxf, &track->sequence->structural_components_refs[j]);
1967  if (!component)
1968  continue;
1969  break;
1970  }
1971  if (!component)
1972  return 0;
1973 
1974  st = avformat_new_stream(mxf->fc, NULL);
1975  if (!st) {
1976  av_log(mxf->fc, AV_LOG_ERROR, "could not allocate metadata stream\n");
1977  return AVERROR(ENOMEM);
1978  }
1979 
1982  st->id = track->track_id;
1983 
1984  memcpy(&tmp_package.package_ul, component->source_package_ul, 16);
1985  memcpy(&tmp_package.package_uid, component->source_package_uid, 16);
1986  mxf_add_umid_metadata(&st->metadata, "file_package_umid", &tmp_package);
1987  if (track->name && track->name[0])
1988  av_dict_set(&st->metadata, "track_name", track->name, 0);
1989 
1991  av_dict_set(&st->metadata, "data_type", av_get_media_type_string(codec_ul->id), 0);
1992  return 0;
1993 }
1994 
1996 {
1997  MXFPackage *material_package = NULL;
1998  int i, j, k, ret;
1999 
2000  av_log(mxf->fc, AV_LOG_TRACE, "metadata sets count %d\n", mxf->metadata_sets_count);
2001  /* TODO: handle multiple material packages (OP3x) */
2002  for (i = 0; i < mxf->packages_count; i++) {
2003  material_package = mxf_resolve_strong_ref(mxf, &mxf->packages_refs[i], MaterialPackage);
2004  if (material_package) break;
2005  }
2006  if (!material_package) {
2007  av_log(mxf->fc, AV_LOG_ERROR, "no material package found\n");
2008  return AVERROR_INVALIDDATA;
2009  }
2010 
2011  mxf_add_umid_metadata(&mxf->fc->metadata, "material_package_umid", material_package);
2012  if (material_package->name && material_package->name[0])
2013  av_dict_set(&mxf->fc->metadata, "material_package_name", material_package->name, 0);
2014  mxf_parse_package_comments(mxf, &mxf->fc->metadata, material_package);
2015 
2016  for (i = 0; i < material_package->tracks_count; i++) {
2017  MXFPackage *source_package = NULL;
2018  MXFTrack *material_track = NULL;
2019  MXFTrack *source_track = NULL;
2020  MXFTrack *temp_track = NULL;
2021  MXFDescriptor *descriptor = NULL;
2022  MXFStructuralComponent *component = NULL;
2023  MXFTimecodeComponent *mxf_tc = NULL;
2024  UID *essence_container_ul = NULL;
2025  const MXFCodecUL *codec_ul = NULL;
2026  const MXFCodecUL *container_ul = NULL;
2027  const MXFCodecUL *pix_fmt_ul = NULL;
2028  AVStream *st;
2029  AVTimecode tc;
2030  int flags;
2031 
2032  if (!(material_track = mxf_resolve_strong_ref(mxf, &material_package->tracks_refs[i], Track))) {
2033  av_log(mxf->fc, AV_LOG_ERROR, "could not resolve material track strong ref\n");
2034  continue;
2035  }
2036 
2037  if ((component = mxf_resolve_strong_ref(mxf, &material_track->sequence_ref, TimecodeComponent))) {
2038  mxf_tc = (MXFTimecodeComponent*)component;
2039  flags = mxf_tc->drop_frame == 1 ? AV_TIMECODE_FLAG_DROPFRAME : 0;
2040  if (av_timecode_init(&tc, mxf_tc->rate, flags, mxf_tc->start_frame, mxf->fc) == 0) {
2041  mxf_add_timecode_metadata(&mxf->fc->metadata, "timecode", &tc);
2042  }
2043  }
2044 
2045  if (!(material_track->sequence = mxf_resolve_strong_ref(mxf, &material_track->sequence_ref, Sequence))) {
2046  av_log(mxf->fc, AV_LOG_ERROR, "could not resolve material track sequence strong ref\n");
2047  continue;
2048  }
2049 
2050  for (j = 0; j < material_track->sequence->structural_components_count; j++) {
2051  component = mxf_resolve_strong_ref(mxf, &material_track->sequence->structural_components_refs[j], TimecodeComponent);
2052  if (!component)
2053  continue;
2054 
2055  mxf_tc = (MXFTimecodeComponent*)component;
2056  flags = mxf_tc->drop_frame == 1 ? AV_TIMECODE_FLAG_DROPFRAME : 0;
2057  if (av_timecode_init(&tc, mxf_tc->rate, flags, mxf_tc->start_frame, mxf->fc) == 0) {
2058  mxf_add_timecode_metadata(&mxf->fc->metadata, "timecode", &tc);
2059  break;
2060  }
2061  }
2062 
2063  /* TODO: handle multiple source clips, only finds first valid source clip */
2064  if(material_track->sequence->structural_components_count > 1)
2065  av_log(mxf->fc, AV_LOG_WARNING, "material track %d: has %d components\n",
2066  material_track->track_id, material_track->sequence->structural_components_count);
2067 
2068  for (j = 0; j < material_track->sequence->structural_components_count; j++) {
2069  component = mxf_resolve_sourceclip(mxf, &material_track->sequence->structural_components_refs[j]);
2070  if (!component)
2071  continue;
2072 
2073  source_package = mxf_resolve_source_package(mxf, component->source_package_ul, component->source_package_uid);
2074  if (!source_package) {
2075  av_log(mxf->fc, AV_LOG_TRACE, "material track %d: no corresponding source package found\n", material_track->track_id);
2076  continue;
2077  }
2078  for (k = 0; k < source_package->tracks_count; k++) {
2079  if (!(temp_track = mxf_resolve_strong_ref(mxf, &source_package->tracks_refs[k], Track))) {
2080  av_log(mxf->fc, AV_LOG_ERROR, "could not resolve source track strong ref\n");
2081  ret = AVERROR_INVALIDDATA;
2082  goto fail_and_free;
2083  }
2084  if (temp_track->track_id == component->source_track_id) {
2085  source_track = temp_track;
2086  break;
2087  }
2088  }
2089  if (!source_track) {
2090  av_log(mxf->fc, AV_LOG_ERROR, "material track %d: no corresponding source track found\n", material_track->track_id);
2091  break;
2092  }
2093 
2094  for (k = 0; k < mxf->essence_container_data_count; k++) {
2095  MXFEssenceContainerData *essence_data;
2096 
2097  if (!(essence_data = mxf_resolve_strong_ref(mxf, &mxf->essence_container_data_refs[k], EssenceContainerData))) {
2098  av_log(mxf, AV_LOG_TRACE, "could not resolve essence container data strong ref\n");
2099  continue;
2100  }
2101  if (!memcmp(component->source_package_ul, essence_data->package_ul, sizeof(UID)) && !memcmp(component->source_package_uid, essence_data->package_uid, sizeof(UID))) {
2102  source_track->body_sid = essence_data->body_sid;
2103  source_track->index_sid = essence_data->index_sid;
2104  break;
2105  }
2106  }
2107 
2108  if(source_track && component)
2109  break;
2110  }
2111  if (!source_track || !component || !source_package) {
2112  if((ret = mxf_add_metadata_stream(mxf, material_track)))
2113  goto fail_and_free;
2114  continue;
2115  }
2116 
2117  if (!(source_track->sequence = mxf_resolve_strong_ref(mxf, &source_track->sequence_ref, Sequence))) {
2118  av_log(mxf->fc, AV_LOG_ERROR, "could not resolve source track sequence strong ref\n");
2119  ret = AVERROR_INVALIDDATA;
2120  goto fail_and_free;
2121  }
2122 
2123  /* 0001GL00.MXF.A1.mxf_opatom.mxf has the same SourcePackageID as 0001GL.MXF.V1.mxf_opatom.mxf
2124  * This would result in both files appearing to have two streams. Work around this by sanity checking DataDefinition */
2125  if (memcmp(material_track->sequence->data_definition_ul, source_track->sequence->data_definition_ul, 16)) {
2126  av_log(mxf->fc, AV_LOG_ERROR, "material track %d: DataDefinition mismatch\n", material_track->track_id);
2127  continue;
2128  }
2129 
2130  st = avformat_new_stream(mxf->fc, NULL);
2131  if (!st) {
2132  av_log(mxf->fc, AV_LOG_ERROR, "could not allocate stream\n");
2133  ret = AVERROR(ENOMEM);
2134  goto fail_and_free;
2135  }
2136  st->id = material_track->track_id;
2137  st->priv_data = source_track;
2138 
2139  source_package->descriptor = mxf_resolve_strong_ref(mxf, &source_package->descriptor_ref, AnyType);
2140  descriptor = mxf_resolve_multidescriptor(mxf, source_package->descriptor, source_track->track_id);
2141 
2142  /* A SourceClip from a EssenceGroup may only be a single frame of essence data. The clips duration is then how many
2143  * frames its suppose to repeat for. Descriptor->duration, if present, contains the real duration of the essence data */
2144  if (descriptor && descriptor->duration != AV_NOPTS_VALUE)
2145  source_track->original_duration = st->duration = FFMIN(descriptor->duration, component->duration);
2146  else
2147  source_track->original_duration = st->duration = component->duration;
2148 
2149  if (st->duration == -1)
2150  st->duration = AV_NOPTS_VALUE;
2151  st->start_time = component->start_position;
2152  if (material_track->edit_rate.num <= 0 ||
2153  material_track->edit_rate.den <= 0) {
2154  av_log(mxf->fc, AV_LOG_WARNING,
2155  "Invalid edit rate (%d/%d) found on stream #%d, "
2156  "defaulting to 25/1\n",
2157  material_track->edit_rate.num,
2158  material_track->edit_rate.den, st->index);
2159  material_track->edit_rate = (AVRational){25, 1};
2160  }
2161  avpriv_set_pts_info(st, 64, material_track->edit_rate.den, material_track->edit_rate.num);
2162 
2163  /* ensure SourceTrack EditRate == MaterialTrack EditRate since only
2164  * the former is accessible via st->priv_data */
2165  source_track->edit_rate = material_track->edit_rate;
2166 
2167  PRINT_KEY(mxf->fc, "data definition ul", source_track->sequence->data_definition_ul);
2169  st->codecpar->codec_type = codec_ul->id;
2170 
2171  if (!descriptor) {
2172  av_log(mxf->fc, AV_LOG_INFO, "source track %d: stream %d, no descriptor found\n", source_track->track_id, st->index);
2173  continue;
2174  }
2175  PRINT_KEY(mxf->fc, "essence codec ul", descriptor->essence_codec_ul);
2176  PRINT_KEY(mxf->fc, "essence container ul", descriptor->essence_container_ul);
2177  essence_container_ul = &descriptor->essence_container_ul;
2178  /* HACK: replacing the original key with mxf_encrypted_essence_container
2179  * is not allowed according to s429-6, try to find correct information anyway */
2180  if (IS_KLV_KEY(essence_container_ul, mxf_encrypted_essence_container)) {
2181  av_log(mxf->fc, AV_LOG_INFO, "broken encrypted mxf file\n");
2182  for (k = 0; k < mxf->metadata_sets_count; k++) {
2183  MXFMetadataSet *metadata = mxf->metadata_sets[k];
2184  if (metadata->type == CryptoContext) {
2185  essence_container_ul = &((MXFCryptoContext *)metadata)->source_container_ul;
2186  break;
2187  }
2188  }
2189  }
2190 
2191  /* TODO: drop PictureEssenceCoding and SoundEssenceCompression, only check EssenceContainer */
2192  codec_ul = mxf_get_codec_ul(ff_mxf_codec_uls, &descriptor->essence_codec_ul);
2193  st->codecpar->codec_id = (enum AVCodecID)codec_ul->id;
2194  if (st->codecpar->codec_id == AV_CODEC_ID_NONE) {
2195  codec_ul = mxf_get_codec_ul(ff_mxf_codec_uls, &descriptor->codec_ul);
2196  st->codecpar->codec_id = (enum AVCodecID)codec_ul->id;
2197  }
2198 
2199  av_log(mxf->fc, AV_LOG_VERBOSE, "%s: Universal Label: ",
2201  for (k = 0; k < 16; k++) {
2202  av_log(mxf->fc, AV_LOG_VERBOSE, "%.2x",
2203  descriptor->essence_codec_ul[k]);
2204  if (!(k+1 & 19) || k == 5)
2205  av_log(mxf->fc, AV_LOG_VERBOSE, ".");
2206  }
2207  av_log(mxf->fc, AV_LOG_VERBOSE, "\n");
2208 
2209  mxf_add_umid_metadata(&st->metadata, "file_package_umid", source_package);
2210  if (source_package->name && source_package->name[0])
2211  av_dict_set(&st->metadata, "file_package_name", source_package->name, 0);
2212  if (material_track->name && material_track->name[0])
2213  av_dict_set(&st->metadata, "track_name", material_track->name, 0);
2214 
2215  mxf_parse_physical_source_package(mxf, source_track, st);
2216 
2217  if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
2218  source_track->intra_only = mxf_is_intra_only(descriptor);
2219  container_ul = mxf_get_codec_ul(mxf_picture_essence_container_uls, essence_container_ul);
2220  if (st->codecpar->codec_id == AV_CODEC_ID_NONE)
2221  st->codecpar->codec_id = container_ul->id;
2222  st->codecpar->width = descriptor->width;
2223  st->codecpar->height = descriptor->height; /* Field height, not frame height */
2224  switch (descriptor->frame_layout) {
2225  case FullFrame:
2227  break;
2228  case OneField:
2229  /* Every other line is stored and needs to be duplicated. */
2230  av_log(mxf->fc, AV_LOG_INFO, "OneField frame layout isn't currently supported\n");
2231  break; /* The correct thing to do here is fall through, but by breaking we might be
2232  able to decode some streams at half the vertical resolution, rather than not al all.
2233  It's also for compatibility with the old behavior. */
2234  case MixedFields:
2235  break;
2236  case SegmentedFrame:
2238  case SeparateFields:
2239  av_log(mxf->fc, AV_LOG_DEBUG, "video_line_map: (%d, %d), field_dominance: %d\n",
2240  descriptor->video_line_map[0], descriptor->video_line_map[1],
2241  descriptor->field_dominance);
2242  if ((descriptor->video_line_map[0] > 0) && (descriptor->video_line_map[1] > 0)) {
2243  /* Detect coded field order from VideoLineMap:
2244  * (even, even) => bottom field coded first
2245  * (even, odd) => top field coded first
2246  * (odd, even) => top field coded first
2247  * (odd, odd) => bottom field coded first
2248  */
2249  if ((descriptor->video_line_map[0] + descriptor->video_line_map[1]) % 2) {
2250  switch (descriptor->field_dominance) {
2254  break;
2257  break;
2258  default:
2260  "Field dominance %d support",
2261  descriptor->field_dominance);
2262  }
2263  } else {
2264  switch (descriptor->field_dominance) {
2268  break;
2271  break;
2272  default:
2274  "Field dominance %d support",
2275  descriptor->field_dominance);
2276  }
2277  }
2278  }
2279  /* Turn field height into frame height. */
2280  st->codecpar->height *= 2;
2281  break;
2282  default:
2283  av_log(mxf->fc, AV_LOG_INFO, "Unknown frame layout type: %d\n", descriptor->frame_layout);
2284  }
2285  if (st->codecpar->codec_id == AV_CODEC_ID_RAWVIDEO) {
2286  st->codecpar->format = descriptor->pix_fmt;
2287  if (st->codecpar->format == AV_PIX_FMT_NONE) {
2289  &descriptor->essence_codec_ul);
2290  st->codecpar->format = (enum AVPixelFormat)pix_fmt_ul->id;
2291  if (st->codecpar->format== AV_PIX_FMT_NONE) {
2293  &descriptor->essence_codec_ul)->id;
2294  if (!st->codecpar->codec_tag) {
2295  /* support files created before RP224v10 by defaulting to UYVY422
2296  if subsampling is 4:2:2 and component depth is 8-bit */
2297  if (descriptor->horiz_subsampling == 2 &&
2298  descriptor->vert_subsampling == 1 &&
2299  descriptor->component_depth == 8) {
2301  }
2302  }
2303  }
2304  }
2305  }
2307  if (material_track->sequence->origin) {
2308  av_dict_set_int(&st->metadata, "material_track_origin", material_track->sequence->origin, 0);
2309  }
2310  if (source_track->sequence->origin) {
2311  av_dict_set_int(&st->metadata, "source_track_origin", source_track->sequence->origin, 0);
2312  }
2313  if (descriptor->aspect_ratio.num && descriptor->aspect_ratio.den)
2314  st->display_aspect_ratio = descriptor->aspect_ratio;
2315  } else if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
2316  container_ul = mxf_get_codec_ul(mxf_sound_essence_container_uls, essence_container_ul);
2317  /* Only overwrite existing codec ID if it is unset or A-law, which is the default according to SMPTE RP 224. */
2318  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))
2319  st->codecpar->codec_id = (enum AVCodecID)container_ul->id;
2320  st->codecpar->channels = descriptor->channels;
2321  st->codecpar->bits_per_coded_sample = descriptor->bits_per_sample;
2322 
2323  if (descriptor->sample_rate.den > 0) {
2324  st->codecpar->sample_rate = descriptor->sample_rate.num / descriptor->sample_rate.den;
2325  avpriv_set_pts_info(st, 64, descriptor->sample_rate.den, descriptor->sample_rate.num);
2326  } else {
2327  av_log(mxf->fc, AV_LOG_WARNING, "invalid sample rate (%d/%d) "
2328  "found for stream #%d, time base forced to 1/48000\n",
2329  descriptor->sample_rate.num, descriptor->sample_rate.den,
2330  st->index);
2331  avpriv_set_pts_info(st, 64, 1, 48000);
2332  }
2333 
2334  /* if duration is set, rescale it from EditRate to SampleRate */
2335  if (st->duration != AV_NOPTS_VALUE)
2336  st->duration = av_rescale_q(st->duration,
2337  av_inv_q(material_track->edit_rate),
2338  st->time_base);
2339 
2340  /* TODO: implement AV_CODEC_ID_RAWAUDIO */
2341  if (st->codecpar->codec_id == AV_CODEC_ID_PCM_S16LE) {
2342  if (descriptor->bits_per_sample > 16 && descriptor->bits_per_sample <= 24)
2344  else if (descriptor->bits_per_sample == 32)
2346  } else if (st->codecpar->codec_id == AV_CODEC_ID_PCM_S16BE) {
2347  if (descriptor->bits_per_sample > 16 && descriptor->bits_per_sample <= 24)
2349  else if (descriptor->bits_per_sample == 32)
2351  } else if (st->codecpar->codec_id == AV_CODEC_ID_MP2) {
2353  }
2354  } else if (st->codecpar->codec_type == AVMEDIA_TYPE_DATA) {
2355  enum AVMediaType type;
2356  container_ul = mxf_get_codec_ul(mxf_data_essence_container_uls, essence_container_ul);
2357  if (st->codecpar->codec_id == AV_CODEC_ID_NONE)
2358  st->codecpar->codec_id = container_ul->id;
2359  type = avcodec_get_type(st->codecpar->codec_id);
2360  if (type == AVMEDIA_TYPE_SUBTITLE)
2361  st->codecpar->codec_type = type;
2362  if (container_ul->desc)
2363  av_dict_set(&st->metadata, "data_type", container_ul->desc, 0);
2364  }
2365  if (descriptor->extradata) {
2366  if (!ff_alloc_extradata(st->codecpar, descriptor->extradata_size)) {
2367  memcpy(st->codecpar->extradata, descriptor->extradata, descriptor->extradata_size);
2368  }
2369  } else if (st->codecpar->codec_id == AV_CODEC_ID_H264) {
2370  int coded_width = mxf_get_codec_ul(mxf_intra_only_picture_coded_width,
2371  &descriptor->essence_codec_ul)->id;
2372  if (coded_width)
2373  st->codecpar->width = coded_width;
2374  ret = ff_generate_avci_extradata(st);
2375  if (ret < 0)
2376  return ret;
2377  }
2378  if (st->codecpar->codec_type != AVMEDIA_TYPE_DATA && (*essence_container_ul)[15] > 0x01) {
2379  /* TODO: decode timestamps */
2381  }
2382  }
2383 
2384  ret = 0;
2385 fail_and_free:
2386  return ret;
2387 }
2388 
2389 static int64_t mxf_timestamp_to_int64(uint64_t timestamp)
2390 {
2391  struct tm time = { 0 };
2392  time.tm_year = (timestamp >> 48) - 1900;
2393  time.tm_mon = (timestamp >> 40 & 0xFF) - 1;
2394  time.tm_mday = (timestamp >> 32 & 0xFF);
2395  time.tm_hour = (timestamp >> 24 & 0xFF);
2396  time.tm_min = (timestamp >> 16 & 0xFF);
2397  time.tm_sec = (timestamp >> 8 & 0xFF);
2398 
2399  /* msvcrt versions of strftime calls the invalid parameter handler
2400  * (aborting the process if one isn't set) if the parameters are out
2401  * of range. */
2402  time.tm_mon = av_clip(time.tm_mon, 0, 11);
2403  time.tm_mday = av_clip(time.tm_mday, 1, 31);
2404  time.tm_hour = av_clip(time.tm_hour, 0, 23);
2405  time.tm_min = av_clip(time.tm_min, 0, 59);
2406  time.tm_sec = av_clip(time.tm_sec, 0, 59);
2407 
2408  return (int64_t)av_timegm(&time) * 1000000;
2409 }
2410 
2411 #define SET_STR_METADATA(pb, name, str) do { \
2412  if ((ret = mxf_read_utf16be_string(pb, size, &str)) < 0) \
2413  return ret; \
2414  av_dict_set(&s->metadata, name, str, AV_DICT_DONT_STRDUP_VAL); \
2415 } while (0)
2416 
2417 #define SET_UID_METADATA(pb, name, var, str) do { \
2418  avio_read(pb, var, 16); \
2419  if ((ret = mxf_uid_to_str(var, &str)) < 0) \
2420  return ret; \
2421  av_dict_set(&s->metadata, name, str, AV_DICT_DONT_STRDUP_VAL); \
2422 } while (0)
2423 
2424 #define SET_TS_METADATA(pb, name, var, str) do { \
2425  var = avio_rb64(pb); \
2426  if ((ret = avpriv_dict_set_timestamp(&s->metadata, name, mxf_timestamp_to_int64(var)) < 0)) \
2427  return ret; \
2428 } while (0)
2429 
2430 static int mxf_read_identification_metadata(void *arg, AVIOContext *pb, int tag, int size, UID _uid, int64_t klv_offset)
2431 {
2432  MXFContext *mxf = arg;
2433  AVFormatContext *s = mxf->fc;
2434  int ret;
2435  UID uid = { 0 };
2436  char *str = NULL;
2437  uint64_t ts;
2438  switch (tag) {
2439  case 0x3C01:
2440  SET_STR_METADATA(pb, "company_name", str);
2441  break;
2442  case 0x3C02:
2443  SET_STR_METADATA(pb, "product_name", str);
2444  break;
2445  case 0x3C04:
2446  SET_STR_METADATA(pb, "product_version", str);
2447  break;
2448  case 0x3C05:
2449  SET_UID_METADATA(pb, "product_uid", uid, str);
2450  break;
2451  case 0x3C06:
2452  SET_TS_METADATA(pb, "modification_date", ts, str);
2453  break;
2454  case 0x3C08:
2455  SET_STR_METADATA(pb, "application_platform", str);
2456  break;
2457  case 0x3C09:
2458  SET_UID_METADATA(pb, "generation_uid", uid, str);
2459  break;
2460  case 0x3C0A:
2461  SET_UID_METADATA(pb, "uid", uid, str);
2462  break;
2463  }
2464  return 0;
2465 }
2466 
2467 static int mxf_read_preface_metadata(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
2468 {
2469  MXFContext *mxf = arg;
2470  AVFormatContext *s = mxf->fc;
2471  int ret;
2472  char *str = NULL;
2473 
2474  if (tag >= 0x8000 && (IS_KLV_KEY(uid, mxf_avid_project_name))) {
2475  SET_STR_METADATA(pb, "project_name", str);
2476  }
2477  return 0;
2478 }
2479 
2481  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x05,0x01,0x00 }, mxf_read_primer_pack },
2482  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02,0x01,0x00 }, mxf_read_partition_pack },
2483  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02,0x02,0x00 }, mxf_read_partition_pack },
2484  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02,0x03,0x00 }, mxf_read_partition_pack },
2485  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x02,0x04,0x00 }, mxf_read_partition_pack },
2486  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x03,0x01,0x00 }, mxf_read_partition_pack },
2487  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x03,0x02,0x00 }, mxf_read_partition_pack },
2488  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x03,0x03,0x00 }, mxf_read_partition_pack },
2489  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x03,0x04,0x00 }, mxf_read_partition_pack },
2490  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x04,0x02,0x00 }, mxf_read_partition_pack },
2491  { { 0x06,0x0e,0x2b,0x34,0x02,0x05,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x04,0x04,0x00 }, mxf_read_partition_pack },
2492  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x2f,0x00 }, mxf_read_preface_metadata },
2493  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x30,0x00 }, mxf_read_identification_metadata },
2494  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x18,0x00 }, mxf_read_content_storage, 0, AnyType },
2495  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x37,0x00 }, mxf_read_package, sizeof(MXFPackage), SourcePackage },
2496  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x36,0x00 }, mxf_read_package, sizeof(MXFPackage), MaterialPackage },
2497  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x0f,0x00 }, mxf_read_sequence, sizeof(MXFSequence), Sequence },
2498  { { 0x06,0x0E,0x2B,0x34,0x02,0x53,0x01,0x01,0x0D,0x01,0x01,0x01,0x01,0x01,0x05,0x00 }, mxf_read_essence_group, sizeof(MXFEssenceGroup), EssenceGroup},
2499  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x11,0x00 }, mxf_read_source_clip, sizeof(MXFStructuralComponent), SourceClip },
2500  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x3f,0x00 }, mxf_read_tagged_value, sizeof(MXFTaggedValue), TaggedValue },
2501  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x44,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), MultipleDescriptor },
2502  { { 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 */
2503  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x28,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* CDCI */
2504  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x29,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* RGBA */
2505  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x48,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* Wave */
2506  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x47,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* AES3 */
2507  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x51,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* MPEG2VideoDescriptor */
2508  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x5b,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* VBI - SMPTE 436M */
2509  { { 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 */
2510  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x5e,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* MPEG2AudioDescriptor */
2511  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x64,0x00 }, mxf_read_generic_descriptor, sizeof(MXFDescriptor), Descriptor }, /* DC Timed Text Descriptor */
2512  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x3A,0x00 }, mxf_read_track, sizeof(MXFTrack), Track }, /* Static Track */
2513  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x3B,0x00 }, mxf_read_track, sizeof(MXFTrack), Track }, /* Generic Track */
2514  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x14,0x00 }, mxf_read_timecode_component, sizeof(MXFTimecodeComponent), TimecodeComponent },
2515  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x0c,0x00 }, mxf_read_pulldown_component, sizeof(MXFPulldownComponent), PulldownComponent },
2516  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x04,0x01,0x02,0x02,0x00,0x00 }, mxf_read_cryptographic_context, sizeof(MXFCryptoContext), CryptoContext },
2517  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x10,0x01,0x00 }, mxf_read_index_table_segment, sizeof(MXFIndexTableSegment), IndexTableSegment },
2518  { { 0x06,0x0e,0x2b,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x23,0x00 }, mxf_read_essence_container_data, sizeof(MXFEssenceContainerData), EssenceContainerData },
2519  { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, NULL, 0, AnyType },
2520 };
2521 
2523 {
2524  switch (type){
2525  case MultipleDescriptor:
2526  case Descriptor:
2527  ((MXFDescriptor*)ctx)->pix_fmt = AV_PIX_FMT_NONE;
2528  ((MXFDescriptor*)ctx)->duration = AV_NOPTS_VALUE;
2529  break;
2530  default:
2531  break;
2532  }
2533  return 0;
2534 }
2535 
2536 static int mxf_read_local_tags(MXFContext *mxf, KLVPacket *klv, MXFMetadataReadFunc *read_child, int ctx_size, enum MXFMetadataSetType type)
2537 {
2538  AVIOContext *pb = mxf->fc->pb;
2539  MXFMetadataSet *ctx = ctx_size ? av_mallocz(ctx_size) : mxf;
2540  uint64_t klv_end = avio_tell(pb) + klv->length;
2541 
2542  if (!ctx)
2543  return AVERROR(ENOMEM);
2544  mxf_metadataset_init(ctx, type);
2545  while (avio_tell(pb) + 4 < klv_end && !avio_feof(pb)) {
2546  int ret;
2547  int tag = avio_rb16(pb);
2548  int size = avio_rb16(pb); /* KLV specified by 0x53 */
2549  uint64_t next = avio_tell(pb) + size;
2550  UID uid = {0};
2551 
2552  av_log(mxf->fc, AV_LOG_TRACE, "local tag %#04x size %d\n", tag, size);
2553  if (!size) { /* ignore empty tag, needed for some files with empty UMID tag */
2554  av_log(mxf->fc, AV_LOG_ERROR, "local tag %#04x with 0 size\n", tag);
2555  continue;
2556  }
2557  if (tag > 0x7FFF) { /* dynamic tag */
2558  int i;
2559  for (i = 0; i < mxf->local_tags_count; i++) {
2560  int local_tag = AV_RB16(mxf->local_tags+i*18);
2561  if (local_tag == tag) {
2562  memcpy(uid, mxf->local_tags+i*18+2, 16);
2563  av_log(mxf->fc, AV_LOG_TRACE, "local tag %#04x\n", local_tag);
2564  PRINT_KEY(mxf->fc, "uid", uid);
2565  }
2566  }
2567  }
2568  if (ctx_size && tag == 0x3C0A) {
2569  avio_read(pb, ctx->uid, 16);
2570  } else if ((ret = read_child(ctx, pb, tag, size, uid, -1)) < 0) {
2571  mxf_free_metadataset(&ctx, !!ctx_size);
2572  return ret;
2573  }
2574 
2575  /* Accept the 64k local set limit being exceeded (Avid). Don't accept
2576  * it extending past the end of the KLV though (zzuf5.mxf). */
2577  if (avio_tell(pb) > klv_end) {
2578  if (ctx_size) {
2579  ctx->type = type;
2580  mxf_free_metadataset(&ctx, !!ctx_size);
2581  }
2582 
2583  av_log(mxf->fc, AV_LOG_ERROR,
2584  "local tag %#04x extends past end of local set @ %#"PRIx64"\n",
2585  tag, klv->offset);
2586  return AVERROR_INVALIDDATA;
2587  } else if (avio_tell(pb) <= next) /* only seek forward, else this can loop for a long time */
2588  avio_seek(pb, next, SEEK_SET);
2589  }
2590  if (ctx_size) ctx->type = type;
2591  return ctx_size ? mxf_add_metadata_set(mxf, ctx) : 0;
2592 }
2593 
2594 /**
2595  * Matches any partition pack key, in other words:
2596  * - HeaderPartition
2597  * - BodyPartition
2598  * - FooterPartition
2599  * @return non-zero if the key is a partition pack key, zero otherwise
2600  */
2602 {
2603  //NOTE: this is a little lax since it doesn't constraint key[14]
2604  return !memcmp(key, mxf_header_partition_pack_key, 13) &&
2605  key[13] >= 2 && key[13] <= 4;
2606 }
2607 
2608 /**
2609  * Parses a metadata KLV
2610  * @return <0 on error, 0 otherwise
2611  */
2613  int ctx_size, enum MXFMetadataSetType type)
2614 {
2615  AVFormatContext *s = mxf->fc;
2616  int res;
2617  if (klv.key[5] == 0x53) {
2618  res = mxf_read_local_tags(mxf, &klv, read, ctx_size, type);
2619  } else {
2620  uint64_t next = avio_tell(s->pb) + klv.length;
2621  res = read(mxf, s->pb, 0, klv.length, klv.key, klv.offset);
2622 
2623  /* only seek forward, else this can loop for a long time */
2624  if (avio_tell(s->pb) > next) {
2625  av_log(s, AV_LOG_ERROR, "read past end of KLV @ %#"PRIx64"\n",
2626  klv.offset);
2627  return AVERROR_INVALIDDATA;
2628  }
2629 
2630  avio_seek(s->pb, next, SEEK_SET);
2631  }
2632  if (res < 0) {
2633  av_log(s, AV_LOG_ERROR, "error reading header metadata\n");
2634  return res;
2635  }
2636  return 0;
2637 }
2638 
2639 /**
2640  * Seeks to the previous partition and parses it, if possible
2641  * @return <= 0 if we should stop parsing, > 0 if we should keep going
2642  */
2644 {
2645  AVIOContext *pb = mxf->fc->pb;
2646  KLVPacket klv;
2647  int64_t current_partition_ofs;
2648  int ret;
2649 
2650  if (!mxf->current_partition ||
2652  return 0; /* we've parsed all partitions */
2653 
2654  /* seek to previous partition */
2655  current_partition_ofs = mxf->current_partition->pack_ofs; //includes run-in
2656  avio_seek(pb, mxf->run_in + mxf->current_partition->previous_partition, SEEK_SET);
2657  mxf->current_partition = NULL;
2658 
2659  av_log(mxf->fc, AV_LOG_TRACE, "seeking to previous partition\n");
2660 
2661  /* Make sure this is actually a PartitionPack, and if so parse it.
2662  * See deadlock2.mxf
2663  */
2664  if ((ret = klv_read_packet(&klv, pb)) < 0) {
2665  av_log(mxf->fc, AV_LOG_ERROR, "failed to read PartitionPack KLV\n");
2666  return ret;
2667  }
2668 
2669  if (!mxf_is_partition_pack_key(klv.key)) {
2670  av_log(mxf->fc, AV_LOG_ERROR, "PreviousPartition @ %" PRIx64 " isn't a PartitionPack\n", klv.offset);
2671  return AVERROR_INVALIDDATA;
2672  }
2673 
2674  /* We can't just check ofs >= current_partition_ofs because PreviousPartition
2675  * can point to just before the current partition, causing klv_read_packet()
2676  * to sync back up to it. See deadlock3.mxf
2677  */
2678  if (klv.offset >= current_partition_ofs) {
2679  av_log(mxf->fc, AV_LOG_ERROR, "PreviousPartition for PartitionPack @ %"
2680  PRIx64 " indirectly points to itself\n", current_partition_ofs);
2681  return AVERROR_INVALIDDATA;
2682  }
2683 
2684  if ((ret = mxf_parse_klv(mxf, klv, mxf_read_partition_pack, 0, 0)) < 0)
2685  return ret;
2686 
2687  return 1;
2688 }
2689 
2690 /**
2691  * Called when essence is encountered
2692  * @return <= 0 if we should stop parsing, > 0 if we should keep going
2693  */
2695 {
2696  AVIOContext *pb = mxf->fc->pb;
2697  int64_t ret;
2698 
2699  if (mxf->parsing_backward) {
2700  return mxf_seek_to_previous_partition(mxf);
2701  } else {
2702  if (!mxf->footer_partition) {
2703  av_log(mxf->fc, AV_LOG_TRACE, "no FooterPartition\n");
2704  return 0;
2705  }
2706 
2707  av_log(mxf->fc, AV_LOG_TRACE, "seeking to FooterPartition\n");
2708 
2709  /* remember where we were so we don't end up seeking further back than this */
2710  mxf->last_forward_tell = avio_tell(pb);
2711 
2712  if (!(pb->seekable & AVIO_SEEKABLE_NORMAL)) {
2713  av_log(mxf->fc, AV_LOG_INFO, "file is not seekable - not parsing FooterPartition\n");
2714  return -1;
2715  }
2716 
2717  /* seek to FooterPartition and parse backward */
2718  if ((ret = avio_seek(pb, mxf->run_in + mxf->footer_partition, SEEK_SET)) < 0) {
2719  av_log(mxf->fc, AV_LOG_ERROR,
2720  "failed to seek to FooterPartition @ 0x%" PRIx64
2721  " (%"PRId64") - partial file?\n",
2722  mxf->run_in + mxf->footer_partition, ret);
2723  return ret;
2724  }
2725 
2726  mxf->current_partition = NULL;
2727  mxf->parsing_backward = 1;
2728  }
2729 
2730  return 1;
2731 }
2732 
2733 /**
2734  * Called when the next partition or EOF is encountered
2735  * @return <= 0 if we should stop parsing, > 0 if we should keep going
2736  */
2738 {
2739  return mxf->parsing_backward ? mxf_seek_to_previous_partition(mxf) : 1;
2740 }
2741 
2742 /**
2743  * Figures out the proper offset and length of the essence container in each partition
2744  */
2746 {
2747  int x;
2748 
2749  /* everything is already correct */
2750  if (mxf->op == OPAtom)
2751  return;
2752 
2753  for (x = 0; x < mxf->partitions_count; x++) {
2754  MXFPartition *p = &mxf->partitions[x];
2755 
2756  if (!p->body_sid)
2757  continue; /* BodySID == 0 -> no essence */
2758 
2759  if (x >= mxf->partitions_count - 1)
2760  break; /* FooterPartition - can't compute length (and we don't need to) */
2761 
2762  /* essence container spans to the next partition */
2764 
2765  if (p->essence_length < 0) {
2766  /* next ThisPartition < essence_offset */
2767  p->essence_length = 0;
2768  av_log(mxf->fc, AV_LOG_ERROR,
2769  "partition %i: bad ThisPartition = %"PRIX64"\n",
2770  x+1, mxf->partitions[x+1].this_partition);
2771  }
2772  }
2773 }
2774 
2775 static int64_t round_to_kag(int64_t position, int kag_size)
2776 {
2777  /* TODO: account for run-in? the spec isn't clear whether KAG should account for it */
2778  /* NOTE: kag_size may be any integer between 1 - 2^10 */
2779  int64_t ret = (position / kag_size) * kag_size;
2780  return ret == position ? ret : ret + kag_size;
2781 }
2782 
2783 static int is_pcm(enum AVCodecID codec_id)
2784 {
2785  /* we only care about "normal" PCM codecs until we get samples */
2786  return codec_id >= AV_CODEC_ID_PCM_S16LE && codec_id < AV_CODEC_ID_PCM_S24DAUD;
2787 }
2788 
2790 {
2791  int i;
2792 
2793  if (mxf->op != OPAtom)
2794  return NULL;
2795 
2796  for (i = 0; i < mxf->fc->nb_streams; i++) {
2797  if (mxf->fc->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_DATA)
2798  continue;
2799  return mxf->fc->streams[i];
2800  }
2801  return NULL;
2802 }
2803 
2804 /**
2805  * Deal with the case where for some audio atoms EditUnitByteCount is
2806  * very small (2, 4..). In those cases we should read more than one
2807  * sample per call to mxf_read_packet().
2808  */
2810 {
2811  MXFContext *mxf = s->priv_data;
2812  MXFTrack *track;
2813 
2814  /* assuming non-OPAtom == frame wrapped
2815  * no sane writer would wrap 2 byte PCM packets with 20 byte headers.. */
2816  AVStream *st = mxf_get_opatom_stream(mxf);
2817  if (!st)
2818  return;
2819 
2820  /* expect PCM with exactly one index table segment and a small (< 32) EUBC */
2821  if (st->codecpar->codec_type != AVMEDIA_TYPE_AUDIO ||
2822  !is_pcm(st->codecpar->codec_id) ||
2823  mxf->nb_index_tables != 1 ||
2824  mxf->index_tables[0].nb_segments != 1 ||
2825  mxf->index_tables[0].segments[0]->edit_unit_byte_count >= 32)
2826  return;
2827 
2828  /* arbitrarily default to 48 kHz PAL audio frame size */
2829  /* TODO: We could compute this from the ratio between the audio
2830  * and video edit rates for 48 kHz NTSC we could use the
2831  * 1802-1802-1802-1802-1801 pattern. */
2832  track = st->priv_data;
2833  mxf->edit_units_per_packet = FFMAX(1, track->edit_rate.num / track->edit_rate.den / 25);
2834 }
2835 
2836 /**
2837  * Deal with the case where OPAtom files does not have any IndexTableSegments.
2838  */
2840 {
2841  AVFormatContext *s = mxf->fc;
2842  AVStream *st = NULL;
2844  MXFPartition *p = NULL;
2845  int essence_partition_count = 0;
2846  int i, ret;
2847 
2848  st = mxf_get_opatom_stream(mxf);
2849  if (!st)
2850  return 0;
2851 
2852  /* TODO: support raw video without an index if they exist */
2854  return 0;
2855 
2856  /* check if file already has a IndexTableSegment */
2857  for (i = 0; i < mxf->metadata_sets_count; i++) {
2858  if (mxf->metadata_sets[i]->type == IndexTableSegment)
2859  return 0;
2860  }
2861 
2862  /* find the essence partition */
2863  for (i = 0; i < mxf->partitions_count; i++) {
2864  /* BodySID == 0 -> no essence */
2865  if (!mxf->partitions[i].body_sid)
2866  continue;
2867 
2868  p = &mxf->partitions[i];
2869  essence_partition_count++;
2870  }
2871 
2872  /* only handle files with a single essence partition */
2873  if (essence_partition_count != 1)
2874  return 0;
2875 
2876  if (!(segment = av_mallocz(sizeof(*segment))))
2877  return AVERROR(ENOMEM);
2878 
2879  if ((ret = mxf_add_metadata_set(mxf, segment))) {
2880  mxf_free_metadataset((MXFMetadataSet**)&segment, 1);
2881  return ret;
2882  }
2883 
2884  segment->type = IndexTableSegment;
2885  /* stream will be treated as small EditUnitByteCount */
2887  segment->index_start_position = 0;
2888  segment->index_duration = s->streams[0]->duration;
2889  segment->index_sid = p->index_sid;
2890  segment->body_sid = p->body_sid;
2891  return 0;
2892 }
2893 
2895 {
2896  MXFContext *mxf = s->priv_data;
2897  uint32_t length;
2898  int64_t file_size, max_rip_length, min_rip_length;
2899  KLVPacket klv;
2900 
2901  if (!(s->pb->seekable & AVIO_SEEKABLE_NORMAL))
2902  return;
2903 
2904  file_size = avio_size(s->pb);
2905 
2906  /* S377m says to check the RIP length for "silly" values, without defining "silly".
2907  * The limit below assumes a file with nothing but partition packs and a RIP.
2908  * Before changing this, consider that a muxer may place each sample in its own partition.
2909  *
2910  * 105 is the size of the smallest possible PartitionPack
2911  * 12 is the size of each RIP entry
2912  * 28 is the size of the RIP header and footer, assuming an 8-byte BER
2913  */
2914  max_rip_length = ((file_size - mxf->run_in) / 105) * 12 + 28;
2915  max_rip_length = FFMIN(max_rip_length, INT_MAX); //2 GiB and up is also silly
2916 
2917  /* We're only interested in RIPs with at least two entries.. */
2918  min_rip_length = 16+1+24+4;
2919 
2920  /* See S377m section 11 */
2921  avio_seek(s->pb, file_size - 4, SEEK_SET);
2922  length = avio_rb32(s->pb);
2923 
2924  if (length < min_rip_length || length > max_rip_length)
2925  goto end;
2926  avio_seek(s->pb, file_size - length, SEEK_SET);
2927  if (klv_read_packet(&klv, s->pb) < 0 ||
2929  klv.length != length - 20)
2930  goto end;
2931 
2932  avio_skip(s->pb, klv.length - 12);
2933  mxf->footer_partition = avio_rb64(s->pb);
2934 
2935  /* sanity check */
2936  if (mxf->run_in + mxf->footer_partition >= file_size) {
2937  av_log(s, AV_LOG_WARNING, "bad FooterPartition in RIP - ignoring\n");
2938  mxf->footer_partition = 0;
2939  }
2940 
2941 end:
2942  avio_seek(s->pb, mxf->run_in, SEEK_SET);
2943 }
2944 
2946 {
2947  MXFContext *mxf = s->priv_data;
2948  KLVPacket klv;
2949  int64_t essence_offset = 0;
2950  int ret;
2951 
2952  mxf->last_forward_tell = INT64_MAX;
2953  mxf->edit_units_per_packet = 1;
2954 
2956  av_log(s, AV_LOG_ERROR, "could not find header partition pack key\n");
2957  return AVERROR_INVALIDDATA;
2958  }
2959  avio_seek(s->pb, -14, SEEK_CUR);
2960  mxf->fc = s;
2961  mxf->run_in = avio_tell(s->pb);
2962 
2964 
2965  while (!avio_feof(s->pb)) {
2966  const MXFMetadataReadTableEntry *metadata;
2967 
2968  if (klv_read_packet(&klv, s->pb) < 0) {
2969  /* EOF - seek to previous partition or stop */
2970  if(mxf_parse_handle_partition_or_eof(mxf) <= 0)
2971  break;
2972  else
2973  continue;
2974  }
2975 
2976  PRINT_KEY(s, "read header", klv.key);
2977  av_log(s, AV_LOG_TRACE, "size %"PRIu64" offset %#"PRIx64"\n", klv.length, klv.offset);
2983 
2984  if (!mxf->current_partition) {
2985  av_log(mxf->fc, AV_LOG_ERROR, "found essence prior to first PartitionPack\n");
2986  return AVERROR_INVALIDDATA;
2987  }
2988 
2989  if (!mxf->current_partition->essence_offset) {
2990  /* for OP1a we compute essence_offset
2991  * for OPAtom we point essence_offset after the KL (usually op1a_essence_offset + 20 or 25)
2992  * TODO: for OP1a we could eliminate this entire if statement, always stopping parsing at op1a_essence_offset
2993  * for OPAtom we still need the actual essence_offset though (the KL's length can vary)
2994  */
2995  int64_t op1a_essence_offset =
3000 
3001  if (mxf->op == OPAtom) {
3002  /* point essence_offset to the actual data
3003  * OPAtom has all the essence in one big KLV
3004  */
3007  } else {
3008  /* NOTE: op1a_essence_offset may be less than to klv.offset (C0023S01.mxf) */
3011  else
3012  mxf->current_partition->essence_offset = op1a_essence_offset;
3013  }
3014  }
3015 
3016  if (!essence_offset)
3017  essence_offset = klv.offset;
3018 
3019  /* seek to footer, previous partition or stop */
3020  if (mxf_parse_handle_essence(mxf) <= 0)
3021  break;
3022  continue;
3023  } else if (mxf_is_partition_pack_key(klv.key) && mxf->current_partition) {
3024  /* next partition pack - keep going, seek to previous partition or stop */
3025  if(mxf_parse_handle_partition_or_eof(mxf) <= 0)
3026  break;
3027  else if (mxf->parsing_backward)
3028  continue;
3029  /* we're still parsing forward. proceed to parsing this partition pack */
3030  }
3031 
3032  for (metadata = mxf_metadata_read_table; metadata->read; metadata++) {
3033  if (IS_KLV_KEY(klv.key, metadata->key)) {
3034  if ((ret = mxf_parse_klv(mxf, klv, metadata->read, metadata->ctx_size, metadata->type)) < 0)
3035  goto fail;
3036  break;
3037  }
3038  }
3039  if (!metadata->read) {
3040  av_log(s, AV_LOG_VERBOSE, "Dark key " PRIxUID "\n",
3041  UID_ARG(klv.key));
3042  avio_skip(s->pb, klv.length);
3043  }
3044  }
3045  /* FIXME avoid seek */
3046  if (!essence_offset) {
3047  av_log(s, AV_LOG_ERROR, "no essence\n");
3048  ret = AVERROR_INVALIDDATA;
3049  goto fail;
3050  }
3051  avio_seek(s->pb, essence_offset, SEEK_SET);
3052 
3054 
3055  /* we need to do this before computing the index tables
3056  * to be able to fill in zero IndexDurations with st->duration */
3057  if ((ret = mxf_parse_structural_metadata(mxf)) < 0)
3058  goto fail;
3059 
3061  if ((ret = mxf_compute_index_tables(mxf)) < 0)
3062  goto fail;
3063 
3064  if (mxf->nb_index_tables > 1) {
3065  /* TODO: look up which IndexSID to use via EssenceContainerData */
3066  av_log(mxf->fc, AV_LOG_INFO, "got %i index tables - only the first one (IndexSID %i) will be used\n",
3067  mxf->nb_index_tables, mxf->index_tables[0].index_sid);
3068  } else if (mxf->nb_index_tables == 0 && mxf->op == OPAtom) {
3069  av_log(mxf->fc, AV_LOG_ERROR, "cannot demux OPAtom without an index\n");
3070  ret = AVERROR_INVALIDDATA;
3071  goto fail;
3072  }
3073 
3075 
3076  return 0;
3077 fail:
3078  mxf_read_close(s);
3079 
3080  return ret;
3081 }
3082 
3083 static MXFIndexTable *mxf_find_index_table(MXFContext *mxf, int index_sid)
3084 {
3085  int i;
3086  for (i = 0; i < mxf->nb_index_tables; i++)
3087  if (mxf->index_tables[i].index_sid == index_sid)
3088  return &mxf->index_tables[i];
3089  return NULL;
3090 }
3091 
3092 /* Get the edit unit of the next packet from current_offset in a track. The returned edit unit can be original_duration as well! */
3093 static int mxf_get_next_track_edit_unit(MXFContext *mxf, MXFTrack *track, int64_t current_offset, int64_t *edit_unit_out)
3094 {
3095  int64_t a, b, m, offset;
3096  MXFIndexTable *t = mxf_find_index_table(mxf, track->index_sid);
3097 
3098  if (!t || track->original_duration <= 0)
3099  return -1;
3100 
3101  a = -1;
3102  b = track->original_duration;
3103 
3104  while (b - a > 1) {
3105  m = (a + b) >> 1;
3106  if (mxf_edit_unit_absolute_offset(mxf, t, m, NULL, &offset, 0) < 0)
3107  return -1;
3108  if (offset < current_offset)
3109  a = m;
3110  else
3111  b = m;
3112  }
3113 
3114  *edit_unit_out = b;
3115 
3116  return 0;
3117 }
3118 
3119 /**
3120  * Sets mxf->current_edit_unit based on what offset we're currently at.
3121  * @return next_ofs if OK, <0 on error
3122  */
3123 static int64_t mxf_set_current_edit_unit(MXFContext *mxf, int64_t current_offset)
3124 {
3125  int64_t last_ofs = -1, next_ofs = -1;
3126  MXFIndexTable *t = &mxf->index_tables[0];
3127 
3128  /* this is called from the OP1a demuxing logic, which means there
3129  * may be no index tables */
3130  if (mxf->nb_index_tables <= 0)
3131  return -1;
3132 
3133  /* find mxf->current_edit_unit so that the next edit unit starts ahead of current_offset */
3134  while (mxf->current_edit_unit >= 0) {
3135  if (mxf_edit_unit_absolute_offset(mxf, t, mxf->current_edit_unit + 1, NULL, &next_ofs, 0) < 0)
3136  return -2;
3137 
3138  if (next_ofs <= last_ofs) {
3139  /* large next_ofs didn't change or current_edit_unit wrapped
3140  * around this fixes the infinite loop on zzuf3.mxf */
3141  av_log(mxf->fc, AV_LOG_ERROR,
3142  "next_ofs didn't change. not deriving packet timestamps\n");
3143  return -1;
3144  }
3145 
3146  if (next_ofs > current_offset)
3147  break;
3148 
3149  last_ofs = next_ofs;
3150  mxf->current_edit_unit++;
3151  }
3152 
3153  /* not checking mxf->current_edit_unit >= t->nb_ptses here since CBR files may lack IndexEntryArrays */
3154  if (mxf->current_edit_unit < 0)
3155  return -1;
3156 
3157  return next_ofs;
3158 }
3159 
3160 static int mxf_compute_sample_count(MXFContext *mxf, int stream_index,
3161  uint64_t *sample_count)
3162 {
3163  int i, total = 0, size = 0;
3164  AVStream *st = mxf->fc->streams[stream_index];
3165  MXFTrack *track = st->priv_data;
3166  AVRational time_base = av_inv_q(track->edit_rate);
3168  const MXFSamplesPerFrame *spf = NULL;
3169 
3170  if ((sample_rate.num / sample_rate.den) == 48000)
3171  spf = ff_mxf_get_samples_per_frame(mxf->fc, time_base);
3172  if (!spf) {
3173  int remainder = (sample_rate.num * time_base.num) %
3174  (time_base.den * sample_rate.den);
3175  *sample_count = av_rescale_q(mxf->current_edit_unit, sample_rate, track->edit_rate);
3176  if (remainder)
3177  av_log(mxf->fc, AV_LOG_WARNING,
3178  "seeking detected on stream #%d with time base (%d/%d) and "
3179  "sample rate (%d/%d), audio pts won't be accurate.\n",
3180  stream_index, time_base.num, time_base.den,
3181  sample_rate.num, sample_rate.den);
3182  return 0;
3183  }
3184 
3185  while (spf->samples_per_frame[size]) {
3186  total += spf->samples_per_frame[size];
3187  size++;
3188  }
3189 
3190  av_assert2(size);
3191 
3192  *sample_count = (mxf->current_edit_unit / size) * (uint64_t)total;
3193  for (i = 0; i < mxf->current_edit_unit % size; i++) {
3194  *sample_count += spf->samples_per_frame[i];
3195  }
3196 
3197  return 0;
3198 }
3199 
3201  AVPacket *pkt)
3202 {
3203  MXFTrack *track = mxf->fc->streams[pkt->stream_index]->priv_data;
3204  int64_t bits_per_sample = par->bits_per_coded_sample;
3205 
3206  if (!bits_per_sample)
3207  bits_per_sample = av_get_bits_per_sample(par->codec_id);
3208 
3209  pkt->pts = track->sample_count;
3210 
3211  if ( par->channels <= 0
3212  || bits_per_sample <= 0
3213  || par->channels * (int64_t)bits_per_sample < 8)
3214  return AVERROR(EINVAL);
3215  track->sample_count += pkt->size / (par->channels * (int64_t)bits_per_sample / 8);
3216  return 0;
3217 }
3218 
3219 static int mxf_set_pts(MXFContext *mxf, AVStream *st, AVPacket *pkt, int64_t next_ofs)
3220 {
3221  AVCodecParameters *par = st->codecpar;
3222  MXFTrack *track = st->priv_data;
3223 
3224  if (par->codec_type == AVMEDIA_TYPE_VIDEO && (next_ofs >= 0 || next_ofs == -2 && st->duration == mxf->current_edit_unit + 1)) {
3225  /* mxf->current_edit_unit good - see if we have an
3226  * index table to derive timestamps from */
3227  MXFIndexTable *t = &mxf->index_tables[0];
3228 
3229  if (mxf->nb_index_tables >= 1 && mxf->current_edit_unit < t->nb_ptses) {
3230  pkt->dts = mxf->current_edit_unit + t->first_dts;
3231  pkt->pts = t->ptses[mxf->current_edit_unit];
3232  } else if (track->intra_only) {
3233  /* intra-only -> PTS = EditUnit.
3234  * let utils.c figure out DTS since it can be < PTS if low_delay = 0 (Sony IMX30) */
3235  pkt->pts = mxf->current_edit_unit;
3236  }
3237  } else if (par->codec_type == AVMEDIA_TYPE_AUDIO) {
3238  int ret = mxf_set_audio_pts(mxf, par, pkt);
3239  if (ret < 0)
3240  return ret;
3241  }
3242  return 0;
3243 }
3244 
3246 {
3247  KLVPacket klv;
3248  MXFContext *mxf = s->priv_data;
3249  int ret;
3250 
3251  while ((ret = klv_read_packet(&klv, s->pb)) == 0) {
3252  PRINT_KEY(s, "read packet", klv.key);
3253  av_log(s, AV_LOG_TRACE, "size %"PRIu64" offset %#"PRIx64"\n", klv.length, klv.offset);
3255  ret = mxf_decrypt_triplet(s, pkt, &klv);
3256  if (ret < 0) {
3257  av_log(s, AV_LOG_ERROR, "invalid encoded triplet\n");
3258  return ret;
3259  }
3260  return 0;
3261  }
3265  int body_sid = find_body_sid_by_offset(mxf, klv.offset);
3266  int index = mxf_get_stream_index(s, &klv, body_sid);
3267  int64_t next_ofs, next_klv;
3268  AVStream *st;
3269 
3270  if (index < 0) {
3271  av_log(s, AV_LOG_ERROR,
3272  "error getting stream index %"PRIu32"\n",
3273  AV_RB32(klv.key + 12));
3274  goto skip;
3275  }
3276 
3277  st = s->streams[index];
3278 
3279  if (s->streams[index]->discard == AVDISCARD_ALL)
3280  goto skip;
3281 
3282  next_klv = avio_tell(s->pb) + klv.length;
3283  next_ofs = mxf_set_current_edit_unit(mxf, klv.offset);
3284 
3285  if (next_ofs >= 0 && next_klv > next_ofs) {
3286  /* if this check is hit then it's possible OPAtom was treated as OP1a
3287  * truncate the packet since it's probably very large (>2 GiB is common) */
3289  "OPAtom misinterpreted as OP1a? "
3290  "KLV for edit unit %"PRId64" extending into "
3291  "next edit unit",
3292  mxf->current_edit_unit);
3293  klv.length = next_ofs - avio_tell(s->pb);
3294  }
3295 
3296  /* check for 8 channels AES3 element */
3297  if (klv.key[12] == 0x06 && klv.key[13] == 0x01 && klv.key[14] == 0x10) {
3298  ret = mxf_get_d10_aes3_packet(s->pb, s->streams[index],
3299  pkt, klv.length);
3300  if (ret < 0) {
3301  av_log(s, AV_LOG_ERROR, "error reading D-10 aes3 frame\n");
3302  return ret;
3303  }
3304  } else {
3305  ret = av_get_packet(s->pb, pkt, klv.length);
3306  if (ret < 0)
3307  return ret;
3308  }
3309  pkt->stream_index = index;
3310  pkt->pos = klv.offset;
3311 
3312  ret = mxf_set_pts(mxf, st, pkt, next_ofs);
3313  if (ret < 0)
3314  return ret;
3315 
3316  /* seek for truncated packets */
3317  avio_seek(s->pb, next_klv, SEEK_SET);
3318 
3319  return 0;
3320  } else
3321  skip:
3322  avio_skip(s->pb, klv.length);
3323  }
3324  return avio_feof(s->pb) ? AVERROR_EOF : ret;
3325 }
3326 
3328 {
3329  MXFContext *mxf = s->priv_data;
3330  int ret, size;
3331  int64_t ret64, pos, next_pos;
3332  AVStream *st;
3333  MXFIndexTable *t;
3334  MXFTrack *track;
3335  int edit_units;
3336 
3337  if (mxf->op != OPAtom)
3338  return mxf_read_packet_old(s, pkt);
3339 
3340  // If we have no streams then we basically are at EOF
3341  st = mxf_get_opatom_stream(mxf);
3342  if (!st)
3343  return AVERROR_EOF;
3344 
3345  track = st->priv_data;
3346 
3347  /* OPAtom - clip wrapped demuxing */
3348  /* NOTE: mxf_read_header() makes sure nb_index_tables > 0 for OPAtom */
3349  t = &mxf->index_tables[0];
3350 
3351  if (mxf->current_edit_unit >= track->original_duration)
3352  return AVERROR_EOF;
3353 
3354  edit_units = FFMIN(mxf->edit_units_per_packet, track->original_duration - mxf->current_edit_unit);
3355 
3356  if ((ret = mxf_edit_unit_absolute_offset(mxf, t, mxf->current_edit_unit, NULL, &pos, 1)) < 0)
3357  return ret;
3358 
3359  /* compute size by finding the next edit unit or the end of the essence container
3360  * not pretty, but it works */
3361  if ((ret = mxf_edit_unit_absolute_offset(mxf, t, mxf->current_edit_unit + edit_units, NULL, &next_pos, 0)) < 0 &&
3362  (next_pos = mxf_essence_container_end(mxf, t->body_sid)) <= 0) {
3363  av_log(s, AV_LOG_ERROR, "unable to compute the size of the last packet\n");
3364  return AVERROR_INVALIDDATA;
3365  }
3366 
3367  if ((size = next_pos - pos) <= 0) {
3368  av_log(s, AV_LOG_ERROR, "bad size: %i\n", size);
3369  return AVERROR_INVALIDDATA;
3370  }
3371 
3372  if ((ret64 = avio_seek(s->pb, pos, SEEK_SET)) < 0)
3373  return ret64;
3374 
3375  if ((size = av_get_packet(s->pb, pkt, size)) < 0)
3376  return size;
3377 
3378  pkt->stream_index = st->index;
3379 
3380  ret = mxf_set_pts(mxf, st, pkt, next_pos);
3381  if (ret < 0)
3382  return ret;
3383 
3384  mxf->current_edit_unit += edit_units;
3385 
3386  return 0;
3387 }
3388 
3390 {
3391  MXFContext *mxf = s->priv_data;
3392  int i;
3393 
3394  av_freep(&mxf->packages_refs);
3396 
3397  for (i = 0; i < s->nb_streams; i++)
3398  s->streams[i]->priv_data = NULL;
3399 
3400  for (i = 0; i < mxf->metadata_sets_count; i++) {
3401  mxf_free_metadataset(mxf->metadata_sets + i, 1);
3402  }
3403  av_freep(&mxf->partitions);
3404  av_freep(&mxf->metadata_sets);
3405  av_freep(&mxf->aesc);
3406  av_freep(&mxf->local_tags);
3407 
3408  if (mxf->index_tables) {
3409  for (i = 0; i < mxf->nb_index_tables; i++) {
3410  av_freep(&mxf->index_tables[i].segments);
3411  av_freep(&mxf->index_tables[i].ptses);
3412  av_freep(&mxf->index_tables[i].fake_index);
3413  av_freep(&mxf->index_tables[i].offsets);
3414  }
3415  }
3416  av_freep(&mxf->index_tables);
3417 
3418  return 0;
3419 }
3420 
3421 static int mxf_probe(AVProbeData *p) {
3422  const uint8_t *bufp = p->buf;
3423  const uint8_t *end = p->buf + p->buf_size;
3424 
3425  if (p->buf_size < sizeof(mxf_header_partition_pack_key))
3426  return 0;
3427 
3428  /* Must skip Run-In Sequence and search for MXF header partition pack key SMPTE 377M 5.5 */
3429  end -= sizeof(mxf_header_partition_pack_key);
3430 
3431  for (; bufp < end;) {
3432  if (!((bufp[13] - 1) & 0xF2)){
3433  if (AV_RN32(bufp ) == AV_RN32(mxf_header_partition_pack_key ) &&
3434  AV_RN32(bufp+ 4) == AV_RN32(mxf_header_partition_pack_key+ 4) &&
3435  AV_RN32(bufp+ 8) == AV_RN32(mxf_header_partition_pack_key+ 8) &&
3437  return AVPROBE_SCORE_MAX;
3438  bufp ++;
3439  } else
3440  bufp += 10;
3441  }
3442 
3443  return 0;
3444 }
3445 
3446 /* rudimentary byte seek */
3447 /* XXX: use MXF Index */
3448 static int mxf_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
3449 {
3450  AVStream *st = s->streams[stream_index];
3451  int64_t seconds;
3452  MXFContext* mxf = s->priv_data;
3453  int64_t seekpos;
3454  int i, ret;
3455  MXFIndexTable *t;
3456  MXFTrack *source_track = st->priv_data;
3457 
3458  if (!source_track)
3459  return 0;
3460 
3461  /* if audio then truncate sample_time to EditRate */
3463  sample_time = av_rescale_q(sample_time, st->time_base,
3464  av_inv_q(source_track->edit_rate));
3465 
3466  if (mxf->nb_index_tables <= 0) {
3467  if (!s->bit_rate)
3468  return AVERROR_INVALIDDATA;
3469  if (sample_time < 0)
3470  sample_time = 0;
3471  seconds = av_rescale(sample_time, st->time_base.num, st->time_base.den);
3472 
3473  seekpos = avio_seek(s->pb, (s->bit_rate * seconds) >> 3, SEEK_SET);
3474  if (seekpos < 0)
3475  return seekpos;
3476 
3477  ff_update_cur_dts(s, st, sample_time);
3478  mxf->current_edit_unit = sample_time;
3479  } else {
3480  t = &mxf->index_tables[0];
3481  if (t->index_sid != source_track->index_sid) {
3482  /* If the first index table does not belong to the stream, then find a stream which does belong to the index table */
3483  for (i = 0; i < s->nb_streams; i++) {
3484  MXFTrack *new_source_track = s->streams[i]->priv_data;
3485  if (new_source_track && new_source_track->index_sid == t->index_sid) {
3486  sample_time = av_rescale_q(sample_time, new_source_track->edit_rate, source_track->edit_rate);
3487  source_track = new_source_track;
3488  st = s->streams[i];
3489  break;
3490  }
3491  }
3492  if (i == s->nb_streams)
3493  return AVERROR_INVALIDDATA;
3494  }
3495 
3496  /* clamp above zero, else ff_index_search_timestamp() returns negative
3497  * this also means we allow seeking before the start */
3498  sample_time = FFMAX(sample_time, 0);
3499 
3500  if (t->fake_index) {
3501  /* The first frames may not be keyframes in presentation order, so
3502  * we have to advance the target to be able to find the first
3503  * keyframe backwards... */
3504  if (!(flags & AVSEEK_FLAG_ANY) &&
3505  (flags & AVSEEK_FLAG_BACKWARD) &&
3506  t->ptses[0] != AV_NOPTS_VALUE &&
3507  sample_time < t->ptses[0] &&
3508  (t->fake_index[t->ptses[0]].flags & AVINDEX_KEYFRAME))
3509  sample_time = t->ptses[0];
3510 
3511  /* behave as if we have a proper index */
3512  if ((sample_time = ff_index_search_timestamp(t->fake_index, t->nb_ptses, sample_time, flags)) < 0)
3513  return sample_time;
3514  /* get the stored order index from the display order index */
3515  sample_time += t->offsets[sample_time];
3516  } else {
3517  /* no IndexEntryArray (one or more CBR segments)
3518  * make sure we don't seek past the end */
3519  sample_time = FFMIN(sample_time, source_track->original_duration - 1);
3520  }
3521 
3522  if ((ret = mxf_edit_unit_absolute_offset(mxf, t, sample_time, &sample_time, &seekpos, 1)) < 0)
3523  return ret;
3524 
3525  ff_update_cur_dts(s, st, sample_time);
3526  mxf->current_edit_unit = sample_time;
3527  avio_seek(s->pb, seekpos, SEEK_SET);
3528  }
3529 
3530  // Update all tracks sample count
3531  for (i = 0; i < s->nb_streams; i++) {
3532  AVStream *cur_st = s->streams[i];
3533  MXFTrack *cur_track = cur_st->priv_data;
3534  if (cur_st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
3535  int64_t track_edit_unit;
3536  if (st != cur_st && mxf_get_next_track_edit_unit(mxf, cur_track, seekpos, &track_edit_unit) >= 0) {
3537  cur_track->sample_count = av_rescale_q(track_edit_unit,
3538  av_inv_q(cur_track->edit_rate),
3539  cur_st->time_base);
3540  } else {
3541  uint64_t current_sample_count = 0;
3542  ret = mxf_compute_sample_count(mxf, i, &current_sample_count);
3543  if (ret < 0)
3544  return ret;
3545  cur_track->sample_count = current_sample_count;
3546  }
3547  }
3548  }
3549  return 0;
3550 }
3551 
3553  .name = "mxf",
3554  .long_name = NULL_IF_CONFIG_SMALL("MXF (Material eXchange Format)"),
3555  .flags = AVFMT_SEEK_TO_PTS,
3556  .priv_data_size = sizeof(MXFContext),
3557  .read_probe = mxf_probe,
3562 };
static const uint8_t mxf_crypto_source_container_ul[]
Definition: mxfdec.c:310
static void mxf_compute_essence_containers(MXFContext *mxf)
Figures out the proper offset and length of the essence container in each partition.
Definition: mxfdec.c:2745
time_t av_timegm(struct tm *tm)
Convert the decomposed UTC time in tm to a time_t value.
Definition: parseutils.c:568
packed YUV 4:2:2, 16bpp, Cb Y0 Cr Y1
Definition: pixfmt.h:77
enum MXFMetadataSetType type
Definition: mxfdec.c:156
#define AVSEEK_FLAG_BACKWARD
Definition: avformat.h:2504
#define NULL
Definition: coverity.c:32
full parsing and interpolation of timestamps for frames not starting on a packet boundary ...
Definition: avformat.h:795
static int mxf_parse_structural_metadata(MXFContext *mxf)
Definition: mxfdec.c:1995
enum AVFieldOrder field_order
Video only.
Definition: avcodec.h:3966
MXFMetadataSetType
Definition: mxf.h:30
Bytestream IO Context.
Definition: avio.h:161
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
int64_t avio_size(AVIOContext *s)
Get the filesize.
Definition: aviobuf.c:336
uint8_t origin
Definition: mxfdec.c:121
unsigned int component_depth
Definition: mxfdec.c:189
KLVPacket current_klv_data
Definition: mxfdec.c:271
static int mxf_read_index_table_segment(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:1000
Definition: mxf.h:39
int index_sid
Definition: mxfdec.c:166
AVTimecode tc
Definition: mxfdec.c:130
UID * comment_refs
Definition: mxfdec.c:225
static const MXFMetadataReadTableEntry mxf_metadata_read_table[]
Definition: mxfdec.c:2480
int structural_components_count
Definition: mxfdec.c:143
#define PRINT_KEY(pc, s, x)
Definition: mxf.h:116
int index_sid
Definition: mxfdec.c:245
static int klv_read_packet(KLVPacket *klv, AVIOContext *pb)
Definition: mxfdec.c:393
UID * structural_components_refs
Definition: mxfdec.c:118
enum MXFMetadataSetType type
Definition: mxfdec.c:231
MXFOP
Definition: mxfdec.c:65
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:182
static int mxf_seek_to_previous_partition(MXFContext *mxf)
Seeks to the previous partition and parses it, if possible.
Definition: mxfdec.c:2643
int edit_unit_byte_count
Definition: mxfdec.c:203
enum MXFMetadataSetType type
Definition: mxfdec.c:105
int64_t pos
byte position in stream, -1 if unknown
Definition: avcodec.h:1451
static int mxf_probe(AVProbeData *p)
Definition: mxfdec.c:3421
void av_shrink_packet(AVPacket *pkt, int size)
Reduce packet size, correctly zeroing padding.
Definition: avpacket.c:101
void avpriv_set_pts_info(AVStream *s, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: utils.c:4849
int64_t * ptses
Definition: mxfdec.c:249
#define AVSEEK_FLAG_ANY
seek to any frame, even non-keyframes
Definition: avformat.h:2506
UID * structural_components_refs
Definition: mxfdec.c:142
UID sequence_ref
Definition: mxfdec.c:158
static void mxf_read_random_index_pack(AVFormatContext *s)
Definition: mxfdec.c:2894
static int read_seek(AVFormatContext *ctx, int stream_index, int64_t timestamp, int flags)
Definition: libcdio.c:153
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: avcodec.h:3885
int num
Numerator.
Definition: rational.h:59
int index
stream index in AVFormatContext
Definition: avformat.h:875
#define SET_TS_METADATA(pb, name, var, str)
Definition: mxfdec.c:2424
int size
Definition: avcodec.h:1432
Definition: mxf.h:34
uint64_t footer_partition
Definition: mxfdec.c:270
const char * b
Definition: vf_curves.c:113
int closed
Definition: mxfdec.c:80
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:246
enum MXFMetadataSetType type
Definition: mxfdec.c:296
void av_aes_crypt(AVAES *a, uint8_t *dst, const uint8_t *src, int count, uint8_t *iv, int decrypt)
Encrypt or decrypt a buffer using a previously initialized context.
Definition: aes.c:163
static int mxf_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
Definition: mxfdec.c:3448
static int mxf_uid_to_str(UID uid, char **str)
Definition: mxfdec.c:1698
int64_t bit_rate
Total stream bitrate in bit/s, 0 if not available.
Definition: avformat.h:1473
static int mxf_get_d10_aes3_packet(AVIOContext *pb, AVStream *st, AVPacket *pkt, int64_t length)
Definition: mxfdec.c:446
MXFSequence * sequence
Definition: mxfdec.c:157
#define tc
Definition: regdef.h:69
int linked_track_id
Definition: mxfdec.c:194
UID key
Definition: mxf.h:62
static int mxf_umid_to_str(UID ul, UID uid, char **str)
Definition: mxfdec.c:1716
int64_t offset
Definition: mxf.h:63
void * priv_data
Definition: avformat.h:889
int64_t avio_skip(AVIOContext *s, int64_t offset)
Skip given number of bytes forward.
Definition: aviobuf.c:331
const char * key
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
Definition: mem.c:236
char * name
Definition: mxfdec.c:150
discard all
Definition: avcodec.h:795
static AVPacket pkt
#define MXF_FIELD_DOMINANCE_DEFAULT
Definition: mxfdec.c:182
Definition: mxfdec.c:72
int ff_index_search_timestamp(const AVIndexEntry *entries, int nb_entries, int64_t wanted_timestamp, int flags)
Internal version of av_index_search_timestamp.
Definition: utils.c:2047
unsigned int avio_rb16(AVIOContext *s)
Definition: aviobuf.c:786
static int mxf_is_intra_only(MXFDescriptor *descriptor)
Definition: mxfdec.c:1690
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_WB32 unsigned int_TMPL AV_WB24 unsigned int_TMPL AV_RB16
Definition: bytestream.h:87
#define sample
static int mxf_read_header(AVFormatContext *s)
Definition: mxfdec.c:2945
UID source_container_ul
Definition: mxfdec.c:100
int avio_get_str16le(AVIOContext *pb, int maxlen, char *buf, int buflen)
Read a UTF-16 string from pb and convert it to UTF-8.
int id
Definition: mxf.h:70
This struct describes the properties of an encoded stream.
Definition: avcodec.h:3877
static int find_body_sid_by_offset(MXFContext *mxf, int64_t offset)
Definition: mxfdec.c:422
static int mxf_read_essence_container_data(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:951
static int mxf_parse_klv(MXFContext *mxf, KLVPacket klv, MXFMetadataReadFunc *read, int ctx_size, enum MXFMetadataSetType type)
Parses a metadata KLV.
Definition: mxfdec.c:2612
void * av_calloc(size_t nmemb, size_t size)
Non-inlined equivalent of av_mallocz_array().
Definition: mem.c:244
enum MXFMetadataSetType type
Definition: mxfdec.c:172
#define AV_DICT_DONT_STRDUP_KEY
Take ownership of a key that's been allocated with av_malloc() or another memory allocation function...
Definition: dict.h:73
Definition: mxfdec.c:71
static int mxf_parse_handle_partition_or_eof(MXFContext *mxf)
Called when the next partition or EOF is encountered.
Definition: mxfdec.c:2737
static int mxf_read_local_tags(MXFContext *mxf, KLVPacket *klv, MXFMetadataReadFunc *read_child, int ctx_size, enum MXFMetadataSetType type)
Definition: mxfdec.c:2536
static int mxf_read_packet(AVFormatContext *s, AVPacket *pkt)
Definition: mxfdec.c:3327
Format I/O context.
Definition: avformat.h:1351
Definition: mxfdec.c:69
static int is_pcm(enum AVCodecID codec_id)
Definition: mxfdec.c:2783
uint8_t UID[16]
Definition: mxf.h:28
int frame_layout
Definition: mxfdec.c:180
UID uid
Definition: mxfenc.c:2116
void ff_update_cur_dts(AVFormatContext *s, AVStream *ref_st, int64_t timestamp)
Update cur_dts of all streams based on the given timestamp and AVStream.
Definition: utils.c:1952
static int mxf_handle_missing_index_segment(MXFContext *mxf)
Deal with the case where OPAtom files does not have any IndexTableSegments.
Definition: mxfdec.c:2839
static const uint8_t mxf_indirect_value_utf16le[]
Definition: mxfdec.c:317
Definition: mxfdec.c:62
void void avpriv_request_sample(void *avc, const char *msg,...) av_printf_format(2
Log a generic warning message about a missing feature.
UID * packages_refs
Definition: mxfdec.c:260
#define MXF_FIELD_DOMINANCE_FF
Definition: mxfdec.c:183
uint8_t
UID * tracks_refs
Definition: mxfdec.c:220
#define av_malloc(s)
int64_t first_dts
Definition: mxfdec.c:248
const MXFSamplesPerFrame * ff_mxf_get_samples_per_frame(AVFormatContext *s, AVRational time_base)
Definition: mxf.c:154
Opaque data information usually continuous.
Definition: avutil.h:203
static const uint8_t mxf_avid_essence_element_key[]
Definition: mxfdec.c:304
int bits_per_sample
Definition: mxfdec.c:187
int width
Video only.
Definition: avcodec.h:3951
#define av_assert2(cond)
assert() equivalent, that does lie in speed critical code.
Definition: avassert.h:64
Definition: mxfdec.c:286
#define AV_LOG_TRACE
Extremely verbose debugging, useful for libav* development.
Definition: log.h:202
unsigned int avio_rb32(AVIOContext *s)
Definition: aviobuf.c:801
timecode is drop frame
Definition: timecode.h:36
static av_cold int end(AVCodecContext *avctx)
Definition: avrndec.c:90
AVRational index_edit_rate
Definition: mxfdec.c:206
enum AVStreamParseType need_parsing
Definition: avformat.h:1092
int id
Format-specific stream ID.
Definition: avformat.h:881
static int mxf_read_close(AVFormatContext *s)
Definition: mxfdec.c:3389
Definition: mxf.h:61
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
Definition: utils.c:4456
static int mxf_add_umid_metadata(AVDictionary **pm, const char *key, MXFPackage *package)
Definition: mxfdec.c:1737
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_RB32
Definition: bytestream.h:87
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1419
#define UID_ARG(x)
Definition: mxf.h:93
static int mxf_absolute_bodysid_offset(MXFContext *mxf, int body_sid, int64_t offset, int64_t *offset_out)
Computes the absolute file offset of the given essence container offset.
Definition: mxfdec.c:1361
int complete
Definition: mxfdec.c:81
static const uint8_t mxf_klv_key[]
Definition: mxfdec.c:308
int MXFMetadataReadFunc(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:290
static int mxf_compute_index_tables(MXFContext *mxf)
Sorts and collects index table segments into index tables.
Definition: mxfdec.c:1589
MXFIndexTableSegment ** segments
Definition: mxfdec.c:251
const MXFCodecUL ff_mxf_codec_uls[]
Definition: mxf.c:36
uint8_t * data
Definition: avcodec.h:1431
static const uint8_t mxf_sony_mpeg4_extradata[]
Definition: mxfdec.c:314
uint32_t tag
Definition: movenc.c:1466
Definition: mxfdec.c:285
#define AVERROR_EOF
End of file.
Definition: error.h:55
static const uint8_t mxf_avid_project_name[]
Definition: mxfdec.c:315
static av_cold int read_close(AVFormatContext *ctx)
Definition: libcdio.c:145
#define AV_LOG_VERBOSE
Detailed information.
Definition: log.h:192
int av_get_packet(AVIOContext *s, AVPacket *pkt, int size)
Allocate and read the payload of a packet and initialize its fields with default values.
Definition: utils.c:310
Definition: mxfdec.c:67
Definition: ismindex.c:69
ptrdiff_t size
Definition: opengl_enc.c:101
static int mxf_read_utf16_string(AVIOContext *pb, int size, char **str, int be)
Definition: mxfdec.c:768
uint64_t avio_rb64(AVIOContext *s)
Definition: aviobuf.c:922
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:557
#define flags(name, subs,...)
Definition: cbs_h2645.c:263
static const MXFCodecUL mxf_intra_only_essence_container_uls[]
Definition: mxfdec.c:1249
unsigned int vert_subsampling
Definition: mxfdec.c:191
void * av_realloc_array(void *ptr, size_t nmemb, size_t size)
Definition: mem.c:198
int intra_only
Definition: mxfdec.c:163
#define MXF_FIELD_DOMINANCE_FL
Definition: mxfdec.c:184
#define av_log(a,...)
int32_t kag_size
Definition: mxfdec.c:89
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:648
static const uint8_t mxf_random_index_pack_key[]
Definition: mxfdec.c:313
static int mxf_get_sorted_table_segments(MXFContext *mxf, int *nb_sorted_segments, MXFIndexTableSegment ***sorted_segments)
Definition: mxfdec.c:1287
static MXFDescriptor * mxf_resolve_multidescriptor(MXFContext *mxf, MXFDescriptor *descriptor, int track_id)
Definition: mxfdec.c:1794
static AVStream * mxf_get_opatom_stream(MXFContext *mxf)
Definition: mxfdec.c:2789
uint8_t track_number[4]
Definition: mxfdec.c:161
int64_t original_duration
Definition: mxfdec.c:165
int64_t av_rescale_q(int64_t a, AVRational bq, AVRational cq)
Rescale a 64-bit integer by 2 rational numbers.
Definition: mathematics.c:142
int av_timecode_init(AVTimecode *tc, AVRational rate, int flags, int frame_start, void *log_ctx)
Init a timecode struct with the passed parameters.
Definition: timecode.c:184
#define SET_UID_METADATA(pb, name, var, str)
Definition: mxfdec.c:2417
static int mxf_read_sync(AVIOContext *pb, const uint8_t *key, unsigned size)
Definition: mxfdec.c:380
#define AVINDEX_KEYFRAME
Definition: avformat.h:810
int metadata_sets_count
Definition: mxfdec.c:265
UID essence_container_ul
Definition: mxfdec.c:173
static int mxf_get_stream_index(AVFormatContext *s, KLVPacket *klv, int body_sid)
Definition: mxfdec.c:408
AVCodecID
Identify the syntax and semantics of the bitstream.
Definition: avcodec.h:215
int64_t start_position
Definition: mxfdec.c:110
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
int last_forward_partition
Definition: mxfdec.c:277
int av_get_bits_per_sample(enum AVCodecID codec_id)
Return codec bits per sample.
Definition: utils.c:1495
AVDictionary * metadata
Metadata that applies to the whole file.
Definition: avformat.h:1591
static int mxf_read_timecode_component(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:841
MXFPartitionType
Definition: mxfdec.c:59
static void mxf_read_pixel_layout(AVIOContext *pb, MXFDescriptor *descriptor)
Definition: mxfdec.c:1037
static MXFTimecodeComponent * mxf_resolve_timecode_component(MXFContext *mxf, UID *strong_ref)
Definition: mxfdec.c:1757
UID essence_codec_ul
Definition: mxfdec.c:174
int8_t * temporal_offset_entries
Definition: mxfdec.c:209
static int mxf_add_timecode_metadata(AVDictionary **pm, const char *key, AVTimecode *tc)
Definition: mxfdec.c:1749
#define AVERROR(e)
Definition: error.h:43
static int mxf_read_sequence(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:893
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)
Definition: mxfdec.c:1423
MXFDescriptor * descriptor
Definition: mxfdec.c:222
int64_t timestamp
Timestamp in AVStream.time_base units, preferably the time from which on correctly decoded frames are...
Definition: avformat.h:804
uint64_t index_start_position
Definition: mxfdec.c:207
int nb_ptses
Definition: mxfdec.c:247
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:186
static int mxf_read_index_entry_array(AVIOContext *pb, MXFIndexTableSegment *segment)
Definition: mxfdec.c:970
unsigned matching_len
Definition: mxf.h:69
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:197
static const MXFCodecUL mxf_intra_only_picture_essence_coding_uls[]
Definition: mxfdec.c:1255
Definition: mxf.h:56
enum AVMediaType codec_type
General type of the encoded data.
Definition: avcodec.h:3881
const char * arg
Definition: jacosubdec.c:66
uint64_t length
Definition: mxf.h:64
static const MXFCodecUL mxf_sound_essence_container_uls[]
Definition: mxfdec.c:1270
simple assert() macros that are a bit more flexible than ISO C assert().
enum AVMediaType avcodec_get_type(enum AVCodecID codec_id)
Get the type of the given codec.
Definition: codec_desc.c:3168
GLsizei GLsizei * length
Definition: opengl_enc.c:115
static int mxf_parse_handle_essence(MXFContext *mxf)
Called when essence is encountered.
Definition: mxfdec.c:2694
#define IS_KLV_KEY(x, y)
Definition: mxfdec.c:320
struct AVAES * av_aes_alloc(void)
Allocate an AVAES context.
Definition: aes.c:31
static int64_t round_to_kag(int64_t position, int kag_size)
Definition: mxfdec.c:2775
int track_id
Definition: mxfdec.c:159
static const uint8_t offset[127][2]
Definition: vf_spp.c:92
GLsizei count
Definition: opengl_enc.c:109
UID package_uid
Definition: mxfdec.c:218
#define FFMAX(a, b)
Definition: common.h:94
int64_t essence_length
Definition: mxfdec.c:88
#define fail()
Definition: checkasm.h:117
static int mxf_read_package(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:927
static int mxf_read_source_clip(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:819
int64_t duration
Definition: mxfdec.c:120
int video_line_map[2]
Definition: mxfdec.c:181
static int mxf_match_uid(const UID key, const UID uid, int len)
Definition: mxfdec.c:1195
int packages_count
Definition: mxfdec.c:261
Only parse headers, do not repack.
Definition: avformat.h:794
int avio_r8(AVIOContext *s)
Definition: aviobuf.c:639
static MXFStructuralComponent * mxf_resolve_sourceclip(MXFContext *mxf, UID *strong_ref)
Definition: mxfdec.c:1847
int64_t index_byte_count
Definition: mxfdec.c:91
int buf_size
Size of buf except extra allocated bytes.
Definition: avformat.h:451
int ff_mxf_decode_pixel_layout(const char pixel_layout[16], enum AVPixelFormat *pix_fmt)
Definition: mxf.c:119
const MXFCodecUL ff_mxf_pixel_format_uls[]
Definition: mxf.c:77
unsigned char * buf
Buffer must have AVPROBE_PADDING_SIZE of extra allocated bytes filled with zero.
Definition: avformat.h:450
Definition: hls.c:68
unsigned int nb_streams
Number of elements in AVFormatContext.streams.
Definition: avformat.h:1407
int nb_segments
Definition: mxfdec.c:250
int ff_generate_avci_extradata(AVStream *st)
Generate standard extradata for AVC-Intra based on width/height and field order.
Definition: utils.c:5307
static const MXFCodecUL * mxf_get_codec_ul(const MXFCodecUL *uls, UID *uid)
Definition: mxfdec.c:1205
static const uint8_t mxf_jp2k_rsiz[]
Definition: mxfdec.c:316
MXFPartition * partitions
Definition: mxfdec.c:257
int8_t * offsets
Definition: mxfdec.c:253
enum MXFMetadataSetType type
Definition: mxfdec.c:202
static const uint8_t mxf_essence_element_key[]
Definition: mxfdec.c:303
static void * mxf_resolve_strong_ref(MXFContext *mxf, UID *strong_ref, enum MXFMetadataSetType type)
Definition: mxfdec.c:1215
int seekable
A combination of AVIO_SEEKABLE_ flags or 0 when the stream is not seekable.
Definition: avio.h:260
int local_tags_count
Definition: mxfdec.c:269
int ff_alloc_extradata(AVCodecParameters *par, int size)
Allocate extradata with additional AV_INPUT_BUFFER_PADDING_SIZE at end which is always set to 0...
Definition: utils.c:3286
unsigned int horiz_subsampling
Definition: mxfdec.c:190
static const uint8_t mxf_system_item_key_cp[]
Definition: mxfdec.c:306
#define PRIxUID
Definition: mxf.h:87
int64_t av_rescale(int64_t a, int64_t b, int64_t c)
Rescale a 64-bit integer with rounding to nearest.
Definition: mathematics.c:129
static MXFStructuralComponent * mxf_resolve_essence_group_choice(MXFContext *mxf, MXFEssenceGroup *essence_group)
Definition: mxfdec.c:1820
static const uint8_t mxf_encrypted_triplet_key[]
Definition: mxfdec.c:311
#define FFMIN(a, b)
Definition: common.h:96
static int mxf_get_next_track_edit_unit(MXFContext *mxf, MXFTrack *track, int64_t current_offset, int64_t *edit_unit_out)
Definition: mxfdec.c:3093
#define AV_DICT_DONT_STRDUP_VAL
Take ownership of a value that's been allocated with av_malloc() or another memory allocation functio...
Definition: dict.h:76
enum AVPixelFormat pix_fmt
Definition: mxfdec.c:197
static int read_probe(AVProbeData *pd)
Definition: jvdec.c:55
UID uid
Definition: mxfdec.c:115
int64_t essence_offset
absolute offset of essence
Definition: mxfdec.c:87
struct AVRational rate
Definition: mxfdec.c:129
static int mxf_parse_package_comments(MXFContext *mxf, AVDictionary **pm, MXFPackage *package)
Definition: mxfdec.c:1865
int edit_units_per_packet
how many edit units to read at a time (PCM, OPAtom)
Definition: mxfdec.c:281
GLsizei GLboolean const GLfloat * value
Definition: opengl_enc.c:109
static MXFPackage * mxf_resolve_source_package(MXFContext *mxf, UID package_ul, UID package_uid)
Definition: mxfdec.c:1778
int64_t last_forward_tell
Definition: mxfdec.c:276
static const uint8_t mxf_header_partition_pack_key[]
Definition: mxfdec.c:302
int32_t
AVFormatContext * ctx
Definition: movenc.c:48
static int mxf_read_primer_pack(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:543
static const MXFCodecUL mxf_intra_only_picture_coded_width[]
Definition: mxfdec.c:1262
Definition: mxfdec.c:70
#define s(width, name)
Definition: cbs_vp9.c:257
int n
Definition: avisynth_c.h:684
AVDictionary * metadata
Definition: avformat.h:938
#define FF_PROFILE_JPEG2000_DCINEMA_4K
Definition: avcodec.h:2925
int64_t this_partition
Definition: mxfdec.c:86
static int mxf_decrypt_triplet(AVFormatContext *s, AVPacket *pkt, KLVPacket *klv)
Definition: mxfdec.c:474
enum AVCodecID codec_id
Definition: vaapi_decode.c:362
static int mxf_read_identification_metadata(void *arg, AVIOContext *pb, int tag, int size, UID _uid, int64_t klv_offset)
Definition: mxfdec.c:2430
static void mxf_handle_small_eubc(AVFormatContext *s)
Deal with the case where for some audio atoms EditUnitByteCount is very small (2, 4...
Definition: mxfdec.c:2809
const MXFCodecUL ff_mxf_data_definition_uls[]
SMPTE RP224 http://www.smpte-ra.org/mdd/index.html.
Definition: mxf.c:28
UID * sub_descriptors_refs
Definition: mxfdec.c:192
static const uint8_t mxf_system_item_key_gc[]
Definition: mxfdec.c:307
static MXFIndexTable * mxf_find_index_table(MXFContext *mxf, int index_sid)
Definition: mxfdec.c:3083
static int mxf_read_essence_group(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:913
preferred ID for MPEG-1/2 video decoding
Definition: avcodec.h:220
const char * avcodec_get_name(enum AVCodecID id)
Get the name of a codec.
Definition: utils.c:1127
static int read_header(FFV1Context *f)
Definition: ffv1dec.c:530
uint64_t index_duration
Definition: mxfdec.c:208
Stream structure.
Definition: avformat.h:874
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
Definition: error.h:62
static int mxf_read_strong_ref_array(AVIOContext *pb, UID **refs, int *count)
Definition: mxfdec.c:755
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
Definition: avio_reading.c:42
int av_aes_init(AVAES *a, const uint8_t *key, int key_bits, int decrypt)
Initialize an AVAES context.
Definition: aes.c:195
#define AVIO_SEEKABLE_NORMAL
Seeking works like for a local file.
Definition: avio.h:40
static int mxf_read_tagged_value(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:1179
int64_t current_edit_unit
Definition: mxfdec.c:278
sample_rate
#define AV_LOG_INFO
Standard information.
Definition: log.h:187
UID * essence_container_data_refs
Definition: mxfdec.c:262
Timecode helpers header.
int sub_descriptors_count
Definition: mxfdec.c:193
int nb_index_tables
Definition: mxfdec.c:279
AVIOContext * pb
I/O context.
Definition: avformat.h:1393
static int mxf_read_generic_descriptor(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:1057
static const uint8_t mxf_encrypted_essence_container[]
Definition: mxfdec.c:312
static int mxf_metadataset_init(MXFMetadataSet *ctx, enum MXFMetadataSetType type)
Definition: mxfdec.c:2522
Definition: mxfdec.c:66
int samples_per_frame[6]
Definition: mxf.h:76
#define AV_RN16(p)
Definition: intreadwrite.h:360
enum MXFMetadataSetType type
Definition: mxfdec.c:116
void * buf
Definition: avisynth_c.h:690
int run_in
Definition: mxfdec.c:273
GLint GLenum type
Definition: opengl_enc.c:105
int av_dict_set(AVDictionary **pm, const char *key, const char *value, int flags)
Set the given entry in *pm, overwriting an existing entry.
Definition: dict.c:70
int comment_count
Definition: mxfdec.c:226
static const MXFCodecUL mxf_data_essence_container_uls[]
Definition: mxfdec.c:1280
#define AV_PIX_FMT_XYZ12
Definition: pixfmt.h:414
AVRational aspect_ratio
Definition: mxfdec.c:177
int index
Definition: gxfenc.c:89
Definition: mxf.h:54
Rational number (pair of numerator and denominator).
Definition: rational.h:58
static int mxf_read_partition_pack(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:569
MXFPartitionType type
Definition: mxfdec.c:82
static const uint8_t mxf_canopus_essence_element_key[]
Definition: mxfdec.c:305
AVRational display_aspect_ratio
display aspect ratio (0 if unknown)
Definition: avformat.h:1225
static int mxf_set_pts(MXFContext *mxf, AVStream *st, AVPacket *pkt, int64_t next_ofs)
Definition: mxfdec.c:3219
struct AVAES * aesc
Definition: mxfdec.c:267
AVMediaType
Definition: avutil.h:199
AVFormatContext * fc
Definition: mxfdec.c:266
#define AVFMT_SEEK_TO_PTS
Seeking is based on PTS.
Definition: avformat.h:500
uint8_t * extradata
Definition: mxfdec.c:195
UID package_ul
Definition: mxfdec.c:219
unsigned partitions_count
Definition: mxfdec.c:258
enum MXFMetadataSetType type
Definition: mxfdec.c:99
AVRational sample_rate
Definition: mxfdec.c:176
#define snprintf
Definition: snprintf.h:34
This structure contains the data a format has to probe a file.
Definition: avformat.h:448
Definition: mxf.h:67
#define AV_RN32(p)
Definition: intreadwrite.h:364
misc parsing utilities
int essence_container_data_count
Definition: mxfdec.c:263
uint8_t * local_tags
Definition: mxfdec.c:268
int channels
Definition: mxfdec.c:186
#define READ_STR16(type, big_endian)
Definition: mxfdec.c:794
const char * av_get_media_type_string(enum AVMediaType media_type)
Return a string describing the media_type enum, NULL if media_type is unknown.
Definition: utils.c:76
static const uint8_t mxf_indirect_value_utf16be[]
Definition: mxfdec.c:318
static int64_t klv_decode_ber_length(AVIOContext *pb)
Definition: mxfdec.c:363
MXFWrappingScheme
Definition: mxfdec.c:284
int64_t duration
Definition: mxfdec.c:144
char * av_timecode_make_string(const AVTimecode *tc, char *buf, int framenum)
Load timecode string in buf.
Definition: timecode.c:84
char * name
Definition: mxfdec.c:160
static int mxf_read_pulldown_component(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:858
int64_t duration
Decoding: duration of the stream, in stream time base.
Definition: avformat.h:923
static av_always_inline AVRational av_inv_q(AVRational q)
Invert a rational.
Definition: rational.h:159
int body_sid
Definition: mxfdec.c:167
int sample_rate
Audio only.
Definition: avcodec.h:3995
Definition: mxfdec.c:73
static int mxf_parse_physical_source_package(MXFContext *mxf, MXFTrack *source_track, AVStream *st)
Definition: mxfdec.c:1887
#define AVPROBE_SCORE_MAX
maximum score
Definition: avformat.h:460
const uint8_t * key
Definition: avformat.h:1527
int parsing_backward
Definition: mxfdec.c:275
static int op(uint8_t **dst, const uint8_t *dst_end, GetByteContext *gb, int pixel, int count, int *x, int width, int linesize)
Perform decode operation.
Definition: anm.c:78
full parsing and repack
Definition: avformat.h:793
Main libavformat public API header.
static int mxf_read_content_storage(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:803
int tracks_count
Definition: mxfdec.c:221
Definition: mxf.h:37
int body_sid
Definition: mxfdec.c:246
if(ret< 0)
Definition: vf_mcdeint.c:279
const char * desc
Definition: mxf.h:71
static const MXFCodecUL mxf_picture_essence_container_uls[]
Definition: mxfdec.c:1230
int avio_get_str16be(AVIOContext *pb, int maxlen, char *buf, int buflen)
int height
Definition: mxfdec.c:179
UID codec_ul
Definition: mxfdec.c:175
Definition: mxfdec.c:75
int64_t start_time
Decoding: pts of the first frame of the stream in presentation order, in stream time base...
Definition: avformat.h:913
MXFMetadataReadFunc * read
Definition: mxfdec.c:294
int av_dict_set_int(AVDictionary **pm, const char *key, int64_t value, int flags)
Convenience wrapper for av_dict_set that converts the value to a string and stores it...
Definition: dict.c:147
Definition: mxfdec.c:68
Definition: mxfdec.c:74
static int mxf_add_metadata_stream(MXFContext *mxf, MXFTrack *track)
Definition: mxfdec.c:1957
int field_dominance
Definition: mxfdec.c:185
static void mxf_free_metadataset(MXFMetadataSet **ctx, int freectx)
Definition: mxfdec.c:322
AVInputFormat ff_mxf_demuxer
Definition: mxfdec.c:3552
UID uid
Definition: mxfdec.c:216
int den
Denominator.
Definition: rational.h:60
AVIndexEntry * fake_index
Definition: mxfdec.c:252
int64_t pack_ofs
absolute offset of pack in file, including run-in
Definition: mxfdec.c:93
int structural_components_count
Definition: mxfdec.c:119
UID data_definition_ul
Definition: mxfdec.c:117
static int64_t mxf_timestamp_to_int64(uint64_t timestamp)
Definition: mxfdec.c:2389
static int mxf_read_cryptographic_context(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:745
enum MXFMetadataSetType type
Definition: mxfdec.c:240
int body_sid
Definition: mxfdec.c:85
static int mxf_compute_ptses_fake_index(MXFContext *mxf, MXFIndexTable *index_table)
Definition: mxfdec.c:1471
#define av_free(p)
static int mxf_set_audio_pts(MXFContext *mxf, AVCodecParameters *par, AVPacket *pkt)
Definition: mxfdec.c:3200
int current_klv_index
Definition: mxfdec.c:272
const MXFCodecUL ff_mxf_codec_tag_uls[]
Definition: mxf.c:83
int len
uint64_t * stream_offset_entries
Definition: mxfdec.c:211
static int mxf_is_partition_pack_key(UID key)
Matches any partition pack key, in other words:
Definition: mxfdec.c:2601
static int mxf_read_preface_metadata(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:2467
static int mxf_add_metadata_set(MXFContext *mxf, void *metadata_set)
Definition: mxfdec.c:732
MXFOP op
Definition: mxfdec.c:259
void * priv_data
Format private data.
Definition: avformat.h:1379
static int64_t mxf_essence_container_end(MXFContext *mxf, int body_sid)
Returns the end position of the essence container with given BodySID, or zero if unknown.
Definition: mxfdec.c:1402
#define FF_PROFILE_JPEG2000_DCINEMA_2K
Definition: avcodec.h:2924
int extradata_size
Definition: mxfdec.c:196
UID uid
Definition: mxfdec.c:155
static int mxf_read_indirect_value(void *arg, AVIOContext *pb, int size)
Definition: mxfdec.c:1161
uint64_t layout
int bits_per_coded_sample
The number of bits per sample in the codedwords.
Definition: avcodec.h:3927
uint8_t * extradata
Extra binary data needed for initializing the decoder, codec-dependent.
Definition: avcodec.h:3899
static int mxf_read_track(void *arg, AVIOContext *pb, int tag, int size, UID uid, int64_t klv_offset)
Definition: mxfdec.c:869
#define SET_STR_METADATA(pb, name, str)
Definition: mxfdec.c:2411
int channels
Audio only.
Definition: avcodec.h:3991
int64_t dts
Decompression timestamp in AVStream->time_base units; the time at which the packet is decompressed...
Definition: avcodec.h:1430
static int64_t mxf_set_current_edit_unit(MXFContext *mxf, int64_t current_offset)
Sets mxf->current_edit_unit based on what offset we're currently at.
Definition: mxfdec.c:3123
int64_t duration
Definition: mxfdec.c:188
MXFMetadataSet ** metadata_sets
Definition: mxfdec.c:264
static int mxf_read_packet_old(AVFormatContext *s, AVPacket *pkt)
Definition: mxfdec.c:3245
static int mxf_compute_sample_count(MXFContext *mxf, int stream_index, uint64_t *sample_count)
Definition: mxfdec.c:3160
#define av_freep(p)
const char * name
A comma separated list of short names for the format.
Definition: avformat.h:647
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1021
int avio_feof(AVIOContext *s)
feof() equivalent for AVIOContext.
Definition: aviobuf.c:358
uint32_t codec_tag
Additional information about the codec (corresponds to the AVI FOURCC).
Definition: avcodec.h:3889
MXFIndexTable * index_tables
Definition: mxfdec.c:280
UID uid
Definition: mxf.h:68
int pack_length
Definition: mxfdec.c:92
int stream_index
Definition: avcodec.h:1433
AVRational time_base
This is the fundamental unit of time (in seconds) in terms of which frame timestamps are represented...
Definition: avformat.h:903
int64_t body_offset
Definition: mxfdec.c:94
AVRational edit_rate
Definition: mxfdec.c:162
enum AVDiscard discard
Selects which packets can be discarded at will and do not need to be demuxed.
Definition: avformat.h:929
char * name
Definition: mxfdec.c:224
AVPixelFormat
Pixel format.
Definition: pixfmt.h:60
This structure stores compressed data.
Definition: avcodec.h:1408
int index_sid
Definition: mxfdec.c:84
Definition: mxf.h:31
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: avcodec.h:1424
enum MXFMetadataSetType type
Definition: mxfdec.c:217
for(j=16;j >0;--j)
Definition: mxfdec.c:60
uint64_t sample_count
Definition: mxfdec.c:164
MXFPartition * current_partition
Definition: mxfdec.c:274
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:248
char * value
Definition: mxfdec.c:151
uint64_t previous_partition
Definition: mxfdec.c:83
#define AV_TIMECODE_STR_SIZE
Definition: timecode.h:33
void * av_mallocz_array(size_t nmemb, size_t size)
Definition: mem.c:191
const char * name
Definition: opengl_enc.c:103
int64_t header_byte_count
Definition: mxfdec.c:90
static uint8_t tmp[11]
Definition: aes_ctr.c:26
UID descriptor_ref
Definition: mxfdec.c:223