d/source/zap/MessageBuilder.d

// Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors
// Licensed under the MIT License:
//
// 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.

module zap.MessageBuilder;

import java.nio.ByteBuffer;

import zap.AnyPointer;
import zap.BuilderArena;
import zap.ReaderArena;
import zap.ReaderOptions;
import zap.SegmentBuilder;

final class MessageBuilder
{
public: //Methods.
	this()
	{
		this.arena = new BuilderArena(BuilderArena.SUGGESTED_FIRST_SEGMENT_WORDS, BuilderArena.SUGGESTED_ALLOCATION_STRATEGY);
	}
	
	this(int firstSegmentWords)
	{
		this.arena = new BuilderArena(firstSegmentWords, BuilderArena.SUGGESTED_ALLOCATION_STRATEGY);
	}
	
	this(int firstSegmentWords, BuilderArena.AllocationStrategy allocationStrategy)
	{
		this.arena = new BuilderArena(firstSegmentWords, allocationStrategy);
	}
	
	T.Builder getRoot(T)()
	{
		return this.getRootInternal().getAs!T();
	}
	
	void setRoot(T, U)(U reader)
	{
		this.getRootInternal().setAs!T(reader);
	}
	
	T.Builder initRoot(T)()
	{
		return this.getRootInternal().initAs!T();
	}
	
	ByteBuffer[] getSegmentsForOutput()
	{
		return this.arena.getSegmentsForOutput();
	}

private: //Methods.
	AnyPointer.Builder getRootInternal()
	{
		auto rootSegment = &this.arena.segments.data[0];
		if(rootSegment.currentSize() == 0)
		{
			auto location = rootSegment.allocate(1);
			if(location == SegmentBuilder.FAILED_ALLOCATION)
				throw new Error("Could not allocate root pointer.");
			if(location != 0)
				throw new Error("First allocated word of new segment was not at offset 0.");
			return AnyPointer.Builder(rootSegment, cast(int)location);
		}
		return AnyPointer.Builder(rootSegment, 0);
	}

private: //Variables.
	BuilderArena arena;
}