[v3,08/11] ipa: c3-isp: Add CCM Algorithm
diff mbox series

Message ID 20250227105733.187611-9-keke.li@amlogic.com
State New
Headers show
Series
  • Add Amlogic C3 ISP pipeline handler and IPA
Related show

Commit Message

Keke Li Feb. 27, 2025, 10:57 a.m. UTC 
Add a new Colour Correction Matrix algorithm

Signed-off-by: Keke Li <keke.li@amlogic.com>
---
 src/ipa/c3-isp/algorithms/ccm.cpp     | 92 +++++++++++++++++++++++++++
 src/ipa/c3-isp/algorithms/ccm.h       | 36 +++++++++++
 src/ipa/c3-isp/algorithms/meson.build |  1 +
 3 files changed, 129 insertions(+)
 create mode 100644 src/ipa/c3-isp/algorithms/ccm.cpp
 create mode 100644 src/ipa/c3-isp/algorithms/ccm.h

Patch
diff mbox series 

diff --git a/src/ipa/c3-isp/algorithms/ccm.cpp b/src/ipa/c3-isp/algorithms/ccm.cpp
new file mode 100644
index 00000000..c0efb539
--- /dev/null
+++ b/src/ipa/c3-isp/algorithms/ccm.cpp
@@ -0,0 +1,92 @@ 
+/* SPDX-License-Identifier: LGPL-2.1-or-later */
+/*
+ * Copyright (C) 2024, Amlogic
+ *
+ * C3ISP Color Correction Matrix control
+ */
+
+#include "ccm.h"
+
+#include <libcamera/base/log.h>
+#include <libcamera/base/utils.h>
+
+#include <libcamera/control_ids.h>
+
+#include "libcamera/internal/yaml_parser.h"
+
+/**
+ * \file ccm.h
+ */
+
+namespace libcamera {
+
+namespace ipa::c3isp::algorithms {
+
+/**
+ * \class Ccm
+ * \brief A color correction matrix algorithm
+ */
+
+LOG_DEFINE_CATEGORY(C3ISPCcm)
+
+Ccm::Ccm()
+{
+}
+
+/**
+ * \copydoc libcamera::ipa::Algorithm::init
+ */
+int Ccm::init([[maybe_unused]] IPAContext &context, const YamlObject &tuningData)
+{
+	ccmCoeff_ = tuningData["ccmCoeff"].getList<int16_t>().value_or(std::vector<int16_t>{});
+	if (ccmCoeff_.size() != 9) {
+		LOG(C3ISPCcm, Error) << "Invalid CCM coeff size";
+		return -EINVAL;
+	}
+
+	return 0;
+}
+
+/**
+ * \copydoc libcamera::ipa::Algorithm::prepare
+ */
+void Ccm::prepare([[maybe_unused]] IPAContext &context,
+		  const uint32_t frame,
+		  [[maybe_unused]] IPAFrameContext &frameContext, C3ISPParams *params)
+{
+	if (frame > 0)
+		return;
+
+	auto CcmCfg = params->block<BlockType::Ccm>();
+	CcmCfg.setEnabled(C3_ISP_PARAMS_BLOCK_FL_ENABLE);
+
+	for (unsigned int i = 0; i < 3; i++) {
+		for (unsigned int j = 0; j < 3; j++) {
+			CcmCfg->matrix[i][j] = ccmCoeff_[i * 3 + j];
+		}
+	}
+}
+
+/**
+ * \copydoc libcamera::ipa::Algorithm::process
+ */
+void Ccm::process([[maybe_unused]] IPAContext &context,
+		  [[maybe_unused]] const uint32_t frame,
+		  [[maybe_unused]] IPAFrameContext &frameContext,
+		  [[maybe_unused]] const c3_isp_stats_info *stats,
+		  ControlList &metadata)
+{
+	float m[9];
+	for (unsigned int i = 0; i < 3; i++) {
+		for (unsigned int j = 0; j < 3; j++)
+			m[i * 3 + j] = ccmCoeff_[i * 3 + j] / 256.0;
+	}
+
+	metadata.set(controls::ColourCorrectionMatrix, m);
+}
+
+REGISTER_IPA_ALGORITHM(Ccm, "Ccm")
+
+} /* namespace ipa::c3isp::algorithms */
+
+} /* namespace libcamera */
diff --git a/src/ipa/c3-isp/algorithms/ccm.h b/src/ipa/c3-isp/algorithms/ccm.h
new file mode 100644
index 00000000..c65e7edf
--- /dev/null
+++ b/src/ipa/c3-isp/algorithms/ccm.h
@@ -0,0 +1,36 @@ 
+/* SPDX-License-Identifier: LGPL-2.1-or-later */
+/*
+ * Copyright (C) 2024, Amlogic
+ *
+ * C3ISP Color Correction Matrix control
+ */
+
+#pragma once
+
+#include "algorithm.h"
+
+namespace libcamera {
+
+namespace ipa::c3isp::algorithms {
+
+class Ccm : public Algorithm
+{
+public:
+	Ccm();
+	~Ccm() = default;
+
+	int init(IPAContext &context, const YamlObject &tuningData) override;
+	void prepare(IPAContext &context, const uint32_t frame,
+		     IPAFrameContext &frameContext,
+		     C3ISPParams *params) override;
+	void process(IPAContext &context, const uint32_t frame,
+		     IPAFrameContext &frameContext, const c3_isp_stats_info *stats,
+		     ControlList &metadata) override;
+
+private:
+	std::vector<int16_t> ccmCoeff_;
+};
+
+} /* namespace ipa::c3isp::algorithms */
+
+} /* namespace libcamera */
diff --git a/src/ipa/c3-isp/algorithms/meson.build b/src/ipa/c3-isp/algorithms/meson.build
index d91973e3..591c2433 100644
--- a/src/ipa/c3-isp/algorithms/meson.build
+++ b/src/ipa/c3-isp/algorithms/meson.build
@@ -4,4 +4,5 @@  c3isp_ipa_algorithms = files([
     'agc.cpp',
     'awb.cpp',
     'blc.cpp',
+    'ccm.cpp',
 ])