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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
/* SPDX-License-Identifier: LGPL-2.1-or-later */
/*
* Copyright (C) 2024, Red Hat Inc.
*
* Auto white balance
*/
#include "awb.h"
#include <numeric>
#include <stdint.h>
#include <libcamera/base/log.h>
#include <libcamera/control_ids.h>
#include "libipa/colours.h"
#include "simple/ipa_context.h"
#include "control_ids.h"
namespace libcamera {
LOG_DEFINE_CATEGORY(IPASoftAwb)
namespace ipa::soft::algorithms {
int Awb::configure(IPAContext &context,
[[maybe_unused]] const IPAConfigInfo &configInfo)
{
auto &gains = context.activeState.awb.gains;
gains = { { 1.0, 1.0, 1.0 } };
return 0;
}
void Awb::prepare(IPAContext &context,
[[maybe_unused]] const uint32_t frame,
IPAFrameContext &frameContext,
[[maybe_unused]] DebayerParams *params)
{
auto &gains = context.activeState.awb.gains;
frameContext.gains.red = gains.r();
frameContext.gains.blue = gains.b();
}
void Awb::process(IPAContext &context,
[[maybe_unused]] const uint32_t frame,
IPAFrameContext &frameContext,
const SwIspStats *stats,
ControlList &metadata)
{
const SwIspStats::Histogram &histogram = stats->yHistogram;
const uint8_t blackLevel = context.activeState.blc.level;
const float maxGain = 1024.0;
const float mdGains[] = {
static_cast<float>(frameContext.gains.red / maxGain),
static_cast<float>(frameContext.gains.blue / maxGain)
};
metadata.set(controls::ColourGains, mdGains);
/*
* Black level must be subtracted to get the correct AWB ratios, they
* would be off if they were computed from the whole brightness range
* rather than from the sensor range.
*/
const uint64_t nPixels = std::accumulate(
histogram.begin(), histogram.end(), 0);
const uint64_t offset = blackLevel * nPixels;
const uint64_t sumR = stats->sumR_ - offset / 4;
const uint64_t sumG = stats->sumG_ - offset / 2;
const uint64_t sumB = stats->sumB_ - offset / 4;
/*
* Calculate red and blue gains for AWB.
* Clamp max gain at 4.0, this also avoids 0 division.
*/
auto &gains = context.activeState.awb.gains;
gains = { {
sumR <= sumG / 4 ? 4.0f : static_cast<float>(sumG) / sumR,
1.0,
sumB <= sumG / 4 ? 4.0f : static_cast<float>(sumG) / sumB,
} };
RGB<double> rgbGains{ { 1 / gains.r(), 1 / gains.g(), 1 / gains.b() } };
context.activeState.awb.temperatureK = estimateCCT(rgbGains);
metadata.set(controls::ColourTemperature, context.activeState.awb.temperatureK);
LOG(IPASoftAwb, Debug)
<< "gain R/B: " << gains << "; temperature: "
<< context.activeState.awb.temperatureK;
}
REGISTER_IPA_ALGORITHM(Awb, "Awb")
} /* namespace ipa::soft::algorithms */
} /* namespace libcamera */
|
