C++11 tidying.
[dcpomatic.git] / src / lib / audio_analysis.cc
index b133dec3def90ab2d74517364d24ea637ed27cc2..b16fa708ad85d74bad44d134b2302745b4d8602c 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
 
 */
 
+
 #include "audio_analysis.h"
 #include "cross.h"
 #include "util.h"
-#include "raw_convert.h"
 #include "playlist.h"
 #include "audio_content.h"
+#include "warnings.h"
+#include <dcp/raw_convert.h>
+DCPOMATIC_DISABLE_WARNINGS
 #include <libxml++/libxml++.h>
+DCPOMATIC_ENABLE_WARNINGS
 #include <boost/filesystem.hpp>
-#include <boost/foreach.hpp>
-#include <boost/make_shared.hpp>
 #include <stdint.h>
 #include <cmath>
 #include <cstdio>
 #include <iostream>
 #include <inttypes.h>
 
-using std::ostream;
+
+using std::cout;
+using std::dynamic_pointer_cast;
 using std::istream;
+using std::list;
+using std::make_pair;
+using std::make_shared;
+using std::max;
+using std::ostream;
+using std::pair;
+using std::shared_ptr;
 using std::string;
 using std::vector;
-using std::cout;
-using std::max;
-using std::list;
-using boost::shared_ptr;
-using boost::make_shared;
-using boost::dynamic_pointer_cast;
+using boost::optional;
+using dcp::raw_convert;
+using namespace dcpomatic;
+
+
+int const AudioAnalysis::_current_state_version = 3;
+
 
 AudioAnalysis::AudioAnalysis (int channels)
 {
        _data.resize (channels);
 }
 
