package kotlin.reflect.jvm.internal.impl.serialization.deserialization;

import defpackage.cr4;
import defpackage.kp6;
import defpackage.pq6;
import defpackage.sq3;
import defpackage.vx8;
import defpackage.wo5;
import defpackage.y41;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import kotlin.reflect.jvm.internal.impl.descriptors.SourceElement;
import kotlin.reflect.jvm.internal.impl.metadata.ProtoBuf;
import kotlin.reflect.jvm.internal.impl.metadata.deserialization.BinaryVersion;
import kotlin.reflect.jvm.internal.impl.metadata.deserialization.NameResolver;
import kotlin.reflect.jvm.internal.impl.name.ClassId;

/* JADX INFO: loaded from: classes3.dex */
@vx8
public final class ProtoBasedClassDataFinder implements ClassDataFinder {

    @kp6
    private final Map<ClassId, ProtoBuf.Class> classIdToProto;

    @kp6
    private final sq3<ClassId, SourceElement> classSource;

    @kp6
    private final BinaryVersion metadataVersion;

    @kp6
    private final NameResolver nameResolver;

    /* JADX WARN: Multi-variable type inference failed */
    public ProtoBasedClassDataFinder(@kp6 ProtoBuf.PackageFragment packageFragment, @kp6 NameResolver nameResolver, @kp6 BinaryVersion binaryVersion, @kp6 sq3<? super ClassId, ? extends SourceElement> sq3Var) {
        cr4.g(packageFragment, "proto");
        cr4.g(nameResolver, "nameResolver");
        cr4.g(binaryVersion, "metadataVersion");
        cr4.g(sq3Var, "classSource");
        this.nameResolver = nameResolver;
        this.metadataVersion = binaryVersion;
        this.classSource = sq3Var;
        List<ProtoBuf.Class> class_List = packageFragment.getClass_List();
        cr4.f(class_List, "getClass_List(...)");
        List<ProtoBuf.Class> list = class_List;
        int iH = wo5.h(y41.p(list, 10));
        LinkedHashMap linkedHashMap = new LinkedHashMap(iH < 16 ? 16 : iH);
        for (Object obj : list) {
            linkedHashMap.put(NameResolverUtilKt.getClassId(this.nameResolver, ((ProtoBuf.Class) obj).getFqName()), obj);
        }
        this.classIdToProto = linkedHashMap;
    }

    @Override // kotlin.reflect.jvm.internal.impl.serialization.deserialization.ClassDataFinder
    @pq6
    public ClassData findClassData(@kp6 ClassId classId) {
        cr4.g(classId, "classId");
        ProtoBuf.Class r0 = this.classIdToProto.get(classId);
        if (r0 == null) {
            return null;
        }
        return new ClassData(this.nameResolver, r0, this.metadataVersion, (SourceElement) this.classSource.invoke(classId));
    }

    @kp6
    public final Collection<ClassId> getAllClassIds() {
        return this.classIdToProto.keySet();
    }
}