Some compilers don't like x = {} where x is a boost::optional<string>
[libdcp.git] / src / sound_asset_writer.cc
index ff1d02c23057fd5458aa5826ff1675a0884e8ed7..f54c960e7ed4027582c80844d572149cc61da9d6 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2012-2020 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net>
 
     This file is part of libdcp.
 
     files in the program, then also delete it here.
 */
 
+
+/** @file  src/sound_asset_writer.cc
+ *  @brief SoundAssetWriter class
+ */
+
+
 #include "bitstream.h"
 #include "sound_asset_writer.h"
 #include "sound_asset.h"
 #include "compose.hpp"
 #include "crypto_context.h"
 #include <asdcp/AS_DCP.h>
+#include <asdcp/Metadata.h>
 #include <iostream>
 
+
 using std::min;
 using std::max;
 using std::cout;
@@ -48,6 +56,7 @@ using std::string;
 using std::vector;
 using namespace dcp;
 
+
 struct SoundAssetWriter::ASDCPState
 {
        ASDCP::PCM::MXFWriter mxf_writer;
@@ -56,27 +65,27 @@ struct SoundAssetWriter::ASDCPState
        ASDCP::PCM::AudioDescriptor desc;
 };
 
-SoundAssetWriter::SoundAssetWriter (SoundAsset* asset, boost::filesystem::path file, bool sync)
+
+SoundAssetWriter::SoundAssetWriter (SoundAsset* asset, boost::filesystem::path file, vector<Channel> active_channels, bool sync)
        : AssetWriter (asset, file)
        , _state (new SoundAssetWriter::ASDCPState)
        , _asset (asset)
-       , _frame_buffer_offset (0)
        , _sync (sync)
