bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2012 Google Inc. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
| 6 | */ |
| 7 | |
| 8 | #include "GrMemoryPool.h" |
Herb Derby | b549cc3 | 2017-03-27 13:35:15 -0400 | [diff] [blame] | 9 | #include "SkMalloc.h" |
Brian Salomon | a5002c3 | 2017-03-28 16:51:02 -0400 | [diff] [blame] | 10 | #ifdef SK_DEBUG |
| 11 | #include "SkAtomics.h" |
| 12 | #endif |
Robert Phillips | 7c525e6 | 2018-06-12 10:11:12 -0400 | [diff] [blame] | 13 | #include "ops/GrOp.h" |
Ethan Nicholas | 222e275 | 2018-10-11 11:21:34 -0400 | [diff] [blame^] | 14 | #include "effects/GrSkSLFP.h" |
Herb Derby | d7b34a5 | 2017-03-20 11:19:23 -0400 | [diff] [blame] | 15 | |
commit-bot@chromium.org | 515dcd3 | 2013-08-28 14:17:03 +0000 | [diff] [blame] | 16 | #ifdef SK_DEBUG |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 17 | #define VALIDATE this->validate() |
| 18 | #else |
| 19 | #define VALIDATE |
| 20 | #endif |
| 21 | |
Ethan Nicholas | 222e275 | 2018-10-11 11:21:34 -0400 | [diff] [blame^] | 22 | enum { |
| 23 | // We assume this alignment is good enough for everybody. |
| 24 | kAlignment = alignof(GrSkSLFP), |
| 25 | kHeaderSize = GR_CT_ALIGN_UP(GrMemoryPool::kBlockHeaderSize, kAlignment), |
| 26 | kPerAllocPad = GR_CT_ALIGN_UP(GrMemoryPool::kAllocHeaderSize, kAlignment), |
| 27 | }; |
| 28 | |
Robert Phillips | 7c525e6 | 2018-06-12 10:11:12 -0400 | [diff] [blame] | 29 | void GrOpMemoryPool::release(std::unique_ptr<GrOp> op) { |
| 30 | GrOp* tmp = op.release(); |
| 31 | SkASSERT(tmp); |
| 32 | tmp->~GrOp(); |
| 33 | fMemoryPool.release(tmp); |
| 34 | } |
| 35 | |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 36 | constexpr size_t GrMemoryPool::kSmallestMinAllocSize; |
| 37 | |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 38 | GrMemoryPool::GrMemoryPool(size_t preallocSize, size_t minAllocSize) { |
commit-bot@chromium.org | 1acc3d7 | 2013-09-06 23:13:05 +0000 | [diff] [blame] | 39 | SkDEBUGCODE(fAllocationCnt = 0); |
joshualitt | b7133be | 2015-04-08 09:08:31 -0700 | [diff] [blame] | 40 | SkDEBUGCODE(fAllocBlockCnt = 0); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 41 | |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 42 | minAllocSize = SkTMax<size_t>(GrSizeAlignUp(minAllocSize, kAlignment), kSmallestMinAllocSize); |
| 43 | preallocSize = SkTMax<size_t>(GrSizeAlignUp(preallocSize, kAlignment), minAllocSize); |
| 44 | |
| 45 | fMinAllocSize = minAllocSize; |
joshualitt | 22c6f5c | 2016-01-05 08:07:18 -0800 | [diff] [blame] | 46 | fSize = 0; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 47 | |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 48 | fHead = CreateBlock(preallocSize); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 49 | fTail = fHead; |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 50 | fHead->fNext = nullptr; |
| 51 | fHead->fPrev = nullptr; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 52 | VALIDATE; |
| 53 | }; |
| 54 | |
| 55 | GrMemoryPool::~GrMemoryPool() { |
| 56 | VALIDATE; |
Brian Salomon | a5002c3 | 2017-03-28 16:51:02 -0400 | [diff] [blame] | 57 | #ifdef SK_DEBUG |
| 58 | int i = 0; |
| 59 | int n = fAllocatedIDs.count(); |
| 60 | fAllocatedIDs.foreach([&i, n] (int32_t id) { |
| 61 | if (++i == 1) { |
| 62 | SkDebugf("Leaked IDs (in no particular order): %d", id); |
| 63 | } else if (i < 11) { |
| 64 | SkDebugf(", %d%s", id, (n == i ? "\n" : "")); |
| 65 | } else if (i == 11) { |
| 66 | SkDebugf(", ...\n"); |
| 67 | } |
| 68 | }); |
| 69 | #endif |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 70 | SkASSERT(0 == fAllocationCnt); |
| 71 | SkASSERT(fHead == fTail); |
| 72 | SkASSERT(0 == fHead->fLiveCount); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 73 | DeleteBlock(fHead); |
| 74 | }; |
| 75 | |
| 76 | void* GrMemoryPool::allocate(size_t size) { |
| 77 | VALIDATE; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 78 | size += kPerAllocPad; |
robertphillips | 19c6250 | 2016-01-06 07:04:46 -0800 | [diff] [blame] | 79 | size = GrSizeAlignUp(size, kAlignment); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 80 | if (fTail->fFreeSize < size) { |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 81 | size_t blockSize = size + kHeaderSize; |
commit-bot@chromium.org | 972f9cd | 2014-03-28 17:58:28 +0000 | [diff] [blame] | 82 | blockSize = SkTMax<size_t>(blockSize, fMinAllocSize); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 83 | BlockHeader* block = CreateBlock(blockSize); |
| 84 | |
| 85 | block->fPrev = fTail; |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 86 | block->fNext = nullptr; |
| 87 | SkASSERT(nullptr == fTail->fNext); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 88 | fTail->fNext = block; |
| 89 | fTail = block; |
joshualitt | b7133be | 2015-04-08 09:08:31 -0700 | [diff] [blame] | 90 | fSize += block->fSize; |
| 91 | SkDEBUGCODE(++fAllocBlockCnt); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 92 | } |
robertphillips | b7f4b8e | 2016-01-07 10:12:16 -0800 | [diff] [blame] | 93 | SkASSERT(kAssignedMarker == fTail->fBlockSentinal); |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 94 | SkASSERT(fTail->fFreeSize >= size); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 95 | intptr_t ptr = fTail->fCurrPtr; |
| 96 | // We stash a pointer to the block header, just before the allocated space, |
| 97 | // so that we can decrement the live count on delete in constant time. |
robertphillips | 19c6250 | 2016-01-06 07:04:46 -0800 | [diff] [blame] | 98 | AllocHeader* allocData = reinterpret_cast<AllocHeader*>(ptr); |
| 99 | SkDEBUGCODE(allocData->fSentinal = kAssignedMarker); |
Brian Salomon | a5002c3 | 2017-03-28 16:51:02 -0400 | [diff] [blame] | 100 | SkDEBUGCODE(allocData->fID = []{static int32_t gID; return sk_atomic_inc(&gID) + 1;}()); |
| 101 | // You can set a breakpoint here when a leaked ID is allocated to see the stack frame. |
| 102 | SkDEBUGCODE(fAllocatedIDs.add(allocData->fID)); |
robertphillips | 19c6250 | 2016-01-06 07:04:46 -0800 | [diff] [blame] | 103 | allocData->fHeader = fTail; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 104 | ptr += kPerAllocPad; |
tomhudson@google.com | dcba4c2 | 2012-07-24 21:36:16 +0000 | [diff] [blame] | 105 | fTail->fPrevPtr = fTail->fCurrPtr; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 106 | fTail->fCurrPtr += size; |
| 107 | fTail->fFreeSize -= size; |
| 108 | fTail->fLiveCount += 1; |
commit-bot@chromium.org | 1acc3d7 | 2013-09-06 23:13:05 +0000 | [diff] [blame] | 109 | SkDEBUGCODE(++fAllocationCnt); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 110 | VALIDATE; |
| 111 | return reinterpret_cast<void*>(ptr); |
| 112 | } |
| 113 | |
| 114 | void GrMemoryPool::release(void* p) { |
| 115 | VALIDATE; |
| 116 | intptr_t ptr = reinterpret_cast<intptr_t>(p) - kPerAllocPad; |
robertphillips | 19c6250 | 2016-01-06 07:04:46 -0800 | [diff] [blame] | 117 | AllocHeader* allocData = reinterpret_cast<AllocHeader*>(ptr); |
| 118 | SkASSERT(kAssignedMarker == allocData->fSentinal); |
| 119 | SkDEBUGCODE(allocData->fSentinal = kFreedMarker); |
Brian Salomon | a5002c3 | 2017-03-28 16:51:02 -0400 | [diff] [blame] | 120 | SkDEBUGCODE(fAllocatedIDs.remove(allocData->fID)); |
robertphillips | 19c6250 | 2016-01-06 07:04:46 -0800 | [diff] [blame] | 121 | BlockHeader* block = allocData->fHeader; |
robertphillips | b7f4b8e | 2016-01-07 10:12:16 -0800 | [diff] [blame] | 122 | SkASSERT(kAssignedMarker == block->fBlockSentinal); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 123 | if (1 == block->fLiveCount) { |
| 124 | // the head block is special, it is reset rather than deleted |
| 125 | if (fHead == block) { |
joshualitt | b7133be | 2015-04-08 09:08:31 -0700 | [diff] [blame] | 126 | fHead->fCurrPtr = reinterpret_cast<intptr_t>(fHead) + kHeaderSize; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 127 | fHead->fLiveCount = 0; |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 128 | fHead->fFreeSize = fHead->fSize - kHeaderSize; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 129 | } else { |
| 130 | BlockHeader* prev = block->fPrev; |
| 131 | BlockHeader* next = block->fNext; |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 132 | SkASSERT(prev); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 133 | prev->fNext = next; |
| 134 | if (next) { |
| 135 | next->fPrev = prev; |
| 136 | } else { |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 137 | SkASSERT(fTail == block); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 138 | fTail = prev; |
| 139 | } |
joshualitt | b7133be | 2015-04-08 09:08:31 -0700 | [diff] [blame] | 140 | fSize -= block->fSize; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 141 | DeleteBlock(block); |
joshualitt | b7133be | 2015-04-08 09:08:31 -0700 | [diff] [blame] | 142 | SkDEBUGCODE(fAllocBlockCnt--); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 143 | } |
| 144 | } else { |
| 145 | --block->fLiveCount; |
tomhudson@google.com | dcba4c2 | 2012-07-24 21:36:16 +0000 | [diff] [blame] | 146 | // Trivial reclaim: if we're releasing the most recent allocation, reuse it |
| 147 | if (block->fPrevPtr == ptr) { |
| 148 | block->fFreeSize += (block->fCurrPtr - block->fPrevPtr); |
| 149 | block->fCurrPtr = block->fPrevPtr; |
| 150 | } |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 151 | } |
commit-bot@chromium.org | 1acc3d7 | 2013-09-06 23:13:05 +0000 | [diff] [blame] | 152 | SkDEBUGCODE(--fAllocationCnt); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 153 | VALIDATE; |
| 154 | } |
| 155 | |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 156 | GrMemoryPool::BlockHeader* GrMemoryPool::CreateBlock(size_t blockSize) { |
| 157 | blockSize = SkTMax<size_t>(blockSize, kHeaderSize); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 158 | BlockHeader* block = |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 159 | reinterpret_cast<BlockHeader*>(sk_malloc_throw(blockSize)); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 160 | // we assume malloc gives us aligned memory |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 161 | SkASSERT(!(reinterpret_cast<intptr_t>(block) % kAlignment)); |
robertphillips | b7f4b8e | 2016-01-07 10:12:16 -0800 | [diff] [blame] | 162 | SkDEBUGCODE(block->fBlockSentinal = kAssignedMarker); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 163 | block->fLiveCount = 0; |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 164 | block->fFreeSize = blockSize - kHeaderSize; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 165 | block->fCurrPtr = reinterpret_cast<intptr_t>(block) + kHeaderSize; |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 166 | block->fPrevPtr = 0; // gcc warns on assigning nullptr to an intptr_t. |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 167 | block->fSize = blockSize; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 168 | return block; |
| 169 | } |
| 170 | |
| 171 | void GrMemoryPool::DeleteBlock(BlockHeader* block) { |
robertphillips | b7f4b8e | 2016-01-07 10:12:16 -0800 | [diff] [blame] | 172 | SkASSERT(kAssignedMarker == block->fBlockSentinal); |
| 173 | SkDEBUGCODE(block->fBlockSentinal = kFreedMarker); // FWIW |
reed@google.com | 939ca7c | 2013-09-26 19:56:51 +0000 | [diff] [blame] | 174 | sk_free(block); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 175 | } |
| 176 | |
| 177 | void GrMemoryPool::validate() { |
humper@google.com | 0e51577 | 2013-01-07 19:54:40 +0000 | [diff] [blame] | 178 | #ifdef SK_DEBUG |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 179 | BlockHeader* block = fHead; |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 180 | BlockHeader* prev = nullptr; |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 181 | SkASSERT(block); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 182 | int allocCount = 0; |
| 183 | do { |
robertphillips | b7f4b8e | 2016-01-07 10:12:16 -0800 | [diff] [blame] | 184 | SkASSERT(kAssignedMarker == block->fBlockSentinal); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 185 | allocCount += block->fLiveCount; |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 186 | SkASSERT(prev == block->fPrev); |
bsalomon | 49f085d | 2014-09-05 13:34:00 -0700 | [diff] [blame] | 187 | if (prev) { |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 188 | SkASSERT(prev->fNext == block); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 189 | } |
| 190 | |
| 191 | intptr_t b = reinterpret_cast<intptr_t>(block); |
| 192 | size_t ptrOffset = block->fCurrPtr - b; |
| 193 | size_t totalSize = ptrOffset + block->fFreeSize; |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 194 | intptr_t userStart = b + kHeaderSize; |
| 195 | |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 196 | SkASSERT(!(b % kAlignment)); |
| 197 | SkASSERT(!(totalSize % kAlignment)); |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 198 | SkASSERT(!(block->fCurrPtr % kAlignment)); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 199 | if (fHead != block) { |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 200 | SkASSERT(block->fLiveCount); |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 201 | SkASSERT(totalSize >= fMinAllocSize); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 202 | } else { |
dskiba | e4cd006 | 2016-11-29 06:50:35 -0800 | [diff] [blame] | 203 | SkASSERT(totalSize == block->fSize); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 204 | } |
| 205 | if (!block->fLiveCount) { |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 206 | SkASSERT(ptrOffset == kHeaderSize); |
| 207 | SkASSERT(userStart == block->fCurrPtr); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 208 | } else { |
robertphillips | 19c6250 | 2016-01-06 07:04:46 -0800 | [diff] [blame] | 209 | AllocHeader* allocData = reinterpret_cast<AllocHeader*>(userStart); |
halcanary | 9d524f2 | 2016-03-29 09:03:52 -0700 | [diff] [blame] | 210 | SkASSERT(allocData->fSentinal == kAssignedMarker || |
robertphillips | 19c6250 | 2016-01-06 07:04:46 -0800 | [diff] [blame] | 211 | allocData->fSentinal == kFreedMarker); |
| 212 | SkASSERT(block == allocData->fHeader); |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 213 | } |
robertphillips | 19c6250 | 2016-01-06 07:04:46 -0800 | [diff] [blame] | 214 | |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 215 | prev = block; |
| 216 | } while ((block = block->fNext)); |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 217 | SkASSERT(allocCount == fAllocationCnt); |
Brian Salomon | a5002c3 | 2017-03-28 16:51:02 -0400 | [diff] [blame] | 218 | SkASSERT(fAllocationCnt == fAllocatedIDs.count()); |
tfarina@chromium.org | f6de475 | 2013-08-17 00:02:59 +0000 | [diff] [blame] | 219 | SkASSERT(prev == fTail); |
joshualitt | 22c6f5c | 2016-01-05 08:07:18 -0800 | [diff] [blame] | 220 | SkASSERT(fAllocBlockCnt != 0 || fSize == 0); |
humper@google.com | 0e51577 | 2013-01-07 19:54:40 +0000 | [diff] [blame] | 221 | #endif |
bsalomon@google.com | 4da34e3 | 2012-06-19 15:40:27 +0000 | [diff] [blame] | 222 | } |