mxfenc.c 83 KB

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