-       , _sync_packet (0)
+       , _active_channels (active_channels)
 {
        DCP_ASSERT (!_sync || _asset->channels() >= 14);
-       DCP_ASSERT (!_sync || _asset->standard() == SMPTE);
+       DCP_ASSERT (!_sync || _asset->standard() == Standard::SMPTE);
 
        /* Derived from ASDCP::Wav::SimpleWaveHeader::FillADesc */
        _state->desc.EditRate = ASDCP::Rational (_asset->edit_rate().numerator, _asset->edit_rate().denominator);
        _state->desc.AudioSamplingRate = ASDCP::Rational (_asset->sampling_rate(), 1);
        _state->desc.Locked = 0;
-       _state->desc.ChannelCount = _asset->channels ();
+       _state->desc.ChannelCount = _asset->channels();
        _state->desc.QuantizationBits = 24;
        _state->desc.BlockAlign = 3 * _asset->channels();
        _state->desc.AvgBps = _asset->sampling_rate() * _state->desc.BlockAlign;
        _state->desc.LinkedTrackID = 0;
-       if (asset->standard() == INTEROP) {
+       if (asset->standard() == Standard::INTEROP) {
                _state->desc.ChannelFormat = ASDCP::PCM::CF_NONE;
        } else {
                /* Just use WTF ("wild track format") for SMPTE for now; searches suggest that this
@@ -101,6 +110,62 @@ SoundAssetWriter::SoundAssetWriter (SoundAsset* asset, boost::filesystem::path f
        }
 }
 
+
+void
+SoundAssetWriter::start ()
+{
+       auto r = _state->mxf_writer.OpenWrite (_file.string().c_str(), _state->writer_info, _state->desc);
+       if (ASDCP_FAILURE(r)) {
+               boost::throw_exception (FileError("could not open audio MXF for writing", _file.string(), r));
+       }
+
+       if (_asset->standard() == Standard::SMPTE && !_active_channels.empty()) {
+
+               ASDCP::MXF::WaveAudioDescriptor* essence_descriptor = nullptr;
+               _state->mxf_writer.OP1aHeader().GetMDObjectByType(
+                       asdcp_smpte_dict->ul(ASDCP::MDD_WaveAudioDescriptor), reinterpret_cast<ASDCP::MXF::InterchangeObject**>(&essence_descriptor)
+                       );
+               DCP_ASSERT (essence_descriptor);
+               essence_descriptor->ChannelAssignment = asdcp_smpte_dict->ul(ASDCP::MDD_DCAudioChannelCfg_MCA);
+
+               auto soundfield = new ASDCP::MXF::SoundfieldGroupLabelSubDescriptor(asdcp_smpte_dict);
+               GenRandomValue (soundfield->MCALinkID);
+               soundfield->RFC5646SpokenLanguage = _asset->language();
+
+               const MCASoundField field = _asset->channels() > 10 ? MCASoundField::SEVEN_POINT_ONE : MCASoundField::FIVE_POINT_ONE;
+
+               if (field == MCASoundField::SEVEN_POINT_ONE) {
+                       soundfield->MCATagSymbol = "sg71";
+                       soundfield->MCATagName = "7.1DS";
+                       soundfield->MCALabelDictionaryID = asdcp_smpte_dict->ul(ASDCP::MDD_DCAudioSoundfield_71);
+               } else {
+                       soundfield->MCATagSymbol = "sg51";
+                       soundfield->MCATagName = "5.1";
+                       soundfield->MCALabelDictionaryID = asdcp_smpte_dict->ul(ASDCP::MDD_DCAudioSoundfield_51);
+               }
+
+               _state->mxf_writer.OP1aHeader().AddChildObject(soundfield);
+               essence_descriptor->SubDescriptors.push_back(soundfield->InstanceUID);
+
+               for (auto i: _active_channels) {
+                       auto channel = new ASDCP::MXF::AudioChannelLabelSubDescriptor(asdcp_smpte_dict);
+                       GenRandomValue (channel->MCALinkID);
+                       channel->SoundfieldGroupLinkID = soundfield->MCALinkID;
+                       channel->MCAChannelID = static_cast<int>(i) + 1;
+                       channel->MCATagSymbol = "ch" + channel_to_mca_id(i, field);
+                       channel->MCATagName = channel_to_mca_name(i, field);
+                       channel->RFC5646SpokenLanguage = _asset->language();
+                       channel->MCALabelDictionaryID = channel_to_mca_universal_label(i, field, asdcp_smpte_dict);
+                       _state->mxf_writer.OP1aHeader().AddChildObject(channel);
+                       essence_descriptor->SubDescriptors.push_back(channel->InstanceUID);
+               }
+       }
+
+       _asset->set_file (_file);
+       _started = true;
+}
+
+
 void
 SoundAssetWriter::write (float const * const * data, int frames)
 {
@@ -110,13 +175,7 @@ SoundAssetWriter::write (float const * const * data, int frames)
        static float const clip = 1.0f - (1.0f / pow (2, 23));
 
        if (!_started) {
-               Kumu::Result_t r = _state->mxf_writer.OpenWrite (_file.string().c_str(), _state->writer_info, _state->desc);
-               if (ASDCP_FAILURE (r)) {
-                       boost::throw_exception (FileError ("could not open audio MXF for writing", _file.string(), r));
-               }
-
-               _asset->set_file (_file);
-               _started = true;
+               start ();
        }
 
        int const ch = _asset->channels ();
@@ -146,7 +205,7 @@ SoundAssetWriter::write (float const * const * data, int frames)
                }
                _frame_buffer_offset += 3 * ch;
 
-               DCP_ASSERT (_frame_buffer_offset <= int (_state->frame_buffer.Capacity()));
+               DCP_ASSERT (_frame_buffer_offset <= int(_state->frame_buffer.Capacity()));
 
                /* Finish the MXF frame if required */
                if (_frame_buffer_offset == int (_state->frame_buffer.Capacity())) {
@@ -160,9 +219,9 @@ SoundAssetWriter::write (float const * const * data, int frames)
 void
 SoundAssetWriter::write_current_frame ()
 {
-       ASDCP::Result_t const r = _state->mxf_writer.WriteFrame (_state->frame_buffer, _crypto_context->context(), _crypto_context->hmac());
-       if (ASDCP_FAILURE (r)) {
-               boost::throw_exception (MiscError (String::compose ("could not write audio MXF frame (%1)", int (r))));
+       auto const r = _state->mxf_writer.WriteFrame (_state->frame_buffer, _crypto_context->context(), _crypto_context->hmac());
+       if (ASDCP_FAILURE(r)) {
+               boost::throw_exception (MiscError(String::compose("could not write audio MXF frame (%1)", static_cast<int>(r))));
        }
 
        ++_frames_written;
@@ -181,9 +240,9 @@ SoundAssetWriter::finalize ()
        }
 
        if (_started) {
-               ASDCP::Result_t const r = _state->mxf_writer.Finalize();
+               auto const r = _state->mxf_writer.Finalize();
                if (ASDCP_FAILURE(r)) {
-                       boost::throw_exception (MiscError (String::compose ("could not finalise audio MXF (%1)", int(r))));
+                       boost::throw_exception (MiscError(String::compose ("could not finalise audio MXF (%1)", static_cast<int>(r))));
                }
        }
 
@@ -205,7 +264,7 @@ SoundAssetWriter::create_sync_packets ()
        int remaining_bits = 0;
        /* How many packets in this edit unit (i.e. "frame") */
        int packets = 0;
-       Fraction const edit_rate = _asset->edit_rate ();
+       auto const edit_rate = _asset->edit_rate ();
        if (edit_rate == Fraction(24, 1)) {
                edit_rate_code = 0;
                remaining_bits = 25;