mtklein | 8113dd1 | 2014-11-12 15:15:28 -0800 | [diff] [blame] | 1 | #ifndef SkVarAlloc_DEFINED |
| 2 | #define SkVarAlloc_DEFINED |
| 3 | |
| 4 | #include "SkTypes.h" |
| 5 | |
| 6 | class SkVarAlloc : SkNoncopyable { |
| 7 | public: |
mtklein | f2950b1 | 2014-11-13 12:41:14 -0800 | [diff] [blame] | 8 | SkVarAlloc(); |
mtklein | 8113dd1 | 2014-11-12 15:15:28 -0800 | [diff] [blame] | 9 | ~SkVarAlloc(); |
| 10 | |
| 11 | // Returns contiguous bytes aligned at least for pointers. You may pass SK_MALLOC_THROW, etc. |
| 12 | char* alloc(size_t bytes, unsigned sk_malloc_flags) { |
| 13 | bytes = SkAlignPtr(bytes); |
| 14 | |
mtklein | f2950b1 | 2014-11-13 12:41:14 -0800 | [diff] [blame] | 15 | if (bytes > fRemaining) { |
mtklein | 8113dd1 | 2014-11-12 15:15:28 -0800 | [diff] [blame] | 16 | this->makeSpace(bytes, sk_malloc_flags); |
| 17 | } |
mtklein | f2950b1 | 2014-11-13 12:41:14 -0800 | [diff] [blame] | 18 | SkASSERT(bytes <= fRemaining); |
mtklein | 8113dd1 | 2014-11-12 15:15:28 -0800 | [diff] [blame] | 19 | |
| 20 | char* ptr = fByte; |
| 21 | fByte += bytes; |
mtklein | f2950b1 | 2014-11-13 12:41:14 -0800 | [diff] [blame] | 22 | fRemaining -= bytes; |
mtklein | 8113dd1 | 2014-11-12 15:15:28 -0800 | [diff] [blame] | 23 | return ptr; |
| 24 | } |
| 25 | |
mtklein | 0bd57b2 | 2014-11-18 09:32:36 -0800 | [diff] [blame^] | 26 | // Returns our best estimate of the number of bytes we've allocated. |
| 27 | // (We intentionally do not track this precisely to save space.) |
| 28 | size_t approxBytesAllocated() const; |
| 29 | |
mtklein | 8113dd1 | 2014-11-12 15:15:28 -0800 | [diff] [blame] | 30 | private: |
| 31 | void makeSpace(size_t bytes, unsigned flags); |
| 32 | |
| 33 | char* fByte; |
mtklein | f2950b1 | 2014-11-13 12:41:14 -0800 | [diff] [blame] | 34 | unsigned fRemaining; |
mtklein | 975ae5e | 2014-11-13 13:55:22 -0800 | [diff] [blame] | 35 | unsigned fLgSize; // This is always in the range [4, 16], so it really only needs 4 bits. |
mtklein | 8113dd1 | 2014-11-12 15:15:28 -0800 | [diff] [blame] | 36 | |
| 37 | struct Block; |
| 38 | Block* fBlock; |
| 39 | }; |
| 40 | |
| 41 | #endif//SkVarAlloc_DEFINED |