summaryrefslogtreecommitdiff
path: root/src/lib/audio_analysis.cc
blob: 9d708bbfdd07f30e07327781cc328828a4ecce06 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*
    Copyright (C) 2012 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.

*/

#include <stdint.h>
#include <cmath>
#include <cassert>
#include <fstream>
#include <boost/filesystem.hpp>
#include "audio_analysis.h"

using std::ostream;
using std::istream;
using std::string;
using std::ofstream;
using std::ifstream;
using std::vector;
using std::cout;
using std::max;
using std::list;

AudioPoint::AudioPoint ()
{
	for (int i = 0; i < COUNT; ++i) {
		_data[i] = 0;
	}
}

AudioPoint::AudioPoint (istream& s)
{
	for (int i = 0; i < COUNT; ++i) {
		s >> _data[i];
	}
}

void
AudioPoint::write (ostream& s) const
{
	for (int i = 0; i < COUNT; ++i) {
		s << _data[i] << "\n";
	}
}
	

AudioAnalysis::AudioAnalysis (int channels)
{
	_data.resize (channels);
}

AudioAnalysis::AudioAnalysis (string filename)
{
	ifstream f (filename.c_str ());

	int channels;
	f >> channels;
	_data.resize (channels);

	for (int i = 0; i < channels; ++i) {
		int points;
		f >> points;
		for (int j = 0; j < points; ++j) {
			_data[i].push_back (AudioPoint (f));
		}
	}
}

void
AudioAnalysis::add_point (int c, AudioPoint const & p)
{
	assert (c < channels ());
	_data[c].push_back (p);
}

AudioPoint
AudioAnalysis::get_point (int c, int p) const
{
	assert (p < points (c));
	return _data[c][p];
}

int
AudioAnalysis::channels () const
{
	return _data.size ();
}

int
AudioAnalysis::points (int c) const
{
	assert (c < channels ());
	return _data[c].size ();
}

void
AudioAnalysis::write (string filename)
{
	string tmp = filename + ".tmp";
	
	ofstream f (tmp.c_str ());
	f << _data.size() << "\n";
	for (vector<vector<AudioPoint> >::iterator i = _data.begin(); i != _data.end(); ++i) {
		f << i->size () << "\n";
		for (vector<AudioPoint>::iterator j = i->begin(); j != i->end(); ++j) {
			j->write (f);
		}
	}

	f.close ();
	boost::filesystem::rename (tmp, filename);
}