YuvConverter.smali

.class Lorg/webrtc/ali/YuvConverter;
.super Ljava/lang/Object;
.source "YuvConverter.java"


# static fields
.field private static final DEVICE_RECTANGLE:Ljava/nio/FloatBuffer;

.field private static final FRAGMENT_SHADER:Ljava/lang/String; = "#extension GL_OES_EGL_image_external : require\nprecision mediump float;\nvarying vec2 interp_tc;\n\nuniform samplerExternalOES oesTex;\nuniform vec2 xUnit;\nuniform vec4 coeffs;\n\nvoid main() {\n  gl_FragColor.r = coeffs.a + dot(coeffs.rgb,\n      texture2D(oesTex, interp_tc - 1.5 * xUnit).rgb);\n  gl_FragColor.g = coeffs.a + dot(coeffs.rgb,\n      texture2D(oesTex, interp_tc - 0.5 * xUnit).rgb);\n  gl_FragColor.b = coeffs.a + dot(coeffs.rgb,\n      texture2D(oesTex, interp_tc + 0.5 * xUnit).rgb);\n  gl_FragColor.a = coeffs.a + dot(coeffs.rgb,\n      texture2D(oesTex, interp_tc + 1.5 * xUnit).rgb);\n}\n"

.field private static final TEXTURE_RECTANGLE:Ljava/nio/FloatBuffer;

.field private static final VERTEX_SHADER:Ljava/lang/String; = "varying vec2 interp_tc;\nattribute vec4 in_pos;\nattribute vec4 in_tc;\n\nuniform mat4 texMatrix;\n\nvoid main() {\n    gl_Position = in_pos;\n    interp_tc = (texMatrix * in_tc).xy;\n}\n"


# instance fields
.field private final coeffsLoc:I

.field private released:Z

.field private final shader:Lorg/webrtc/ali/GlShader;

.field private final texMatrixLoc:I

.field private final textureFrameBuffer:Lorg/webrtc/ali/GlTextureFrameBuffer;

.field private final threadChecker:Lorg/webrtc/ali/ThreadUtils$ThreadChecker;

.field private final xUnitLoc:I


