/* $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(),
- fChi2(-999.),
- fID(-999),
+ AliVTrack(),
+ fChi2perNDF(-999.),
+ fChi2MatchTrigger(0.),
+ fFlags(0),
fLabel(-999),
- fCovMatrix(NULL),
- fProdVertex(0x0),
+ fITSMuonClusterMap(0),
+ fFilterMap(0),
+ fID(-999),
fCharge(-99),
- fITSClusterMap(0),
- fType(kUndef)
+ fType(kUndef),
+ 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,
UChar_t itsClusMap,
Double_t pid[10],
AliAODVertex *prodVertex,
+ Bool_t usedForVtxFit,
Bool_t usedForPrimVtxFit,
- AODTrk_t ttype) :
- AliVirtualParticle(),
- fChi2(-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),
- fITSClusterMap(itsClusMap),
- fType(ttype)
+ fType(ttype),
+ 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,
UChar_t itsClusMap,
Float_t pid[10],
AliAODVertex *prodVertex,
+ Bool_t usedForVtxFit,
Bool_t usedForPrimVtxFit,
- AODTrk_t ttype) :
- AliVirtualParticle(),
- fChi2(-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),
- fITSClusterMap(itsClusMap),
- fType(ttype)
+ fType(ttype),
+ 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);
}
//______________________________________________________________________________
{
// destructor
delete fCovMatrix;
+ delete fDetPid;
}
//______________________________________________________________________________
AliAODTrack::AliAODTrack(const AliAODTrack& trk) :
- AliVirtualParticle(trk),
- fChi2(trk.fChi2),
- fID(trk.fID),
+ AliVTrack(trk),
+ fChi2perNDF(trk.fChi2perNDF),
+ fChi2MatchTrigger(trk.fChi2MatchTrigger),
+ fFlags(trk.fFlags),
fLabel(trk.fLabel),
- fCovMatrix(NULL),
- fProdVertex(trk.fProdVertex),
+ fITSMuonClusterMap(trk.fITSMuonClusterMap),
+ fFilterMap(trk.fFilterMap),
+ fID(trk.fID),
fCharge(trk.fCharge),
- fITSClusterMap(trk.fITSClusterMap),
- fType(trk.fType)
+ fType(trk.fType),
+ 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);
-
}
//______________________________________________________________________________
// Assignment operator
if(this!=&trk) {
- AliVirtualParticle::operator=(trk);
+ AliVTrack::operator=(trk);
trk.GetP(fMomentum);
trk.GetPosition(fPosition);
trk.GetPID(fPID);
- fChi2 = trk.fChi2;
+ 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;
- fITSClusterMap = trk.fITSClusterMap;
+ SetUsedForVtxFit(trk.GetUsedForVtxFit());
SetUsedForPrimVtxFit(trk.GetUsedForPrimVtxFit());
- fType = trk.fType;
+
+ delete fDetPid;
+ if(trk.fDetPid) fDetPid=new AliAODPid(*trk.fDetPid);
+ else fDetPid=NULL;
}
return *this;
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 error on the values does 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 :
//______________________________________________________________________________
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);
//______________________________________________________________________________
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);
// 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;
}
//______________________________________________________________________________
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
printf(" 1/pt = %f\n", OneOverPt());
printf(" theta = %f\n", Theta());
printf(" phi = %f\n", Phi());
- printf(" chi2 = %f\n", Chi2());
+ printf(" chi2/NDF = %f\n", Chi2perNDF());
printf(" charge = %d\n", Charge());
- printf(" PID object: %p\n", PID());
+}
+
+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;
+ case 1: // 1 track match but does not pass pt cut
+ fITSMuonClusterMap=(fITSMuonClusterMap&0x3fffffff)|0x40000000;
+ break;
+ case 2: // 2 track match Low pt cut
+ fITSMuonClusterMap=(fITSMuonClusterMap&0x3fffffff)|0x80000000;
+ break;
+ case 3: // 3 track match High pt cut
+ fITSMuonClusterMap=fITSMuonClusterMap|0xc0000000;
+ break;
+ default:
+ fITSMuonClusterMap=fITSMuonClusterMap&0x3fffffff;
+ AliWarning(Form("unknown case for matchTrig: %d\n",matchTrig));
+ }
+}
+
+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){
+ if(iplane==1)
+ return (fITSMuonClusterMap&0x8000)?1:0;
+ else if(iplane==2)
+ return (fITSMuonClusterMap&0x4000)?1:0;
+ else
+ return 0;
+ }else if(istation==2){
+ if(iplane==1)
+ return (fITSMuonClusterMap&0x2000)?1:0;
+ else if(iplane==2)
+ return (fITSMuonClusterMap&0x1000)?1:0;
+ else
+ return 0;
+ }else{
+ return 0;
+ }
+ }else if(cathode[0]=='y'||cathode[0]=='Y'){
+ if(istation==1){
+ if(iplane==1)
+ return (fITSMuonClusterMap&0x0800)?1:0;
+ else if(iplane==2)
+ return (fITSMuonClusterMap&0x0400)?1:0;
+ else
+ return 0;
+ }else if(istation==2){
+ if(iplane==1)
+ return (fITSMuonClusterMap&0x0200)?1:0;
+ else if(iplane==2)
+ return (fITSMuonClusterMap&0x0100)?1:0;
+ else
+ return 0;
+ }else{
+ return 0;
+ }
+ }else{
+ return 0;
+ }
+ }else{
+ if(istation==1){
+ if(iplane==1)
+ return (HitsMT(1,1,"X")||HitsMT(1,1,"Y"))?1:0;
+ else if(iplane==2)
+ return (HitsMT(1,2,"X")||HitsMT(1,2,"Y"))?1:0;
+ else
+ return 0;
+ }else if(istation==2){
+ if(iplane==1)
+ return (HitsMT(2,1,"X")||HitsMT(2,1,"Y"))?1:0;
+ else if(iplane==2)
+ return (HitsMT(2,2,"X")||HitsMT(2,2,"Y"))?1:0;
+ else
+ return 0;
+ }else{
+ return 0;
+ }
+ }
+}
+
+Int_t AliAODTrack::HitsMuonChamber(Int_t MuonChamber){
+ //
+ // 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;
+ }
+ }
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff