9387ec896b67b92f072a3d31b0509a4373dc03e3
[dcpomatic.git] / src / lib / audio_analysis.h
1 /*
2     Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #ifndef DCPOMATIC_AUDIO_ANALYSIS_H
21 #define DCPOMATIC_AUDIO_ANALYSIS_H
22
23 #include <vector>
24 #include <list>
25 #include <boost/filesystem.hpp>
26 #include <boost/optional.hpp>
27 #include <libcxml/cxml.h>
28 #include "types.h"
29
30 class AudioPoint
31 {
32 public:
33         enum Type {
34                 PEAK,
35                 RMS,
36                 COUNT
37         };
38
39         AudioPoint ();
40         AudioPoint (cxml::ConstNodePtr node);
41         AudioPoint (AudioPoint const &);
42         AudioPoint& operator= (AudioPoint const &);
43
44         void as_xml (xmlpp::Element *) const;
45         
46         float& operator[] (int t) {
47                 return _data[t];
48         }
49
50 private:
51         float _data[COUNT];
52 };
53
54 class AudioAnalysis : public boost::noncopyable
55 {
56 public:
57         AudioAnalysis (int c);
58         AudioAnalysis (boost::filesystem::path);
59
60         void add_point (int c, AudioPoint const & p);
61         void set_peak (float peak, DCPTime time) {
62                 _peak = peak;
63                 _peak_time = time;
64         }
65         
66         AudioPoint get_point (int c, int p) const;
67         int points (int c) const;
68         int channels () const;
69
70         boost::optional<float> peak () const {
71                 return _peak;
72         }
73
74         boost::optional<DCPTime> peak_time () const {
75                 return _peak_time;
76         }
77
78         void write (boost::filesystem::path);
79
80 private:
81         std::vector<std::vector<AudioPoint> > _data;
82         boost::optional<float> _peak;
83         boost::optional<DCPTime> _peak_time;
84 };
85
86 #endif