package com.google.android.gms.internal.firebase_ml;

import android.os.SystemClock;
import android.util.SparseArray;
import com.google.android.gms.common.internal.GmsLogger;
import com.google.android.gms.internal.firebase_ml.zzmd;
import com.google.firebase.FirebaseApp;
import com.google.firebase.ml.common.FirebaseMLException;
import com.google.firebase.ml.common.modeldownload.FirebaseLocalModel;
import com.google.firebase.ml.common.modeldownload.FirebaseRemoteModel;
import com.google.firebase.ml.custom.FirebaseModelOutputs;
import java.nio.MappedByteBuffer;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.atomic.AtomicLong;
import o.C15516gsl;
import org.tensorflow.lite.TensorFlowLite;

/* loaded from: classes2.dex */
public final class zzpe implements zznm<FirebaseModelOutputs, zzpa>, zznw {
    private final FirebaseLocalModel zzaqk;
    private final zzov zzaql;
    private final zzog zzaqm;
    private final zznu zzaqs;
    private final FirebaseRemoteModel zzaqt;
    private final boolean zzast;
    private AtomicLong zzatb = new AtomicLong(0);
    private zzpj zzatc = null;
    public static final String zzasz = TensorFlowLite.version();
    private static final GmsLogger zzaoz = new GmsLogger("ModelInterpreterTask", "");
    private static boolean zzata = true;

    public zzpe(FirebaseApp firebaseApp, FirebaseLocalModel firebaseLocalModel, FirebaseRemoteModel firebaseRemoteModel, boolean z) {
        this.zzaqk = firebaseLocalModel;
        this.zzaqt = firebaseRemoteModel;
        this.zzast = z;
        this.zzaqs = zznu.zza(firebaseApp, 2);
        if (firebaseRemoteModel != null) {
            this.zzaql = zzov.zza(firebaseApp, firebaseRemoteModel, new zzoz(), new zzpb(firebaseApp.getApplicationContext()), zzok.CUSTOM);
            GmsLogger gmsLogger = zzaoz;
            String valueOf = String.valueOf(firebaseRemoteModel);
            StringBuilder sb = new StringBuilder(String.valueOf(valueOf).length() + 27);
            sb.append("Cloud model source is set: ");
            sb.append(valueOf);
            gmsLogger.d("ModelInterpreterTask", sb.toString());
        } else {
            this.zzaql = null;
        }
        if (firebaseLocalModel == null) {
            this.zzaqm = null;
            return;
        }
        this.zzaqm = new zzog(firebaseApp.getApplicationContext(), firebaseLocalModel);
        GmsLogger gmsLogger2 = zzaoz;
        String valueOf2 = String.valueOf(firebaseLocalModel);
        StringBuilder sb2 = new StringBuilder(String.valueOf(valueOf2).length() + 27);
        sb2.append("Local model source is set: ");
        sb2.append(valueOf2);
        gmsLogger2.d("ModelInterpreterTask", sb2.toString());
    }

    @Override // com.google.android.gms.internal.firebase_ml.zznm
    public final zznw zzlm() {
        return this;
    }

    /* JADX INFO: Access modifiers changed from: private */
    /* JADX WARN: Removed duplicated region for block: B:96:0x022a  */
    /* JADX WARN: Removed duplicated region for block: B:99:0x022c  */
    @Override // com.google.android.gms.internal.firebase_ml.zznm
    /*
        Code decompiled incorrectly, please refer to instructions dump.
        To view partially-correct add '--show-bad-code' argument
    */
    public final com.google.firebase.ml.custom.FirebaseModelOutputs zza(com.google.android.gms.internal.firebase_ml.zzpa r19) throws com.google.firebase.ml.common.FirebaseMLException {
        /*
            Method dump skipped, instructions count: 596
            To view this dump add '--comments-level debug' option
        */
        throw new UnsupportedOperationException("Method not decompiled: com.google.android.gms.internal.firebase_ml.zzpe.zza(com.google.android.gms.internal.firebase_ml.zzpa):com.google.firebase.ml.custom.FirebaseModelOutputs");
    }

    @Override // com.google.android.gms.internal.firebase_ml.zznw
    public final void zzlp() throws FirebaseMLException {
        synchronized (this) {
            zza(zzpf.zzatd);
        }
    }

    private final void zza(zzpi zzpiVar) throws FirebaseMLException {
        Exception exc;
        boolean z;
        synchronized (this) {
            long elapsedRealtime = SystemClock.elapsedRealtime();
            ArrayList arrayList = new ArrayList();
            Exception e = null;
            boolean z2 = false;
            try {
                z = zza(elapsedRealtime, zzpiVar, arrayList);
                exc = null;
            } catch (Exception e2) {
                exc = e2;
                z = false;
            }
            if (z) {
                zze(arrayList);
                return;
            }
            try {
                z2 = zzb(elapsedRealtime, zzpiVar, arrayList);
            } catch (Exception e3) {
                e = e3;
            }
            if (z2) {
                zze(arrayList);
                return;
            }
            arrayList.add(zzmk.NO_VALID_MODEL);
            zze(arrayList);
            if (exc != null) {
                throw new FirebaseMLException("Cloud model load failed: ", 14, exc);
            }
            if (e != null) {
                throw new FirebaseMLException("Local model load failed: ", 14, e);
            }
            throw new FirebaseMLException("Cannot load custom model", 14);
        }
    }

    private final boolean zza(long j, zzpi zzpiVar, List<zzmk> list) throws FirebaseMLException {
        synchronized (this) {
            zzov zzovVar = this.zzaql;
            if (zzovVar != null) {
                try {
                    MappedByteBuffer load = zzovVar.load();
                    if (load != null) {
                        zzpj zzd = zzpiVar.zzd(load);
                        this.zzatc = zzd;
                        zzd.setUseNNAPI(this.zzast);
                        this.zzatb.set(SystemClock.elapsedRealtime() - j);
                        zzaoz.d("ModelInterpreterTask", "Cloud model source is loaded successfully");
                        return true;
                    }
                    zzaoz.d("ModelInterpreterTask", "Cloud model source can NOT be loaded, try local model.");
                    list.add(zzmk.CLOUD_MODEL_LOADER_LOADS_NO_MODEL);
                } catch (FirebaseMLException e) {
                    zzaoz.d("ModelInterpreterTask", "Cloud model source can NOT be loaded, try local model.");
                    list.add(zzmk.CLOUD_MODEL_LOADER_ERROR);
                    throw e;
                }
            }
            return false;
        }
    }

    private final boolean zzb(long j, zzpi zzpiVar, List<zzmk> list) throws FirebaseMLException {
        MappedByteBuffer load;
        synchronized (this) {
            zzog zzogVar = this.zzaqm;
            if (zzogVar == null || (load = zzogVar.load()) == null) {
                return false;
            }
            try {
                zzpj zzd = zzpiVar.zzd(load);
                this.zzatc = zzd;
                zzd.setUseNNAPI(this.zzast);
                this.zzatb.set(SystemClock.elapsedRealtime() - j);
                zzaoz.d("ModelInterpreterTask", "Local model source is loaded successfully");
                return true;
            } catch (RuntimeException e) {
                list.add(zzmk.LOCAL_MODEL_INVALID);
                throw e;
            }
        }
    }

    @Override // com.google.android.gms.internal.firebase_ml.zznw
    public final void release() {
        synchronized (this) {
            zzpj zzpjVar = this.zzatc;
            if (zzpjVar != null) {
                zzpjVar.close();
                this.zzatc = null;
            }
            zzata = true;
        }
    }

    public final int getInputIndex(String str) throws FirebaseMLException {
        int inputIndex;
        synchronized (this) {
            zzpj zzpjVar = this.zzatc;
            if (zzpjVar == null) {
                throw new FirebaseMLException("Trying to get input index while model has not been initialized yet, or has been released.", 14);
            }
            inputIndex = zzpjVar.getInputIndex(str);
        }
        return inputIndex;
    }

    public final int getOutputIndex(String str) throws FirebaseMLException {
        int outputIndex;
        synchronized (this) {
            zzpj zzpjVar = this.zzatc;
            if (zzpjVar == null) {
                throw new FirebaseMLException("Trying to get output index while model has not been initialized yet, or has been released.", 14);
            }
            outputIndex = zzpjVar.getOutputIndex(str);
        }
        return outputIndex;
    }

