X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fanalyse_audio_job.cc;h=461b84b4f5dbaf4c2d6884770156550520ae29d2;hb=3df4d6271a6a660fdce143dcd65467c402e98976;hp=588e9fc3d2632c645cff1fa9d2585141f3b76fdc;hpb=36fdb78ea9973d1a797171d762802e707577c960;p=dcpomatic.git diff --git a/src/lib/analyse_audio_job.cc b/src/lib/analyse_audio_job.cc index 588e9fc3d..461b84b4f 100644 --- a/src/lib/analyse_audio_job.cc +++ b/src/lib/analyse_audio_job.cc @@ -1,109 +1,218 @@ /* - Copyright (C) 2012 Carl Hetherington + Copyright (C) 2012-2018 Carl Hetherington - This program is free software; you can redistribute it and/or modify + This file is part of DCP-o-matic. + + DCP-o-matic 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, + DCP-o-matic 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. + along with DCP-o-matic. If not, see . */ #include "audio_analysis.h" +#include "audio_buffers.h" #include "analyse_audio_job.h" +#include "audio_content.h" #include "compose.hpp" #include "film.h" -#include "options.h" -#include "decoder_factory.h" -#include "audio_decoder.h" +#include "player.h" +#include "playlist.h" +#include "filter.h" +#include "audio_filter_graph.h" +#include "config.h" +extern "C" { +#include +#ifdef DCPOMATIC_HAVE_EBUR128_PATCHED_FFMPEG +#include +#endif +} +#include +#include + +#include "i18n.h" using std::string; +using std::vector; using std::max; +using std::min; using std::cout; using boost::shared_ptr; +using boost::dynamic_pointer_cast; int const AnalyseAudioJob::_num_points = 1024; -AnalyseAudioJob::AnalyseAudioJob (shared_ptr f) - : Job (f) +/** @param from_zero true to analyse audio from time 0 in the playlist, otherwise begin at Playlist::start */ +AnalyseAudioJob::AnalyseAudioJob (shared_ptr film, shared_ptr playlist, bool from_zero) + : Job (film) + , _playlist (playlist) + , _from_zero (from_zero) , _done (0) , _samples_per_point (1) + , _current (0) + , _sample_peak (new float[film->audio_channels()]) + , _sample_peak_frame (new Frame[film->audio_channels()]) +#ifdef DCPOMATIC_HAVE_EBUR128_PATCHED_FFMPEG + , _ebur128 (new AudioFilterGraph (film->audio_frame_rate(), film->audio_channels())) +#endif { +#ifdef DCPOMATIC_HAVE_EBUR128_PATCHED_FFMPEG + _filters.push_back (new Filter ("ebur128", "ebur128", "audio", "ebur128=peak=true")); + _ebur128->setup (_filters); +#endif + + for (int i = 0; i < film->audio_channels(); ++i) { + _sample_peak[i] = 0; + _sample_peak_frame[i] = 0; + } + if (!_from_zero) { + _start = _playlist->start().get_value_or(DCPTime()); + } +} + +AnalyseAudioJob::~AnalyseAudioJob () +{ + BOOST_FOREACH (Filter const * i, _filters) { + delete const_cast (i); + } + delete[] _current; + delete[] _sample_peak; + delete[] _sample_peak_frame; } string AnalyseAudioJob::name () const { - return String::compose ("Analyse audio of %1", _film->name()); + return _("Analyse audio"); +} + +string +AnalyseAudioJob::json_name () const +{ + return N_("analyse_audio"); } void AnalyseAudioJob::run () { - if (!_film->audio_stream () || !_film->length()) { - set_progress (1); - set_state (FINISHED_ERROR); - return; + shared_ptr player (new Player (_film, _playlist)); + player->set_ignore_video (); + player->set_ignore_text (); + player->set_fast (); + player->set_play_referenced (); + player->Audio.connect (bind (&AnalyseAudioJob::analyse, this, _1, _2)); + + DCPTime const length = _playlist->length (); + + Frame const len = DCPTime (length - _start).frames_round (_film->audio_frame_rate()); + _samples_per_point = max (int64_t (1), len / _num_points); + + delete[] _current; + _current = new AudioPoint[_film->audio_channels ()]; + _analysis.reset (new AudioAnalysis (_film->audio_channels ())); + + bool has_any_audio = false; + BOOST_FOREACH (shared_ptr c, _playlist->content ()) { + if (c->audio) { + has_any_audio = true; + } } - - DecodeOptions options; - options.decode_video = false; - - Decoders decoders = decoder_factory (_film, options); - assert (decoders.audio); - - decoders.audio->set_audio_stream (_film->audio_stream ()); - decoders.audio->Audio.connect (bind (&AnalyseAudioJob::audio, this, _1)); - - int64_t total_audio_frames = video_frames_to_audio_frames (_film->length().get(), _film->audio_stream()->sample_rate(), _film->frames_per_second()); - _samples_per_point = total_audio_frames / _num_points; - - _current.resize (_film->audio_stream()->channels ()); - _analysis.reset (new AudioAnalysis (_film->audio_stream()->channels())); - - while (!decoders.audio->pass()) { - set_progress (float (_done) / total_audio_frames); + + if (has_any_audio) { + player->seek (_start, true); + _done = 0; + while (!player->pass ()) {} } - _analysis->write (_film->audio_analysis_path ()); - + vector sample_peak; + for (int i = 0; i < _film->audio_channels(); ++i) { + sample_peak.push_back ( + AudioAnalysis::PeakTime (_sample_peak[i], DCPTime::from_frames (_sample_peak_frame[i], _film->audio_frame_rate ())) + ); + } + _analysis->set_sample_peak (sample_peak); + +#ifdef DCPOMATIC_HAVE_EBUR128_PATCHED_FFMPEG + if (Config::instance()->analyse_ebur128 ()) { + void* eb = _ebur128->get("Parsed_ebur128_0")->priv; + vector true_peak; + for (int i = 0; i < _film->audio_channels(); ++i) { + true_peak.push_back (av_ebur128_get_true_peaks(eb)[i]); + } + _analysis->set_true_peak (true_peak); + _analysis->set_integrated_loudness (av_ebur128_get_integrated_loudness(eb)); + _analysis->set_loudness_range (av_ebur128_get_loudness_range(eb)); + } +#endif + + if (_playlist->content().size() == 1) { + /* If there was only one piece of content in this analysis we may later need to know what its + gain was when we analysed it. + */ + shared_ptr ac = _playlist->content().front()->audio; + DCPOMATIC_ASSERT (ac); + _analysis->set_analysis_gain (ac->gain ()); + } + + _analysis->set_samples_per_point (_samples_per_point); + _analysis->set_sample_rate (_film->audio_frame_rate ()); + _analysis->write (_film->audio_analysis_path (_playlist)); + set_progress (1); set_state (FINISHED_OK); } void -AnalyseAudioJob::audio (shared_ptr b) +AnalyseAudioJob::analyse (shared_ptr b, DCPTime 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 + DCPOMATIC_ASSERT (time >= _start); + +#ifdef DCPOMATIC_HAVE_EBUR128_PATCHED_FFMPEG + if (Config::instance()->analyse_ebur128 ()) { + _ebur128->process (b); + } +#endif + + int const frames = b->frames (); + int const channels = b->channels (); + + for (int j = 0; j < channels; ++j) { + float* data = b->data(j); + for (int i = 0; i < frames; ++i) { + float s = data[i]; + float as = fabsf (s); + if (as < 10e-7) { + /* We may struggle to serialise and recover inf or -inf, so prevent such values by replacing with this (140dB down) */ - s = 10e-7; + s = as = 10e-7; } _current[j][AudioPoint::RMS] += pow (s, 2); - _current[j][AudioPoint::PEAK] = max (_current[j][AudioPoint::PEAK], fabsf (s)); + _current[j][AudioPoint::PEAK] = max (_current[j][AudioPoint::PEAK], as); + + if (as > _sample_peak[j]) { + _sample_peak[j] = as; + _sample_peak_frame[j] = _done + i; + } - if ((_done % _samples_per_point) == 0) { - _current[j][AudioPoint::RMS] = 20 * log10 (sqrt (_current[j][AudioPoint::RMS] / _samples_per_point)); - _current[j][AudioPoint::PEAK] = 20 * log10 (_current[j][AudioPoint::PEAK]); + if (((_done + i) % _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; } -} + _done += frames; + + DCPTime const length = _playlist->length (); + set_progress ((time.seconds() - _start.seconds()) / (length.seconds() - _start.seconds())); +}