]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliAODTrack.cxx
The track number and track pid associated with the cluster are included
[u/mrichter/AliRoot.git] / STEER / AliAODTrack.cxx
index e30a527111eac83240208143361a0734158c20e5..752c0ab2ae2b49379c46edffc41d65eb4bbe564f 100644 (file)
 /* $Id$ */
 
 //-------------------------------------------------------------------------
-//     AOD track base class
+//     AOD track implementation of AliVTrack
 //     Author: Markus Oldenburg, CERN
+//     Markus.Oldenburg@cern.ch
 //-------------------------------------------------------------------------
 
+#include "AliLog.h"
 #include "AliAODTrack.h"
 
 ClassImp(AliAODTrack)
 
 //______________________________________________________________________________
 AliAODTrack::AliAODTrack() : 
-  AliVirtualParticle(),
+  AliVTrack(),
   fChi2perNDF(-999.),
-  fID(-999),
+  fChi2MatchTrigger(0.),
+  fFlags(0),
   fLabel(-999),
-  fCovMatrix(NULL),
-  fProdVertex(0x0),
-  fCharge(-99),
   fITSMuonClusterMap(0),
+  fFilterMap(0),
+  fID(-999),
+  fCharge(-99),
   fType(kUndef),
-  fChi2MatchTrigger(0.)
+  fCovMatrix(NULL),
+  fDetPid(NULL),
+  fProdVertex(NULL)
 {
   // default constructor
 
   SetP();
   SetPosition((Float_t*)NULL);
+  SetXYAtDCA(-999., -999.);
+  SetPxPyPzAtDCA(-999., -999., -999.);
   SetPID((Float_t*)NULL);
 }
 
 //______________________________________________________________________________
