ipa: Sort algorithm operations based on calling order
Reorder functions in the base ipa::Algorithm and its derived classes to match the calling order: queueRequest(), prepare() and process(). This makes the code flow easier to read. No functional change intended. Signed-off-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com> Reviewed-by: Paul Elder <paul.elder@ideasonboard.com> Reviewed-by: Jacopo Mondi <jacopo@jmondi.org>
This commit is contained in:
Laurent Pinchart
parent
bc0d9a6f5d
commit
02bc2a8a24
7 changed files with 209 additions and 209 deletions
|
|
@ -113,22 +113,6 @@ Af::Af()
|
|||
{
|
||||
}
|
||||
|
||||
/**
|
||||
* \copydoc libcamera::ipa::Algorithm::prepare
|
||||
*/
|
||||
void Af::prepare(IPAContext &context,
|
||||
[[maybe_unused]] const uint32_t frame,
|
||||
[[maybe_unused]] IPAFrameContext &frameContext,
|
||||
ipu3_uapi_params *params)
|
||||
{
|
||||
const struct ipu3_uapi_grid_config &grid = context.configuration.af.afGrid;
|
||||
params->acc_param.af.grid_cfg = grid;
|
||||
params->acc_param.af.filter_config = afFilterConfigDefault;
|
||||
|
||||
/* Enable AF processing block */
|
||||
params->use.acc_af = 1;
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Configure the Af given a configInfo
|
||||
* \param[in] context The shared IPA context
|
||||
|
|
@ -197,6 +181,22 @@ int Af::configure(IPAContext &context, const IPAConfigInfo &configInfo)
|
|||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* \copydoc libcamera::ipa::Algorithm::prepare
|
||||
*/
|
||||
void Af::prepare(IPAContext &context,
|
||||
[[maybe_unused]] const uint32_t frame,
|
||||
[[maybe_unused]] IPAFrameContext &frameContext,
|
||||
ipu3_uapi_params *params)
|
||||
{
|
||||
const struct ipu3_uapi_grid_config &grid = context.configuration.af.afGrid;
|
||||
params->acc_param.af.grid_cfg = grid;
|
||||
params->acc_param.af.filter_config = afFilterConfigDefault;
|
||||
|
||||
/* Enable AF processing block */
|
||||
params->use.acc_af = 1;
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief AF coarse scan
|
||||
* \param[in] context The shared IPA context
|
||||
|
|
|
|||
|
|
@ -218,6 +218,89 @@ int Awb::configure(IPAContext &context,
|
|||
return 0;
|
||||
}
|
||||
|
||||
constexpr uint16_t Awb::threshold(float value)
|
||||
{
|
||||
/* AWB thresholds are in the range [0, 8191] */
|
||||
return value * 8191;
|
||||
}
|
||||
|
||||
constexpr uint16_t Awb::gainValue(double gain)
|
||||
{
|
||||
/*
|
||||
* The colour gains applied by the BNR for the four channels (Gr, R, B
|
||||
* and Gb) are expressed in the parameters structure as 16-bit integers
|
||||
* that store a fixed-point U3.13 value in the range [0, 8[.
|
||||
*
|
||||
* The real gain value is equal to the gain parameter plus one, i.e.
|
||||
*
|
||||
* Pout = Pin * (1 + gain / 8192)
|
||||
*
|
||||
* where 'Pin' is the input pixel value, 'Pout' the output pixel value,
|
||||
* and 'gain' the gain in the parameters structure as a 16-bit integer.
|
||||
*/
|
||||
return std::clamp((gain - 1.0) * 8192, 0.0, 65535.0);
|
||||
}
|
||||
|
||||
/**
|
||||
* \copydoc libcamera::ipa::Algorithm::prepare
|
||||
*/
|
||||
void Awb::prepare(IPAContext &context,
|
||||
[[maybe_unused]] const uint32_t frame,
|
||||
[[maybe_unused]] IPAFrameContext &frameContext,
|
||||
ipu3_uapi_params *params)
|
||||
{
|
||||
/*
|
||||
* Green saturation thresholds are reduced because we are using the
|
||||
* green channel only in the exposure computation.
|
||||
*/
|
||||
params->acc_param.awb.config.rgbs_thr_r = threshold(1.0);
|
||||
params->acc_param.awb.config.rgbs_thr_gr = threshold(0.9);
|
||||
params->acc_param.awb.config.rgbs_thr_gb = threshold(0.9);
|
||||
params->acc_param.awb.config.rgbs_thr_b = threshold(1.0);
|
||||
|
||||
/*
|
||||
* Enable saturation inclusion on thr_b for ImgU to update the
|
||||
* ipu3_uapi_awb_set_item->sat_ratio field.
|
||||
*/
|
||||
params->acc_param.awb.config.rgbs_thr_b |= IPU3_UAPI_AWB_RGBS_THR_B_INCL_SAT |
|
||||
IPU3_UAPI_AWB_RGBS_THR_B_EN;
|
||||
|
||||
const ipu3_uapi_grid_config &grid = context.configuration.grid.bdsGrid;
|
||||
|
||||
params->acc_param.awb.config.grid = context.configuration.grid.bdsGrid;
|
||||
|
||||
/*
|
||||
* Optical center is column start (respectively row start) of the
|
||||
* cell of interest minus its X center (respectively Y center).
|
||||
*
|
||||
* For the moment use BDS as a first approximation, but it should
|
||||
* be calculated based on Shading (SHD) parameters.
|
||||
*/
|
||||
params->acc_param.bnr = imguCssBnrDefaults;
|
||||
Size &bdsOutputSize = context.configuration.grid.bdsOutputSize;
|
||||
params->acc_param.bnr.column_size = bdsOutputSize.width;
|
||||
params->acc_param.bnr.opt_center.x_reset = grid.x_start - (bdsOutputSize.width / 2);
|
||||
params->acc_param.bnr.opt_center.y_reset = grid.y_start - (bdsOutputSize.height / 2);
|
||||
params->acc_param.bnr.opt_center_sqr.x_sqr_reset = params->acc_param.bnr.opt_center.x_reset
|
||||
* params->acc_param.bnr.opt_center.x_reset;
|
||||
params->acc_param.bnr.opt_center_sqr.y_sqr_reset = params->acc_param.bnr.opt_center.y_reset
|
||||
* params->acc_param.bnr.opt_center.y_reset;
|
||||
|
||||
params->acc_param.bnr.wb_gains.gr = gainValue(context.activeState.awb.gains.green);
|
||||
params->acc_param.bnr.wb_gains.r = gainValue(context.activeState.awb.gains.red);
|
||||
params->acc_param.bnr.wb_gains.b = gainValue(context.activeState.awb.gains.blue);
|
||||
params->acc_param.bnr.wb_gains.gb = gainValue(context.activeState.awb.gains.green);
|
||||
|
||||
LOG(IPU3Awb, Debug) << "Color temperature estimated: " << asyncResults_.temperatureK;
|
||||
|
||||
/* The CCM matrix may change when color temperature will be used */
|
||||
params->acc_param.ccm = imguCssCcmDefault;
|
||||
|
||||
params->use.acc_awb = 1;
|
||||
params->use.acc_bnr = 1;
|
||||
params->use.acc_ccm = 1;
|
||||
}
|
||||
|
||||
/**
|
||||
* The function estimates the correlated color temperature using
|
||||
* from RGB color space input.
|
||||
|
|
@ -415,89 +498,6 @@ void Awb::process(IPAContext &context, [[maybe_unused]] const uint32_t frame,
|
|||
context.activeState.awb.temperatureK);
|
||||
}
|
||||
|
||||
constexpr uint16_t Awb::threshold(float value)
|
||||
{
|
||||
/* AWB thresholds are in the range [0, 8191] */
|
||||
return value * 8191;
|
||||
}
|
||||
|
||||
constexpr uint16_t Awb::gainValue(double gain)
|
||||
{
|
||||
/*
|
||||
* The colour gains applied by the BNR for the four channels (Gr, R, B
|
||||
* and Gb) are expressed in the parameters structure as 16-bit integers
|
||||
* that store a fixed-point U3.13 value in the range [0, 8[.
|
||||
*
|
||||
* The real gain value is equal to the gain parameter plus one, i.e.
|
||||
*
|
||||
* Pout = Pin * (1 + gain / 8192)
|
||||
*
|
||||
* where 'Pin' is the input pixel value, 'Pout' the output pixel value,
|
||||
* and 'gain' the gain in the parameters structure as a 16-bit integer.
|
||||
*/
|
||||
return std::clamp((gain - 1.0) * 8192, 0.0, 65535.0);
|
||||
}
|
||||
|
||||
/**
|
||||
* \copydoc libcamera::ipa::Algorithm::prepare
|
||||
*/
|
||||
void Awb::prepare(IPAContext &context,
|
||||
[[maybe_unused]] const uint32_t frame,
|
||||
[[maybe_unused]] IPAFrameContext &frameContext,
|
||||
ipu3_uapi_params *params)
|
||||
{
|
||||
/*
|
||||
* Green saturation thresholds are reduced because we are using the
|
||||
* green channel only in the exposure computation.
|
||||
*/
|
||||
params->acc_param.awb.config.rgbs_thr_r = threshold(1.0);
|
||||
params->acc_param.awb.config.rgbs_thr_gr = threshold(0.9);
|
||||
params->acc_param.awb.config.rgbs_thr_gb = threshold(0.9);
|
||||
params->acc_param.awb.config.rgbs_thr_b = threshold(1.0);
|
||||
|
||||
/*
|
||||
* Enable saturation inclusion on thr_b for ImgU to update the
|
||||
* ipu3_uapi_awb_set_item->sat_ratio field.
|
||||
*/
|
||||
params->acc_param.awb.config.rgbs_thr_b |= IPU3_UAPI_AWB_RGBS_THR_B_INCL_SAT |
|
||||
IPU3_UAPI_AWB_RGBS_THR_B_EN;
|
||||
|
||||
const ipu3_uapi_grid_config &grid = context.configuration.grid.bdsGrid;
|
||||
|
||||
params->acc_param.awb.config.grid = context.configuration.grid.bdsGrid;
|
||||
|
||||
/*
|
||||
* Optical center is column start (respectively row start) of the
|
||||
* cell of interest minus its X center (respectively Y center).
|
||||
*
|
||||
* For the moment use BDS as a first approximation, but it should
|
||||
* be calculated based on Shading (SHD) parameters.
|
||||
*/
|
||||
params->acc_param.bnr = imguCssBnrDefaults;
|
||||
Size &bdsOutputSize = context.configuration.grid.bdsOutputSize;
|
||||
params->acc_param.bnr.column_size = bdsOutputSize.width;
|
||||
params->acc_param.bnr.opt_center.x_reset = grid.x_start - (bdsOutputSize.width / 2);
|
||||
params->acc_param.bnr.opt_center.y_reset = grid.y_start - (bdsOutputSize.height / 2);
|
||||
params->acc_param.bnr.opt_center_sqr.x_sqr_reset = params->acc_param.bnr.opt_center.x_reset
|
||||
* params->acc_param.bnr.opt_center.x_reset;
|
||||
params->acc_param.bnr.opt_center_sqr.y_sqr_reset = params->acc_param.bnr.opt_center.y_reset
|
||||
* params->acc_param.bnr.opt_center.y_reset;
|
||||
|
||||
params->acc_param.bnr.wb_gains.gr = gainValue(context.activeState.awb.gains.green);
|
||||
params->acc_param.bnr.wb_gains.r = gainValue(context.activeState.awb.gains.red);
|
||||
params->acc_param.bnr.wb_gains.b = gainValue(context.activeState.awb.gains.blue);
|
||||
params->acc_param.bnr.wb_gains.gb = gainValue(context.activeState.awb.gains.green);
|
||||
|
||||
LOG(IPU3Awb, Debug) << "Color temperature estimated: " << asyncResults_.temperatureK;
|
||||
|
||||
/* The CCM matrix may change when color temperature will be used */
|
||||
params->acc_param.ccm = imguCssCcmDefault;
|
||||
|
||||
params->use.acc_awb = 1;
|
||||
params->use.acc_bnr = 1;
|
||||
params->use.acc_ccm = 1;
|
||||
}
|
||||
|
||||
REGISTER_IPA_ALGORITHM(Awb, "Awb")
|
||||
|
||||
} /* namespace ipa::ipu3::algorithms */
|
||||
|
|
|
|||
|
|
@ -66,23 +66,6 @@ namespace ipa {
|
|||
* \return 0 if successful, an error code otherwise
|
||||
*/
|
||||
|
||||
/**
|
||||
* \fn Algorithm::prepare()
|
||||
* \brief Fill the \a params buffer with ISP processing parameters for a frame
|
||||
* \param[in] context The shared IPA context
|
||||
* \param[in] frame The frame context sequence number
|
||||
* \param[in] frameContext The FrameContext for this frame
|
||||
* \param[out] params The ISP specific parameters
|
||||
*
|
||||
* This function is called for every frame when the camera is running before it
|
||||
* is processed by the ISP to prepare the ISP processing parameters for that
|
||||
* frame.
|
||||
*
|
||||
* Algorithms shall fill in the parameter structure fields appropriately to
|
||||
* configure the ISP processing blocks that they are responsible for. This
|
||||
* includes setting fields and flags that enable those processing blocks.
|
||||
*/
|
||||
|
||||
/**
|
||||
* \fn Algorithm::queueRequest()
|
||||
* \brief Provide control values to the algorithm
|
||||
|
|
@ -100,6 +83,23 @@ namespace ipa {
|
|||
* use during frame processing.
|
||||
*/
|
||||
|
||||
/**
|
||||
* \fn Algorithm::prepare()
|
||||
* \brief Fill the \a params buffer with ISP processing parameters for a frame
|
||||
* \param[in] context The shared IPA context
|
||||
* \param[in] frame The frame context sequence number
|
||||
* \param[in] frameContext The FrameContext for this frame
|
||||
* \param[out] params The ISP specific parameters
|
||||
*
|
||||
* This function is called for every frame when the camera is running before it
|
||||
* is processed by the ISP to prepare the ISP processing parameters for that
|
||||
* frame.
|
||||
*
|
||||
* Algorithms shall fill in the parameter structure fields appropriately to
|
||||
* configure the ISP processing blocks that they are responsible for. This
|
||||
* includes setting fields and flags that enable those processing blocks.
|
||||
*/
|
||||
|
||||
/**
|
||||
* \fn Algorithm::process()
|
||||
* \brief Process ISP statistics, and run algorithm operations
|
||||
|
|
|
|||
|
|
@ -38,13 +38,6 @@ public:
|
|||
return 0;
|
||||
}
|
||||
|
||||
virtual void prepare([[maybe_unused]] typename Module::Context &context,
|
||||
[[maybe_unused]] const uint32_t frame,
|
||||
[[maybe_unused]] typename Module::FrameContext &frameContext,
|
||||
[[maybe_unused]] typename Module::Params *params)
|
||||
{
|
||||
}
|
||||
|
||||
virtual void queueRequest([[maybe_unused]] typename Module::Context &context,
|
||||
[[maybe_unused]] const uint32_t frame,
|
||||
[[maybe_unused]] typename Module::FrameContext &frameContext,
|
||||
|
|
@ -52,6 +45,13 @@ public:
|
|||
{
|
||||
}
|
||||
|
||||
virtual void prepare([[maybe_unused]] typename Module::Context &context,
|
||||
[[maybe_unused]] const uint32_t frame,
|
||||
[[maybe_unused]] typename Module::FrameContext &frameContext,
|
||||
[[maybe_unused]] typename Module::Params *params)
|
||||
{
|
||||
}
|
||||
|
||||
virtual void process([[maybe_unused]] typename Module::Context &context,
|
||||
[[maybe_unused]] const uint32_t frame,
|
||||
[[maybe_unused]] typename Module::FrameContext &frameContext,
|
||||
|
|
|
|||
|
|
@ -107,6 +107,46 @@ int Agc::configure(IPAContext &context, const IPACameraSensorInfo &configInfo)
|
|||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* \copydoc libcamera::ipa::Algorithm::prepare
|
||||
*/
|
||||
void Agc::prepare(IPAContext &context, const uint32_t frame,
|
||||
IPAFrameContext &frameContext, rkisp1_params_cfg *params)
|
||||
{
|
||||
frameContext.agc.exposure = context.activeState.agc.exposure;
|
||||
frameContext.agc.gain = context.activeState.agc.gain;
|
||||
|
||||
if (frame > 0)
|
||||
return;
|
||||
|
||||
/* Configure the measurement window. */
|
||||
params->meas.aec_config.meas_window = context.configuration.agc.measureWindow;
|
||||
/* Use a continuous method for measure. */
|
||||
params->meas.aec_config.autostop = RKISP1_CIF_ISP_EXP_CTRL_AUTOSTOP_0;
|
||||
/* Estimate Y as (R + G + B) x (85/256). */
|
||||
params->meas.aec_config.mode = RKISP1_CIF_ISP_EXP_MEASURING_MODE_1;
|
||||
|
||||
params->module_cfg_update |= RKISP1_CIF_ISP_MODULE_AEC;
|
||||
params->module_ens |= RKISP1_CIF_ISP_MODULE_AEC;
|
||||
params->module_en_update |= RKISP1_CIF_ISP_MODULE_AEC;
|
||||
|
||||
/* Configure histogram. */
|
||||
params->meas.hst_config.meas_window = context.configuration.agc.measureWindow;
|
||||
/* Produce the luminance histogram. */
|
||||
params->meas.hst_config.mode = RKISP1_CIF_ISP_HISTOGRAM_MODE_Y_HISTOGRAM;
|
||||
/* Set an average weighted histogram. */
|
||||
for (unsigned int histBin = 0; histBin < numHistBins_; histBin++)
|
||||
params->meas.hst_config.hist_weight[histBin] = 1;
|
||||
/* Step size can't be less than 3. */
|
||||
params->meas.hst_config.histogram_predivider = 4;
|
||||
|
||||
/* Update the configuration for histogram. */
|
||||
params->module_cfg_update |= RKISP1_CIF_ISP_MODULE_HST;
|
||||
/* Enable the histogram measure unit. */
|
||||
params->module_ens |= RKISP1_CIF_ISP_MODULE_HST;
|
||||
params->module_en_update |= RKISP1_CIF_ISP_MODULE_HST;
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Apply a filter on the exposure value to limit the speed of changes
|
||||
* \param[in] exposureValue The target exposure from the AGC algorithm
|
||||
|
|
@ -348,46 +388,6 @@ void Agc::process(IPAContext &context, [[maybe_unused]] const uint32_t frame,
|
|||
metadata.set(controls::FrameDuration, frameDuration.get<std::micro>());
|
||||
}
|
||||
|
||||
/**
|
||||
* \copydoc libcamera::ipa::Algorithm::prepare
|
||||
*/
|
||||
void Agc::prepare(IPAContext &context, const uint32_t frame,
|
||||
IPAFrameContext &frameContext, rkisp1_params_cfg *params)
|
||||
{
|
||||
frameContext.agc.exposure = context.activeState.agc.exposure;
|
||||
frameContext.agc.gain = context.activeState.agc.gain;
|
||||
|
||||
if (frame > 0)
|
||||
return;
|
||||
|
||||
/* Configure the measurement window. */
|
||||
params->meas.aec_config.meas_window = context.configuration.agc.measureWindow;
|
||||
/* Use a continuous method for measure. */
|
||||
params->meas.aec_config.autostop = RKISP1_CIF_ISP_EXP_CTRL_AUTOSTOP_0;
|
||||
/* Estimate Y as (R + G + B) x (85/256). */
|
||||
params->meas.aec_config.mode = RKISP1_CIF_ISP_EXP_MEASURING_MODE_1;
|
||||
|
||||
params->module_cfg_update |= RKISP1_CIF_ISP_MODULE_AEC;
|
||||
params->module_ens |= RKISP1_CIF_ISP_MODULE_AEC;
|
||||
params->module_en_update |= RKISP1_CIF_ISP_MODULE_AEC;
|
||||
|
||||
/* Configure histogram. */
|
||||
params->meas.hst_config.meas_window = context.configuration.agc.measureWindow;
|
||||
/* Produce the luminance histogram. */
|
||||
params->meas.hst_config.mode = RKISP1_CIF_ISP_HISTOGRAM_MODE_Y_HISTOGRAM;
|
||||
/* Set an average weighted histogram. */
|
||||
for (unsigned int histBin = 0; histBin < numHistBins_; histBin++)
|
||||
params->meas.hst_config.hist_weight[histBin] = 1;
|
||||
/* Step size can't be less than 3. */
|
||||
params->meas.hst_config.histogram_predivider = 4;
|
||||
|
||||
/* Update the configuration for histogram. */
|
||||
params->module_cfg_update |= RKISP1_CIF_ISP_MODULE_HST;
|
||||
/* Enable the histogram measure unit. */
|
||||
params->module_ens |= RKISP1_CIF_ISP_MODULE_HST;
|
||||
params->module_en_update |= RKISP1_CIF_ISP_MODULE_HST;
|
||||
}
|
||||
|
||||
REGISTER_IPA_ALGORITHM(Agc, "Agc")
|
||||
|
||||
} /* namespace ipa::rkisp1::algorithms */
|
||||
|
|
|
|||
|
|
@ -67,20 +67,41 @@ int Awb::configure(IPAContext &context,
|
|||
return 0;
|
||||
}
|
||||
|
||||
uint32_t Awb::estimateCCT(double red, double green, double blue)
|
||||
/**
|
||||
* \copydoc libcamera::ipa::Algorithm::queueRequest
|
||||
*/
|
||||
void Awb::queueRequest(IPAContext &context,
|
||||
[[maybe_unused]] const uint32_t frame,
|
||||
IPAFrameContext &frameContext,
|
||||
const ControlList &controls)
|
||||
{
|
||||
/* Convert the RGB values to CIE tristimulus values (XYZ) */
|
||||
double X = (-0.14282) * (red) + (1.54924) * (green) + (-0.95641) * (blue);
|
||||
double Y = (-0.32466) * (red) + (1.57837) * (green) + (-0.73191) * (blue);
|
||||
double Z = (-0.68202) * (red) + (0.77073) * (green) + (0.56332) * (blue);
|
||||
auto &awb = context.activeState.awb;
|
||||
|
||||
/* Calculate the normalized chromaticity values */
|
||||
double x = X / (X + Y + Z);
|
||||
double y = Y / (X + Y + Z);
|
||||
const auto &awbEnable = controls.get(controls::AwbEnable);
|
||||
if (awbEnable && *awbEnable != awb.autoEnabled) {
|
||||
awb.autoEnabled = *awbEnable;
|
||||
|
||||
/* Calculate CCT */
|
||||
double n = (x - 0.3320) / (0.1858 - y);
|
||||
return 449 * n * n * n + 3525 * n * n + 6823.3 * n + 5520.33;
|
||||
LOG(RkISP1Awb, Debug)
|
||||
<< (*awbEnable ? "Enabling" : "Disabling") << " AWB";
|
||||
}
|
||||
|
||||
const auto &colourGains = controls.get(controls::ColourGains);
|
||||
if (colourGains && !awb.autoEnabled) {
|
||||
awb.gains.manual.red = (*colourGains)[0];
|
||||
awb.gains.manual.blue = (*colourGains)[1];
|
||||
|
||||
LOG(RkISP1Awb, Debug)
|
||||
<< "Set colour gains to red: " << awb.gains.manual.red
|
||||
<< ", blue: " << awb.gains.manual.blue;
|
||||
}
|
||||
|
||||
frameContext.awb.autoEnabled = awb.autoEnabled;
|
||||
|
||||
if (!awb.autoEnabled) {
|
||||
frameContext.awb.gains.red = awb.gains.manual.red;
|
||||
frameContext.awb.gains.green = 1.0;
|
||||
frameContext.awb.gains.blue = awb.gains.manual.blue;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
@ -165,41 +186,20 @@ void Awb::prepare(IPAContext &context, const uint32_t frame,
|
|||
params->module_ens |= RKISP1_CIF_ISP_MODULE_AWB;
|
||||
}
|
||||
|
||||
/**
|
||||
* \copydoc libcamera::ipa::Algorithm::queueRequest
|
||||
*/
|
||||
void Awb::queueRequest(IPAContext &context,
|
||||
[[maybe_unused]] const uint32_t frame,
|
||||
IPAFrameContext &frameContext,
|
||||
const ControlList &controls)
|
||||
uint32_t Awb::estimateCCT(double red, double green, double blue)
|
||||
{
|
||||
auto &awb = context.activeState.awb;
|
||||
/* Convert the RGB values to CIE tristimulus values (XYZ) */
|
||||
double X = (-0.14282) * (red) + (1.54924) * (green) + (-0.95641) * (blue);
|
||||
double Y = (-0.32466) * (red) + (1.57837) * (green) + (-0.73191) * (blue);
|
||||
double Z = (-0.68202) * (red) + (0.77073) * (green) + (0.56332) * (blue);
|
||||
|
||||
const auto &awbEnable = controls.get(controls::AwbEnable);
|
||||
if (awbEnable && *awbEnable != awb.autoEnabled) {
|
||||
awb.autoEnabled = *awbEnable;
|
||||
/* Calculate the normalized chromaticity values */
|
||||
double x = X / (X + Y + Z);
|
||||
double y = Y / (X + Y + Z);
|
||||
|
||||
LOG(RkISP1Awb, Debug)
|
||||
<< (*awbEnable ? "Enabling" : "Disabling") << " AWB";
|
||||
}
|
||||
|
||||
const auto &colourGains = controls.get(controls::ColourGains);
|
||||
if (colourGains && !awb.autoEnabled) {
|
||||
awb.gains.manual.red = (*colourGains)[0];
|
||||
awb.gains.manual.blue = (*colourGains)[1];
|
||||
|
||||
LOG(RkISP1Awb, Debug)
|
||||
<< "Set colour gains to red: " << awb.gains.manual.red
|
||||
<< ", blue: " << awb.gains.manual.blue;
|
||||
}
|
||||
|
||||
frameContext.awb.autoEnabled = awb.autoEnabled;
|
||||
|
||||
if (!awb.autoEnabled) {
|
||||
frameContext.awb.gains.red = awb.gains.manual.red;
|
||||
frameContext.awb.gains.green = 1.0;
|
||||
frameContext.awb.gains.blue = awb.gains.manual.blue;
|
||||
}
|
||||
/* Calculate CCT */
|
||||
double n = (x - 0.3320) / (0.1858 - y);
|
||||
return 449 * n * n * n + 3525 * n * n + 6823.3 * n + 5520.33;
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
|||
|
|
@ -20,12 +20,12 @@ public:
|
|||
~Awb() = default;
|
||||
|
||||
int configure(IPAContext &context, const IPACameraSensorInfo &configInfo) override;
|
||||
void prepare(IPAContext &context, const uint32_t frame,
|
||||
IPAFrameContext &frameContext,
|
||||
rkisp1_params_cfg *params) override;
|
||||
void queueRequest(IPAContext &context, const uint32_t frame,
|
||||
IPAFrameContext &frameContext,
|
||||
const ControlList &controls) override;
|
||||
void prepare(IPAContext &context, const uint32_t frame,
|
||||
IPAFrameContext &frameContext,
|
||||
rkisp1_params_cfg *params) override;
|
||||
void process(IPAContext &context, const uint32_t frame,
|
||||
IPAFrameContext &frameContext,
|
||||
const rkisp1_stat_buffer *stats,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue