BaseTreeFiller.h

Go to the documentation of this file.

#ifndef PhysicsTools_TagAndProbe_BaseTreeFiller_h
#define PhysicsTools_TagAndProbe_BaseTreeFiller_h

#include <vector>
#include <string>
#include <cstdint>

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "DataFormats/PatCandidates/interface/MET.h"
#include "DataFormats/METReco/interface/MET.h"
#include "DataFormats/METReco/interface/METCollection.h"
#include "DataFormats/METReco/interface/CaloMET.h"
#include "DataFormats/METReco/interface/CaloMETCollection.h"
#include "DataFormats/METReco/interface/PFMET.h"
#include "DataFormats/METReco/interface/PFMETCollection.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "CommonTools/Utils/interface/StringCutObjectSelector.h"
#include "CommonTools/Utils/interface/StringObjectFunction.h"

#include "SimDataFormats/PileupSummaryInfo/interface/PileupSummaryInfo.h"
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"

#include <TTree.h>
#include <boost/utility.hpp>

namespace tnp {

  class ProbeVariable {
  public:
    ProbeVariable(const std::string &name, const std::string &expression)
        : name_(name), external_(false), function_(expression) {}

    ProbeVariable(const std::string &name, const edm::EDGetTokenT<edm::ValueMap<float> > &srcToken)
        : name_(name), external_(true), function_("-1"), srcToken_(srcToken) {}

    ~ProbeVariable();

    float *address() const { return &value_; }

    const std::string &name() const { return name_; }

    void init(const edm::Event &iEvent) const {
      if (external_)
        iEvent.getByToken(srcToken_, handle_);
    }

    void fill(const reco::CandidateBaseRef &probe) const { value_ = external_ ? (*handle_)[probe] : function_(*probe); }

  private:
    std::string name_;
    mutable float value_;

    bool external_;
    // ---- this below is used if 'external_' is false
    StringObjectFunction<reco::Candidate, true> function_;
    // In releases older than 3.4.X, it can be parially emulated by PATStringObjectFunction (PhysicsTools/PatUtils/interface/StringParserTools.h)
    // Or you can use StringObjectFunction<reco::Candidate> and get only reco::Candidate methods

    // ---- this below is used if 'external_' is true
    edm::EDGetTokenT<edm::ValueMap<float> > srcToken_;
    mutable edm::Handle<edm::ValueMap<float> > handle_;
  };

  class ProbeFlag {
  public:
    ProbeFlag(const std::string &name, const std::string &cut) : name_(name), external_(false), cut_(cut) {}

    ProbeFlag(const std::string &name, const edm::EDGetTokenT<edm::View<reco::Candidate> > &srcToken)
        : name_(name), external_(true), cut_(""), srcToken_(srcToken) {}

    ~ProbeFlag();

    int32_t *address() const { return &value_; }

    const std::string &name() const { return name_; }

    void init(const edm::Event &iEvent) const;

    void fill(const reco::CandidateBaseRef &probe) const;

  private:
    std::string name_;
    mutable int32_t value_;

    bool external_;
    // ---- this below is used if 'external_' is false
    StringCutObjectSelector<reco::Candidate, true> cut_;

    // ---- this below is used if 'external_' is true
    edm::EDGetTokenT<edm::View<reco::Candidate> > srcToken_;
    mutable std::vector<reco::CandidateBaseRef> passingProbes_;
  };

  // This class inherits from boost::noncopyable, as copying it would break the addresses in the TTree
  class BaseTreeFiller : boost::noncopyable {
  public:
    BaseTreeFiller(const char *name, const edm::ParameterSet &config, edm::ConsumesCollector &&iC)
        : BaseTreeFiller(name, config, iC){};
    BaseTreeFiller(const char *name, const edm::ParameterSet &config, edm::ConsumesCollector &iC);

    BaseTreeFiller(BaseTreeFiller &main,
                   const edm::ParameterSet &iConfig,
                   edm::ConsumesCollector &&iC,
                   const std::string &branchNamePrefix);

    ~BaseTreeFiller();

    //implementation notice: we declare 'const' the methods which don't change the configuration
    //                       and that can't mess up the addresses in the TTree.

    void init(const edm::Event &iEvent) const;

    void fill(const reco::CandidateBaseRef &probe) const;

    void writeProvenance(const edm::ParameterSet &pset) const;

    //get the pileup weight informations
    bool storePUweight() const { return storePUweight_; };

  protected:
    std::vector<ProbeVariable> vars_;
    std::vector<ProbeFlag> flags_;

    enum WeightMode { None, Fixed, External };
    WeightMode weightMode_;
    edm::EDGetTokenT<GenEventInfoProduct> weightSrcToken_;
    edm::EDGetTokenT<double> PUweightSrcToken_;
    edm::EDGetTokenT<double> rhoToken_;
    edm::EDGetTokenT<reco::VertexCollection> recVtxsToken_;
    edm::EDGetTokenT<reco::BeamSpot> beamSpotToken_;
    edm::EDGetTokenT<reco::CaloMETCollection> metToken_;
    edm::EDGetTokenT<reco::METCollection> tcmetToken_;
    edm::EDGetTokenT<reco::PFMETCollection> pfmetToken_;
    edm::EDGetTokenT<pat::METCollection> pfmetTokenMiniAOD_;
    edm::EDGetTokenT<std::vector<PileupSummaryInfo> > pileupInfoToken_;

    bool addRunLumiInfo_;

    bool storePUweight_;

    bool addEventVariablesInfo_;
    bool addRho_;
    bool addCaloMet_;

    void addBranches_(TTree *tree,
                      const edm::ParameterSet &iConfig,
                      edm::ConsumesCollector &iC,
                      const std::string &branchNamePrefix = "");

    //implementation notice: these two are 'mutable' because we will fill them from a 'const' method
    mutable TTree *tree_;
    mutable float weight_, PUweight_, totWeight_;
    mutable uint32_t run_, lumi_, mNPV_;
    mutable uint64_t event_;
    mutable int truePU_;

    mutable float mPVx_, mPVy_, mPVz_, mBSx_, mBSy_, mBSz_;
    mutable float rho_;
    mutable float mMET_, mSumET_, mMETSign_, mtcMET_, mtcSumET_, mtcMETSign_, mpfMET_, mpfSumET_, mpfMETSign_, mpfPhi_;
  };

}  // namespace tnp

#endif