Tree.h

Go to the documentation of this file.

// Tree.h

#ifndef L1Trigger_L1TMuonEndCap_emtf_Tree
#define L1Trigger_L1TMuonEndCap_emtf_Tree

#include <list>
#include "Node.h"
#include "TXMLEngine.h"
#include "CondFormats/L1TObjects/interface/L1TMuonEndCapForest.h"

namespace emtf {

//class Node;

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

        Tree(const Tree &tree);
        Tree& operator=(const Tree &tree);
        Tree(Tree && tree);

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

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

        int getNumTerminalNodes();

        void buildTree(int 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 loadFromCondPayload(const L1TMuonEndCapForest::DTree& tree);
        void loadFromCondPayloadRecursive(const L1TMuonEndCapForest::DTree& tree, const L1TMuonEndCapForest::DTreeNode& node, Node* tnode);

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

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

        double getBoostWeight(void) const   { return boostWeight; }
        void     setBoostWeight(double wgt) { boostWeight =  wgt; }

    private:
        Node *rootNode;
        std::list<Node*> terminalNodes;
        int numTerminalNodes;
        double rmsError;
        double boostWeight;
        unsigned xmlVersion; // affects only XML loading part, save uses an old format and looses the boostWeight

        // this is the main recursive workhorse function that compensates for Nodes being non-copyable
        Node* copyFrom(const Node *local_root); // no garantees if throws in the process
        // a dumb DFS tree traversal
        void findLeafs(Node *local_root, std::list<Node*> &tn);
};

} // end of emtf namespace

#endif