1
2 /*
3 * Copyright 2011 Google Inc.
4 *
5 * Use of this source code is governed by a BSD-style license that can be
6 * found in the LICENSE file.
7 */
8 #include "SkPtrRecorder.h"
9 #include "SkTSearch.h"
10
reset()11 void SkPtrSet::reset() {
12 Pair* p = fList.begin();
13 Pair* stop = fList.end();
14 while (p < stop) {
15 this->decPtr(p->fPtr);
16 p += 1;
17 }
18 fList.reset();
19 }
20
Less(const Pair & a,const Pair & b)21 bool SkPtrSet::Less(const Pair& a, const Pair& b) {
22 return (char*)a.fPtr < (char*)b.fPtr;
23 }
24
find(void * ptr) const25 uint32_t SkPtrSet::find(void* ptr) const {
26 if (NULL == ptr) {
27 return 0;
28 }
29
30 int count = fList.count();
31 Pair pair;
32 pair.fPtr = ptr;
33
34 int index = SkTSearch<Pair, Less>(fList.begin(), count, pair, sizeof(pair));
35 if (index < 0) {
36 return 0;
37 }
38 return fList[index].fIndex;
39 }
40
add(void * ptr)41 uint32_t SkPtrSet::add(void* ptr) {
42 if (NULL == ptr) {
43 return 0;
44 }
45
46 int count = fList.count();
47 Pair pair;
48 pair.fPtr = ptr;
49
50 int index = SkTSearch<Pair, Less>(fList.begin(), count, pair, sizeof(pair));
51 if (index < 0) {
52 index = ~index; // turn it back into an index for insertion
53 this->incPtr(ptr);
54 pair.fIndex = count + 1;
55 *fList.insert(index) = pair;
56 return count + 1;
57 } else {
58 return fList[index].fIndex;
59 }
60 }
61
copyToArray(void * array[]) const62 void SkPtrSet::copyToArray(void* array[]) const {
63 int count = fList.count();
64 if (count > 0) {
65 SkASSERT(array);
66 const Pair* p = fList.begin();
67 // p->fIndex is base-1, so we need to subtract to find its slot
68 for (int i = 0; i < count; i++) {
69 int index = p[i].fIndex - 1;
70 SkASSERT((unsigned)index < (unsigned)count);
71 array[index] = p[i].fPtr;
72 }
73 }
74 }
75