Merge ImageMagick and FFmpeg content into VideoContent list; remove seek_to_last...
[dcpomatic.git] / 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
31 using std::list;
32 using std::cout;
33 using boost::shared_ptr;
34 using boost::weak_ptr;
35 using boost::dynamic_pointer_cast;
36
37 Player::Player (shared_ptr<const Film> f, shared_ptr<const Playlist> p)
38         : _film (f)
39         , _playlist (p)
40         , _video (true)
41         , _audio (true)
42         , _subtitles (true)
43         , _have_valid_decoders (false)
44         , _video_sync (true)
45 {
46         _playlist->Changed.connect (bind (&Player::playlist_changed, this));
47         _playlist->ContentChanged.connect (bind (&Player::content_changed, this, _1, _2));
48 }
49
50 void
51 Player::disable_video ()
52 {
53         _video = false;
54 }
55
56 void
57 Player::disable_audio ()
58 {
59         _audio = false;
60 }
61
62 void
63 Player::disable_subtitles ()
64 {
65         _subtitles = false;
66 }
67
68 bool
69 Player::pass ()
70 {
71         if (!_have_valid_decoders) {
72                 setup_decoders ();
73                 _have_valid_decoders = true;
74         }
75         
76         bool done = true;
77         
78         if (_video_decoder != _video_decoders.end ()) {
79                 if ((*_video_decoder)->pass ()) {
80                         _video_decoder++;
81                 }
82                 
83                 if (_video_decoder != _video_decoders.end ()) {
84                         done = false;
85                 }
86         }
87
88         if (_playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
89                 for (list<shared_ptr<SndfileDecoder> >::iterator i = _sndfile_decoders.begin(); i != _sndfile_decoders.end(); ++i) {
90                         if (!(*i)->pass ()) {
91                                 done = false;
92                         }
93                 }
94
95                 Audio (_sndfile_buffers);
96                 _sndfile_buffers.reset ();
97         }
98
99         return done;
100 }
101
102 void
103 Player::set_progress (shared_ptr<Job> job)
104 {
105         /* Assume progress can be divined from how far through the video we are */
106
107         if (_video_decoder == _video_decoders.end() || !_playlist->video_length()) {
108                 return;
109         }
110         
111         ContentVideoFrame p = 0;
112         list<shared_ptr<VideoDecoder> >::iterator i = _video_decoders.begin ();
113         while (i != _video_decoders.end() && i != _video_decoder) {
114                 p += (*i)->video_length ();
115         }
116
117         job->set_progress (float ((*_video_decoder)->video_frame ()) / _playlist->video_length ());
118 }
119
120 void
121 Player::process_video (shared_ptr<Image> i, bool same, shared_ptr<Subtitle> s)
122 {
123         Video (i, same, s);
124 }
125
126 void
127 Player::process_audio (weak_ptr<const AudioContent> c, shared_ptr<AudioBuffers> b)
128 {
129         if (_playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
130                 AudioMapping mapping = _film->audio_mapping ();
131                 if (!_sndfile_buffers) {
132                         _sndfile_buffers.reset (new AudioBuffers (mapping.dcp_channels(), b->frames ()));
133                         _sndfile_buffers->make_silent ();
134                 }
135
136                 for (int i = 0; i < b->channels(); ++i) {
137                         list<libdcp::Channel> dcp = mapping.content_to_dcp (AudioMapping::Channel (c, i));
138                         for (list<libdcp::Channel>::iterator j = dcp.begin(); j != dcp.end(); ++j) {
139                                 _sndfile_buffers->accumulate (b, i, static_cast<int> (*j));
140                         }
141                 }
142
143         } else {
144                 Audio (b);
145         }
146 }
147
148 /** @return true on error */
149 bool
150 Player::seek (double t)
151 {
152         if (!_have_valid_decoders) {
153                 setup_decoders ();
154                 _have_valid_decoders = true;
155         }
156         
157         /* Find the decoder that contains this position */
158         _video_decoder = _video_decoders.begin ();
159         while (_video_decoder != _video_decoders.end ()) {
160                 double const this_length = (*_video_decoder)->video_length() / _film->video_frame_rate ();
161                 if (t < this_length) {
162                         break;
163                 }
164                 t -= this_length;
165                 ++_video_decoder;
166         }
167         
168         if (_video_decoder != _video_decoders.end()) {
169                 (*_video_decoder)->seek (t);
170         } else {
171                 return true;
172         }
173
174         /* XXX: don't seek audio because we don't need to... */
175
176         return false;
177 }
178
179 void
180 Player::setup_decoders ()
181 {
182         if (_video) {
183                 list<shared_ptr<const VideoContent> > vc = _playlist->video ();
184                 for (list<shared_ptr<const VideoContent> >::iterator i = vc.begin(); i != vc.end(); ++i) {
185
186                         shared_ptr<VideoDecoder> d;
187                         
188                         /* XXX: into content? */
189                         
190                         shared_ptr<const FFmpegContent> fc = dynamic_pointer_cast<const FFmpegContent> (*i);
191                         if (fc) {
192                                 shared_ptr<FFmpegDecoder> fd (
193                                         new FFmpegDecoder (
194                                                 _film, fc, _video,
195                                                 _audio && _playlist->audio_from() == Playlist::AUDIO_FFMPEG,
196                                                 _subtitles && _film->with_subtitles(),
197                                                 _video_sync
198                                                 )
199                                         );
200
201                                 if (_playlist->audio_from() == Playlist::AUDIO_FFMPEG) {
202                                         fd->Audio.connect (bind (&Player::process_audio, this, fc, _1));
203                                 }
204
205                                 d = fd;
206                         }
207
208                         shared_ptr<const ImageMagickContent> ic = dynamic_pointer_cast<const ImageMagickContent> (*i);
209                         if (ic) {
210                                 d.reset (new ImageMagickDecoder (_film, ic));
211                         }
212
213                         d->connect_video (shared_from_this ());
214                         _video_decoders.push_back (d);
215                 }
216
217                 _video_decoder = _video_decoders.begin ();
218         }
219
220         if (_audio && _playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
221                 list<shared_ptr<const SndfileContent> > sc = _playlist->sndfile ();
222                 for (list<shared_ptr<const SndfileContent> >::iterator i = sc.begin(); i != sc.end(); ++i) {
223                         shared_ptr<SndfileDecoder> d (new SndfileDecoder (_film, *i));
224                         _sndfile_decoders.push_back (d);
225                         d->Audio.connect (bind (&Player::process_audio, this, *i, _1));
226                 }
227         }
228 }
229
230 void
231 Player::disable_video_sync ()
232 {
233         _video_sync = false;
234 }
235
236 double
237 Player::last_video_time () const
238 {
239         double t = 0;
240         for (list<shared_ptr<VideoDecoder> >::const_iterator i = _video_decoders.begin(); i != _video_decoder; ++i) {
241                 t += (*i)->video_length() / (*i)->video_frame_rate ();
242         }
243
244         return t + (*_video_decoder)->last_content_time ();
245 }
246
247 void
248 Player::content_changed (weak_ptr<Content> w, int p)
249 {
250         shared_ptr<Content> c = w.lock ();
251         if (!c) {
252                 return;
253         }
254
255         if (p == VideoContentProperty::VIDEO_LENGTH) {
256                 if (dynamic_pointer_cast<FFmpegContent> (c)) {
257                         /* FFmpeg content length changes are serious; we need new decoders */
258                         _have_valid_decoders = false;
259                 }
260         }
261 }
262
263 void
264 Player::playlist_changed ()
265 {
266         _have_valid_decoders = false;
267 }