    private final void zza(final zzmk zzmkVar, final long j, final zzpa zzpaVar, final boolean z) {
        this.zzaqs.zza(new zznv(this, j, zzpaVar, zzmkVar, z) { // from class: com.google.android.gms.internal.firebase_ml.zzpg
            private final zzpe zzate;
            private final long zzatf;
            private final zzpa zzatg;
            private final zzmk zzath;
            private final boolean zzati;

            @Override // com.google.android.gms.internal.firebase_ml.zznv
            public final zzmd.zzq.zza zzm() {
                return this.zzate.zza(this.zzatf, this.zzatg, this.zzath, this.zzati);
            }

            /* JADX INFO: Access modifiers changed from: package-private */
            {
                this.zzate = this;
                this.zzatf = j;
                this.zzatg = zzpaVar;
                this.zzath = zzmkVar;
                this.zzati = z;
            }
        }, zzmn.CUSTOM_MODEL_RUN);
    }

    private final void zze(final List<zzmk> list) {
        this.zzaqs.zza(new zznv(this, list) { // from class: com.google.android.gms.internal.firebase_ml.zzph
            private final zzpe zzate;
            private final List zzatj;

            @Override // com.google.android.gms.internal.firebase_ml.zznv
            public final zzmd.zzq.zza zzm() {
                return this.zzate.zzf(this.zzatj);
            }

            /* JADX INFO: Access modifiers changed from: package-private */
            {
                this.zzate = this;
                this.zzatj = list;
            }
        }, zzmn.CUSTOM_MODEL_LOAD);
    }

    /* JADX INFO: Access modifiers changed from: package-private */
    public final /* synthetic */ zzmd.zzq.zza zzf(List list) {
        if (list.isEmpty()) {
            list.add(zzmk.NO_ERROR);
        }
        zzmd.zzn.zza zzl = zzmd.zzn.zzjo().zzl(this.zzatb.get()).zzh(list).zzl(this.zzast);
        FirebaseRemoteModel firebaseRemoteModel = this.zzaqt;
        if (firebaseRemoteModel != null) {
            zzl.zzf(firebaseRemoteModel.zzmh());
        }
        FirebaseLocalModel firebaseLocalModel = this.zzaqk;
        if (firebaseLocalModel != null) {
            zzl.zzg(firebaseLocalModel.zzmh());
        }
        return zzmd.zzq.zzjx().zzb(zzmd.zzac.zzli().zzbu(zzasz)).zzb(zzl);
    }

    /* JADX INFO: Access modifiers changed from: package-private */
    public final /* synthetic */ zzmd.zzq.zza zza(long j, zzpa zzpaVar, zzmk zzmkVar, boolean z) {
        zzmd.zzo zzmh;
        long elapsedRealtime = SystemClock.elapsedRealtime();
        ArrayList arrayList = new ArrayList();
        ArrayList arrayList2 = new ArrayList();
        if (zzpaVar != null) {
            SparseArray<zzpc> zzml = zzpaVar.zzasw.zzml();
            for (int i = 0; i < zzml.size(); i++) {
                arrayList.add(zzml.valueAt(i).zzmu());
            }
            SparseArray<zzpc> zzmm = zzpaVar.zzasw.zzmm();
            for (int i2 = 0; i2 < zzmm.size(); i2++) {
                arrayList2.add(zzmm.valueAt(i2).zzmu());
            }
        }
        zzmd.zzm.zza zzb = zzmd.zzm.zzjk().zzb(zzmd.zzs.zzkb().zzn(elapsedRealtime - j).zzc(zzmkVar).zzz(zzata).zzaa(true).zzab(true).zzac(this.zzast));
        if (z) {
            zzmh = this.zzaqt.zzmh();
        } else {
            zzmh = this.zzaqk.zzmh();
        }
        return zzmd.zzq.zzjx().zzb(zzmd.zzac.zzli().zzbu(zzasz)).zzb(zzb.zzc(zzmh).zzc(arrayList).zzd(arrayList2).zzk(this.zzatb.get()));
    }

    /* JADX INFO: Access modifiers changed from: package-private */
    public static final /* synthetic */ zzpj zzc(MappedByteBuffer mappedByteBuffer) {
        return new zzpj(new C15516gsl(mappedByteBuffer));
    }
}