src/lib/player.cc

   1 /*
   2     Copyright (C) 2013 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 #include "player.h"
  21 #include "film.h"
  22 #include "ffmpeg_decoder.h"
  23 #include "ffmpeg_content.h"
  24 #include "imagemagick_decoder.h"
  25 #include "imagemagick_content.h"
  26 #include "sndfile_decoder.h"
  27 #include "sndfile_content.h"
  28 #include "playlist.h"
  29 #include "job.h"
  30 #include "image.h"
  31 #include "null_content.h"
  32 #include "black_decoder.h"
  33 #include "silence_decoder.h"
  34
  35 using std::list;
  36 using std::cout;
  37 using std::min;
  38 using std::max;
  39 using std::vector;
  40 using boost::shared_ptr;
  41 using boost::weak_ptr;
  42 using boost::dynamic_pointer_cast;
  43
  44 struct Piece
  45 {
  46         Piece (shared_ptr<Content> c, shared_ptr<Decoder> d)
  47                 : content (c)
  48                 , decoder (d)
  49         {}
  50
  51         shared_ptr<Content> content;
  52         shared_ptr<Decoder> decoder;
  53 };
  54
  55 Player::Player (shared_ptr<const Film> f, shared_ptr<const Playlist> p)
  56         : _film (f)
  57         , _playlist (p)
  58         , _video (true)
  59         , _audio (true)
  60         , _subtitles (true)
  61         , _have_valid_pieces (false)
  62         , _position (0)
  63         , _audio_buffers (f->dcp_audio_channels(), 0)
  64         , _next_audio (0)
  65 {
  66         _playlist->Changed.connect (bind (&Player::playlist_changed, this));
  67         _playlist->ContentChanged.connect (bind (&Player::content_changed, this, _1, _2));
  68 }
  69
  70 void
  71 Player::disable_video ()
  72 {
  73         _video = false;
  74 }
  75
  76 void
  77 Player::disable_audio ()
  78 {
  79         _audio = false;
  80 }
  81
  82 void
  83 Player::disable_subtitles ()
  84 {
  85         _subtitles = false;
  86 }
  87
  88 bool
  89 Player::pass ()
  90 {
  91         if (!_have_valid_pieces) {
  92                 setup_pieces ();
  93                 _have_valid_pieces = true;
  94         }
  95
  96         /* Here we are just finding the active decoder with the earliest last emission time, then
  97            calling pass on it.
  98         */
  99
 100         Time earliest_t = TIME_MAX;
 101         shared_ptr<Piece> earliest;
 102
 103         for (list<shared_ptr<Piece> >::iterator i = _pieces.begin(); i != _pieces.end(); ++i) {
 104 //              cout << "check " << (*i)->content->file() << " start=" << (*i)->content->start() << ", next=" << (*i)->decoder->next() << ", end=" << (*i)->content->end() << "\n";
 105                 if (((*i)->decoder->next() + (*i)->content->start()) >= (*i)->content->end()) {
 106                         continue;
 107                 }
 108
 109                 if (!_audio && dynamic_pointer_cast<AudioDecoder> ((*i)->decoder) && !dynamic_pointer_cast<VideoDecoder> ((*i)->decoder)) {
 110                         continue;
 111                 }
 112
 113                 Time const t = (*i)->content->start() + (*i)->decoder->next();
 114                 if (t < earliest_t) {
 115 //                      cout << "\t candidate; " << t << " " << (t / TIME_HZ) << ".\n";
 116                         earliest_t = t;
 117                         earliest = *i;
 118                 }
 119         }
 120
 121         if (!earliest) {
 122                 return true;
 123         }
 124
 125         cout << "PASS:\n";
 126         cout << "\tpass " << earliest->content->file() << " ";
 127         if (dynamic_pointer_cast<FFmpegContent> (earliest->content)) {
 128                 cout << " FFmpeg.\n";
 129         } else if (dynamic_pointer_cast<ImageMagickContent> (earliest->content)) {
 130                 cout << " ImageMagickContent.\n";
 131         } else if (dynamic_pointer_cast<SndfileContent> (earliest->content)) {
 132                 cout << " SndfileContent.\n";
 133         } else if (dynamic_pointer_cast<BlackDecoder> (earliest->decoder)) {
 134                 cout << " Black.\n";
 135         } else if (dynamic_pointer_cast<SilenceDecoder> (earliest->decoder)) {
 136                 cout << " Silence.\n";
 137         }
 138
 139         earliest->decoder->pass ();
 140         _position = earliest->content->start() + earliest->decoder->next ();
 141         cout << "\tpassed to " << _position << " " << (_position / TIME_HZ) << "\n";
 142
 143         return false;
 144 }
 145
 146 void
 147 Player::process_video (weak_ptr<Content> weak_content, shared_ptr<const Image> image, bool same, shared_ptr<Subtitle> sub, Time time)
 148 {
 149         shared_ptr<Content> content = weak_content.lock ();
 150         if (!content) {
 151                 return;
 152         }
 153
 154         time += content->start ();
 155
 156         Video (image, same, sub, time);
 157 }
 158
 159 void
 160 Player::process_audio (weak_ptr<Content> weak_content, shared_ptr<const AudioBuffers> audio, Time time)
 161 {
 162         shared_ptr<Content> content = weak_content.lock ();
 163         if (!content) {
 164                 return;
 165         }
 166
 167         /* The time of this audio may indicate that some of our buffered audio is not going to
 168            be added to any more, so it can be emitted.
 169         */
 170
 171         time += content->start ();
 172
 173         if (time > _next_audio) {
 174                 /* We can emit some audio from our buffers */
 175                 assert (_film->time_to_audio_frames (time - _next_audio) <= _audio_buffers.frames());
 176                 OutputAudioFrame const N = _film->time_to_audio_frames (time - _next_audio);
 177                 shared_ptr<AudioBuffers> emit (new AudioBuffers (_audio_buffers.channels(), N));
 178                 emit->copy_from (&_audio_buffers, N, 0, 0);
 179                 Audio (emit, _next_audio);
 180                 _next_audio += _film->audio_frames_to_time (N);
 181
 182                 /* And remove it from our buffers */
 183                 if (_audio_buffers.frames() > N) {
 184                         _audio_buffers.move (N, 0, _audio_buffers.frames() - N);
 185                 }
 186                 _audio_buffers.set_frames (_audio_buffers.frames() - N);
 187         }
 188
 189         /* Now accumulate the new audio into our buffers */
 190         _audio_buffers.ensure_size (_audio_buffers.frames() + audio->frames());
 191         _audio_buffers.accumulate_frames (audio.get(), 0, 0, audio->frames ());
 192 }
 193
 194 /** @return true on error */
 195 void
 196 Player::seek (Time t)
 197 {
 198         if (!_have_valid_pieces) {
 199                 setup_pieces ();
 200                 _have_valid_pieces = true;
 201         }
 202
 203         if (_pieces.empty ()) {
 204                 return;
 205         }
 206
 207 //      cout << "seek to " << t << " " << (t / TIME_HZ) << "\n";
 208
 209         for (list<shared_ptr<Piece> >::iterator i = _pieces.begin(); i != _pieces.end(); ++i) {
 210                 Time s = t - (*i)->content->start ();
 211                 s = max (static_cast<Time> (0), s);
 212                 s = min ((*i)->content->length(), s);
 213 //              cout << "seek [" << (*i)->content->file() << "," << (*i)->content->start() << "," << (*i)->content->end() << "] to " << s << "\n";
 214                 (*i)->decoder->seek (s);
 215         }
 216
 217         /* XXX: don't seek audio because we don't need to... */
 218 }
 219
 220
 221 void
 222 Player::seek_back ()
 223 {
 224
 225 }
 226
 227 void
 228 Player::seek_forward ()
 229 {
 230
 231 }
 232
 233 void
 234 Player::add_black_piece (Time s, Time len)
 235 {
 236         shared_ptr<NullContent> nc (new NullContent (_film, s, len));
 237         shared_ptr<BlackDecoder> bd (new BlackDecoder (_film, nc));
 238         bd->Video.connect (bind (&Player::process_video, this, nc, _1, _2, _3, _4));
 239         _pieces.push_back (shared_ptr<Piece> (new Piece (nc, bd)));
 240         cout << "\tblack @ " << s << " -- " << (s + len) << "\n";
 241 }
 242
 243 void
 244 Player::add_silent_piece (Time s, Time len)
 245 {
 246         shared_ptr<NullContent> nc (new NullContent (_film, s, len));
 247         shared_ptr<SilenceDecoder> sd (new SilenceDecoder (_film, nc));
 248         sd->Audio.connect (bind (&Player::process_audio, this, nc, _1, _2));
 249         _pieces.push_back (shared_ptr<Piece> (new Piece (nc, sd)));
 250         cout << "\tsilence @ " << s << " -- " << (s + len) << "\n";
 251 }
 252
 253
 254 struct ContentSorter
 255 {
 256         bool operator() (shared_ptr<Content> a, shared_ptr<Content> b)
 257         {
 258                 return a->start() < b->start();
 259         }
 260 };
 261
 262 void
 263 Player::setup_pieces ()
 264 {
 265         cout << "----- Player SETUP PIECES.\n";
 266
 267         list<shared_ptr<Piece> > old_pieces = _pieces;
 268
 269         _pieces.clear ();
 270
 271         Playlist::ContentList content = _playlist->content ();
 272         sort (content.begin(), content.end(), ContentSorter ());
 273
 274         for (Playlist::ContentList::iterator i = content.begin(); i != content.end(); ++i) {
 275
 276                 shared_ptr<Decoder> decoder;
 277
 278                 /* XXX: into content? */
 279
 280                 shared_ptr<const FFmpegContent> fc = dynamic_pointer_cast<const FFmpegContent> (*i);
 281                 if (fc) {
 282                         shared_ptr<FFmpegDecoder> fd (new FFmpegDecoder (_film, fc, _video, _audio, _subtitles));
 283
 284                         fd->Video.connect (bind (&Player::process_video, this, *i, _1, _2, _3, _4));
 285                         fd->Audio.connect (bind (&Player::process_audio, this, *i, _1, _2));
 286
 287                         decoder = fd;
 288                         cout << "\tFFmpeg @ " << fc->start() << " -- " << fc->end() << "\n";
 289                 }
 290
 291                 shared_ptr<const ImageMagickContent> ic = dynamic_pointer_cast<const ImageMagickContent> (*i);
 292                 if (ic) {
 293                         shared_ptr<ImageMagickDecoder> id;
 294
 295                         /* See if we can re-use an old ImageMagickDecoder */
 296                         for (list<shared_ptr<Piece> >::const_iterator j = old_pieces.begin(); j != old_pieces.end(); ++j) {
 297                                 shared_ptr<ImageMagickDecoder> imd = dynamic_pointer_cast<ImageMagickDecoder> ((*j)->decoder);
 298                                 if (imd && imd->content() == ic) {
 299                                         id = imd;
 300                                 }
 301                         }
 302
 303                         if (!id) {
 304                                 id.reset (new ImageMagickDecoder (_film, ic));
 305                                 id->Video.connect (bind (&Player::process_video, this, *i, _1, _2, _3, _4));
 306                         }
 307
 308                         decoder = id;
 309                         cout << "\tImageMagick @ " << ic->start() << " -- " << ic->end() << "\n";
 310                 }
 311
 312                 shared_ptr<const SndfileContent> sc = dynamic_pointer_cast<const SndfileContent> (*i);
 313                 if (sc) {
 314                         shared_ptr<AudioDecoder> sd (new SndfileDecoder (_film, sc));
 315                         sd->Audio.connect (bind (&Player::process_audio, this, *i, _1, _2));
 316
 317                         decoder = sd;
 318                         cout << "\tSndfile @ " << sc->start() << " -- " << sc->end() << "\n";
 319                 }
 320
 321                 _pieces.push_back (shared_ptr<Piece> (new Piece (*i, decoder)));
 322         }
 323
 324         /* Fill in visual gaps with black and audio gaps with silence */
 325
 326         Time video_pos = 0;
 327         Time audio_pos = 0;
 328         list<shared_ptr<Piece> > pieces_copy = _pieces;
 329         for (list<shared_ptr<Piece> >::iterator i = pieces_copy.begin(); i != pieces_copy.end(); ++i) {
 330                 if (dynamic_pointer_cast<VideoContent> ((*i)->content)) {
 331                         Time const diff = (*i)->content->start() - video_pos;
 332                         if (diff > 0) {
 333                                 add_black_piece (video_pos, diff);
 334                         }
 335
 336                         video_pos = (*i)->content->end();
 337                 } else {
 338                         Time const diff = (*i)->content->start() - audio_pos;
 339                         if (diff > 0) {
 340                                 add_silent_piece (video_pos, diff);
 341                         }
 342                         audio_pos = (*i)->content->end();
 343                 }
 344         }
 345
 346         if (video_pos < audio_pos) {
 347                 add_black_piece (video_pos, audio_pos - video_pos);
 348         } else if (audio_pos < video_pos) {
 349                 add_silent_piece (audio_pos, video_pos - audio_pos);
 350         }
 351 }
 352
 353 void
 354 Player::content_changed (weak_ptr<Content> w, int p)
 355 {
 356         shared_ptr<Content> c = w.lock ();
 357         if (!c) {
 358                 return;
 359         }
 360
 361         if (p == ContentProperty::START || p == VideoContentProperty::VIDEO_LENGTH) {
 362                 _have_valid_pieces = false;
 363         }
 364 }
 365
 366 void
 367 Player::playlist_changed ()
 368 {
 369         _have_valid_pieces = false;
 370 }