PattRecoTree

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52
/** \class reco::PattRecoTree
 *
 * \short Class for storing FFTJet sparse clustering trees
 *
 * This is a pure storage class with limited functionality.
 * Applications should use fftjet::SparseClusteringTree
 *
 * \author Igor Volobouev, TTU, June 16, 2010
 ************************************************************/

#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