]>
git.saurik.com Git - wxWidgets.git/blob - include/wx/buffer.h
1 ///////////////////////////////////////////////////////////////////////////////
3 // Purpose: auto buffer classes: buffers which automatically free memory
4 // Author: Vadim Zeitlin
8 // Copyright: (c) 1998 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
9 // Licence: wxWindows licence
10 ///////////////////////////////////////////////////////////////////////////////
12 // these classes are for private use only for now, they're not documented
17 #include "wx/wxchar.h"
19 #include <stdlib.h> // malloc() and free()
21 // ----------------------------------------------------------------------------
22 // Special classes for (wide) character strings: they use malloc/free instead
24 // ----------------------------------------------------------------------------
26 #define DEFINE_BUFFER(classname, chartype, strdupfunc) \
30 classname(const chartype *str) \
31 : m_str(str ? strdupfunc(str) : NULL) \
35 classname(size_t len) \
36 : m_str((chartype *)malloc((len + 1)*sizeof(chartype))) \
38 m_str[len] = (chartype)0; \
41 /* no need to check for NULL, free() does it */ \
42 ~classname() { free(m_str); } \
47 the copy ctor and assignment operators change the passed in object \
48 even although it is declared as "const", so: \
50 a) it shouldn't be really const \
51 b) you shouldn't use it afterwards (or know that it was reset) \
53 This is very ugly but is unfortunately needed to make the normal use\
54 of classname buffer objects possible and is very similar to what \
55 std::auto_ptr<> does (as if it were an excuse...) \
59 because of the remark above, release() is declared const even if it \
62 chartype *release() const \
64 chartype *p = m_str; \
65 ((classname *)this)->m_str = NULL; \
69 classname(const classname& src) \
70 : m_str(src.release()) \
74 classname& operator=(const chartype *str) \
77 m_str = str ? strdupfunc(str) : NULL; \
81 classname& operator=(const classname& src) \
84 m_str = src.release(); \
89 chartype *data() { return m_str; } \
90 const chartype *data() const { return m_str; } \
91 operator const chartype *() const { return m_str; } \
92 chartype operator[](size_t n) const { return m_str[n]; } \
98 DEFINE_BUFFER(wxCharBuffer
, char, wxStrdupA
);
102 DEFINE_BUFFER(wxWCharBuffer
, wchar_t, wxStrdupW
);
104 #endif // wxUSE_WCHAR_T
109 #define wxMB2WXbuf wxWCharBuffer
110 #define wxWX2MBbuf wxCharBuffer
111 #define wxWC2WXbuf wxChar*
112 #define wxWX2WCbuf wxChar*
114 #define wxMB2WXbuf wxChar*
115 #define wxWX2MBbuf wxChar*
116 #define wxWC2WXbuf wxCharBuffer
117 #define wxWX2WCbuf wxWCharBuffer
118 #endif // Unicode/ANSI
120 // ----------------------------------------------------------------------------
121 // A class for holding growable data buffers (not necessarily strings)
122 // ----------------------------------------------------------------------------
124 // This class manages the actual data buffer pointer and is ref-counted.
125 class wxMemoryBufferData
128 // the initial size and also the size added by ResizeIfNeeded()
129 enum { BLOCK_SIZE
= 1024 };
131 friend class wxMemoryBuffer
;
133 // everyting is private as it can only be used by wxMemoryBuffer
135 wxMemoryBufferData(size_t size
= wxMemoryBufferData::BLOCK_SIZE
)
136 : m_data(size
? malloc(size
) : NULL
), m_size(size
), m_len(0), m_ref(0)
139 ~wxMemoryBufferData() { free(m_data
); }
142 void ResizeIfNeeded(size_t newSize
)
144 if (newSize
> m_size
)
146 void *dataOld
= m_data
;
147 m_data
= realloc(m_data
, newSize
+ wxMemoryBufferData::BLOCK_SIZE
);
153 m_size
= newSize
+ wxMemoryBufferData::BLOCK_SIZE
;
157 void IncRef() { m_ref
+= 1; }
161 if (m_ref
== 0) // are there no more references?
166 // the buffer containing the data
169 // the size of the buffer
172 // the amount of data currently in the buffer
175 // the reference count
178 DECLARE_NO_COPY_CLASS(wxMemoryBufferData
)
186 wxMemoryBuffer(size_t size
= wxMemoryBufferData::BLOCK_SIZE
)
188 m_bufdata
= new wxMemoryBufferData(size
);
192 ~wxMemoryBuffer() { m_bufdata
->DecRef(); }
195 // copy and assignment
196 wxMemoryBuffer(const wxMemoryBuffer
& src
)
197 : m_bufdata(src
.m_bufdata
)
202 wxMemoryBuffer
& operator=(const wxMemoryBuffer
& src
)
205 m_bufdata
= src
.m_bufdata
;
212 void *GetData() const { return m_bufdata
->m_data
; }
213 size_t GetBufSize() const { return m_bufdata
->m_size
; }
214 size_t GetDataLen() const { return m_bufdata
->m_len
; }
216 void SetBufSize(size_t size
) { m_bufdata
->ResizeIfNeeded(size
); }
217 void SetDataLen(size_t len
)
219 wxASSERT(len
<= m_bufdata
->m_size
);
220 m_bufdata
->m_len
= len
;
223 // Ensure the buffer is big enough and return a pointer to it
224 void *GetWriteBuf(size_t sizeNeeded
)
226 m_bufdata
->ResizeIfNeeded(sizeNeeded
);
227 return m_bufdata
->m_data
;
230 // Update the length after the write
231 void UngetWriteBuf(size_t sizeUsed
) { SetDataLen(sizeUsed
); }
233 // Like the above, but appends to the buffer
234 void *GetAppendBuf(size_t sizeNeeded
)
236 m_bufdata
->ResizeIfNeeded(m_bufdata
->m_len
+ sizeNeeded
);
237 return (char*)m_bufdata
->m_data
+ m_bufdata
->m_len
;
240 // Update the length after the append
241 void UngetAppendBuf(size_t sizeUsed
)
243 SetDataLen(m_bufdata
->m_len
+ sizeUsed
);
246 // Other ways to append to the buffer
247 void AppendByte(char data
)
249 wxCHECK_RET( m_bufdata
->m_data
, _T("invalid wxMemoryBuffer") );
251 m_bufdata
->ResizeIfNeeded(m_bufdata
->m_len
+ 1);
252 *(((char*)m_bufdata
->m_data
) + m_bufdata
->m_len
) = data
;
253 m_bufdata
->m_len
+= 1;
256 void AppendData(void* data
, size_t len
)
258 memcpy(GetAppendBuf(len
), data
, len
);
262 operator const char *() const { return (const char*)GetData(); }
265 wxMemoryBufferData
* m_bufdata
;
268 // ----------------------------------------------------------------------------
269 // template class for any kind of data
270 // ----------------------------------------------------------------------------
274 #endif // _WX_BUFFER_H