X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Faudio_analysis.h;h=263e8a8a08fc1e86fd71376f6bf62b2d31df0c83;hb=3339d3bce70afe9ae2ca10e9fcfc4b54b748fbf4;hp=2411b4316c7f1474ff363b9e71e1baaee82ddaf6;hpb=e60bb3e51bd1508b149e6b8f6608f09b5196ae26;p=dcpomatic.git

diff --git a/src/lib/audio_analysis.h b/src/lib/audio_analysis.h
index 2411b4316..263e8a8a0 100644
--- a/src/lib/audio_analysis.h
+++ b/src/lib/audio_analysis.h
@@ -1,86 +1,155 @@
 /*
-    Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2021 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
 
-    This program is distributed in the hope that it will be useful,
+    DCP-o-matic is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU General Public License for more details.
 
     You should have received a copy of the GNU General Public License
-    along with this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
+
 #ifndef DCPOMATIC_AUDIO_ANALYSIS_H
 #define DCPOMATIC_AUDIO_ANALYSIS_H
 
-#include <vector>
-#include <list>
-#include <boost/filesystem.hpp>
-#include <boost/optional.hpp>
+
+#include "dcpomatic_time.h"
+#include "audio_point.h"
 #include <libcxml/cxml.h>
-#include "types.h"
+#include <boost/optional.hpp>
+#include <boost/filesystem.hpp>
+#include <vector>
 
-class AudioPoint
-{
-public:
-	enum Type {
-		PEAK,
-		RMS,
-		COUNT
-	};
 
-	AudioPoint ();
-	AudioPoint (cxml::ConstNodePtr node);
-	AudioPoint (AudioPoint const &);
-	AudioPoint& operator= (AudioPoint const &);
+namespace xmlpp {
+	class Element;
+}
 
-	void as_xml (xmlpp::Element *) const;
 
-	float& operator[] (int t) {
-		return _data[t];
-	}
+class Playlist;
 
-private:
-	float _data[COUNT];
-};
 
 class AudioAnalysis : public boost::noncopyable
 {
 public:
-	AudioAnalysis (int c);
-	AudioAnalysis (boost::filesystem::path);
+	explicit AudioAnalysis (int c);
+	explicit AudioAnalysis (boost::filesystem::path);
 
 	void add_point (int c, AudioPoint const & p);
-	void set_peak (float peak, DCPTime time) {
-		_peak = peak;
-		_peak_time = time;
+
+	struct PeakTime {
+		PeakTime (float p, dcpomatic::DCPTime t)
+			: peak (p)
+			, time (t)
+		{}
+
+		float peak;
+		dcpomatic::DCPTime time;
+	};
+
+	void set_sample_peak (std::vector<PeakTime> peak) {
+		_sample_peak = peak;
+	}
+
+	void set_true_peak (std::vector<float> peak) {
+		_true_peak = peak;
+	}
+
+	void set_integrated_loudness (float l) {
+		_integrated_loudness = l;
+	}
+
+	void set_loudness_range (float r) {
+		_loudness_range = r;
 	}
 
 	AudioPoint get_point (int c, int p) const;
 	int points (int c) const;
 	int channels () const;
 
-	boost::optional<float> peak () const {
-		return _peak;
+	std::vector<PeakTime> sample_peak () const {
+		return _sample_peak;
 	}
 
-	boost::optional<DCPTime> peak_time () const {
-		return _peak_time;
+	std::pair<PeakTime, int> overall_sample_peak () const;
+
+	std::vector<float> true_peak () const {
+		return _true_peak;
+	}
+
+	boost::optional<float> overall_true_peak () const;
+
+	boost::optional<float> integrated_loudness () const {
+		return _integrated_loudness;
+	}
+
+	boost::optional<float> loudness_range () const {
+		return _loudness_range;
+	}
+
+	boost::optional<double> analysis_gain () const {
+		return _analysis_gain;
+	}
+
+	void set_analysis_gain (double gain) {
+		_analysis_gain = gain;
+	}
+
+	int64_t samples_per_point () const {
+		return _samples_per_point;
+	}
+
+	void set_samples_per_point (int64_t spp) {
+		_samples_per_point = spp;
+	}
+
+	int sample_rate () const {
+		return _sample_rate;
+	}
+
+	void set_sample_rate (int sr) {
+		_sample_rate = sr;
+	}
+
+	void set_leqm (double leqm) {
+		_leqm = leqm;
+	}
+
+	boost::optional<double> leqm () const {
+		return _leqm;
 	}
 
 	void write (boost::filesystem::path);
 
+	float gain_correction (std::shared_ptr<const Playlist> playlist);
+
 private:
-	std::vector<std::vector<AudioPoint> > _data;
-	boost::optional<float> _peak;
-	boost::optional<DCPTime> _peak_time;
+	std::vector<std::vector<AudioPoint>> _data;
+	std::vector<PeakTime> _sample_peak;
+	std::vector<float> _true_peak;
+	boost::optional<float> _integrated_loudness;
+	boost::optional<float> _loudness_range;
+	boost::optional<double> _leqm;
+	/** If this analysis was run on a single piece of
+	 *  content we store its gain in dB when the analysis
+	 *  happened.
+	 */
+	boost::optional<double> _analysis_gain;
+	int64_t _samples_per_point = 0;
+	int _sample_rate = 0;
+
+	static int const _current_state_version;
 };
 
+
 #endif