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;

namespace emtf{
  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