/*
    Copyright (C) 2012-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
    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,
    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.

*/

#ifndef DCPOMATIC_AUDIO_ANALYSIS_H
#define DCPOMATIC_AUDIO_ANALYSIS_H

#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 Playlist;

class AudioAnalysis : public boost::noncopyable
{
public:
	AudioAnalysis (int c);
	AudioAnalysis (boost::filesystem::path);

	void add_point (int c, AudioPoint const & p);

	void set_sample_peak (float peak, DCPTime time) {
		_sample_peak = peak;
		_sample_peak_time = time;
	}

	void set_true_peak (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> sample_peak () const {
		return _sample_peak;
	}

	boost::optional<DCPTime> sample_peak_time () const {
		return _sample_peak_time;
	}

	boost::optional<float> true_peak () const {
		return _true_peak;
	}

	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;
	}

	void write (boost::filesystem::path);

	float gain_correction (boost::shared_ptr<const Playlist> playlist);

private:
	std::vector<std::vector<AudioPoint> > _data;
	boost::optional<float> _sample_peak;
	boost::optional<DCPTime> _sample_peak_time;
	boost::optional<float> _true_peak;
	boost::optional<float> _integrated_loudness;
	boost::optional<float> _loudness_range;
	/** 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;
};

#endif