Partial addition of DCP audio channel count to Film.
[dcpomatic.git] / src / lib / player.cc
1 /* -*- c-basic-offset: 8; default-tab-width: 8; -*- */
2
3 /*
4     Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
5
6     This program 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     This program 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 this program; if not, write to the Free Software
18     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19
20 */
21
22 #include "player.h"
23 #include "film.h"
24 #include "ffmpeg_decoder.h"
25 #include "ffmpeg_content.h"
26 #include "imagemagick_decoder.h"
27 #include "imagemagick_content.h"
28 #include "sndfile_decoder.h"
29 #include "sndfile_content.h"
30 #include "playlist.h"
31 #include "job.h"
32 #include "image.h"
33 #include "null_content.h"
34 #include "black_decoder.h"
35 #include "silence_decoder.h"
36
37 using std::list;
38 using std::cout;
39 using std::min;
40 using std::max;
41 using std::vector;
42 using boost::shared_ptr;
43 using boost::weak_ptr;
44 using boost::dynamic_pointer_cast;
45
46 struct Piece
47 {
48         Piece (shared_ptr<Content> c, shared_ptr<Decoder> d)
49                 : content (c)
50                 , decoder (d)
51         {}
52         
53         shared_ptr<Content> content;
54         shared_ptr<Decoder> decoder;
55 };
56
57 Player::Player (shared_ptr<const Film> f, shared_ptr<const Playlist> p)
58         : _film (f)
59         , _playlist (p)
60         , _video (true)
61         , _audio (true)
62         , _subtitles (true)
63         , _have_valid_pieces (false)
64         , _position (0)
65         , _audio_buffers (f->dcp_audio_channels(), 0)
66         , _next_audio (0)
67 {
68         _playlist->Changed.connect (bind (&Player::playlist_changed, this));
69         _playlist->ContentChanged.connect (bind (&Player::content_changed, this, _1, _2));
70 }
71
72 void
73 Player::disable_video ()
74 {
75         _video = false;
76 }
77
78 void
79 Player::disable_audio ()
80 {
81         _audio = false;
82 }
83
84 void
85 Player::disable_subtitles ()
86 {
87         _subtitles = false;
88 }
89
90 bool
91 Player::pass ()
92 {
93         if (!_have_valid_pieces) {
94                 setup_pieces ();
95                 _have_valid_pieces = true;
96         }
97
98         /* Here we are just finding the active decoder with the earliest last emission time, then
99            calling pass on it.
100         */
101
102         Time earliest_t = TIME_MAX;
103         shared_ptr<Piece> earliest;
104
105         for (list<shared_ptr<Piece> >::iterator i = _pieces.begin(); i != _pieces.end(); ++i) {
106                 cout << "check " << (*i)->content->file() << " start=" << (*i)->content->start() << ", next=" << (*i)->decoder->next() << ", end=" << (*i)->content->end() << "\n";
107                 if (((*i)->decoder->next() + (*i)->content->start()) >= (*i)->content->end()) {
108                         continue;
109                 }
110                 
111                 Time const t = (*i)->content->start() + (*i)->decoder->next();
112                 if (t < earliest_t) {
113                         cout << "\t candidate; " << t << " " << (t / TIME_HZ) << ".\n";
114                         earliest_t = t;
115                         earliest = *i;
116                 }
117         }
118
119         if (!earliest) {
120                 return true;
121         }
122
123         cout << "PASS:\n";
124         cout << "\tpass " << earliest->content->file() << " ";
125         if (dynamic_pointer_cast<FFmpegContent> (earliest->content)) {
126                 cout << " FFmpeg.\n";
127         } else if (dynamic_pointer_cast<ImageMagickContent> (earliest->content)) {
128                 cout << " ImageMagickContent.\n";
129         } else if (dynamic_pointer_cast<SndfileContent> (earliest->content)) {
130                 cout << " SndfileContent.\n";
131         } else if (dynamic_pointer_cast<BlackDecoder> (earliest->decoder)) {
132                 cout << " Black.\n";
133         } else if (dynamic_pointer_cast<SilenceDecoder> (earliest->decoder)) {
134                 cout << " Silence.\n";
135         }
136         
137         earliest->decoder->pass ();
138         _position = earliest->content->start() + earliest->decoder->next ();
139         cout << "\tpassed to " << _position << " " << (_position / TIME_HZ) << "\n";
140
141         return false;
142 }
143
144 void
145 Player::process_video (weak_ptr<Content> weak_content, shared_ptr<const Image> image, bool same, shared_ptr<Subtitle> sub, Time time)
146 {
147         cout << "[V]\n";
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, 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 struct ContentSorter
234 {
235         bool operator() (shared_ptr<Content> a, shared_ptr<Content> b)
236         {
237                 return a->start() < b->start();
238         }
239 };
240
241 void
242 Player::setup_pieces ()
243 {
244 //      cout << "----- Player SETUP PIECES.\n";
245
246         list<shared_ptr<Piece> > old_pieces = _pieces;
247
248         _pieces.clear ();
249
250         Playlist::ContentList content = _playlist->content ();
251         sort (content.begin(), content.end(), ContentSorter ());
252         
253         for (Playlist::ContentList::iterator i = content.begin(); i != content.end(); ++i) {
254
255                 shared_ptr<Decoder> decoder;
256                 
257                 /* XXX: into content? */
258
259                 shared_ptr<const FFmpegContent> fc = dynamic_pointer_cast<const FFmpegContent> (*i);
260                 if (fc) {
261                         shared_ptr<FFmpegDecoder> fd (new FFmpegDecoder (_film, fc, _video, _audio, _subtitles));
262                         
263                         fd->Video.connect (bind (&Player::process_video, this, *i, _1, _2, _3, _4));
264                         fd->Audio.connect (bind (&Player::process_audio, this, *i, _1, _2));
265
266                         decoder = fd;
267 //                      cout << "\tFFmpeg @ " << fc->start() << " -- " << fc->end() << "\n";
268                 }
269                 
270                 shared_ptr<const ImageMagickContent> ic = dynamic_pointer_cast<const ImageMagickContent> (*i);
271                 if (ic) {
272                         shared_ptr<ImageMagickDecoder> id;
273                         
274                         /* See if we can re-use an old ImageMagickDecoder */
275                         for (list<shared_ptr<Piece> >::const_iterator j = old_pieces.begin(); j != old_pieces.end(); ++j) {
276                                 shared_ptr<ImageMagickDecoder> imd = dynamic_pointer_cast<ImageMagickDecoder> ((*j)->decoder);
277                                 if (imd && imd->content() == ic) {
278                                         id = imd;
279                                 }
280                         }
281
282                         if (!id) {
283                                 id.reset (new ImageMagickDecoder (_film, ic));
284                                 id->Video.connect (bind (&Player::process_video, this, *i, _1, _2, _3, _4));
285                         }
286
287                         decoder = id;
288 //                      cout << "\tImageMagick @ " << ic->start() << " -- " << ic->end() << "\n";
289                 }
290
291                 shared_ptr<const SndfileContent> sc = dynamic_pointer_cast<const SndfileContent> (*i);
292                 if (sc) {
293                         shared_ptr<AudioDecoder> sd (new SndfileDecoder (_film, sc));
294                         sd->Audio.connect (bind (&Player::process_audio, this, *i, _1, _2));
295
296                         decoder = sd;
297 //                      cout << "\tSndfile @ " << sc->start() << " -- " << sc->end() << "\n";
298                 }
299
300                 _pieces.push_back (shared_ptr<Piece> (new Piece (*i, decoder)));
301         }
302
303         /* Fill in visual gaps with black and audio gaps with silence */
304
305         Time video_pos = 0;
306         Time audio_pos = 0;
307         list<shared_ptr<Piece> > pieces_copy = _pieces;
308         for (list<shared_ptr<Piece> >::iterator i = pieces_copy.begin(); i != pieces_copy.end(); ++i) {
309                 if (dynamic_pointer_cast<VideoContent> ((*i)->content)) {
310                         Time const diff = (*i)->content->start() - video_pos;
311                         if (diff > 0) {
312                                 shared_ptr<NullContent> nc (new NullContent (_film, video_pos, diff));
313                                 shared_ptr<BlackDecoder> bd (new BlackDecoder (_film, nc));
314                                 bd->Video.connect (bind (&Player::process_video, this, nc, _1, _2, _3, _4));
315                                 _pieces.push_back (shared_ptr<Piece> (new Piece (nc, bd)));
316 //                              cout << "\tblack @ " << video_pos << " -- " << (video_pos + diff) << "\n";
317                         }
318                                                 
319                         video_pos = (*i)->content->end();
320                 } else {
321                         Time const diff = (*i)->content->start() - audio_pos;
322                         if (diff > 0) {
323                                 shared_ptr<NullContent> nc (new NullContent (_film, audio_pos, diff));
324                                 shared_ptr<SilenceDecoder> sd (new SilenceDecoder (_film, nc));
325                                 sd->Audio.connect (bind (&Player::process_audio, this, nc, _1, _2));
326                                 _pieces.push_back (shared_ptr<Piece> (new Piece (nc, sd)));
327 //                              cout << "\tsilence @ " << audio_pos << " -- " << (audio_pos + diff) << "\n";
328                         }
329                         audio_pos = (*i)->content->end();
330                 }
331         }
332 }
333
334 void
335 Player::content_changed (weak_ptr<Content> w, int p)
336 {
337         shared_ptr<Content> c = w.lock ();
338         if (!c) {
339                 return;
340         }
341
342         if (p == ContentProperty::START || p == VideoContentProperty::VIDEO_LENGTH) {
343                 _have_valid_pieces = false;
344         }
345 }
346
347 void
348 Player::playlist_changed ()
349 {
350         _have_valid_pieces = false;
351 }