ipa: raspberrypi: Generalise the contrast algorithm
Generalise the contrast algorithm code by removing any hard-coded assumptions about the target hardware platform. Instead, the algorithm code creates a generic Pwl that gets returned to the IPA, where it gets converted to the bcm2835 hardware specific lookup table. As a drive-by, remove an unused mutex. Signed-off-by: Naushir Patuck <naush@raspberrypi.com> Reviewed-by: David Plowman <david.plowman@raspberrypi.com> Reviewed-by: Jacopo Mondi <jacopo.mondi@ideasonboard.com> Signed-off-by: Kieran Bingham <kieran.bingham@ideasonboard.com>
This commit is contained in:
parent
a82d08973f
commit
d8685a579c
4 changed files with 20 additions and 37 deletions
|
@ -6,20 +6,15 @@
|
||||||
*/
|
*/
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
|
#include "pwl.h"
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* The "contrast" algorithm creates a gamma curve, optionally doing a little bit
|
* The "contrast" algorithm creates a gamma curve, optionally doing a little bit
|
||||||
* of contrast stretching based on the AGC histogram.
|
* of contrast stretching based on the AGC histogram.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
constexpr unsigned int ContrastNumPoints = 33;
|
|
||||||
|
|
||||||
struct ContrastPoint {
|
|
||||||
uint16_t x;
|
|
||||||
uint16_t y;
|
|
||||||
};
|
|
||||||
|
|
||||||
struct ContrastStatus {
|
struct ContrastStatus {
|
||||||
struct ContrastPoint points[ContrastNumPoints];
|
RPiController::Pwl gammaCurve;
|
||||||
double brightness;
|
double brightness;
|
||||||
double contrast;
|
double contrast;
|
||||||
};
|
};
|
||||||
|
|
|
@ -65,34 +65,19 @@ void Contrast::setContrast(double contrast)
|
||||||
contrast_ = contrast;
|
contrast_ = contrast;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void fillInStatus(ContrastStatus &status, double brightness,
|
|
||||||
double contrast, Pwl &gammaCurve)
|
|
||||||
{
|
|
||||||
status.brightness = brightness;
|
|
||||||
status.contrast = contrast;
|
|
||||||
for (unsigned int i = 0; i < ContrastNumPoints - 1; i++) {
|
|
||||||
int x = i < 16 ? i * 1024
|
|
||||||
: (i < 24 ? (i - 16) * 2048 + 16384
|
|
||||||
: (i - 24) * 4096 + 32768);
|
|
||||||
status.points[i].x = x;
|
|
||||||
status.points[i].y = std::min(65535.0, gammaCurve.eval(x));
|
|
||||||
}
|
|
||||||
status.points[ContrastNumPoints - 1].x = 65535;
|
|
||||||
status.points[ContrastNumPoints - 1].y = 65535;
|
|
||||||
}
|
|
||||||
|
|
||||||
void Contrast::initialise()
|
void Contrast::initialise()
|
||||||
{
|
{
|
||||||
/*
|
/*
|
||||||
* Fill in some default values as Prepare will run before Process gets
|
* Fill in some default values as Prepare will run before Process gets
|
||||||
* called.
|
* called.
|
||||||
*/
|
*/
|
||||||
fillInStatus(status_, brightness_, contrast_, config_.gammaCurve);
|
status_.brightness = brightness_;
|
||||||
|
status_.contrast = contrast_;
|
||||||
|
status_.gammaCurve = config_.gammaCurve;
|
||||||
}
|
}
|
||||||
|
|
||||||
void Contrast::prepare(Metadata *imageMetadata)
|
void Contrast::prepare(Metadata *imageMetadata)
|
||||||
{
|
{
|
||||||
std::unique_lock<std::mutex> lock(mutex_);
|
|
||||||
imageMetadata->set("contrast.status", status_);
|
imageMetadata->set("contrast.status", status_);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -183,12 +168,9 @@ void Contrast::process(StatisticsPtr &stats,
|
||||||
* And fill in the status for output. Use more points towards the bottom
|
* And fill in the status for output. Use more points towards the bottom
|
||||||
* of the curve.
|
* of the curve.
|
||||||
*/
|
*/
|
||||||
ContrastStatus status;
|
status_.brightness = brightness_;
|
||||||
fillInStatus(status, brightness_, contrast_, gammaCurve);
|
status_.contrast = contrast_;
|
||||||
{
|
status_.gammaCurve = std::move(gammaCurve);
|
||||||
std::unique_lock<std::mutex> lock(mutex_);
|
|
||||||
status_ = status;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Register algorithm with the system. */
|
/* Register algorithm with the system. */
|
||||||
|
|
|
@ -46,7 +46,6 @@ private:
|
||||||
double brightness_;
|
double brightness_;
|
||||||
double contrast_;
|
double contrast_;
|
||||||
ContrastStatus status_;
|
ContrastStatus status_;
|
||||||
std::mutex mutex_;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
} /* namespace RPiController */
|
} /* namespace RPiController */
|
||||||
|
|
|
@ -1593,14 +1593,21 @@ void IPARPi::applyCCM(const struct CcmStatus *ccmStatus, ControlList &ctrls)
|
||||||
|
|
||||||
void IPARPi::applyGamma(const struct ContrastStatus *contrastStatus, ControlList &ctrls)
|
void IPARPi::applyGamma(const struct ContrastStatus *contrastStatus, ControlList &ctrls)
|
||||||
{
|
{
|
||||||
|
const unsigned int numGammaPoints = controller_.getHardwareConfig().numGammaPoints;
|
||||||
struct bcm2835_isp_gamma gamma;
|
struct bcm2835_isp_gamma gamma;
|
||||||
|
|
||||||
gamma.enabled = 1;
|
for (unsigned int i = 0; i < numGammaPoints - 1; i++) {
|
||||||
for (unsigned int i = 0; i < ContrastNumPoints; i++) {
|
int x = i < 16 ? i * 1024
|
||||||
gamma.x[i] = contrastStatus->points[i].x;
|
: (i < 24 ? (i - 16) * 2048 + 16384
|
||||||
gamma.y[i] = contrastStatus->points[i].y;
|
: (i - 24) * 4096 + 32768);
|
||||||
|
gamma.x[i] = x;
|
||||||
|
gamma.y[i] = std::min<uint16_t>(65535, contrastStatus->gammaCurve.eval(x));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
gamma.x[numGammaPoints - 1] = 65535;
|
||||||
|
gamma.y[numGammaPoints - 1] = 65535;
|
||||||
|
gamma.enabled = 1;
|
||||||
|
|
||||||
ControlValue c(Span<const uint8_t>{ reinterpret_cast<uint8_t *>(&gamma),
|
ControlValue c(Span<const uint8_t>{ reinterpret_cast<uint8_t *>(&gamma),
|
||||||
sizeof(gamma) });
|
sizeof(gamma) });
|
||||||
ctrls.set(V4L2_CID_USER_BCM2835_ISP_GAMMA, c);
|
ctrls.set(V4L2_CID_USER_BCM2835_ISP_GAMMA, c);
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue