Merge branch 'sobotka-histogram-updates'
This commit is contained in:
@@ -0,0 +1,37 @@
|
||||
#version 150
|
||||
|
||||
uniform sampler2D ove_maintex;
|
||||
uniform vec2 ove_resolution;
|
||||
uniform vec2 ove_viewport;
|
||||
|
||||
uniform float histogram_scale;
|
||||
|
||||
in vec2 ove_texcoord;
|
||||
|
||||
out vec4 fragColor;
|
||||
|
||||
void main(void) {
|
||||
float histogram_width = ceil(histogram_scale * ove_viewport.y);
|
||||
float quantisation = 1.0 / (histogram_width - 1.0);
|
||||
vec3 cur_col = vec3(0.0);
|
||||
vec3 sum = vec3(0.0);
|
||||
float ratio = 0.0;
|
||||
|
||||
for (int i = 0; i < histogram_width; i++) {
|
||||
ratio = float(i) / float(histogram_width - 1);
|
||||
cur_col = texture(
|
||||
ove_maintex,
|
||||
vec2(ove_texcoord.y, ratio)
|
||||
).rgb;
|
||||
|
||||
sum += step(vec3(ove_texcoord.x - quantisation), cur_col) *
|
||||
step(cur_col, vec3(ove_texcoord.x + quantisation)) +
|
||||
(
|
||||
// Account for values beyond the upper x limit.
|
||||
step(1.0 - quantisation, ove_texcoord.x) *
|
||||
step(vec3(1.0 - quantisation), cur_col)
|
||||
);
|
||||
}
|
||||
|
||||
fragColor = vec4(sum, 1.0);
|
||||
}
|
||||
@@ -0,0 +1,29 @@
|
||||
#version 150
|
||||
|
||||
uniform float histogram_scale;
|
||||
uniform vec2 ove_resolution;
|
||||
|
||||
in vec4 a_position;
|
||||
in vec2 a_texcoord;
|
||||
|
||||
out vec2 ove_texcoord;
|
||||
|
||||
mat4 scale_mat4(vec3 scale) {
|
||||
return mat4(
|
||||
scale.x, 0.0, 0.0, 0.0,
|
||||
0.0, scale.y, 0.0, 0.0,
|
||||
0.0, 0.0, scale.z, 0.0,
|
||||
0.0, 0.0, 0.0, 1.0
|
||||
);
|
||||
}
|
||||
|
||||
void main() {
|
||||
// Create identity matrix
|
||||
mat4 transform = mat4(1.0);
|
||||
|
||||
// Scale the scope
|
||||
transform *= scale_mat4(vec3(histogram_scale, histogram_scale, 1.0));
|
||||
|
||||
gl_Position = transform * a_position;
|
||||
ove_texcoord = a_texcoord;
|
||||
}
|
||||
@@ -0,0 +1,35 @@
|
||||
#version 150
|
||||
|
||||
uniform sampler2D ove_maintex;
|
||||
uniform vec2 ove_resolution;
|
||||
uniform vec2 ove_viewport;
|
||||
|
||||
uniform float histogram_scale;
|
||||
uniform float histogram_power;
|
||||
|
||||
in vec2 ove_texcoord;
|
||||
|
||||
out vec4 fragColor;
|
||||
|
||||
void main(void) {
|
||||
vec3 col = vec3(0.0);
|
||||
float histogram_height = ceil(ove_viewport.y * histogram_scale);
|
||||
vec3 histogram_ratio = vec3(0.0);
|
||||
vec3 sum = vec3(0.0);
|
||||
float ratio = 0.0;
|
||||
vec3 total_pixels = vec3(ceil(ove_viewport.x * ove_resolution.y *
|
||||
histogram_scale));
|
||||
|
||||
for (int i = 0; i < histogram_height; i++) {
|
||||
ratio = float(i) / float(histogram_height - 1.0);
|
||||
sum += texture(
|
||||
ove_maintex,
|
||||
vec2(ove_texcoord.x, ratio)
|
||||
).rgb;
|
||||
}
|
||||
|
||||
histogram_ratio = pow(sum / total_pixels, vec3(histogram_power));
|
||||
col = step(vec3(ove_texcoord.y), histogram_ratio);
|
||||
|
||||
fragColor = vec4(col, 1.0);
|
||||
}
|
||||
@@ -23,8 +23,9 @@
|
||||
#include <QPainter>
|
||||
#include <QtMath>
|
||||
|
||||
#include "common/clamp.h"
|
||||
#include "common/functiontimer.h"
|
||||
#include "common/qtutils.h"
|
||||
#include "node/node.h"
|
||||
#include "render/backend/opengl/openglrenderfunctions.h"
|
||||
|
||||
OLIVE_NAMESPACE_ENTER
|
||||
|
||||
@@ -33,4 +34,170 @@ HistogramScope::HistogramScope(QWidget* parent) :
|
||||
{
|
||||
}
|
||||
|
||||
HistogramScope::~HistogramScope()
|
||||
{
|
||||
CleanUp();
|
||||
|
||||
if (context()) {
|
||||
disconnect(context(), &QOpenGLContext::aboutToBeDestroyed, this,
|
||||
&HistogramScope::CleanUp);
|
||||
}
|
||||
}
|
||||
|
||||
void HistogramScope::initializeGL()
|
||||
{
|
||||
ScopeBase::initializeGL();
|
||||
|
||||
pipeline_secondary_ = CreateSecondaryShader();
|
||||
|
||||
connect(context(), &QOpenGLContext::aboutToBeDestroyed, this,
|
||||
&HistogramScope::CleanUp, Qt::DirectConnection);
|
||||
}
|
||||
|
||||
void HistogramScope::AssertAdditionalTextures()
|
||||
{
|
||||
if (!texture_row_sums_.IsCreated()
|
||||
|| texture_row_sums_.width() != width()
|
||||
|| texture_row_sums_.height() != height()) {
|
||||
texture_row_sums_.Destroy();
|
||||
texture_row_sums_.Create(context(), VideoParams(width(),
|
||||
height(), managed_tex().format()));
|
||||
}
|
||||
}
|
||||
|
||||
void HistogramScope::CleanUp()
|
||||
{
|
||||
makeCurrent();
|
||||
|
||||
pipeline_secondary_ = nullptr;
|
||||
texture_row_sums_.Destroy();
|
||||
|
||||
doneCurrent();
|
||||
}
|
||||
|
||||
OpenGLShaderPtr HistogramScope::CreateShader()
|
||||
{
|
||||
OpenGLShaderPtr pipeline = OpenGLShader::Create();
|
||||
|
||||
pipeline->create();
|
||||
pipeline->addShaderFromSourceCode(QOpenGLShader::Vertex,
|
||||
OpenGLShader::CodeDefaultVertex());
|
||||
pipeline->addShaderFromSourceCode(QOpenGLShader::Fragment,
|
||||
Node::ReadFileAsString(":/shaders/rgbhistogram.frag"));
|
||||
pipeline->link();
|
||||
|
||||
return pipeline;
|
||||
}
|
||||
|
||||
OpenGLShaderPtr HistogramScope::CreateSecondaryShader()
|
||||
{
|
||||
OpenGLShaderPtr pipeline_secondary_ = OpenGLShader::Create();
|
||||
|
||||
pipeline_secondary_->create();
|
||||
pipeline_secondary_->addShaderFromSourceCode(QOpenGLShader::Vertex,
|
||||
Node::ReadFileAsString(":/shaders/rgbhistogram.vert"));
|
||||
pipeline_secondary_->addShaderFromSourceCode(QOpenGLShader::Fragment,
|
||||
Node::ReadFileAsString(":/shaders/rgbhistogram_secondary.frag"));
|
||||
pipeline_secondary_->link();
|
||||
|
||||
return pipeline_secondary_;
|
||||
}
|
||||
|
||||
void HistogramScope::DrawScope()
|
||||
{
|
||||
float histogram_scale = 0.80f;
|
||||
// This value is eyeballed for usefulness. Until we have a geometry
|
||||
// shader approach, it is impossible to normalize against a peak
|
||||
// sum of image values.
|
||||
float histogram_base = 2.5f;
|
||||
float histogram_power = 1.0f / histogram_base;
|
||||
|
||||
pipeline()->bind();
|
||||
pipeline()->setUniformValue("ove_resolution", managed_tex().width(),
|
||||
managed_tex().height());
|
||||
pipeline()->setUniformValue("ove_viewport", width(), height());
|
||||
pipeline()->setUniformValue("histogram_scale", histogram_scale);
|
||||
pipeline()->release();
|
||||
|
||||
AssertAdditionalTextures();
|
||||
|
||||
framebuffer().Attach(&texture_row_sums_, true);
|
||||
framebuffer().Bind();
|
||||
|
||||
managed_tex().Bind();
|
||||
|
||||
OpenGLRenderFunctions::Blit(pipeline());
|
||||
|
||||
managed_tex().Release();
|
||||
|
||||
framebuffer().Release();
|
||||
framebuffer().Detach();
|
||||
|
||||
pipeline_secondary_->bind();
|
||||
pipeline_secondary_->setUniformValue("ove_resolution",
|
||||
texture_row_sums_.width(), texture_row_sums_.height());
|
||||
pipeline_secondary_->setUniformValue("ove_viewport", width(), height());
|
||||
pipeline_secondary_->setUniformValue("histogram_scale", histogram_scale);
|
||||
pipeline_secondary_->setUniformValue("histogram_power", histogram_power);
|
||||
pipeline_secondary_->release();
|
||||
|
||||
texture_row_sums_.Bind();
|
||||
|
||||
OpenGLRenderFunctions::Blit(pipeline_secondary_);
|
||||
|
||||
texture_row_sums_.Release();
|
||||
|
||||
// Draw line overlays
|
||||
QPainter p(this);
|
||||
QFont font = p.font();
|
||||
font.setPixelSize(10);
|
||||
QFontMetrics font_metrics = QFontMetrics(font);
|
||||
QString label;
|
||||
std::vector<float> histogram_increments = {
|
||||
0.00,
|
||||
0.25,
|
||||
0.50,
|
||||
1.0
|
||||
};
|
||||
|
||||
int histogram_steps = histogram_increments.size();
|
||||
QVector<QLine> histogram_lines(histogram_steps + 1);
|
||||
int font_x_offset = 0;
|
||||
int font_y_offset = font_metrics.capHeight() / 2.0f;
|
||||
|
||||
p.setCompositionMode(QPainter::CompositionMode_Plus);
|
||||
|
||||
p.setPen(QColor(0.0, 0.6 * 255.0, 0.0));
|
||||
p.setFont(font);
|
||||
|
||||
float histogram_dim_x = ceil((width() - 1.0) * histogram_scale);
|
||||
float histogram_dim_y = ceil((height() - 1.0) * histogram_scale);
|
||||
float histogram_start_dim_x =
|
||||
((width() - 1.0) - histogram_dim_x) / 2.0f;
|
||||
float histogram_start_dim_y =
|
||||
((height() - 1.0) - histogram_dim_y) / 2.0f;
|
||||
float histogram_end_dim_x = (width() - 1.0) - histogram_start_dim_x;
|
||||
|
||||
// for (int i=0; i <= histogram_steps; i++) {
|
||||
for(std::vector<float>::iterator it = histogram_increments.begin();
|
||||
it != histogram_increments.end(); it++) {
|
||||
histogram_lines[it - histogram_increments.begin()].setLine(
|
||||
histogram_start_dim_x,
|
||||
(histogram_dim_y * pow(1.0 - *it, histogram_base)) +
|
||||
histogram_start_dim_y,
|
||||
histogram_end_dim_x,
|
||||
(histogram_dim_y * pow(1.0 - *it, histogram_base)) +
|
||||
histogram_start_dim_y);
|
||||
label = QString::number(
|
||||
*it * 100, 'f', 1) + "%";
|
||||
font_x_offset = QFontMetricsWidth(font_metrics, label) + 4;
|
||||
|
||||
p.drawText(
|
||||
histogram_start_dim_x - font_x_offset,
|
||||
(histogram_dim_y * pow(1.0 - *it, histogram_base)) +
|
||||
histogram_start_dim_y + font_y_offset, label);
|
||||
}
|
||||
p.drawLines(histogram_lines);
|
||||
}
|
||||
|
||||
OLIVE_NAMESPACE_EXIT
|
||||
|
||||
@@ -31,11 +31,24 @@ class HistogramScope : public ScopeBase
|
||||
public:
|
||||
HistogramScope(QWidget* parent = nullptr);
|
||||
|
||||
virtual ~HistogramScope() override;
|
||||
|
||||
protected:
|
||||
//virtual OpenGLShaderPtr CreateShader() override;
|
||||
virtual void initializeGL() override;
|
||||
|
||||
//virtual void DrawScope() override;
|
||||
virtual OpenGLShaderPtr CreateShader() override;
|
||||
OpenGLShaderPtr CreateSecondaryShader();
|
||||
|
||||
void AssertAdditionalTextures();
|
||||
|
||||
virtual void DrawScope() override;
|
||||
|
||||
private:
|
||||
OpenGLShaderPtr pipeline_secondary_;
|
||||
OpenGLTexture texture_row_sums_;
|
||||
|
||||
private slots:
|
||||
void CleanUp();
|
||||
};
|
||||
|
||||
OLIVE_NAMESPACE_EXIT
|
||||
|
||||
@@ -68,16 +68,6 @@ void ScopeBase::DrawScope()
|
||||
managed_tex().Release();
|
||||
}
|
||||
|
||||
OpenGLShaderPtr ScopeBase::pipeline()
|
||||
{
|
||||
return pipeline_;
|
||||
}
|
||||
|
||||
OpenGLTexture &ScopeBase::managed_tex()
|
||||
{
|
||||
return managed_tex_;
|
||||
}
|
||||
|
||||
void ScopeBase::UploadTextureFromBuffer()
|
||||
{
|
||||
if (!isVisible()) {
|
||||
|
||||
@@ -51,9 +51,20 @@ protected:
|
||||
|
||||
virtual void DrawScope();
|
||||
|
||||
OpenGLShaderPtr pipeline();
|
||||
OpenGLShaderPtr pipeline()
|
||||
{
|
||||
return pipeline_;
|
||||
}
|
||||
|
||||
OpenGLTexture& managed_tex();
|
||||
OpenGLTexture& managed_tex()
|
||||
{
|
||||
return managed_tex_;
|
||||
}
|
||||
|
||||
OpenGLFramebuffer& framebuffer()
|
||||
{
|
||||
return framebuffer_;
|
||||
}
|
||||
|
||||
private:
|
||||
void UploadTextureFromBuffer();
|
||||
|
||||
Reference in New Issue
Block a user