summaryrefslogtreecommitdiff
path: root/src/ipa/raspberrypi/controller/rpi/agc.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/ipa/raspberrypi/controller/rpi/agc.hpp')
-rw-r--r--src/ipa/raspberrypi/controller/rpi/agc.hpp123
1 files changed, 123 insertions, 0 deletions
diff --git a/src/ipa/raspberrypi/controller/rpi/agc.hpp b/src/ipa/raspberrypi/controller/rpi/agc.hpp
new file mode 100644
index 00000000..dbcefba6
--- /dev/null
+++ b/src/ipa/raspberrypi/controller/rpi/agc.hpp
@@ -0,0 +1,123 @@
+/* SPDX-License-Identifier: BSD-2-Clause */
+/*
+ * Copyright (C) 2019, Raspberry Pi (Trading) Limited
+ *
+ * agc.hpp - AGC/AEC control algorithm
+ */
+#pragma once
+
+#include <vector>
+#include <mutex>
+
+#include "../agc_algorithm.hpp"
+#include "../agc_status.h"
+#include "../pwl.hpp"
+
+// This is our implementation of AGC.
+
+// This is the number actually set up by the firmware, not the maximum possible
+// number (which is 16).
+
+#define AGC_STATS_SIZE 15
+
+namespace RPi {
+
+struct AgcMeteringMode {
+ double weights[AGC_STATS_SIZE];
+ void Read(boost::property_tree::ptree const &params);
+};
+
+struct AgcExposureMode {
+ std::vector<double> shutter;
+ std::vector<double> gain;
+ void Read(boost::property_tree::ptree const &params);
+};
+
+struct AgcConstraint {
+ enum class Bound { LOWER = 0, UPPER = 1 };
+ Bound bound;
+ double q_lo;
+ double q_hi;
+ Pwl Y_target;
+ void Read(boost::property_tree::ptree const &params);
+};
+
+typedef std::vector<AgcConstraint> AgcConstraintMode;
+
+struct AgcConfig {
+ void Read(boost::property_tree::ptree const &params);
+ std::map<std::string, AgcMeteringMode> metering_modes;
+ std::map<std::string, AgcExposureMode> exposure_modes;
+ std::map<std::string, AgcConstraintMode> constraint_modes;
+ Pwl Y_target;
+ double speed;
+ uint16_t startup_frames;
+ double max_change;
+ double min_change;
+ double fast_reduce_threshold;
+ double speed_up_threshold;
+ std::string default_metering_mode;
+ std::string default_exposure_mode;
+ std::string default_constraint_mode;
+ double base_ev;
+};
+
+class Agc : public AgcAlgorithm
+{
+public:
+ Agc(Controller *controller);
+ char const *Name() const override;
+ void Read(boost::property_tree::ptree const &params) override;
+ void SetEv(double ev) override;
+ void SetFlickerPeriod(double flicker_period) override;
+ void SetFixedShutter(double fixed_shutter) override; // microseconds
+ void SetFixedAnalogueGain(double fixed_analogue_gain) override;
+ void SetMeteringMode(std::string const &metering_mode_name) override;
+ void SetExposureMode(std::string const &exposure_mode_name) override;
+ void SetConstraintMode(std::string const &contraint_mode_name) override;
+ void Prepare(Metadata *image_metadata) override;
+ void Process(StatisticsPtr &stats, Metadata *image_metadata) override;
+
+private:
+ AgcConfig config_;
+ void housekeepConfig();
+ void fetchCurrentExposure(Metadata *image_metadata);
+ void computeGain(bcm2835_isp_stats *statistics, Metadata *image_metadata,
+ double &gain, double &target_Y);
+ void computeTargetExposure(double gain);
+ bool applyDigitalGain(Metadata *image_metadata, double gain,
+ double target_Y);
+ void filterExposure(bool desaturate);
+ void divvyupExposure();
+ void writeAndFinish(Metadata *image_metadata, bool desaturate);
+ AgcMeteringMode *metering_mode_;
+ AgcExposureMode *exposure_mode_;
+ AgcConstraintMode *constraint_mode_;
+ uint64_t frame_count_;
+ struct ExposureValues {
+ ExposureValues() : shutter(0), analogue_gain(0),
+ total_exposure(0), total_exposure_no_dg(0) {}
+ double shutter;
+ double analogue_gain;
+ double total_exposure;
+ double total_exposure_no_dg; // without digital gain
+ };
+ ExposureValues current_; // values for the current frame
+ ExposureValues target_; // calculate the values we want here
+ ExposureValues filtered_; // these values are filtered towards target
+ AgcStatus status_; // to "latch" settings so they can't change
+ AgcStatus output_status_; // the status we will write out
+ std::mutex output_mutex_;
+ int lock_count_;
+ // Below here the "settings" that applications can change.
+ std::mutex settings_mutex_;
+ std::string metering_mode_name_;
+ std::string exposure_mode_name_;
+ std::string constraint_mode_name_;
+ double ev_;
+ double flicker_period_;
+ double fixed_shutter_;
+ double fixed_analogue_gain_;
+};
+
+} // namespace RPi