blob: 4f6788c928b13427943e33b6cbe0c848b7721ad9 [file] [log] [blame]
epoger@google.comec3ed6a2011-07-28 14:26:00 +00001
reed@android.com8a1c16f2008-12-17 15:59:43 +00002/*
epoger@google.comec3ed6a2011-07-28 14:26:00 +00003 * Copyright 2006 The Android Open Source Project
reed@android.com8a1c16f2008-12-17 15:59:43 +00004 *
epoger@google.comec3ed6a2011-07-28 14:26:00 +00005 * Use of this source code is governed by a BSD-style license that can be
6 * found in the LICENSE file.
reed@android.com8a1c16f2008-12-17 15:59:43 +00007 */
8
epoger@google.comec3ed6a2011-07-28 14:26:00 +00009
reed@android.com8a1c16f2008-12-17 15:59:43 +000010#ifndef SkTypes_DEFINED
11#define SkTypes_DEFINED
12
13#include "SkPreConfig.h"
14#include "SkUserConfig.h"
15#include "SkPostConfig.h"
16
17#ifndef SK_IGNORE_STDINT_DOT_H
18 #include <stdint.h>
19#endif
20
21#include <stdio.h>
22
23/** \file SkTypes.h
24*/
25
reed@android.com9aa8b322010-04-13 13:22:54 +000026/** See SkGraphics::GetVersion() to retrieve these at runtime
27 */
28#define SKIA_VERSION_MAJOR 1
29#define SKIA_VERSION_MINOR 0
30#define SKIA_VERSION_PATCH 0
31
reed@android.com8a1c16f2008-12-17 15:59:43 +000032/*
33 memory wrappers to be implemented by the porting layer (platform)
34*/
35
36/** Called internally if we run out of memory. The platform implementation must
37 not return, but should either throw an exception or otherwise exit.
38*/
reed@google.comde916c82011-10-19 19:50:48 +000039SK_API extern void sk_out_of_memory(void);
reed@android.com8a1c16f2008-12-17 15:59:43 +000040/** Called internally if we hit an unrecoverable error.
41 The platform implementation must not return, but should either throw
42 an exception or otherwise exit.
43*/
reed@google.comde916c82011-10-19 19:50:48 +000044SK_API extern void sk_throw(void);
reed@android.com8a1c16f2008-12-17 15:59:43 +000045
46enum {
47 SK_MALLOC_TEMP = 0x01, //!< hint to sk_malloc that the requested memory will be freed in the scope of the stack frame
48 SK_MALLOC_THROW = 0x02 //!< instructs sk_malloc to call sk_throw if the memory cannot be allocated.
49};
50/** Return a block of memory (at least 4-byte aligned) of at least the
51 specified size. If the requested memory cannot be returned, either
52 return null (if SK_MALLOC_TEMP bit is clear) or call sk_throw()
53 (if SK_MALLOC_TEMP bit is set). To free the memory, call sk_free().
54*/
ctguil@chromium.org7ffb1b22011-03-15 21:27:08 +000055SK_API extern void* sk_malloc_flags(size_t size, unsigned flags);
reed@android.com8a1c16f2008-12-17 15:59:43 +000056/** Same as sk_malloc(), but hard coded to pass SK_MALLOC_THROW as the flag
57*/
reed@google.comde916c82011-10-19 19:50:48 +000058SK_API extern void* sk_malloc_throw(size_t size);
reed@android.com8a1c16f2008-12-17 15:59:43 +000059/** Same as standard realloc(), but this one never returns null on failure. It will throw
60 an exception if it fails.
61*/
reed@google.comde916c82011-10-19 19:50:48 +000062SK_API extern void* sk_realloc_throw(void* buffer, size_t size);
reed@android.com8a1c16f2008-12-17 15:59:43 +000063/** Free memory returned by sk_malloc(). It is safe to pass null.
64*/
reed@google.comde916c82011-10-19 19:50:48 +000065SK_API extern void sk_free(void*);
reed@android.com8a1c16f2008-12-17 15:59:43 +000066
reed@android.com4516f472009-06-29 16:25:36 +000067// bzero is safer than memset, but we can't rely on it, so... sk_bzero()
68static inline void sk_bzero(void* buffer, size_t size) {
69 memset(buffer, 0, size);
70}
71
reed@google.combdf73612011-09-06 14:56:20 +000072///////////////////////////////////////////////////////////////////////////////
73
74#ifdef SK_OVERRIDE_GLOBAL_NEW
75#include <new>
76
77inline void* operator new(size_t size) {
78 return sk_malloc_throw(size);
79}
80
81inline void operator delete(void* p) {
82 sk_free(p);
83}
84#endif
85
86///////////////////////////////////////////////////////////////////////////////
reed@android.com8a1c16f2008-12-17 15:59:43 +000087
88#define SK_INIT_TO_AVOID_WARNING = 0
89
90#ifndef SkDebugf
91 void SkDebugf(const char format[], ...);
92#endif
93
94#ifdef SK_DEBUG
95 #define SkASSERT(cond) SK_DEBUGBREAK(cond)
tomhudson@google.com0c00f212011-12-28 14:59:50 +000096 #define SkDEBUGFAIL(message) SkASSERT(false && message)
reed@android.com8a1c16f2008-12-17 15:59:43 +000097 #define SkDEBUGCODE(code) code
98 #define SkDECLAREPARAM(type, var) , type var
99 #define SkPARAM(var) , var
100// #define SkDEBUGF(args ) SkDebugf##args
101 #define SkDEBUGF(args ) SkDebugf args
102 #define SkAssertResult(cond) SkASSERT(cond)
103#else
104 #define SkASSERT(cond)
tomhudson@google.com0c00f212011-12-28 14:59:50 +0000105 #define SkDEBUGFAIL(message)
reed@android.com8a1c16f2008-12-17 15:59:43 +0000106 #define SkDEBUGCODE(code)
107 #define SkDEBUGF(args)
108 #define SkDECLAREPARAM(type, var)
109 #define SkPARAM(var)
110
111 // unlike SkASSERT, this guy executes its condition in the non-debug build
112 #define SkAssertResult(cond) cond
113#endif
114
robertphillips@google.com76f9e932013-01-15 20:17:47 +0000115#ifdef SK_DEVELOPER
116 #define SkDEVCODE(code) code
117 // the 'toString' helper functions convert Sk* objects to human-readable
118 // form in developer mode
119 #define SK_DEVELOPER_TO_STRING() virtual void toString(SkString* str) const SK_OVERRIDE;
120#else
121 #define SkDEVCODE(code)
122 #define SK_DEVELOPER_TO_STRING()
123#endif
124
vandebo@chromium.org28be72b2010-11-11 21:37:00 +0000125template <bool>
126struct SkCompileAssert {
127};
128
vandebo@chromium.org28be72b2010-11-11 21:37:00 +0000129#define SK_COMPILE_ASSERT(expr, msg) \
130 typedef SkCompileAssert<(bool(expr))> msg[bool(expr) ? 1 : -1]
131
reed@google.com49a5b192012-10-25 17:31:39 +0000132/*
133 * Usage: SK_MACRO_CONCAT(a, b) to construct the symbol ab
134 *
135 * SK_MACRO_CONCAT_IMPL_PRIV just exists to make this work. Do not use directly
136 *
137 */
138#define SK_MACRO_CONCAT(X, Y) SK_MACRO_CONCAT_IMPL_PRIV(X, Y)
139#define SK_MACRO_CONCAT_IMPL_PRIV(X, Y) X ## Y
140
141/*
142 * Usage: SK_MACRO_APPEND_LINE(foo) to make foo123, where 123 is the current
143 * line number. Easy way to construct
144 * unique names for local functions or
145 * variables.
146 */
147#define SK_MACRO_APPEND_LINE(name) SK_MACRO_CONCAT(name, __LINE__)
148
reed@android.com8a1c16f2008-12-17 15:59:43 +0000149///////////////////////////////////////////////////////////////////////
150
reed@google.com37a31332011-01-25 14:55:42 +0000151/**
152 * Fast type for signed 8 bits. Use for parameter passing and local variables,
153 * not for storage.
154 */
155typedef int S8CPU;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000156
reed@google.com37a31332011-01-25 14:55:42 +0000157/**
158 * Fast type for unsigned 8 bits. Use for parameter passing and local
159 * variables, not for storage
160 */
161typedef unsigned U8CPU;
162
163/**
164 * Fast type for signed 16 bits. Use for parameter passing and local variables,
165 * not for storage
166 */
167typedef int S16CPU;
168
169/**
170 * Fast type for unsigned 16 bits. Use for parameter passing and local
171 * variables, not for storage
172 */
173typedef unsigned U16CPU;
174
175/**
176 * Meant to be faster than bool (doesn't promise to be 0 or 1,
177 * just 0 or non-zero
178 */
179typedef int SkBool;
180
181/**
182 * Meant to be a small version of bool, for storage purposes. Will be 0 or 1
183 */
184typedef uint8_t SkBool8;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000185
186#ifdef SK_DEBUG
ctguil@chromium.org7ffb1b22011-03-15 21:27:08 +0000187 SK_API int8_t SkToS8(long);
188 SK_API uint8_t SkToU8(size_t);
189 SK_API int16_t SkToS16(long);
190 SK_API uint16_t SkToU16(size_t);
191 SK_API int32_t SkToS32(long);
192 SK_API uint32_t SkToU32(size_t);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000193#else
194 #define SkToS8(x) ((int8_t)(x))
195 #define SkToU8(x) ((uint8_t)(x))
196 #define SkToS16(x) ((int16_t)(x))
197 #define SkToU16(x) ((uint16_t)(x))
198 #define SkToS32(x) ((int32_t)(x))
199 #define SkToU32(x) ((uint32_t)(x))
200#endif
201
202/** Returns 0 or 1 based on the condition
203*/
204#define SkToBool(cond) ((cond) != 0)
205
206#define SK_MaxS16 32767
207#define SK_MinS16 -32767
208#define SK_MaxU16 0xFFFF
209#define SK_MinU16 0
210#define SK_MaxS32 0x7FFFFFFF
caryclark@google.com594dd3c2012-09-24 19:33:57 +0000211#define SK_MinS32 -SK_MaxS32
reed@android.com8a1c16f2008-12-17 15:59:43 +0000212#define SK_MaxU32 0xFFFFFFFF
213#define SK_MinU32 0
humper@google.com0e515772013-01-07 19:54:40 +0000214#define SK_NaN32 (1 << 31)
reed@android.com8a1c16f2008-12-17 15:59:43 +0000215
reed@android.comd4577752009-11-21 02:48:11 +0000216/** Returns true if the value can be represented with signed 16bits
217 */
reed@android.com90209ca2009-11-21 19:58:04 +0000218static inline bool SkIsS16(long x) {
reed@android.comd4577752009-11-21 02:48:11 +0000219 return (int16_t)x == x;
220}
221
222/** Returns true if the value can be represented with unsigned 16bits
223 */
reed@android.com90209ca2009-11-21 19:58:04 +0000224static inline bool SkIsU16(long x) {
reed@android.comd4577752009-11-21 02:48:11 +0000225 return (uint16_t)x == x;
226}
227
228//////////////////////////////////////////////////////////////////////////////
reed@android.com8a1c16f2008-12-17 15:59:43 +0000229#ifndef SK_OFFSETOF
reed@google.com56d3a232012-03-21 12:25:48 +0000230 #define SK_OFFSETOF(type, field) (size_t)((char*)&(((type*)1)->field) - (char*)1)
reed@android.com8a1c16f2008-12-17 15:59:43 +0000231#endif
232
233/** Returns the number of entries in an array (not a pointer)
234*/
235#define SK_ARRAY_COUNT(array) (sizeof(array) / sizeof(array[0]))
236
reed@android.com8a1c16f2008-12-17 15:59:43 +0000237#define SkAlign2(x) (((x) + 1) >> 1 << 1)
reed@google.comc6faa5a2012-06-27 15:07:11 +0000238#define SkIsAlign2(x) (0 == ((x) & 1))
reed@android.com8a1c16f2008-12-17 15:59:43 +0000239
reed@google.comc6faa5a2012-06-27 15:07:11 +0000240#define SkAlign4(x) (((x) + 3) >> 2 << 2)
241#define SkIsAlign4(x) (0 == ((x) & 3))
242
243#define SkAlign8(x) (((x) + 7) >> 3 << 3)
244#define SkIsAlign8(x) (0 == ((x) & 7))
tomhudson@google.com01224d52011-11-28 18:22:01 +0000245
reed@android.com8a1c16f2008-12-17 15:59:43 +0000246typedef uint32_t SkFourByteTag;
247#define SkSetFourByteTag(a, b, c, d) (((a) << 24) | ((b) << 16) | ((c) << 8) | (d))
248
249/** 32 bit integer to hold a unicode value
250*/
251typedef int32_t SkUnichar;
252/** 32 bit value to hold a millisecond count
253*/
254typedef uint32_t SkMSec;
255/** 1 second measured in milliseconds
256*/
257#define SK_MSec1 1000
258/** maximum representable milliseconds
259*/
260#define SK_MSecMax 0x7FFFFFFF
261/** Returns a < b for milliseconds, correctly handling wrap-around from 0xFFFFFFFF to 0
262*/
263#define SkMSec_LT(a, b) ((int32_t)(a) - (int32_t)(b) < 0)
264/** Returns a <= b for milliseconds, correctly handling wrap-around from 0xFFFFFFFF to 0
265*/
266#define SkMSec_LE(a, b) ((int32_t)(a) - (int32_t)(b) <= 0)
267
reed@android.com8a1c16f2008-12-17 15:59:43 +0000268/****************************************************************************
269 The rest of these only build with C++
270*/
271#ifdef __cplusplus
272
273/** Faster than SkToBool for integral conditions. Returns 0 or 1
274*/
reed@android.comd4577752009-11-21 02:48:11 +0000275static inline int Sk32ToBool(uint32_t n) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000276 return (n | (0-n)) >> 31;
277}
278
bsalomon@google.comff436612013-02-27 19:07:32 +0000279/** Generic swap function. Classes with efficient swaps should specialize this function to take
280 their fast path. This function is used by SkTSort. */
reed@android.comd4577752009-11-21 02:48:11 +0000281template <typename T> inline void SkTSwap(T& a, T& b) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000282 T c(a);
283 a = b;
284 b = c;
285}
286
reed@android.comd4577752009-11-21 02:48:11 +0000287static inline int32_t SkAbs32(int32_t value) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000288#ifdef SK_CPU_HAS_CONDITIONAL_INSTR
289 if (value < 0)
290 value = -value;
291 return value;
292#else
293 int32_t mask = value >> 31;
294 return (value ^ mask) - mask;
295#endif
296}
297
reed@google.com2b57dc62013-01-08 13:23:32 +0000298template <typename T> inline T SkTAbs(T value) {
299 if (value < 0) {
300 value = -value;
301 }
302 return value;
303}
304
reed@android.comd4577752009-11-21 02:48:11 +0000305static inline int32_t SkMax32(int32_t a, int32_t b) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000306 if (a < b)
307 a = b;
308 return a;
309}
310
reed@android.comd4577752009-11-21 02:48:11 +0000311static inline int32_t SkMin32(int32_t a, int32_t b) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000312 if (a > b)
313 a = b;
314 return a;
315}
316
caryclark@google.com3b97af52013-04-23 11:56:44 +0000317template <typename T> const T& SkTMin(const T& a, const T& b) {
318 return (a < b) ? a : b;
319}
320
321template <typename T> const T& SkTMax(const T& a, const T& b) {
322 return (b < a) ? a : b;
323}
324
reed@android.comd4577752009-11-21 02:48:11 +0000325static inline int32_t SkSign32(int32_t a) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000326 return (a >> 31) | ((unsigned) -a >> 31);
327}
328
reed@android.comd4577752009-11-21 02:48:11 +0000329static inline int32_t SkFastMin32(int32_t value, int32_t max) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000330#ifdef SK_CPU_HAS_CONDITIONAL_INSTR
331 if (value > max)
332 value = max;
333 return value;
334#else
335 int diff = max - value;
336 // clear diff if it is negative (clear if value > max)
337 diff &= (diff >> 31);
338 return value + diff;
339#endif
340}
341
342/** Returns signed 32 bit value pinned between min and max, inclusively
343*/
reed@android.comd4577752009-11-21 02:48:11 +0000344static inline int32_t SkPin32(int32_t value, int32_t min, int32_t max) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000345#ifdef SK_CPU_HAS_CONDITIONAL_INSTR
346 if (value < min)
347 value = min;
348 if (value > max)
349 value = max;
350#else
351 if (value < min)
352 value = min;
353 else if (value > max)
354 value = max;
355#endif
356 return value;
357}
358
reed@android.comd4577752009-11-21 02:48:11 +0000359static inline uint32_t SkSetClearShift(uint32_t bits, bool cond,
360 unsigned shift) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000361 SkASSERT((int)cond == 0 || (int)cond == 1);
362 return (bits & ~(1 << shift)) | ((int)cond << shift);
363}
364
reed@android.comd4577752009-11-21 02:48:11 +0000365static inline uint32_t SkSetClearMask(uint32_t bits, bool cond,
366 uint32_t mask) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000367 return cond ? bits | mask : bits & ~mask;
368}
369
reed@google.com1fcd51e2011-01-05 15:50:27 +0000370///////////////////////////////////////////////////////////////////////////////
371
vandebo@chromium.org325cb9a2011-03-30 18:36:29 +0000372/** Use to combine multiple bits in a bitmask in a type safe way.
373 */
374template <typename T>
375T SkTBitOr(T a, T b) {
376 return (T)(a | b);
377}
378
reed@google.com1fcd51e2011-01-05 15:50:27 +0000379/**
380 * Use to cast a pointer to a different type, and maintaining strict-aliasing
381 */
382template <typename Dst> Dst SkTCast(const void* ptr) {
383 union {
384 const void* src;
385 Dst dst;
386 } data;
387 data.src = ptr;
388 return data.dst;
389}
390
reed@android.com8a1c16f2008-12-17 15:59:43 +0000391//////////////////////////////////////////////////////////////////////////////
392
393/** \class SkNoncopyable
394
395SkNoncopyable is the base class for objects that may do not want to
396be copied. It hides its copy-constructor and its assignment-operator.
397*/
ctguil@chromium.org7ffb1b22011-03-15 21:27:08 +0000398class SK_API SkNoncopyable {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000399public:
400 SkNoncopyable() {}
reed@google.com1fcd51e2011-01-05 15:50:27 +0000401
reed@android.com8a1c16f2008-12-17 15:59:43 +0000402private:
403 SkNoncopyable(const SkNoncopyable&);
404 SkNoncopyable& operator=(const SkNoncopyable&);
405};
406
407class SkAutoFree : SkNoncopyable {
408public:
409 SkAutoFree() : fPtr(NULL) {}
410 explicit SkAutoFree(void* ptr) : fPtr(ptr) {}
411 ~SkAutoFree() { sk_free(fPtr); }
reed@google.com1fcd51e2011-01-05 15:50:27 +0000412
reed@android.com8a1c16f2008-12-17 15:59:43 +0000413 /** Return the currently allocate buffer, or null
414 */
415 void* get() const { return fPtr; }
416
417 /** Assign a new ptr allocated with sk_malloc (or null), and return the
418 previous ptr. Note it is the caller's responsibility to sk_free the
419 returned ptr.
420 */
421 void* set(void* ptr) {
422 void* prev = fPtr;
423 fPtr = ptr;
424 return prev;
425 }
reed@google.com1fcd51e2011-01-05 15:50:27 +0000426
reed@android.com8a1c16f2008-12-17 15:59:43 +0000427 /** Transfer ownership of the current ptr to the caller, setting the
428 internal reference to null. Note the caller is reponsible for calling
429 sk_free on the returned address.
430 */
431 void* detach() { return this->set(NULL); }
432
433 /** Free the current buffer, and set the internal reference to NULL. Same
434 as calling sk_free(detach())
435 */
436 void free() {
437 sk_free(fPtr);
438 fPtr = NULL;
439 }
440
441private:
442 void* fPtr;
443 // illegal
444 SkAutoFree(const SkAutoFree&);
445 SkAutoFree& operator=(const SkAutoFree&);
446};
447
bsalomon@google.com7d4679a2011-09-02 22:06:24 +0000448/**
449 * Manage an allocated block of heap memory. This object is the sole manager of
450 * the lifetime of the block, so the caller must not call sk_free() or delete
reed@google.com1c401d82011-10-18 18:52:03 +0000451 * on the block, unless detach() was called.
bsalomon@google.com7d4679a2011-09-02 22:06:24 +0000452 */
453class SkAutoMalloc : public SkNoncopyable {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000454public:
reed@google.com3ab41952011-10-18 18:32:46 +0000455 explicit SkAutoMalloc(size_t size = 0) {
456 fPtr = size ? sk_malloc_throw(size) : NULL;
bsalomon@google.com7d4679a2011-09-02 22:06:24 +0000457 fSize = size;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000458 }
bsalomon@google.com7d4679a2011-09-02 22:06:24 +0000459
460 ~SkAutoMalloc() {
461 sk_free(fPtr);
462 }
463
464 /**
reed@google.com1c401d82011-10-18 18:52:03 +0000465 * Passed to reset to specify what happens if the requested size is smaller
466 * than the current size (and the current block was dynamically allocated).
467 */
468 enum OnShrink {
469 /**
470 * If the requested size is smaller than the current size, and the
471 * current block is dynamically allocated, free the old block and
472 * malloc a new block of the smaller size.
473 */
474 kAlloc_OnShrink,
rmistry@google.comfbfcd562012-08-23 18:09:54 +0000475
reed@google.com1c401d82011-10-18 18:52:03 +0000476 /**
477 * If the requested size is smaller than the current size, and the
478 * current block is dynamically allocated, just return the old
479 * block.
480 */
tomhudson@google.com1f902872012-06-01 13:15:47 +0000481 kReuse_OnShrink
reed@google.com1c401d82011-10-18 18:52:03 +0000482 };
483
484 /**
bsalomon@google.com7d4679a2011-09-02 22:06:24 +0000485 * Reallocates the block to a new size. The ptr may or may not change.
486 */
reed@google.com1c401d82011-10-18 18:52:03 +0000487 void* reset(size_t size, OnShrink shrink = kAlloc_OnShrink) {
488 if (size == fSize || (kReuse_OnShrink == shrink && size < fSize)) {
489 return fPtr;
bsalomon@google.com7d4679a2011-09-02 22:06:24 +0000490 }
reed@google.com1c401d82011-10-18 18:52:03 +0000491
492 sk_free(fPtr);
493 fPtr = size ? sk_malloc_throw(size) : NULL;
494 fSize = size;
495
bsalomon@google.com7d4679a2011-09-02 22:06:24 +0000496 return fPtr;
497 }
498
499 /**
500 * Releases the block back to the heap
501 */
502 void free() {
503 this->reset(0);
504 }
505
506 /**
507 * Return the allocated block.
508 */
509 void* get() { return fPtr; }
510 const void* get() const { return fPtr; }
511
bsalomon@google.com6dcd27c2011-09-06 15:02:33 +0000512 /** Transfer ownership of the current ptr to the caller, setting the
513 internal reference to null. Note the caller is reponsible for calling
514 sk_free on the returned address.
515 */
516 void* detach() {
517 void* ptr = fPtr;
518 fPtr = NULL;
519 fSize = 0;
520 return ptr;
521 }
522
bsalomon@google.com7d4679a2011-09-02 22:06:24 +0000523private:
524 void* fPtr;
reed@google.com1c401d82011-10-18 18:52:03 +0000525 size_t fSize; // can be larger than the requested size (see kReuse)
reed@android.com8a1c16f2008-12-17 15:59:43 +0000526};
527
reed@google.com63a60602011-03-10 13:07:35 +0000528/**
529 * Manage an allocated block of memory. If the requested size is <= kSize, then
530 * the allocation will come from the stack rather than the heap. This object
531 * is the sole manager of the lifetime of the block, so the caller must not
532 * call sk_free() or delete on the block.
533 */
reed@android.com8a1c16f2008-12-17 15:59:43 +0000534template <size_t kSize> class SkAutoSMalloc : SkNoncopyable {
535public:
reed@google.com63a60602011-03-10 13:07:35 +0000536 /**
537 * Creates initially empty storage. get() returns a ptr, but it is to
bsalomon@google.com7d4679a2011-09-02 22:06:24 +0000538 * a zero-byte allocation. Must call reset(size) to return an allocated
reed@google.com63a60602011-03-10 13:07:35 +0000539 * block.
540 */
541 SkAutoSMalloc() {
542 fPtr = fStorage;
reed@google.com1c401d82011-10-18 18:52:03 +0000543 fSize = 0;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000544 }
reed@google.com63a60602011-03-10 13:07:35 +0000545
546 /**
547 * Allocate a block of the specified size. If size <= kSize, then the
548 * allocation will come from the stack, otherwise it will be dynamically
549 * allocated.
550 */
551 explicit SkAutoSMalloc(size_t size) {
552 fPtr = fStorage;
reed@google.com7a17e3a2011-10-18 18:58:06 +0000553 fSize = 0;
bsalomon@google.com7d4679a2011-09-02 22:06:24 +0000554 this->reset(size);
reed@google.com63a60602011-03-10 13:07:35 +0000555 }
556
557 /**
558 * Free the allocated block (if any). If the block was small enought to
559 * have been allocated on the stack (size <= kSize) then this does nothing.
560 */
561 ~SkAutoSMalloc() {
562 if (fPtr != (void*)fStorage) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000563 sk_free(fPtr);
reed@google.com63a60602011-03-10 13:07:35 +0000564 }
reed@android.com8a1c16f2008-12-17 15:59:43 +0000565 }
reed@google.com63a60602011-03-10 13:07:35 +0000566
567 /**
568 * Return the allocated block. May return non-null even if the block is
569 * of zero size. Since this may be on the stack or dynamically allocated,
570 * the caller must not call sk_free() on it, but must rely on SkAutoSMalloc
571 * to manage it.
572 */
reed@android.com8a1c16f2008-12-17 15:59:43 +0000573 void* get() const { return fPtr; }
reed@google.com63a60602011-03-10 13:07:35 +0000574
575 /**
576 * Return a new block of the requested size, freeing (as necessary) any
577 * previously allocated block. As with the constructor, if size <= kSize
578 * then the return block may be allocated locally, rather than from the
579 * heap.
580 */
reed@google.com1c401d82011-10-18 18:52:03 +0000581 void* reset(size_t size,
582 SkAutoMalloc::OnShrink shrink = SkAutoMalloc::kAlloc_OnShrink) {
583 if (size == fSize || (SkAutoMalloc::kReuse_OnShrink == shrink &&
584 size < fSize)) {
585 return fPtr;
586 }
587
reed@google.com63a60602011-03-10 13:07:35 +0000588 if (fPtr != (void*)fStorage) {
589 sk_free(fPtr);
590 }
591
592 if (size <= kSize) {
593 fPtr = fStorage;
594 } else {
595 fPtr = sk_malloc_flags(size, SK_MALLOC_THROW | SK_MALLOC_TEMP);
596 }
597 return fPtr;
598 }
599
reed@android.com8a1c16f2008-12-17 15:59:43 +0000600private:
601 void* fPtr;
reed@google.com1c401d82011-10-18 18:52:03 +0000602 size_t fSize; // can be larger than the requested size (see kReuse)
reed@android.com8a1c16f2008-12-17 15:59:43 +0000603 uint32_t fStorage[(kSize + 3) >> 2];
reed@android.com8a1c16f2008-12-17 15:59:43 +0000604};
605
606#endif /* C++ */
607
608#endif