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();
 
    std::map <std::string, PrimitiveP> mAllPrimitives_;                 
    std::map <std::string, Primitives::terminal_type> mTypeNames_;
 
protected:
    StateP state_;
 
    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