NekoCWD/libcamera
NekoCWD/libcamera
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 7 Wiki Activity Actions

ipa: raspberrypi: awb: Update colour temperature whenever manual gains change

Browse source 
Previously we only did this when the system starts (on the first
switch_mode). Now we do it whenever the manual colour gains are
updated. To facilitate this, this R/B vs. colour temperature inverse
functions are stored persistently in the AwbConfig.

Signed-off-by: David Plowman <david.plowman@raspberrypi.com>
Reviewed-by: Naushir Patuck <naush@raspberrypi.com>
Reviewed-by: Kieran Bingham <kieran.bingham@ideasonboard.com>
Signed-off-by: Kieran Bingham <kieran.bingham@ideasonboard.com>
This commit is contained in:
David Plowman 2022-10-27 12:39:56 +01:00 • committed by Kieran Bingham
parent 30d704732b
commit 375a70d43e
2 changed files with 12 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/ipa/raspberrypi/controller/rpi/awb.cpp
View file

@ -104,6 +104,9 @@ int AwbConfig::read(const libcamera::YamlObject &params)
ret = readCtCurve(ctR, ctB, params["ct_curve"]);
if (ret)
return ret;
/* We will want the inverse functions of these too. */
ctRInverse = ctR.inverse();
ctBInverse = ctB.inverse();
}
if (params.contains("priors")) {
@ -174,7 +177,6 @@ Awb::Awb(Controller *controller)
asyncAbort_ = asyncStart_ = asyncStarted_ = asyncFinished_ = false;
mode_ = nullptr;
manualR_ = manualB_ = 0.0;
firstSwitchMode_ = true;
asyncThread_ = std::thread(std::bind(&Awb::asyncFunc, this));
}
@ -270,27 +272,21 @@ void Awb::setManualGains(double manualR, double manualB)
syncResults_.gainR = prevSyncResults_.gainR = manualR_;
syncResults_.gainG = prevSyncResults_.gainG = 1.0;
syncResults_.gainB = prevSyncResults_.gainB = manualB_;
if (config_.bayes) {
/* Also estimate the best corresponding colour temperature from the curves. */
double ctR = config_.ctRInverse.eval(config_.ctRInverse.domain().clip(1 / manualR_));
double ctB = config_.ctBInverse.eval(config_.ctBInverse.domain().clip(1 / manualB_));
prevSyncResults_.temperatureK = (ctR + ctB) / 2;
syncResults_.temperatureK = prevSyncResults_.temperatureK;
}
}
}
void Awb::switchMode([[maybe_unused]] CameraMode const &cameraMode,
Metadata *metadata)
{
/*
* On the first mode switch we'll have no meaningful colour
* temperature, so try to dead reckon one if in manual mode.
*/
if (!isAutoEnabled() && firstSwitchMode_ && config_.bayes) {
Pwl ctRInverse = config_.ctR.inverse();
Pwl ctBInverse = config_.ctB.inverse();
double ctR = ctRInverse.eval(ctRInverse.domain().clip(1 / manualR_));
double ctB = ctBInverse.eval(ctBInverse.domain().clip(1 / manualB_));
prevSyncResults_.temperatureK = (ctR + ctB) / 2;
syncResults_.temperatureK = prevSyncResults_.temperatureK;
}
/* Let other algorithms know the current white balance values. */
metadata->set("awb.status", prevSyncResults_);
firstSwitchMode_ = false;
}
bool Awb::isAutoEnabled() const