|
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
|
#ifndef CommonTools_PFCandProducer_PFPileUpAlgo_
#define CommonTools_PFCandProducer_PFPileUpAlgo_
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
class PFPileUpAlgo {
public:
typedef std::vector<edm::FwdPtr<reco::PFCandidate> > PFCollection;
PFPileUpAlgo() : checkClosestZVertex_(true), verbose_(false) { ; }
PFPileUpAlgo(bool checkClosestZVertex, bool verbose = false)
: checkClosestZVertex_(checkClosestZVertex), verbose_(verbose) {
;
}
~PFPileUpAlgo() { ; }
// the last parameter is needed if you want to use the sourceCandidatePtr
void process(const PFCollection& pfCandidates, const reco::VertexCollection& vertices);
inline void setVerbose(bool verbose) { verbose_ = verbose; }
inline void setCheckClosestZVertex(bool val) { checkClosestZVertex_ = val; }
inline void setNumOfPUVtxsForCharged(unsigned int val) { fNumOfPUVtxsForCharged_ = val; }
inline void setDzCutForChargedFromPUVtxs(double val) { fDzCutForChargedFromPUVtxs_ = val; }
const PFCollection& getPFCandidatesFromPU() const { return pfCandidatesFromPU_; }
const PFCollection& getPFCandidatesFromVtx() const { return pfCandidatesFromVtx_; }
int chargedHadronVertex(const reco::VertexCollection& vertices, const reco::PFCandidate& pfcand) const;
private:
/// use the closest z vertex if a track is not in a vertex
bool checkClosestZVertex_;
unsigned int fNumOfPUVtxsForCharged_;
double fDzCutForChargedFromPUVtxs_;
/// verbose ?
bool verbose_;
PFCollection pfCandidatesFromVtx_;
PFCollection pfCandidatesFromPU_;
};
#endif
|