summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/ipa/raspberrypi/controller/rpi/awb.cpp14
-rw-r--r--src/ipa/raspberrypi/controller/rpi/awb.hpp1
2 files changed, 15 insertions, 0 deletions
diff --git a/src/ipa/raspberrypi/controller/rpi/awb.cpp b/src/ipa/raspberrypi/controller/rpi/awb.cpp
index 469d0e53..020825e3 100644
--- a/src/ipa/raspberrypi/controller/rpi/awb.cpp
+++ b/src/ipa/raspberrypi/controller/rpi/awb.cpp
@@ -179,6 +179,20 @@ void Awb::SetManualGains(double manual_r, double manual_b)
manual_b_ = manual_b;
}
+void Awb::SwitchMode([[maybe_unused]] CameraMode const &camera_mode,
+ Metadata *metadata)
+{
+ // If fixed colour gains have been set, we should let other algorithms
+ // know by writing it into the image metadata.
+ if (manual_r_ != 0.0 && manual_b_ != 0.0) {
+ prev_sync_results_.gain_r = manual_r_;
+ prev_sync_results_.gain_g = 1.0;
+ prev_sync_results_.gain_b = manual_b_;
+ sync_results_ = prev_sync_results_;
+ }
+ metadata->Set("awb.status", prev_sync_results_);
+}
+
void Awb::fetchAsyncResults()
{
RPI_LOG("Fetch AWB results");
diff --git a/src/ipa/raspberrypi/controller/rpi/awb.hpp b/src/ipa/raspberrypi/controller/rpi/awb.hpp
index 9124d042..545d85a8 100644
--- a/src/ipa/raspberrypi/controller/rpi/awb.hpp
+++ b/src/ipa/raspberrypi/controller/rpi/awb.hpp
@@ -84,6 +84,7 @@ public:
void Read(boost::property_tree::ptree const &params) override;
void SetMode(std::string const &name) override;
void SetManualGains(double manual_r, double manual_b) override;
+ void SwitchMode(CameraMode const &camera_mode, Metadata *metadata) override;
void Prepare(Metadata *image_metadata) override;
void Process(StatisticsPtr &stats, Metadata *image_metadata) override;
struct RGB {