--- parser3/src/include/pa_pool.h 2001/01/26 14:47:52 1.1 +++ parser3/src/include/pa_pool.h 2003/02/14 16:52:21 1.86.2.28 @@ -1,96 +1,384 @@ -/* +/** @file + Parser: Parser: reference counting classes decls. - String Chunk0 - ====== ======== - head_chunk--------->[ptr, size] - append_here-------->[ptr, size] - chunk_link_row ........ - . . - . [ptr, size] - ...........>[link to the next chunk] + Copyright (c) 2001-2003 ArtLebedev Group (http://www.artlebedev.com) + Author: Alexandr Petrosian (http://paf.design.ru) */ #ifndef PA_POOL_H #define PA_POOL_H -#include +static const char* IDENT_POOL_H="$Date: 2003/02/14 16:52:21 $"; -class Pool; +// include -class String { - friend Pool; +#include "pa_config_includes.h" - // the pool I'm allocated on - Pool *pool; +#ifdef XML +# include "gdome.h" +// for xmlChar +# include "libxml/tree.h" +#endif - // the number of records per chunk - int chunk_items; - enum { - CI_CONST_STRING_GROW=5, - CI_DYNAMIC_STRING_GROW_DEFAULT=10 - }; +// forwards + +void *pa_malloc(size_t size); +void *pa_calloc(size_t size); +void pa_free(void *ptr); +void *pa_realloc(void *ptr, size_t size); + +class Exception; +class String; +class Charset; +class GdomeDOMString_auto_ptr; + +/* * + Pool mechanizm allows users not to free up allocated memory, + leaving that problem to 'pools'. + + @see Pooled +* / + +class Pool { +public: + + Pool(void *astorage); - struct Chunk { - union Row { - // chunk item - struct { - char *ptr; // pointer to the start of string fragment - size_t size; // length of the fragment - } item; - Chunk *link; // link to the next chunk in chain - } first; - // next rows are here + //{@ statistics + size_t total_allocated() { return ftotal_allocated; } + unsigned int total_times() { return ftotal_times; } + //}@ + + void set_context(void *acontext) { fcontext=acontext; } + void *get_context() { return fcontext; } + + /// allocates some bytes on pool + void *malloc(size_t size, int place=0) { + return check(real_malloc(size, place), size); + } + /// allocates some bytes clearing them with zeros + void *calloc(size_t size) { + return check(real_calloc(size), size); } - *head_chunk; // the head chunk of the chunk chain - // next append would write to this record - Chunk::Row *append_here; + /// registers a routine to clean up non-pooled allocations + void register_cleanup(void (*cleanup) (void *), void *data) { + if(!real_register_cleanup(cleanup, data)) + fail_register_cleanup(); + } + + //{@ helpers + void *copy(const void *buf, const size_t size); + char *copy(const char* cstr); + //}@ + + //{@ source charset + void set_source_charset(Charset& acharset); + Charset& get_source_charset(); + //}@ + + //{@ client charset + void set_client_charset(Charset& charset); + Charset& get_client_charset(); + //}@ + +#ifdef XML + + /// @see Charset::transcode_cstr(xmlChar *s); + const char* transcode_cstr(xmlChar *s); + /// @see Charset::transcode(xmlChar *s); + String& transcode(xmlChar *s +#ifndef NO_STRING_ORIGIN + , const String *origin +#endif + ); + /// @see Charset::transcode_cstr(GdomeDOMString *s); + const char* transcode_cstr(GdomeDOMString *s); + /// @see Charset::transcode(GdomeDOMString *s); + String& transcode(GdomeDOMString *s +#ifndef NO_STRING_ORIGIN + , const String *origin +#endif + ); + /// @see Charset::transcode_cstr(const char* buf, size_t buf_size=0); + xmlChar *transcode_buf2xchar(const char* buf, size_t buf_size=0); + /// @see Charset::transcode(const String& s) + GdomeDOMString_auto_ptr transcode(const String& s); + +#endif + +private: + + void *fstorage; + void *fcontext; + Charset *source_charset; + Charset *client_charset; + +private: - // the address of place where lies address - // of the link to the next chunk to allocate - Chunk::Row *chunk_link_row; + //{ + /// @name implementation defined + void *real_malloc(size_t size, int place); + void *real_calloc(size_t size); + bool real_register_cleanup(void (*cleanup) (void *), void *data); + //} + +private: + + /// checks whether mem allocated OK. throws exception otherwise + void *check(void *ptr, size_t size) { + if(ptr) { + ftotal_allocated+=size; + ftotal_times++; + return ptr; + } - // new&constructors made private to enforce factory manufacturing at pool - void *operator new(size_t size, Pool *apool); + fail_alloc(size); - void construct(Pool *apool, int achunk_items); - String(Pool *apool, int achunk_items=CI_DYNAMIC_STRING_GROW_DEFAULT) { - construct(apool, achunk_items); + // never reached + return 0; } - String(Pool *apool, char *src) { - construct(apool, CI_CONST_STRING_GROW); - *this+=src; + /// throws allocation exception + void fail_alloc(size_t size) const; + + /// throws register cleanup exception + void fail_register_cleanup() const; + +private: // statistics + + size_t ftotal_allocated; + unsigned int ftotal_times; + +private: //disabled + + Pool(const Pool&); + Pool& operator= (const Pool&); +}; +*/ + + +/** + Base for all classes that are allocated in 'pools'. + + Holds Pool object. Contains useful wrappers to it's methods. + + @see NEW +* / + +// all classes that are members parents of packed class [String] +// sould be packed also to avoid sparc odd st/lduh problem +#include "pa_pragma_pack_begin.h" +class Pooled { + // the pool i'm allocated on + Pool *fpool; +public: + + /// the Pooled-sole: Pooled instances can be allocated in Pool rather then on heap + static void *operator new(size_t size, Pool& apool) { + return apool.malloc(size, 1); } - bool chunk_is_full() { - return append_here == chunk_link_row; + Pooled(Pool& apool) : fpool(&apool) {} + + /// my pool + Pool& pool() const { return *fpool; } + + /** used for moving objects from one pool to another. + in between object can have no pool and can not be used + @see SQL_Driver_manager + * / + void set_pool(Pool *apool) { fpool=apool; } + + //{ + /// @name useful wrapper around pool + void *malloc(size_t size, int place=0) const { return fpool->malloc(size, place); } + void *calloc(size_t size) const { return fpool->calloc(size); } + void register_cleanup(void (*cleanup) (void *), void *data) { fpool->register_cleanup(cleanup, data); } + void *copy(const void *buf, const size_t size) { return fpool->copy(buf, size); } + char *copy(const char* cstr) { return fpool->copy(cstr); } +#ifdef XML + + const char* transcode_cstr(GdomeDOMString *s) { return fpool->transcode_cstr(s); } + String& transcode(GdomeDOMString *s +#ifndef NO_STRING_ORIGIN + , const String *origin +#endif + ) { + return fpool->transcode(s +#ifndef NO_STRING_ORIGIN + , origin +#endif + ); } - void expand(); +#endif + //} +}; +#include "pa_pragma_pack_end.h" +/// useful macro for creating objects on current Pooled object Pooled::pool() +#define NEW new(pool()) +*/ + + +template class object_ptr { + T *ptr; public: + typedef T element_type; - size_t size(); - char *c_str(); - String& operator += (char *src); + explicit object_ptr(T *ptr = 0) { + this->ptr=ptr; + if(ptr) + ptr->ref(); + } + template object_ptr(const object_ptr& src) { + this->ptr=src.get(); + if(ptr) + ptr->ref(); + } + object_ptr(const object_ptr& src) { + ptr=src.get(); + ptr->ref(); + } + object_ptr(T *ptr, int) { + this->ptr=ptr; + } + object_ptr& operator=(const object_ptr& src) { + if(this!=&src) + if(ptr!=src.get()) { + if(ptr) + ptr->unref(); + ptr=src.get(); + if(ptr) + ptr->ref(); + } + return *this; + } + ~object_ptr() { + if(ptr) + ptr->unref(); + } + T& operator*() const { + return *get(); + } + T *operator->() const { + return get(); + } + T *get() const { + return ptr; + } + operator bool() const { + return get()!=0; + } + bool operator !() const { + return get()==0; + } }; -class Pool { +#define DECLARE_OBJECT_PTR(name) \ + typedef object_ptr name##Ptr + +// typedef object_ptr name##Ptr; \ +// const object_ptr name##PtrZero(0); + +/// TEMPLATE CLASS smart_ptr, stolen from stl:auto_ptr & renamed field/params [not compiled on gcc] +template + class smart_ptr { + bool owns; + T *ptr; public: - Pool(); - ~Pool(); - void *alloc(size_t size); - void *calloc(size_t size); + typedef T element_type; + explicit smart_ptr(T *aptr = 0) + : owns(aptr != 0), ptr(aptr) {} + smart_ptr(const smart_ptr& src) + : owns(src.owns), ptr(src.release()) {} + smart_ptr& operator=(const smart_ptr& src) + {if (this != &src) + {if (ptr != src.get()) + {if (owns) + delete ptr; + owns = src.owns; } + else if (src.owns) + owns = true; + ptr = src.release(); } + return (*this); } + ~smart_ptr() + {if (owns) + delete ptr; } + T& operator*() const + {return (*get()); } + T *get() const + {return ptr; } + T *release() const + {((smart_ptr *)this)->owns = false; + return (ptr); } + operator bool() const { + return get()!=0; + } + operator T*() const { + return ptr; + } + }; + +// defines + +#define override +#define rethrow throw + +// memory - String *makeString() { - return new(this) String(this); +inline void *operator new[] (size_t size) { + return pa_malloc(size); +} +inline void operator delete[] (void *ptr) { + pa_free(ptr); +} + +class PA_Allocated { +public: + /// the sole: instances allocated using our functions + static void *operator new(size_t size) { + return pa_malloc(size); + } + static void operator delete(void *ptr) { + pa_free(ptr); + } + static void *malloc(size_t size) { + return pa_malloc(size); } - String *makeString(int chunk_items) { - return new(this) String(this, chunk_items); + static void *calloc(size_t size) { + return pa_calloc(size); } - String *makeString(char *src) { - return new(this) String(this, src); + static void *realloc(void *ptr, size_t size) { + return pa_realloc(ptr, size); } + }; +/** + Base for all Parser classes, memory allocation/dallocation goes via pa_malloc/pa_free. +*/ +class PA_Object: public PA_Allocated { + mutable unsigned long freferences; +public: + PA_Object(): freferences(0) {} + virtual ~PA_Object() {} + + void ref() const { + freferences++; + } + void unref() const { + if(freferences) { + if(--freferences==0) + delete this; + } + } +}; +DECLARE_OBJECT_PTR(PA_Object); + +// convinient types + +typedef smart_ptr CharPtr; +//const CharPtr CharPtrZero(0); + #endif