#ifndef PrimitiveSet_h
#define PrimitiveSet_h
#include <vector>
#include "Primitive.h"
#include "Terminal.h"

class PrimitiveSet
{
	friend class Tree;
public:
	PrimitiveSet(void);
	void addFunction(PrimitiveP functionPrimitive);
	void addTerminal(PrimitiveP terminalPrimitive);
	PrimitiveP getRandomTerminal();
	PrimitiveP getRandomFunction();
	PrimitiveP getRandomPrimitive();
	virtual ~PrimitiveSet(void);
	uint getFunctionSetSize();
	uint getTerminalSetSize();

	std::vector<PrimitiveP> terminalSet; // actual used terminals
	std::map<std::string, PrimitiveP> mPrimitives_;	// map of all registered primitive functions
	StateP state_;
	std::map <std::string, terminal_type> mTypeNames_;
private:
	std::vector<PrimitiveP> functionSet;	// actual used functions
	std::vector<PrimitiveP> primitives;		// used functions and terminals
};
typedef boost::shared_ptr<PrimitiveSet> PrimitiveSetP;

#endif