+
 AudioAnalysis::AudioAnalysis (boost::filesystem::path filename)
 {
        cxml::Document f ("AudioAnalysis");
        f.read_file (filename);
 
-       BOOST_FOREACH (cxml::NodePtr i, f.node_children ("Channel")) {
+       if (f.optional_number_child<int>("Version").get_value_or(1) < _current_state_version) {
+               /* Too old.  Throw an exception so that this analysis is re-run. */
+               throw OldFormatError ("Audio analysis file is too old");
+       }
+
+       for (auto i: f.node_children("Channel")) {
                vector<AudioPoint> channel;
 
-               BOOST_FOREACH (cxml::NodePtr j, i->node_children ("Point")) {
-                       channel.push_back (AudioPoint (j));
+               for (auto j: i->node_children("Point")) {
+                       channel.push_back (AudioPoint(j));
                }
 
                _data.push_back (channel);
        }
 
-       _sample_peak = f.optional_number_child<float> ("Peak");
-       if (!_sample_peak) {
-               /* New key */
-               _sample_peak = f.optional_number_child<float> ("SamplePeak");
+       for (auto i: f.node_children ("SamplePeak")) {
+               _sample_peak.push_back (
+                       PeakTime(
+                               dcp::raw_convert<float>(i->content()), DCPTime(i->number_attribute<Frame>("Time"))
+                               )
+                       );
        }
 
-       if (f.optional_number_child<DCPTime::Type> ("PeakTime")) {
-               _sample_peak_time = DCPTime (f.number_child<DCPTime::Type> ("PeakTime"));
-       } else if (f.optional_number_child<DCPTime::Type> ("SamplePeakTime")) {
-               _sample_peak_time = DCPTime (f.number_child<DCPTime::Type> ("SamplePeakTime"));
+       for (auto i: f.node_children("TruePeak")) {
+               _true_peak.push_back (dcp::raw_convert<float>(i->content()));
        }
 
-       _true_peak = f.optional_number_child<float> ("TruePeak");
-       _integrated_loudness = f.optional_number_child<float> ("IntegratedLoudness");
-       _loudness_range = f.optional_number_child<float> ("LoudnessRange");
+       _integrated_loudness = f.optional_number_child<float>("IntegratedLoudness");
+       _loudness_range = f.optional_number_child<float>("LoudnessRange");
+
+       _analysis_gain = f.optional_number_child<double>("AnalysisGain");
+       _samples_per_point = f.number_child<int64_t>("SamplesPerPoint");
+       _sample_rate = f.number_child<int64_t>("SampleRate");
 
-       _analysis_gain = f.optional_number_child<double> ("AnalysisGain");
+       _leqm = f.optional_number_child<double>("Leqm");
 }
 
+
 void
 AudioAnalysis::add_point (int c, AudioPoint const & p)
 {
@@ -91,46 +113,53 @@ AudioAnalysis::add_point (int c, AudioPoint const & p)
        _data[c].push_back (p);
 }
 
+
 AudioPoint
 AudioAnalysis::get_point (int c, int p) const
 {
-       DCPOMATIC_ASSERT (p < points (c));
+       DCPOMATIC_ASSERT (p < points(c));
        return _data[c][p];
 }
 
+
 int
 AudioAnalysis::channels () const
 {
        return _data.size ();
 }
 
+
 int
 AudioAnalysis::points (int c) const
 {
-       DCPOMATIC_ASSERT (c < channels ());
+       DCPOMATIC_ASSERT (c < channels());
        return _data[c].size ();
 }
 
+
 void
 AudioAnalysis::write (boost::filesystem::path filename)
 {
-       shared_ptr<xmlpp::Document> doc = make_shared<xmlpp::Document> ();
+       auto doc = make_shared<xmlpp::Document>();
        xmlpp::Element* root = doc->create_root_node ("AudioAnalysis");
 
-       BOOST_FOREACH (vector<AudioPoint>& i, _data) {
-               xmlpp::Element* channel = root->add_child ("Channel");
-               BOOST_FOREACH (AudioPoint& j, i) {
+       root->add_child("Version")->add_child_text(raw_convert<string>(_current_state_version));
+
+       for (auto& i: _data) {
+               auto channel = root->add_child ("Channel");
+               for (auto& j: i) {
                        j.as_xml (channel->add_child ("Point"));
                }
        }
 
-       if (_sample_peak) {
-               root->add_child("SamplePeak")->add_child_text (raw_convert<string> (_sample_peak.get ()));
-               root->add_child("SamplePeakTime")->add_child_text (raw_convert<string> (_sample_peak_time.get().get ()));
+       for (size_t i = 0; i < _sample_peak.size(); ++i) {
+               auto n = root->add_child("SamplePeak");
+               n->add_child_text (raw_convert<string> (_sample_peak[i].peak));
+               n->set_attribute ("Time", raw_convert<string> (_sample_peak[i].time.get()));
        }
 
-       if (_true_peak) {
-               root->add_child("TruePeak")->add_child_text (raw_convert<string> (_true_peak.get ()));
+       for (auto i: _true_peak) {
+               root->add_child("TruePeak")->add_child_text (raw_convert<string> (i));
        }
 
        if (_integrated_loudness) {
@@ -145,9 +174,17 @@ AudioAnalysis::write (boost::filesystem::path filename)
                root->add_child("AnalysisGain")->add_child_text (raw_convert<string> (_analysis_gain.get ()));
        }
 
+       root->add_child("SamplesPerPoint")->add_child_text (raw_convert<string> (_samples_per_point));
+       root->add_child("SampleRate")->add_child_text (raw_convert<string> (_sample_rate));
+
+       if (_leqm) {
+               root->add_child("Leqm")->add_child_text(raw_convert<string>(*_leqm));
+       }
+
        doc->write_to_file_formatted (filename.string ());
 }
 
+
 float
 AudioAnalysis::gain_correction (shared_ptr<const Playlist> playlist)
 {
@@ -157,8 +194,44 @@ AudioAnalysis::gain_correction (shared_ptr<const Playlist> playlist)
                   what correction is now needed to make it look `right'.
                */
                DCPOMATIC_ASSERT (playlist->content().front()->audio);
-               return playlist->content().front()->audio->gain() - analysis_gain().get ();
+               return playlist->content().front()->audio->gain() - analysis_gain().get();
        }
 
        return 0.0f;
 }
+
+
+/** @return Peak across all channels, and the channel number it is on */
+pair<AudioAnalysis::PeakTime, int>
+AudioAnalysis::overall_sample_peak () const
+{
+       DCPOMATIC_ASSERT (!_sample_peak.empty());
+
+       optional<PeakTime> pt;
+       int c = 0;
+
+       for (size_t i = 0; i < _sample_peak.size(); ++i) {
+               if (!pt || _sample_peak[i].peak > pt->peak) {
+                       pt = _sample_peak[i];
+                       c = i;
+               }
+       }
+
+       return make_pair (pt.get(), c);
+}
+
+
+optional<float>
+AudioAnalysis::overall_true_peak () const
+{
+       optional<float> p;
+
+       for (auto i: _true_peak) {
+               if (!p || i > *p) {
+                       p = i;
+               }
+       }
+
+       return p;
+}
+