--- a/src/databuffer.h Fri Jan 10 21:58:42 2014 +0200 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,264 +0,0 @@ -#ifndef BOTC_DATABUFFER_H -#define BOTC_DATABUFFER_H -#include <stdio.h> -#include <string.h> -#include "main.h" -#include "stringtable.h" - -#define MAX_MARKS 512 - -extern int g_NextMark; - -// ============================================================================ -// DataBuffer: A dynamic data buffer. -class DataBuffer { -public: - // The actual buffer - byte* buffer; - - // Allocated size of the buffer - unsigned int allocsize; - - // Written size of the buffer - unsigned int writesize; - - // Marks and references - ScriptMark* marks[MAX_MARKS]; - ScriptMarkReference* refs[MAX_MARKS]; - - // ==================================================================== - // METHODS - - // ==================================================================== - // Constructor - DataBuffer (unsigned int size=128) { - writesize = 0; - - buffer = new unsigned char[size]; - allocsize = size; - - // Clear the marks table out - for (unsigned int u = 0; u < MAX_MARKS; u++) { - marks[u] = null; - refs[u] = null; - } - } - - // ==================================================================== - ~DataBuffer () { - delete buffer; - - // Delete any marks and references - for (unsigned int u = 0; u < MAX_MARKS; u++) { - if (marks[u]) - delete marks[u]; - - if (refs[u]) - delete refs[u]; - } - } - - // ==================================================================== - // Write stuff to the buffer - template<class T> void Write (T stuff) { - if (writesize + sizeof (T) >= allocsize) { - // We don't have enough space in the buffer to write - // the stuff - thus resize. First, store the old - // buffer temporarily: - char* copy = new char[allocsize]; - memcpy (copy, buffer, allocsize); - - // Remake the buffer with the new size. Have enough space - // for the stuff we're going to write, as well as a bit - // of leeway so we don't have to resize immediately again. - size_t newsize = allocsize + sizeof (T) + 128; - - delete buffer; - buffer = new unsigned char[newsize]; - allocsize = newsize; - - // Now, copy the stuff back. - memcpy (buffer, copy, allocsize); - delete copy; - } - - // Buffer is now guaranteed to have enough space. - // Write the stuff one byte at a time. - union_t<T> uni; - uni.val = stuff; - for (unsigned int x = 0; x < sizeof (T); x++) { - if (writesize >= allocsize) // should NEVER happen because resizing is done above - error ("DataBuffer: written size exceeds allocated size!\n"); - - buffer[writesize] = uni.b[x]; - writesize++; - } - } - - // ==================================================================== - // Merge another data buffer into this one. - void Merge (DataBuffer* other) { - if (!other) - return; - int oldsize = writesize; - - for (unsigned int x = 0; x < other->writesize; x++) - Write (*(other->buffer+x)); - - // Merge its marks and references - unsigned int u = 0; - for (u = 0; u < MAX_MARKS; u++) { - if (other->marks[u]) { - // Merge the mark and offset its position. - if (marks[u]) - error ("DataBuffer: duplicate mark %d!\n"); - - marks[u] = other->marks[u]; - marks[u]->pos += oldsize; - - // The original mark becomes null so that the deconstructor - // will not delete it prematurely. (should it delete any - // marks in the first place since there is no such thing - // as at temporary mark?) - other->marks[u] = null; - } - - if (other->refs[u]) { - // Same for references - // TODO: add a g_NextRef system like here, akin to marks! - unsigned int r = AddMarkReference (other->refs[u]->num, false); - refs[r]->pos = other->refs[u]->pos + oldsize; - } - } - - delete other; - } - - // Clones this databuffer to a new one and returns it. - DataBuffer* Clone () { - DataBuffer* other = new DataBuffer; - for (unsigned int x = 0; x < writesize; x++) - other->Write (*(buffer+x)); - return other; - } - - // ==================================================================== - // Adds a mark to the buffer. A mark is a "pointer" to a particular - // position in the bytecode. The actual permanent position cannot - // be predicted in any way or form, thus these things will be used - // to "mark" a position like that for future use. - unsigned int AddMark (string name) { - // Find a free slot for the mark - unsigned int u = g_NextMark++; - - if (marks[u]) - error ("DataBuffer: attempted to re-create mark %u!\n", u); - - if (u >= MAX_MARKS) - error ("mark quota exceeded, all labels, if-structs and loops add marks\n"); - - ScriptMark* m = new ScriptMark; - m->name = name; - m->pos = writesize; - marks[u] = m; - return u; - } - - // ==================================================================== - // A ref is another "mark" that references a mark. When the bytecode - // is written to file, they are changed into their marks' current - // positions. Marks themselves are never written to files, only refs are - unsigned int AddMarkReference (unsigned int marknum, bool placeholder = true) { - unsigned int u; - for (u = 0; u < MAX_MARKS; u++) - if (!refs[u]) - break; - - if (u == MAX_MARKS) - error ("mark reference quota exceeded, all goto-statements, if-structs and loops add refs\n"); - - ScriptMarkReference* r = new ScriptMarkReference; - r->num = marknum; - r->pos = writesize; - refs[u] = r; - - // Write a dummy placeholder for the reference - if (placeholder) - Write (1234); - - return u; - } - - // Delete a mark and all references to it. - void DeleteMark (unsigned int marknum) { - if (!marks[marknum]) - return; - - // Delete the mark - delete marks[marknum]; - marks[marknum] = null; - - // Delete its references - for (unsigned int u = 0; u < MAX_MARKS; u++) { - if (refs[u]->num == marknum) { - delete refs[u]; - refs[u] = null; - } - } - } - - // Adjusts a mark to the current position - void MoveMark (unsigned int mark) { - if (!marks[mark]) - return; - - marks[mark]->pos = writesize; - } - - void OffsetMark (unsigned int mark, int offset) { - if (!marks[mark]) - return; - - marks[mark]->pos += offset; - } - - // Dump the buffer (for debugging purposes) - void Dump() { - for (unsigned int x = 0; x < writesize; x++) - printf ("%d. [%d]\n", x, *(buffer+x)); - } - - // Count the amount of marks - unsigned int CountMarks () { - unsigned int count = 0; - for (unsigned int u = 0; u < MAX_MARKS; u++) - count += !!marks[u]; - return count; - } - - // Count the amount of refs - unsigned int CountReferences () { - unsigned int count = 0; - for (unsigned int u = 0; u < MAX_MARKS; u++) - count += !!refs[u]; - return count; - } - - // Write a float into the buffer - void WriteFloat (string floatstring) { - // TODO: Casting float to word causes the decimal to be lost. - // Find a way to store the number without such loss. - float val = atof (floatstring); - Write (DH_PUSHNUMBER); - Write (static_cast<word> ((val > 0) ? val : -val)); - if (val < 0) - Write (DH_UNARYMINUS); - } - - void WriteString (string a) { - Write (DH_PUSHSTRINGINDEX); - Write (get_string_table_index (a)); - } -}; - -#endif // BOTC_DATABUFFER_H \ No newline at end of file