diff options
Diffstat (limited to 'src/lib/analyse_audio_job.cc')
| -rw-r--r-- | src/lib/analyse_audio_job.cc | 108 |
1 files changed, 108 insertions, 0 deletions
diff --git a/src/lib/analyse_audio_job.cc b/src/lib/analyse_audio_job.cc new file mode 100644 index 000000000..8186f9de4 --- /dev/null +++ b/src/lib/analyse_audio_job.cc @@ -0,0 +1,108 @@ +/* + 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 "audio_analysis.h" +#include "analyse_audio_job.h" +#include "compose.hpp" +#include "film.h" +#include "player.h" + +#include "i18n.h" + +using std::string; +using std::max; +using std::min; +using std::cout; +using boost::shared_ptr; + +int const AnalyseAudioJob::_num_points = 1024; + +AnalyseAudioJob::AnalyseAudioJob (shared_ptr<const Film> f, shared_ptr<AudioContent> c) + : Job (f) + , _content (c) + , _done (0) + , _samples_per_point (1) +{ + +} + +string +AnalyseAudioJob::name () const +{ + return _("Analyse audio"); +} + +void +AnalyseAudioJob::run () +{ + shared_ptr<AudioContent> content = _content.lock (); + if (!content) { + return; + } + + shared_ptr<Playlist> playlist (new Playlist); + playlist->add (content); + shared_ptr<Player> player (new Player (_film, playlist)); + player->disable_video (); + + player->Audio.connect (bind (&AnalyseAudioJob::audio, this, _1, _2)); + + _samples_per_point = max (int64_t (1), _film->time_to_audio_frames (_film->length()) / _num_points); + + _current.resize (_film->audio_channels ()); + _analysis.reset (new AudioAnalysis (_film->audio_channels ())); + + _done = 0; + OutputAudioFrame const len = _film->time_to_audio_frames (_film->length ()); + while (!player->pass ()) { + set_progress (double (_done) / len); + } + + _analysis->write (content->audio_analysis_path ()); + + set_progress (1); + set_state (FINISHED_OK); +} + +void +AnalyseAudioJob::audio (shared_ptr<const AudioBuffers> b, Time) +{ + for (int i = 0; i < b->frames(); ++i) { + for (int j = 0; j < b->channels(); ++j) { + float s = b->data(j)[i]; + if (fabsf (s) < 10e-7) { + /* stringstream can't serialise and recover inf or -inf, so prevent such + values by replacing with this (140dB down) */ + s = 10e-7; + } + _current[j][AudioPoint::RMS] += pow (s, 2); + _current[j][AudioPoint::PEAK] = max (_current[j][AudioPoint::PEAK], fabsf (s)); + + if ((_done % _samples_per_point) == 0) { + _current[j][AudioPoint::RMS] = sqrt (_current[j][AudioPoint::RMS] / _samples_per_point); + _analysis->add_point (j, _current[j]); + + _current[j] = AudioPoint (); + } + } + + ++_done; + } +} + |