summaryrefslogtreecommitdiffstats
path: root/src/libs/whitebalance/whitebalance.h
blob: 466fdc672eb2ad2d3a701a4c703ff9d097037e86 (plain)
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
/* ============================================================
 *
 * This file is a part of digiKam project
 * http://www.digikam.org
 *
 * Date        : 2007-16-01
 * Description : white balance color correction.
 * 
 * Copyright (C) 2007-2008 by Gilles Caulier <caulier dot gilles at gmail dot com>
 * Copyright (C) 2008 by Guillaume Castagnino <casta at xwing dot info>
 *
 * This program is free software; you can redistribute it
 * and/or modify it under the terms of the GNU General
 * Public License as published by the Free Software Foundation;
 * either version 2, or (at your option)
 * any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * ============================================================ */
 
#ifndef WHITEBALANCE_H
#define WHITEBALANCE_H

// Digikam includes.

#include "digikam_export.h"

class TQColor;

namespace Digikam
{

class WhiteBalancePriv;

class DIGIKAM_EXPORT WhiteBalance
{

public:
    
    WhiteBalance(bool sixteenBit);
    ~WhiteBalance();

    void whiteBalance(uchar *data, int width, int height, bool sixteenBit, 
                      double black=0.0, double exposition=0.0,
                      double temperature=6500.0, double green=1.0, double dark=0.5, 
                      double gamma=1.0, double saturation=1.0);

    static void autoExposureAdjustement(uchar* data, int width, int height, bool sb,
                                        double &black, double &expo); 

    static void autoWBAdjustementFromColor(const TQColor &tc, double &temperature, double &green); 

private:

    void setRGBmult();
    static void setRGBmult(double &temperature, double &green, float &mr, float &mg, float &mb);
    void setLUTv();
    void adjustWhiteBalance(uchar *data, int width, int height, bool sixteenBit);
    inline unsigned short pixelColor(int colorMult, int index, int value);

private:

    WhiteBalancePriv* d;
};

}  // NameSpace Digikam

#endif /* WHITEBALANCE_H */