blob: 5c298d3b6f69a8f2c3e3ea645570e1a8a43cde67 (
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
|
/* SPDX-License-Identifier: LGPL-2.1-or-later */
/*
* Copyright (C) 2024 Ideas on Board Oy
*
* Generic AWB algorithms
*/
#pragma once
#include <map>
#include <libcamera/control_ids.h>
#include <libcamera/controls.h>
#include "libcamera/internal/vector.h"
#include "libcamera/internal/yaml_parser.h"
namespace libcamera {
namespace ipa {
struct AwbResult {
RGB<double> gains;
double colourTemperature;
};
struct AwbStats {
virtual double computeColourError(const RGB<double> &gains) const = 0;
virtual RGB<double> getRGBMeans() const = 0;
};
class AwbAlgorithm
{
public:
virtual ~AwbAlgorithm() = default;
virtual int init(const YamlObject &tuningData) = 0;
virtual AwbResult calculateAwb(const AwbStats &stats, unsigned int lux) = 0;
virtual RGB<double> gainsFromColourTemperature(double colourTemperature) = 0;
const ControlInfoMap::Map &controls() const
{
return controls_;
}
virtual void handleControls([[maybe_unused]] const ControlList &controls) {}
protected:
int parseModeConfigs(const YamlObject &tuningData,
const ControlValue &def = {});
struct ModeConfig {
double ctHi;
double ctLo;
};
ControlInfoMap::Map controls_;
std::map<controls::AwbModeEnum, AwbAlgorithm::ModeConfig> modes_;
};
} /* namespace ipa */
} /* namespace libcamera */
|
