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