|
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
|
#ifndef DataFormats_Candidate_VertexCompositeCandidate_H
#define DataFormats_Candidate_VertexCompositeCandidate_H
/** \class reco::VertexCompositeCandidate
*
* A composite Candidate with error
* matrix and other vertex fix information.
*
* \author Luca Lista, INFN
*
*
*/
#include "DataFormats/Candidate/interface/VertexCompositeCandidateFwd.h"
#include "DataFormats/Candidate/interface/CompositeCandidate.h"
namespace reco {
class VertexCompositeCandidate : public CompositeCandidate {
public:
VertexCompositeCandidate() : CompositeCandidate() {}
/// constructor from values
VertexCompositeCandidate(
Charge q, const LorentzVector &p4, const Point &vtx, int pdgId = 0, int status = 0, bool integerCharge = true)
: CompositeCandidate(q, p4, vtx, pdgId, status, integerCharge), chi2_(0), ndof_(0) {}
/// constructor from values
VertexCompositeCandidate(Charge q,
const LorentzVector &p4,
const Point &vtx,
const CovarianceMatrix &err,
double chi2,
double ndof,
int pdgId = 0,
int status = 0,
bool integerCharge = true);
/// constructor from values
explicit VertexCompositeCandidate(const Candidate &p) : CompositeCandidate(p), chi2_(0), ndof_(0) {}
/// constructor from values
explicit VertexCompositeCandidate(const CompositeCandidate &p) : CompositeCandidate(p), chi2_(0), ndof_(0) {}
/// destructor
~VertexCompositeCandidate() override;
/// returns a clone of the candidate
VertexCompositeCandidate *clone() const override;
/// chi-squares
double vertexChi2() const override { return chi2_; }
/** Number of degrees of freedom
* Meant to be Double32_t for soft-assignment fitters:
* tracks may contribute to the vertex with fractional weights.
* The ndof is then = to the sum of the track weights.
* see e.g. CMS NOTE-2006/032, CMS NOTE-2004/002
*/
double vertexNdof() const override { return ndof_; }
/// chi-squared divided by n.d.o.f.
double vertexNormalizedChi2() const override { return chi2_ / ndof_; }
/// (i, j)-th element of error matrix, i, j = 0, ... 2
double vertexCovariance(int i, int j) const override { return covariance_[idx(i, j)]; }
using reco::LeafCandidate::vertexCovariance; // avoid hiding the
/// fill SMatrix
void fillVertexCovariance(CovarianceMatrix &v) const override;
/// set chi2 and ndof
void setChi2AndNdof(double chi2, double ndof) {
chi2_ = chi2;
ndof_ = ndof;
}
/// set covariance matrix
void setCovariance(const CovarianceMatrix &m);
private:
/// chi-sqared
Double32_t chi2_;
/// number of degrees of freedom
Double32_t ndof_;
/// covariance matrix (3x3) as vector
Double32_t covariance_[size];
/// position index
index idx(index i, index j) const {
int a = (i <= j ? i : j), b = (i <= j ? j : i);
return b * (b + 1) / 2 + a;
}
};
} // namespace reco
#endif
|