Android源码分析之SparseArray
本来接下来应该分析MessageQueue了,可是我这几天正好在实际开发中又再次用到了SparseArray(之前有用到过一次,那次只是
大概浏览了下源码,没做深入研究),于是在兴趣的推动下,花了些时间深入研究了下,趁着记忆还是新鲜的,就先在这里分析了。
MessageQueue的分析应该会在本周末给出。
和以往一样,首先我们来看看关键字段和ctor:
private static final Object DELETED = new Object(); private boolean mGarbage = false; private int[] mKeys; private Object[] mValues; private int mSize; /** * Creates a new SparseArray containing no mappings. */ public SparseArray() { this(10); } /** * Creates a new SparseArray containing no mappings that will not * require any additional memory allocation to store the specified * number of mappings. If you supply an initial capacity of 0, the * sparse array will be initialized with a light-weight representation * not requiring any additional array allocations. */ public SparseArray(int initialCapacity) { if (initialCapacity == 0) { mKeys = ContainerHelpers.EMPTY_INTS; mValues = ContainerHelpers.EMPTY_OBJECTS; } else { initialCapacity = ArrayUtils.idealIntArraySize(initialCapacity); mKeys = new int[initialCapacity]; mValues = new Object[initialCapacity]; } mSize = 0; }
常量DELETED对象用来标记删除,如果某个位置的值是DELETED则表示这个位置没对应的元素(被删除了);
mGarbage在删除操作中会被设置(置为true),表示接下来需要清理压缩了(compacting);
mkeys,mValues分别表示SparseArray的内部存储,即分别是key、value的存储数组;
mSize表示有效的key-value对的数目;
接下来来看ctor,无参版本会调用带参数的版本并且传递10,表示初始容量。我们可以看到如果initialCapacity=0的话,mkeys、mValues
会分别被初始化为EMPTY的东西,实际是长度为0的数组,不会产生内存分配;否则初始化2个数组为具体的大小,这里要留意一点就是代码
里并没有直接用你传递进来的值,而是调用了initialCapacity = ArrayUtils.idealIntArraySize(initialCapacity);来我们顺便看一下:
public static int idealIntArraySize(int need) { return idealByteArraySize(need * 4) / 4; } public static int idealByteArraySize(int need) { for (int i = 4; i < 32; i++) if (need <= (1 << i) - 12) return (1 << i) - 12; return need; }
Android认为它这个方式是比较ideal的,比客户端直接传的值要好,所以经过这一堆运算后,initialCapacity很可能已经不是你当初传的值了。
最后都设置了mSize为0。
接下来看2个依据key来得到value的方法:
/** * Gets the Object mapped from the specified key, or <code>null</code> * if no such mapping has been made. */ public E get(int key) { return get(key, null); } /** * Gets the Object mapped from the specified key, or the specified Object * if no such mapping has been made. */ @SuppressWarnings("unchecked") public E get(int key, E valueIfKeyNotFound) { int i = ContainerHelpers.binarySearch(mKeys, mSize, key); if (i < 0 || mValues[i] == DELETED) { return valueIfKeyNotFound; } else { return (E) mValues[i]; } }
看2个参数的get方法,你可以提供一个fallback的值,表示如果没找到key对应的值就返回你提供的这个值;SparseArray内部是通过二分
查找算法来search key的,顺便看看:
// This is Arrays.binarySearch(), but doesn't do any argument validation. static int binarySearch(int[] array, int size, int value) { int lo = 0; int hi = size - 1; while (lo <= hi) { final int mid = (lo + hi) >>> 1; final int midVal = array[mid]; if (midVal < value) { lo = mid + 1; } else if (midVal > value) { hi = mid - 1; } else { return mid; // value found } } return ~lo; // value not present }
如源码所说,这个版本和java.util.Arrays.java里的实现一样,只是省略了参数检查。二分查找大家大学都接触过,应该印象都比较深刻,
这里只说一点即最后没找到时的返回值~lo。如方法的doc所说,没找到的情况下会返回一个负值,那到底返回哪个负值呢,-1行不?其实
这里的~lo(取反)就相当于-(lo+1)(参看Arrays.binarySearch的实现)。为什么要这样做,因为我们不仅想表示没找到,还想返回
更多信息,即这个key如果要插进来应该在的位置(外面的代码只需要再次~即取反就可以得到这个信息)。接下来回到刚才的get方法,
明白了这里使用的二分查找这个方法就非常简单明了了。get内部通过binarySearch的返回值来做判断,如果是负的或i>=0但是位置i已经
被标记为删除了则返回valueIfKeyNotFound,否则直接返回(E) mValues[i]。
接下来看一组delete/remove方法:
/** * Removes the mapping from the specified key, if there was any. */ public void delete(int key) { int i = ContainerHelpers.binarySearch(mKeys, mSize, key); if (i >= 0) { if (mValues[i] != DELETED) { mValues[i] = DELETED; mGarbage = true; } } } /** * Alias for {@link #delete(int)}. */ public void remove(int key) { delete(key); } /** * Removes the mapping at the specified index. */ public void removeAt(int index) { if (mValues[index] != DELETED) { mValues[index] = DELETED; mGarbage = true; } } /** * Remove a range of mappings as a batch. * * @param index Index to begin at * @param size Number of mappings to remove */ public void removeAtRange(int index, int size) { final int end = Math.min(mSize, index + size); for (int i = index; i < end; i++) { removeAt(i); } }
delete(key)和remove(key)一样,都是先通过二分查找来找这个key,如果不存在则do nothing否则如果这个位置还没被标记为删除
则标记之,顺便设置mGarbage(之前提到过删除之类的操作都会设置这个值,表示接下来需要执行清理压缩操作了)。注意这里数组
的不同处理,没有直接移动数组元素(压缩处理)来删除这个key,而仅仅是标记操作。这2个方法都是通过key来进行操作,有时你可能
想基于index来执行某些操作,下面的removeAt(index)和removeAtRange(index, size)就是这样的方法,处理也都是如果位置i没标记
删除则标记之,并设置mGarbage的值。只是在removeAtRange中对上限做了保险处理即end = Math.min(mSize, index+size);
防止数组越界。
接下来该打起来精神,睁大眼睛了,让我们一起来看看SparseArray的关键,我叫它清理压缩,上代码:
private void gc() { // Log.e("SparseArray", "gc start with " + mSize); int n = mSize; int reusedIndex = 0; int[] keys = mKeys; Object[] values = mValues; for (int i = 0; i < n; i++) { Object val = values[i]; if (val != DELETED) { if (i != reusedIndex) { keys[reusedIndex] = keys[i]; values[reusedIndex] = val; values[i] = null; } reusedIndex++; } } mGarbage = false; mSize = reusedIndex; // Log.e("SparseArray", "gc end with " + mSize); }
gc方法只有在mGarbage设置的时候才有可能调用,其总体思想是把靠后的有效元素(即没被删除的)往前(reusedIndex的位置)提,
从而达到清理压缩的目的。我们仔细分析下,首先初始化一些接下来要用到的值,这里特别留意下reusedIndex(源码中叫o,实在不好
理解,我按自己的理解重新命名了),初始化为0,指向第一个元素。接下来是遍历mValues数组的for循环,其内部是如果当前元素val是
DELETED则啥也不做,接着看下一个元素(注意此时reusedIndex不做任何改动);否则当前是个有效的元素,执行1,2:
1. reusedIndex不是当前的index则应该把当前元素拷贝到reusedIndex的位置(key和value都复制),当前位置的value清空(置null);
2. reusedIndex往前+1(每遇到一个有效元素),准备下次循环。
结束遍历之后reset mGarbage(因为刚刚做过了,暂时不需要了),更新mSize为reusedIndex(因为每遇到一个有效元素,reusedIndex
就+1,所以它的值就是新的mSize)。
看完了删除我们来看put相关的方法,代码如下:
/** * Adds a mapping from the specified key to the specified value, * replacing the previous mapping from the specified key if there * was one. */ public void put(int key, E value) { int i = ContainerHelpers.binarySearch(mKeys, mSize, key); if (i >= 0) { mValues[i] = value; } else { i = ~i; if (i < mSize && mValues[i] == DELETED) { mKeys[i] = key; mValues[i] = value; return; } if (mGarbage && mSize >= mKeys.length) { gc(); // Search again because indices may have changed. i = ~ContainerHelpers.binarySearch(mKeys, mSize, key); } if (mSize >= mKeys.length) { int n = ArrayUtils.idealIntArraySize(mSize + 1); int[] nkeys = new int[n]; Object[] nvalues = new Object[n]; // Log.e("SparseArray", "grow " + mKeys.length + " to " + n); System.arraycopy(mKeys, 0, nkeys, 0, mKeys.length); System.arraycopy(mValues, 0, nvalues, 0, mValues.length); mKeys = nkeys; mValues = nvalues; } if (mSize - i != 0) { // Log.e("SparseArray", "move " + (mSize - i)); System.arraycopy(mKeys, i, mKeys, i + 1, mSize - i); System.arraycopy(mValues, i, mValues, i + 1, mSize - i); } mKeys[i] = key; mValues[i] = value; mSize++; } }
put的实现一般都是有这个key则覆盖原先的value,否则新插入一个。这里的也一样,首先先调用二分查找算法去找key,如果找到了
(i>=0)则直接更新mValues[i]为新值;否则就得想办法插入这对key-value,具体做法是:
1. 首先把二分查找的返回值取反(~),拿到要插入的位置信息;
2. 接下来看如果这个位置i在有效范围内(即不需要额外分配空间)且此位置被标记为删除了,则这就是个可以重用的位置,也就是我们
要找的插入位置,插入之,完事(return);
3. 不然的话(也就是说i超出了有效范围或者没超出但位置i是有效元素),如果mGarbage被设置了且mSize >= mKeys.length,
表示该执行gc算法了,执行之,接着重新利用二分查找算法确定下key的新位置(因为index可能变了)。
接下来,如果mSize >= mKeys.length(即key数组不够用了或者说到了要分配更多内存的临界点了),利用mSize计算新的capacity,
int n = ArrayUtils.idealIntArraySize(mSize + 1); 分配新的数组,将旧内容拷过去,接着将mKeys, mValues指向新分配
的数组。然后看下i如果在有效范围内,则应该把现在从位置i开始的元素移动到位置i+1处(把i的位置腾出来给新put的元素让位),
以此类推,到mSize-1的位置为止(mSize-1位置的元素移动到mSize的位置)。最后将put的元素插在第i的位置上,mSize++,
最终结束整个put操作。
长舒一口气啊,接着来看看几个非常简单的方法,
/** * Returns the number of key-value mappings that this SparseArray * currently stores. */ public int size() { if (mGarbage) { gc(); } return mSize; } /** * Given an index in the range <code>0...size()-1</code>, returns * the key from the <code>index</code>th key-value mapping that this * SparseArray stores. * * <p>The keys corresponding to indices in ascending order are guaranteed to * be in ascending order, e.g., <code>keyAt(0)</code> will return the * smallest key and <code>keyAt(size()-1)</code> will return the largest * key.</p> */ public int keyAt(int index) { if (mGarbage) { gc(); } return mKeys[index]; } /** * Given an index in the range <code>0...size()-1</code>, returns * the value from the <code>index</code>th key-value mapping that this * SparseArray stores. * * <p>The values corresponding to indices in ascending order are guaranteed * to be associated with keys in ascending order, e.g., * <code>valueAt(0)</code> will return the value associated with the * smallest key and <code>valueAt(size()-1)</code> will return the value * associated with the largest key.</p> */ @SuppressWarnings("unchecked") public E valueAt(int index) { if (mGarbage) { gc(); } return (E) mValues[index]; } /** * Given an index in the range <code>0...size()-1</code>, sets a new * value for the <code>index</code>th key-value mapping that this * SparseArray stores. */ public void setValueAt(int index, E value) { if (mGarbage) { gc(); } mValues[index] = value; } /** * Returns the index for which {@link #keyAt} would return the * specified key, or a negative number if the specified * key is not mapped. */ public int indexOfKey(int key) { if (mGarbage) { gc(); } return ContainerHelpers.binarySearch(mKeys, mSize, key); } /** * Returns an index for which {@link #valueAt} would return the * specified key, or a negative number if no keys map to the * specified value. * <p>Beware that this is a linear search, unlike lookups by key, * and that multiple keys can map to the same value and this will * find only one of them. * <p>Note also that unlike most collections' {@code indexOf} methods, * this method compares values using {@code ==} rather than {@code equals}. */ public int indexOfValue(E value) { if (mGarbage) { gc(); } for (int i = 0; i < mSize; i++) if (mValues[i] == value) return i; return -1; } /** * Removes all key-value mappings from this SparseArray. */ public void clear() { int n = mSize; Object[] values = mValues; for (int i = 0; i < n; i++) { values[i] = null; } mSize = 0; mGarbage = false; }
size()方法返回当前有效的key-value对的数目,值得注意的是在其内部都会有条件的执行gc方法,因为之前的操作可能导致mGarbage
被设置了,所以必须执行下清理压缩才能返回最新的值;keyAt(index)和valueAt(index)提供了遍历SparseArray的方法,如下:
for (int i = 0; i < sparseArray.size(); i++) { System.out.println("key = " + sparseArray.keyAt(i) + ", value = " + sparseArray.valueAt(i)); }
同样和size()方法一样其内部也都会有条件的执行gc方法,注意你传递的index必须在[0, size()-1]之间,否则会数组越界。
setValueAt让你像使用数组一样设置某个位置处的值,同样会有条件的执行gc方法。
indexOfKey通过二分查找来search key的位置;indexOfValue在整个mValues数组中遍历查找value,有则返回对应的index
否则返回-1。
clear()方法清空了mValues数组,重置了mSize,mGarbage,但请注意并没动mKeys数组;
最后我们看下SparseArray的最后一个方法append,源码如下:
/** * Puts a key/value pair into the array, optimizing for the case where * the key is greater than all existing keys in the array. */ public void append(int key, E value) { if (mSize != 0 && key <= mKeys[mSize - 1]) { put(key, value); return; } if (mGarbage && mSize >= mKeys.length) { gc(); } int pos = mSize; if (pos >= mKeys.length) { int n = ArrayUtils.idealIntArraySize(pos + 1); int[] nkeys = new int[n]; Object[] nvalues = new Object[n]; // Log.e("SparseArray", "grow " + mKeys.length + " to " + n); System.arraycopy(mKeys, 0, nkeys, 0, mKeys.length); System.arraycopy(mValues, 0, nvalues, 0, mValues.length); mKeys = nkeys; mValues = nvalues; } mKeys[pos] = key; mValues[pos] = value; mSize = pos + 1; }
看源码我们知道,当mSize != 0且key<=SparseArray中最大的key时,则直接调用put方法;否则当key比现存所有的key都大,
这种情况下我们执行的只是put方法中i==mSize的部分(小分支,所以说是个优化相比直接调用put方法)。
目前为止,SparseArray类的所有关键代码都已经分析完毕,希望对各位平时的开发有所帮助(由于本人水平有限,欢迎批评指正)。