diff options
Diffstat (limited to 'src/ipa')
| -rw-r--r-- | src/ipa/rpi/controller/rpi/cac.cpp | 82 | ||||
| -rw-r--r-- | src/ipa/rpi/controller/rpi/cac.h | 5 |
2 files changed, 55 insertions, 32 deletions
diff --git a/src/ipa/rpi/controller/rpi/cac.cpp b/src/ipa/rpi/controller/rpi/cac.cpp index 7c123da1..f2c8d282 100644 --- a/src/ipa/rpi/controller/rpi/cac.cpp +++ b/src/ipa/rpi/controller/rpi/cac.cpp @@ -27,40 +27,23 @@ char const *Cac::name() const return NAME; } -int Cac::read(const libcamera::YamlObject ¶ms) -{ - arrayToSet(params["lut_rx"], config_.lutRx); - arrayToSet(params["lut_ry"], config_.lutRy); - arrayToSet(params["lut_bx"], config_.lutBx); - arrayToSet(params["lut_by"], config_.lutBy); - cacStatus_.lutRx = config_.lutRx; - cacStatus_.lutRy = config_.lutRy; - cacStatus_.lutBx = config_.lutBx; - cacStatus_.lutBy = config_.lutBy; - double strength = params["strength"].get<double>(1); - setStrength(config_.lutRx, cacStatus_.lutRx, strength); - setStrength(config_.lutBx, cacStatus_.lutBx, strength); - setStrength(config_.lutRy, cacStatus_.lutRy, strength); - setStrength(config_.lutBy, cacStatus_.lutBy, strength); - return 0; -} - -void Cac::initialise() -{ -} - -void Cac::arrayToSet(const libcamera::YamlObject ¶ms, std::vector<double> &inputArray) +static bool arrayToSet(const libcamera::YamlObject ¶ms, std::vector<double> &inputArray, const Size &size) { int num = 0; - const Size &size = getHardwareConfig().cacRegions; - inputArray.resize((size.width + 1) * (size.height + 1)); + int max_num = (size.width + 1) * (size.height + 1); + inputArray.resize(max_num); + for (const auto &p : params.asList()) { + if (num == max_num) + return false; inputArray[num++] = p.get<double>(0); } + + return num == max_num; } -void Cac::setStrength(std::vector<double> &inputArray, std::vector<double> &outputArray, - double strengthFactor) +static void setStrength(std::vector<double> &inputArray, std::vector<double> &outputArray, + double strengthFactor) { int num = 0; for (const auto &p : inputArray) { @@ -68,9 +51,52 @@ void Cac::setStrength(std::vector<double> &inputArray, std::vector<double> &outp } } +int Cac::read(const libcamera::YamlObject ¶ms) +{ + config_.enabled = params.contains("lut_rx") && params.contains("lut_ry") && + params.contains("lut_bx") && params.contains("lut_by"); + if (!config_.enabled) + return 0; + + const Size &size = getHardwareConfig().cacRegions; + + if (!arrayToSet(params["lut_rx"], config_.lutRx, size)) { + LOG(RPiCac, Error) << "Bad CAC lut_rx table"; + return -EINVAL; + } + + if (!arrayToSet(params["lut_ry"], config_.lutRy, size)) { + LOG(RPiCac, Error) << "Bad CAC lut_ry table"; + return -EINVAL; + } + + if (!arrayToSet(params["lut_bx"], config_.lutBx, size)) { + LOG(RPiCac, Error) << "Bad CAC lut_bx table"; + return -EINVAL; + } + + if (!arrayToSet(params["lut_by"], config_.lutBy, size)) { + LOG(RPiCac, Error) << "Bad CAC lut_by table"; + return -EINVAL; + } + + double strength = params["strength"].get<double>(1); + cacStatus_.lutRx = config_.lutRx; + cacStatus_.lutRy = config_.lutRy; + cacStatus_.lutBx = config_.lutBx; + cacStatus_.lutBy = config_.lutBy; + setStrength(config_.lutRx, cacStatus_.lutRx, strength); + setStrength(config_.lutBx, cacStatus_.lutBx, strength); + setStrength(config_.lutRy, cacStatus_.lutRy, strength); + setStrength(config_.lutBy, cacStatus_.lutBy, strength); + + return 0; +} + void Cac::prepare(Metadata *imageMetadata) { - imageMetadata->set("cac.status", cacStatus_); + if (config_.enabled) + imageMetadata->set("cac.status", cacStatus_); } // Register algorithm with the system. diff --git a/src/ipa/rpi/controller/rpi/cac.h b/src/ipa/rpi/controller/rpi/cac.h index 419180ab..a7b14c00 100644 --- a/src/ipa/rpi/controller/rpi/cac.h +++ b/src/ipa/rpi/controller/rpi/cac.h @@ -12,6 +12,7 @@ namespace RPiController { struct CacConfig { + bool enabled; std::vector<double> lutRx; std::vector<double> lutRy; std::vector<double> lutBx; @@ -24,15 +25,11 @@ public: Cac(Controller *controller = NULL); char const *name() const override; int read(const libcamera::YamlObject ¶ms) override; - void initialise() override; void prepare(Metadata *imageMetadata) override; - void setStrength(std::vector<double> &inputArray, std::vector<double> &outputArray, - double strengthFactor); private: CacConfig config_; CacStatus cacStatus_; - void arrayToSet(const libcamera::YamlObject ¶ms, std::vector<double> &inputArray); }; } // namespace RPiController |