a.smali

.class public abstract Lwl/a;
.super Ljava/lang/Object;
.source "TensorBuffer.java"


# instance fields
.field public a:Ljava/nio/ByteBuffer;

.field public b:[I

.field public c:I

.field public final d:Z


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

    return-void
.end method

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

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

    const/4 v0, -0x1

    iput v0, p0, Lwl/a;->c:I

    const/4 v0, 0x1

    iput-boolean v0, p0, Lwl/a;->d:Z

    new-array v0, v0, [I

    const/4 v1, 0x0

    aput v1, v0, v1

    invoke-virtual {p0, v0}, Lwl/a;->a([I)V

    return-void
.end method

.method public constructor <init>([I)V
    .registers 3
    .param p1    # [I
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param

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

    const/4 v0, -0x1

    iput v0, p0, Lwl/a;->c:I

    const/4 v0, 0x0

    iput-boolean v0, p0, Lwl/a;->d:Z

    invoke-virtual {p0, p1}, Lwl/a;->a([I)V

    return-void
.end method

.method public static a(Lorg/tensorflow/lite/DataType;)Lwl/a;
    .registers 4
    .annotation build Landroidx/annotation/NonNull;
    .end annotation

    sget-object v0, Lwl/a$a;->a:[I

    invoke-virtual {p0}, Ljava/lang/Enum;->ordinal()I

    move-result v1

    aget v0, v0, v1

    const/4 v1, 0x1

    if-eq v0, v1, :cond_2b

    const/4 v1, 0x2

    if-ne v0, v1, :cond_14

    new-instance p0, Lwl/c;

    invoke-direct {p0}, Lwl/c;-><init>()V

    return-object p0

    :cond_14
    new-instance v0, Ljava/lang/AssertionError;

    new-instance v1, Ljava/lang/StringBuilder;

    invoke-direct {v1}, Ljava/lang/StringBuilder;-><init>()V

    const-string v2, "TensorBuffer does not support data type: "

    invoke-virtual {v1, v2}, Ljava/lang/StringBuilder;->append(Ljava/lang/String;)Ljava/lang/StringBuilder;

    invoke-virtual {v1, p0}, Ljava/lang/StringBuilder;->append(Ljava/lang/Object;)Ljava/lang/StringBuilder;

    invoke-virtual {v1}, Ljava/lang/StringBuilder;->toString()Ljava/lang/String;

    move-result-object p0

    invoke-direct {v0, p0}, Ljava/lang/AssertionError;-><init>(Ljava/lang/Object;)V

    throw v0

    :cond_2b
    new-instance p0, Lwl/b;

    invoke-direct {p0}, Lwl/b;-><init>()V

    return-object p0
.end method

.method public static a(Lwl/a;Lorg/tensorflow/lite/DataType;)Lwl/a;
    .registers 5
    .param p0    # Lwl/a;
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param
    .annotation build Landroidx/annotation/NonNull;
    .end annotation

    const-string v0, "Cannot create a buffer from null"

    invoke-static {p0, v0}, Lrl/d;->a(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;

    invoke-virtual {p0}, Lwl/a;->h()Z

    move-result v0

    if-eqz v0, :cond_10

    invoke-static {p1}, Lwl/a;->a(Lorg/tensorflow/lite/DataType;)Lwl/a;

    move-result-object v0

    goto :goto_16

    :cond_10
    iget-object v0, p0, Lwl/a;->b:[I

    invoke-static {v0, p1}, Lwl/a;->a([ILorg/tensorflow/lite/DataType;)Lwl/a;

    move-result-object v0

    :goto_16
    invoke-virtual {p0}, Lwl/a;->c()Lorg/tensorflow/lite/DataType;

    move-result-object v1

    sget-object v2, Lorg/tensorflow/lite/DataType;->FLOAT32:Lorg/tensorflow/lite/DataType;

    if-ne v1, v2, :cond_2a

    if-ne p1, v2, :cond_2a

    invoke-virtual {p0}, Lwl/a;->d()[F

    move-result-object p1

    iget-object p0, p0, Lwl/a;->b:[I

    invoke-virtual {v0, p1, p0}, Lwl/a;->a([F[I)V

    goto :goto_33

    :cond_2a
    invoke-virtual {p0}, Lwl/a;->e()[I

    move-result-object p1

    iget-object p0, p0, Lwl/a;->b:[I

    invoke-virtual {v0, p1, p0}, Lwl/a;->a([I[I)V

    :goto_33
    return-object v0
.end method

.method public static a([ILorg/tensorflow/lite/DataType;)Lwl/a;
    .registers 4
    .param p0    # [I
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param
    .annotation build Landroidx/annotation/NonNull;
    .end annotation

    sget-object v0, Lwl/a$a;->a:[I

    invoke-virtual {p1}, Ljava/lang/Enum;->ordinal()I

    move-result v1

    aget v0, v0, v1

    const/4 v1, 0x1

    if-eq v0, v1, :cond_2b

    const/4 v1, 0x2

    if-ne v0, v1, :cond_14

    new-instance p1, Lwl/c;

    invoke-direct {p1, p0}, Lwl/c;-><init>([I)V

    return-object p1

    :cond_14
    new-instance p0, Ljava/lang/AssertionError;

    new-instance v0, Ljava/lang/StringBuilder;

    invoke-direct {v0}, Ljava/lang/StringBuilder;-><init>()V

    const-string v1, "TensorBuffer does not support data type: "

    invoke-virtual {v0, v1}, Ljava/lang/StringBuilder;->append(Ljava/lang/String;)Ljava/lang/StringBuilder;

    invoke-virtual {v0, p1}, Ljava/lang/StringBuilder;->append(Ljava/lang/Object;)Ljava/lang/StringBuilder;

    invoke-virtual {v0}, Ljava/lang/StringBuilder;->toString()Ljava/lang/String;

    move-result-object p1

    invoke-direct {p0, p1}, Ljava/lang/AssertionError;-><init>(Ljava/lang/Object;)V

    throw p0

    :cond_2b
    new-instance p1, Lwl/b;

    invoke-direct {p1, p0}, Lwl/b;-><init>([I)V

    return-object p1
.end method

.method public static c([I)I
    .registers 5
    .param p0    # [I
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param

    const-string v0, "Shape cannot be null."

    invoke-static {p0, v0}, Lrl/d;->a(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;

    array-length v0, p0

    const/4 v1, 0x1

    const/4 v2, 0x0

    :goto_8
    if-ge v2, v0, :cond_11

    aget v3, p0, v2

    mul-int v1, v1, v3

    add-int/lit8 v2, v2, 0x1

    goto :goto_8

    :cond_11
    return v1
.end method

.method public static d([I)Z
    .registers 6
    .param p0    # [I
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param

    array-length v0, p0

    const/4 v1, 0x1

    if-nez v0, :cond_5

    return v1

    :cond_5
    array-length v0, p0

    const/4 v2, 0x0

    const/4 v3, 0x0

    :goto_8
    if-ge v3, v0, :cond_12

    aget v4, p0, v3

    if-gez v4, :cond_f

    return v2

    :cond_f
    add-int/lit8 v3, v3, 0x1

    goto :goto_8

    :cond_12
    return v1
.end method


# virtual methods
.method public final a()V
    .registers 6

    iget-object v0, p0, Lwl/a;->b:[I

    invoke-static {v0}, Lwl/a;->c([I)I

    move-result v0

    iget-object v1, p0, Lwl/a;->a:Ljava/nio/ByteBuffer;

    invoke-virtual {v1}, Ljava/nio/ByteBuffer;->limit()I

    move-result v1

    invoke-virtual {p0}, Lwl/a;->g()I

    move-result v2

    mul-int v2, v2, v0

    const/4 v0, 0x1

    const/4 v3, 0x0

    if-ne v1, v2, :cond_18

    const/4 v1, 0x1

    goto :goto_19

    :cond_18
    const/4 v1, 0x0

    :goto_19
    const/4 v2, 0x2

    new-array v2, v2, [Ljava/lang/Object;

    iget-object v4, p0, Lwl/a;->a:Ljava/nio/ByteBuffer;

    invoke-virtual {v4}, Ljava/nio/ByteBuffer;->limit()I

    move-result v4

    invoke-static {v4}, Ljava/lang/Integer;->valueOf(I)Ljava/lang/Integer;

    move-result-object v4

    aput-object v4, v2, v3

    iget-object v3, p0, Lwl/a;->b:[I

    invoke-static {v3}, Ljava/util/Arrays;->toString([I)Ljava/lang/String;

    move-result-object v3

    aput-object v3, v2, v0

    const-string v0, "The size of underlying ByteBuffer (%d) and the shape (%s) do not match. The ByteBuffer may have been changed."

    invoke-static {v0, v2}, Ljava/lang/String;->format(Ljava/lang/String;[Ljava/lang/Object;)Ljava/lang/String;

    move-result-object v0

    invoke-static {v1, v0}, Lrl/d;->b(ZLjava/lang/Object;)V

    return-void
.end method

.method public a(Ljava/nio/ByteBuffer;[I)V
    .registers 6
    .param p1    # Ljava/nio/ByteBuffer;
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param
    .param p2    # [I
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param

    const-string v0, "Byte buffer cannot be null."

    invoke-static {p1, v0}, Lrl/d;->a(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;

    invoke-static {p2}, Lwl/a;->c([I)I

    move-result v0

    invoke-virtual {p1}, Ljava/nio/ByteBuffer;->limit()I

    move-result v1

    invoke-virtual {p0}, Lwl/a;->g()I

    move-result v2

    mul-int v2, v2, v0

    if-ne v1, v2, :cond_17

    const/4 v0, 0x1

    goto :goto_18

    :cond_17
    const/4 v0, 0x0

    :goto_18
    const-string v1, "The size of byte buffer and the shape do not match."

    invoke-static {v0, v1}, Lrl/d;->a(ZLjava/lang/Object;)V

    invoke-virtual {p0, p2}, Lwl/a;->b([I)V

    invoke-virtual {p1}, Ljava/nio/ByteBuffer;->rewind()Ljava/nio/Buffer;

    iput-object p1, p0, Lwl/a;->a:Ljava/nio/ByteBuffer;

    return-void
.end method

.method public abstract a([F[I)V
    .param p1    # [F
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param
    .param p2    # [I
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param
.end method

.method public final a([I)V
    .registers 4
    .param p1    # [I
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param

    const-string v0, "TensorBuffer shape cannot be null."

    invoke-static {p1, v0}, Lrl/d;->a(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;

    invoke-static {p1}, Lwl/a;->d([I)Z

    move-result v0

    const-string v1, "Values in TensorBuffer shape should be non-negative."

    invoke-static {v0, v1}, Lrl/d;->a(ZLjava/lang/Object;)V

    invoke-static {p1}, Lwl/a;->c([I)I

    move-result v0

    invoke-virtual {p1}, [I->clone()Ljava/lang/Object;

    move-result-object p1

    check-cast p1, [I

    iput-object p1, p0, Lwl/a;->b:[I

    iget p1, p0, Lwl/a;->c:I

    if-ne p1, v0, :cond_1f

    return-void

    :cond_1f
    iput v0, p0, Lwl/a;->c:I

    invoke-virtual {p0}, Lwl/a;->g()I

    move-result p1

    mul-int v0, v0, p1

    invoke-static {v0}, Ljava/nio/ByteBuffer;->allocateDirect(I)Ljava/nio/ByteBuffer;

    move-result-object p1

    iput-object p1, p0, Lwl/a;->a:Ljava/nio/ByteBuffer;

    invoke-static {}, Ljava/nio/ByteOrder;->nativeOrder()Ljava/nio/ByteOrder;

    move-result-object v0

    invoke-virtual {p1, v0}, Ljava/nio/ByteBuffer;->order(Ljava/nio/ByteOrder;)Ljava/nio/ByteBuffer;

    return-void
.end method

.method public abstract a([I[I)V
    .param p1    # [I
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param
    .param p2    # [I
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param
.end method

.method public b()Ljava/nio/ByteBuffer;
    .registers 2
    .annotation build Landroidx/annotation/NonNull;
    .end annotation

    iget-object v0, p0, Lwl/a;->a:Ljava/nio/ByteBuffer;

    return-object v0
.end method

.method public b([I)V
    .registers 3
    .param p1    # [I
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param

    iget-boolean v0, p0, Lwl/a;->d:Z

    if-eqz v0, :cond_8

    invoke-virtual {p0, p1}, Lwl/a;->a([I)V

    goto :goto_19

    :cond_8
    iget-object v0, p0, Lwl/a;->b:[I

    invoke-static {p1, v0}, Ljava/util/Arrays;->equals([I[I)Z

    move-result v0

    invoke-static {v0}, Lrl/d;->a(Z)V

    invoke-virtual {p1}, [I->clone()Ljava/lang/Object;

    move-result-object p1

    check-cast p1, [I

    iput-object p1, p0, Lwl/a;->b:[I

    :goto_19
    return-void
.end method

.method public abstract c()Lorg/tensorflow/lite/DataType;
.end method

.method public abstract d()[F
    .annotation build Landroidx/annotation/NonNull;
    .end annotation
.end method

.method public abstract e()[I
    .annotation build Landroidx/annotation/NonNull;
    .end annotation
.end method

.method public f()[I
    .registers 3
    .annotation build Landroidx/annotation/NonNull;
    .end annotation

    invoke-virtual {p0}, Lwl/a;->a()V

    iget-object v0, p0, Lwl/a;->b:[I

    array-length v1, v0

    invoke-static {v0, v1}, Ljava/util/Arrays;->copyOf([II)[I

    move-result-object v0

    return-object v0
.end method

.method public abstract g()I
.end method

.method public h()Z
    .registers 2

    iget-boolean v0, p0, Lwl/a;->d:Z

    return v0
.end method