1/* 2 * MXF muxer 3 * Copyright (c) 2008 GUCAS, Zhentan Feng <spyfeng at gmail dot com> 4 * Copyright (c) 2008 Baptiste Coudurier <baptiste dot coudurier at gmail dot com> 5 * 6 * This file is part of FFmpeg. 7 * 8 * FFmpeg is free software; you can redistribute it and/or 9 * modify it under the terms of the GNU Lesser General Public 10 * License as published by the Free Software Foundation; either 11 * version 2.1 of the License, or (at your option) any later version. 12 * 13 * FFmpeg is distributed in the hope that it will be useful, 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 16 * Lesser General Public License for more details. 17 * 18 * You should have received a copy of the GNU Lesser General Public 19 * License along with FFmpeg; if not, write to the Free Software 20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 21 */ 22 23/* 24 * References 25 * SMPTE 336M KLV Data Encoding Protocol Using Key-Length-Value 26 * SMPTE 377M MXF File Format Specifications 27 * SMPTE 379M MXF Generic Container 28 * SMPTE 381M Mapping MPEG Streams into the MXF Generic Container 29 * SMPTE RP210: SMPTE Metadata Dictionary 30 * SMPTE RP224: Registry of SMPTE Universal Labels 31 */ 32 33//#define DEBUG 34 35#include <math.h> 36#include <time.h> 37 38#include "libavutil/random_seed.h" 39#include "libavcodec/bytestream.h" 40#include "audiointerleave.h" 41#include "avformat.h" 42#include "mxf.h" 43 44static const int NTSC_samples_per_frame[] = { 1602, 1601, 1602, 1601, 1602, 0 }; 45static const int PAL_samples_per_frame[] = { 1920, 0 }; 46 47AVOutputFormat mxf_d10_muxer; 48 49#define EDIT_UNITS_PER_BODY 250 50#define KAG_SIZE 512 51 52typedef struct { 53 int local_tag; 54 UID uid; 55} MXFLocalTagPair; 56 57typedef struct { 58 uint8_t flags; 59 uint64_t offset; 60 unsigned slice_offset; ///< offset of audio slice 61} MXFIndexEntry; 62 63typedef struct { 64 AudioInterleaveContext aic; 65 UID track_essence_element_key; 66 int index; ///< index in mxf_essence_container_uls table 67 const UID *codec_ul; 68 int order; ///< interleaving order if dts are equal 69 int interlaced; ///< wether picture is interlaced 70 int temporal_reordering; 71 AVRational aspect_ratio; ///< display aspect ratio 72 int closed_gop; ///< gop is closed, used in mpeg-2 frame parsing 73} MXFStreamContext; 74 75typedef struct { 76 UID container_ul; 77 UID element_ul; 78 UID codec_ul; 79 void (*write_desc)(); 80} MXFContainerEssenceEntry; 81 82static const struct { 83 enum CodecID id; 84 int index; 85} mxf_essence_mappings[] = { 86 { CODEC_ID_MPEG2VIDEO, 0 }, 87 { CODEC_ID_PCM_S24LE, 1 }, 88 { CODEC_ID_PCM_S16LE, 1 }, 89 { CODEC_ID_NONE } 90}; 91 92static void mxf_write_wav_desc(AVFormatContext *s, AVStream *st); 93static void mxf_write_aes3_desc(AVFormatContext *s, AVStream *st); 94static void mxf_write_mpegvideo_desc(AVFormatContext *s, AVStream *st); 95static void mxf_write_cdci_desc(AVFormatContext *s, AVStream *st); 96static void mxf_write_generic_sound_desc(AVFormatContext *s, AVStream *st); 97 98static const MXFContainerEssenceEntry mxf_essence_container_uls[] = { 99 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x02,0x0D,0x01,0x03,0x01,0x02,0x04,0x60,0x01 }, 100 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x15,0x01,0x05,0x00 }, 101 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x03,0x04,0x01,0x02,0x02,0x01,0x00,0x00,0x00 }, 102 mxf_write_mpegvideo_desc }, 103 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x06,0x03,0x00 }, 104 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x16,0x01,0x03,0x00 }, 105 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x02,0x02,0x01,0x00,0x00,0x00,0x00 }, 106 mxf_write_aes3_desc }, 107 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x06,0x01,0x00 }, 108 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x16,0x01,0x01,0x00 }, 109 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x02,0x02,0x01,0x00,0x00,0x00,0x00 }, 110 mxf_write_wav_desc }, 111 // D-10 625/50 PAL 50mb/s 112 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x01,0x01 }, 113 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x05,0x01,0x01,0x00 }, 114 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x01,0x02,0x02,0x01,0x02,0x01,0x01 }, 115 mxf_write_cdci_desc }, 116 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x01,0x01 }, 117 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x06,0x01,0x10,0x00 }, 118 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x02,0x02,0x01,0x00,0x00,0x00,0x00 }, 119 mxf_write_generic_sound_desc }, 120 // D-10 525/60 NTSC 50mb/s 121 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x02,0x01 }, 122 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x05,0x01,0x01,0x00 }, 123 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x01,0x02,0x02,0x01,0x02,0x01,0x02 }, 124 mxf_write_cdci_desc }, 125 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x02,0x01 }, 126 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x06,0x01,0x10,0x00 }, 127 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x02,0x02,0x01,0x00,0x00,0x00,0x00 }, 128 mxf_write_generic_sound_desc }, 129 // D-10 625/50 PAL 40mb/s 130 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x03,0x01 }, 131 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x05,0x01,0x01,0x00 }, 132 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x01,0x02,0x02,0x01,0x02,0x01,0x03 }, 133 mxf_write_cdci_desc }, 134 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x03,0x01 }, 135 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x06,0x01,0x10,0x00 }, 136 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x02,0x02,0x01,0x00,0x00,0x00,0x00 }, 137 mxf_write_generic_sound_desc }, 138 // D-10 525/60 NTSC 40mb/s 139 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x04,0x01 }, 140 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x05,0x01,0x01,0x00 }, 141 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x01,0x02,0x02,0x01,0x02,0x01,0x04 }, 142 mxf_write_cdci_desc }, 143 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x04,0x01 }, 144 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x06,0x01,0x10,0x00 }, 145 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x02,0x02,0x01,0x00,0x00,0x00,0x00 }, 146 mxf_write_generic_sound_desc }, 147 // D-10 625/50 PAL 30mb/s 148 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x05,0x01 }, 149 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x05,0x01,0x01,0x00 }, 150 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x01,0x02,0x02,0x01,0x02,0x01,0x05 }, 151 mxf_write_cdci_desc }, 152 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x05,0x01 }, 153 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x06,0x01,0x10,0x00 }, 154 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x02,0x02,0x01,0x00,0x00,0x00,0x00 }, 155 mxf_write_generic_sound_desc }, 156 // D-10 525/60 NTSC 30mb/s 157 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x06,0x01 }, 158 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x05,0x01,0x01,0x00 }, 159 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x01,0x02,0x02,0x01,0x02,0x01,0x06 }, 160 mxf_write_cdci_desc }, 161 { { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x03,0x01,0x02,0x01,0x06,0x01 }, 162 { 0x06,0x0E,0x2B,0x34,0x01,0x02,0x01,0x01,0x0D,0x01,0x03,0x01,0x06,0x01,0x10,0x00 }, 163 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x04,0x02,0x02,0x01,0x00,0x00,0x00,0x00 }, 164 mxf_write_generic_sound_desc }, 165 { { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 166 { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 167 { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 }, 168 NULL }, 169}; 170 171typedef struct MXFContext { 172 int64_t footer_partition_offset; 173 int essence_container_count; 174 AVRational time_base; 175 int header_written; 176 MXFIndexEntry *index_entries; 177 unsigned edit_units_count; 178 uint64_t timestamp; ///< timestamp, as year(16),month(8),day(8),hour(8),minutes(8),msec/4(8) 179 uint8_t slice_count; ///< index slice count minus 1 (1 if no audio, 0 otherwise) 180 int last_indexed_edit_unit; 181 uint64_t *body_partition_offset; 182 unsigned body_partitions_count; 183 int last_key_index; ///< index of last key frame 184 uint64_t duration; 185 AVStream *timecode_track; 186 int timecode_base; ///< rounded time code base (25 or 30) 187 int timecode_start; ///< frame number computed from mpeg-2 gop header timecode 188 int timecode_drop_frame; ///< time code use drop frame method frop mpeg-2 essence gop header 189 int edit_unit_byte_count; ///< fixed edit unit byte count 190 uint64_t body_offset; 191 uint32_t instance_number; 192 uint8_t umid[16]; ///< unique material identifier 193} MXFContext; 194 195static const uint8_t uuid_base[] = { 0xAD,0xAB,0x44,0x24,0x2f,0x25,0x4d,0xc7,0x92,0xff,0x29,0xbd }; 196static const uint8_t umid_ul[] = { 0x06,0x0A,0x2B,0x34,0x01,0x01,0x01,0x05,0x01,0x01,0x0D,0x00,0x13 }; 197 198/** 199 * complete key for operation pattern, partitions, and primer pack 200 */ 201static const uint8_t op1a_ul[] = { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x0D,0x01,0x02,0x01,0x01,0x01,0x09,0x00 }; 202static const uint8_t footer_partition_key[] = { 0x06,0x0E,0x2B,0x34,0x02,0x05,0x01,0x01,0x0D,0x01,0x02,0x01,0x01,0x04,0x04,0x00 }; // ClosedComplete 203static const uint8_t primer_pack_key[] = { 0x06,0x0E,0x2B,0x34,0x02,0x05,0x01,0x01,0x0D,0x01,0x02,0x01,0x01,0x05,0x01,0x00 }; 204static const uint8_t index_table_segment_key[] = { 0x06,0x0E,0x2B,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x02,0x01,0x01,0x10,0x01,0x00 }; 205static const uint8_t random_index_pack_key[] = { 0x06,0x0E,0x2B,0x34,0x02,0x05,0x01,0x01,0x0D,0x01,0x02,0x01,0x01,0x11,0x01,0x00 }; 206static const uint8_t header_open_partition_key[] = { 0x06,0x0E,0x2B,0x34,0x02,0x05,0x01,0x01,0x0D,0x01,0x02,0x01,0x01,0x02,0x01,0x00 }; // OpenIncomplete 207static const uint8_t header_closed_partition_key[] = { 0x06,0x0E,0x2B,0x34,0x02,0x05,0x01,0x01,0x0D,0x01,0x02,0x01,0x01,0x02,0x04,0x00 }; // ClosedComplete 208static const uint8_t klv_fill_key[] = { 0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x03,0x01,0x02,0x10,0x01,0x00,0x00,0x00 }; 209static const uint8_t body_partition_key[] = { 0x06,0x0E,0x2B,0x34,0x02,0x05,0x01,0x01,0x0D,0x01,0x02,0x01,0x01,0x03,0x04,0x00 }; // ClosedComplete 210 211/** 212 * partial key for header metadata 213 */ 214static const uint8_t header_metadata_key[] = { 0x06,0x0E,0x2B,0x34,0x02,0x53,0x01,0x01,0x0D,0x01,0x01,0x01,0x01 }; 215static const uint8_t multiple_desc_ul[] = { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x03,0x0D,0x01,0x03,0x01,0x02,0x7F,0x01,0x00 }; 216 217/** 218 * SMPTE RP210 http://www.smpte-ra.org/mdd/index.html 219 */ 220static const MXFLocalTagPair mxf_local_tag_batch[] = { 221 // preface set 222 { 0x3C0A, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x01,0x01,0x15,0x02,0x00,0x00,0x00,0x00}}, /* Instance UID */ 223 { 0x3B02, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x07,0x02,0x01,0x10,0x02,0x04,0x00,0x00}}, /* Last Modified Date */ 224 { 0x3B05, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x03,0x01,0x02,0x01,0x05,0x00,0x00,0x00}}, /* Version */ 225 { 0x3B06, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x04,0x06,0x04,0x00,0x00}}, /* Identifications reference */ 226 { 0x3B03, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x04,0x02,0x01,0x00,0x00}}, /* Content Storage reference */ 227 { 0x3B09, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x01,0x02,0x02,0x03,0x00,0x00,0x00,0x00}}, /* Operational Pattern UL */ 228 { 0x3B0A, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x01,0x02,0x02,0x10,0x02,0x01,0x00,0x00}}, /* Essence Containers UL batch */ 229 { 0x3B0B, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x01,0x02,0x02,0x10,0x02,0x02,0x00,0x00}}, /* DM Schemes UL batch */ 230 // Identification 231 { 0x3C09, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x05,0x20,0x07,0x01,0x01,0x00,0x00,0x00}}, /* This Generation UID */ 232 { 0x3C01, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x05,0x20,0x07,0x01,0x02,0x01,0x00,0x00}}, /* Company Name */ 233 { 0x3C02, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x05,0x20,0x07,0x01,0x03,0x01,0x00,0x00}}, /* Product Name */ 234 { 0x3C04, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x05,0x20,0x07,0x01,0x05,0x01,0x00,0x00}}, /* Version String */ 235 { 0x3C05, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x05,0x20,0x07,0x01,0x07,0x00,0x00,0x00}}, /* Product ID */ 236 { 0x3C06, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x07,0x02,0x01,0x10,0x02,0x03,0x00,0x00}}, /* Modification Date */ 237 // Content Storage 238 { 0x1901, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x04,0x05,0x01,0x00,0x00}}, /* Package strong reference batch */ 239 { 0x1902, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x04,0x05,0x02,0x00,0x00}}, /* Package strong reference batch */ 240 // Essence Container Data 241 { 0x2701, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x06,0x01,0x00,0x00,0x00}}, /* Linked Package UID */ 242 { 0x3F07, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x04,0x01,0x03,0x04,0x04,0x00,0x00,0x00,0x00}}, /* BodySID */ 243 // Package 244 { 0x4401, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x01,0x01,0x15,0x10,0x00,0x00,0x00,0x00}}, /* Package UID */ 245 { 0x4405, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x07,0x02,0x01,0x10,0x01,0x03,0x00,0x00}}, /* Package Creation Date */ 246 { 0x4404, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x07,0x02,0x01,0x10,0x02,0x05,0x00,0x00}}, /* Package Modified Date */ 247 { 0x4403, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x04,0x06,0x05,0x00,0x00}}, /* Tracks Strong reference array */ 248 { 0x4701, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x04,0x02,0x03,0x00,0x00}}, /* Descriptor */ 249 // Track 250 { 0x4801, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x01,0x07,0x01,0x01,0x00,0x00,0x00,0x00}}, /* Track ID */ 251 { 0x4804, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x01,0x04,0x01,0x03,0x00,0x00,0x00,0x00}}, /* Track Number */ 252 { 0x4B01, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x05,0x30,0x04,0x05,0x00,0x00,0x00,0x00}}, /* Edit Rate */ 253 { 0x4B02, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x07,0x02,0x01,0x03,0x01,0x03,0x00,0x00}}, /* Origin */ 254 { 0x4803, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x04,0x02,0x04,0x00,0x00}}, /* Sequence reference */ 255 // Sequence 256 { 0x0201, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x04,0x07,0x01,0x00,0x00,0x00,0x00,0x00}}, /* Data Definition UL */ 257 { 0x0202, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x07,0x02,0x02,0x01,0x01,0x03,0x00,0x00}}, /* Duration */ 258 { 0x1001, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x04,0x06,0x09,0x00,0x00}}, /* Structural Components reference array */ 259 // Source Clip 260 { 0x1201, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x07,0x02,0x01,0x03,0x01,0x04,0x00,0x00}}, /* Start position */ 261 { 0x1101, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x03,0x01,0x00,0x00,0x00}}, /* SourcePackageID */ 262 { 0x1102, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x03,0x02,0x00,0x00,0x00}}, /* SourceTrackID */ 263 // Timecode Component 264 { 0x1501, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x07,0x02,0x01,0x03,0x01,0x05,0x00,0x00}}, /* Start Time Code */ 265 { 0x1502, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x04,0x04,0x01,0x01,0x02,0x06,0x00,0x00}}, /* Rounded Time Code Base */ 266 { 0x1503, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x04,0x04,0x01,0x01,0x05,0x00,0x00,0x00}}, /* Drop Frame */ 267 // File Descriptor 268 { 0x3F01, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x04,0x06,0x01,0x01,0x04,0x06,0x0B,0x00,0x00}}, /* Sub Descriptors reference array */ 269 { 0x3006, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x06,0x01,0x01,0x03,0x05,0x00,0x00,0x00}}, /* Linked Track ID */ 270 { 0x3001, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x04,0x06,0x01,0x01,0x00,0x00,0x00,0x00}}, /* SampleRate */ 271 { 0x3004, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x06,0x01,0x01,0x04,0x01,0x02,0x00,0x00}}, /* Essence Container */ 272 // Generic Picture Essence Descriptor 273 { 0x320C, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x04,0x01,0x03,0x01,0x04,0x00,0x00,0x00}}, /* Frame Layout */ 274 { 0x320D, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x04,0x01,0x03,0x02,0x05,0x00,0x00,0x00}}, /* Video Line Map */ 275 { 0x3203, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x04,0x01,0x05,0x02,0x02,0x00,0x00,0x00}}, /* Stored Width */ 276 { 0x3202, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x04,0x01,0x05,0x02,0x01,0x00,0x00,0x00}}, /* Stored Height */ 277 { 0x3209, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x04,0x01,0x05,0x01,0x0C,0x00,0x00,0x00}}, /* Display Width */ 278 { 0x3208, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x04,0x01,0x05,0x01,0x0B,0x00,0x00,0x00}}, /* Display Height */ 279 { 0x320E, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x04,0x01,0x01,0x01,0x01,0x00,0x00,0x00}}, /* Aspect Ratio */ 280 { 0x3201, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x04,0x01,0x06,0x01,0x00,0x00,0x00,0x00}}, /* Picture Essence Coding */ 281 // CDCI Picture Essence Descriptor 282 { 0x3301, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x04,0x01,0x05,0x03,0x0A,0x00,0x00,0x00}}, /* Component Depth */ 283 { 0x3302, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x01,0x04,0x01,0x05,0x01,0x05,0x00,0x00,0x00}}, /* Horizontal Subsampling */ 284 // Generic Sound Essence Descriptor 285 { 0x3D02, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x04,0x04,0x02,0x03,0x01,0x04,0x00,0x00,0x00}}, /* Locked/Unlocked */ 286 { 0x3D03, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x04,0x02,0x03,0x01,0x01,0x01,0x00,0x00}}, /* Audio sampling rate */ 287 { 0x3D07, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x04,0x02,0x01,0x01,0x04,0x00,0x00,0x00}}, /* ChannelCount */ 288 { 0x3D01, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x04,0x04,0x02,0x03,0x03,0x04,0x00,0x00,0x00}}, /* Quantization bits */ 289 { 0x3D06, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x02,0x04,0x02,0x04,0x02,0x00,0x00,0x00,0x00}}, /* Sound Essence Compression */ 290 // Index Table Segment 291 { 0x3F0B, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x05,0x30,0x04,0x06,0x00,0x00,0x00,0x00}}, /* Index Edit Rate */ 292 { 0x3F0C, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x07,0x02,0x01,0x03,0x01,0x0A,0x00,0x00}}, /* Index Start Position */ 293 { 0x3F0D, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x07,0x02,0x02,0x01,0x01,0x02,0x00,0x00}}, /* Index Duration */ 294 { 0x3F05, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x04,0x04,0x06,0x02,0x01,0x00,0x00,0x00,0x00}}, /* Edit Unit Byte Count */ 295 { 0x3F06, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x04,0x01,0x03,0x04,0x05,0x00,0x00,0x00,0x00}}, /* IndexSID */ 296 { 0x3F08, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x04,0x04,0x04,0x04,0x01,0x01,0x00,0x00,0x00}}, /* Slice Count */ 297 { 0x3F09, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x04,0x04,0x04,0x01,0x06,0x00,0x00,0x00}}, /* Delta Entry Array */ 298 { 0x3F0A, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x04,0x04,0x04,0x02,0x05,0x00,0x00,0x00}}, /* Index Entry Array */ 299 // MPEG video Descriptor 300 { 0x8000, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x04,0x01,0x06,0x02,0x01,0x0B,0x00,0x00}}, /* BitRate */ 301 { 0x8007, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x04,0x01,0x06,0x02,0x01,0x0A,0x00,0x00}}, /* ProfileAndLevel */ 302 // Wave Audio Essence Descriptor 303 { 0x3D09, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x04,0x02,0x03,0x03,0x05,0x00,0x00,0x00}}, /* Average Bytes Per Second */ 304 { 0x3D0A, {0x06,0x0E,0x2B,0x34,0x01,0x01,0x01,0x05,0x04,0x02,0x03,0x02,0x01,0x00,0x00,0x00}}, /* Block Align */ 305}; 306 307static void mxf_write_uuid(ByteIOContext *pb, enum MXFMetadataSetType type, int value) 308{ 309 put_buffer(pb, uuid_base, 12); 310 put_be16(pb, type); 311 put_be16(pb, value); 312} 313 314static void mxf_write_umid(AVFormatContext *s, int type) 315{ 316 MXFContext *mxf = s->priv_data; 317 put_buffer(s->pb, umid_ul, 13); 318 put_be24(s->pb, mxf->instance_number); 319 put_buffer(s->pb, mxf->umid, 15); 320 put_byte(s->pb, type); 321} 322 323static void mxf_write_refs_count(ByteIOContext *pb, int ref_count) 324{ 325 put_be32(pb, ref_count); 326 put_be32(pb, 16); 327} 328 329static int klv_ber_length(uint64_t len) 330{ 331 if (len < 128) 332 return 1; 333 else 334 return (av_log2(len) >> 3) + 2; 335} 336 337static int klv_encode_ber_length(ByteIOContext *pb, uint64_t len) 338{ 339 // Determine the best BER size 340 int size; 341 if (len < 128) { 342 //short form 343 put_byte(pb, len); 344 return 1; 345 } 346 347 size = (av_log2(len) >> 3) + 1; 348 349 // long form 350 put_byte(pb, 0x80 + size); 351 while(size) { 352 size--; 353 put_byte(pb, len >> 8 * size & 0xff); 354 } 355 return 0; 356} 357 358static void klv_encode_ber4_length(ByteIOContext *pb, int len) 359{ 360 put_byte(pb, 0x80 + 3); 361 put_be24(pb, len); 362} 363 364/* 365 * Get essence container ul index 366 */ 367static int mxf_get_essence_container_ul_index(enum CodecID id) 368{ 369 int i; 370 for (i = 0; mxf_essence_mappings[i].id; i++) 371 if (mxf_essence_mappings[i].id == id) 372 return mxf_essence_mappings[i].index; 373 return -1; 374} 375 376static void mxf_write_primer_pack(AVFormatContext *s) 377{ 378 ByteIOContext *pb = s->pb; 379 int local_tag_number, i = 0; 380 381 local_tag_number = FF_ARRAY_ELEMS(mxf_local_tag_batch); 382 383 put_buffer(pb, primer_pack_key, 16); 384 klv_encode_ber_length(pb, local_tag_number * 18 + 8); 385 386 put_be32(pb, local_tag_number); // local_tag num 387 put_be32(pb, 18); // item size, always 18 according to the specs 388 389 for (i = 0; i < local_tag_number; i++) { 390 put_be16(pb, mxf_local_tag_batch[i].local_tag); 391 put_buffer(pb, mxf_local_tag_batch[i].uid, 16); 392 } 393} 394 395static void mxf_write_local_tag(ByteIOContext *pb, int size, int tag) 396{ 397 put_be16(pb, tag); 398 put_be16(pb, size); 399} 400 401static void mxf_write_metadata_key(ByteIOContext *pb, unsigned int value) 402{ 403 put_buffer(pb, header_metadata_key, 13); 404 put_be24(pb, value); 405} 406 407static void mxf_free(AVFormatContext *s) 408{ 409 int i; 410 411 for (i = 0; i < s->nb_streams; i++) { 412 AVStream *st = s->streams[i]; 413 av_freep(&st->priv_data); 414 } 415} 416 417static const MXFCodecUL *mxf_get_data_definition_ul(int type) 418{ 419 const MXFCodecUL *uls = ff_mxf_data_definition_uls; 420 while (uls->uid[0]) { 421 if (type == uls->id) 422 break; 423 uls++; 424 } 425 return uls; 426} 427 428static void mxf_write_essence_container_refs(AVFormatContext *s) 429{ 430 MXFContext *c = s->priv_data; 431 ByteIOContext *pb = s->pb; 432 int i; 433 434 mxf_write_refs_count(pb, c->essence_container_count); 435 av_log(s,AV_LOG_DEBUG, "essence container count:%d\n", c->essence_container_count); 436 for (i = 0; i < c->essence_container_count; i++) { 437 MXFStreamContext *sc = s->streams[i]->priv_data; 438 put_buffer(pb, mxf_essence_container_uls[sc->index].container_ul, 16); 439 } 440} 441 442static void mxf_write_preface(AVFormatContext *s) 443{ 444 MXFContext *mxf = s->priv_data; 445 ByteIOContext *pb = s->pb; 446 447 mxf_write_metadata_key(pb, 0x012f00); 448 PRINT_KEY(s, "preface key", pb->buf_ptr - 16); 449 klv_encode_ber_length(pb, 130 + 16 * mxf->essence_container_count); 450 451 // write preface set uid 452 mxf_write_local_tag(pb, 16, 0x3C0A); 453 mxf_write_uuid(pb, Preface, 0); 454 PRINT_KEY(s, "preface uid", pb->buf_ptr - 16); 455 456 // last modified date 457 mxf_write_local_tag(pb, 8, 0x3B02); 458 put_be64(pb, mxf->timestamp); 459 460 // write version 461 mxf_write_local_tag(pb, 2, 0x3B05); 462 put_be16(pb, 258); // v1.2 463 464 // write identification_refs 465 mxf_write_local_tag(pb, 16 + 8, 0x3B06); 466 mxf_write_refs_count(pb, 1); 467 mxf_write_uuid(pb, Identification, 0); 468 469 // write content_storage_refs 470 mxf_write_local_tag(pb, 16, 0x3B03); 471 mxf_write_uuid(pb, ContentStorage, 0); 472 473 // operational pattern 474 mxf_write_local_tag(pb, 16, 0x3B09); 475 put_buffer(pb, op1a_ul, 16); 476 477 // write essence_container_refs 478 mxf_write_local_tag(pb, 8 + 16 * mxf->essence_container_count, 0x3B0A); 479 mxf_write_essence_container_refs(s); 480 481 // write dm_scheme_refs 482 mxf_write_local_tag(pb, 8, 0x3B0B); 483 put_be64(pb, 0); 484} 485 486/* 487 * Write a local tag containing an ascii string as utf-16 488 */ 489static void mxf_write_local_tag_utf16(ByteIOContext *pb, int tag, const char *value) 490{ 491 int i, size = strlen(value); 492 mxf_write_local_tag(pb, size*2, tag); 493 for (i = 0; i < size; i++) 494 put_be16(pb, value[i]); 495} 496 497static void mxf_write_identification(AVFormatContext *s) 498{ 499 MXFContext *mxf = s->priv_data; 500 ByteIOContext *pb = s->pb; 501 const char *company = "FFmpeg"; 502 const char *product = "OP1a Muxer"; 503 const char *version; 504 int length; 505 506 mxf_write_metadata_key(pb, 0x013000); 507 PRINT_KEY(s, "identification key", pb->buf_ptr - 16); 508 509 version = s->streams[0]->codec->flags & CODEC_FLAG_BITEXACT ? 510 "0.0.0" : AV_STRINGIFY(LIBAVFORMAT_VERSION); 511 length = 84 + (strlen(company)+strlen(product)+strlen(version))*2; // utf-16 512 klv_encode_ber_length(pb, length); 513 514 // write uid 515 mxf_write_local_tag(pb, 16, 0x3C0A); 516 mxf_write_uuid(pb, Identification, 0); 517 PRINT_KEY(s, "identification uid", pb->buf_ptr - 16); 518 519 // write generation uid 520 mxf_write_local_tag(pb, 16, 0x3C09); 521 mxf_write_uuid(pb, Identification, 1); 522 523 mxf_write_local_tag_utf16(pb, 0x3C01, company); // Company Name 524 mxf_write_local_tag_utf16(pb, 0x3C02, product); // Product Name 525 mxf_write_local_tag_utf16(pb, 0x3C04, version); // Version String 526 527 // write product uid 528 mxf_write_local_tag(pb, 16, 0x3C05); 529 mxf_write_uuid(pb, Identification, 2); 530 531 // modification date 532 mxf_write_local_tag(pb, 8, 0x3C06); 533 put_be64(pb, mxf->timestamp); 534} 535 536static void mxf_write_content_storage(AVFormatContext *s) 537{ 538 ByteIOContext *pb = s->pb; 539 540 mxf_write_metadata_key(pb, 0x011800); 541 PRINT_KEY(s, "content storage key", pb->buf_ptr - 16); 542 klv_encode_ber_length(pb, 92); 543 544 // write uid 545 mxf_write_local_tag(pb, 16, 0x3C0A); 546 mxf_write_uuid(pb, ContentStorage, 0); 547 PRINT_KEY(s, "content storage uid", pb->buf_ptr - 16); 548 549 // write package reference 550 mxf_write_local_tag(pb, 16 * 2 + 8, 0x1901); 551 mxf_write_refs_count(pb, 2); 552 mxf_write_uuid(pb, MaterialPackage, 0); 553 mxf_write_uuid(pb, SourcePackage, 0); 554 555 // write essence container data 556 mxf_write_local_tag(pb, 8 + 16, 0x1902); 557 mxf_write_refs_count(pb, 1); 558 mxf_write_uuid(pb, EssenceContainerData, 0); 559} 560 561static void mxf_write_track(AVFormatContext *s, AVStream *st, enum MXFMetadataSetType type) 562{ 563 MXFContext *mxf = s->priv_data; 564 ByteIOContext *pb = s->pb; 565 MXFStreamContext *sc = st->priv_data; 566 567 mxf_write_metadata_key(pb, 0x013b00); 568 PRINT_KEY(s, "track key", pb->buf_ptr - 16); 569 klv_encode_ber_length(pb, 80); 570 571 // write track uid 572 mxf_write_local_tag(pb, 16, 0x3C0A); 573 mxf_write_uuid(pb, type == MaterialPackage ? Track : Track + TypeBottom, st->index); 574 PRINT_KEY(s, "track uid", pb->buf_ptr - 16); 575 576 // write track id 577 mxf_write_local_tag(pb, 4, 0x4801); 578 put_be32(pb, st->index+2); 579 580 // write track number 581 mxf_write_local_tag(pb, 4, 0x4804); 582 if (type == MaterialPackage) 583 put_be32(pb, 0); // track number of material package is 0 584 else 585 put_buffer(pb, sc->track_essence_element_key + 12, 4); 586 587 mxf_write_local_tag(pb, 8, 0x4B01); 588 put_be32(pb, mxf->time_base.den); 589 put_be32(pb, mxf->time_base.num); 590 591 // write origin 592 mxf_write_local_tag(pb, 8, 0x4B02); 593 put_be64(pb, 0); 594 595 // write sequence refs 596 mxf_write_local_tag(pb, 16, 0x4803); 597 mxf_write_uuid(pb, type == MaterialPackage ? Sequence: Sequence + TypeBottom, st->index); 598} 599 600static const uint8_t smpte_12m_timecode_track_data_ul[] = { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x01,0x01,0x03,0x02,0x01,0x01,0x00,0x00,0x00 }; 601 602static void mxf_write_common_fields(AVFormatContext *s, AVStream *st) 603{ 604 MXFContext *mxf = s->priv_data; 605 ByteIOContext *pb = s->pb; 606 607 // find data define uls 608 mxf_write_local_tag(pb, 16, 0x0201); 609 if (st == mxf->timecode_track) 610 put_buffer(pb, smpte_12m_timecode_track_data_ul, 16); 611 else { 612 const MXFCodecUL *data_def_ul = mxf_get_data_definition_ul(st->codec->codec_type); 613 put_buffer(pb, data_def_ul->uid, 16); 614 } 615 616 // write duration 617 mxf_write_local_tag(pb, 8, 0x0202); 618 put_be64(pb, mxf->duration); 619} 620 621static void mxf_write_sequence(AVFormatContext *s, AVStream *st, enum MXFMetadataSetType type) 622{ 623 MXFContext *mxf = s->priv_data; 624 ByteIOContext *pb = s->pb; 625 enum MXFMetadataSetType component; 626 627 mxf_write_metadata_key(pb, 0x010f00); 628 PRINT_KEY(s, "sequence key", pb->buf_ptr - 16); 629 klv_encode_ber_length(pb, 80); 630 631 mxf_write_local_tag(pb, 16, 0x3C0A); 632 mxf_write_uuid(pb, type == MaterialPackage ? Sequence: Sequence + TypeBottom, st->index); 633 634 PRINT_KEY(s, "sequence uid", pb->buf_ptr - 16); 635 mxf_write_common_fields(s, st); 636 637 // write structural component 638 mxf_write_local_tag(pb, 16 + 8, 0x1001); 639 mxf_write_refs_count(pb, 1); 640 if (st == mxf->timecode_track) 641 component = TimecodeComponent; 642 else 643 component = SourceClip; 644 if (type == SourcePackage) 645 component += TypeBottom; 646 mxf_write_uuid(pb, component, st->index); 647} 648 649static void mxf_write_timecode_component(AVFormatContext *s, AVStream *st, enum MXFMetadataSetType type) 650{ 651 MXFContext *mxf = s->priv_data; 652 ByteIOContext *pb = s->pb; 653 654 mxf_write_metadata_key(pb, 0x011400); 655 klv_encode_ber_length(pb, 75); 656 657 // UID 658 mxf_write_local_tag(pb, 16, 0x3C0A); 659 mxf_write_uuid(pb, type == MaterialPackage ? TimecodeComponent : 660 TimecodeComponent + TypeBottom, st->index); 661 662 mxf_write_common_fields(s, st); 663 664 // Start Time Code 665 mxf_write_local_tag(pb, 8, 0x1501); 666 put_be64(pb, mxf->timecode_start); 667 668 // Rounded Time Code Base 669 mxf_write_local_tag(pb, 2, 0x1502); 670 put_be16(pb, mxf->timecode_base); 671 672 // Drop Frame 673 mxf_write_local_tag(pb, 1, 0x1503); 674 put_byte(pb, mxf->timecode_drop_frame); 675} 676 677static void mxf_write_structural_component(AVFormatContext *s, AVStream *st, enum MXFMetadataSetType type) 678{ 679 ByteIOContext *pb = s->pb; 680 int i; 681 682 mxf_write_metadata_key(pb, 0x011100); 683 PRINT_KEY(s, "sturctural component key", pb->buf_ptr - 16); 684 klv_encode_ber_length(pb, 108); 685 686 // write uid 687 mxf_write_local_tag(pb, 16, 0x3C0A); 688 mxf_write_uuid(pb, type == MaterialPackage ? SourceClip: SourceClip + TypeBottom, st->index); 689 690 PRINT_KEY(s, "structural component uid", pb->buf_ptr - 16); 691 mxf_write_common_fields(s, st); 692 693 // write start_position 694 mxf_write_local_tag(pb, 8, 0x1201); 695 put_be64(pb, 0); 696 697 // write source package uid, end of the reference 698 mxf_write_local_tag(pb, 32, 0x1101); 699 if (type == SourcePackage) { 700 for (i = 0; i < 4; i++) 701 put_be64(pb, 0); 702 } else 703 mxf_write_umid(s, 1); 704 705 // write source track id 706 mxf_write_local_tag(pb, 4, 0x1102); 707 if (type == SourcePackage) 708 put_be32(pb, 0); 709 else 710 put_be32(pb, st->index+2); 711} 712 713static void mxf_write_multi_descriptor(AVFormatContext *s) 714{ 715 MXFContext *mxf = s->priv_data; 716 ByteIOContext *pb = s->pb; 717 const uint8_t *ul; 718 int i; 719 720 mxf_write_metadata_key(pb, 0x014400); 721 PRINT_KEY(s, "multiple descriptor key", pb->buf_ptr - 16); 722 klv_encode_ber_length(pb, 64 + 16 * s->nb_streams); 723 724 mxf_write_local_tag(pb, 16, 0x3C0A); 725 mxf_write_uuid(pb, MultipleDescriptor, 0); 726 PRINT_KEY(s, "multi_desc uid", pb->buf_ptr - 16); 727 728 // write sample rate 729 mxf_write_local_tag(pb, 8, 0x3001); 730 put_be32(pb, mxf->time_base.den); 731 put_be32(pb, mxf->time_base.num); 732 733 // write essence container ul 734 mxf_write_local_tag(pb, 16, 0x3004); 735 if (mxf->essence_container_count > 1) 736 ul = multiple_desc_ul; 737 else { 738 MXFStreamContext *sc = s->streams[0]->priv_data; 739 ul = mxf_essence_container_uls[sc->index].container_ul; 740 } 741 put_buffer(pb, ul, 16); 742 743 // write sub descriptor refs 744 mxf_write_local_tag(pb, s->nb_streams * 16 + 8, 0x3F01); 745 mxf_write_refs_count(pb, s->nb_streams); 746 for (i = 0; i < s->nb_streams; i++) 747 mxf_write_uuid(pb, SubDescriptor, i); 748} 749 750static void mxf_write_generic_desc(AVFormatContext *s, AVStream *st, const UID key, unsigned size) 751{ 752 MXFContext *mxf = s->priv_data; 753 MXFStreamContext *sc = st->priv_data; 754 ByteIOContext *pb = s->pb; 755 756 put_buffer(pb, key, 16); 757 klv_encode_ber4_length(pb, size+20+8+12+20); 758 759 mxf_write_local_tag(pb, 16, 0x3C0A); 760 mxf_write_uuid(pb, SubDescriptor, st->index); 761 762 mxf_write_local_tag(pb, 4, 0x3006); 763 put_be32(pb, st->index+2); 764 765 mxf_write_local_tag(pb, 8, 0x3001); 766 put_be32(pb, mxf->time_base.den); 767 put_be32(pb, mxf->time_base.num); 768 769 mxf_write_local_tag(pb, 16, 0x3004); 770 put_buffer(pb, mxf_essence_container_uls[sc->index].container_ul, 16); 771} 772 773static const UID mxf_mpegvideo_descriptor_key = { 0x06,0x0E,0x2B,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x51,0x00 }; 774static const UID mxf_wav_descriptor_key = { 0x06,0x0E,0x2B,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x48,0x00 }; 775static const UID mxf_aes3_descriptor_key = { 0x06,0x0E,0x2B,0x34,0x02,0x53,0x01,0x01,0x0d,0x01,0x01,0x01,0x01,0x01,0x47,0x00 }; 776static const UID mxf_cdci_descriptor_key = { 0x06,0x0E,0x2B,0x34,0x02,0x53,0x01,0x01,0x0D,0x01,0x01,0x01,0x01,0x01,0x28,0x00 }; 777static const UID mxf_generic_sound_descriptor_key = { 0x06,0x0E,0x2B,0x34,0x02,0x53,0x01,0x01,0x0D,0x01,0x01,0x01,0x01,0x01,0x42,0x00 }; 778 779static void mxf_write_cdci_common(AVFormatContext *s, AVStream *st, const UID key, unsigned size) 780{ 781 MXFStreamContext *sc = st->priv_data; 782 ByteIOContext *pb = s->pb; 783 int stored_height = (st->codec->height+15)/16*16; 784 int display_height; 785 int f1, f2; 786 787 mxf_write_generic_desc(s, st, key, size+8+8+8+8+8+8+5+16+sc->interlaced*4+12+20); 788 789 mxf_write_local_tag(pb, 4, 0x3203); 790 put_be32(pb, st->codec->width); 791 792 mxf_write_local_tag(pb, 4, 0x3202); 793 put_be32(pb, stored_height>>sc->interlaced); 794 795 mxf_write_local_tag(pb, 4, 0x3209); 796 put_be32(pb, st->codec->width); 797 798 if (st->codec->height == 608) // PAL + VBI 799 display_height = 576; 800 else if (st->codec->height == 512) // NTSC + VBI 801 display_height = 486; 802 else 803 display_height = st->codec->height; 804 805 mxf_write_local_tag(pb, 4, 0x3208); 806 put_be32(pb, display_height>>sc->interlaced); 807 808 // component depth 809 mxf_write_local_tag(pb, 4, 0x3301); 810 put_be32(pb, 8); 811 812 // horizontal subsampling 813 mxf_write_local_tag(pb, 4, 0x3302); 814 put_be32(pb, 2); 815 816 // frame layout 817 mxf_write_local_tag(pb, 1, 0x320C); 818 put_byte(pb, sc->interlaced); 819 820 // video line map 821 switch (st->codec->height) { 822 case 576: f1 = 23; f2 = 336; break; 823 case 608: f1 = 7; f2 = 320; break; 824 case 480: f1 = 20; f2 = 283; break; 825 case 512: f1 = 7; f2 = 270; break; 826 case 720: f1 = 26; f2 = 0; break; // progressive 827 case 1080: f1 = 21; f2 = 584; break; 828 default: f1 = 0; f2 = 0; break; 829 } 830 831 if (!sc->interlaced) { 832 f2 = 0; 833 f1 *= 2; 834 } 835 836 mxf_write_local_tag(pb, 12+sc->interlaced*4, 0x320D); 837 put_be32(pb, sc->interlaced ? 2 : 1); 838 put_be32(pb, 4); 839 put_be32(pb, f1); 840 if (sc->interlaced) 841 put_be32(pb, f2); 842 843 mxf_write_local_tag(pb, 8, 0x320E); 844 put_be32(pb, sc->aspect_ratio.num); 845 put_be32(pb, sc->aspect_ratio.den); 846 847 mxf_write_local_tag(pb, 16, 0x3201); 848 put_buffer(pb, *sc->codec_ul, 16); 849} 850 851static void mxf_write_cdci_desc(AVFormatContext *s, AVStream *st) 852{ 853 mxf_write_cdci_common(s, st, mxf_cdci_descriptor_key, 0); 854} 855 856static void mxf_write_mpegvideo_desc(AVFormatContext *s, AVStream *st) 857{ 858 ByteIOContext *pb = s->pb; 859 int profile_and_level = (st->codec->profile<<4) | st->codec->level; 860 861 mxf_write_cdci_common(s, st, mxf_mpegvideo_descriptor_key, 8+5); 862 863 // bit rate 864 mxf_write_local_tag(pb, 4, 0x8000); 865 put_be32(pb, st->codec->bit_rate); 866 867 // profile and level 868 mxf_write_local_tag(pb, 1, 0x8007); 869 if (!st->codec->profile) 870 profile_and_level |= 0x80; // escape bit 871 put_byte(pb, profile_and_level); 872} 873 874static void mxf_write_generic_sound_common(AVFormatContext *s, AVStream *st, const UID key, unsigned size) 875{ 876 ByteIOContext *pb = s->pb; 877 878 mxf_write_generic_desc(s, st, key, size+5+12+8+8); 879 880 // audio locked 881 mxf_write_local_tag(pb, 1, 0x3D02); 882 put_byte(pb, 1); 883 884 // write audio sampling rate 885 mxf_write_local_tag(pb, 8, 0x3D03); 886 put_be32(pb, st->codec->sample_rate); 887 put_be32(pb, 1); 888 889 mxf_write_local_tag(pb, 4, 0x3D07); 890 put_be32(pb, st->codec->channels); 891 892 mxf_write_local_tag(pb, 4, 0x3D01); 893 put_be32(pb, av_get_bits_per_sample(st->codec->codec_id)); 894} 895 896static void mxf_write_wav_common(AVFormatContext *s, AVStream *st, const UID key, unsigned size) 897{ 898 ByteIOContext *pb = s->pb; 899 900 mxf_write_generic_sound_common(s, st, key, size+6+8); 901 902 mxf_write_local_tag(pb, 2, 0x3D0A); 903 put_be16(pb, st->codec->block_align); 904 905 // avg bytes per sec 906 mxf_write_local_tag(pb, 4, 0x3D09); 907 put_be32(pb, st->codec->block_align*st->codec->sample_rate); 908} 909 910static void mxf_write_wav_desc(AVFormatContext *s, AVStream *st) 911{ 912 mxf_write_wav_common(s, st, mxf_wav_descriptor_key, 0); 913} 914 915static void mxf_write_aes3_desc(AVFormatContext *s, AVStream *st) 916{ 917 mxf_write_wav_common(s, st, mxf_aes3_descriptor_key, 0); 918} 919 920static void mxf_write_generic_sound_desc(AVFormatContext *s, AVStream *st) 921{ 922 mxf_write_generic_sound_common(s, st, mxf_generic_sound_descriptor_key, 0); 923} 924 925static void mxf_write_package(AVFormatContext *s, enum MXFMetadataSetType type) 926{ 927 MXFContext *mxf = s->priv_data; 928 ByteIOContext *pb = s->pb; 929 int i, track_count = s->nb_streams+1; 930 931 if (type == MaterialPackage) { 932 mxf_write_metadata_key(pb, 0x013600); 933 PRINT_KEY(s, "Material Package key", pb->buf_ptr - 16); 934 klv_encode_ber_length(pb, 92 + 16*track_count); 935 } else { 936 mxf_write_metadata_key(pb, 0x013700); 937 PRINT_KEY(s, "Source Package key", pb->buf_ptr - 16); 938 klv_encode_ber_length(pb, 112 + 16*track_count); // 20 bytes length for descriptor reference 939 } 940 941 // write uid 942 mxf_write_local_tag(pb, 16, 0x3C0A); 943 mxf_write_uuid(pb, type, 0); 944 av_log(s,AV_LOG_DEBUG, "package type:%d\n", type); 945 PRINT_KEY(s, "package uid", pb->buf_ptr - 16); 946 947 // write package umid 948 mxf_write_local_tag(pb, 32, 0x4401); 949 mxf_write_umid(s, type == SourcePackage); 950 PRINT_KEY(s, "package umid second part", pb->buf_ptr - 16); 951 952 // package creation date 953 mxf_write_local_tag(pb, 8, 0x4405); 954 put_be64(pb, mxf->timestamp); 955 956 // package modified date 957 mxf_write_local_tag(pb, 8, 0x4404); 958 put_be64(pb, mxf->timestamp); 959 960 // write track refs 961 mxf_write_local_tag(pb, track_count*16 + 8, 0x4403); 962 mxf_write_refs_count(pb, track_count); 963 mxf_write_uuid(pb, type == MaterialPackage ? Track : 964 Track + TypeBottom, -1); // timecode track 965 for (i = 0; i < s->nb_streams; i++) 966 mxf_write_uuid(pb, type == MaterialPackage ? Track : Track + TypeBottom, i); 967 968 // write multiple descriptor reference 969 if (type == SourcePackage) { 970 mxf_write_local_tag(pb, 16, 0x4701); 971 if (s->nb_streams > 1) { 972 mxf_write_uuid(pb, MultipleDescriptor, 0); 973 mxf_write_multi_descriptor(s); 974 } else 975 mxf_write_uuid(pb, SubDescriptor, 0); 976 } 977 978 // write timecode track 979 mxf_write_track(s, mxf->timecode_track, type); 980 mxf_write_sequence(s, mxf->timecode_track, type); 981 mxf_write_timecode_component(s, mxf->timecode_track, type); 982 983 for (i = 0; i < s->nb_streams; i++) { 984 AVStream *st = s->streams[i]; 985 mxf_write_track(s, st, type); 986 mxf_write_sequence(s, st, type); 987 mxf_write_structural_component(s, st, type); 988 989 if (type == SourcePackage) { 990 MXFStreamContext *sc = st->priv_data; 991 mxf_essence_container_uls[sc->index].write_desc(s, st); 992 } 993 } 994} 995 996static int mxf_write_essence_container_data(AVFormatContext *s) 997{ 998 ByteIOContext *pb = s->pb; 999 1000 mxf_write_metadata_key(pb, 0x012300); 1001 klv_encode_ber_length(pb, 72); 1002 1003 mxf_write_local_tag(pb, 16, 0x3C0A); // Instance UID 1004 mxf_write_uuid(pb, EssenceContainerData, 0); 1005 1006 mxf_write_local_tag(pb, 32, 0x2701); // Linked Package UID 1007 mxf_write_umid(s, 1); 1008 1009 mxf_write_local_tag(pb, 4, 0x3F07); // BodySID 1010 put_be32(pb, 1); 1011 1012 mxf_write_local_tag(pb, 4, 0x3F06); // IndexSID 1013 put_be32(pb, 2); 1014 1015 return 0; 1016} 1017 1018static int mxf_write_header_metadata_sets(AVFormatContext *s) 1019{ 1020 mxf_write_preface(s); 1021 mxf_write_identification(s); 1022 mxf_write_content_storage(s); 1023 mxf_write_package(s, MaterialPackage); 1024 mxf_write_package(s, SourcePackage); 1025 mxf_write_essence_container_data(s); 1026 return 0; 1027} 1028 1029static unsigned klv_fill_size(uint64_t size) 1030{ 1031 unsigned pad = KAG_SIZE - (size & (KAG_SIZE-1)); 1032 if (pad < 20) // smallest fill item possible 1033 return pad + KAG_SIZE; 1034 else 1035 return pad & (KAG_SIZE-1); 1036} 1037 1038static void mxf_write_index_table_segment(AVFormatContext *s) 1039{ 1040 MXFContext *mxf = s->priv_data; 1041 ByteIOContext *pb = s->pb; 1042 int i, j; 1043 int temporal_reordering = 0; 1044 int key_index = mxf->last_key_index; 1045 1046 av_log(s, AV_LOG_DEBUG, "edit units count %d\n", mxf->edit_units_count); 1047 1048 if (!mxf->edit_units_count && !mxf->edit_unit_byte_count) 1049 return; 1050 1051 put_buffer(pb, index_table_segment_key, 16); 1052 1053 if (mxf->edit_unit_byte_count) { 1054 klv_encode_ber_length(pb, 80); 1055 } else { 1056 klv_encode_ber_length(pb, 85 + 12+(s->nb_streams+1)*6 + 1057 12+mxf->edit_units_count*(11+mxf->slice_count*4)); 1058 } 1059 1060 // instance id 1061 mxf_write_local_tag(pb, 16, 0x3C0A); 1062 mxf_write_uuid(pb, IndexTableSegment, 0); 1063 1064 // index edit rate 1065 mxf_write_local_tag(pb, 8, 0x3F0B); 1066 put_be32(pb, mxf->time_base.den); 1067 put_be32(pb, mxf->time_base.num); 1068 1069 // index start position 1070 mxf_write_local_tag(pb, 8, 0x3F0C); 1071 put_be64(pb, mxf->last_indexed_edit_unit); 1072 1073 // index duration 1074 mxf_write_local_tag(pb, 8, 0x3F0D); 1075 if (mxf->edit_unit_byte_count) 1076 put_be64(pb, 0); // index table covers whole container 1077 else 1078 put_be64(pb, mxf->edit_units_count); 1079 1080 // edit unit byte count 1081 mxf_write_local_tag(pb, 4, 0x3F05); 1082 put_be32(pb, mxf->edit_unit_byte_count); 1083 1084 // index sid 1085 mxf_write_local_tag(pb, 4, 0x3F06); 1086 put_be32(pb, 2); 1087 1088 // body sid 1089 mxf_write_local_tag(pb, 4, 0x3F07); 1090 put_be32(pb, 1); 1091 1092 if (!mxf->edit_unit_byte_count) { 1093 // real slice count - 1 1094 mxf_write_local_tag(pb, 1, 0x3F08); 1095 put_byte(pb, mxf->slice_count); 1096 1097 // delta entry array 1098 mxf_write_local_tag(pb, 8 + (s->nb_streams+1)*6, 0x3F09); 1099 put_be32(pb, s->nb_streams+1); // num of entries 1100 put_be32(pb, 6); // size of one entry 1101 // write system item delta entry 1102 put_byte(pb, 0); 1103 put_byte(pb, 0); // slice entry 1104 put_be32(pb, 0); // element delta 1105 for (i = 0; i < s->nb_streams; i++) { 1106 AVStream *st = s->streams[i]; 1107 MXFStreamContext *sc = st->priv_data; 1108 put_byte(pb, sc->temporal_reordering); 1109 if (sc->temporal_reordering) 1110 temporal_reordering = 1; 1111 if (i == 0) { // video track 1112 put_byte(pb, 0); // slice number 1113 put_be32(pb, KAG_SIZE); // system item size including klv fill 1114 } else { // audio track 1115 unsigned audio_frame_size = sc->aic.samples[0]*sc->aic.sample_size; 1116 audio_frame_size += klv_fill_size(audio_frame_size); 1117 put_byte(pb, 1); 1118 put_be32(pb, (i-1)*audio_frame_size); // element delta 1119 } 1120 } 1121 1122 mxf_write_local_tag(pb, 8 + mxf->edit_units_count*(11+mxf->slice_count*4), 0x3F0A); 1123 put_be32(pb, mxf->edit_units_count); // num of entries 1124 put_be32(pb, 11+mxf->slice_count*4); // size of one entry 1125 for (i = 0; i < mxf->edit_units_count; i++) { 1126 int temporal_offset = 0; 1127 if (temporal_reordering) { 1128 for (j = i+1; j < mxf->edit_units_count; j++) { 1129 temporal_offset++; 1130 if (mxf->index_entries[j].flags & 0x10) { // backward prediction 1131 // next is not b, so is reordered 1132 if (!(mxf->index_entries[i+1].flags & 0x10)) { 1133 if ((mxf->index_entries[i].flags & 0x11) == 0) // I frame 1134 temporal_offset = 0; 1135 else 1136 temporal_offset = -temporal_offset; 1137 } 1138 break; 1139 } 1140 } 1141 } 1142 put_byte(pb, temporal_offset); 1143 1144 if (!(mxf->index_entries[i].flags & 0x33)) { // I frame 1145 if (mxf->index_entries[i].flags & 0x40 && // seq header 1146 (!temporal_reordering || !temporal_offset)) 1147 mxf->index_entries[i].flags |= 0x80; // random access 1148 mxf->last_key_index = key_index; 1149 key_index = i; 1150 } 1151 if ((mxf->index_entries[i].flags & 0x30) == 0x30) { // back and forward prediction 1152 put_byte(pb, mxf->last_key_index - i); 1153 } else { 1154 put_byte(pb, key_index - i); // key frame offset 1155 if ((mxf->index_entries[i].flags & 0x20) == 0x20) // only forward 1156 mxf->last_key_index = key_index; 1157 } 1158 put_byte(pb, mxf->index_entries[i].flags); 1159 // stream offset 1160 put_be64(pb, mxf->index_entries[i].offset); 1161 if (s->nb_streams > 1) 1162 put_be32(pb, mxf->index_entries[i].slice_offset); 1163 } 1164 1165 mxf->last_key_index = key_index - mxf->edit_units_count; 1166 mxf->last_indexed_edit_unit += mxf->edit_units_count; 1167 mxf->edit_units_count = 0; 1168 } 1169} 1170 1171static void mxf_write_klv_fill(AVFormatContext *s) 1172{ 1173 unsigned pad = klv_fill_size(url_ftell(s->pb)); 1174 if (pad) { 1175 put_buffer(s->pb, klv_fill_key, 16); 1176 pad -= 16 + 4; 1177 klv_encode_ber4_length(s->pb, pad); 1178 for (; pad; pad--) 1179 put_byte(s->pb, 0); 1180 assert(!(url_ftell(s->pb) & (KAG_SIZE-1))); 1181 } 1182} 1183 1184static void mxf_write_partition(AVFormatContext *s, int bodysid, 1185 int indexsid, 1186 const uint8_t *key, int write_metadata) 1187{ 1188 MXFContext *mxf = s->priv_data; 1189 ByteIOContext *pb = s->pb; 1190 int64_t header_byte_count_offset; 1191 unsigned index_byte_count = 0; 1192 uint64_t partition_offset = url_ftell(pb); 1193 1194 if (!mxf->edit_unit_byte_count && mxf->edit_units_count) 1195 index_byte_count = 85 + 12+(s->nb_streams+1)*6 + 1196 12+mxf->edit_units_count*(11+mxf->slice_count*4); 1197 else if (mxf->edit_unit_byte_count && indexsid) 1198 index_byte_count = 80; 1199 1200 if (index_byte_count) { 1201 // add encoded ber length 1202 index_byte_count += 16 + klv_ber_length(index_byte_count); 1203 index_byte_count += klv_fill_size(index_byte_count); 1204 } 1205 1206 if (!memcmp(key, body_partition_key, 16)) { 1207 mxf->body_partition_offset = 1208 av_realloc(mxf->body_partition_offset, 1209 (mxf->body_partitions_count+1)* 1210 sizeof(*mxf->body_partition_offset)); 1211 mxf->body_partition_offset[mxf->body_partitions_count++] = partition_offset; 1212 } 1213 1214 // write klv 1215 put_buffer(pb, key, 16); 1216 klv_encode_ber_length(pb, 88 + 16 * mxf->essence_container_count); 1217 1218 // write partition value 1219 put_be16(pb, 1); // majorVersion 1220 put_be16(pb, 2); // minorVersion 1221 put_be32(pb, KAG_SIZE); // KAGSize 1222 1223 put_be64(pb, partition_offset); // ThisPartition 1224 1225 if (!memcmp(key, body_partition_key, 16) && mxf->body_partitions_count > 1) 1226 put_be64(pb, mxf->body_partition_offset[mxf->body_partitions_count-2]); // PreviousPartition 1227 else if (!memcmp(key, footer_partition_key, 16) && mxf->body_partitions_count) 1228 put_be64(pb, mxf->body_partition_offset[mxf->body_partitions_count-1]); // PreviousPartition 1229 else 1230 put_be64(pb, 0); 1231 1232 put_be64(pb, mxf->footer_partition_offset); // footerPartition 1233 1234 // set offset 1235 header_byte_count_offset = url_ftell(pb); 1236 put_be64(pb, 0); // headerByteCount, update later 1237 1238 // indexTable 1239 put_be64(pb, index_byte_count); // indexByteCount 1240 put_be32(pb, index_byte_count ? indexsid : 0); // indexSID 1241 1242 // BodyOffset 1243 if (bodysid && mxf->edit_units_count && mxf->body_partitions_count) { 1244 put_be64(pb, mxf->body_offset); 1245 } else 1246 put_be64(pb, 0); 1247 1248 put_be32(pb, bodysid); // bodySID 1249 1250 // operational pattern 1251 put_buffer(pb, op1a_ul, 16); 1252 1253 // essence container 1254 mxf_write_essence_container_refs(s); 1255 1256 if (write_metadata) { 1257 // mark the start of the headermetadata and calculate metadata size 1258 int64_t pos, start; 1259 unsigned header_byte_count; 1260 1261 mxf_write_klv_fill(s); 1262 start = url_ftell(s->pb); 1263 mxf_write_primer_pack(s); 1264 mxf_write_header_metadata_sets(s); 1265 pos = url_ftell(s->pb); 1266 header_byte_count = pos - start + klv_fill_size(pos); 1267 1268 // update header_byte_count 1269 url_fseek(pb, header_byte_count_offset, SEEK_SET); 1270 put_be64(pb, header_byte_count); 1271 url_fseek(pb, pos, SEEK_SET); 1272 } 1273 1274 put_flush_packet(pb); 1275} 1276 1277static const UID mxf_mpeg2_codec_uls[] = { 1278 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x03,0x04,0x01,0x02,0x02,0x01,0x01,0x10,0x00 }, // MP-ML I-Frame 1279 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x03,0x04,0x01,0x02,0x02,0x01,0x01,0x11,0x00 }, // MP-ML Long GOP 1280 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x03,0x04,0x01,0x02,0x02,0x01,0x02,0x02,0x00 }, // 422P-ML I-Frame 1281 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x03,0x04,0x01,0x02,0x02,0x01,0x02,0x03,0x00 }, // 422P-ML Long GOP 1282 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x03,0x04,0x01,0x02,0x02,0x01,0x03,0x02,0x00 }, // MP-HL I-Frame 1283 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x03,0x04,0x01,0x02,0x02,0x01,0x03,0x03,0x00 }, // MP-HL Long GOP 1284 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x03,0x04,0x01,0x02,0x02,0x01,0x04,0x02,0x00 }, // 422P-HL I-Frame 1285 { 0x06,0x0E,0x2B,0x34,0x04,0x01,0x01,0x03,0x04,0x01,0x02,0x02,0x01,0x04,0x03,0x00 }, // 422P-HL Long GOP 1286}; 1287 1288static const UID *mxf_get_mpeg2_codec_ul(AVCodecContext *avctx) 1289{ 1290 int long_gop = avctx->gop_size > 1 || avctx->has_b_frames; 1291 1292 if (avctx->profile == 4) { // Main 1293 if (avctx->level == 8) // Main 1294 return &mxf_mpeg2_codec_uls[0+long_gop]; 1295 else if (avctx->level == 4) // High 1296 return &mxf_mpeg2_codec_uls[4+long_gop]; 1297 } else if (avctx->profile == 0) { // 422 1298 if (avctx->level == 5) // Main 1299 return &mxf_mpeg2_codec_uls[2+long_gop]; 1300 else if (avctx->level == 2) // High 1301 return &mxf_mpeg2_codec_uls[6+long_gop]; 1302 } 1303 return NULL; 1304} 1305 1306static int mxf_parse_mpeg2_frame(AVFormatContext *s, AVStream *st, AVPacket *pkt, int *flags) 1307{ 1308 MXFStreamContext *sc = st->priv_data; 1309 MXFContext *mxf = s->priv_data; 1310 uint32_t c = -1; 1311 int i; 1312 1313 *flags = 0; 1314 1315 for(i = 0; i < pkt->size - 4; i++) { 1316 c = (c<<8) + pkt->data[i]; 1317 if (c == 0x1b5) { 1318 if ((pkt->data[i+1] & 0xf0) == 0x10) { // seq ext 1319 st->codec->profile = pkt->data[i+1] & 0x07; 1320 st->codec->level = pkt->data[i+2] >> 4; 1321 } else if (i + 5 < pkt->size && (pkt->data[i+1] & 0xf0) == 0x80) { // pict coding ext 1322 sc->interlaced = !(pkt->data[i+5] & 0x80); // progressive frame 1323 break; 1324 } 1325 } else if (c == 0x1b8) { // gop 1326 if (pkt->data[i+4]>>6 & 0x01) { // closed 1327 sc->closed_gop = 1; 1328 if (*flags & 0x40) // sequence header present 1329 *flags |= 0x80; // random access 1330 } 1331 if (!mxf->header_written) { 1332 unsigned hours = (pkt->data[i+1]>>2) & 0x1f; 1333 unsigned minutes = ((pkt->data[i+1] & 0x03) << 4) | (pkt->data[i+2]>>4); 1334 unsigned seconds = ((pkt->data[i+2] & 0x07) << 3) | (pkt->data[i+3]>>5); 1335 unsigned frames = ((pkt->data[i+3] & 0x1f) << 1) | (pkt->data[i+4]>>7); 1336 mxf->timecode_drop_frame = !!(pkt->data[i+1] & 0x80); 1337 mxf->timecode_start = (hours*3600 + minutes*60 + seconds) * 1338 mxf->timecode_base + frames; 1339 if (mxf->timecode_drop_frame) { 1340 unsigned tminutes = 60 * hours + minutes; 1341 mxf->timecode_start -= 2 * (tminutes - tminutes / 10); 1342 } 1343 av_log(s, AV_LOG_DEBUG, "frame %d %d:%d:%d%c%d\n", mxf->timecode_start, 1344 hours, minutes, seconds, mxf->timecode_drop_frame ? ';':':', frames); 1345 } 1346 } else if (c == 0x1b3) { // seq 1347 *flags |= 0x40; 1348 switch ((pkt->data[i+4]>>4) & 0xf) { 1349 case 2: sc->aspect_ratio = (AVRational){ 4, 3}; break; 1350 case 3: sc->aspect_ratio = (AVRational){ 16, 9}; break; 1351 case 4: sc->aspect_ratio = (AVRational){221,100}; break; 1352 default: 1353 av_reduce(&sc->aspect_ratio.num, &sc->aspect_ratio.den, 1354 st->codec->width, st->codec->height, 1024*1024); 1355 } 1356 } else if (c == 0x100) { // pic 1357 int pict_type = (pkt->data[i+2]>>3) & 0x07; 1358 if (pict_type == 2) { // P frame 1359 *flags |= 0x22; 1360 sc->closed_gop = 0; // reset closed gop, don't matter anymore 1361 } else if (pict_type == 3) { // B frame 1362 if (sc->closed_gop) 1363 *flags |= 0x13; // only backward prediction 1364 else 1365 *flags |= 0x33; 1366 sc->temporal_reordering = -1; 1367 } else if (!pict_type) { 1368 av_log(s, AV_LOG_ERROR, "error parsing mpeg2 frame\n"); 1369 return 0; 1370 } 1371 } 1372 } 1373 if (s->oformat != &mxf_d10_muxer) 1374 sc->codec_ul = mxf_get_mpeg2_codec_ul(st->codec); 1375 return !!sc->codec_ul; 1376} 1377 1378static uint64_t mxf_parse_timestamp(time_t timestamp) 1379{ 1380 struct tm *time = gmtime(×tamp); 1381 return (uint64_t)(time->tm_year+1900) << 48 | 1382 (uint64_t)(time->tm_mon+1) << 40 | 1383 (uint64_t) time->tm_mday << 32 | 1384 time->tm_hour << 24 | 1385 time->tm_min << 16 | 1386 time->tm_sec << 8; 1387} 1388 1389static void mxf_gen_umid(AVFormatContext *s) 1390{ 1391 MXFContext *mxf = s->priv_data; 1392 uint32_t seed = ff_random_get_seed(); 1393 uint64_t umid = seed + 0x5294713400000000LL; 1394 1395 AV_WB64(mxf->umid , umid); 1396 AV_WB64(mxf->umid+8, umid>>8); 1397 1398 mxf->instance_number = seed; 1399} 1400 1401static int mxf_write_header(AVFormatContext *s) 1402{ 1403 MXFContext *mxf = s->priv_data; 1404 int i; 1405 uint8_t present[FF_ARRAY_ELEMS(mxf_essence_container_uls)] = {0}; 1406 const int *samples_per_frame = NULL; 1407 1408 if (!s->nb_streams) 1409 return -1; 1410 1411 for (i = 0; i < s->nb_streams; i++) { 1412 AVStream *st = s->streams[i]; 1413 MXFStreamContext *sc = av_mallocz(sizeof(*sc)); 1414 if (!sc) 1415 return AVERROR(ENOMEM); 1416 st->priv_data = sc; 1417 1418 if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) { 1419 if (i != 0) { 1420 av_log(s, AV_LOG_ERROR, "video stream must be first track\n"); 1421 return -1; 1422 } 1423 if (fabs(av_q2d(st->codec->time_base) - 1/25.0) < 0.0001) { 1424 samples_per_frame = PAL_samples_per_frame; 1425 mxf->time_base = (AVRational){ 1, 25 }; 1426 mxf->timecode_base = 25; 1427 } else if (fabs(av_q2d(st->codec->time_base) - 1001/30000.0) < 0.0001) { 1428 samples_per_frame = NTSC_samples_per_frame; 1429 mxf->time_base = (AVRational){ 1001, 30000 }; 1430 mxf->timecode_base = 30; 1431 } else { 1432 av_log(s, AV_LOG_ERROR, "unsupported video frame rate\n"); 1433 return -1; 1434 } 1435 av_set_pts_info(st, 64, mxf->time_base.num, mxf->time_base.den); 1436 if (s->oformat == &mxf_d10_muxer) { 1437 if (st->codec->bit_rate == 50000000) 1438 if (mxf->time_base.den == 25) sc->index = 3; 1439 else sc->index = 5; 1440 else if (st->codec->bit_rate == 40000000) 1441 if (mxf->time_base.den == 25) sc->index = 7; 1442 else sc->index = 9; 1443 else if (st->codec->bit_rate == 30000000) 1444 if (mxf->time_base.den == 25) sc->index = 11; 1445 else sc->index = 13; 1446 else { 1447 av_log(s, AV_LOG_ERROR, "error MXF D-10 only support 30/40/50 mbit/s\n"); 1448 return -1; 1449 } 1450 1451 mxf->edit_unit_byte_count = KAG_SIZE; // system element 1452 mxf->edit_unit_byte_count += 16 + 4 + (uint64_t)st->codec->bit_rate * 1453 mxf->time_base.num / (8*mxf->time_base.den); 1454 mxf->edit_unit_byte_count += klv_fill_size(mxf->edit_unit_byte_count); 1455 mxf->edit_unit_byte_count += 16 + 4 + 4 + samples_per_frame[0]*8*4; 1456 mxf->edit_unit_byte_count += klv_fill_size(mxf->edit_unit_byte_count); 1457 } 1458 } else if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) { 1459 if (st->codec->sample_rate != 48000) { 1460 av_log(s, AV_LOG_ERROR, "only 48khz is implemented\n"); 1461 return -1; 1462 } 1463 av_set_pts_info(st, 64, 1, st->codec->sample_rate); 1464 if (s->oformat == &mxf_d10_muxer) { 1465 if (st->index != 1) { 1466 av_log(s, AV_LOG_ERROR, "MXF D-10 only support one audio track\n"); 1467 return -1; 1468 } 1469 if (st->codec->codec_id != CODEC_ID_PCM_S16LE && 1470 st->codec->codec_id != CODEC_ID_PCM_S24LE) { 1471 av_log(s, AV_LOG_ERROR, "MXF D-10 only support 16 or 24 bits le audio\n"); 1472 } 1473 sc->index = ((MXFStreamContext*)s->streams[0]->priv_data)->index + 1; 1474 } else 1475 mxf->slice_count = 1; 1476 } 1477 1478 if (!sc->index) { 1479 sc->index = mxf_get_essence_container_ul_index(st->codec->codec_id); 1480 if (sc->index == -1) { 1481 av_log(s, AV_LOG_ERROR, "track %d: could not find essence container ul, " 1482 "codec not currently supported in container\n", i); 1483 return -1; 1484 } 1485 } 1486 1487 sc->codec_ul = &mxf_essence_container_uls[sc->index].codec_ul; 1488 1489 memcpy(sc->track_essence_element_key, mxf_essence_container_uls[sc->index].element_ul, 15); 1490 sc->track_essence_element_key[15] = present[sc->index]; 1491 PRINT_KEY(s, "track essence element key", sc->track_essence_element_key); 1492 1493 if (!present[sc->index]) 1494 mxf->essence_container_count++; 1495 present[sc->index]++; 1496 } 1497 1498 if (s->oformat == &mxf_d10_muxer) { 1499 mxf->essence_container_count = 1; 1500 } 1501 1502 if (!(s->streams[0]->codec->flags & CODEC_FLAG_BITEXACT)) 1503 mxf_gen_umid(s); 1504 1505 for (i = 0; i < s->nb_streams; i++) { 1506 MXFStreamContext *sc = s->streams[i]->priv_data; 1507 // update element count 1508 sc->track_essence_element_key[13] = present[sc->index]; 1509 sc->order = AV_RB32(sc->track_essence_element_key+12); 1510 } 1511 1512 if (s->timestamp) 1513 mxf->timestamp = mxf_parse_timestamp(s->timestamp); 1514 mxf->duration = -1; 1515 1516 mxf->timecode_track = av_mallocz(sizeof(*mxf->timecode_track)); 1517 if (!mxf->timecode_track) 1518 return AVERROR(ENOMEM); 1519 mxf->timecode_track->priv_data = av_mallocz(sizeof(MXFStreamContext)); 1520 if (!mxf->timecode_track->priv_data) 1521 return AVERROR(ENOMEM); 1522 mxf->timecode_track->index = -1; 1523 1524 if (!samples_per_frame) 1525 samples_per_frame = PAL_samples_per_frame; 1526 1527 if (ff_audio_interleave_init(s, samples_per_frame, mxf->time_base) < 0) 1528 return -1; 1529 1530 return 0; 1531} 1532 1533static const uint8_t system_metadata_pack_key[] = { 0x06,0x0E,0x2B,0x34,0x02,0x05,0x01,0x01,0x0D,0x01,0x03,0x01,0x04,0x01,0x01,0x00 }; 1534static const uint8_t system_metadata_package_set_key[] = { 0x06,0x0E,0x2B,0x34,0x02,0x43,0x01,0x01,0x0D,0x01,0x03,0x01,0x04,0x01,0x02,0x01 }; 1535 1536static uint32_t ff_framenum_to_12m_time_code(unsigned frame, int drop, int fps) 1537{ 1538 return (0 << 31) | // color frame flag 1539 (0 << 30) | // drop frame flag 1540 ( ((frame % fps) / 10) << 28) | // tens of frames 1541 ( ((frame % fps) % 10) << 24) | // units of frames 1542 (0 << 23) | // field phase (NTSC), b0 (PAL) 1543 ((((frame / fps) % 60) / 10) << 20) | // tens of seconds 1544 ((((frame / fps) % 60) % 10) << 16) | // units of seconds 1545 (0 << 15) | // b0 (NTSC), b2 (PAL) 1546 ((((frame / (fps * 60)) % 60) / 10) << 12) | // tens of minutes 1547 ((((frame / (fps * 60)) % 60) % 10) << 8) | // units of minutes 1548 (0 << 7) | // b1 1549 (0 << 6) | // b2 (NSC), field phase (PAL) 1550 ((((frame / (fps * 3600) % 24)) / 10) << 4) | // tens of hours 1551 ( (frame / (fps * 3600) % 24)) % 10; // units of hours 1552} 1553 1554static void mxf_write_system_item(AVFormatContext *s) 1555{ 1556 MXFContext *mxf = s->priv_data; 1557 ByteIOContext *pb = s->pb; 1558 unsigned frame; 1559 uint32_t time_code; 1560 1561 frame = mxf->timecode_start + mxf->last_indexed_edit_unit + mxf->edit_units_count; 1562 1563 // write system metadata pack 1564 put_buffer(pb, system_metadata_pack_key, 16); 1565 klv_encode_ber4_length(pb, 57); 1566 put_byte(pb, 0x5c); // UL, user date/time stamp, picture and sound item present 1567 put_byte(pb, 0x04); // content package rate 1568 put_byte(pb, 0x00); // content package type 1569 put_be16(pb, 0x00); // channel handle 1570 put_be16(pb, frame); // continuity count 1571 if (mxf->essence_container_count > 1) 1572 put_buffer(pb, multiple_desc_ul, 16); 1573 else { 1574 MXFStreamContext *sc = s->streams[0]->priv_data; 1575 put_buffer(pb, mxf_essence_container_uls[sc->index].container_ul, 16); 1576 } 1577 put_byte(pb, 0); 1578 put_be64(pb, 0); 1579 put_be64(pb, 0); // creation date/time stamp 1580 1581 put_byte(pb, 0x81); // SMPTE 12M time code 1582 time_code = ff_framenum_to_12m_time_code(frame, mxf->timecode_drop_frame, mxf->timecode_base); 1583 put_be32(pb, time_code); 1584 put_be32(pb, 0); // binary group data 1585 put_be64(pb, 0); 1586 1587 // write system metadata package set 1588 put_buffer(pb, system_metadata_package_set_key, 16); 1589 klv_encode_ber4_length(pb, 35); 1590 put_byte(pb, 0x83); // UMID 1591 put_be16(pb, 0x20); 1592 mxf_write_umid(s, 1); 1593} 1594 1595static void mxf_write_d10_video_packet(AVFormatContext *s, AVStream *st, AVPacket *pkt) 1596{ 1597 MXFContext *mxf = s->priv_data; 1598 ByteIOContext *pb = s->pb; 1599 int packet_size = (uint64_t)st->codec->bit_rate*mxf->time_base.num / 1600 (8*mxf->time_base.den); // frame size 1601 int pad; 1602 1603 packet_size += 16 + 4; 1604 packet_size += klv_fill_size(packet_size); 1605 1606 klv_encode_ber4_length(pb, pkt->size); 1607 put_buffer(pb, pkt->data, pkt->size); 1608 1609 // ensure CBR muxing by padding to correct video frame size 1610 pad = packet_size - pkt->size - 16 - 4; 1611 if (pad > 20) { 1612 put_buffer(s->pb, klv_fill_key, 16); 1613 pad -= 16 + 4; 1614 klv_encode_ber4_length(s->pb, pad); 1615 for (; pad; pad--) 1616 put_byte(s->pb, 0); 1617 assert(!(url_ftell(s->pb) & (KAG_SIZE-1))); 1618 } else { 1619 av_log(s, AV_LOG_WARNING, "cannot fill d-10 video packet\n"); 1620 for (; pad > 0; pad--) 1621 put_byte(s->pb, 0); 1622 } 1623} 1624 1625static void mxf_write_d10_audio_packet(AVFormatContext *s, AVStream *st, AVPacket *pkt) 1626{ 1627 MXFContext *mxf = s->priv_data; 1628 ByteIOContext *pb = s->pb; 1629 int frame_size = pkt->size / st->codec->block_align; 1630 uint8_t *samples = pkt->data; 1631 uint8_t *end = pkt->data + pkt->size; 1632 int i; 1633 1634 klv_encode_ber4_length(pb, 4 + frame_size*4*8); 1635 1636 put_byte(pb, (frame_size == 1920 ? 0 : (mxf->edit_units_count-1) % 5 + 1)); 1637 put_le16(pb, frame_size); 1638 put_byte(pb, (1<<st->codec->channels)-1); 1639 1640 while (samples < end) { 1641 for (i = 0; i < st->codec->channels; i++) { 1642 uint32_t sample; 1643 if (st->codec->codec_id == CODEC_ID_PCM_S24LE) { 1644 sample = AV_RL24(samples)<< 4; 1645 samples += 3; 1646 } else { 1647 sample = AV_RL16(samples)<<12; 1648 samples += 2; 1649 } 1650 put_le32(pb, sample | i); 1651 } 1652 for (; i < 8; i++) 1653 put_le32(pb, i); 1654 } 1655} 1656 1657static int mxf_write_packet(AVFormatContext *s, AVPacket *pkt) 1658{ 1659 MXFContext *mxf = s->priv_data; 1660 ByteIOContext *pb = s->pb; 1661 AVStream *st = s->streams[pkt->stream_index]; 1662 MXFStreamContext *sc = st->priv_data; 1663 int flags = 0; 1664 1665 if (!mxf->edit_unit_byte_count && !(mxf->edit_units_count % EDIT_UNITS_PER_BODY)) { 1666 mxf->index_entries = av_realloc(mxf->index_entries, 1667 (mxf->edit_units_count + EDIT_UNITS_PER_BODY)*sizeof(*mxf->index_entries)); 1668 if (!mxf->index_entries) { 1669 av_log(s, AV_LOG_ERROR, "could not allocate index entries\n"); 1670 return -1; 1671 } 1672 } 1673 1674 if (st->codec->codec_id == CODEC_ID_MPEG2VIDEO) { 1675 if (!mxf_parse_mpeg2_frame(s, st, pkt, &flags)) { 1676 av_log(s, AV_LOG_ERROR, "could not get mpeg2 profile and level\n"); 1677 return -1; 1678 } 1679 } 1680 1681 if (!mxf->header_written) { 1682 if (mxf->edit_unit_byte_count) { 1683 mxf_write_partition(s, 1, 2, header_open_partition_key, 1); 1684 mxf_write_klv_fill(s); 1685 mxf_write_index_table_segment(s); 1686 } else { 1687 mxf_write_partition(s, 0, 0, header_open_partition_key, 1); 1688 } 1689 mxf->header_written = 1; 1690 } 1691 1692 if (st->index == 0) { 1693 if (!mxf->edit_unit_byte_count && 1694 (!mxf->edit_units_count || mxf->edit_units_count > EDIT_UNITS_PER_BODY) && 1695 !(flags & 0x33)) { // I frame, Gop start 1696 mxf_write_klv_fill(s); 1697 mxf_write_partition(s, 1, 2, body_partition_key, 0); 1698 1699 mxf_write_klv_fill(s); 1700 mxf_write_index_table_segment(s); 1701 } 1702 1703 mxf_write_klv_fill(s); 1704 mxf_write_system_item(s); 1705 1706 if (!mxf->edit_unit_byte_count) { 1707 mxf->index_entries[mxf->edit_units_count].offset = mxf->body_offset; 1708 mxf->index_entries[mxf->edit_units_count].flags = flags; 1709 mxf->body_offset += KAG_SIZE; // size of system element 1710 } 1711 mxf->edit_units_count++; 1712 } else if (!mxf->edit_unit_byte_count && st->index == 1) { 1713 mxf->index_entries[mxf->edit_units_count-1].slice_offset = 1714 mxf->body_offset - mxf->index_entries[mxf->edit_units_count-1].offset; 1715 } 1716 1717 mxf_write_klv_fill(s); 1718 put_buffer(pb, sc->track_essence_element_key, 16); // write key 1719 if (s->oformat == &mxf_d10_muxer) { 1720 if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) 1721 mxf_write_d10_video_packet(s, st, pkt); 1722 else 1723 mxf_write_d10_audio_packet(s, st, pkt); 1724 } else { 1725 klv_encode_ber4_length(pb, pkt->size); // write length 1726 put_buffer(pb, pkt->data, pkt->size); 1727 mxf->body_offset += 16+4+pkt->size + klv_fill_size(16+4+pkt->size); 1728 } 1729 1730 put_flush_packet(pb); 1731 1732 return 0; 1733} 1734 1735static void mxf_write_random_index_pack(AVFormatContext *s) 1736{ 1737 MXFContext *mxf = s->priv_data; 1738 ByteIOContext *pb = s->pb; 1739 uint64_t pos = url_ftell(pb); 1740 int i; 1741 1742 put_buffer(pb, random_index_pack_key, 16); 1743 klv_encode_ber_length(pb, 28 + 12*mxf->body_partitions_count); 1744 1745 if (mxf->edit_unit_byte_count) 1746 put_be32(pb, 1); // BodySID of header partition 1747 else 1748 put_be32(pb, 0); 1749 put_be64(pb, 0); // offset of header partition 1750 1751 for (i = 0; i < mxf->body_partitions_count; i++) { 1752 put_be32(pb, 1); // BodySID 1753 put_be64(pb, mxf->body_partition_offset[i]); 1754 } 1755 1756 put_be32(pb, 0); // BodySID of footer partition 1757 put_be64(pb, mxf->footer_partition_offset); 1758 1759 put_be32(pb, url_ftell(pb) - pos + 4); 1760} 1761 1762static int mxf_write_footer(AVFormatContext *s) 1763{ 1764 MXFContext *mxf = s->priv_data; 1765 ByteIOContext *pb = s->pb; 1766 1767 mxf->duration = mxf->last_indexed_edit_unit + mxf->edit_units_count; 1768 1769 mxf_write_klv_fill(s); 1770 mxf->footer_partition_offset = url_ftell(pb); 1771 if (mxf->edit_unit_byte_count) { // no need to repeat index 1772 mxf_write_partition(s, 0, 0, footer_partition_key, 0); 1773 } else { 1774 mxf_write_partition(s, 0, 2, footer_partition_key, 0); 1775 1776 mxf_write_klv_fill(s); 1777 mxf_write_index_table_segment(s); 1778 } 1779 1780 mxf_write_klv_fill(s); 1781 mxf_write_random_index_pack(s); 1782 1783 if (!url_is_streamed(s->pb)) { 1784 url_fseek(pb, 0, SEEK_SET); 1785 if (mxf->edit_unit_byte_count) { 1786 mxf_write_partition(s, 1, 2, header_closed_partition_key, 1); 1787 mxf_write_klv_fill(s); 1788 mxf_write_index_table_segment(s); 1789 } else { 1790 mxf_write_partition(s, 0, 0, header_closed_partition_key, 1); 1791 } 1792 } 1793 1794 put_flush_packet(pb); 1795 1796 ff_audio_interleave_close(s); 1797 1798 av_freep(&mxf->index_entries); 1799 av_freep(&mxf->body_partition_offset); 1800 av_freep(&mxf->timecode_track->priv_data); 1801 av_freep(&mxf->timecode_track); 1802 1803 mxf_free(s); 1804 1805 return 0; 1806} 1807 1808static int mxf_interleave_get_packet(AVFormatContext *s, AVPacket *out, AVPacket *pkt, int flush) 1809{ 1810 int i, stream_count = 0; 1811 1812 for (i = 0; i < s->nb_streams; i++) 1813 stream_count += !!s->streams[i]->last_in_packet_buffer; 1814 1815 if (stream_count && (s->nb_streams == stream_count || flush)) { 1816 AVPacketList *pktl = s->packet_buffer; 1817 if (s->nb_streams != stream_count) { 1818 AVPacketList *last = NULL; 1819 // find last packet in edit unit 1820 while (pktl) { 1821 if (!stream_count || pktl->pkt.stream_index == 0) 1822 break; 1823 last = pktl; 1824 pktl = pktl->next; 1825 stream_count--; 1826 } 1827 // purge packet queue 1828 while (pktl) { 1829 AVPacketList *next = pktl->next; 1830 1831 if(s->streams[pktl->pkt.stream_index]->last_in_packet_buffer == pktl) 1832 s->streams[pktl->pkt.stream_index]->last_in_packet_buffer= NULL; 1833 av_free_packet(&pktl->pkt); 1834 av_freep(&pktl); 1835 pktl = next; 1836 } 1837 if (last) 1838 last->next = NULL; 1839 else { 1840 s->packet_buffer = NULL; 1841 s->packet_buffer_end= NULL; 1842 goto out; 1843 } 1844 pktl = s->packet_buffer; 1845 } 1846 1847 *out = pktl->pkt; 1848 //av_log(s, AV_LOG_DEBUG, "out st:%d dts:%lld\n", (*out).stream_index, (*out).dts); 1849 s->packet_buffer = pktl->next; 1850 if(s->streams[pktl->pkt.stream_index]->last_in_packet_buffer == pktl) 1851 s->streams[pktl->pkt.stream_index]->last_in_packet_buffer= NULL; 1852 if(!s->packet_buffer) 1853 s->packet_buffer_end= NULL; 1854 av_freep(&pktl); 1855 return 1; 1856 } else { 1857 out: 1858 av_init_packet(out); 1859 return 0; 1860 } 1861} 1862 1863static int mxf_compare_timestamps(AVFormatContext *s, AVPacket *next, AVPacket *pkt) 1864{ 1865 MXFStreamContext *sc = s->streams[pkt ->stream_index]->priv_data; 1866 MXFStreamContext *sc2 = s->streams[next->stream_index]->priv_data; 1867 1868 return next->dts > pkt->dts || 1869 (next->dts == pkt->dts && sc->order < sc2->order); 1870} 1871 1872static int mxf_interleave(AVFormatContext *s, AVPacket *out, AVPacket *pkt, int flush) 1873{ 1874 return ff_audio_rechunk_interleave(s, out, pkt, flush, 1875 mxf_interleave_get_packet, mxf_compare_timestamps); 1876} 1877 1878AVOutputFormat mxf_muxer = { 1879 "mxf", 1880 NULL_IF_CONFIG_SMALL("Material eXchange Format"), 1881 "application/mxf", 1882 "mxf", 1883 sizeof(MXFContext), 1884 CODEC_ID_PCM_S16LE, 1885 CODEC_ID_MPEG2VIDEO, 1886 mxf_write_header, 1887 mxf_write_packet, 1888 mxf_write_footer, 1889 AVFMT_NOTIMESTAMPS, 1890 NULL, 1891 mxf_interleave, 1892}; 1893 1894AVOutputFormat mxf_d10_muxer = { 1895 "mxf_d10", 1896 NULL_IF_CONFIG_SMALL("Material eXchange Format, D-10 Mapping"), 1897 "application/mxf", 1898 NULL, 1899 sizeof(MXFContext), 1900 CODEC_ID_PCM_S16LE, 1901 CODEC_ID_MPEG2VIDEO, 1902 mxf_write_header, 1903 mxf_write_packet, 1904 mxf_write_footer, 1905 AVFMT_NOTIMESTAMPS, 1906 NULL, 1907 mxf_interleave, 1908}; 1909