X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=src%2Flib%2Faudio_analysis.h;h=e5ee24d49918a52e19bbcaaa6ee9c59acc6bdaa8;hp=b91a1cf5123d3c43c3a1b40418e279ff273b4630;hb=HEAD;hpb=3e12c68dc0451e73b5bc1a84d1d70f4999f7b4b5

diff --git a/src/lib/audio_analysis.h b/src/lib/audio_analysis.h
index b91a1cf51..038059502 100644
--- a/src/lib/audio_analysis.h
+++ b/src/lib/audio_analysis.h
@@ -1,84 +1,155 @@
 /*
-    Copyright (C) 2012-2014 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/>.
 
 */
 
-/** @file  src/lib/audio_analysis.h
- *  @brief AudioAnalysis and AudioPoint classes.
- */
 
 #ifndef DCPOMATIC_AUDIO_ANALYSIS_H
 #define DCPOMATIC_AUDIO_ANALYSIS_H
 
-#include <vector>
-#include <list>
+
+#include "dcpomatic_time.h"
+#include "audio_point.h"
+#include <libcxml/cxml.h>
+#include <boost/optional.hpp>
 #include <boost/filesystem.hpp>
+#include <vector>
+
+
+namespace xmlpp {
+	class Element;
+}
+
 
-/** @class AudioPoint
- *  @brief A single point of an audio analysis for one portion of one channel.
- */
-class AudioPoint
+class Playlist;
+
+
+class AudioAnalysis
 {
 public:
-	enum Type {
-		PEAK,
-		RMS,
-		COUNT
+	explicit AudioAnalysis (int c);
+	explicit AudioAnalysis (boost::filesystem::path);
+
+	void add_point (int c, AudioPoint const & p);
+
+	struct PeakTime {
+		PeakTime (float p, dcpomatic::DCPTime t)
+			: peak (p)
+			, time (t)
+		{}
+
+		float peak;
+		dcpomatic::DCPTime time;
 	};
 
-	AudioPoint ();
-	AudioPoint (FILE *);
-	AudioPoint (AudioPoint const &);
-	AudioPoint& operator= (AudioPoint const &);
+	void set_sample_peak (std::vector<PeakTime> peak) {
+		_sample_peak = peak;
+	}
 
-	void write (FILE *) const;
-	
-	float& operator[] (int t) {
-		return _data[t];
+	void set_true_peak (std::vector<float> peak) {
+		_true_peak = peak;
 	}
 
-private:
-	float _data[COUNT];
-};
+	void set_integrated_loudness (float l) {
+		_integrated_loudness = l;
+	}
 
-/** @class AudioAnalysis
- *  @brief An analysis of the audio data in a piece of AudioContent.
- *
- *  This is a set of AudioPoints for each channel.  The AudioPoints
- *  each represent some measurement of the audio over a portion of the
- *  content.  For example each AudioPoint may give the RMS level of
- *  a 1-minute portion of the audio.
- */
-class AudioAnalysis : public boost::noncopyable
-{
-public:
-	AudioAnalysis (int c);
-	AudioAnalysis (boost::filesystem::path);
+	void set_loudness_range (float r) {
+		_loudness_range = r;
+	}
 
-	void add_point (int c, AudioPoint const & p);
-	
 	AudioPoint get_point (int c, int p) const;
 	int points (int c) const;
 	int channels () const;
 
+	std::vector<PeakTime> sample_peak () const {
+		return _sample_peak;
+	}
+
+	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;
+	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