#ifndef NEURAL_CONTAINER_H
#define NEURAL_CONTAINER_H

#include "neural/node.h"
#include <bu/list.h>

namespace Neural
{
	template<typename sigtype>
	class Container : public Node<sigtype>
	{
	public:
		typedef Bu::List<Neural::Node<sigtype> *> NodeList;

		Container()
		{
		}

		virtual ~Container()
		{
			for( typename NodeList::iterator i = lNodes.begin(); i; i++ )
				delete *i;
		}

		virtual int setWeights( const sigtype *pWeights )
		{
			int iOffset = 0;
			for( typename Container<sigtype>::NodeList::iterator i =
				Container<sigtype>::getNodeList().begin(); i; i++ )
			{
				iOffset += (*i)->setWeights( &pWeights[iOffset] );
			}

			return iOffset;
		}

		virtual int setBiases( const sigtype *pBiases )
		{
			int iOffset = 0;
			for( typename Container<sigtype>::NodeList::iterator i =
				Container<sigtype>::getNodeList().begin(); i; i++ )
			{
				iOffset += (*i)->setBiases( &pBiases[iOffset] );
			}

			return iOffset;
		}

		virtual void addNode( Node<sigtype> *pNode )
		{
			lNodes.append( pNode );
		}

		virtual const NodeList &getNodeList() const
		{
			return lNodes;
		}

	protected:
		virtual NodeList &getNodeList()
		{
			return lNodes;
		}

	private:
		NodeList lNodes;
	};
};

#endif