X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/53c75dc2f27fb1679f47a4a297f36d5066d354da..ad60f9e7b66d326505592cc6dae034cab5826f3b:/include/wx/buffer.h diff --git a/include/wx/buffer.h b/include/wx/buffer.h index 81a445afe5..4cc340aa51 100644 --- a/include/wx/buffer.h +++ b/include/wx/buffer.h @@ -15,7 +15,9 @@ #include "wx/chartype.h" #include "wx/wxcrtbase.h" +#ifndef __WXPALMOS5__ #include // malloc() and free() +#endif // ! __WXPALMOS5__ class WXDLLIMPEXP_FWD_BASE wxCStrData; @@ -24,135 +26,199 @@ class WXDLLIMPEXP_FWD_BASE wxCStrData; // of new/delete // ---------------------------------------------------------------------------- +// helpers used by wxCharTypeBuffer +namespace wxPrivate +{ + +struct UntypedBufferData +{ + enum Kind + { + Owned, + NonOwned + }; + + UntypedBufferData(void *str, Kind kind = Owned) + : m_str(str), m_ref(1), m_owned(kind == Owned) {} + + ~UntypedBufferData() + { + if ( m_owned ) + free(m_str); + } + + void *m_str; + + // "short" to have sizeof(Data)=8 on 32bit archs + unsigned short m_ref; + + bool m_owned; +}; + +// this has to be defined inside the DLL (and not e.g. as a static variable +// inside an inline function) as otherwise MSVC gives link errors when the +// functions are effectively inlined (i.e. in non-debug build) +// +// NB: this is defined in string.cpp and not the (non-existent) buffer.cpp +extern WXDLLIMPEXP_DATA_BASE(UntypedBufferData * const) untypedNullDataPtr; + +} // namespace wxPrivate + template -class WXDLLIMPEXP_BASE wxCharTypeBuffer +class wxCharTypeBuffer { public: typedef T CharType; wxCharTypeBuffer(const CharType *str = NULL) - : m_str(str ? wxStrdup(str) : NULL), - m_owned(true) { + if ( str ) + m_data = new Data(wxStrdup(str)); + else + m_data = GetNullData(); } wxCharTypeBuffer(size_t len) - : m_str((CharType *)malloc((len + 1)*sizeof(CharType))), - m_owned(true) { - m_str[len] = (CharType)0; + m_data = new Data((CharType *)malloc((len + 1)*sizeof(CharType))); + m_data->Get()[len] = (CharType)0; } static const wxCharTypeBuffer CreateNonOwned(const CharType *str) { wxCharTypeBuffer buf; - buf.m_str = wx_const_cast(CharType*, str); - buf.m_owned = false; + if ( str ) + buf.m_data = new Data(const_cast(str), Data::NonOwned); return buf; } - /* no need to check for NULL, free() does it */ ~wxCharTypeBuffer() { - if ( m_owned) - free(m_str); + DecRef(); } - /* - WARNING: + // NB: this method is only const for backward compatibility. It used to + // be needed for auto_ptr-like semantics of the copy ctor, but now + // that ref-counting is used, it's not really needed. + CharType *release() const + { + if ( m_data == GetNullData() ) + return NULL; - the copy ctor and assignment operators change the passed in object - even although it is declared as "const", so: + wxASSERT_MSG( m_data->m_owned, _T("can't release non-owned buffer") ); + wxASSERT_MSG( m_data->m_ref == 1, _T("can't release shared buffer") ); - a) it shouldn't be really const - b) you shouldn't use it afterwards (or know that it was reset) + CharType * const p = m_data->Get(); - This is very ugly but is unfortunately needed to make the normal use - of wxCharTypeBuffer buffer objects possible and is very similar to what - std::auto_ptr<> does (as if it were an excuse...) - */ + wxCharTypeBuffer *self = const_cast(this); + self->m_data->Set(NULL); + self->DecRef(); - /* - because of the remark above, release() is declared const even if it - isn't really const - */ - CharType *release() const - { - wxASSERT_MSG( m_owned, _T("can't release non-owned buffer") ); - return DoRelease(); + return p; } void reset() { - if ( m_owned ) - free(m_str); - m_str = NULL; + DecRef(); } wxCharTypeBuffer(const wxCharTypeBuffer& src) { - CopyFrom(src); + m_data = src.m_data; + IncRef(); } wxCharTypeBuffer& operator=(const CharType *str) { - if ( m_owned ) - free(m_str); - m_str = str ? wxStrdup(str) : NULL; - m_owned = true; + DecRef(); + + if ( str ) + m_data = new Data(wxStrdup(str)); return *this; } wxCharTypeBuffer& operator=(const wxCharTypeBuffer& src) { - if ( m_owned ) - free(m_str); - CopyFrom(src); + if ( &src == this ) + return *this; + + DecRef(); + m_data = src.m_data; + IncRef(); + return *this; } bool extend(size_t len) { - wxASSERT_MSG( m_owned, _T("cannot extend non-owned buffer") ); + wxASSERT_MSG( m_data->m_owned, _T("cannot extend non-owned buffer") ); + wxASSERT_MSG( m_data->m_ref == 1, _T("can't extend shared buffer") ); - CharType * - str = (CharType *)realloc(m_str, (len + 1)*sizeof(CharType)); + CharType *str = + (CharType *)realloc(data(), (len + 1) * sizeof(CharType)); if ( !str ) return false; - m_str = str; + if ( m_data == GetNullData() ) + { + m_data = new Data(str); + } + else + { + m_data->Set(str); + m_data->m_owned = true; + } return true; } - CharType *data() { return m_str; } - const CharType *data() const { return m_str; } - operator const CharType *() const { return m_str; } - CharType operator[](size_t n) const { return m_str[n]; } - + CharType *data() { return m_data->Get(); } + const CharType *data() const { return m_data->Get(); } + operator const CharType *() const { return data(); } + CharType operator[](size_t n) const { return data()[n]; } private: - CharType *DoRelease() const + // reference-counted data + struct Data : public wxPrivate::UntypedBufferData { - CharType *p = m_str; - ((wxCharTypeBuffer *)this)->m_str = NULL; - return p; + Data(CharType *str, Kind kind = Owned) + : wxPrivate::UntypedBufferData(str, kind) + { + } + + CharType *Get() const { return static_cast(m_str); } + void Set(CharType *str) { m_str = str; } + }; + + // placeholder for NULL string, to simplify this code + static Data *GetNullData() + { + return static_cast(wxPrivate::untypedNullDataPtr); } - void CopyFrom(const wxCharTypeBuffer& src) + void IncRef() { - m_owned = src.m_owned; - m_str = src.DoRelease(); + if ( m_data == GetNullData() ) // exception, not ref-counted + return; + m_data->m_ref++; + } + + void DecRef() + { + if ( m_data == GetNullData() ) // exception, not ref-counted + return; + if ( --m_data->m_ref == 0 ) + delete m_data; + m_data = GetNullData(); } private: - CharType *m_str; - bool m_owned; + Data *m_data; }; WXDLLIMPEXP_TEMPLATE_INSTANCE_BASE( wxCharTypeBuffer ) -class WXDLLIMPEXP_BASE wxCharBuffer : public wxCharTypeBuffer +class wxCharBuffer : public wxCharTypeBuffer { public: typedef wxCharTypeBuffer wxCharTypeBufferBase; @@ -169,7 +235,7 @@ public: #if wxUSE_WCHAR_T WXDLLIMPEXP_TEMPLATE_INSTANCE_BASE( wxCharTypeBuffer ) -class WXDLLIMPEXP_BASE wxWCharBuffer : public wxCharTypeBuffer +class wxWCharBuffer : public wxCharTypeBuffer { public: typedef wxCharTypeBuffer wxCharTypeBufferBase; @@ -295,7 +361,7 @@ private: }; -class WXDLLIMPEXP_BASE wxMemoryBuffer +class wxMemoryBuffer { public: // ctor and dtor @@ -317,9 +383,12 @@ public: wxMemoryBuffer& operator=(const wxMemoryBuffer& src) { - m_bufdata->DecRef(); - m_bufdata = src.m_bufdata; - m_bufdata->IncRef(); + if (&src != this) + { + m_bufdata->DecRef(); + m_bufdata = src.m_bufdata; + m_bufdata->IncRef(); + } return *this; }