libRocket/Source/Core/StreamMemory.cpp

/*
 * This source file is part of libRocket, the HTML/CSS Interface Middleware
 *
 * For the latest information, see http://www.librocket.com
 *
 * Copyright (c) 2008-2010 CodePoint Ltd, Shift Technology Ltd
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 */

#include "precompiled.h"
#include "../../Include/Rocket/Core/StreamMemory.h"
#include <stdio.h>

namespace Rocket {
namespace Core {

const int DEFAULT_BUFFER_SIZE = 256;
const int BUFFER_INCREMENTS = 256;

StreamMemory::StreamMemory() 
{
	buffer = NULL;
	buffer_ptr = NULL;
	buffer_size = 0;
	buffer_used = 0;
	owns_buffer = true;
	Reallocate(DEFAULT_BUFFER_SIZE);
}

StreamMemory::StreamMemory(size_t initial_size)
{
	buffer = NULL;
	buffer_ptr = NULL;
	buffer_size = 0;
	buffer_used = 0;
	owns_buffer = true;
	Reallocate(initial_size);
}

StreamMemory::StreamMemory(const byte* _buffer, size_t _buffer_size)
{
	buffer = (byte*)_buffer;
	buffer_size = _buffer_size;
	buffer_used = _buffer_size;
	owns_buffer = false;
	buffer_ptr = buffer;	
}

StreamMemory::StreamMemory(const StreamMemory& copy) : Stream(copy)
{
	buffer = NULL;
	buffer_ptr = NULL;
	buffer_size = 0;
	buffer_used = 0;
	owns_buffer = true;
	
	// Copy the buffer and pointer offsets
	Reallocate( ( ( copy.buffer_used + BUFFER_INCREMENTS ) / BUFFER_INCREMENTS ) * BUFFER_INCREMENTS );
	ROCKET_ASSERTMSG(buffer != NULL, "Could not allocate buffer for StreamMemory reader.");
	if(buffer != NULL)
	{
		memcpy( buffer, copy.buffer, copy.buffer_used );
		buffer_ptr = buffer + ( copy.buffer_ptr - copy.buffer );
	}
}

StreamMemory::~StreamMemory() 
{
	if ( owns_buffer )
		free( buffer );
}

StreamMemory& StreamMemory::operator=( const StreamMemory& copy )
{
	// Copy the buffer and pointer offsets
	Reallocate( ( ( copy.buffer_used + BUFFER_INCREMENTS ) / BUFFER_INCREMENTS ) * BUFFER_INCREMENTS );
	memcpy( buffer, copy.buffer, copy.buffer_used );
	buffer_ptr = buffer + ( copy.buffer_ptr - copy.buffer );
	
	return *this;
}

void StreamMemory::Close() 
{
	Stream::Close();
}

bool StreamMemory::IsEOS() const 
{
	return buffer_ptr >= buffer + buffer_used;
}

// Get current offset
size_t StreamMemory::Tell() const 
{
	return buffer_ptr - buffer;
}

size_t StreamMemory::Length() const 
{
	return buffer_used;
}

// Read bytes from the buffer, advancing the internal pointer
size_t StreamMemory::Read(void *_buffer, size_t bytes) const
{
	bytes = Math::ClampUpper(bytes, (size_t) (buffer + buffer_used - buffer_ptr));

	memcpy(_buffer, buffer_ptr, bytes);

	buffer_ptr += bytes;

	return bytes;
}

// Read bytes from the buffer, not advancing the internal pointer
size_t StreamMemory::Peek( void *_buffer, size_t bytes ) const
{
	bytes = Math::ClampUpper(bytes, (size_t) (buffer + buffer_used - buffer_ptr));

	memcpy(_buffer, buffer_ptr, bytes);

	return bytes;
}

// Read bytes from the buffer, advancing the internal pointer
size_t StreamMemory::Write( const void *_buffer, size_t bytes ) 
{
	if ( buffer_ptr + bytes > buffer + buffer_size )
		if ( !Reallocate( bytes + BUFFER_INCREMENTS ) )
			return 0;
					
	memcpy( buffer_ptr, _buffer, bytes );

	buffer_ptr += bytes;
	buffer_used = Math::Max( (size_t)(buffer_ptr - buffer), buffer_used );
	
	return bytes;
}

// Truncate the stream to the specified length
size_t StreamMemory::Truncate( size_t bytes )
{
	if ( bytes > buffer_used )
		return 0;

	size_t old_size = buffer_used;
	buffer_used = bytes;
	buffer_ptr = buffer + bytes;
	return old_size - buffer_used;
}

// Set pointer to the specified offset
bool StreamMemory::Seek( long offset, int origin ) const
{
	byte* new_ptr = NULL;

	switch ( origin )
	{
	case SEEK_SET:	
		new_ptr = buffer + offset;
		break;
	case SEEK_END:		
		new_ptr = buffer + ( buffer_used - offset );
		break;
	case SEEK_CUR:
		new_ptr = buffer_ptr + offset;
	}

	// Check of overruns
	if ( new_ptr < buffer || new_ptr > buffer + buffer_used )
		return false;

	buffer_ptr = new_ptr;
  
	return true;
}

size_t StreamMemory::PushFront( const void* _buffer, size_t bytes )
{
	if ( buffer_used + bytes > buffer_size )
		if ( !Reallocate( bytes + BUFFER_INCREMENTS ) )
			return 0;

	memmove( &buffer[ bytes ], &buffer[ 0 ], buffer_used );
	memcpy( buffer, _buffer, bytes );
	buffer_used += bytes;
	buffer_ptr += bytes;
	return bytes;
}

size_t StreamMemory::PopFront( size_t bytes )
{
	Erase( 0, bytes );
	buffer_ptr -= bytes;
	buffer_ptr = Math::ClampLower(buffer_ptr, buffer);
	return bytes;
}

const byte* StreamMemory::RawStream() const 
{
	return buffer;
}

void StreamMemory::Erase( size_t offset, size_t bytes )
{
	bytes = Math::ClampUpper(bytes, buffer_used - offset);
	memmove(&buffer[offset], &buffer[offset + bytes], buffer_used - offset - bytes);
	buffer_used -= bytes;	
}

bool StreamMemory::IsReadReady()
{
	return !IsEOS();
}

bool StreamMemory::IsWriteReady()
{
	return true;
}

void StreamMemory::SetSourceURL(const URL& url)
{
	SetStreamDetails(url, Stream::MODE_READ | (owns_buffer ? Stream::MODE_WRITE : 0));
}

// Resize the buffer
bool StreamMemory::Reallocate( size_t size ) 
{	
	ROCKET_ASSERT( owns_buffer );
	if ( !owns_buffer )
		return false;
	
	byte *new_buffer = (byte*)realloc( buffer, buffer_size + size );
	if ( new_buffer == NULL )
		return false;

	buffer_ptr = new_buffer + ( buffer_ptr - buffer );

	buffer = new_buffer;
	buffer_size += size;
  
	return true;
}

}
}