PrimitiveSetAP.h

#ifndef PrimitiveSetAP_h
#define PrimitiveSetAP_h
#include <vector>
#include "../ECF_base.h"
#include "../tree/Primitive.h"
 
 
namespace Tree
{
 
    class PrimitiveSetAP
    {
        
    public:
        PrimitiveSetAP();
        virtual ~PrimitiveSetAP();
 
        bool initialize(StateP state);
 
        //void addFunction(PrimitiveP functionPrimitive);
        bool addFunction(std::string name);
        void addTerminal(PrimitiveP terminalPrimitive);
 
        PrimitiveP getTerminalByName(std::string name);
        PrimitiveP getGFSOneByName(std::string name);
        PrimitiveP getGFSTwoByName(std::string name);
        PrimitiveP getPrimitiveByName(std::string name);
 
        PrimitiveP getTerminalByIndex(uint index);
        PrimitiveP getGFSOneByIndex(uint index);
        PrimitiveP getGFSTwoByIndex(uint index);
        PrimitiveP getPrimitiveByIndex(uint index);
 
        uint getGFSOneSetSize();
        uint getGFSTwoSetSize();
        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> GFSOneSet_;                               
        std::map <std::string, PrimitiveP> mGFSOneSet_;                   
        std::vector<PrimitiveP> GFSTwoSet_;                               
        std::map <std::string, PrimitiveP> mGFSTwoSet_;                   
        std::vector<PrimitiveP> primitives_;                                
        std::map <std::string, PrimitiveP> mPrimitiveSet_;                  
    };
    typedef boost::shared_ptr<PrimitiveSetAP> PrimitiveSetAPP;
 
    typedef std::map <std::string, PrimitiveP>::iterator prim_iter;
    typedef std::map <std::string, Primitives::terminal_type>::iterator type_iter;
 
}
#endif