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
73
74
75
|
/* SPDX-License-Identifier: BSD-2-Clause */
/*
* Copyright (C) 2019, Raspberry Pi (Trading) Limited
*
* ccm.h - CCM (colour correction matrix) control algorithm
*/
#pragma once
#include <vector>
#include "../ccm_algorithm.h"
#include "../pwl.h"
namespace RPiController {
/* Algorithm to calculate colour matrix. Should be placed after AWB. */
struct Matrix {
Matrix(double m0, double m1, double m2, double m3, double m4, double m5,
double m6, double m7, double m8);
Matrix();
double m[3][3];
void read(boost::property_tree::ptree const ¶ms);
};
static inline Matrix operator*(double d, Matrix const &m)
{
return Matrix(m.m[0][0] * d, m.m[0][1] * d, m.m[0][2] * d,
m.m[1][0] * d, m.m[1][1] * d, m.m[1][2] * d,
m.m[2][0] * d, m.m[2][1] * d, m.m[2][2] * d);
}
static inline Matrix operator*(Matrix const &m1, Matrix const &m2)
{
Matrix m;
for (int i = 0; i < 3; i++)
for (int j = 0; j < 3; j++)
m.m[i][j] = m1.m[i][0] * m2.m[0][j] +
m1.m[i][1] * m2.m[1][j] +
m1.m[i][2] * m2.m[2][j];
return m;
}
static inline Matrix operator+(Matrix const &m1, Matrix const &m2)
{
Matrix m;
for (int i = 0; i < 3; i++)
for (int j = 0; j < 3; j++)
m.m[i][j] = m1.m[i][j] + m2.m[i][j];
return m;
}
struct CtCcm {
double ct;
Matrix ccm;
};
struct CcmConfig {
std::vector<CtCcm> ccms;
Pwl saturation;
};
class Ccm : public CcmAlgorithm
{
public:
Ccm(Controller *controller = NULL);
char const *name() const override;
void read(boost::property_tree::ptree const ¶ms) override;
void setSaturation(double saturation) override;
void initialise() override;
void prepare(Metadata *imageMetadata) override;
private:
CcmConfig config_;
double saturation_;
};
} /* namespace RPiController */
|
