1 files changed, 67 insertions, 0 deletions
diff --git a/src/ipa/libipa/awb_bayes.h b/src/ipa/libipa/awb_bayes.h
new file mode 100644
index 00000000..47db7243
--- /dev/null
+++ b/src/ipa/libipa/awb_bayes.h
@@ -0,0 +1,67 @@
+/* SPDX-License-Identifier: LGPL-2.1-or-later */
+/*
+ * Copyright (C) 2024 Ideas on Board Oy
+ *
+ * Base class for bayes AWB algorithms
+ */
+
+#pragma once
+
+#include <map>
+#include <memory>
+#include <tuple>
+#include <vector>
+
+#include <libcamera/base/utils.h>
+
+#include <libcamera/control_ids.h>
+#include <libcamera/controls.h>
+
+#include "libcamera/internal/vector.h"
+#include "libcamera/internal/yaml_parser.h"
+
+#include "awb.h"
+#include "interpolator.h"
+#include "pwl.h"
+
+namespace libcamera {
+
+namespace ipa {
+
+class AwbBayes : public AwbAlgorithm
+{
+public:
+	AwbBayes() = default;
+
+	int init(const YamlObject &tuningData) override;
+	AwbResult calculateAwb(const AwbStats &stats, int lux) override;
+	RGB<double> gainsFromColourTemperature(double temperatureK) override;
+	void handleControls(const ControlList &controls) override;
+
+private:
+	int readPriors(const YamlObject &tuningData);
+
+	void fineSearch(double &t, double &r, double &b, ipa::Pwl const &prior,
+			const AwbStats &stats) const;
+	double coarseSearch(const ipa::Pwl &prior, const AwbStats &stats) const;
+	double interpolateQuadratic(ipa::Pwl::Point const &a,
+				    ipa::Pwl::Point const &b,
+				    ipa::Pwl::Point const &c) const;
+
+	Interpolator<Pwl> priors_;
+	Interpolator<Vector<double, 2>> colourGainCurve_;
+
+	ipa::Pwl ctR_;
+	ipa::Pwl ctB_;
+	ipa::Pwl ctRInverse_;
+	ipa::Pwl ctBInverse_;
+
+	double transversePos_;
+	double transverseNeg_;
+
+	ModeConfig *currentMode_ = nullptr;
+};
+
+} /* namespace ipa */
+
+} /* namespace libcamera */