histogram: implemented color management

This commit is contained in:
itsmattkc
2020-04-21 01:59:06 +10:00
parent bdb13e88f5
commit 0ced48d4df
11 changed files with 68 additions and 11 deletions
+28 -7
View File
@@ -30,7 +30,8 @@ OLIVE_NAMESPACE_ENTER
HistogramScope::HistogramScope(QWidget* parent) :
QOpenGLWidget(parent),
buffer_(nullptr)
buffer_(nullptr),
processor_(nullptr)
{
connect(&worker_, &HistogramScopeWorker::Finished, this, &HistogramScope::FinishedProcessing, Qt::QueuedConnection);
worker_.start(QThread::IdlePriority);
@@ -45,12 +46,18 @@ HistogramScope::~HistogramScope()
void HistogramScope::SetBuffer(Frame* frame)
{
// We'll be multithreading this, so we make a copy to prevent collisions
buffer_ = frame;
StartUpdate();
}
void HistogramScope::SetColorProcessor(ColorProcessorPtr processor)
{
processor_ = processor;
StartUpdate();
}
void HistogramScope::FinishedProcessing(QVector<double> red, QVector<double> green, QVector<double> blue)
{
red_val_ = red;
@@ -89,14 +96,20 @@ void HistogramScope::resizeEvent(QResizeEvent *e)
{
QOpenGLWidget::resizeEvent(e);
if (buffer_){
StartUpdate();
}
StartUpdate();
}
void HistogramScope::StartUpdate()
{
worker_.QueueNext(*buffer_, width());
if (buffer_) {
worker_.QueueNext(*buffer_, processor_, width());
} else {
// Update with nothing
red_val_.clear();
green_val_.clear();
blue_val_.clear();
update();
}
}
HistogramScopeWorker::HistogramScopeWorker() :
@@ -120,10 +133,13 @@ void HistogramScopeWorker::run()
// Copy values
Frame f = next_;
int w = next_width_;
ColorProcessorPtr processor = next_processor_;
next_.destroy();
next_lock_.unlock();
// Color manage frame
QVector<int> data(w * kRGBChannels, 0);
int max_w = w-1;
@@ -132,6 +148,10 @@ void HistogramScopeWorker::run()
for (int y=0;y<f.height();y++) {
Color c = f.get_pixel(x, y);
if (processor) {
c = processor->ConvertColor(c);
}
data[qFloor(clamp(c.red(), 0.0f, 1.0f) * max_w)]++;
data[qFloor(clamp(c.green(), 0.0f, 1.0f) * max_w) + w]++;
data[qFloor(clamp(c.blue(), 0.0f, 1.0f) * max_w) + w * 2]++;
@@ -165,12 +185,13 @@ void HistogramScopeWorker::run()
}
}
void HistogramScopeWorker::QueueNext(const Frame &f, int width)
void HistogramScopeWorker::QueueNext(const Frame &f, ColorProcessorPtr processor, int width)
{
next_lock_.lock();
next_ = f;
next_width_ = width;
next_processor_ = processor;
next_wait_.wakeOne();