Barry
7 years ago
parent
c7c1f30252
commit
c00954b6a9
@ -1,228 +0,0 @@
|
||||
#include <QOpenGLShaderProgram>
|
||||
|
||||
#include "yuvglwidget.h"
|
||||
|
||||
static QString vertShader = R"(
|
||||
#version 150 core
|
||||
uniform mat4 u_pm;
|
||||
uniform vec4 draw_pos;
|
||||
|
||||
const vec2 verts[4] = vec2[] (
|
||||
vec2(-0.5, 0.5),
|
||||
vec2(-0.5, -0.5),
|
||||
vec2( 0.5, 0.5),
|
||||
vec2( 0.5, -0.5)
|
||||
);
|
||||
|
||||
const vec2 texcoords[4] = vec2[] (
|
||||
vec2(0.0, 1.0),
|
||||
vec2(0.0, 0.0),
|
||||
vec2(1.0, 1.0),
|
||||
vec2(1.0, 0.0)
|
||||
);
|
||||
|
||||
out vec2 v_coord;
|
||||
|
||||
void main() {
|
||||
vec2 vert = verts[gl_VertexID];
|
||||
vec4 p = vec4((0.5 * draw_pos.z) + draw_pos.x + (vert.x * draw_pos.z),
|
||||
(0.5 * draw_pos.w) + draw_pos.y + (vert.y * draw_pos.w),
|
||||
0, 1);
|
||||
gl_Position = u_pm * p;
|
||||
v_coord = texcoords[gl_VertexID];
|
||||
}
|
||||
)";
|
||||
|
||||
static QString fragShader = R"(
|
||||
#version 150 core
|
||||
uniform sampler2D y_tex;
|
||||
uniform sampler2D u_tex;
|
||||
uniform sampler2D v_tex;
|
||||
in vec2 v_coord;
|
||||
//layout( location = 0 ) out vec4 fragcolor;
|
||||
out vec4 fragcolor;
|
||||
|
||||
const vec3 R_cf = vec3(1.164383, 0.000000, 1.596027);
|
||||
const vec3 G_cf = vec3(1.164383, -0.391762, -0.812968);
|
||||
const vec3 B_cf = vec3(1.164383, 2.017232, 0.000000);
|
||||
const vec3 offset = vec3(-0.0625, -0.5, -0.5);
|
||||
|
||||
void main() {
|
||||
float y = texture(y_tex, v_coord).r;
|
||||
float u = texture(u_tex, v_coord).r;
|
||||
float v = texture(v_tex, v_coord).r;
|
||||
vec3 yuv = vec3(y,u,v);
|
||||
yuv += offset;
|
||||
fragcolor = vec4(0.0, 0.0, 0.0, 1.0);
|
||||
fragcolor.r = dot(yuv, R_cf);
|
||||
fragcolor.g = dot(yuv, G_cf);
|
||||
fragcolor.b = dot(yuv, B_cf);
|
||||
}
|
||||
)";
|
||||
|
||||
YUVGLWidget::YUVGLWidget(QWidget *parent) :
|
||||
QOpenGLWidget(parent)
|
||||
{
|
||||
/*
|
||||
// 设置opengl兼容性格式为CoreProfile
|
||||
QSurfaceFormat defaultFormat = QSurfaceFormat::defaultFormat();
|
||||
defaultFormat.setProfile(QSurfaceFormat::CoreProfile);
|
||||
defaultFormat.setVersion(3, 3); // Adapt to your system
|
||||
QSurfaceFormat::setDefaultFormat(defaultFormat);
|
||||
setFormat(defaultFormat);
|
||||
*/
|
||||
}
|
||||
|
||||
YUVGLWidget::~YUVGLWidget()
|
||||
{
|
||||
makeCurrent();
|
||||
deInitTextures();
|
||||
doneCurrent();
|
||||
}
|
||||
|
||||
QSize YUVGLWidget::minimumSizeHint() const
|
||||
{
|
||||
return QSize(50, 50);
|
||||
}
|
||||
|
||||
QSize YUVGLWidget::sizeHint() const
|
||||
{
|
||||
return size();
|
||||
}
|
||||
|
||||
void YUVGLWidget::setFrameSize(const QSize& frameSize)
|
||||
{
|
||||
if (m_frameSize != frameSize) {
|
||||
m_frameSize = frameSize;
|
||||
m_needInit = true;
|
||||
}
|
||||
}
|
||||
|
||||
void YUVGLWidget::initializeGL()
|
||||
{
|
||||
initializeOpenGLFunctions();
|
||||
glDisable(GL_DEPTH_TEST);
|
||||
m_vao.create();
|
||||
}
|
||||
|
||||
void YUVGLWidget::paintGL()
|
||||
{
|
||||
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
|
||||
|
||||
// init
|
||||
if (m_needInit) {
|
||||
initShader();
|
||||
initTextures();
|
||||
m_needInit = false;
|
||||
}
|
||||
m_vao.bind();
|
||||
if (m_program) {
|
||||
QMatrix4x4 matrix;
|
||||
matrix.ortho(0, width(), height(), 0, 0.0, 100.0f);
|
||||
m_program->setUniformValue("u_pm", matrix);
|
||||
glUniform4f(m_drawPos, 0, 0, width(), height());
|
||||
}
|
||||
|
||||
glActiveTexture(GL_TEXTURE0);
|
||||
glBindTexture(GL_TEXTURE_2D, m_texture[Texture_Y]);
|
||||
|
||||
glActiveTexture(GL_TEXTURE1);
|
||||
glBindTexture(GL_TEXTURE_2D, m_texture[Texture_U]);
|
||||
|
||||
glActiveTexture(GL_TEXTURE2);
|
||||
glBindTexture(GL_TEXTURE_2D, m_texture[Texture_V]);
|
||||
|
||||
glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
|
||||
}
|
||||
|
||||
void YUVGLWidget::resizeGL(int width, int height)
|
||||
{
|
||||
glViewport(0, 0, width, height);
|
||||
}
|
||||
|
||||
void YUVGLWidget::initShader()
|
||||
{
|
||||
if (m_program) {
|
||||
m_program->release();
|
||||
delete m_program;
|
||||
m_program = Q_NULLPTR;
|
||||
}
|
||||
m_program = new QOpenGLShaderProgram(this);
|
||||
// Setup shaders
|
||||
m_program->addShaderFromSourceCode(QOpenGLShader::Vertex, vertShader);
|
||||
m_program->addShaderFromSourceCode(QOpenGLShader::Fragment, fragShader);
|
||||
m_program->link();
|
||||
m_program->bind();
|
||||
m_program->setUniformValue("y_tex", Texture_Y);
|
||||
m_program->setUniformValue("u_tex", Texture_U);
|
||||
m_program->setUniformValue("v_tex", Texture_V);
|
||||
m_drawPos = m_program->uniformLocation("draw_pos");
|
||||
}
|
||||
|
||||
void YUVGLWidget::updateTextures(quint8 *dataY, quint8 *dataU, quint8 *dataV, quint32 linesizeY, quint32 linesizeU, quint32 linesizeV)
|
||||
{
|
||||
bindPixelTexture(m_texture[Texture_Y], Texture_Y, dataY, linesizeY);
|
||||
bindPixelTexture(m_texture[Texture_U], Texture_U, dataU, linesizeU);
|
||||
bindPixelTexture(m_texture[Texture_V], Texture_V, dataV, linesizeV);
|
||||
update();
|
||||
}
|
||||
|
||||
void YUVGLWidget::bindPixelTexture(GLuint texture, YUVTextureType textureType, quint8* pixels, quint32 stride)
|
||||
{
|
||||
if (!pixels)
|
||||
return;
|
||||
|
||||
QSize size(0, 0);
|
||||
calcTextureSize(textureType, size);
|
||||
|
||||
makeCurrent();
|
||||
glBindTexture(GL_TEXTURE_2D, texture);
|
||||
glPixelStorei(GL_UNPACK_ROW_LENGTH, stride);
|
||||
glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, size.width(), size.height(), GL_LUMINANCE, GL_UNSIGNED_BYTE, pixels);
|
||||
doneCurrent();
|
||||
}
|
||||
|
||||
void YUVGLWidget::initTexture(qint32 textureType)
|
||||
{
|
||||
if (Texture_NULL >= textureType || Texture_Size <= textureType) {
|
||||
return;
|
||||
}
|
||||
glGenTextures(1, &m_texture[textureType]);
|
||||
glBindTexture(GL_TEXTURE_2D, m_texture[textureType]);
|
||||
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
|
||||
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
|
||||
glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
|
||||
glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
|
||||
|
||||
QSize size(0, 0);
|
||||
calcTextureSize(textureType, size);
|
||||
glTexImage2D(GL_TEXTURE_2D, 0, GL_LUMINANCE, size.width(), size.height(), 0, GL_LUMINANCE, GL_UNSIGNED_BYTE, NULL);
|
||||
}
|
||||
|
||||
void YUVGLWidget::initTextures()
|
||||
{
|
||||
//TODO: use FBO?
|
||||
|
||||
deInitTextures();
|
||||
for (int i = Texture_Y; i < Texture_Size; i++) {
|
||||
initTexture(i);
|
||||
}
|
||||
}
|
||||
|
||||
void YUVGLWidget::deInitTextures()
|
||||
{
|
||||
glDeleteTextures(3, m_texture);
|
||||
memset(m_texture, 0, Texture_Size);
|
||||
}
|
||||
|
||||
void YUVGLWidget::calcTextureSize(qint32 textureType, QSize &size)
|
||||
{
|
||||
if (Texture_NULL >= textureType || Texture_Size <= textureType) {
|
||||
return;
|
||||
}
|
||||
if (Texture_Y == textureType) {
|
||||
size = m_frameSize;
|
||||
} else {
|
||||
size = m_frameSize / 2;
|
||||
}
|
||||
}
|
||||
@ -1,56 +0,0 @@
|
||||
#ifndef YUVGLWIDGET_H
|
||||
#define YUVGLWIDGET_H
|
||||
|
||||
#include <QOpenGLFunctions>
|
||||
#include <QOpenGLVertexArrayObject>
|
||||
#include <QOpenGLWidget>
|
||||
//#include <QSurfaceFormat>
|
||||
|
||||
class QOpenGLShaderProgram;
|
||||
class YUVGLWidget : public QOpenGLWidget, protected QOpenGLFunctions
|
||||
{
|
||||
Q_OBJECT
|
||||
public:
|
||||
explicit YUVGLWidget(QWidget* parent = 0);
|
||||
~YUVGLWidget();
|
||||
|
||||
QSize minimumSizeHint() const override;
|
||||
QSize sizeHint() const override;
|
||||
|
||||
void setFrameSize(const QSize& frameSize);
|
||||
void updateTextures(quint8* dataY, quint8* dataU, quint8* dataV, quint32 linesizeY, quint32 linesizeU, quint32 linesizeV);
|
||||
|
||||
protected:
|
||||
void initializeGL() override;
|
||||
void paintGL() override;
|
||||
void resizeGL(int width, int height) override;
|
||||
|
||||
private:
|
||||
enum YUVTextureType {
|
||||
Texture_NULL = -1,
|
||||
Texture_Y,
|
||||
Texture_U,
|
||||
Texture_V,
|
||||
Texture_Size
|
||||
};
|
||||
|
||||
void initShader();
|
||||
void initTexture(qint32 textureType);
|
||||
void initTextures();
|
||||
void deInitTextures();
|
||||
void calcTextureSize(qint32 textureType, QSize& size);
|
||||
|
||||
void bindPixelTexture(GLuint texture, YUVTextureType textureType, quint8* pixels, quint32 stride);
|
||||
|
||||
private:
|
||||
QSize m_frameSize = {0, 0};
|
||||
bool m_needInit = false;
|
||||
|
||||
QOpenGLShaderProgram* m_program = Q_NULLPTR;
|
||||
QOpenGLVertexArrayObject m_vao;
|
||||
// yuv texture
|
||||
GLuint m_texture[Texture_Size] = {0};
|
||||
GLint m_drawPos = -1;
|
||||
};
|
||||
|
||||
#endif // YUVGLWIDGET_H
|
||||
Loading…
Reference in new issue