TagAndProbe/interface/TPTreeFiller.h

0001 #ifndef PhysicsTools_TagAndProbe_TreeFiller_h
0002 #define PhysicsTools_TagAndProbe_TreeFiller_h
0003
0004 #include "PhysicsTools/TagAndProbe/interface/BaseTreeFiller.h"
0005
0006 namespace tnp {
0007   class TPTreeFiller : public BaseTreeFiller {
0008   public:
0009     TPTreeFiller(const edm::ParameterSet& config, edm::ConsumesCollector&& iC) : TPTreeFiller(config, iC){};
0010     TPTreeFiller(const edm::ParameterSet& config, edm::ConsumesCollector& iC);
0011     ~TPTreeFiller();
0012
0013     // We declare 'const' the methods which don't change the configuration
0014     void init(const edm::Event& iEvent) const;
0015     void fill(const reco::CandidateBaseRef& probe, double mass, bool mcTrue = false, float mcMass = 0.0) const;
0016
0017   protected:
0018     /// extra branch for the mass
0019     mutable float mass_;
0020     /// extra branch for the mc truth
0021     mutable int32_t mcTrue_;
0022     /// extra branch for the mc-truth mass
0023     mutable float mcMass_;
0024   };
0025 }  // namespace tnp
0026
0027 #endif