source/effects/colorcurve.cpp

   1 #include <cmath>
   2 #include "color.h"
   3 #include "colorcurve.h"
   4 #include "mesh.h"
   5 #include "renderer.h"
   6 #include "resources.h"
   7 #include "texture2d.h"
   8
   9 using namespace std;
  10
  11 namespace Msp {
  12 namespace GL {
  13
  14 ColorCurve::ColorCurve():
  15         shprog(Resources::get_global().get<Program>("_colorcurve.glsl.shader")),
  16         quad(Resources::get_global().get<Mesh>("_fullscreen_quad.mesh")),
  17         linear_sampler(Resources::get_global().get<Sampler>("_linear_clamp.samp")),
  18         nearest_sampler(Resources::get_global().get<Sampler>("_nearest_clamp.samp"))
  19 {
  20         curve.storage(LUMINANCE8, 256, 1);
  21
  22         set_exposure_adjust(0.0f);
  23         set_brightness_response(0.4f);
  24         set_linear();
  25 }
  26
  27 void ColorCurve::set_exposure_adjust(float e)
  28 {
  29         shdata.uniform("exposure", pow(2.0f, e));
  30 }
  31
  32 void ColorCurve::set_brightness_response(float b)
  33 {
  34         if(b<=0 || b>1)
  35                 throw invalid_argument("ColorCurve::set_brightness_response");
  36         // Calculate an offset value to obtain a derivative of 1 at zero intensity.
  37         float t = (b<1 ? pow(b, 1/(1-b)) : 0.0f);
  38         shdata.uniform("brightness_response", b, t, pow(t, b));
  39 }
  40
  41 void ColorCurve::set_gamma(float g)
  42 {
  43         if(g<0.1 || g>10)
  44                 throw invalid_argument("ColorCurve::set_gamma");
  45
  46         unsigned char curve_data[256];
  47         for(unsigned i=0; i<256; ++i)
  48                 curve_data[i] = pow(i/255.0f, 1/g)*255+0.5f;
  49         curve.image(0, curve_data);
  50 }
  51
  52 void ColorCurve::set_srgb()
  53 {
  54         unsigned char curve_data[256];
  55         curve_data[0] = 0;
  56         for(unsigned i=1; i<256; ++i)
  57                 curve_data[i] = to_srgb(i/255.0f)*255+0.5f;
  58         curve.image(0, curve_data);
  59 }
  60
  61 void ColorCurve::set_linear()
  62 {
  63         unsigned char curve_data[256];
  64         for(unsigned i=0; i<256; ++i)
  65                 curve_data[i] = i;
  66         curve.image(0, curve_data);
  67 }
  68
  69 void ColorCurve::render(Renderer &renderer, const Texture2D &color_buf, const Texture2D &)
  70 {
  71         Renderer::Push push(renderer);
  72         renderer.set_pipeline_key(this);
  73         renderer.clear(0);
  74         renderer.set_shader_program(&shprog, &shdata);
  75         renderer.set_texture("source", &color_buf, &nearest_sampler);
  76         renderer.set_texture("curve", &curve, &linear_sampler);
  77         quad.draw(renderer);
  78 }
  79
  80 void ColorCurve::set_debug_name(const string &name)
  81 {
  82 #ifdef DEBUG
  83         shdata.set_debug_name(name+" [UBO]");
  84         curve.set_debug_name(name+"/curve.tex");
  85 #else
  86         (void)name;
  87 #endif
  88 }
  89
  90
  91 ColorCurve *ColorCurve::Template::create(unsigned, unsigned) const
  92 {
  93         RefPtr<ColorCurve> colorcurve = new ColorCurve();
  94         colorcurve->set_exposure_adjust(exposure_adjust);
  95         colorcurve->set_brightness_response(brightness_response);
  96         if(srgb)
  97                 colorcurve->set_srgb();
  98         else
  99                 colorcurve->set_gamma(gamma);
 100         return colorcurve.release();
 101 }
 102
 103
 104 ColorCurve::Template::Loader::Loader(Template &t):
 105         DataFile::DerivedObjectLoader<Template, PostProcessor::Template::Loader>(t)
 106 {
 107         add("brightness_response", &Template::brightness_response);
 108         add("exposure_adjust", &Template::exposure_adjust);
 109         add("gamma", &Loader::gamma);
 110         add("srgb", &Loader::srgb);
 111 }
 112
 113 void ColorCurve::Template::Loader::gamma(float g)
 114 {
 115         obj.gamma = g;
 116         obj.srgb = false;
 117 }
 118
 119 void ColorCurve::Template::Loader::srgb()
 120 {
 121         obj.srgb = true;
 122 }
 123
 124 } // namespace GL
 125 } // namespace Msp