1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
/*
Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#include <iostream>
#include <boost/shared_ptr.hpp>
#include <sigc++/bind.h>
#include "ab_transcoder.h"
#include "film.h"
#include "decoder.h"
#include "encoder.h"
#include "job.h"
#include "film_state.h"
#include "options.h"
#include "image.h"
#include "decoder_factory.h"
/** @file src/ab_transcoder.cc
* @brief A transcoder which uses one FilmState for the left half of the screen, and a different one
* for the right half (to facilitate A/B comparisons of settings)
*/
using namespace std;
using namespace boost;
/** @param a FilmState to use for the left half of the screen.
* @param b FilmState to use for the right half of the screen.
* @param o Options.
* @param j Job that we are associated with.
* @param l Log.
* @param e Encoder to use.
*/
ABTranscoder::ABTranscoder (
shared_ptr<const FilmState> a, shared_ptr<const FilmState> b, shared_ptr<const Options> o, Job* j, Log* l, shared_ptr<Encoder> e)
: _fs_a (a)
, _fs_b (b)
, _opt (o)
, _job (j)
, _log (l)
, _encoder (e)
, _last_frame (0)
{
_da = decoder_factory (_fs_a, o, j, _log);
_db = decoder_factory (_fs_b, o, j, _log);
_da->Video.connect (sigc::bind (sigc::mem_fun (*this, &ABTranscoder::process_video), 0));
_db->Video.connect (sigc::bind (sigc::mem_fun (*this, &ABTranscoder::process_video), 1));
_da->Audio.connect (sigc::mem_fun (*e, &Encoder::process_audio));
}
ABTranscoder::~ABTranscoder ()
{
}
void
ABTranscoder::process_video (shared_ptr<Image> yuv, int frame, int index)
{
if (index == 0) {
/* Keep this image around until we get the other half */
_image = yuv;
} else {
/* Copy the right half of yuv into _image */
for (int i = 0; i < yuv->components(); ++i) {
int const line_size = yuv->line_size()[i];
int const half_line_size = line_size / 2;
uint8_t* p = _image->data()[i];
uint8_t* q = yuv->data()[i];
for (int j = 0; j < yuv->lines (i); ++j) {
memcpy (p + half_line_size, q + half_line_size, half_line_size);
p += line_size;
q += line_size;
}
}
/* And pass it to the encoder */
_encoder->process_video (_image, frame);
_image.reset ();
}
_last_frame = frame;
}
void
ABTranscoder::go ()
{
_encoder->process_begin ();
_da->process_begin ();
_db->process_begin ();
while (1) {
bool const a = _da->pass ();
bool const b = _db->pass ();
if (_job) {
_job->set_progress (float (_last_frame) / _da->decoding_frames ());
}
if (a && b) {
break;
}
}
_encoder->process_end ();
_da->process_end ();
_db->process_end ();
}
|
