#include "PipeLine.h"

PipeMaterial::PipeMaterial() : buffer(nullptr), buffSize(0), former(nullptr)
{
}

PipeLine::PipeLine() : global_params_map(), elem_create_func_map(), elems()
{
}

PipeLine::~PipeLine()
{
	// pipe stop
	
	for(elem_vec_t::iterator iter = elems.begin(); iter != elems.end(); ++iter)
	{
		PipeLineElem* elem = *iter;
		if (elem != nullptr)
		{
			elem->finit();
			delete *iter;
		}
	}
	
	elems.clear();
}

bool PipeLine::register_elem_creator(const std::string& type, elem_create_func_t func)
{
	if (type.empty() || func == nullptr)
		return false;

	elem_create_func_map_t::iterator iter = elem_create_func_map.find(type);
	if (iter != elem_create_func_map.end())
		return false;
	
	elem_create_func_map.insert(std::make_pair(type, func));
	return true;
}

void PipeLine::push_elem(PipeLineElem* elem)
{
	if(elem != nullptr)
	{
		elem->manager = this;
		elems.push_back(elem);
	}
}

PipeLineElem* PipeLine::push_elem(const std::string& type)
{
	elem_create_func_map_t::iterator iter = elem_create_func_map.find(type);
	if (iter == elem_create_func_map.end())
		return nullptr;
	
	elem_create_func_t func = iter->second;
	if (func == nullptr)
		return nullptr;
	
	PipeLineElem* elem = func();
	if (elem == nullptr)
		return nullptr;
	
	elem->manager = this;
	push_elem(elem);
	return elem;
}

PipeLineElem* PipeLine::pipe(PipeMaterial* pm /*= nullptr*/)
{
	PipeLineElem* elem_begin = *elems.begin();
	PipeLineElem* elem_last = *elems.rbegin();

	if (elems.empty() || elem_begin == nullptr || elem_last == nullptr)
		return nullptr;

	uint8_t pmPlacement[sizeof(PipeMaterial)];
	if (pm == nullptr)
		pm = new (pmPlacement) PipeMaterial;

	if (elems.size() == 1)
	{
		elem_begin->gain(*pm);
		return elem_begin;
	}
	else if (elems.size() == 2)
	{
		if (elem_begin->gain(*pm))
			elem_last->pay(*pm);
		else
			return elem_begin;
		return elem_last;
	}
	else
	{
		if (!elem_begin->gain(*pm))
			return elem_begin;
		
		bool lastRet = true;
		elem_vec_t::iterator iter = elems.begin();
		++iter;
		elem_begin = *iter;
		while (elem_begin != elem_last)
		{
			if (lastRet && (lastRet = elem_begin->pay(*pm)) )
				lastRet = elem_begin->gain(*pm);
			else
				return elem_begin;
			
			++iter;
			elem_begin = *iter;
		}
	
		if (lastRet)
			elem_last->pay(*pm);
		return elem_last;
	}
	
	return nullptr;
}

void PipeLine::set_global_param(const std::string& name, const std::string& value)
{
	if (name.empty())
		return;

	global_params_map_t::iterator iter = global_params_map.find(name);
	if (iter == global_params_map.end())
		global_params_map.insert(std::make_pair(name, value));
	else
		iter->second = value;
}

std::string PipeLine::get_global_param(const std::string& name) const
{
	global_params_map_t::const_iterator iter = global_params_map.find(name);
	if (iter == global_params_map.end())
		return "";
	else
		return iter->second;
}