D:/Projekt/ECF_trunk/ECF/tree/PrimitiveSet.h

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


namespace Tree
{

class PrimitiveSet
{
    friend class Tree;
public:
    PrimitiveSet();
    virtual ~PrimitiveSet();

    bool initialize(StateP state);

    //void addFunction(PrimitiveP functionPrimitive);
    bool addFunction(std::string name);
    void addTerminal(PrimitiveP terminalPrimitive);

    PrimitiveP getRandomTerminal();
    PrimitiveP getRandomFunction();
    PrimitiveP getRandomPrimitive();
    
    PrimitiveP getTerminalByName(std::string name);
    PrimitiveP getFunctionByName(std::string name);
    PrimitiveP getPrimitiveByName(std::string name);
    
    uint getFunctionSetSize();
    uint getTerminalSetSize();
    uint getPrimitivesSize();

protected:
    StateP state_;

    std::map <std::string, PrimitiveP> mAllPrimitives_;                 
    std::map <std::string, Primitives::terminal_type> mTypeNames_;

    std::vector<PrimitiveP> terminalSet_;                               
    std::map <std::string, PrimitiveP> mTerminalSet_;                   
    std::vector<PrimitiveP> functionSet_;                               
    std::map <std::string, PrimitiveP> mFunctionSet_;                   
    std::vector<PrimitiveP> primitives_;                                
    std::map <std::string, PrimitiveP> mPrimitiveSet_;                  
};
typedef boost::shared_ptr<PrimitiveSet> PrimitiveSetP;

typedef std::map <std::string, PrimitiveP>::iterator prim_iter;
typedef std::map <std::string, Primitives::terminal_type>::iterator type_iter;

}
#endif