Various work on audio mapping.
[dcpomatic.git] / src / lib / audio_mapping.cc
index 3fc423e101a891074eada7261bff77d622311cba..fc3909a8aa0ab5240dd7d17b968bf102e2783d70 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2013-2015 Carl Hetherington <cth@carlh.net>
 
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
 */
 
 #include "audio_mapping.h"
+#include "util.h"
+#include "md5_digester.h"
+#include "raw_convert.h"
+#include <libcxml/cxml.h>
+#include <libxml++/libxml++.h>
+
+using std::list;
+using std::cout;
+using std::make_pair;
+using std::pair;
+using std::string;
+using std::min;
+using std::vector;
+using boost::shared_ptr;
+using boost::dynamic_pointer_cast;
+
+AudioMapping::AudioMapping ()
+       : _input_channels (0)
+       , _output_channels (0)
+{
 
-using std::map;
-using boost::optional;
+}
 
-AutomaticAudioMapping::AutomaticAudioMapping (int c)
-       : _source_channels (c)
+/** Create an empty AudioMapping.
+ *  @param input_channels Number of input channels.
+ *  @param output_channels Number of output channels.
+ */
+AudioMapping::AudioMapping (int input_channels, int output_channels)
 {
-
+       setup (input_channels, output_channels);
 }
 
-optional<libdcp::Channel>
-AutomaticAudioMapping::source_to_dcp (int c) const
+void
+AudioMapping::setup (int input_channels, int output_channels)
 {
-       if (c >= _source_channels) {
-               return optional<libdcp::Channel> ();
+       _input_channels = input_channels;
+       _output_channels = output_channels;
+       
+       _gain.resize (_input_channels);
+       for (int i = 0; i < _input_channels; ++i) {
+               _gain[i].resize (_output_channels);
        }
 
-       if (_source_channels == 1) {
-               /* mono sources to centre */
-               return libdcp::CENTRE;
-       }
-       
-       return static_cast<libdcp::Channel> (c);
+       make_zero ();
 }
 
-optional<int>
-AutomaticAudioMapping::dcp_to_source (libdcp::Channel c) const
+void
+AudioMapping::make_zero ()
 {
-       if (_source_channels == 1) {
-               if (c == libdcp::CENTRE) {
-                       return 0;
-               } else {
-                       return optional<int> ();
+       for (int i = 0; i < _input_channels; ++i) {
+               for (int j = 0; j < _output_channels; ++j) {
+                       _gain[i][j] = 0;
                }
        }
+}
 
-       if (static_cast<int> (c) >= _source_channels) {
-               return optional<int> ();
+AudioMapping::AudioMapping (cxml::ConstNodePtr node, int state_version)
+{
+       if (state_version < 32) {
+               setup (node->number_child<int> ("ContentChannels"), MAX_DCP_AUDIO_CHANNELS);
+       } else {
+               setup (node->number_child<int> ("InputChannels"), node->number_child<int> ("OutputChannels"));
+       }
+
+       if (state_version <= 5) {
+               /* Old-style: on/off mapping */
+               list<cxml::NodePtr> const c = node->node_children ("Map");
+               for (list<cxml::NodePtr>::const_iterator i = c.begin(); i != c.end(); ++i) {
+                       set ((*i)->number_child<int> ("ContentIndex"), static_cast<dcp::Channel> ((*i)->number_child<int> ("DCP")), 1);
+               }
+       } else {
+               list<cxml::NodePtr> const c = node->node_children ("Gain");
+               for (list<cxml::NodePtr>::const_iterator i = c.begin(); i != c.end(); ++i) {
+                       if (state_version < 32) {
+                               set (
+                                       (*i)->number_attribute<int> ("Content"),
+                                       static_cast<dcp::Channel> ((*i)->number_attribute<int> ("DCP")),
+                                       raw_convert<float> ((*i)->content ())
+                                       );
+                       } else {
+                               set (
+                                       (*i)->number_attribute<int> ("Input"),
+                                       (*i)->number_attribute<int> ("Output"),
+                                       raw_convert<float> ((*i)->content ())
+                                       );
+                       }
+               }
        }
-       
-       return static_cast<int> (c);
 }
 
-int
-AutomaticAudioMapping::dcp_channels () const
+void
+AudioMapping::set (int input_channel, int output_channel, float g)
 {
-       if (_source_channels == 1) {
-               /* The source is mono, so to put the mono channel into
-                  the centre we need to generate a 5.1 soundtrack.
-               */
-               return 6;
-       }
+       _gain[input_channel][output_channel] = g;
+}
 
-       return _source_channels;
+float
+AudioMapping::get (int input_channel, int output_channel) const
+{
+       return _gain[input_channel][output_channel];
 }
 
-optional<int>
-ConfiguredAudioMapping::dcp_to_source (libdcp::Channel c) const
+void
+AudioMapping::as_xml (xmlpp::Node* node) const
 {
-       map<int, libdcp::Channel>::const_iterator i = _source_to_dcp.begin ();
-       while (i != _source_to_dcp.end() && i->second != c) {
-               ++i;
+       node->add_child ("InputChannels")->add_child_text (raw_convert<string> (_input_channels));
+       node->add_child ("OutputChannels")->add_child_text (raw_convert<string> (_output_channels));
+
+       for (int c = 0; c < _input_channels; ++c) {
+               for (int d = 0; d < _output_channels; ++d) {
+                       xmlpp::Element* t = node->add_child ("Gain");
+                       t->set_attribute ("Input", raw_convert<string> (c));
+                       t->set_attribute ("Output", raw_convert<string> (d));
+                       t->add_child_text (raw_convert<string> (get (c, d)));
+               }
        }
+}
 
-       if (i == _source_to_dcp.end ()) {
-               return boost::none;
+/** @return a string which is unique for a given AudioMapping configuration, for
+ *  differentiation between different AudioMappings.
+ */
+string
+AudioMapping::digest () const
+{
+       MD5Digester digester;
+       digester.add (_input_channels);
+       digester.add (_output_channels);
+       for (int i = 0; i < _input_channels; ++i) {
+               for (int j = 0; j < _output_channels; ++j) {
+                       digester.add (_gain[i][j]);
+               }
        }
 
-       return i->first;
+       return digester.get ();
 }
 
-optional<libdcp::Channel>
-ConfiguredAudioMapping::source_to_dcp (int c) const
+list<int>
+AudioMapping::mapped_output_channels () const
 {
-       map<int, libdcp::Channel>::const_iterator i = _source_to_dcp.find (c);
-       if (i == _source_to_dcp.end ()) {
-               return boost::none;
+       static float const minus_96_db = 0.000015849;
+
+       list<int> mapped;
+       
+       for (vector<vector<float> >::const_iterator i = _gain.begin(); i != _gain.end(); ++i) {
+               for (size_t j = 0; j < i->size(); ++j) {
+                       if (abs ((*i)[j]) > minus_96_db) {
+                               mapped.push_back (j);
+                       }
+               }
        }
 
-       return i->second;
+       mapped.sort ();
+       mapped.unique ();
+       
+       return mapped;
 }
 
-       
+void
+AudioMapping::unmap_all ()
+{
+       for (vector<vector<float> >::iterator i = _gain.begin(); i != _gain.end(); ++i) {
+               for (vector<float>::iterator j = i->begin(); j != i->end(); ++j) {
+                       *j = 0;
+               }
+       }
+}