]> git.sesse.net Git - movit/blob - ycbcr_conversion_effect.cpp
Add a mode for YCbCrInput where Cb and Cr are in the same texture.
[movit] / ycbcr_conversion_effect.cpp
1 #include <epoxy/gl.h>
2 #include <assert.h>
3 #include <stdio.h>
4 #include <algorithm>
5 #include <Eigen/Core>
6 #include <Eigen/LU>
7
8 #include "ycbcr_conversion_effect.h"
9 #include "effect_util.h"
10 #include "util.h"
11 #include "ycbcr.h"
12
13 using namespace std;
14 using namespace Eigen;
15
16 namespace movit {
17
18 YCbCrConversionEffect::YCbCrConversionEffect(const YCbCrFormat &ycbcr_format)
19         : ycbcr_format(ycbcr_format)
20 {
21 }
22
23 string YCbCrConversionEffect::output_fragment_shader()
24 {
25         float offset[3];
26         Matrix3d ycbcr_to_rgb;
27         compute_ycbcr_matrix(ycbcr_format, offset, &ycbcr_to_rgb);
28
29         string frag_shader = output_glsl_mat3("PREFIX(ycbcr_matrix)", ycbcr_to_rgb.inverse());
30         frag_shader += output_glsl_vec3("PREFIX(offset)", offset[0], offset[1], offset[2]);
31
32         if (ycbcr_format.full_range) {
33                 // The card will clamp for us later.
34                 frag_shader += "#define YCBCR_CLAMP_RANGE 0\n";
35         } else {
36                 frag_shader += "#define YCBCR_CLAMP_RANGE 1\n";
37
38                 // These limits come from BT.601 page 8, or BT.701, page 5.
39                 // TODO: Use num_levels. Currently we support 8-bit levels only.
40                 frag_shader += output_glsl_vec3("PREFIX(ycbcr_min)", 16.0 / 255.0, 16.0 / 255.0, 16.0 / 255.0);
41                 frag_shader += output_glsl_vec3("PREFIX(ycbcr_max)", 235.0 / 255.0, 240.0 / 255.0, 240.0 / 255.0);
42         }
43
44         return frag_shader + read_file("ycbcr_conversion_effect.frag");
45 }
46
47 }  // namespace movit