libRocket/Source/Core/Stream.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/Stream.h"
#include <algorithm>
#include <stdio.h>

namespace Rocket {
namespace Core {

const size_t READ_BLOCK_SIZE = 1024;

Stream::Stream() : ReferenceCountable(1)
{
	stream_mode = 0;
}

Stream::~Stream()
{
}

void Stream::Close()
{
	stream_mode = 0;
}

// Returns the mode the stream was opened in.
int Stream::GetStreamMode() const
{
	return stream_mode;
}

// Returns the source url (if available)
const URL& Stream::GetSourceURL() const
{
	return url;
}

bool Stream::IsEOS() const
{
	return Tell() >= Length();
}

size_t Stream::Peek(void* buffer, size_t bytes) const
{
	size_t pos = Tell();
	size_t read = Read( buffer, bytes );
	Seek( (long)pos, SEEK_SET );
	return read;
}

// Read from one stream into another
size_t Stream::Read(Stream* stream, size_t bytes) const
{
	byte buffer[ READ_BLOCK_SIZE ];
	size_t total_bytes_read = 0;
	while (total_bytes_read < bytes)
	{
		size_t bytes_read = this->Read(buffer, Core::Math::Min(READ_BLOCK_SIZE, bytes - total_bytes_read));
		if (bytes_read < 1)
			return total_bytes_read;
		stream->Write(buffer, bytes_read);
		total_bytes_read += bytes_read;
	}
	return total_bytes_read;
}

// Read from one stream into another
size_t Stream::Read(String& string, size_t bytes) const
{
	size_t string_size = string.Length();
	string.Resize(string_size + bytes + 1);
	size_t read = Read(&string[string_size], bytes);
	string[string_size + read] = '\0';
	string.Resize(string_size + read);
	return read;
}

/// Write to this stream from another stream
size_t Stream::Write(const Stream* stream, size_t bytes)
{
	return stream->Read(this, bytes);
}

size_t Stream::Write(const char* string)
{
	return Write(string, strlen(string));
}

size_t Stream::Write(const String& string)
{
	return Write(string.CString(), string.Length());
}

// Push onto the front of the stream
size_t Stream::PushFront(const void* ROCKET_UNUSED_PARAMETER(buffer), size_t ROCKET_UNUSED_PARAMETER(bytes))
{
	ROCKET_UNUSED(buffer);
	ROCKET_UNUSED(bytes);

	ROCKET_ERRORMSG("No generic way to PushFront to a stream.");
	return false;
}

// Push onto the back of the stream
size_t Stream::PushBack(const void* buffer, size_t bytes)
{	
	size_t pos = Tell();
	Seek(0, SEEK_END);
	size_t wrote = Write(buffer, bytes);
	Seek((long)pos, SEEK_SET);
	return wrote;
}

// Push onto the front of the stream
size_t Stream::PopFront(size_t ROCKET_UNUSED_PARAMETER(bytes))
{
	ROCKET_UNUSED(bytes);

	ROCKET_ERRORMSG("No generic way to PopFront from a stream.");
	return 0;
}

// Push onto the back of the stream
size_t Stream::PopBack(size_t bytes)
{
	return Truncate(Length() - bytes);
}

// Sets the mode on the stream; should be called by a stream when it is opened.
void Stream::SetStreamDetails(const URL& _url, int _stream_mode)
{
	url = _url;
	stream_mode = _stream_mode;
}

// Deletes the stream.
void Stream::OnReferenceDeactivate()
{
	Close();
	delete this;
}

}
}