255 lines
8.5 KiB
C++
255 lines
8.5 KiB
C++
/***
|
|
|
|
Olive - Non-Linear Video Editor
|
|
Copyright (C) 2019 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 "shadergenerators.h"
|
|
|
|
#include <QOpenGLExtraFunctions>
|
|
|
|
namespace olive {
|
|
|
|
ShaderPtr ShaderGenerator::DefaultPipeline(const QString& function_name, const QString& shader_code)
|
|
{
|
|
ShaderPtr program = std::make_shared<QOpenGLShaderProgram>();
|
|
|
|
// Generate vertex shader
|
|
QString vert_shader = "#version 110\n"
|
|
"\n"
|
|
"#ifdef GL_ES\n"
|
|
"precision mediump int;\n"
|
|
"precision mediump float;\n"
|
|
"#endif\n"
|
|
"\n"
|
|
"uniform mat4 mvp_matrix;\n"
|
|
"\n"
|
|
"attribute vec4 a_position;\n"
|
|
"attribute vec2 a_texcoord;\n"
|
|
"\n"
|
|
"varying vec2 v_texcoord;\n"
|
|
"\n"
|
|
"void main() {\n"
|
|
" gl_Position = mvp_matrix * a_position;\n"
|
|
" v_texcoord = a_texcoord;\n"
|
|
"}\n";
|
|
|
|
// Generate fragment shader
|
|
QString frag_shader = "#version 110\n"
|
|
"\n"
|
|
"#ifdef GL_ES\n"
|
|
"precision mediump int;\n"
|
|
"precision mediump float;\n"
|
|
"#endif\n"
|
|
"\n"
|
|
"uniform sampler2D texture;\n"
|
|
"uniform float opacity;\n"
|
|
"uniform bool color_only;\n"
|
|
"uniform vec4 color_only_color;\n"
|
|
"varying vec2 v_texcoord;\n"
|
|
"\n";
|
|
|
|
// Finish the function with the main function
|
|
|
|
// Check if additional code was passed to this function, add it here
|
|
if (shader_code.isEmpty()) {
|
|
|
|
// If not, just add a pure main() function
|
|
|
|
frag_shader.append("\n"
|
|
"void main() {\n"
|
|
" if (color_only) {\n"
|
|
" gl_FragColor = color_only_color;"
|
|
" } else {\n"
|
|
" vec4 color = texture2D(texture, v_texcoord)*opacity;\n"
|
|
" gl_FragColor = color;\n"
|
|
" }\n"
|
|
"}\n");
|
|
|
|
} else {
|
|
|
|
// If additional code was passed, add it and reference it in main().
|
|
//
|
|
// The function in the additional code is expected to be `vec4 function_name(vec4 color)`. The texture coordinate
|
|
// can be acquired through `v_texcoord`.
|
|
|
|
frag_shader.append(shader_code);
|
|
|
|
frag_shader.append(QString("\n"
|
|
"void main() {\n"
|
|
" vec4 color = %1(texture2D(texture, v_texcoord))*opacity;\n"
|
|
" gl_FragColor = color;\n"
|
|
"}\n").arg(function_name));
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Add shaders to program
|
|
program->addShaderFromSourceCode(QOpenGLShader::Vertex, vert_shader);
|
|
program->addShaderFromSourceCode(QOpenGLShader::Fragment, frag_shader);
|
|
program->link();
|
|
|
|
// Set opacity default to 100%
|
|
program->bind();
|
|
program->setUniformValue("opacity", 1.0f);
|
|
program->release();
|
|
|
|
return program;
|
|
}
|
|
|
|
QString ShaderGenerator::AlphaDisassociateFunction(const QString &function_name)
|
|
{
|
|
return QString("vec4 %1(vec4 col) {\n"
|
|
" if (col.a > 0.0) {\n"
|
|
" return vec4(col.rgb / col.a, col.a);"
|
|
" }\n"
|
|
" return col;\n"
|
|
"}\n").arg(function_name);
|
|
}
|
|
|
|
QString ShaderGenerator::AlphaReassociateFunction(const QString &function_name)
|
|
{
|
|
return QString("vec4 %1(vec4 col) {\n"
|
|
" if (col.a > 0.0) {\n"
|
|
" return vec4(col.rgb * col.a, col.a);"
|
|
" }\n"
|
|
" return col;\n"
|
|
"}\n").arg(function_name);
|
|
}
|
|
|
|
QString ShaderGenerator::AlphaAssociateFunction(const QString &function_name)
|
|
{
|
|
return QString("vec4 %1(vec4 col) {\n"
|
|
" return vec4(col.rgb * col.a, col.a);\n"
|
|
"}\n").arg(function_name);
|
|
}
|
|
|
|
// copied from source code to OCIODisplay
|
|
const int OCIO_LUT3D_EDGE_SIZE = 32;
|
|
|
|
// copied from source code to OCIODisplay, expanded from 3*LUT3D_EDGE_SIZE*LUT3D_EDGE_SIZE*LUT3D_EDGE_SIZE
|
|
const int OCIO_NUM_3D_ENTRIES = 98304;
|
|
|
|
ShaderPtr ShaderGenerator::OCIOPipeline(QOpenGLContext* ctx,
|
|
GLuint& lut_texture,
|
|
OCIO::ConstProcessorRcPtr processor,
|
|
bool alpha_is_associated)
|
|
{
|
|
|
|
QOpenGLExtraFunctions* xf = ctx->extraFunctions();
|
|
|
|
// Create LUT texture
|
|
xf->glGenTextures(1, &lut_texture);
|
|
|
|
// Bind LUT
|
|
xf->glBindTexture(GL_TEXTURE_3D, lut_texture);
|
|
|
|
// Set texture parameters
|
|
xf->glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
|
|
xf->glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
|
|
xf->glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
|
|
xf->glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
|
|
xf->glTexParameteri(GL_TEXTURE_3D, GL_TEXTURE_WRAP_R, GL_CLAMP_TO_EDGE);
|
|
|
|
// Allocate storage for texture
|
|
xf->glTexImage3D(GL_TEXTURE_3D, 0, GL_RGB16F_ARB,
|
|
OCIO_LUT3D_EDGE_SIZE, OCIO_LUT3D_EDGE_SIZE, OCIO_LUT3D_EDGE_SIZE,
|
|
0, GL_RGB,GL_FLOAT, nullptr);
|
|
|
|
//
|
|
// SET UP GLSL SHADER
|
|
//
|
|
|
|
OCIO::GpuShaderDesc shaderDesc;
|
|
const char* ocio_func_name = "OCIODisplay";
|
|
shaderDesc.setLanguage(OCIO::GPU_LANGUAGE_GLSL_1_0);
|
|
shaderDesc.setFunctionName(ocio_func_name);
|
|
shaderDesc.setLut3DEdgeLen(OCIO_LUT3D_EDGE_SIZE);
|
|
|
|
//
|
|
// COMPUTE 3D LUT
|
|
//
|
|
|
|
GLfloat* ocio_lut_data = new GLfloat[OCIO_NUM_3D_ENTRIES];
|
|
processor->getGpuLut3D(ocio_lut_data, shaderDesc);
|
|
|
|
// Upload LUT data to texture
|
|
xf->glTexSubImage3D(GL_TEXTURE_3D, 0,
|
|
0, 0, 0,
|
|
OCIO_LUT3D_EDGE_SIZE, OCIO_LUT3D_EDGE_SIZE, OCIO_LUT3D_EDGE_SIZE,
|
|
GL_RGB, GL_FLOAT, ocio_lut_data);
|
|
|
|
delete [] ocio_lut_data;
|
|
|
|
// Create OCIO shader code
|
|
QString shader_text(processor->getGpuShaderText(shaderDesc));
|
|
|
|
QString shader_call;
|
|
|
|
// Enforce alpha association
|
|
if (alpha_is_associated) {
|
|
|
|
// If alpha is already associated, we'll need to disassociate and reassociate
|
|
shader_text.append("\n");
|
|
|
|
QString disassociate_func_name = "disassoc";
|
|
shader_text.append(AlphaDisassociateFunction(disassociate_func_name));
|
|
|
|
QString reassociate_func_name = "reassoc";
|
|
shader_text.append(AlphaReassociateFunction(reassociate_func_name));
|
|
|
|
// Make OCIO call pass through disassociate and reassociate function
|
|
shader_call = QString("%3(%1(%2(col), tex2));").arg(ocio_func_name,
|
|
disassociate_func_name,
|
|
reassociate_func_name);
|
|
|
|
} else {
|
|
|
|
// If alpha is not already associated, we can just associate after OCIO
|
|
|
|
// Add associate function
|
|
QString associate_func_name = "assoc";
|
|
shader_text.append(AlphaAssociateFunction(associate_func_name));
|
|
|
|
// Make OCIO call pass through associate function
|
|
shader_call = QString("%2(%1(col, tex2));").arg(ocio_func_name, associate_func_name);
|
|
|
|
}
|
|
|
|
// Add process() function, which GetPipeline() will call if specified
|
|
QString process_function_name = "process";
|
|
shader_text.append(QString("\n"
|
|
"uniform sampler3D tex2;\n"
|
|
"\n"
|
|
"vec4 %2(vec4 col) {\n"
|
|
" return %1\n"
|
|
"}\n").arg(shader_call, process_function_name));
|
|
|
|
|
|
// Get pipeline-based shader to inject OCIO shader into
|
|
ShaderPtr shader = ShaderGenerator::DefaultPipeline(process_function_name, shader_text);
|
|
|
|
// Release LUT
|
|
xf->glBindTexture(GL_TEXTURE_3D, 0);
|
|
|
|
return shader;
|
|
}
|
|
|
|
}
|