Compositor: Full frame Brightness node
Adds full frame implementation to this node operation. No functional changes. 2.4x faster than tiled fallback. Reviewed By: jbakker Differential Revision: https://developer.blender.org/D11677
This commit is contained in:
parent
b35efa932e
commit
9aa88a6604
|
@ -28,6 +28,7 @@ BrightnessOperation::BrightnessOperation()
|
|||
this->addOutputSocket(DataType::Color);
|
||||
this->m_inputProgram = nullptr;
|
||||
this->m_use_premultiply = false;
|
||||
flags.can_be_constant = true;
|
||||
}
|
||||
|
||||
void BrightnessOperation::setUsePremultiply(bool use_premultiply)
|
||||
|
@ -85,6 +86,50 @@ void BrightnessOperation::executePixelSampled(float output[4],
|
|||
}
|
||||
}
|
||||
|
||||
void BrightnessOperation::update_memory_buffer_partial(MemoryBuffer *output,
|
||||
const rcti &area,
|
||||
Span<MemoryBuffer *> inputs)
|
||||
{
|
||||
float tmp_color[4];
|
||||
for (BuffersIterator<float> it = output->iterate_with(inputs, area); !it.is_end(); ++it) {
|
||||
const float *in_color = it.in(0);
|
||||
const float brightness = *it.in(1) / 100.0f;
|
||||
const float contrast = *it.in(2);
|
||||
float delta = contrast / 200.0f;
|
||||
/*
|
||||
* The algorithm is by Werner D. Streidt
|
||||
* (http://visca.com/ffactory/archives/5-99/msg00021.html)
|
||||
* Extracted of OpenCV demhist.c
|
||||
*/
|
||||
float a, b;
|
||||
if (contrast > 0) {
|
||||
a = 1.0f - delta * 2.0f;
|
||||
a = 1.0f / max_ff(a, FLT_EPSILON);
|
||||
b = a * (brightness - delta);
|
||||
}
|
||||
else {
|
||||
delta *= -1;
|
||||
a = max_ff(1.0f - delta * 2.0f, 0.0f);
|
||||
b = a * brightness + delta;
|
||||
}
|
||||
const float *color;
|
||||
if (this->m_use_premultiply) {
|
||||
premul_to_straight_v4_v4(tmp_color, in_color);
|
||||
color = tmp_color;
|
||||
}
|
||||
else {
|
||||
color = in_color;
|
||||
}
|
||||
it.out[0] = a * color[0] + b;
|
||||
it.out[1] = a * color[1] + b;
|
||||
it.out[2] = a * color[2] + b;
|
||||
it.out[3] = color[3];
|
||||
if (this->m_use_premultiply) {
|
||||
straight_to_premul_v4(it.out);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void BrightnessOperation::deinitExecution()
|
||||
{
|
||||
this->m_inputProgram = nullptr;
|
||||
|
|
|
@ -18,11 +18,11 @@
|
|||
|
||||
#pragma once
|
||||
|
||||
#include "COM_NodeOperation.h"
|
||||
#include "COM_MultiThreadedOperation.h"
|
||||
|
||||
namespace blender::compositor {
|
||||
|
||||
class BrightnessOperation : public NodeOperation {
|
||||
class BrightnessOperation : public MultiThreadedOperation {
|
||||
private:
|
||||
/**
|
||||
* Cached reference to the inputProgram
|
||||
|
@ -52,6 +52,10 @@ class BrightnessOperation : public NodeOperation {
|
|||
void deinitExecution() override;
|
||||
|
||||
void setUsePremultiply(bool use_premultiply);
|
||||
|
||||
void update_memory_buffer_partial(MemoryBuffer *output,
|
||||
const rcti &area,
|
||||
Span<MemoryBuffer *> inputs) override;
|
||||
};
|
||||
|
||||
} // namespace blender::compositor
|
||||
|
|
Loading…
Reference in New Issue