PattRecoTree.h

Go to the documentation of this file.

#ifndef DataFormats_JetReco_PattRecoTree_h
#define DataFormats_JetReco_PattRecoTree_h

#include "DataFormats/JetReco/interface/PattRecoNode.h"

#include <vector>

namespace reco {
    template<typename ScaleType, class Cluster>
    class PattRecoTree
    {
    public:
        typedef PattRecoNode<Cluster> Node;

        inline PattRecoTree() : sparse_(false) {}

        // Inspectors
        inline bool isSparse() const {return sparse_;}
        inline const std::vector<Node>& getNodes() const {return nodes_;}
        inline const std::vector<ScaleType>& getScales() const
        {return scales_;}

        // Modifiers
        inline void setSparse(const bool b) {sparse_ = b;}

        inline void clear()
        {nodes_.clear(); scales_.clear(); sparse_ = false;}

        inline void reserveNodes(const unsigned n) {nodes_.reserve(n);}
        inline void reserveScales(const unsigned n) {scales_.reserve(n);}
        inline void addNode(const Node& node) {nodes_.push_back(node);}
        inline void addScale(const double s)
        {scales_.push_back(static_cast<ScaleType>(s));}

    private:
        std::vector<Node> nodes_;
        std::vector<ScaleType> scales_;
        bool sparse_;
    };
}

#endif // JetReco_PattRecoTree_h