151 lines
4.6 KiB
C++
151 lines
4.6 KiB
C++
/***
|
|
|
|
Olive - Non-Linear Video Editor
|
|
Copyright (C) 2022 Olive Team
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
***/
|
|
|
|
#include "histogram.h"
|
|
|
|
#include <QPainter>
|
|
#include <QtMath>
|
|
#include <QVector2D>
|
|
|
|
#include "common/qtutils.h"
|
|
#include "node/node.h"
|
|
|
|
namespace olive
|
|
{
|
|
|
|
#define super ScopeBase
|
|
|
|
HistogramScope::HistogramScope(QWidget *parent)
|
|
: super(parent)
|
|
{
|
|
}
|
|
|
|
void HistogramScope::OnInit()
|
|
{
|
|
super::OnInit();
|
|
|
|
ShaderCode secondary_code(
|
|
FileFunctions::ReadFileAsString(
|
|
":/shaders/rgbhistogram_secondary.frag"),
|
|
FileFunctions::ReadFileAsString(":/shaders/rgbhistogram.vert"));
|
|
pipeline_secondary_ = renderer()->CreateNativeShader(secondary_code);
|
|
}
|
|
|
|
void HistogramScope::OnDestroy()
|
|
{
|
|
pipeline_secondary_.clear();
|
|
texture_row_sums_ = nullptr;
|
|
|
|
super::OnDestroy();
|
|
}
|
|
|
|
ShaderCode HistogramScope::GenerateShaderCode()
|
|
{
|
|
return ShaderCode(
|
|
FileFunctions::ReadFileAsString(":/shaders/rgbhistogram.frag"),
|
|
FileFunctions::ReadFileAsString(":/shaders/default.vert"));
|
|
}
|
|
|
|
void HistogramScope::DrawScope(TexturePtr managed_tex, QVariant pipeline)
|
|
{
|
|
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;
|
|
|
|
ShaderJob shader_job;
|
|
|
|
shader_job.Insert(QStringLiteral("viewport"),
|
|
NodeValue(NodeValue::kVec2,
|
|
QVector2D(width(), height())));
|
|
shader_job.Insert(QStringLiteral("histogram_scale"),
|
|
NodeValue(NodeValue::kFloat, histogram_scale));
|
|
shader_job.Insert(QStringLiteral("histogram_power"),
|
|
NodeValue(NodeValue::kFloat, histogram_power));
|
|
|
|
if (!texture_row_sums_ || texture_row_sums_->width() != this->width() ||
|
|
texture_row_sums_->height() != this->height()) {
|
|
texture_row_sums_ = renderer()->CreateTexture(
|
|
VideoParams(width(), height(), managed_tex->format(),
|
|
managed_tex->channel_count()));
|
|
}
|
|
|
|
// Draw managed texture to a sums texture
|
|
shader_job.Insert(QStringLiteral("ove_maintex"),
|
|
NodeValue(NodeValue::kTexture,
|
|
QVariant::fromValue(managed_tex)));
|
|
renderer()->BlitToTexture(pipeline, shader_job, texture_row_sums_.get());
|
|
|
|
// Draw sums into a histogram
|
|
shader_job.Insert(QStringLiteral("ove_maintex"),
|
|
NodeValue(NodeValue::kTexture,
|
|
QVariant::fromValue(texture_row_sums_)));
|
|
renderer()->Blit(pipeline_secondary_, shader_job,
|
|
texture_row_sums_->params());
|
|
|
|
// Draw line overlays
|
|
QPainter p(paint_device());
|
|
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 = QtUtils::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);
|
|
}
|
|
|
|
}
|