#include "AliKalmanTrack.h"
#endif
-#ifndef ALIESDTRACK_H
-#include "AliESDtrack.h"
-#endif
-
#ifndef ALITRDSEEDV1_H
#include "AliTRDseedV1.h"
#endif
-class AliTRDcluster;
class AliESDtrack;
+class AliTRDcluster;
class AliTRDReconstructor;
class AliTRDtrackV1 : public AliKalmanTrack
{
+ friend class AliHLTTRDTrack; // allow HLT special access
public:
- enum { kMAXCLUSTERSPERTRACK = 210 };
-
- enum { kNdet = 540
- , kNstacks = 90
- , kNplane = AliESDtrack::kTRDnPlanes
- , kNcham = 5
- , kNsect = 18
- , kNslice = 3
- , kNMLPslice = 8 };
+ enum ETRDtrackSize {
+ kNdet = AliTRDgeometry::kNdet
+ ,kNstacks = AliTRDgeometry::kNstack*AliTRDgeometry::kNsector
+ ,kNplane = AliTRDgeometry::kNlayer
+ ,kNcham = AliTRDgeometry::kNstack
+ ,kNsect = AliTRDgeometry::kNsector
+ ,kNslice = 3
+ ,kNMLPslice = 8
+ ,kMAXCLUSTERSPERTRACK = 210
+ };
- enum AliTRDPIDMethod {
- kNN = 0
- , kLQ = 1 };
+ // bits from 0-13 are reserved by ROOT (see TObject.h)
+ enum ETRDtrackStatus {
+ kOwner = BIT(14)
+ ,kStopped = BIT(15)
+ ,kKink = BIT(16)
+ ,kPrimary = BIT(17)
+ };
+
+ // propagation/update error codes (up to 4 bits)
+ enum ETRDtrackError {
+ kProlongation = 1
+ ,kPropagation
+ ,kAdjustSector
+ ,kSnp
+ ,kTrackletInit
+ ,kUpdate
+ ,kUnknown = 0xff
+ };
+
+ // data/clusters/tracklet error codes (up to 4 bits/layer)
+ enum ETRDlayerError {
+ kGeometry = 1
+ ,kBoundary
+ ,kNoClusters
+ ,kNoAttach
+ ,kNoClustersTracklet
+ ,kNoFit
+ ,kChi2
+ };
AliTRDtrackV1();
- AliTRDtrackV1(AliTRDseedV1 *trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha);
+ AliTRDtrackV1(AliTRDseedV1 * const trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha);
AliTRDtrackV1(const AliESDtrack &ref);
AliTRDtrackV1(const AliTRDtrackV1 &ref);
virtual ~AliTRDtrackV1();
- AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref) { *(new(this) AliTRDtrackV1(ref)); return *this; }
-
+ AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref);
+ virtual void Copy(TObject &ref) const;
+
Bool_t CookPID();
Bool_t CookLabel(Float_t wrong);
AliTRDtrackV1* GetBackupTrack() const {return fBackupTrack;}
Double_t GetBudget(Int_t i) const { return fBudget[i];}
- Double_t GetC() const { return AliExternalTrackParam::GetC(GetBz());}
+ AliTRDcluster* GetCluster(Int_t id);
Int_t GetClusterIndex(Int_t id) const;
Float_t GetEdep() const {return fDE;}
- inline Float_t GetMomentum(Int_t plane) const;
+ Int_t GetESDid() const {return fESDid;}
+ inline Float_t GetMomentum(Int_t plane=-1) const;
inline Int_t GetNCross();
inline Int_t GetNumberOfTracklets() const;
- Double_t GetPIDsignal() const {return 0.;}
+ Double_t GetPIDsignal() const { return 0.;}
Double_t GetPID(Int_t is) const { return (is >=0 && is < AliPID::kSPECIES) ? fPID[is] : -1.;}
- Double_t GetPredictedChi2(const AliTRDseedV1 *tracklet) const;
+ UChar_t GetNumberOfTrackletsPID() const;
+ Double_t GetPredictedChi2(const AliTRDseedV1 *tracklet, Double_t *cov) const;
Double_t GetPredictedChi2(const AliCluster* /*c*/) const { return 0.0; }
- Int_t GetProlongation(Double_t xk, Double_t &y, Double_t &z);
- AliTRDseedV1* GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : 0x0;}
- Int_t GetTrackletIndex(Int_t plane) const { return (plane>=0 && plane<kNplane) ? fTrackletIndex[plane] : -1;}
- UShort_t* GetTrackletIndexes() {return &fTrackletIndex[0];}
-
- Bool_t IsOwner() const {return TestBit(1);};
- Bool_t IsStopped() const {return TestBit(2);};
+ Int_t GetProlongation(Double_t xk, Double_t &y, Double_t &z) const;
+ inline UChar_t GetStatusTRD(Int_t ly=-1) const;
+ Int_t GetSector() const;
+ AliTRDseedV1* GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : NULL;}
+ Int_t GetTrackletIndex(Int_t plane) const { return (plane>=0 && plane<kNplane) ? fTrackletIndex[plane] : -1;}
+ AliExternalTrackParam*
+ GetTrackIn() const { return fTrackLow;}
+ AliExternalTrackParam*
+ GetTrackOut() const { return fTrackHigh;}
+ const Int_t* GetTrackletIndexes() const { return &fTrackletIndex[0];}
- void MakeBackupTrack();
+ Bool_t IsEqual(const TObject *inTrack) const;
+ Bool_t IsKink() const { return TestBit(kKink);}
+ Bool_t IsOwner() const { return TestBit(kOwner);};
+ Bool_t IsPrimary() const { return TestBit(kPrimary);};
+ Bool_t IsStopped() const { return TestBit(kStopped);};
+ Bool_t IsElectron() const;
+ inline static Bool_t IsTrackError(ETRDtrackError error, UInt_t status);
+ inline static Bool_t IsLayerError(ETRDlayerError error, Int_t layer, UInt_t status);
+
+ Int_t MakeBackupTrack();
+ void Print(Option_t *o="") const;
+
Bool_t PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
Int_t PropagateToR(Double_t xr, Double_t step);
Bool_t Rotate(Double_t angle, Bool_t absolute = kFALSE);
void SetBudget(Int_t i, Double_t b) {if(i>=0 && i<3) fBudget[i] = b;}
+ void SetEdep(Double32_t inDE){fDE = inDE;};
+ void SetESDid(Int_t id) {fESDid = id;}
+ void SetKink(Bool_t k) { SetBit(kKink, k);}
+ void SetPrimary(Bool_t k) { SetBit(kPrimary, k);}
void SetNumberOfClusters();
+ UChar_t SetNumberOfTrackletsPID(Bool_t recalc);
void SetOwner();
- void SetStopped(Bool_t stop) {SetBit(2, stop);}
- void SetTracklet(AliTRDseedV1 *trklt, Int_t index);
+ void SetPID(Short_t is, Double_t inPID){if (is >=0 && is < AliPID::kSPECIES) fPID[is]=inPID;};
+ void SetPIDquality(UChar_t /*inPIDquality*/) const {/*fPIDquality = inPIDquality*/;};
+ inline void SetStatus(UChar_t stat, Int_t ly=-1);
+ void SetStopped(Bool_t stop) {SetBit(kStopped, stop);}
+ void SetTracklet(AliTRDseedV1 *const trklt, Int_t index);
+ void SetTrackIn();
+ void SetTrackOut(const AliExternalTrackParam *op=NULL);
inline void SetReconstructor(const AliTRDReconstructor *rec);
inline Float_t StatusForTOF();
- Bool_t Update(AliTRDseedV1 *tracklet, Double_t chi2);
- //Bool_t Update(const AliTRDcluster *c, Double_t chi2, Int_t index, Double_t h01){ return AliTRDtrack::Update(c,chi2,index,h01); };
- Bool_t Update(const AliCluster *, Double_t, Int_t) { return kFALSE; };
+ void UnsetTracklet(Int_t plane);
+ Bool_t Update(const AliCluster *, Double_t, Int_t) { return kFALSE; };
+ void UpdateChi2(Float_t chi2);
void UpdateESDtrack(AliESDtrack *t);
-protected:
- Double_t GetBz() const;
-
private:
- UChar_t fPIDquality; // No of planes used for PID calculation
- UShort_t fTrackletIndex[kNplane];// Tracklets index in the tracker list
- Double32_t fPID[AliPID::kSPECIES];// PID probabilities
- Double32_t fBudget[3]; // Integrated material budget
- Double32_t fDE; // Integrated delta energy
- AliTRDseedV1 *fTracklet[kNplane]; // Tracklets array defining the track
- AliTRDtrackV1 *fBackupTrack; // Backup track
-
-
- ClassDef(AliTRDtrackV1, 2) // new TRD track
+ UInt_t fStatus; // Bit map for the status of propagation
+ Int_t fTrackletIndex[kNplane];// Tracklets index in the tracker list
+ Int_t fESDid; // ESD track id
+ Double32_t fPID[AliPID::kSPECIES]; // PID probabilities
+ Double32_t fBudget[3]; // Integrated material budget
+ Double32_t fDE; // Integrated delta energy
+ const AliTRDReconstructor *fkReconstructor;//! reconstructor link
+ AliTRDtrackV1 *fBackupTrack; //! Backup track
+ AliTRDseedV1 *fTracklet[kNplane]; // Tracklets array defining the track
+ AliExternalTrackParam *fTrackLow; // parameters of the track which enter TRD from below (TPC)
+ AliExternalTrackParam *fTrackHigh; // parameters of the track which enter TRD from above (HMPID, PHOS)
+
+ ClassDef(AliTRDtrackV1, 7) // TRD track - tracklet based
};
//____________________________________________________
inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const
{
- return plane >=0 && plane < kNplane && fTrackletIndex[plane] != 0xff ? fTracklet[plane]->GetMomentum() : -1.;
+// Return ESD momentum stored in the tracklet reconstructed in layer = "plane".
+// By default returns the ESD momentum in first tracklet attached to track
+ if(plane==-1){
+ for(Int_t i(0); i<kNplane; i++){
+ if(fTracklet[i]) return fTracklet[i]->GetMomentum();
+ }
+ } else if( plane >=0 && plane < kNplane){
+ if(fTracklet[plane]) return fTracklet[plane]->GetMomentum();
+ }
+ return -1.;
}
//____________________________________________________
return n;
}
+//____________________________________________________
+inline UChar_t AliTRDtrackV1::GetStatusTRD(Int_t ly) const
+{
+ if(ly>=-1 && ly<kNplane) return (fStatus>>((ly+1)*4))&0xf;
+ return kUnknown;
+}
+
+//____________________________________________________
+inline Bool_t AliTRDtrackV1::IsTrackError(ETRDtrackError error, UInt_t status)
+{
+ return (status&0xf)==UChar_t(error);
+}
+
+//____________________________________________________
+inline Bool_t AliTRDtrackV1::IsLayerError(ETRDlayerError error, Int_t ly, UInt_t status)
+{
+ if(ly>=kNplane || ly<0) return kFALSE;
+ return ((status>>((ly+1)*4))&0xf) == UChar_t(error);
+}
//____________________________________________________
inline void AliTRDtrackV1::SetReconstructor(const AliTRDReconstructor *rec)
if(!fTracklet[ip]) continue;
fTracklet[ip]->SetReconstructor(rec);
}
+ fkReconstructor = rec;
+}
+
+//____________________________________________________
+inline void AliTRDtrackV1::SetStatus(UChar_t status, Int_t ly)
+{
+ if(ly<kNplane) fStatus|=((status&0xf)<<((ly+1)*4));
+ return;
}
//____________________________________________________________________________
inline Float_t AliTRDtrackV1::StatusForTOF()
{
- //
+ // OBSOLETE
// Defines the status of the TOF extrapolation
//
#endif
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff