CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
ZMuMuUserData.cc
Go to the documentation of this file.
10 
16 
20 
21 #include <vector>
22 
23 using namespace edm;
24 using namespace std;
25 using namespace reco;
26 using namespace isodeposit;
27 //using namespace pat;
28 
30 public:
33 private:
34  void produce( edm::Event &, const edm::EventSetup & ) override;
35 
40  double alpha_, beta_;
41  string hltPath_;
42  int counter;
43 
44 
45 };
46 
47 
48 
50  srcToken_(consumes<std::vector<reco::CompositeCandidate> > ( cfg.getParameter<InputTag>( "src" ) ) ),
51  beamSpotToken_(consumes<BeamSpot> (cfg.getParameter<InputTag>( "beamSpot" ) ) ),
52  primaryVerticesToken_(consumes<VertexCollection> (cfg.getParameter<InputTag>( "primaryVertices" ) ) ),
53  zGenParticlesMatchToken_(consumes<GenParticleMatch> (cfg.getParameter<InputTag>( "zGenParticlesMatch" ) ) ),
54  alpha_(cfg.getParameter<double>("alpha") ),
55  beta_(cfg.getParameter<double>("beta") ),
56  hltPath_(cfg.getParameter<std::string >("hltPath") ){
57  produces<vector<pat::CompositeCandidate> >();
58 }
59 
60 void ZMuMuUserData::produce( Event & evt, const EventSetup & ) {
62  evt.getByToken(srcToken_,dimuons);
63 
64  Handle<BeamSpot> beamSpotHandle;
65  evt.getByToken(beamSpotToken_, beamSpotHandle);
66 
67  Handle<VertexCollection> primaryVertices; // Collection of primary Vertices
68  evt.getByToken(primaryVerticesToken_, primaryVertices);
69 
70 
71  bool isMCMatchTrue=false;
72 
73  Handle<GenParticleMatch> zGenParticlesMatch;
74  if(evt.getByToken( zGenParticlesMatchToken_, zGenParticlesMatch )){
75  isMCMatchTrue=true;
76  }
77 
78  //cout<<"isMCMatchTrue"<<isMCMatchTrue <<endl;
79  auto_ptr<vector<pat::CompositeCandidate> > dimuonColl( new vector<pat::CompositeCandidate> () );
80 
81 
82  for (unsigned int i = 0; i< dimuons->size();++i){
83  const CompositeCandidate & z = (*dimuons)[i];
84  //CandidateBaseRef zRef = dimuons ->refAt(i);
86  pat::CompositeCandidate dimuon(z);
87 
88  float trueMass,truePt,trueEta,truePhi,trueY;
89  if (isMCMatchTrue){
90  GenParticleRef trueZRef = (*zGenParticlesMatch)[zRef];
91  //CandidateRef trueZRef = trueZIter->val;
92  if( trueZRef.isNonnull() ) {
93  const Candidate & z = * trueZRef;
94  trueMass = z.mass();
95  truePt = z.pt();
96  trueEta = z.eta();
97  truePhi = z.phi();
98  trueY = z.rapidity();
99  } else {
100  trueMass = -100;
101  truePt = -100;
102  trueEta = -100;
103  truePhi = -100;
104  trueY = -100;
105  }
106 
107  dimuon.addUserFloat("TrueMass",trueMass);
108  dimuon.addUserFloat("TruePt",truePt);
109  dimuon.addUserFloat("TrueEta",trueEta);
110  dimuon.addUserFloat("TruePhi",truePhi);
111  dimuon.addUserFloat("TrueY",trueY);
112 
113  }
114  const Candidate * dau1 = z.daughter(0);
115  const Candidate * dau2 = z.daughter(1);
116  const pat::Muon & mu1 = dynamic_cast<const pat::Muon&>(*dau1->masterClone());
117  const pat::Muon & mu2 = dynamic_cast<const pat::Muon&>(*dau2->masterClone());
118 
119  /*cout<<"mu1 is null? "<<mu1.isMuon()<<endl;
120  cout<<"mu2 is null? "<<mu2.isMuon()<<endl;
121  cout<<"mu1 is global?"<<mu1.isGlobalMuon()<<endl;
122  cout<<"mu2 is global?"<<mu2.isGlobalMuon()<<endl;
123  */
124 
125  if(mu1.isGlobalMuon()==true && mu2.isGlobalMuon()==true){
126  TrackRef stAloneTrack1;
127  TrackRef stAloneTrack2;
128  Vector momentum;
130  double mu_mass;
131  stAloneTrack1 = dau1->get<TrackRef,reco::StandAloneMuonTag>();
132  stAloneTrack2 = dau2->get<TrackRef,reco::StandAloneMuonTag>();
133  float zDau1SaEta = stAloneTrack1->eta();
134  float zDau2SaEta = stAloneTrack2->eta();
135  float zDau1SaPhi = stAloneTrack1->phi();
136  float zDau2SaPhi = stAloneTrack2->phi();
137  float zDau1SaPt,zDau2SaPt;
138  if(counter % 2 == 0) {
139  momentum = stAloneTrack1->momentum();
140  p4_1 = dau2->polarP4();
141  mu_mass = dau1->mass();
143  zDau1SaPt = stAloneTrack1->pt();
144  zDau2SaPt = - stAloneTrack2->pt();
145  }else{
146  momentum = stAloneTrack2->momentum();
147  p4_1= dau1->polarP4();
148  mu_mass = dau2->mass();
150  zDau1SaPt = - stAloneTrack1->pt();
151  zDau2SaPt = stAloneTrack2->pt();
152  }
153 
154  Candidate::PolarLorentzVector p4_2(momentum.rho(), momentum.eta(),momentum.phi(), mu_mass);
155  double mass = (p4_1+p4_2).mass();
156  float zMassSa = mass;
157  //cout<<"zMassSa "<<zMassSa;
158  dimuon.addUserFloat("MassSa",zMassSa);
159  dimuon.addUserFloat("Dau1SaPt",zDau1SaPt);
160  dimuon.addUserFloat("Dau2SaPt",zDau2SaPt);
161  dimuon.addUserFloat("Dau1SaPhi",zDau1SaPhi);
162  dimuon.addUserFloat("Dau2SaPhi",zDau2SaPhi);
163  dimuon.addUserFloat("Dau1SaEta",zDau1SaEta);
164  dimuon.addUserFloat("Dau2SaEta",zDau2SaEta);
165  ++counter;
166  }
167  dimuonColl->push_back(dimuon);
168 
169  }
170 
171 
172  evt.put( dimuonColl);
173 }
174 
176 
178 
Analysis-level particle class.
int i
Definition: DBlmapReader.cc:9
virtual float mass() const =0
mass
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:434
virtual float eta() const =0
momentum pseudorapidity
#define DEFINE_FWK_MODULE(type)
Definition: MakerMacros.h:17
stand alone muon component tag
Definition: RecoCandidate.h:72
bool isGlobalMuon() const
Definition: Muon.h:218
virtual float phi() const =0
momentum azimuthal angle
std::vector< Vertex > VertexCollection
collection of Vertex objects
Definition: VertexFwd.h:9
virtual double rapidity() const =0
rapidity
virtual const PolarLorentzVector & polarP4() const =0
four-momentum Lorentz vector
float float float z
EDGetTokenT< BeamSpot > beamSpotToken_
EDGetTokenT< VertexCollection > primaryVerticesToken_
bool isNonnull() const
Checks for non-null.
Definition: Ref.h:250
virtual float pt() const =0
transverse momentum
ZMuMuUserData(const edm::ParameterSet &)
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:116
void addUserFloat(const std::string &label, float data)
Set user-defined float.
Definition: PATObject.h:765
void produce(edm::Event &, const edm::EventSetup &) override
EDGetTokenT< GenParticleMatch > zGenParticlesMatchToken_
XYZVectorD XYZVector
spatial vector with cartesian internal representation
Definition: Vector3D.h:30
edm::Ref< TrackCollection > TrackRef
persistent reference to a Track
Definition: TrackFwd.h:14
EDGetTokenT< std::vector< reco::CompositeCandidate > > srcToken_
virtual const Candidate * daughter(size_type) const
return daughter at a given position, i = 0, ... numberOfDaughters() - 1 (read only mode) ...
T get() const
get a component
Definition: Candidate.h:219
Analysis-level muon class.
Definition: Muon.h:50
math::XYZVector Vector
virtual const CandidateBaseRef & masterClone() const =0
math::PtEtaPhiMLorentzVector PolarLorentzVector
Lorentz vector.
Definition: Candidate.h:43