|
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
|
#ifndef DataFormats_TauReco_PFTau3ProngSummary_h
#define DataFormats_TauReco_PFTau3ProngSummary_h
/* class PFTau3ProngSummary
*
* Stores information on the 3 prong summary for a fully reconstructed tau lepton
*
* author: Ian M. Nugent
* The idea of the fully reconstructing the tau using a kinematic fit comes from
* Lars Perchalla and Philip Sauerland Theses under Achim Stahl supervision. This
* code is a result of the continuation of this work by Ian M. Nugent and Vladimir Cherepanov.
*/
#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/Math/interface/Point3D.h"
#include "DataFormats/Math/interface/Error.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/TauReco/interface/PFTauTransverseImpactParameter.h"
#include "DataFormats/TauReco/interface/PFTauTransverseImpactParameterFwd.h"
#include "TVector3.h"
#include "TLorentzVector.h"
#include "TMath.h"
namespace reco {
class PFTau3ProngSummary {
public:
enum { ambiguity, minus, plus, nsolutions };
enum { dimension = 3 };
enum { covarianceSize = dimension * (dimension + 1) / 2 };
typedef math::Error<dimension>::type CovMatrix;
typedef math::XYZPoint Point;
typedef math::XYZVector Vector;
/// constructor from values
PFTau3ProngSummary();
PFTau3ProngSummary(reco::PFTauTransverseImpactParameterRef TIP,
TLorentzVector a1,
double vertex_chi2,
double vertex_ndf);
PFTau3ProngSummary(reco::PFTauTransverseImpactParameterRef TIP,
TLorentzVector a1,
double vertex_chi2,
double vertex_ndf,
TVector3 sv,
CovMatrix svcov);
virtual ~PFTau3ProngSummary() {}
PFTau3ProngSummary* clone() const;
virtual bool AddSolution(unsigned int solution,
const TLorentzVector& tau,
const std::vector<TLorentzVector>& daughter_p4,
const std::vector<int>& daughter_charge,
const std::vector<int>& daughter_PDGID,
bool has3ProngSolution,
double solutionChi2,
double thetaGJsig);
const reco::PFTauTransverseImpactParameterRef& PFTauTIP() const { return TIP_; }
// interface for relevant TIP functions;
const VertexRef& primaryVertex() const { return TIP_->primaryVertex(); }
CovMatrix primaryVertexCov() const { return TIP_->primaryVertexCov(); }
bool hasSecondaryVertex() const { return TIP_->hasSecondaryVertex(); }
const VertexRef& secondaryVertex() const { return TIP_->secondaryVertex(); }
CovMatrix secondaryVertexCov() const { return TIP_->secondaryVertexCov(); }
const Vector& flightLength() const { return TIP_->flightLength(); }
double flightLengthSig() const { return TIP_->flightLengthSig(); }
CovMatrix flightLenghtCov() const { return TIP_->flightLengthCov(); }
// Tau 3 prong functions
const TLorentzVector& A1_LV() const { return a1_; }
double M_A1() const { return a1_.M(); }
double M_12() const; //pi-pi-
double M_13() const; //pi-pi+ Dalitz masses
double M_23() const; //pi-pi+ Dalitz masses
int Tau_Charge() const;
const TVector3& HelixFitSecondaryVertex() const { return sv_; }
const CovMatrix& HelixFitSecondaryVertexCov() const { return svcov_; }
double Vertex_chi2() const { return vertex_chi2_; }
double Vertex_ndf() const { return vertex_ndf_; }
double Vertex_Prob() const { return TMath::Prob(vertex_chi2_, vertex_ndf_); }
bool has3ProngSolution(unsigned int i) const { return has3ProngSolution_[i]; }
double Solution_Chi2(unsigned int i) const { return solution_Chi2_[i]; }
double SignificanceOfThetaGJ(unsigned int i) const { return thetaGJsig_[i]; } // 0 or less means the theta_GF has
// a physical solution
const TLorentzVector& Tau(unsigned int i) const { return tau_p4_[i]; }
const std::vector<int>& Daughter_PDGID(unsigned int i) const { return daughter_PDGID_[i]; }
const std::vector<int>& Daughter_Charge(unsigned int i) const { return daughter_charge_[i]; }
const std::vector<TLorentzVector>& Daughter_P4(unsigned int i) const { return daughter_p4_[i]; }
private:
reco::PFTauTransverseImpactParameterRef TIP_;
TLorentzVector a1_;
TVector3 sv_;
CovMatrix svcov_;
double vertex_chi2_;
double vertex_ndf_;
std::vector<bool> has3ProngSolution_;
std::vector<double> solution_Chi2_;
std::vector<double> thetaGJsig_;
std::vector<TLorentzVector> tau_p4_;
std::vector<std::vector<int> > daughter_PDGID_;
std::vector<std::vector<int> > daughter_charge_;
std::vector<std::vector<TLorentzVector> > daughter_p4_;
};
} // namespace reco
#endif
|