src/databuffer.h

#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