c2d8db8760622662a59e11541dbdfbf874aa336c
[dcpomatic.git] / src / lib / audio_analysis.h
1 /*
2     Copyright (C) 2012 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 DVDOMATIC_AUDIO_ANALYSIS_H
21 #define DVDOMATIC_AUDIO_ANALYSIS_H
22
23 #include <iostream>
24 #include <vector>
25
26 class AudioPoint
27 {
28 public:
29         enum Type {
30                 PEAK,
31                 RMS,
32                 COUNT
33         };
34
35         AudioPoint ();
36         AudioPoint (std::istream &);
37
38         void write (std::ostream &) const;
39         
40         float& operator[] (int t) {
41                 return _data[t];
42         }
43
44 private:
45         float _data[COUNT];
46 };
47
48 class AudioAnalysis
49 {
50 public:
51         AudioAnalysis (int c);
52         AudioAnalysis (std::string);
53
54         void add_point (int c, AudioPoint const & p);
55         
56         AudioPoint get_point (int c, int p) const;
57         int points (int c) const;
58         int channels () const;
59
60         void write (std::string);
61
62
63 private:
64         std::vector<std::vector<AudioPoint> > _data;
65 };
66
67 #endif