|
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
|
#include <memory>
#include <vector>
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Common/interface/View.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/Candidate/interface/VertexCompositeCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESHandle.h"
class V0VertexTrackProducer : public edm::global::EDProducer<> {
public:
explicit V0VertexTrackProducer(const edm::ParameterSet&);
~V0VertexTrackProducer() override = default;
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
void produce(edm::StreamID streamID, edm::Event& iEvent, edm::EventSetup const& iSetup) const override;
private:
const edm::EDGetTokenT<reco::VertexCompositeCandidateCollection> vccToken_;
};
V0VertexTrackProducer::V0VertexTrackProducer(const edm::ParameterSet& iConfig)
: vccToken_{consumes<reco::VertexCompositeCandidateCollection>(
iConfig.getParameter<edm::InputTag>("vertexCompositeCandidates"))} {
produces<std::vector<reco::Track>>();
}
void V0VertexTrackProducer::produce(edm::StreamID streamID, edm::Event& iEvent, edm::EventSetup const& iSetup) const {
edm::Handle<reco::VertexCompositeCandidateCollection> vccHandle;
iEvent.getByToken(vccToken_, vccHandle);
std::unique_ptr<std::vector<reco::Track>> outputTracks(new std::vector<reco::Track>());
if (vccHandle.isValid()) {
// Loop over VertexCompositeCandidates and associate tracks
for (const auto& vcc : *vccHandle) {
for (size_t i = 0; i < vcc.numberOfDaughters(); ++i) {
const reco::Candidate* daughter = vcc.daughter(i);
const reco::RecoChargedCandidate* chargedDaughter = dynamic_cast<const reco::RecoChargedCandidate*>(daughter);
if (chargedDaughter) {
const reco::TrackRef trackRef = chargedDaughter->track();
if (trackRef.isNonnull()) {
outputTracks->push_back(*trackRef);
}
}
}
}
} else {
edm::LogError("V0VertexTrackProducer") << "Error >> Failed to get VertexCompositeCandidateCollection";
}
iEvent.put(std::move(outputTracks));
}
void V0VertexTrackProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("vertexCompositeCandidates", edm::InputTag("generalV0Candidates:Kshort"));
descriptions.addWithDefaultLabel(desc);
}
DEFINE_FWK_MODULE(V0VertexTrackProducer);
|