Use an enumeration type rather than unnamed integers to specify the different types...
[synfig.git] / synfig-core / trunk / src / synfig / gamma.h
1 /* === S Y N F I G ========================================================= */
2 /*!     \file gamma.h
3 **      \brief Template Header
4 **
5 **      $Id$
6 **
7 **      \legal
8 **      Copyright (c) 2002-2005 Robert B. Quattlebaum Jr., Adrian Bentley
9 **
10 **      This package is free software; you can redistribute it and/or
11 **      modify it under the terms of the GNU General Public License as
12 **      published by the Free Software Foundation; either version 2 of
13 **      the License, or (at your option) any later version.
14 **
15 **      This package is distributed in the hope that it will be useful,
16 **      but WITHOUT ANY WARRANTY; without even the implied warranty of
17 **      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 **      General Public License for more details.
19 **      \endlegal
20 */
21 /* ========================================================================= */
22
23 /* === S T A R T =========================================================== */
24
25 #ifndef __SYNFIG_GAMMA_H
26 #define __SYNFIG_GAMMA_H
27
28 /* === H E A D E R S ======================================================= */
29
30 #include <cmath>
31
32 /* === M A C R O S ========================================================= */
33
34 /* === T Y P E D E F S ===================================================== */
35
36 /* === C L A S S E S & S T R U C T S ======================================= */
37
38 namespace synfig {
39
40 /*!     \class Gamma
41 **      \brief This class performs color correction on Color classes.
42 **      \stub
43 */
44 class Gamma
45 {
46         float gamma_r;
47         float gamma_g;
48         float gamma_b;
49         float black_level;
50         float red_blue_level;
51
52         unsigned char table_r_U16_to_U8[65536];
53         unsigned char table_g_U16_to_U8[65536];
54         unsigned char table_b_U16_to_U8[65536];
55
56         float table_r_U8_to_F32[256];
57         float table_g_U8_to_F32[256];
58         float table_b_U8_to_F32[256];
59
60 public:
61         Gamma(float x=1):black_level(0) { set_gamma(x); }
62
63         void set_gamma(float x);
64         void set_gamma_r(float x);
65         void set_gamma_g(float x);
66         void set_gamma_b(float x);
67         void set_black_level(float x);
68
69         void set_red_blue_level(float x);
70         void set_all(float r, float g, float b, float black, float red_blue=1.0f);
71
72         float get_gamma()const { return (gamma_r+gamma_g+gamma_b)*0.33333333; }
73         float get_gamma_r()const { return gamma_r; }
74         float get_gamma_g()const { return gamma_g; }
75         float get_gamma_b()const { return gamma_b; }
76         float get_black_level()const { return black_level; }
77         float get_red_blue_level()const { return red_blue_level; }
78
79         void refresh_gamma_r();
80         void refresh_gamma_g();
81         void refresh_gamma_b();
82
83         const unsigned char &r_U16_to_U8(int i)const { return table_r_U16_to_U8[i]; }
84         const unsigned char &g_U16_to_U8(int i)const { return table_g_U16_to_U8[i]; }
85         const unsigned char &b_U16_to_U8(int i)const { return table_b_U16_to_U8[i]; }
86
87         const unsigned char &r_F32_to_U8(float x)const { return table_r_U16_to_U8[(int)(x*65535.0f)]; }
88         const unsigned char &g_F32_to_U8(float x)const { return table_g_U16_to_U8[(int)(x*65535.0f)]; }
89         const unsigned char &b_F32_to_U8(float x)const { return table_b_U16_to_U8[(int)(x*65535.0f)]; }
90
91         unsigned short r_F32_to_U16(float x)const { return (unsigned short)table_r_U16_to_U8[(int)(x*65535.0f)]<<8; }
92         unsigned short g_F32_to_U16(float x)const { return (unsigned short)table_g_U16_to_U8[(int)(x*65535.0f)]<<8; }
93         unsigned short b_F32_to_U16(float x)const { return (unsigned short)table_b_U16_to_U8[(int)(x*65535.0f)]<<8; }
94
95         const float& r_U8_to_F32(int i)const { return table_r_U8_to_F32[i]; }
96         const float& g_U8_to_F32(int i)const { return table_g_U8_to_F32[i]; }
97         const float& b_U8_to_F32(int i)const { return table_b_U8_to_F32[i]; }
98
99         float r_F32_to_F32(float x)const { return static_cast<float>(pow(x,gamma_r)*(1.0f-black_level)+black_level); }
100         float g_F32_to_F32(float x)const { return static_cast<float>(pow(x,gamma_g)*(1.0f-black_level)+black_level); }
101         float b_F32_to_F32(float x)const { return static_cast<float>(pow(x,gamma_b)*(1.0f-black_level)+black_level); }
102 }; // END of class Gamma
103
104 }; // END of namespace synfig
105
106 /* === E N D =============================================================== */
107
108 #endif