5 #include "dither_effect.h"
6 #include "effect_util.h"
12 // A simple LCG (linear congruental generator) random generator.
13 // We implement our own so we can be deterministic from frame to frame
14 // and run to run; we don't have special needs for speed or quality,
15 // as long as the period is reasonably long. The output is in range
18 // This comes from http://en.wikipedia.org/wiki/Linear_congruential_generator.
19 unsigned lcg_rand(unsigned x)
21 return (x * 1103515245U + 12345U) & ((1U << 31) - 1);
26 DitherEffect::DitherEffect()
27 : width(1280), height(720), num_bits(8),
28 last_width(-1), last_height(-1), last_num_bits(-1)
30 register_int("output_width", &width);
31 register_int("output_height", &height);
32 register_int("num_bits", &num_bits);
34 glGenTextures(1, &texnum);
37 DitherEffect::~DitherEffect()
39 glDeleteTextures(1, &texnum);
42 std::string DitherEffect::output_fragment_shader()
45 sprintf(buf, "#define NEED_EXPLICIT_ROUND %d\n", (movit_num_wrongly_rounded > 0));
46 return buf + read_file("dither_effect.frag");
49 void DitherEffect::update_texture(GLuint glsl_program_num, const std::string &prefix, unsigned *sampler_num)
51 float *dither_noise = new float[width * height];
52 float dither_double_amplitude = 1.0f / (1 << num_bits);
54 // We don't need a strictly nonrepeating dither; reducing the resolution
55 // to max 128x128 saves a lot of texture bandwidth, without causing any
56 // noticeable harm to the dither's performance.
57 texture_width = std::min(width, 128);
58 texture_height = std::min(height, 128);
60 // Using the resolution as a seed gives us a consistent dither from frame to frame.
61 // It also gives a different dither for e.g. different aspect ratios, which _feels_
62 // good, but probably shouldn't matter.
63 unsigned seed = (width << 16) ^ height;
64 for (int i = 0; i < texture_width * texture_height; ++i) {
65 seed = lcg_rand(seed);
66 float normalized_rand = seed * (1.0f / (1U << 31)) - 0.5; // [-0.5, 0.5>
67 dither_noise[i] = dither_double_amplitude * normalized_rand;
70 glActiveTexture(GL_TEXTURE0 + *sampler_num);
72 glBindTexture(GL_TEXTURE_2D, texnum);
74 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
76 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
78 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_REPEAT);
80 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_REPEAT);
82 glTexImage2D(GL_TEXTURE_2D, 0, GL_LUMINANCE16F_ARB, texture_width, texture_height, 0, GL_LUMINANCE, GL_FLOAT, dither_noise);
85 delete[] dither_noise;
88 void DitherEffect::set_gl_state(GLuint glsl_program_num, const std::string &prefix, unsigned *sampler_num)
90 Effect::set_gl_state(glsl_program_num, prefix, sampler_num);
96 if (width != last_width || height != last_height || num_bits != last_num_bits) {
97 update_texture(glsl_program_num, prefix, sampler_num);
100 last_num_bits = num_bits;
103 glActiveTexture(GL_TEXTURE0 + *sampler_num);
105 glBindTexture(GL_TEXTURE_2D, texnum);
108 set_uniform_int(glsl_program_num, prefix, "dither_tex", *sampler_num);
111 // In theory, we should adjust for the texel centers that have moved here as well,
112 // but since we use GL_NEAREST and we don't really care a lot what texel we sample,
113 // we don't have to worry about it.
114 float tc_scale[] = { float(width) / float(texture_width), float(height) / float(texture_height) };
115 set_uniform_vec2(glsl_program_num, prefix, "tc_scale", tc_scale);
117 // Used if the shader needs to do explicit rounding.
118 int round_fac = (1 << num_bits) - 1;
119 set_uniform_float(glsl_program_num, prefix, "round_fac", round_fac);
120 set_uniform_float(glsl_program_num, prefix, "inv_round_fac", 1.0f / round_fac);