|
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
|
#include <memory>
#include "DataFormats/Provenance/interface/ProductProvenance.h"
#include "FWCore/Common/interface/Provenance.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/Registry.h"
#include "HeavyFlavorAnalysis/Onia2MuMu/interface/OniaVtxReProducer.h"
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
OniaVtxReProducer::OniaVtxReProducer(const edm::Handle<reco::VertexCollection> &handle, const edm::Event &iEvent) {
const edm::Provenance *prov = handle.provenance();
if (prov == nullptr)
throw cms::Exception("CorruptData") << "Vertex handle doesn't have provenance.";
edm::ParameterSet psetFromProvenance = edm::parameterSet(prov->stable(), iEvent.processHistory());
bool is_primary_available = false;
const edm::Provenance *parent_prov = prov;
if (edm::moduleName(prov->stable(), iEvent.processHistory()) != "PrimaryVertexProducer") {
std::vector<edm::BranchID> parents = prov->productProvenance()->parentage().parents();
for (std::vector<edm::BranchID>::const_iterator it = parents.begin(), ed = parents.end(); it != ed; ++it) {
edm::Provenance parprov = iEvent.getProvenance(*it);
if (parprov.friendlyClassName() == "recoVertexs") { // for AOD actually this the parent we should look for
parent_prov = &parprov;
psetFromProvenance = edm::parameterSet(parprov.stable(), iEvent.processHistory());
is_primary_available = true;
break;
}
}
} else
is_primary_available = true;
if (is_primary_available)
prov = parent_prov;
else
throw cms::Exception("Configuration") << "Vertices to re-produce don't come from a PrimaryVertexProducer \n";
configure(psetFromProvenance);
// Now we also dig out the ProcessName used for the reco::Tracks and reco::Vertices
std::vector<edm::BranchID> parents = prov->productProvenance()->parentage().parents();
bool foundTracks = false;
bool foundBeamSpot = false;
for (std::vector<edm::BranchID>::const_iterator it = parents.begin(), ed = parents.end(); it != ed; ++it) {
const edm::Provenance &parprov = iEvent.getProvenance(*it);
if (parprov.friendlyClassName() == "recoTracks") {
tracksTag_ = edm::InputTag(parprov.moduleLabel(), parprov.productInstanceName(), parprov.processName());
foundTracks = true;
if (parprov.moduleLabel() != "generalTracks")
foundTracks = false; // this is necessary since we are asking for that in onia2mumu
} else if (parprov.friendlyClassName() == "recoBeamSpot") {
beamSpotTag_ = edm::InputTag(parprov.moduleLabel(), parprov.productInstanceName(), parprov.processName());
foundBeamSpot = true;
if (parprov.moduleLabel() != "offlineBeamSpot")
foundBeamSpot = false; // this is necessary since we are asking for that in onia2mumu
}
}
if (!foundTracks || !foundBeamSpot) {
//edm::LogWarning("OniaVtxReProducer_MissingParentage") <<
throw cms::Exception("Configuration")
<< "Can't find correct parentage info for vertex collection inputs: " << (foundTracks ? "" : "generalTracks ")
<< (foundBeamSpot ? "" : "offlineBeamSpot") << "\n";
}
}
void OniaVtxReProducer::configure(const edm::ParameterSet &iConfig) {
config_ = iConfig;
tracksTag_ = iConfig.getParameter<edm::InputTag>("TrackLabel");
beamSpotTag_ = iConfig.getParameter<edm::InputTag>("beamSpotLabel");
algo_ = std::make_unique<PrimaryVertexProducerAlgorithm>(iConfig);
}
std::vector<TransientVertex> OniaVtxReProducer::makeVertices(const reco::TrackCollection &tracks,
const reco::BeamSpot &bs,
const TransientTrackBuilder &theB) const {
std::vector<reco::TransientTrack> t_tks;
t_tks.reserve(tracks.size());
for (reco::TrackCollection::const_iterator it = tracks.begin(), ed = tracks.end(); it != ed; ++it) {
t_tks.push_back(theB.build(*it));
t_tks.back().setBeamSpot(bs);
}
return algo_->vertices(t_tks, bs, "AdaptiveVertexFitter");
}
|