src/lib/audio_decoder.cc

   1 /*
   2     Copyright (C) 2012-2018 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of DCP-o-matic.
   5
   6     DCP-o-matic is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     DCP-o-matic is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 */
  20
  21 #include "audio_decoder.h"
  22 #include "audio_buffers.h"
  23 #include "audio_content.h"
  24 #include "dcpomatic_log.h"
  25 #include "log.h"
  26 #include "resampler.h"
  27 #include "compose.hpp"
  28 #include <boost/foreach.hpp>
  29 #include <iostream>
  30
  31 #include "i18n.h"
  32
  33 using std::cout;
  34 using std::map;
  35 using std::pair;
  36 using boost::shared_ptr;
  37 using boost::optional;
  38 using namespace dcpomatic;
  39
  40 AudioDecoder::AudioDecoder (Decoder* parent, shared_ptr<const AudioContent> content, bool fast)
  41         : DecoderPart (parent)
  42         , _content (content)
  43         , _fast (fast)
  44 {
  45         /* Set up _positions so that we have one for each stream */
  46         BOOST_FOREACH (AudioStreamPtr i, content->streams ()) {
  47                 _positions[i] = 0;
  48         }
  49 }
  50
  51 void
  52 AudioDecoder::emit (shared_ptr<const Film> film, AudioStreamPtr stream, shared_ptr<const AudioBuffers> data, ContentTime time)
  53 {
  54         if (ignore ()) {
  55                 return;
  56         }
  57
  58         if (_positions[stream] == 0) {
  59                 /* This is the first data we have received since initialisation or seek.  Set
  60                    the position based on the ContentTime that was given.  After this first time
  61                    we just count samples, as it seems that ContentTimes are unreliable from
  62                    FFmpegDecoder (not quite continuous; perhaps due to some rounding error).
  63                 */
  64                 if (_content->delay() > 0) {
  65                         /* Insert silence to give the delay */
  66                         silence (_content->delay ());
  67                 }
  68                 time += ContentTime::from_seconds (_content->delay() / 1000.0);
  69                 _positions[stream] = time.frames_round (_content->resampled_frame_rate(film));
  70         }
  71
  72         shared_ptr<Resampler> resampler;
  73         ResamplerMap::iterator i = _resamplers.find(stream);
  74         if (i != _resamplers.end ()) {
  75                 resampler = i->second;
  76         } else {
  77                 if (stream->frame_rate() != _content->resampled_frame_rate(film)) {
  78                         LOG_GENERAL (
  79                                 "Creating new resampler from %1 to %2 with %3 channels",
  80                                 stream->frame_rate(),
  81                                 _content->resampled_frame_rate(film),
  82                                 stream->channels()
  83                                 );
  84
  85                         resampler.reset (new Resampler (stream->frame_rate(), _content->resampled_frame_rate(film), stream->channels()));
  86                         if (_fast) {
  87                                 resampler->set_fast ();
  88                         }
  89                         _resamplers[stream] = resampler;
  90                 }
  91         }
  92
  93         if (resampler) {
  94                 shared_ptr<const AudioBuffers> ro = resampler->run (data);
  95                 if (ro->frames() == 0) {
  96                         return;
  97                 }
  98                 data = ro;
  99         }
 100
 101         Data(stream, ContentAudio (data, _positions[stream]));
 102         _positions[stream] += data->frames();
 103 }
 104
 105 /** @return Time just after the last thing that was emitted from a given stream */
 106 ContentTime
 107 AudioDecoder::stream_position (shared_ptr<const Film> film, AudioStreamPtr stream) const
 108 {
 109         PositionMap::const_iterator i = _positions.find (stream);
 110         DCPOMATIC_ASSERT (i != _positions.end ());
 111         return ContentTime::from_frames (i->second, _content->resampled_frame_rate(film));
 112 }
 113
 114 boost::optional<ContentTime>
 115 AudioDecoder::position (shared_ptr<const Film> film) const
 116 {
 117         optional<ContentTime> p;
 118         for (PositionMap::const_iterator i = _positions.begin(); i != _positions.end(); ++i) {
 119                 ContentTime const ct = stream_position (film, i->first);
 120                 if (!p || ct < *p) {
 121                         p = ct;
 122                 }
 123         }
 124
 125         return p;
 126 }
 127
 128 void
 129 AudioDecoder::seek ()
 130 {
 131         for (ResamplerMap::iterator i = _resamplers.begin(); i != _resamplers.end(); ++i) {
 132                 i->second->flush ();
 133                 i->second->reset ();
 134         }
 135
 136         for (PositionMap::iterator i = _positions.begin(); i != _positions.end(); ++i) {
 137                 i->second = 0;
 138         }
 139 }
 140
 141 void
 142 AudioDecoder::flush ()
 143 {
 144         for (ResamplerMap::iterator i = _resamplers.begin(); i != _resamplers.end(); ++i) {
 145                 shared_ptr<const AudioBuffers> ro = i->second->flush ();
 146                 if (ro->frames() > 0) {
 147                         Data (i->first, ContentAudio (ro, _positions[i->first]));
 148                         _positions[i->first] += ro->frames();
 149                 }
 150         }
 151
 152         if (_content->delay() < 0) {
 153                 /* Finish off with the gap caused by the delay */
 154                 silence (-_content->delay ());
 155         }
 156 }
 157
 158 void
 159 AudioDecoder::silence (int milliseconds)
 160 {
 161         BOOST_FOREACH (AudioStreamPtr i, _content->streams ()) {
 162                 int const samples = ContentTime::from_seconds(milliseconds / 1000.0).frames_round(i->frame_rate());
 163                 shared_ptr<AudioBuffers> silence (new AudioBuffers (i->channels(), samples));
 164                 silence->make_silent ();
 165                 Data (i, ContentAudio (silence, _positions[i]));
 166         }
 167 }