2025-03-31 16:33:42 +02:00

512 lines
16 KiB
Java

package androidx.collection;
import java.lang.reflect.Array;
import java.util.Collection;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
/* loaded from: classes.dex */
public final class ArraySet<E> implements Collection<E>, Set<E> {
private static final int BASE_SIZE = 4;
private static final int CACHE_SIZE = 10;
private static final boolean DEBUG = false;
private static final int[] INT = new int[0];
private static final Object[] OBJECT = new Object[0];
private static final String TAG = "ArraySet";
private static Object[] sBaseCache;
private static int sBaseCacheSize;
private static Object[] sTwiceBaseCache;
private static int sTwiceBaseCacheSize;
Object[] mArray;
private MapCollections<E, E> mCollections;
private int[] mHashes;
int mSize;
@Override // java.util.Collection, java.util.Set
public boolean isEmpty() {
return this.mSize <= 0;
}
@Override // java.util.Collection, java.util.Set
public int size() {
return this.mSize;
}
private int indexOf(Object obj, int i) {
int i2 = this.mSize;
if (i2 == 0) {
return -1;
}
int binarySearch = ContainerHelpers.binarySearch(this.mHashes, i2, i);
if (binarySearch < 0 || obj.equals(this.mArray[binarySearch])) {
return binarySearch;
}
int i3 = binarySearch + 1;
while (i3 < i2 && this.mHashes[i3] == i) {
if (obj.equals(this.mArray[i3])) {
return i3;
}
i3++;
}
for (int i4 = binarySearch - 1; i4 >= 0 && this.mHashes[i4] == i; i4--) {
if (obj.equals(this.mArray[i4])) {
return i4;
}
}
return ~i3;
}
private int indexOfNull() {
int i = this.mSize;
if (i == 0) {
return -1;
}
int binarySearch = ContainerHelpers.binarySearch(this.mHashes, i, 0);
if (binarySearch < 0 || this.mArray[binarySearch] == null) {
return binarySearch;
}
int i2 = binarySearch + 1;
while (i2 < i && this.mHashes[i2] == 0) {
if (this.mArray[i2] == null) {
return i2;
}
i2++;
}
for (int i3 = binarySearch - 1; i3 >= 0 && this.mHashes[i3] == 0; i3--) {
if (this.mArray[i3] == null) {
return i3;
}
}
return ~i2;
}
private void allocArrays(int i) {
if (i == 8) {
synchronized (ArraySet.class) {
Object[] objArr = sTwiceBaseCache;
if (objArr != null) {
this.mArray = objArr;
sTwiceBaseCache = (Object[]) objArr[0];
this.mHashes = (int[]) objArr[1];
objArr[1] = null;
objArr[0] = null;
sTwiceBaseCacheSize--;
return;
}
}
} else if (i == 4) {
synchronized (ArraySet.class) {
Object[] objArr2 = sBaseCache;
if (objArr2 != null) {
this.mArray = objArr2;
sBaseCache = (Object[]) objArr2[0];
this.mHashes = (int[]) objArr2[1];
objArr2[1] = null;
objArr2[0] = null;
sBaseCacheSize--;
return;
}
}
}
this.mHashes = new int[i];
this.mArray = new Object[i];
}
private static void freeArrays(int[] iArr, Object[] objArr, int i) {
if (iArr.length == 8) {
synchronized (ArraySet.class) {
if (sTwiceBaseCacheSize < 10) {
objArr[0] = sTwiceBaseCache;
objArr[1] = iArr;
for (int i2 = i - 1; i2 >= 2; i2--) {
objArr[i2] = null;
}
sTwiceBaseCache = objArr;
sTwiceBaseCacheSize++;
}
}
return;
}
if (iArr.length == 4) {
synchronized (ArraySet.class) {
if (sBaseCacheSize < 10) {
objArr[0] = sBaseCache;
objArr[1] = iArr;
for (int i3 = i - 1; i3 >= 2; i3--) {
objArr[i3] = null;
}
sBaseCache = objArr;
sBaseCacheSize++;
}
}
}
}
public ArraySet() {
this(0);
}
public ArraySet(int i) {
if (i == 0) {
this.mHashes = INT;
this.mArray = OBJECT;
} else {
allocArrays(i);
}
this.mSize = 0;
}
/* JADX WARN: Multi-variable type inference failed */
public ArraySet(ArraySet<E> arraySet) {
this();
if (arraySet != 0) {
addAll((ArraySet) arraySet);
}
}
/* JADX WARN: Multi-variable type inference failed */
public ArraySet(Collection<E> collection) {
this();
if (collection != 0) {
addAll(collection);
}
}
@Override // java.util.Collection, java.util.Set
public void clear() {
int i = this.mSize;
if (i != 0) {
freeArrays(this.mHashes, this.mArray, i);
this.mHashes = INT;
this.mArray = OBJECT;
this.mSize = 0;
}
}
public void ensureCapacity(int i) {
int[] iArr = this.mHashes;
if (iArr.length < i) {
Object[] objArr = this.mArray;
allocArrays(i);
int i2 = this.mSize;
if (i2 > 0) {
System.arraycopy(iArr, 0, this.mHashes, 0, i2);
System.arraycopy(objArr, 0, this.mArray, 0, this.mSize);
}
freeArrays(iArr, objArr, this.mSize);
}
}
@Override // java.util.Collection, java.util.Set
public boolean contains(Object obj) {
return indexOf(obj) >= 0;
}
public int indexOf(Object obj) {
return obj == null ? indexOfNull() : indexOf(obj, obj.hashCode());
}
public E valueAt(int i) {
return (E) this.mArray[i];
}
@Override // java.util.Collection, java.util.Set
public boolean add(E e) {
int i;
int indexOf;
if (e == null) {
indexOf = indexOfNull();
i = 0;
} else {
int hashCode = e.hashCode();
i = hashCode;
indexOf = indexOf(e, hashCode);
}
if (indexOf >= 0) {
return false;
}
int i2 = ~indexOf;
int i3 = this.mSize;
int[] iArr = this.mHashes;
if (i3 >= iArr.length) {
int i4 = 8;
if (i3 >= 8) {
i4 = (i3 >> 1) + i3;
} else if (i3 < 4) {
i4 = 4;
}
Object[] objArr = this.mArray;
allocArrays(i4);
int[] iArr2 = this.mHashes;
if (iArr2.length > 0) {
System.arraycopy(iArr, 0, iArr2, 0, iArr.length);
System.arraycopy(objArr, 0, this.mArray, 0, objArr.length);
}
freeArrays(iArr, objArr, this.mSize);
}
int i5 = this.mSize;
if (i2 < i5) {
int[] iArr3 = this.mHashes;
int i6 = i2 + 1;
System.arraycopy(iArr3, i2, iArr3, i6, i5 - i2);
Object[] objArr2 = this.mArray;
System.arraycopy(objArr2, i2, objArr2, i6, this.mSize - i2);
}
this.mHashes[i2] = i;
this.mArray[i2] = e;
this.mSize++;
return true;
}
public void addAll(ArraySet<? extends E> arraySet) {
int i = arraySet.mSize;
ensureCapacity(this.mSize + i);
if (this.mSize != 0) {
for (int i2 = 0; i2 < i; i2++) {
add(arraySet.valueAt(i2));
}
} else if (i > 0) {
System.arraycopy(arraySet.mHashes, 0, this.mHashes, 0, i);
System.arraycopy(arraySet.mArray, 0, this.mArray, 0, i);
this.mSize = i;
}
}
@Override // java.util.Collection, java.util.Set
public boolean remove(Object obj) {
int indexOf = indexOf(obj);
if (indexOf < 0) {
return false;
}
removeAt(indexOf);
return true;
}
public E removeAt(int i) {
Object[] objArr = this.mArray;
E e = (E) objArr[i];
int i2 = this.mSize;
if (i2 <= 1) {
freeArrays(this.mHashes, objArr, i2);
this.mHashes = INT;
this.mArray = OBJECT;
this.mSize = 0;
} else {
int[] iArr = this.mHashes;
if (iArr.length > 8 && i2 < iArr.length / 3) {
allocArrays(i2 > 8 ? i2 + (i2 >> 1) : 8);
this.mSize--;
if (i > 0) {
System.arraycopy(iArr, 0, this.mHashes, 0, i);
System.arraycopy(objArr, 0, this.mArray, 0, i);
}
int i3 = this.mSize;
if (i < i3) {
int i4 = i + 1;
System.arraycopy(iArr, i4, this.mHashes, i, i3 - i);
System.arraycopy(objArr, i4, this.mArray, i, this.mSize - i);
}
} else {
int i5 = i2 - 1;
this.mSize = i5;
if (i < i5) {
int i6 = i + 1;
System.arraycopy(iArr, i6, iArr, i, i5 - i);
Object[] objArr2 = this.mArray;
System.arraycopy(objArr2, i6, objArr2, i, this.mSize - i);
}
this.mArray[this.mSize] = null;
}
}
return e;
}
public boolean removeAll(ArraySet<? extends E> arraySet) {
int i = arraySet.mSize;
int i2 = this.mSize;
for (int i3 = 0; i3 < i; i3++) {
remove(arraySet.valueAt(i3));
}
return i2 != this.mSize;
}
@Override // java.util.Collection, java.util.Set
public Object[] toArray() {
int i = this.mSize;
Object[] objArr = new Object[i];
System.arraycopy(this.mArray, 0, objArr, 0, i);
return objArr;
}
@Override // java.util.Collection, java.util.Set
public <T> T[] toArray(T[] tArr) {
if (tArr.length < this.mSize) {
tArr = (T[]) ((Object[]) Array.newInstance(tArr.getClass().getComponentType(), this.mSize));
}
System.arraycopy(this.mArray, 0, tArr, 0, this.mSize);
int length = tArr.length;
int i = this.mSize;
if (length > i) {
tArr[i] = null;
}
return tArr;
}
@Override // java.util.Collection, java.util.Set
public boolean equals(Object obj) {
if (this == obj) {
return true;
}
if (obj instanceof Set) {
Set set = (Set) obj;
if (size() != set.size()) {
return false;
}
for (int i = 0; i < this.mSize; i++) {
try {
if (!set.contains(valueAt(i))) {
return false;
}
} catch (ClassCastException | NullPointerException unused) {
}
}
return true;
}
return false;
}
@Override // java.util.Collection, java.util.Set
public int hashCode() {
int[] iArr = this.mHashes;
int i = this.mSize;
int i2 = 0;
for (int i3 = 0; i3 < i; i3++) {
i2 += iArr[i3];
}
return i2;
}
public String toString() {
if (isEmpty()) {
return "{}";
}
StringBuilder sb = new StringBuilder(this.mSize * 14);
sb.append('{');
for (int i = 0; i < this.mSize; i++) {
if (i > 0) {
sb.append(", ");
}
E valueAt = valueAt(i);
if (valueAt != this) {
sb.append(valueAt);
} else {
sb.append("(this Set)");
}
}
sb.append('}');
return sb.toString();
}
private MapCollections<E, E> getCollection() {
if (this.mCollections == null) {
this.mCollections = new MapCollections<E, E>() { // from class: androidx.collection.ArraySet.1
@Override // androidx.collection.MapCollections
protected int colGetSize() {
return ArraySet.this.mSize;
}
@Override // androidx.collection.MapCollections
protected Object colGetEntry(int i, int i2) {
return ArraySet.this.mArray[i];
}
@Override // androidx.collection.MapCollections
protected int colIndexOfKey(Object obj) {
return ArraySet.this.indexOf(obj);
}
@Override // androidx.collection.MapCollections
protected int colIndexOfValue(Object obj) {
return ArraySet.this.indexOf(obj);
}
@Override // androidx.collection.MapCollections
protected Map<E, E> colGetMap() {
throw new UnsupportedOperationException("not a map");
}
@Override // androidx.collection.MapCollections
protected void colPut(E e, E e2) {
ArraySet.this.add(e);
}
@Override // androidx.collection.MapCollections
protected E colSetValue(int i, E e) {
throw new UnsupportedOperationException("not a map");
}
@Override // androidx.collection.MapCollections
protected void colRemoveAt(int i) {
ArraySet.this.removeAt(i);
}
@Override // androidx.collection.MapCollections
protected void colClear() {
ArraySet.this.clear();
}
};
}
return this.mCollections;
}
@Override // java.util.Collection, java.lang.Iterable, java.util.Set
public Iterator<E> iterator() {
return getCollection().getKeySet().iterator();
}
@Override // java.util.Collection, java.util.Set
public boolean containsAll(Collection<?> collection) {
Iterator<?> it = collection.iterator();
while (it.hasNext()) {
if (!contains(it.next())) {
return false;
}
}
return true;
}
@Override // java.util.Collection, java.util.Set
public boolean addAll(Collection<? extends E> collection) {
ensureCapacity(this.mSize + collection.size());
Iterator<? extends E> it = collection.iterator();
boolean z = false;
while (it.hasNext()) {
z |= add(it.next());
}
return z;
}
@Override // java.util.Collection, java.util.Set
public boolean removeAll(Collection<?> collection) {
Iterator<?> it = collection.iterator();
boolean z = false;
while (it.hasNext()) {
z |= remove(it.next());
}
return z;
}
@Override // java.util.Collection, java.util.Set
public boolean retainAll(Collection<?> collection) {
boolean z = false;
for (int i = this.mSize - 1; i >= 0; i--) {
if (!collection.contains(this.mArray[i])) {
removeAt(i);
z = true;
}
}
return z;
}
}