• 如何渲染最原始的yuv视频数据?


    一.整体思路

      我们在用纹理增加细节那篇文章中提到过,要将图片渲染在屏幕上,首先要拿到图片的像素数组数据,然后将像素数组数据通过纹理单元传递到片段着色器中,最后通过纹理采样函数将纹理中对应坐标的颜色值采样出来,然后给最终的片段赋予颜色值。现在换成了yuv视频,我们应该如何处理呢?因为最终的片段颜色值是RGBA格式的,而我们的视频是YUV格式的,所以我们需要做一个转化:即将YUV转化为RGBA。

      我们在渲染图像到屏幕的时候,需要用到glTexImage2D()函数指定二维纹理图像,这个函数各个参数的含义如下:

    • target:指定目标纹理,这个值必须是GL_TEXTURE_2D
    • level:执行细节级别,0是最基本的图像级别,n表示第N级贴图细化级别
    • internalformat:指定纹理中的颜色组件,可选的值有GL_ALPHA,GL_RGB,GL_RGBA,GL_LUMINANCE, GL_LUMINANCE_ALPHA 等几种
    • width:指定纹理图像的宽度,必须是2的n次方
    • height:指定纹理图像的高度,必须是2的n次方
    • border:指定边框的宽度,必须为0
    • format:像素数据的颜色格式, 不需要和internalformat取值必须相同,可选的值参考internalformat
    • type:指定像素数据的数据类型
    • pixels:指定内存中指向图像数据的指针

      我们可以看到这个函数并没有直接支持yuv格式的图像数据,但是,别担心!它又给我们提供了GL_LUMINANCE这种格式,它表示只取一个颜色通道,假如传入的值为r,则在片段着色器中的纹理单元中读出的值为(r,r,r,1)。这样以来,我们就可以将yuv图像拆分为3个通道来读取。但是,拆分为3个通道来读取,最后如何重新合成一个RGBA颜色值呢?这个时候,之前学过的纹理单元就可以派上用场了,我们可以定义3个纹理单元,分别读取yuv图像的3个通道的数据,最后在片段着色器中进行合成,然后转化为RGBA值即可。

    二.读取解析yuv视频文件

      想要读取yuv视频数据,我们首先得清楚它的内部结构。为了方便讲解,这里我们以yuv420p格式的视频文件为例,它是一个由宽640,高360的yuv图像构成的视频,并且帧和帧之间无缝衔接。我们知道yuv420p格式的图像帧是先连续存储所有的y分量,然后再连续存储所有的u分量,最后再连续存储所有的v分量。并且,亮度分量y和色度分量uv的比例为4:1:1,也就是4个亮度分量共享一组色度分量。

      知道了这些之后,我们就可以来读取yuv视频文件了。首先我们将准备的视频文件input.yuv放入assets文件夹下面,然后写一个函数循环地去读取这个视频文件,代码如下:

    复制代码
    fun readYuvData(w:Int,h:Int){
            val input=context.resources.assets.open("input.yuv")
            val y=ByteArray(w*h)
            val u=ByteArray(w*h/4)
            val v=ByteArray(w*h/4)
            while(true){
                val ySize=input.read(y)
                val uSize=input.read(u)
                val vSize=input.read(v)
                if(ySize>0&&uSize>0&&vSize>0){
                    //根据指定的字节数组创建一个新的ByteBuffer对象,对返回的ByteBuffer对象所做的更改会反映在原始字节数组上,因为它们共享相同的存储区域
                    bufferY=ByteBuffer.wrap(y)
                    bufferU=ByteBuffer.wrap(u)
                    bufferV=ByteBuffer.wrap(v)
                    //请求渲染一个新帧,调用requestRender()后,GLSurfaceView会在下一个合适的时机调用OpenGL渲染器的onDrawFrame()方法,从而实现新的场景绘制和渲染
                    glSurfaceView.requestRender()
                    Thread.sleep(1000/30)
                }
                else{
                    break
                }
            }
        }
    复制代码

    三.编写顶点着色器和片段着色器

      首先,我们来编写顶点着色器,代码如下:

    复制代码
    #version 300 es
    layout(location=0) in vec4 a_Position;
    layout(location=1) in vec2 a_Texture_Coordinates;
    layout(location=3) uniform mat4 u_Matrix;
    out vec2 v_Texture_Coordinates;
    void main() {
        gl_Position=u_Matrix*a_Position;
        v_Texture_Coordinates=a_Texture_Coordinates;
    }
    复制代码

      接下来,再来编写片段着色器,代码如下:

    复制代码
    #version 300 es
    precision mediump float;
    in vec2 v_Texture_Coordinates;
    out vec4 fragColor;
    layout(location=0) uniform sampler2D textureY;
    layout(location=1) uniform sampler2D textureU;
    layout(location=2) uniform sampler2D textureV;
    void main() {
        float y,u,v;
        vec3 rgb;
        y=texture(textureY,v_Texture_Coordinates).r;
        u=texture(textureU,v_Texture_Coordinates).g-0.5;
        v=texture(textureV,v_Texture_Coordinates).b-0.5;
        rgb.r = y + 1.540*v;
        rgb.g = y - 0.183*u - 0.459*v;
        rgb.b = y + 1.816*u;
        fragColor=vec4(rgb,1.0);
    }
    复制代码

      在这里,我们定义了三个纹理采样对象,分别用于对yuv图像3个通道的数据进行采样。然后,我们需要知道rgb.r,rgb.g指的是什么。其实,在GLSL中,向量的组件可以通过{x,y,z,w},{r,g,b,a}或{s,t,r,q}来获取,之所以采用这三个不同的命名方法,是因为向量通常会用来表示数学向量,颜色和纹理坐标。所以rgb.r,texture(textureY,v_Texture_Coordinates).r都是指向量中的第一个元素的值。由于我们之前设置的格式是GL_LUMINANCE,假设传入的y分量对应坐标位置的值为r,则在片段着色器中的纹理单元中读出的值为(r,r,r,1),那么我们取r就是取第一个元素的值,其实这里前3个的值都是一样的,取哪个都可以。

      但是,我们注意到,u和v后面都减去了0.5,这是为什么呢?我们先来看下yuv转rgb的公式:

           

       我们首先需要知道的是yuv中的u,v指的是红色R和蓝色B与亮度Y的偏差,u和v的默认值都是128,我们把128代入公式,正好R=Y,R=B。从上面的公式看,代入的u和v都是减去默认值128的,也就是说转化公式中所使用的是u,v和默认值128的偏移值。所以,我们要使用这个公式,也要求出这个偏移值。但是,texture函数计算后得到的是归一化的值,取值范围是[0,1],由于位深是8bit,取值范围是[0,255],减去128相当于减去总范围的一半,所以我们也需要减去总范围的一半,即0.5。

     四.绑定顶点数据和纹理数据

      首先,我们写一个函数用于绑定顶点数据:

    复制代码
    fun bindVertexData(){
         //创建vao glGenVertexArrays(
    1,vao,0)
         //创建vbo glGenBuffers(
    1,vbo,0)
         //一定要先绑定vao,再绑定vbo glBindVertexArray(vao[
    0]) glBindBuffer(GL_ARRAY_BUFFER,vbo[0])
         //将顶点数组数据存入显存 glBufferData(GL_ARRAY_BUFFER,floatBuffer.capacity()
    *4,floatBuffer, GL_STATIC_DRAW)
         //最后一个参数改为偏移值offset glVertexAttribPointer(
    0, position_component_count, GL_FLOAT,false,stride,0) glEnableVertexAttribArray(0) glVertexAttribPointer(1, texture_coordinate_component_count, GL_FLOAT,false,stride, position_component_count*4) glEnableVertexAttribArray(1)
         //解除绑定 glBindBuffer(GL_ARRAY_BUFFER,
    0) glBindVertexArray(0) }
    复制代码

      在这里,我们使用了顶点数组对象vao和顶点缓冲对象vbo,这是opengl es3.0中引入的新特性。在opengl es2.0编程中,用于绘制的顶点数组数据首先保存在cpu内存,在调用glDrawArrays函数进行绘制时,需要将顶点数组数据从cpu内存拷贝到gpu显存中。但是,很多时候我们没必要每次绘制时都进行内存拷贝,如果可以直接在显存中存储这些数据,就可以避免每次拷贝所带来的巨大开销。vbo的出现就是为了解决这个问题的,vbo的作用是提前在显存中开辟好一块内存,用于存储顶点数组数据。

      那vao是用来干嘛的呢?我们现在思考一个问题,假如我们有两份顶点数组数据,一份用来绘制正方体,一份用来绘制长方体,并且我们将它们都存入vbo开辟的显存中,那么gpu怎么知道取哪一部分数据绘制正方体,哪一部分数据绘制长方体呢?vao就是用于解决这个问题的,vao的作用就相当于一个指针,指向我们所开辟的内存的首地址,如下图所示。这样以来,我们可以开辟两处内存分别用于存储正方体数据和长方体数据,然后,我们再使用两个vao对象,分别指向两个内存块的首地址,这样以来,gpu就知道去哪里取数据了。当然,如果只有一份数据,不使用vao也行。

       然后,再写一个函数用来绑定纹理数据,代码如下:

    复制代码
    fun bindTextureData(){
            //y平面
            glActiveTexture(GL_TEXTURE0)
            glBindTexture(GL_TEXTURE_2D,textures[0])
            glTexImage2D(GL_TEXTURE_2D,0, GL_LUMINANCE,W,H,0, GL_LUMINANCE, GL_UNSIGNED_BYTE,bufferY)
            //u平面
            glActiveTexture(GL_TEXTURE1)
            glBindTexture(GL_TEXTURE_2D,textures[1])
            glTexImage2D(GL_TEXTURE_2D,0, GL_LUMINANCE,W/2,H/2,0, GL_LUMINANCE, GL_UNSIGNED_BYTE,bufferU)
            //v平面
            glActiveTexture(GL_TEXTURE2)
            glBindTexture(GL_TEXTURE_2D,textures[2])
            glTexImage2D(GL_TEXTURE_2D,0, GL_LUMINANCE,W/2,H/2,0, GL_LUMINANCE, GL_UNSIGNED_BYTE,bufferV)
        }
    复制代码

      完整的MyRenderer.kt的代码如下:

    复制代码
    class MyRenderer(val context: Context,val glSurfaceView:GLSurfaceView):GLSurfaceView.Renderer {
        private val projectionMatrix:FloatArray=FloatArray(16)//存储投影矩阵
        private val textures=IntArray(3)
        private var floatBuffer:FloatBuffer
        private val vao=IntArray(1)
        private val vbo=IntArray(1)
        private var bufferY:ByteBuffer?=null
        private var bufferU:ByteBuffer?=null
        private var bufferV:ByteBuffer?=null
        init{
            //存储顶点坐标和纹理坐标,并且让矩形的宽高比和视频像素的宽高比一样,都是16:9
            val vertexData= floatArrayOf(
                1f,  9/16f, 1.0f, 0.0f, // top right
                1f, -9/16f, 1.0f, 1.0f, // bottom right
                -1f, 9/16f, 0.0f, 0.0f,  // top left
                1f, -9/16f, 1.0f, 1.0f, // bottom right
                -1f, -9/16f, 0.0f, 1.0f, // bottom left
                -1f,  9/16f, 0.0f, 0.0f  // top left
            )
            floatBuffer= ByteBuffer
                .allocateDirect(vertexData.size*4)//一个浮点数占四个字节
                .order(ByteOrder.nativeOrder())
                .asFloatBuffer()
                .put(vertexData)
            floatBuffer.position(0)
        }
        companion object{
            val position_component_count=2
            val texture_coordinate_component_count=2
            val stride=(position_component_count+ texture_coordinate_component_count)*4
            val W=640
            val H=360
        }
        override fun onSurfaceCreated(gl: GL10?, config: EGLConfig?) {
            glClearColor(1.0f,1.0f,1.0f,1.0f)
            val vertexShaderCode=TextResourceReader.readTextFileFromResource(context,R.raw.vertex_shader)
            val fragmentShaderCode=TextResourceReader.readTextFileFromResource(context,R.raw.fragment_shader)
            ShaderHelper.buildProgram(vertexShaderCode,fragmentShaderCode)
            glUniform1i(0,0)
            glUniform1i(1,1)
            glUniform1i(2,2)
            glGenTextures(3,textures,0)
            for(i in 0..2){
                glBindTexture(GL_TEXTURE_2D,textures[i])
                glTexParameteri(GL_TEXTURE_2D,GL_TEXTURE_MIN_FILTER, GL_NEAREST)//处理图片缩小的情况
                glTexParameteri(GL_TEXTURE_2D,GL_TEXTURE_MAG_FILTER,GL_LINEAR)//处理图片放大的情况
    
                //解绑纹理对象
                glBindTexture(GL_TEXTURE_2D,0)
            }
            bindVertexData()
        }
    
        override fun onSurfaceChanged(gl: GL10?, width: Int, height: Int) {
            glViewport(0,0,width,height)
            //根据屏幕方向生成投影矩阵
            val aspectRatio=if(width>height) width.toFloat()/height.toFloat() else height.toFloat()/width.toFloat()
            if(width>height){
                Matrix.orthoM(projectionMatrix,0,-aspectRatio,aspectRatio,-1f,1f,-1f,1f)
            }
            else{
                Matrix.orthoM(projectionMatrix,0,-1f,1f,-aspectRatio,aspectRatio,-1f,1f)
            }
            //传递正交投影矩阵
            glUniformMatrix4fv(3,1,false,projectionMatrix,0)
            thread{
                readYuvData(W,H)
            }
        }
    
        override fun onDrawFrame(gl: GL10?) {
            glClear(GL_COLOR_BUFFER_BIT)
            bindTextureData()
            glBindVertexArray(vao[0])
            glDrawArrays(GL_TRIANGLES,0,6)
            bufferY?.clear()
            bufferU?.clear()
            bufferV?.clear()
        }
        fun bindVertexData(){
            glGenVertexArrays(1,vao,0)
            glGenBuffers(1,vbo,0)
            glBindVertexArray(vao[0])
            glBindBuffer(GL_ARRAY_BUFFER,vbo[0])
            glBufferData(GL_ARRAY_BUFFER,floatBuffer.capacity()*4,floatBuffer, GL_STATIC_DRAW)
            glVertexAttribPointer(0, position_component_count, GL_FLOAT,false,stride,0)
            glEnableVertexAttribArray(0)
            glVertexAttribPointer(1, texture_coordinate_component_count, GL_FLOAT,false,stride, position_component_count*4)
            glEnableVertexAttribArray(1)
            glBindBuffer(GL_ARRAY_BUFFER,0)
            glBindVertexArray(0)
        }
        fun bindTextureData(){
            //y平面
            glActiveTexture(GL_TEXTURE0)
            glBindTexture(GL_TEXTURE_2D,textures[0])
            glTexImage2D(GL_TEXTURE_2D,0, GL_LUMINANCE,W,H,0, GL_LUMINANCE, GL_UNSIGNED_BYTE,bufferY)
            //u平面
            glActiveTexture(GL_TEXTURE1)
            glBindTexture(GL_TEXTURE_2D,textures[1])
            glTexImage2D(GL_TEXTURE_2D,0, GL_LUMINANCE,W/2,H/2,0, GL_LUMINANCE, GL_UNSIGNED_BYTE,bufferU)
            //v平面
            glActiveTexture(GL_TEXTURE2)
            glBindTexture(GL_TEXTURE_2D,textures[2])
            glTexImage2D(GL_TEXTURE_2D,0, GL_LUMINANCE,W/2,H/2,0, GL_LUMINANCE, GL_UNSIGNED_BYTE,bufferV)
        }
        fun readYuvData(w:Int,h:Int){
            val input=context.resources.assets.open("input.yuv")
            val y=ByteArray(w*h)
            val u=ByteArray(w*h/4)
            val v=ByteArray(w*h/4)
            while(true){
                val ySize=input.read(y)
                val uSize=input.read(u)
                val vSize=input.read(v)
                if(ySize>0&&uSize>0&&vSize>0){
                    //根据指定的字节数组创建一个新的ByteBuffer对象,对返回的ByteBuffer对象所做的更改会反映在原始字节数组上,因为它们共享相同的存储区域
                    bufferY=ByteBuffer.wrap(y)
                    bufferU=ByteBuffer.wrap(u)
                    bufferV=ByteBuffer.wrap(v)
                    //请求渲染一个新帧,调用requestRender()后,GLSurfaceView会在下一个合适的时机调用OpenGL渲染器的onDrawFrame()方法,从而实现新的场景绘制和渲染
                    glSurfaceView.requestRender()
                    Thread.sleep(1000/30)
                }
                else{
                    break
                }
            }
        }
    }
    复制代码

      MainActivity.kt的代码如下:

    复制代码
    class MainActivity : AppCompatActivity() {
        private lateinit var glSurfaceView:GLSurfaceView
        override fun onCreate(savedInstanceState: Bundle?) {
            super.onCreate(savedInstanceState)
            glSurfaceView= GLSurfaceView(this)
            glSurfaceView.setEGLContextClientVersion(3)
            glSurfaceView.setRenderer(MyRenderer(this,glSurfaceView))
            setContentView(glSurfaceView)
        }
        override fun onPause() {
            super.onPause()
            glSurfaceView.onPause()
        }
    
        override fun onResume() {
            super.onResume()
            glSurfaceView.onResume()
        }
    }
    复制代码

      其他代码我之前的文章中有写。

     

  • 相关阅读:
    mysql:create index 和 alter add index
    编程中的delete用法:深入探索与实战应用
    HTTP和HTTPS的区别是什么
    MySQL覆盖索引的使用
    Matlab 信号处理【问答随笔·2】
    微信、支付宝、携程等多款app任意文件读取漏洞
    NPDP如何续证?操作指南来了!
    数据处理任务——知识点总结
    网络编程(一)
    UE5发布时出现SDK NOT Setup错误解决方法
  • 原文地址:https://www.cnblogs.com/luqman/p/18034409