Added classes for UPC analysis

[u/mrichter/AliRoot.git] / PWGUD / UPC / AliUPCTrack.cxx

diff --git a/PWGUD/UPC/AliUPCTrack.cxx b/PWGUD/UPC/AliUPCTrack.cxx
new file mode 100644 (file)
index 0000000..18aec22
--- /dev/null
+++ b/PWGUD/UPC/AliUPCTrack.cxx
@@ -0,0 +1,166 @@
+
+//_____________________________________________________________________________
+//    Class for UPC data
+//    Author: Jaroslav Adam
+//
+//    contains parameters of the central track relevant for UPC analysis
+//_____________________________________________________________________________
+
+#include "TLorentzVector.h"
+
+#include "AliUPCTrack.h"
+
+ClassImp(AliUPCTrack)
+
+//_____________________________________________________________________________
+AliUPCTrack::AliUPCTrack()
+ :TObject(),
+  fCharge(0), fMaskMan(0), fFilterMap(0), fChi2perNDF(0),
+  fTPCmomentum(0), fTPCsignal(0), fTPCncls(0),
+  fTPCrows(0), fTPCnclsF(0), fTPCnclsS(0),
+  fITSchi2perNDF(0), fITSClusterMap(0)
+{
+
+  // Default constructor
+
+  for(Int_t i=0; i<2; i++) {fDZ[i] = 0.; fdzSPD[i] = 0.; fdzIP[i] = 0.;}
+  for(Int_t i=0; i<3; i++) {fP[i] = 0.; fCov[i] = 0.; fCovSPD[i] = 0.; fCovIP[i] = 0.;}
+}
+
+//_____________________________________________________________________________
+void AliUPCTrack::Clear(Option_t *)
+{
+  // clear track flags
+
+  fMaskMan = 0;
+  fFilterMap = 0;
+}
+
+//_____________________________________________________________________________
+void AliUPCTrack::SetImpactParameters(Float_t *p, Float_t *cov)
+{
+  // set impact parameters in XY and Z to default primary vertex
+
+  for(Int_t i=0; i<2; i++) {
+    fDZ[i] = p[i];
+    fCov[i] = cov[i];
+  }
+  fCov[2] = cov[2];
+}
+
+//_____________________________________________________________________________
+void AliUPCTrack::SetImpactParametersSPD(Float_t *p, Float_t *cov)
+{
+  // set SPD impact parameters
+
+  for(Int_t i=0; i<2; i++) {
+    fdzSPD[i] = p[i];
+    fCovSPD[i] = cov[i];
+  }
+  fCovSPD[2] = cov[2];
+}
+
+//_____________________________________________________________________________
+void AliUPCTrack::SetImpactParametersIP(Float_t *p, Float_t *cov)
+{
+  // set impact parameters to nominal interaction point
+
+  for(Int_t i=0; i<2; i++) {
+    fdzIP[i] = p[i];
+    fCovIP[i] = cov[i];
+  }
+  fCovIP[2] = cov[2];
+}
+
+//_____________________________________________________________________________
+void AliUPCTrack::GetMomentum(TLorentzVector *v, Double_t mass) const
+{
+  // get track 4-momentum
+  v->SetXYZM(fP[0],fP[1],fP[2],mass);
+}
+
+Int_t AliUPCTrack::GetITSNcls(void) const
+{
+  // number of points in ITS
+
+  Int_t ncls = 0;
+
+  for(UChar_t i=0; i<6; i++) if( fITSClusterMap & (1 << i) ) ncls++;
+
+  return ncls;
+}
+
+//_____________________________________________________________________________
+void AliUPCTrack::GetImpactParameters(Double_t *p, Double_t *cov) const
+{
+  // get impact parameters in XY and Z to default primary vertex
+
+  for(Int_t i=0; i<2; i++) {
+    p[i] = (Double_t) fDZ[i];
+    cov[i] = (Double_t) fCov[i];
+  }
+  cov[2] = (Double_t) fCov[2];
+}
+
+//_____________________________________________________________________________
+void AliUPCTrack::GetImpactParametersSPD(Double_t *p, Double_t *cov) const
+{
+  // get SPD impact parameters
+
+  for(Int_t i=0; i<2; i++) {
+    p[i] = (Double_t) fdzSPD[i];
+    cov[i] = (Double_t) fCovSPD[i];
+  }
+  cov[2] = (Double_t) fCovSPD[2];
+}
+
+//_____________________________________________________________________________
+void AliUPCTrack::GetImpactParametersIP(Double_t *p, Double_t *cov) const
+{
+  // get impact parameters to nominal interaction point
+
+  for(Int_t i=0; i<2; i++) {
+    p[i] = (Double_t) fdzIP[i];
+    cov[i] = (Double_t) fCovIP[i];
+  }
+  cov[2] = (Double_t) fCovIP[2];
+}
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+