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