Morgan/interface/MCAssociator.h

#ifndef MCAssociator_h
#define MCAssociator_h

#include <memory>
#include <string>
#include <iostream>
#include <map>

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"

#include "../interface/TRootParticle.h"
#include "../interface/TRootMCParticle.h"
#include "../interface/TRootJet.h"

#include "TClonesArray.h"


class MCAssociator
{

   public:
      MCAssociator();
      MCAssociator(const edm::ParameterSet& producersNames, int verbosity);
      ~MCAssociator() {};
      void setVerbosity(int verbosity) {verbosity_ = verbosity; };
      void init(const edm::Event& iEvent, TClonesArray* mcParticles);
      void matchGenParticlesTo(TClonesArray* recoParticles);
      void printParticleAssociation(TClonesArray* recoParticles);
      void processGenJets(TClonesArray* genJets, TClonesArray* recoJets);
      void printJetAssociation(TClonesArray* recoJets);

   private:
      int verbosity_;
      int nMC_;
      TClonesArray* mcParticles_;
      edm::Handle <reco::GenParticleCollection> genParticles_;
      std::map<int,int> mcParticlesMap_; // map between index in genParticle collection and index in mcParticles TClonesArray
      edm::InputTag genParticlesProducer_;

};

#endif
Revision:	1.4
Committed:	Wed Jun 10 11:17:05 2009 UTC (15 years, 10 months ago) by lethuill
Content type:	text/plain
Branch:	MAIN
CVS Tags:	all_3_3_2_01, all_3_2_5_02, all_3_2_5_01, all_2_2_9_03, all_2_2_9_02, all_2_2_9_01, HEAD
Branch point for:	CMSSW_2_2_X_br
Changes since 1.3:	+20 -19 lines
Error occurred while calculating annotation data.
Log Message:	Better protection against missing collection / Cleaning data format selection / Last iteration for migration to PAT of Photons
#	Content
1	#ifndef MCAssociator_h
2	#define MCAssociator_h
3
4	#include <memory>
5	#include <string>
6	#include <iostream>
7	#include <map>
8
9	#include "FWCore/Framework/interface/Event.h"
10	#include "FWCore/ParameterSet/interface/ParameterSet.h"
11	#include "DataFormats/Common/interface/Handle.h"
12	#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
13	#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"
14
15	#include "../interface/TRootParticle.h"
16	#include "../interface/TRootMCParticle.h"
17	#include "../interface/TRootJet.h"
18
19	#include "TClonesArray.h"
20
21
22	class MCAssociator
23	{
24
25	public:
26	MCAssociator();
27	MCAssociator(const edm::ParameterSet& producersNames, int verbosity);
28	~MCAssociator() {};
29	void setVerbosity(int verbosity) {verbosity_ = verbosity; };
30	void init(const edm::Event& iEvent, TClonesArray* mcParticles);
31	void matchGenParticlesTo(TClonesArray* recoParticles);
32	void printParticleAssociation(TClonesArray* recoParticles);
33	void processGenJets(TClonesArray* genJets, TClonesArray* recoJets);
34	void printJetAssociation(TClonesArray* recoJets);
35
36	private:
37	int verbosity_;
38	int nMC_;
39	TClonesArray* mcParticles_;
40	edm::Handle <reco::GenParticleCollection> genParticles_;
41	std::map<int,int> mcParticlesMap_; // map between index in genParticle collection and index in mcParticles TClonesArray
42	edm::InputTag genParticlesProducer_;
43
44	};
45
46	#endif