#include <TagProbeFitTreeProducer.cc>

Inheritance diagram for TagProbeFitTreeProducer:

Public Member Functions
	TagProbeFitTreeProducer (const edm::ParameterSet &)

	~TagProbeFitTreeProducer ()

Public Member Functions inherited from edm::EDAnalyzer
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzer ()

ModuleDescription const &	moduleDescription () const

std::string	workerType () const

virtual	~EDAnalyzer ()

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Member Functions
virtual void	analyze (const edm::Event &, const edm::EventSetup &) override

bool	checkMother (const reco::GenParticleRef &ref) const
	Return true if ref is not null and has an ancestor with pdgId inside 'motherPdgId_'. More...

virtual void	endJob () override

Private Attributes
edm::EDGetTokenT < reco::CandidateView >	allProbesToken_
	InputTag to the collection of all probes. More...

bool	checkMotherInUnbiasEff_
	Check mother pdgId in unbiased inefficiency measurement. More...

bool	isMC_
	Is this sample MC? More...

bool	makeMCUnbiasTree_
	Do we have to compute this. More...

std::auto_ptr < tnp::BaseTreeFiller >	mcFiller_

std::auto_ptr < tnp::BaseTreeFiller >	mcUnbiasFiller_
	The object that actually computes variables and fills the tree for unbiased MC. More...

std::set< int32_t >	motherPdgId_
	Possible pdgids for the mother. If empty, any truth-matched mu will be considered good. More...

std::auto_ptr < tnp::BaseTreeFiller >	oldTagFiller_

std::auto_ptr < tnp::BaseTreeFiller >	pairFiller_

edm::EDGetTokenT < edm::Association < std::vector < reco::GenParticle > > >	probeMatchesToken_

std::auto_ptr < tnp::BaseTreeFiller >	tagFiller_

edm::EDGetTokenT < edm::Association < std::vector < reco::GenParticle > > >	tagMatchesToken_
	Token foran edm::Association<reco::GenParticle> from tags & probes to MC truth. More...

tnp::TagProbePairMaker	tagProbePairMaker_
	The object that produces pairs of tags and probes, making any arbitration needed. More...

std::auto_ptr< tnp::TPTreeFiller >	treeFiller_
	The object that actually computes variables and fills the tree for T&P. More...

Additional Inherited Members
Public Types inherited from edm::EDAnalyzer
typedef EDAnalyzer	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::EDAnalyzer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Description: <one line="" class="" summary>="">

Implementation: <Notes on="" implementation>="">

Definition at line 48 of file TagProbeFitTreeProducer.cc.

Constructor & Destructor Documentation

TagProbeFitTreeProducer::TagProbeFitTreeProducer ( const edm::ParameterSet & iConfig )

explicit

Definition at line 91 of file TagProbeFitTreeProducer.cc.

