Add missing files.
[dcpomatic.git] / src / lib / ab_transcoder.cc
1 /*
2     Copyright (C) 2012 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 <iostream>
21 #include <boost/shared_ptr.hpp>
22 #include "ab_transcoder.h"
23 #include "film.h"
24 #include "encoder.h"
25 #include "job.h"
26 #include "image.h"
27 #include "player.h"
28 #include "matcher.h"
29 #include "delay_line.h"
30 #include "gain.h"
31 #include "combiner.h"
32
33 /** @file src/ab_transcoder.cc
34  *  @brief A transcoder which uses one Film for the left half of the screen, and a different one
35  *  for the right half (to facilitate A/B comparisons of settings)
36  */
37
38 using std::string;
39 using boost::shared_ptr;
40 using boost::dynamic_pointer_cast;
41
42 /** @param a Film to use for the left half of the screen.
43  *  @param b Film to use for the right half of the screen.
44  *  @param o Decoder options.
45  *  @param j Job that we are associated with.
46  *  @param e Encoder to use.
47  */
48
49 ABTranscoder::ABTranscoder (shared_ptr<Film> a, shared_ptr<Film> b, shared_ptr<Job> j)
50         : _film_a (a)
51         , _film_b (b)
52         , _player_a (_film_a->player ())
53         , _player_b (_film_b->player ())
54         , _job (j)
55         , _encoder (new Encoder (_film_a))
56         , _combiner (new Combiner (a->log()))
57 {
58         _matcher.reset (new Matcher (_film_a->log(), _film_a->audio_frame_rate(), _film_a->video_frame_rate()));
59         _delay_line.reset (new DelayLine (_film_a->log(), _film_a->audio_delay() * _film_a->audio_frame_rate() / 1000));
60         _gain.reset (new Gain (_film_a->log(), _film_a->audio_gain()));
61
62         _player_a->Video.connect (bind (&Combiner::process_video, _combiner, _1, _2, _3, _4));
63         _player_b->Video.connect (bind (&Combiner::process_video_b, _combiner, _1, _2, _3, _4));
64
65         _combiner->connect_video (_delay_line);
66         _delay_line->connect_video (_matcher);
67         _matcher->connect_video (_encoder);
68         
69         _player_a->connect_audio (_delay_line);
70         _delay_line->connect_audio (_matcher);
71         _matcher->connect_audio (_gain);
72         _gain->connect_audio (_encoder);
73 }
74
75 void
76 ABTranscoder::go ()
77 {
78         _encoder->process_begin ();
79
80         bool done[2] = { false, false };
81         
82         while (1) {
83                 done[0] = _player_a->pass ();
84                 done[1] = _player_b->pass ();
85
86                 if (_job) {
87                         _player_a->set_progress (_job);
88                 }
89
90                 if (done[0] && done[1]) {
91                         break;
92                 }
93         }
94
95         if (_delay_line) {
96                 _delay_line->process_end ();
97         }
98         if (_matcher) {
99                 _matcher->process_end ();
100         }
101         if (_gain) {
102                 _gain->process_end ();
103         }
104         _encoder->process_end ();
105 }
106