string.h

// The core library - copyright GarageGames.  The core library is released under the MIT Open Source software license.  See /license.txt in this distribution for specific license terms.

class string
{
protected:
	struct string_record
	{
		uint32 ref_count;
		uint32 len;
		char data[1];
	};

	static string_record *_get_null_string_record()
	{
		static string_record null_string = { 0x7FFFFFFF, 0,  { 0 } };
		return &null_string;
	}
	
	string_record *_string_record;
	void _dec_ref()
	{
		if(!--_string_record->ref_count && _string_record != _get_null_string_record())
			memory_deallocate(_string_record);
	}
public:
	string()
	{
		_string_record = _get_null_string_record();
		_string_record->ref_count++;
	}
	
	~string()
	{
		_dec_ref();
	}
	string(const char *string_data)
	{
		_string_record = _get_null_string_record();
		_string_record->ref_count++;
		set(string_data);
	}
	
	string(const char *string_data, uint32 len)
	{
		_string_record = _get_null_string_record();
		_string_record->ref_count++;
		set(string_data, len);		
	}
	
	string(const unsigned char *data, uint32 len)
	{
		_string_record = _get_null_string_record();
		_string_record->ref_count++;
		set(data, len);		
	}
	
	string(const string &the_string)
	{
		_string_record = the_string._string_record;
		_string_record->ref_count++;
	}
	
	string& operator=(const string &the_string)
	{
		_dec_ref();
		_string_record = the_string._string_record;
		_string_record->ref_count++;
		return *this;
	}
	void set(const char *string_data)
	{
		uint32 len = strlen(string_data);
		set(string_data, len);
	}
	
	void set(const char *string_data, uint32 len)
	{
		reserve(len);
		memcpy(_string_record->data, string_data, len);
	}
	void reserve(uint32 len)
	{
		_dec_ref();
		_string_record = (string_record *) memory_allocate(sizeof(string_record) + len);
		_string_record->len = len;
		_string_record->data[len] = 0; // ensure null termination after buffer length
		_string_record->ref_count = 1;

	}

	void set(const unsigned char *string_data, uint32 len)
	{
		set((const char *) string_data, len);
	}
	
	operator const char *()
	{
		return _string_record->data;
	}
	const char *c_str()
	{
		return _string_record->data;
	}
	uint8 *data()
	{
		return (uint8 *) _string_record->data;
	}
	const uint8 *data() const
	{
		return (uint8 *) _string_record->data;
	}
	uint32 len() const
	{
		return _string_record->len;
	}
};

inline string format_string(const char *fmt, ...)
{
	formatted_string_buffer buf;
	
	va_list args;
	va_start(args, fmt);
	uint32 len = buf.format(fmt, args);
	
	string return_value;
	return_value.set(buf.c_str(), len);
	
	return return_value;
}