References edm::ParameterSet::addParameter(), allProbesToken_, edm::EDConsumerBase::consumesCollector(), edm::ParameterSet::empty(), edm::ParameterSet::existsAs(), edm::ParameterSet::getParameter(), isMC_, makeMCUnbiasTree_, mcFiller_, mcUnbiasFiller_, motherPdgId_, pairFiller_, probeMatchesToken_, tagFiller_, tagMatchesToken_, and treeFiller_.

                                                                                :
     isMC_(iConfig.getParameter<bool>("isMC")),
     makeMCUnbiasTree_(isMC_ ? iConfig.getParameter<bool>("makeMCUnbiasTree") : false),
     checkMotherInUnbiasEff_(makeMCUnbiasTree_ ? iConfig.getParameter<bool>("checkMotherInUnbiasEff") : false),
     tagProbePairMaker_(iConfig, consumesCollector()),
     treeFiller_(new tnp::TPTreeFiller(iConfig, consumesCollector())),
     oldTagFiller_((iConfig.existsAs<bool>("fillTagTree") && iConfig.getParameter<bool>("fillTagTree")) ? new tnp::BaseTreeFiller("tag_tree",iConfig, consumesCollector()) : 0)
 {
     if (isMC_) {
         // For mc efficiency we need the MC matches for tags & probes
         tagMatchesToken_ = consumes<edm::Association<std::vector<reco::GenParticle> > >(iConfig.getParameter<edm::InputTag>("tagMatches"));
         probeMatchesToken_ = consumes<edm::Association<std::vector<reco::GenParticle> > >(iConfig.getParameter<edm::InputTag>("probeMatches"));
         //.. and the pdgids of the possible mothers
         if (iConfig.existsAs<int32_t>("motherPdgId")) {
             motherPdgId_.insert(iConfig.getParameter<int32_t>("motherPdgId"));
         } else {
             std::vector<int32_t> motherIds = iConfig.getParameter<std::vector<int32_t> >("motherPdgId");
             motherPdgId_.insert(motherIds.begin(), motherIds.end());
         }
 
         // For unbiased efficiency we also need the collection of all probes
         if (makeMCUnbiasTree_) {
             allProbesToken_ = consumes<reco::CandidateView>(iConfig.getParameter<edm::InputTag>("allProbes"));
             mcUnbiasFiller_.reset(new tnp::BaseTreeFiller("mcUnbias_tree",iConfig, consumesCollector()));
         }
     }
 
     edm::ParameterSet tagPSet;
     if (iConfig.existsAs<edm::ParameterSet>("tagVariables")) tagPSet.addParameter<edm::ParameterSet>("variables", iConfig.getParameter<edm::ParameterSet>("tagVariables"));
     if (iConfig.existsAs<edm::ParameterSet>("tagFlags"    )) tagPSet.addParameter<edm::ParameterSet>("flags",     iConfig.getParameter<edm::ParameterSet>("tagFlags"));
     if (!tagPSet.empty()) { tagFiller_.reset(new tnp::BaseTreeFiller(*treeFiller_, tagPSet, consumesCollector(), "tag_")); }
     edm::ParameterSet mcPSet;
     if (iConfig.existsAs<edm::ParameterSet>("mcVariables")) mcPSet.addParameter<edm::ParameterSet>("variables", iConfig.getParameter<edm::ParameterSet>("mcVariables"));
     if (iConfig.existsAs<edm::ParameterSet>("mcFlags"    )) mcPSet.addParameter<edm::ParameterSet>("flags",     iConfig.getParameter<edm::ParameterSet>("mcFlags"));
     if (!mcPSet.empty()) { mcFiller_.reset(new tnp::BaseTreeFiller(*treeFiller_, mcPSet, consumesCollector(), "mc_")); }
     edm::ParameterSet pairPSet;
     if (iConfig.existsAs<edm::ParameterSet>("pairVariables")) pairPSet.addParameter<edm::ParameterSet>("variables", iConfig.getParameter<edm::ParameterSet>("pairVariables"));
     if (iConfig.existsAs<edm::ParameterSet>("pairFlags"    )) pairPSet.addParameter<edm::ParameterSet>("flags",     iConfig.getParameter<edm::ParameterSet>("pairFlags"));
     if (!pairPSet.empty()) { pairFiller_.reset(new tnp::BaseTreeFiller(*treeFiller_, pairPSet, consumesCollector(), "pair_")); }
 }

TagProbeFitTreeProducer::~TagProbeFitTreeProducer ( )

Definition at line 133 of file TagProbeFitTreeProducer.cc.

134 {

135 }

Member Function Documentation

void TagProbeFitTreeProducer::analyze	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivatevirtual

Implements edm::EDAnalyzer.

Definition at line 144 of file TagProbeFitTreeProducer.cc.

