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
|
#include <epoxy/gl.h>
#include <assert.h>
#include <stdio.h>
#include <algorithm>
#include <Eigen/Core>
#include <Eigen/LU>
#include "ycbcr_conversion_effect.h"
#include "effect_util.h"
#include "util.h"
#include "ycbcr.h"
using namespace std;
using namespace Eigen;
namespace movit {
YCbCrConversionEffect::YCbCrConversionEffect(const YCbCrFormat &ycbcr_format, GLenum type)
: ycbcr_format(ycbcr_format), type(type)
{
register_uniform_mat3("ycbcr_matrix", &uniform_ycbcr_matrix);
register_uniform_vec3("offset", uniform_offset);
register_uniform_bool("clamp_range", &uniform_clamp_range);
// Only used when clamp_range is true.
register_uniform_vec3("ycbcr_min", uniform_ycbcr_min);
register_uniform_vec3("ycbcr_max", uniform_ycbcr_max);
}
string YCbCrConversionEffect::output_fragment_shader()
{
return read_file("ycbcr_conversion_effect.frag");
}
void YCbCrConversionEffect::set_gl_state(GLuint glsl_program_num, const string &prefix, unsigned *sampler_num)
{
Effect::set_gl_state(glsl_program_num, prefix, sampler_num);
Matrix3d ycbcr_to_rgb;
double scale_factor;
compute_ycbcr_matrix(ycbcr_format, uniform_offset, &ycbcr_to_rgb, type, &scale_factor);
uniform_ycbcr_matrix = ycbcr_to_rgb.inverse();
if (ycbcr_format.full_range) {
// The card will clamp for us later.
uniform_clamp_range = false;
} else {
uniform_clamp_range = true;
if (ycbcr_format.num_levels == 0 || ycbcr_format.num_levels == 256) { // 8-bit.
// These limits come from BT.601 page 8, or BT.709, page 5.
uniform_ycbcr_min[0] = 16.0 / 255.0;
uniform_ycbcr_min[1] = 16.0 / 255.0;
uniform_ycbcr_min[2] = 16.0 / 255.0;
uniform_ycbcr_max[0] = 235.0 / 255.0;
uniform_ycbcr_max[1] = 240.0 / 255.0;
uniform_ycbcr_max[2] = 240.0 / 255.0;
} else if (ycbcr_format.num_levels == 1024) { // 10-bit.
// BT.709, page 5, or BT.2020, page 6.
uniform_ycbcr_min[0] = 64.0 / 1023.0;
uniform_ycbcr_min[1] = 64.0 / 1023.0;
uniform_ycbcr_min[2] = 64.0 / 1023.0;
uniform_ycbcr_max[0] = 940.0 / 1023.0;
uniform_ycbcr_max[1] = 960.0 / 1023.0;
uniform_ycbcr_max[2] = 960.0 / 1023.0;
} else if (ycbcr_format.num_levels == 4096) { // 12-bit.
// BT.2020, page 6.
uniform_ycbcr_min[0] = 256.0 / 4095.0;
uniform_ycbcr_min[1] = 256.0 / 4095.0;
uniform_ycbcr_min[2] = 256.0 / 4095.0;
uniform_ycbcr_max[0] = 3760.0 / 4095.0;
uniform_ycbcr_max[1] = 3840.0 / 4095.0;
uniform_ycbcr_max[2] = 3840.0 / 4095.0;
} else {
assert(false);
}
uniform_ycbcr_min[0] /= scale_factor;
uniform_ycbcr_min[1] /= scale_factor;
uniform_ycbcr_min[2] /= scale_factor;
}
}
} // namespace movit
|
