summaryrefslogtreecommitdiff
path: root/src/ipa/raspberrypi/controller/rpi/sdn.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/ipa/raspberrypi/controller/rpi/sdn.cpp')
-rw-r--r--src/ipa/raspberrypi/controller/rpi/sdn.cpp63
1 files changed, 63 insertions, 0 deletions
diff --git a/src/ipa/raspberrypi/controller/rpi/sdn.cpp b/src/ipa/raspberrypi/controller/rpi/sdn.cpp
new file mode 100644
index 00000000..28d9d983
--- /dev/null
+++ b/src/ipa/raspberrypi/controller/rpi/sdn.cpp
@@ -0,0 +1,63 @@
+/* SPDX-License-Identifier: BSD-2-Clause */
+/*
+ * Copyright (C) 2019, Raspberry Pi (Trading) Limited
+ *
+ * sdn.cpp - SDN (spatial denoise) control algorithm
+ */
+
+#include "../noise_status.h"
+#include "../sdn_status.h"
+
+#include "sdn.hpp"
+
+using namespace RPi;
+
+// Calculate settings for the spatial denoise block using the noise profile in
+// the image metadata.
+
+#define NAME "rpi.sdn"
+
+Sdn::Sdn(Controller *controller)
+ : Algorithm(controller)
+{
+}
+
+char const *Sdn::Name() const
+{
+ return NAME;
+}
+
+void Sdn::Read(boost::property_tree::ptree const &params)
+{
+ deviation_ = params.get<double>("deviation", 3.2);
+ strength_ = params.get<double>("strength", 0.75);
+}
+
+void Sdn::Initialise() {}
+
+void Sdn::Prepare(Metadata *image_metadata)
+{
+ struct NoiseStatus noise_status = {};
+ noise_status.noise_slope = 3.0; // in case no metadata
+ if (image_metadata->Get("noise.status", noise_status) != 0)
+ RPI_WARN("Sdn: no noise profile found");
+ RPI_LOG("Noise profile: constant " << noise_status.noise_constant
+ << " slope "
+ << noise_status.noise_slope);
+ struct SdnStatus status;
+ status.noise_constant = noise_status.noise_constant * deviation_;
+ status.noise_slope = noise_status.noise_slope * deviation_;
+ status.strength = strength_;
+ image_metadata->Set("sdn.status", status);
+ RPI_LOG("Sdn: programmed constant " << status.noise_constant
+ << " slope " << status.noise_slope
+ << " strength "
+ << status.strength);
+}
+
+// Register algorithm with the system.
+static Algorithm *Create(Controller *controller)
+{
+ return (Algorithm *)new Sdn(controller);
+}
+static RegisterAlgorithm reg(NAME, &Create);