References allProbesToken_, checkMother(), checkMotherInUnbiasEff_, edm::Event::getByToken(), i, isMC_, edm::Ref< C, T, F >::isNonnull(), makeMCUnbiasTree_, mcFiller_, mcUnbiasFiller_, gen::n, oldTagFiller_, pairFiller_, probeMatchesToken_, tnp::TagProbePairMaker::run(), alcazmumu_cfi::src, tagFiller_, tagMatchesToken_, tagProbePairMaker_, and treeFiller_.

 {
     using namespace edm; using namespace std;
     Handle<reco::CandidateView> src, allProbes;
     Handle<Association<vector<reco::GenParticle> > > tagMatches, probeMatches;
     treeFiller_->init(iEvent); // read out info from the event if needed (external vars, list of passing probes, ...)
     if (oldTagFiller_.get()) oldTagFiller_->init(iEvent);
     if (tagFiller_.get())    tagFiller_->init(iEvent);
     if (pairFiller_.get())   pairFiller_->init(iEvent);
     if (mcFiller_.get())     mcFiller_->init(iEvent);
     
     // on mc we want to load also the MC match info
     if (isMC_) {
         iEvent.getByToken(tagMatchesToken_, tagMatches);
         iEvent.getByToken(probeMatchesToken_, probeMatches);
     }
 
     // get the list of (tag+probe) pairs, performing arbitration
     tnp::TagProbePairs pairs = tagProbePairMaker_.run(iEvent);
     // loop on them to fill the tree
     for (tnp::TagProbePairs::const_iterator it = pairs.begin(), ed = pairs.end(); it != ed; ++it) {
         // on mc, fill mc info (on non-mc, let it to 'true', the treeFiller will ignore it anyway
         bool mcTrue = false;
     float mcMass = 0.f;
         if (isMC_) {
             reco::GenParticleRef mtag = (*tagMatches)[it->tag], mprobe = (*probeMatches)[it->probe];
             mcTrue = checkMother(mtag) && checkMother(mprobe);
         if (mcTrue) {
           mcMass = (mtag->p4() + mprobe->p4()).mass();
           if (mcFiller_.get()) mcFiller_->fill(reco::CandidateBaseRef(mprobe));
         }
     }
         // fill in the variables for this t+p pair
     if (tagFiller_.get())    tagFiller_->fill(it->tag);
     if (oldTagFiller_.get()) oldTagFiller_->fill(it->tag);
     if (pairFiller_.get())   pairFiller_->fill(it->pair);
     treeFiller_->fill(it->probe, it->mass, mcTrue, mcMass);
     }
 
     if (isMC_ && makeMCUnbiasTree_) {
         // read full collection of probes
         iEvent.getByToken(allProbesToken_, allProbes);
         // init the tree filler
         mcUnbiasFiller_->init(iEvent);
         // loop on probes
         for (size_t i = 0, n = allProbes->size(); i < n; ++i) {
             const reco::CandidateBaseRef & probe = allProbes->refAt(i);
             // check mc match, and possibly mother match
             reco::GenParticleRef probeMatch = (*probeMatches)[probe];
             bool probeOk = checkMotherInUnbiasEff_ ? checkMother(probeMatch) : probeMatch.isNonnull();
             // fill the tree only for good ones
             if (probeOk) mcUnbiasFiller_->fill(probe);
         }
     }
 
 }

bool TagProbeFitTreeProducer::checkMother ( const reco::GenParticleRef & ref ) const

private

Return true if ref is not null and has an ancestor with pdgId inside 'motherPdgId_'.

Definition at line 202 of file TagProbeFitTreeProducer.cc.

References funct::abs(), edm::RefVector< C, T, F >::begin(), alignCSCRings::e, edm::RefVector< C, T, F >::end(), edm::Ref< C, T, F >::isNull(), visualization-live-secondInstance_cfg::m, and motherPdgId_.

Referenced by analyze().

                                                                         {
     if (ref.isNull()) return false;
     if (motherPdgId_.empty()) return true;
     if (motherPdgId_.find(abs(ref->pdgId())) != motherPdgId_.end()) return true;
     reco::GenParticle::mothers m = ref->motherRefVector();
     for (reco::GenParticle::mothers::const_iterator it = m.begin(), e = m.end(); it != e; ++it) {
         if (checkMother(*it)) return true;
     }
     return false;
 }

void TagProbeFitTreeProducer::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 216 of file TagProbeFitTreeProducer.cc.

References edm::getProcessParameterSetContainingModule(), edm::EDAnalyzer::moduleDescription(), and treeFiller_.

                                 {
     // ask to write the current PSet info into the TTree header
     treeFiller_->writeProvenance(edm::getProcessParameterSetContainingModule(moduleDescription()));
 }