|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
// -*- C++ -*-
//
// Package: TagProbeMassProducer
// Class: TagProbeMassProducer
//
/**\class TagProbeMassProducer TagProbeMassProducer.cc PhysicsTools/TagProbeMassProducer/src/TagProbeMassProducer.cc
Description: <one line class summary>
Implementation:
<Notes on implementation>
*/
//
// Original Author: Nadia Adam
// Created: Wed Apr 16 09:46:30 CDT 2008
// $Id: TagProbeMassProducer.cc,v 1.1 2010/05/04 09:42:40 azzi Exp $
//
//
// User includes
#include "DPGAnalysis/Skims/interface/TagProbeMassProducer.h"
//#include "PhysicsTools/TagAndProbe/interface/CandidateAssociation.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/Common/interface/AssociationMap.h"
#include "DataFormats/Math/interface/deltaR.h"
//#include "DataFormats/MuonReco/interface/Muon.h"
//#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "Math/GenVector/VectorUtil.h"
TagProbeMassProducer::TagProbeMassProducer(const edm::ParameterSet& iConfig) {
tagCollection_ = iConfig.getParameter<edm::InputTag>("TagCollection");
probeCollection_ = iConfig.getParameter<edm::InputTag>("ProbeCollection");
passingProbeCollection_ = iConfig.getParameter<edm::InputTag>("PassingProbeCollection");
massMinCut_ = iConfig.getUntrackedParameter<double>("MassMinCut", 50.0);
massMaxCut_ = iConfig.getUntrackedParameter<double>("MassMaxCut", 120.0);
delRMinCut_ = iConfig.getUntrackedParameter<double>("DelRMinCut", 0.0);
delRMaxCut_ = iConfig.getUntrackedParameter<double>("DelRMaxCut", 10000.0);
requireOS_ = iConfig.getUntrackedParameter<bool>("RequireOS", true);
produces<std::vector<float> >("TPmass");
}
TagProbeMassProducer::~TagProbeMassProducer() {}
//
// member functions
//
// ------------ method called to produce the data ------------
void TagProbeMassProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
// We need the output Muon association collection to fill
std::unique_ptr<std::vector<float> > TPmass(new std::vector<float>);
if (!iEvent.getByLabel(tagCollection_, tags)) {
edm::LogWarning("TagProbe") << "Could not extract tag muons with input tag " << tagCollection_;
}
if (!iEvent.getByLabel(probeCollection_, probes)) {
edm::LogWarning("TagProbe") << "Could not extract probe muons with input tag " << probeCollection_;
}
if (!iEvent.getByLabel(passingProbeCollection_, passingProbes)) {
edm::LogWarning("TagProbe") << "Could not extract passing probe muons with input tag " << passingProbeCollection_;
}
// Loop over Tag and associate with Probes
if (tags.isValid() && probes.isValid()) {
edm::RefToBaseVector<reco::Candidate> vtags;
for (size_t i = 0; i < tags->size(); ++i) {
vtags.push_back(tags->refAt(i));
}
edm::RefToBaseVector<reco::Candidate> vprobes;
for (size_t i = 0; i < probes->size(); ++i) {
vprobes.push_back(probes->refAt(i));
}
int itag = 0;
edm::RefToBaseVector<reco::Candidate>::const_iterator tag = vtags.begin();
for (; tag != vtags.end(); ++tag, ++itag) {
int iprobe = 0;
edm::RefToBaseVector<reco::Candidate>::const_iterator probe = vprobes.begin();
for (; probe != vprobes.end(); ++probe, ++iprobe) {
// Tag-Probe invariant mass cut
double invMass = ROOT::Math::VectorUtil::InvariantMass((*tag)->p4(), (*probe)->p4());
if (invMass < massMinCut_)
continue;
if (invMass > massMaxCut_)
continue;
// Tag-Probe deltaR cut
double delR = reco::deltaR<double>((*tag)->eta(), (*tag)->phi(), (*probe)->eta(), (*probe)->phi());
if (delR < delRMinCut_)
continue;
if (delR > delRMaxCut_)
continue;
// Tag-Probe opposite sign
int sign = (*tag)->charge() * (*probe)->charge();
if (requireOS_ && sign > 0)
continue;
bool isPassing = isPassingProbe(iprobe);
if (isPassing)
TPmass->push_back(invMass);
}
}
}
// Finally put the tag probe collection in the event
iEvent.put(std::move(TPmass), "TPmass");
}
bool TagProbeMassProducer::isPassingProbe(const unsigned int iProbe) const {
if (iProbe > probes->size())
return false;
edm::RefToBase<reco::Candidate> probeRef = probes->refAt(iProbe);
edm::RefToBase<reco::Candidate> passingProbeRef;
unsigned int numPassingProbes = passingProbes->size();
for (unsigned int iPassProbe = 0; iPassProbe < numPassingProbes; ++iPassProbe) {
passingProbeRef = passingProbes->refAt(iPassProbe);
if (passingProbeRef == probeRef) {
return true;
}
}
return false;
}
//define this as a plug-in
DEFINE_FWK_MODULE(TagProbeMassProducer);
|