# direct methods
.method static constructor <clinit>()V
    .registers 2

    .line 25
    const/16 v0, 0x8

    new-array v1, v0, [F

    fill-array-data v1, :array_1a

    invoke-static {v1}, Lorg/webrtc/ali/GlUtil;->createFloatBuffer([F)Ljava/nio/FloatBuffer;

    move-result-object v1

    sput-object v1, Lorg/webrtc/ali/YuvConverter;->DEVICE_RECTANGLE:Ljava/nio/FloatBuffer;

    .line 33
    new-array v0, v0, [F

    fill-array-data v0, :array_2e

    invoke-static {v0}, Lorg/webrtc/ali/GlUtil;->createFloatBuffer([F)Ljava/nio/FloatBuffer;

    move-result-object v0

    sput-object v0, Lorg/webrtc/ali/YuvConverter;->TEXTURE_RECTANGLE:Ljava/nio/FloatBuffer;

    return-void

    nop

    :array_1a
    .array-data 4
        -0x40800000    # -1.0f
        -0x40800000    # -1.0f
        0x3f800000    # 1.0f
        -0x40800000    # -1.0f
        -0x40800000    # -1.0f
        0x3f800000    # 1.0f
        0x3f800000    # 1.0f
        0x3f800000    # 1.0f
    .end array-data

    :array_2e
    .array-data 4
        0x0
        0x0
        0x3f800000    # 1.0f
        0x0
        0x0
        0x3f800000    # 1.0f
        0x3f800000    # 1.0f
        0x3f800000    # 1.0f
    .end array-data
.end method

.method public constructor <init>()V
    .registers 5

    .line 94
    invoke-direct {p0}, Ljava/lang/Object;-><init>()V

    .line 88
    new-instance v0, Lorg/webrtc/ali/ThreadUtils$ThreadChecker;

    invoke-direct {v0}, Lorg/webrtc/ali/ThreadUtils$ThreadChecker;-><init>()V

    iput-object v0, p0, Lorg/webrtc/ali/YuvConverter;->threadChecker:Lorg/webrtc/ali/ThreadUtils$ThreadChecker;

    .line 89
    const/4 v1, 0x0

    iput-boolean v1, p0, Lorg/webrtc/ali/YuvConverter;->released:Z

    .line 95
    invoke-virtual {v0}, Lorg/webrtc/ali/ThreadUtils$ThreadChecker;->checkIsOnValidThread()V

    .line 96
    new-instance v0, Lorg/webrtc/ali/GlTextureFrameBuffer;

    const/16 v2, 0x1908

    invoke-direct {v0, v2}, Lorg/webrtc/ali/GlTextureFrameBuffer;-><init>(I)V

    iput-object v0, p0, Lorg/webrtc/ali/YuvConverter;->textureFrameBuffer:Lorg/webrtc/ali/GlTextureFrameBuffer;

    .line 97
    new-instance v0, Lorg/webrtc/ali/GlShader;

    const-string v2, "varying vec2 interp_tc;\nattribute vec4 in_pos;\nattribute vec4 in_tc;\n\nuniform mat4 texMatrix;\n\nvoid main() {\n    gl_Position = in_pos;\n    interp_tc = (texMatrix * in_tc).xy;\n}\n"

    const-string v3, "#extension GL_OES_EGL_image_external : require\nprecision mediump float;\nvarying vec2 interp_tc;\n\nuniform samplerExternalOES oesTex;\nuniform vec2 xUnit;\nuniform vec4 coeffs;\n\nvoid main() {\n  gl_FragColor.r = coeffs.a + dot(coeffs.rgb,\n      texture2D(oesTex, interp_tc - 1.5 * xUnit).rgb);\n  gl_FragColor.g = coeffs.a + dot(coeffs.rgb,\n      texture2D(oesTex, interp_tc - 0.5 * xUnit).rgb);\n  gl_FragColor.b = coeffs.a + dot(coeffs.rgb,\n      texture2D(oesTex, interp_tc + 0.5 * xUnit).rgb);\n  gl_FragColor.a = coeffs.a + dot(coeffs.rgb,\n      texture2D(oesTex, interp_tc + 1.5 * xUnit).rgb);\n}\n"

    invoke-direct {v0, v2, v3}, Lorg/webrtc/ali/GlShader;-><init>(Ljava/lang/String;Ljava/lang/String;)V

    iput-object v0, p0, Lorg/webrtc/ali/YuvConverter;->shader:Lorg/webrtc/ali/GlShader;

    .line 98
    invoke-virtual {v0}, Lorg/webrtc/ali/GlShader;->useProgram()V

    .line 99
    iget-object v0, p0, Lorg/webrtc/ali/YuvConverter;->shader:Lorg/webrtc/ali/GlShader;

    const-string v2, "texMatrix"

    invoke-virtual {v0, v2}, Lorg/webrtc/ali/GlShader;->getUniformLocation(Ljava/lang/String;)I

    move-result v0

    iput v0, p0, Lorg/webrtc/ali/YuvConverter;->texMatrixLoc:I

    .line 100
    iget-object v0, p0, Lorg/webrtc/ali/YuvConverter;->shader:Lorg/webrtc/ali/GlShader;

    const-string v2, "xUnit"

    invoke-virtual {v0, v2}, Lorg/webrtc/ali/GlShader;->getUniformLocation(Ljava/lang/String;)I

    move-result v0

    iput v0, p0, Lorg/webrtc/ali/YuvConverter;->xUnitLoc:I

    .line 101
    iget-object v0, p0, Lorg/webrtc/ali/YuvConverter;->shader:Lorg/webrtc/ali/GlShader;

    const-string v2, "coeffs"

    invoke-virtual {v0, v2}, Lorg/webrtc/ali/GlShader;->getUniformLocation(Ljava/lang/String;)I

    move-result v0

    iput v0, p0, Lorg/webrtc/ali/YuvConverter;->coeffsLoc:I

    .line 102
    iget-object v0, p0, Lorg/webrtc/ali/YuvConverter;->shader:Lorg/webrtc/ali/GlShader;

    const-string v2, "oesTex"

    invoke-virtual {v0, v2}, Lorg/webrtc/ali/GlShader;->getUniformLocation(Ljava/lang/String;)I

    move-result v0

    invoke-static {v0, v1}, Landroid/opengl/GLES20;->glUniform1i(II)V

    .line 103
    const-string v0, "Initialize fragment shader uniform values."

    invoke-static {v0}, Lorg/webrtc/ali/GlUtil;->checkNoGLES2Error(Ljava/lang/String;)V

    .line 105
    iget-object v0, p0, Lorg/webrtc/ali/YuvConverter;->shader:Lorg/webrtc/ali/GlShader;

    sget-object v1, Lorg/webrtc/ali/YuvConverter;->DEVICE_RECTANGLE:Ljava/nio/FloatBuffer;

    const/4 v2, 0x2

    const-string v3, "in_pos"

    invoke-virtual {v0, v3, v2, v1}, Lorg/webrtc/ali/GlShader;->setVertexAttribArray(Ljava/lang/String;ILjava/nio/FloatBuffer;)V

    .line 108
    iget-object v0, p0, Lorg/webrtc/ali/YuvConverter;->shader:Lorg/webrtc/ali/GlShader;

    sget-object v1, Lorg/webrtc/ali/YuvConverter;->TEXTURE_RECTANGLE:Ljava/nio/FloatBuffer;

    const-string v3, "in_tc"

    invoke-virtual {v0, v3, v2, v1}, Lorg/webrtc/ali/GlShader;->setVertexAttribArray(Ljava/lang/String;ILjava/nio/FloatBuffer;)V

    .line 109
    return-void
.end method


# virtual methods
.method public convert(Ljava/nio/ByteBuffer;IIII[F)V
    .registers 28
    .param p1, "buf"    # Ljava/nio/ByteBuffer;
    .param p2, "width"    # I
    .param p3, "height"    # I
    .param p4, "stride"    # I
    .param p5, "srcTextureId"    # I
    .param p6, "transformMatrix"    # [F

    .line 113
    move-object/from16 v0, p0

    move/from16 v1, p2

    move/from16 v2, p3

    move/from16 v3, p4

    iget-object v4, v0, Lorg/webrtc/ali/YuvConverter;->threadChecker:Lorg/webrtc/ali/ThreadUtils$ThreadChecker;

    invoke-virtual {v4}, Lorg/webrtc/ali/ThreadUtils$ThreadChecker;->checkIsOnValidThread()V

    .line 114
    iget-boolean v4, v0, Lorg/webrtc/ali/YuvConverter;->released:Z

    if-nez v4, :cond_120

    .line 147
    rem-int/lit8 v4, v3, 0x8

    if-nez v4, :cond_116

    .line 150
    if-lt v3, v1, :cond_10c

    .line 154
    add-int/lit8 v4, v1, 0x3

    const/4 v5, 0x4

    div-int/2addr v4, v5

    .line 155
    .local v4, "y_width":I
    add-int/lit8 v6, v1, 0x7

    div-int/lit8 v6, v6, 0x8

    .line 156
    .local v6, "uv_width":I
    add-int/lit8 v7, v2, 0x1

    div-int/lit8 v7, v7, 0x2

    .line 157
    .local v7, "uv_height":I
    add-int v8, v2, v7

    .line 158
    .local v8, "total_height":I
    mul-int v9, v3, v8

    .line 160
    .local v9, "size":I
    invoke-virtual/range {p1 .. p1}, Ljava/nio/ByteBuffer;->capacity()I

    move-result v10

    if-lt v10, v9, :cond_102

    .line 165
    nop

    .line 166
    invoke-static {}, Lorg/webrtc/ali/RendererCommon;->verticalFlipMatrix()[F

    move-result-object v10

    move-object/from16 v11, p6

    invoke-static {v11, v10}, Lorg/webrtc/ali/RendererCommon;->multiplyMatrices([F[F)[F

    move-result-object v10

    .line 168
    .end local p6    # "transformMatrix":[F
    .local v10, "transformMatrix":[F
    div-int/lit8 v15, v3, 0x4

    .line 169
    .local v15, "frameBufferWidth":I
    move v14, v8

    .line 170
    .local v14, "frameBufferHeight":I
    iget-object v11, v0, Lorg/webrtc/ali/YuvConverter;->textureFrameBuffer:Lorg/webrtc/ali/GlTextureFrameBuffer;

    invoke-virtual {v11, v15, v14}, Lorg/webrtc/ali/GlTextureFrameBuffer;->setSize(II)V

    .line 173
    iget-object v11, v0, Lorg/webrtc/ali/YuvConverter;->textureFrameBuffer:Lorg/webrtc/ali/GlTextureFrameBuffer;

    invoke-virtual {v11}, Lorg/webrtc/ali/GlTextureFrameBuffer;->getFrameBufferId()I

    move-result v11

    const v13, 0x8d40

    invoke-static {v13, v11}, Landroid/opengl/GLES20;->glBindFramebuffer(II)V

    .line 174
    const-string v11, "glBindFramebuffer"

    invoke-static {v11}, Lorg/webrtc/ali/GlUtil;->checkNoGLES2Error(Ljava/lang/String;)V

    .line 176
    const v11, 0x84c0

    invoke-static {v11}, Landroid/opengl/GLES20;->glActiveTexture(I)V

    .line 177
    const v12, 0x8d65

    move/from16 v11, p5

    invoke-static {v12, v11}, Landroid/opengl/GLES20;->glBindTexture(II)V

    .line 178
    iget v12, v0, Lorg/webrtc/ali/YuvConverter;->texMatrixLoc:I

    const/4 v13, 0x1

    const/4 v11, 0x0

    invoke-static {v12, v13, v11, v10, v11}, Landroid/opengl/GLES20;->glUniformMatrix4fv(IIZ[FI)V

    .line 181
    invoke-static {v11, v11, v4, v2}, Landroid/opengl/GLES20;->glViewport(IIII)V

    .line 183
    iget v12, v0, Lorg/webrtc/ali/YuvConverter;->xUnitLoc:I

    aget v17, v10, v11

    int-to-float v5, v1

    div-float v5, v17, v5

    aget v17, v10, v13

    int-to-float v13, v1

    div-float v13, v17, v13

    invoke-static {v12, v5, v13}, Landroid/opengl/GLES20;->glUniform2f(IFF)V

    .line 187
    iget v5, v0, Lorg/webrtc/ali/YuvConverter;->coeffsLoc:I

    const v12, 0x3e991687    # 0.299f

    const v13, 0x3f1645a2    # 0.587f

    const v11, 0x3de978d5    # 0.114f

    move/from16 v19, v4

    .end local v4    # "y_width":I
    .local v19, "y_width":I
    const/4 v4, 0x0

    invoke-static {v5, v12, v13, v11, v4}, Landroid/opengl/GLES20;->glUniform4f(IFFFF)V

    .line 188
    const/4 v4, 0x5

    const/4 v5, 0x4

    const/4 v11, 0x0

    invoke-static {v4, v11, v5}, Landroid/opengl/GLES20;->glDrawArrays(III)V

    .line 191
    invoke-static {v11, v2, v6, v7}, Landroid/opengl/GLES20;->glViewport(IIII)V

    .line 193
    iget v5, v0, Lorg/webrtc/ali/YuvConverter;->xUnitLoc:I

    aget v12, v10, v11

    const/high16 v11, 0x40000000    # 2.0f

    mul-float v12, v12, v11

    int-to-float v13, v1

    div-float/2addr v12, v13

    const/4 v13, 0x1

    aget v13, v10, v13

    mul-float v13, v13, v11

    int-to-float v11, v1

    div-float/2addr v13, v11

    invoke-static {v5, v12, v13}, Landroid/opengl/GLES20;->glUniform2f(IFF)V

    .line 195
    iget v5, v0, Lorg/webrtc/ali/YuvConverter;->coeffsLoc:I

    const v11, -0x41d2f1aa    # -0.169f

    const v12, -0x4156872b    # -0.331f

    const/high16 v13, 0x3f000000    # 0.5f

    const v4, 0x3eff7cee    # 0.499f

    invoke-static {v5, v11, v12, v4, v13}, Landroid/opengl/GLES20;->glUniform4f(IFFFF)V

    .line 196
    const/4 v5, 0x5

    const/4 v11, 0x4

    const/4 v12, 0x0

    invoke-static {v5, v12, v11}, Landroid/opengl/GLES20;->glDrawArrays(III)V

    .line 199
    div-int/lit8 v5, v3, 0x8

    invoke-static {v5, v2, v6, v7}, Landroid/opengl/GLES20;->glViewport(IIII)V

    .line 200
    iget v5, v0, Lorg/webrtc/ali/YuvConverter;->coeffsLoc:I

    const v11, -0x4129fbe7    # -0.418f

    const v12, -0x42597f63    # -0.0813f

    invoke-static {v5, v4, v11, v12, v13}, Landroid/opengl/GLES20;->glUniform4f(IFFFF)V

    .line 201
    const/4 v4, 0x5

    const/4 v5, 0x4

    const/4 v11, 0x0

    invoke-static {v4, v11, v5}, Landroid/opengl/GLES20;->glDrawArrays(III)V

    .line 203
    const/4 v4, 0x0

    const/4 v12, 0x0

    const/16 v5, 0x1908

    const/16 v17, 0x1401

    const/4 v13, 0x0

    move v11, v4

    const v4, 0x8d65

    const/4 v4, 0x0

    move v13, v15

    move/from16 v18, v14

    .end local v14    # "frameBufferHeight":I
    .local v18, "frameBufferHeight":I
    move/from16 v20, v15

    .end local v15    # "frameBufferWidth":I
    .local v20, "frameBufferWidth":I
    move v15, v5

    move/from16 v16, v17

    move-object/from16 v17, p1

    invoke-static/range {v11 .. v17}, Landroid/opengl/GLES20;->glReadPixels(IIIIIILjava/nio/Buffer;)V

    .line 206
    const-string v5, "YuvConverter.convert"

    invoke-static {v5}, Lorg/webrtc/ali/GlUtil;->checkNoGLES2Error(Ljava/lang/String;)V

    .line 209
    const v5, 0x8d40

    invoke-static {v5, v4}, Landroid/opengl/GLES20;->glBindFramebuffer(II)V

    .line 210
    const/16 v5, 0xde1

    invoke-static {v5, v4}, Landroid/opengl/GLES20;->glBindTexture(II)V

    .line 214
    const v5, 0x8d65

    invoke-static {v5, v4}, Landroid/opengl/GLES20;->glBindTexture(II)V

    .line 215
    return-void

    .line 161
    .end local v10    # "transformMatrix":[F
    .end local v18    # "frameBufferHeight":I
    .end local v19    # "y_width":I
    .end local v20    # "frameBufferWidth":I
    .restart local v4    # "y_width":I
    .restart local p6    # "transformMatrix":[F
    :cond_102
    move/from16 v19, v4

    .end local v4    # "y_width":I
    .restart local v19    # "y_width":I
    new-instance v4, Ljava/lang/IllegalArgumentException;

    const-string v5, "YuvConverter.convert called with too small buffer"

    invoke-direct {v4, v5}, Ljava/lang/IllegalArgumentException;-><init>(Ljava/lang/String;)V

    throw v4

    .line 151
    .end local v6    # "uv_width":I
    .end local v7    # "uv_height":I
    .end local v8    # "total_height":I
    .end local v9    # "size":I
    .end local v19    # "y_width":I
    :cond_10c
    move-object/from16 v11, p6

    new-instance v4, Ljava/lang/IllegalArgumentException;

    const-string v5, "Invalid stride, must >= width"

    invoke-direct {v4, v5}, Ljava/lang/IllegalArgumentException;-><init>(Ljava/lang/String;)V

    throw v4

    .line 148
    :cond_116
    move-object/from16 v11, p6

    new-instance v4, Ljava/lang/IllegalArgumentException;

    const-string v5, "Invalid stride, must be a multiple of 8"

    invoke-direct {v4, v5}, Ljava/lang/IllegalArgumentException;-><init>(Ljava/lang/String;)V

    throw v4

    .line 115
    :cond_120
    move-object/from16 v11, p6

    new-instance v4, Ljava/lang/IllegalStateException;

    const-string v5, "YuvConverter.convert called on released object"

    invoke-direct {v4, v5}, Ljava/lang/IllegalStateException;-><init>(Ljava/lang/String;)V

    throw v4
.end method

.method public release()V
    .registers 2

    .line 218
    iget-object v0, p0, Lorg/webrtc/ali/YuvConverter;->threadChecker:Lorg/webrtc/ali/ThreadUtils$ThreadChecker;

    invoke-virtual {v0}, Lorg/webrtc/ali/ThreadUtils$ThreadChecker;->checkIsOnValidThread()V

    .line 219
    const/4 v0, 0x1

    iput-boolean v0, p0, Lorg/webrtc/ali/YuvConverter;->released:Z

    .line 220
    iget-object v0, p0, Lorg/webrtc/ali/YuvConverter;->shader:Lorg/webrtc/ali/GlShader;

    invoke-virtual {v0}, Lorg/webrtc/ali/GlShader;->release()V

    .line 221
    iget-object v0, p0, Lorg/webrtc/ali/YuvConverter;->textureFrameBuffer:Lorg/webrtc/ali/GlTextureFrameBuffer;

    invoke-virtual {v0}, Lorg/webrtc/ali/GlTextureFrameBuffer;->release()V

    .line 222
    return-void
.end method