X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=TRD%2FAliTRDtrackV1.h;h=4466768afb475e41abf8d42f666d6d7aad38bfcf;hb=641e1e0c9ec294390167408c36bcbc9e39cad155;hp=96ccd15991e7d54904ab682f542d23243e14f91f;hpb=181c7f7e7fa75313dff2c8da50b05ba673be733c;p=u%2Fmrichter%2FAliRoot.git diff --git a/TRD/AliTRDtrackV1.h b/TRD/AliTRDtrackV1.h index 96ccd15991e..4466768afb4 100644 --- a/TRD/AliTRDtrackV1.h +++ b/TRD/AliTRDtrackV1.h @@ -11,40 +11,64 @@ // // /////////////////////////////////////////////////////////////////////////////// -#ifndef ALIKALMANTRACK_H +//#ifndef ALIKALMANTRACK_H #include "AliKalmanTrack.h" -#endif - -#ifndef ALIESDTRACK_H -#include "AliESDtrack.h" -#endif +//#endif -#ifndef ALITRDSEEDV1_H +//#ifndef ALITRDSEEDV1_H #include "AliTRDseedV1.h" -#endif +//#endif class AliTRDcluster; class AliESDtrack; class AliTRDReconstructor; class AliTRDtrackV1 : public AliKalmanTrack { + friend class AliHLTTRDTrack; 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(); @@ -54,60 +78,95 @@ public: 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 =0 && plane=0 && plane =0 && plane=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); iGetMomentum(); + } + } else if( plane >=0 && plane < kNplane){ + if(fTracklet[plane]) return fTracklet[plane]->GetMomentum(); + } + return -1.; } //____________________________________________________ @@ -132,6 +191,25 @@ inline Int_t AliTRDtrackV1::GetNumberOfTracklets() const return n; } +//____________________________________________________ +inline UChar_t AliTRDtrackV1::GetStatusTRD(Int_t ly) const +{ + if(ly>=-1 && ly>((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) @@ -140,13 +218,21 @@ 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