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_;
  };
}  // namespace reco
 
#endif  // JetReco_PattRecoTree_h