-AliAODTrack::AliAODTrack(Int_t id,
+AliAODTrack::AliAODTrack(Short_t id,
                         Int_t label, 
                         Double_t p[3],
                         Bool_t cartesian,
@@ -58,31 +65,38 @@ AliAODTrack::AliAODTrack(Int_t id,
                         AliAODVertex *prodVertex,
                         Bool_t usedForVtxFit,
                         Bool_t usedForPrimVtxFit,
-                        AODTrk_t ttype) :
-  AliVirtualParticle(),
-  fChi2perNDF(-999.),
-  fID(id),
+                        AODTrk_t ttype,
+                        UInt_t selectInfo,
+                        Float_t chi2perNDF) :
+  AliVTrack(),
+  fChi2perNDF(chi2perNDF),
+  fChi2MatchTrigger(0.),
+  fFlags(0),
   fLabel(label),
-  fCovMatrix(NULL),
-  fProdVertex(prodVertex),
+  fITSMuonClusterMap(0),
+  fFilterMap(selectInfo),
+  fID(id),
   fCharge(charge),
-  fITSMuonClusterMap(itsClusMap),
   fType(ttype),
-  fChi2MatchTrigger(0.)
+  fCovMatrix(NULL),
+  fDetPid(NULL),
+  fProdVertex(prodVertex)
 {
   // constructor
  
   SetP(p, cartesian);
   SetPosition(x, isDCA);
+  SetXYAtDCA(-999., -999.);
+  SetPxPyPzAtDCA(-999., -999., -999.);
   SetUsedForVtxFit(usedForVtxFit);
   SetUsedForPrimVtxFit(usedForPrimVtxFit);
   if(covMatrix) SetCovMatrix(covMatrix);
   SetPID(pid);
-
+  SetITSClusterMap(itsClusMap);
 }
 
 //______________________________________________________________________________
-AliAODTrack::AliAODTrack(Int_t id,
+AliAODTrack::AliAODTrack(Short_t id,
                         Int_t label, 
                         Float_t p[3],
                         Bool_t cartesian,
@@ -95,26 +109,34 @@ AliAODTrack::AliAODTrack(Int_t id,
                         AliAODVertex *prodVertex,
                         Bool_t usedForVtxFit,
                         Bool_t usedForPrimVtxFit,
-                        AODTrk_t ttype) :
-  AliVirtualParticle(),
-  fChi2perNDF(-999.),
-  fID(id),
+                        AODTrk_t ttype,
+                        UInt_t selectInfo,
+                        Float_t chi2perNDF) :
+  AliVTrack(),
+  fChi2perNDF(chi2perNDF),
+  fChi2MatchTrigger(0.),
+  fFlags(0),
   fLabel(label),
-  fCovMatrix(NULL),
-  fProdVertex(prodVertex),
+  fITSMuonClusterMap(0),
+  fFilterMap(selectInfo),
+  fID(id),
   fCharge(charge),
-  fITSMuonClusterMap(itsClusMap),
   fType(ttype),
-  fChi2MatchTrigger(0.)
+  fCovMatrix(NULL),
+  fDetPid(NULL),
+  fProdVertex(prodVertex)
 {
   // constructor
  
   SetP(p, cartesian);
   SetPosition(x, isDCA);
+  SetXYAtDCA(-999., -999.);
+  SetPxPyPzAtDCA(-999., -999., -999.);
   SetUsedForVtxFit(usedForVtxFit);
   SetUsedForPrimVtxFit(usedForPrimVtxFit);
   if(covMatrix) SetCovMatrix(covMatrix);
   SetPID(pid);
+  SetITSClusterMap(itsClusMap);
 }
 
 //______________________________________________________________________________
@@ -122,31 +144,37 @@ AliAODTrack::~AliAODTrack()
 {
   // destructor
   delete fCovMatrix;
+  delete fDetPid;
 }
 
 
 //______________________________________________________________________________
 AliAODTrack::AliAODTrack(const AliAODTrack& trk) :
-  AliVirtualParticle(trk),
+  AliVTrack(trk),
   fChi2perNDF(trk.fChi2perNDF),
-  fID(trk.fID),
+  fChi2MatchTrigger(trk.fChi2MatchTrigger),
+  fFlags(trk.fFlags),
   fLabel(trk.fLabel),
-  fCovMatrix(NULL),
-  fProdVertex(trk.fProdVertex),
-  fCharge(trk.fCharge),
   fITSMuonClusterMap(trk.fITSMuonClusterMap),
+  fFilterMap(trk.fFilterMap),
+  fID(trk.fID),
+  fCharge(trk.fCharge),
   fType(trk.fType),
-  fChi2MatchTrigger(0.)
+  fCovMatrix(NULL),
+  fDetPid(NULL),
+  fProdVertex(trk.fProdVertex)
 {
   // Copy constructor
 
   trk.GetP(fMomentum);
   trk.GetPosition(fPosition);
+  SetXYAtDCA(trk.XAtDCA(), trk.YAtDCA());
+  SetPxPyPzAtDCA(trk.PxAtDCA(), trk.PyAtDCA(), trk.PzAtDCA());
   SetUsedForVtxFit(trk.GetUsedForVtxFit());
   SetUsedForPrimVtxFit(trk.GetUsedForPrimVtxFit());
   if(trk.fCovMatrix) fCovMatrix=new AliAODRedCov<6>(*trk.fCovMatrix);
+  if(trk.fDetPid) fDetPid=new AliAODPid(*trk.fDetPid);
   SetPID(trk.fPID);
-
 }
 
 //______________________________________________________________________________
@@ -155,27 +183,40 @@ AliAODTrack& AliAODTrack::operator=(const AliAODTrack& trk)
   // Assignment operator
   if(this!=&trk) {
 
-    AliVirtualParticle::operator=(trk);
+    AliVTrack::operator=(trk);
 
     trk.GetP(fMomentum);
     trk.GetPosition(fPosition);
     trk.GetPID(fPID);
 
+    SetXYAtDCA(trk.XAtDCA(), trk.YAtDCA());
+    SetPxPyPzAtDCA(trk.PxAtDCA(), trk.PyAtDCA(), trk.PzAtDCA());
+    
     fChi2perNDF = trk.fChi2perNDF;
+    fChi2MatchTrigger = trk.fChi2MatchTrigger;
 
-    fID = trk.fID;
+    fFlags = trk.fFlags;
     fLabel = trk.fLabel;    
     
+    fITSMuonClusterMap = trk.fITSMuonClusterMap;
+    fFilterMap = trk.fFilterMap;
+
+    fID = trk.fID;
+
+    fCharge = trk.fCharge;
+    fType = trk.fType;
+
     delete fCovMatrix;
     if(trk.fCovMatrix) fCovMatrix=new AliAODRedCov<6>(*trk.fCovMatrix);
     else fCovMatrix=NULL;
     fProdVertex = trk.fProdVertex;
 
-    fCharge = trk.fCharge;
-    fITSMuonClusterMap = trk.fITSMuonClusterMap;
     SetUsedForVtxFit(trk.GetUsedForVtxFit());
     SetUsedForPrimVtxFit(trk.GetUsedForPrimVtxFit());
-    fType = trk.fType;
+
+    delete fDetPid;
+    if(trk.fDetPid) fDetPid=new AliAODPid(*trk.fDetPid);
+    else fDetPid=NULL;
   }
 
   return *this;
@@ -185,47 +226,44 @@ AliAODTrack& AliAODTrack::operator=(const AliAODTrack& trk)
 Double_t AliAODTrack::M(AODTrkPID_t pid) const
 {
   // Returns the mass.
-  // In the case of elementary particles the hard coded mass values were taken 
-  // from the PDG. In all cases the errors on the values do not affect 
-  // the last digit.
-  
+  // Masses for nuclei don't exist in the PDG tables, therefore they were put by hand.
 
   switch (pid) {
 
   case kElectron :
-    return 0.000510999;
+    return 0.000510999; //TDatabasePDG::Instance()->GetParticle(11/*::kElectron*/)->Mass();
     break;
 
   case kMuon :
-    return 0.1056584;
+    return 0.1056584; //TDatabasePDG::Instance()->GetParticle(13/*::kMuonMinus*/)->Mass();
     break;
 
   case kPion :
-    return 0.13957;
+    return 0.13957; //TDatabasePDG::Instance()->GetParticle(211/*::kPiPlus*/)->Mass();
     break;
 
   case kKaon :
-    return 0.4937;
+    return 0.4937; //TDatabasePDG::Instance()->GetParticle(321/*::kKPlus*/)->Mass();
     break;
 
   case kProton :
-    return 0.9382720;
+    return 0.9382720; //TDatabasePDG::Instance()->GetParticle(2212/*::kProton*/)->Mass();
     break;
 
   case kDeuteron :
-    return 1.8756;
+    return 1.8756; //TDatabasePDG::Instance()->GetParticle(1000010020)->Mass();
     break;
 
   case kTriton :
-    return 2.8089;
+    return 2.8089; //TDatabasePDG::Instance()->GetParticle(1000010030)->Mass();
     break;
 
   case kHelium3 :
-    return 2.8084;
+    return 2.8084; //TDatabasePDG::Instance()->GetParticle(1000020030)->Mass();
     break;
 
   case kAlpha :
-    return 3.7274;
+    return 3.7274; //TDatabasePDG::Instance()->GetParticle(1000020040)->Mass();
     break;
 
   case kUnknown :
@@ -253,7 +291,7 @@ Double_t AliAODTrack::E(AODTrkPID_t pid) const
 //______________________________________________________________________________
 Double_t AliAODTrack::Y(AODTrkPID_t pid) const
 {
-  // Returns the energy of the particle of a given pid.
+  // Returns the rapidity of a particle of a given pid.
   
   if (pid != kUnknown) { // particle was identified
     Double_t e = E(pid);
@@ -271,7 +309,7 @@ Double_t AliAODTrack::Y(AODTrkPID_t pid) const
 //______________________________________________________________________________
 Double_t AliAODTrack::Y(Double_t m) const
 {
-  // Returns the energy of the particle of a given mass.
+  // Returns the rapidity of a particle of a given mass.
   
   if (m >= 0.) { // mass makes sense
     Double_t e = E(m);
@@ -323,10 +361,10 @@ void AliAODTrack::ConvertAliPIDtoAODPID()
   // Everything else has to be set to zero.
 
   fPID[kDeuteron] = 0.;
-  fPID[kTriton] = 0.;
-  fPID[kHelium3] = 0.;
-  fPID[kAlpha] = 0.;
-  fPID[kUnknown] = 0.;
+  fPID[kTriton]   = 0.;
+  fPID[kHelium3]  = 0.;
+  fPID[kAlpha]    = 0.;
+  fPID[kUnknown]  = 0.;
   
   return;
 }
@@ -335,16 +373,16 @@ void AliAODTrack::ConvertAliPIDtoAODPID()
 //______________________________________________________________________________
 template <class T> void AliAODTrack::SetP(const T *p, const Bool_t cartesian) 
 {
-  // set the momentum
+  // Set the momentum
 
   if (p) {
     if (cartesian) {
       Double_t pt2 = p[0]*p[0] + p[1]*p[1];
-      Double_t P = TMath::Sqrt(pt2 + p[2]*p[2]);
+      Double_t pp  = TMath::Sqrt(pt2 + p[2]*p[2]);
       
       fMomentum[0] = TMath::Sqrt(pt2); // pt
-      fMomentum[1] = (pt2 != 0.) ? TMath::ATan2(p[1], p[0]) : -999; // phi
-      fMomentum[2] = (P != 0.) ? TMath::ACos(p[2]/P) : -999.; // theta
+      fMomentum[1] = (pt2 != 0.) ? TMath::Pi()+TMath::ATan2(-p[1], -p[0]) : -999; // phi
+      fMomentum[2] = (pp != 0.) ? TMath::ACos(p[2] / pp) : -999.; // theta
     } else {
       fMomentum[0] = p[0];  // pt
       fMomentum[1] = p[1];  // phi
@@ -410,11 +448,12 @@ void AliAODTrack::Print(Option_t* /* option */) const
   printf("       phi = %f\n", Phi());
   printf("  chi2/NDF = %f\n", Chi2perNDF());
   printf("    charge = %d\n", Charge());
-  printf(" PID object: %p\n", PID());
 }
 
-void AliAODTrack::SetMatchTrigger(Int_t MatchTrigger){
-  switch(MatchTrigger){
+void AliAODTrack::SetMatchTrigger(Int_t matchTrig){
+//
+// Set the MUON trigger information
+  switch(matchTrig){
     case 0: // 0 track does not match trigger
       fITSMuonClusterMap=fITSMuonClusterMap&0x3fffffff;
       break;
@@ -429,15 +468,13 @@ void AliAODTrack::SetMatchTrigger(Int_t MatchTrigger){
       break;
     default:
       fITSMuonClusterMap=fITSMuonClusterMap&0x3fffffff;
-      printf("AliAODTrack::SetMatchTrigger unknown case for MatchTrigger: %d\n",MatchTrigger);
+      AliWarning(Form("unknown case for matchTrig: %d\n",matchTrig));
   }
 }
 
-void AliAODTrack::SetHitsPatternInTrigCh(UShort_t hitsPatternInTrigCh){
-  fITSMuonClusterMap=(fITSMuonClusterMap&0xffff00ff)|(hitsPatternInTrigCh<<8);
-}
-
-Int_t AliAODTrack::HitsMT(Int_t istation, Int_t iplane, Char_t *cathode){
+Int_t AliAODTrack::HitsMT(Int_t istation, Int_t iplane, Option_t *cathode){
+//
+// Retrieve hit information for MUON identified by  (station, plane, cathode)
   if(cathode){
     if(cathode[0]=='x'||cathode[0]=='X'){
       if(istation==1){
@@ -500,17 +537,26 @@ Int_t AliAODTrack::HitsMT(Int_t istation, Int_t iplane, Char_t *cathode){
 }
 
 Int_t AliAODTrack::HitsMuonChamber(Int_t MuonChamber){
-  switch(MuonChamber){
-    case 11:
-      return HitsMT(1,1);
-    case 12:
-      return HitsMT(1,2);
-    case 13:
-      return HitsMT(2,1);
-    case 14:
-      return HitsMT(2,2);
-    default:
-      printf("Unknown MUON chamber: %d\n",MuonChamber);
-      return 0;
+  //
+  // Retrieve hit information for MUON Tracker/Trigger Chamber
+  // WARNING: chamber number start from 1 instead of 0
+  
+  if (MuonChamber > 0 && MuonChamber < 11) {
+    return ((GetMUONClusterMap() & BIT(MuonChamber-1)) != 0) ? 1 : 0;
+  } else {
+    switch(MuonChamber){
+      case 11:
+       return HitsMT(1,1);
+      case 12:
+       return HitsMT(1,2);
+      case 13:
+       return HitsMT(2,1);
+      case 14:
+       return HitsMT(2,2);
+      default:
+       printf("Unknown MUON chamber: %d\n",MuonChamber);
+       return 0;
+    }
   }
 }
+