Tree.h

Go to the documentation of this file.

// Tree.h

#ifndef ADD_TREE
#define ADD_TREE

#include <list>
#include "L1Trigger/L1TMuonEndCap/interface/Node.h"
#include "TXMLEngine.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"


//class Node;

class Tree
{
    public:
        Tree();
        Tree(std::vector< std::vector<Event*> >& cEvents);
        ~Tree();

        void setRootNode(Node *sRootNode);
        Node* getRootNode();

        void setTerminalNodes(std::list<Node*>& sTNodes);
        std::list<Node*>& getTerminalNodes();

        Int_t getNumTerminalNodes();

        void buildTree(Int_t nodeLimit);
        void calcError();
        void filterEvents(std::vector<Event*>& tEvents);
        void filterEventsRecursive(Node* node);
        Node* filterEvent(Event* e);
        Node* filterEventRecursive(Node* node, Event* e);

        void saveToXML(const char* filename);
        void saveToXMLRecursive(TXMLEngine* xml, Node* node, XMLNodePointer_t np);
        void addXMLAttributes(TXMLEngine* xml, Node* node, XMLNodePointer_t np);

        void loadFromXML(const char* filename);
        void loadFromXMLRecursive(TXMLEngine* xml, XMLNodePointer_t node, Node* tnode);

        void rankVariables(std::vector<Double_t>& v);
        void rankVariablesRecursive(Node* node, std::vector<Double_t>& v);

        void getSplitValues(std::vector<std::vector<Double_t>>& v);
        void getSplitValuesRecursive(Node* node, std::vector<std::vector<Double_t>>& v);

    private:
        Node *rootNode;
        std::list<Node*> terminalNodes;
        Int_t numTerminalNodes;
        Double_t rmsError;
};

#endif