--- deliantra/server/include/dynbuf.h 2006/09/19 22:05:55 1.3 +++ deliantra/server/include/dynbuf.h 2009/10/12 14:00:58 1.22 @@ -1,69 +1,128 @@ +/* + * This file is part of Deliantra, the Roguelike Realtime MMORPG. + * + * Copyright (©) 2005,2006,2007,2008 Marc Alexander Lehmann / Robin Redeker / the Deliantra team + * Copyright (©) 2002,2007 Mark Wedel & Crossfire Development Team + * Copyright (©) 1992,2007 Frank Tore Johansen + * + * Deliantra is free software: you can redistribute it and/or modify it under + * the terms of the Affero GNU General Public License as published by the + * Free Software Foundation, either version 3 of the License, or (at your + * option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the Affero GNU General Public License + * and the GNU General Public License along with this program. If not, see + * . + * + * The authors can be reached via e-mail to + */ + #ifndef DYNBUF_H__ #define DYNBUF_H__ #include #include +#include "util.h" +#include "shstr.h" + // this is a "buffer" that can grow fast // and is still somewhat space-efficient. // unlike obstacks or other data structures, // it never moves data around. basically, // this is a fast strstream without the overhead. -class dynbuf +struct dynbuf { +protected: struct chunk { chunk *next; + int alloc; int size; char data[0]; }; - char *ptr; - int room; - int ext; + char *ptr, *end; int _size; + int extend, cextend; chunk *first, *last; - void _reserve (int size); - void clear (); - void finish (); + void reserve (int size); + void init (int initial); // allocate sinitial chunk + void free (chunk *&chain); // free chain of chunks + char *_linearise (); + void finalise (); public: - dynbuf (int initial = 4096, int extend = 16384); - ~dynbuf (); + // initial - the size of the initial chunk to be allocated + // extend - first incremental step when buffer size exceeded + dynbuf (int initial = 4096, int extend = 16384) + : extend (extend) + { + init (initial); + } + + ~dynbuf () + { + free (first); + } + + // resets the dynbuf, but does not free the first chunk + // which is either of size "initial" or the size of the last + // linearise + void clear (); - int size () { return _size + (ptr - last->data); } + int size () const { return _size + (ptr - last->data); } + bool empty () const { return !size (); } void linearise (void *data); - char *linearise (); + char *linearise () // does not 0-terminate(!) + { + return first->next ? _linearise () : first->data; + } + + int room () const { return end - ptr; } char *force (int size) { - if (room < size) - _reserve (size); + if (expect_false (ptr + size >= end)) + reserve (size); return ptr; } - char *alloc (int size) + char *falloc (int size) { - char *res = force (size); - - room -= size; + char *res = ptr; ptr += size; - return res; } - void fadd (char c) { --room; *ptr++ = c; } + char *alloc (int size) + { + force (size); + return falloc (size); + } + + void fadd (char c) { *ptr++ = c; } void fadd (unsigned char c) { fadd (char (c)); } + void fadd (const void *p, int len) + { + memcpy (falloc (len), p, len); + } void add (const void *p, int len) { - memcpy (alloc (len), p, len); + force (len); + fadd (p, len); } void add (char c) @@ -76,11 +135,10 @@ add (s, strlen (s)); } - static const int max_sint32_size = 11; - static const int max_sint64_size = 20; - - void add (sint32 i); - void add (sint64 i); + void add (shstr_tmp s) + { + add (s.s, s.length ()); + } //TODO //void add_destructive (dynbuf &buf); @@ -88,6 +146,45 @@ dynbuf &operator << (char c) { add (c); return *this; } dynbuf &operator << (unsigned char c) { return *this << char (c); } dynbuf &operator << (const char *s) { add (s); return *this; } + dynbuf &operator << (shstr_tmp s) { add (s); return *this; } + dynbuf &operator << (const std::string &s) { add (s.data(), s.size ()); return *this; } + + operator std::string (); +}; + +struct dynbuf_text : dynbuf +{ + dynbuf_text (int initial = 4096, int extend = 16384) + : dynbuf (initial, extend) + { } + + using dynbuf::add; + void add (sint32 i); + void add (sint64 i); + + //TODO: should optimise the case printf "(name %+d)" as it comes up extremely often + + //using dynbuf::operator <<; // doesn't work, sometimes C++ just suxx + // instead we use an ugly template function + template + dynbuf_text &operator << (T c) { *(dynbuf *)this << c; return *this; } + + dynbuf_text &operator << (sint16 i) { add (sint32 (i)); return *this; } + dynbuf_text &operator << (uint16 i) { add (sint32 (i)); return *this; } + dynbuf_text &operator << (sint32 i) { add (sint32 (i)); return *this; } + dynbuf_text &operator << (sint64 i) { add (sint64 (i)); return *this; } + dynbuf_text &operator << (uint32 i) { add (sint64 (i)); return *this; } + dynbuf_text &operator << (uint64 i) { add (sint64 (i)); return *this; } + + void printf (const char *format, ...); + void vprintf (const char *format, va_list ap); + + void add_abilities (const char *name, uint32 abilities); + void add_paths (const char *name, uint32 paths); + + // returns the string, linearised and with trailing \0 + operator const char * (); }; #endif +