X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=TRD%2FAliTRDtracker.h;h=63a65dcbf6ab50d575577fb02620180f14589735;hb=30f6c8cbe5b1c7ca16927082fe9cbd03d50fc374;hp=7cdd31a837a70cc9b722312f15e3bdca73c9eff9;hpb=69b55c5549cb0733252b0baadd41ef8972c11990;p=u%2Fmrichter%2FAliRoot.git diff --git a/TRD/AliTRDtracker.h b/TRD/AliTRDtracker.h index 7cdd31a837a..63a65dcbf6a 100644 --- a/TRD/AliTRDtracker.h +++ b/TRD/AliTRDtracker.h @@ -9,91 +9,52 @@ class TFile; class TTree; -class TParticle; -class TParticlePDG; +class TParticle;class TParticlePDG; class AliTRDgeometry; -class AliTRDparameter; class AliTRDtrack; class AliTRDtracklet; class AliTRDcluster; +class AliTRDseed; class AliESD; class TTreeSRedirector; -const unsigned kMaxLayersPerSector = 1000; -const unsigned kMaxTimeBinIndex = 216; // (30 drift + 6 ampl) * 6 planes -const unsigned kMaxClusterPerTimeBin = 7000; -const unsigned kZones = 5; -const Int_t kTrackingSectors = 18; - -class AliTRDseed : public TObject{ - friend class AliTRDtracker; - public: - AliTRDseed(); // default constructor - ~AliTRDseed(){}; // default constructor - static void EvaluateUni(Int_t nvectors, Double_t *data, Double_t &mean, Double_t &sigma, Int_t hh); - static Float_t FitRiemanTilt(AliTRDseed * seed, Bool_t error); - void UseClusters(); // use clusters - void Update(); // update information - without tilt correction - void CookLabels(); // cook label - void UpdateUsed(); - void Reset(); // reset seed - Bool_t isOK(){return fN2>8;} - private: - Float_t fTilt; // tilting angle - Float_t fPadLength; // pad length - Float_t fX0; // x0 position - Float_t fX[25]; // !x position - Float_t fY[25]; // !y position - Float_t fZ[25]; // !z position - Int_t fIndexes[25]; // !indexes - AliTRDcluster *fClusters[25]; // !clusters - Bool_t fUsable[25]; // !indication - usable cluster - Float_t fYref[2]; // reference y - Float_t fZref[2]; // reference z - Float_t fYfit[2]; // y fit position +derivation - Float_t fYfitR[2]; // y fit position +derivation - Float_t fZfit[2]; // z fit position - Float_t fZfitR[2]; // z fit position - Float_t fSigmaY; // "robust" sigma in Y - constant fit - Float_t fSigmaY2; // "robust" sigma in Y - line fit - Float_t fMeanz; // mean vaue of z - Float_t fZProb; // max probbable z - Int_t fLabels[2]; // labels - Int_t fN; // number of associated clusters - Int_t fN2; // number of not crossed - Int_t fNUsed; // number of used clusters - Int_t fFreq; // freq - Int_t fNChange; // change z counter - Float_t fMPads; // mean number of pads per cluster - // global - // - Float_t fC; // curvature - Float_t fCC; // curvature with constrain - Float_t fChi2; // global chi2 - Float_t fChi2Z; // global chi2 - private: - ClassDef(AliTRDseed,1) -}; - +/////////////////////////////////////////////////////////////////////////////// +// // +// The standard TRD tracker // +// Based on Kalman filltering approach // +// // +/////////////////////////////////////////////////////////////////////////////// class AliTRDtracker : public AliTracker { public: + enum { kMaxLayersPerSector = 1000 + , kMaxTimeBinIndex = 216 + , kMaxClusterPerTimeBin = 7000 + , kZones = 5 + , kTrackingSectors = 18 }; AliTRDtracker(); AliTRDtracker(const TFile *in); virtual ~AliTRDtracker(); - static Int_t Freq(Int_t n, const Int_t *inlist, Int_t *outlist, Bool_t down); + + static Int_t Freq(Int_t n, const Int_t *inlist, Int_t *outlist, Bool_t down); Int_t Clusters2Tracks(AliESD* event); Int_t PropagateBack(AliESD* event); Int_t RefitInward(AliESD* event); - + + // transformation + Int_t LocalToGlobalID(Int_t lid); + Int_t GlobalToLocalID(Int_t gid); + Bool_t Transform(AliTRDcluster * cluster); + // Int_t LoadClusters(TTree *cTree); void UnloadClusters(); AliCluster *GetCluster(Int_t index) const { if (index >= fNclusters) return NULL; return (AliCluster*) fClusters->UncheckedAt(index); }; + Bool_t GetTrackPoint(Int_t index, AliTrackPoint& p) const; virtual void CookLabel(AliKalmanTrack *t,Float_t wrong) const; virtual void UseClusters(const AliKalmanTrack *t, Int_t from=0) const; @@ -106,61 +67,45 @@ class AliTRDtracker : public AliTracker { Int_t CookSectorIndex(Int_t gs) const { return kTrackingSectors - 1 - gs; } AliTRDcluster * GetCluster(AliTRDtrack * track, Int_t plane, Int_t timebin, UInt_t &index); Int_t GetLastPlane(AliTRDtrack * track); //return last updated plane - Int_t FindClusters(Int_t sector, Int_t t0, Int_t t1, AliTRDtrack * track, Int_t *clusters, AliTRDtracklet& tracklet); - - Float_t GetSeedGap() const {return fgkSeedGap;} - Int_t GetMaxGap() const {return fMaxGap;} - Int_t GetTimeBinsPerPlane() const {return fTimeBinsPerPlane;} - Float_t GetSeedStep() const {return fgkSeedStep;} - Float_t GetSeedDepth() const {return fgkSeedDepth;} - Float_t GetSkipDepth() const {return fgkSkipDepth;} - Double_t GetMaxChi2() const {return fgkMaxChi2;} - Float_t GetMaxSeedC() const {return fgkMaxSeedC;} - Float_t GetMaxSeedTan() const {return fgkMaxSeedTan;} - Double_t GetSeedErrorSY() const {return fgkSeedErrorSY;} - Double_t GetSeedErrorSY3() const {return fgkSeedErrorSY3;} - Double_t GetSeedErrorSZ() const {return fgkSeedErrorSZ;} - Float_t GetLabelFraction() const {return fgkLabelFraction;} - Float_t GetWideRoad() const {return fgkWideRoad;} - - Float_t GetMinClustersInTrack() const {return fgkMinClustersInTrack;} - Float_t GetMinClustersInSeed() const {return fgkMinClustersInSeed;} - Float_t GetMaxSeedDeltaZ() const {return fgkMaxSeedDeltaZ;} - Float_t GetMaxSeedVertexZ() const {return fgkMaxSeedVertexZ;} + Int_t FindClusters(Int_t sector, Int_t t0, Int_t t1, AliTRDtrack *track + , Int_t *clusters, AliTRDtracklet& tracklet); + + Int_t GetTimeBinsPerPlane() const { return fTimeBinsPerPlane; } + Double_t GetMaxChi2() const { return fgkMaxChi2; } + Float_t GetLabelFraction() const { return fgkLabelFraction; } + Float_t GetMinClustersInTrack() const { return fgkMinClustersInTrack; } // x <-> timebin conversions useful in analysis macros - Double_t GetX(Int_t sec, Int_t plane, Int_t localTB) const; - Double_t GetX(Int_t sec, Int_t pl) const { - return fTrSec[sec]->GetLayer(pl)->GetX(); } - Int_t GetGlobalTimeBin(Int_t sec, Int_t plane, Int_t localTB) const { - return fTrSec[sec]->CookTimeBinIndex(plane,localTB); } - Double_t GetLayerNumber(Int_t sec, Double_t x) const { - return fTrSec[sec]->GetLayerNumber(x); } + Double_t GetX(Int_t sec, Int_t plane, Int_t localTB) const; + Double_t GetX(Int_t sec, Int_t pl) const + { return fTrSec[sec]->GetLayer(pl)->GetX(); } + Int_t GetGlobalTimeBin(Int_t sec, Int_t plane, Int_t localTB) const + { return fTrSec[sec]->CookTimeBinIndex(plane,localTB); } + Double_t GetLayerNumber(Int_t sec, Double_t x) const + { return fTrSec[sec]->GetLayerNumber(x); } + + protected: class AliTRDpropagationLayer { + // ***************** internal class ******************* public: + AliTRDpropagationLayer(Double_t x, Double_t dx, Double_t rho, - Double_t x0, Int_t tbIndex); + Double_t x0, Int_t tbIndex, Int_t plane); + + ~AliTRDpropagationLayer() { if(fTimeBinIndex >= 0) { delete[] fClusters; delete[] fIndex; }} - ~AliTRDpropagationLayer() { - if(fTimeBinIndex >= 0) { delete[] fClusters; delete[] fIndex; } - } - void InsertCluster(AliTRDcluster *c, UInt_t index); + void InsertCluster(AliTRDcluster *c, UInt_t index); operator Int_t() const {return fN;} AliTRDcluster* operator[](Int_t i) {return fClusters[i];} UInt_t GetIndex(Int_t i) const {return fIndex[i];} Double_t GetX() const { return fX; } Double_t GetdX() const { return fdX; } - Double_t GetRho() const { return fRho; } - Double_t GetX0() const { return fX0; } Int_t GetTimeBinIndex() const { return fTimeBinIndex; } - Bool_t GetPropagationParameters(Double_t y, Double_t z, - Double_t &dx, Double_t &rho, Double_t &x0, - Bool_t &lookForCluster) const; - Int_t GetZone( Double_t z) const; + Int_t GetPlane() const { return fPlane;} Int_t Find(Float_t y) const; - Int_t FindNearestCluster(Float_t y, Float_t z, Float_t maxroad, Float_t maxroad) const; + Int_t FindNearestCluster(Float_t y, Float_t z, Float_t maxroad, Float_t maxroadz) const; void SetZmax(Int_t cham, Double_t center, Double_t w) { fZc[cham] = center; fZmax[cham] = w; } void SetZ(Double_t* center, Double_t *w, Double_t *wsensitive); @@ -176,39 +121,45 @@ class AliTRDtracker : public AliTracker { Bool_t IsSensitive() const {return (fTimeBinIndex>=0)? kTRUE: kFALSE;} - void Clear() {for(Int_t i=0; iGetX(); } @@ -217,121 +168,60 @@ class AliTRDtracker : public AliTracker { Int_t GetInnerTimeBin() const; Int_t GetOuterTimeBin() const; Int_t GetLayerNumber(Int_t tb) const {return fTimeBinIndex[tb];} - Float_t GetTzeroShift() const { return fTzeroShift; } Int_t Find(Double_t x) const; void InsertLayer(AliTRDpropagationLayer* pl); + // AliTRDpropagationLayer* operator[](Int_t i) { return fLayers[i]; } AliTRDpropagationLayer* GetLayer(Int_t i) { return fLayers[i]; } Int_t CookTimeBinIndex(Int_t plane, Int_t localTB) const; private: - Int_t fN; // total number of layers - AliTRDgeometry *fGeom; // geometry - AliTRDparameter *fPar; // parameter - AliTRDpropagationLayer *fLayers[kMaxLayersPerSector]; // layers - Int_t fTimeBinIndex[kMaxTimeBinIndex]; // time bin index - Float_t fTzeroShift; // T0 shift in cm - Int_t fGeomSector; // sector # in AliTRDgeometry - }; - - protected: - - friend class AliTRDtracker::AliTRDtrackingSector; - - AliTRDgeometry *fGeom; // Pointer to TRD geometry - AliTRDparameter *fPar; // Pointer to TRD parameter - - AliTRDtrackingSector *fTrSec[kTrackingSectors]; // array of tracking sectors; - Int_t fNclusters; // Number of clusters in TRD - TObjArray *fClusters; // List of clusters for all sectors - - Int_t fNseeds; // Number of track seeds - TObjArray *fSeeds; // List of track seeds - - Int_t fNtracks; // Number of reconstructed tracks - TObjArray *fTracks; // List of reconstructed tracks - - Float_t fSY2corr; // Correction coefficient for - // cluster SigmaY2 - Float_t fSZ2corr; // Correction coefficient for - // cluster SigmaZ2 - - static const Float_t fgkSeedGap; // Distance between inner and outer - // time bin in seeding - // (fraction of all time bins) - - static const Float_t fgkSeedStep; // Step in iterations - static const Float_t fgkSeedDepth; // Fraction of TRD allocated for seeding - static const Float_t fgkSkipDepth; // Fraction of TRD which can be skipped - // in track prolongation - Int_t fTimeBinsPerPlane; // number of sensitive timebins per plane - Int_t fMaxGap; // max gap (in time bins) in the track - // in track prolongation - - static const Double_t fgkMaxChi2; // max increment in track chi2 - - static const Float_t fgkMinClustersInTrack; // min number of clusters in track - // out of total timebins - - static const Float_t fgkMinFractionOfFoundClusters; // min found clusters - // out of expected - - static const Float_t fgkMinClustersInSeed; // min fraction of clusters in seed - static const Float_t fgkMaxSeedDeltaZ; // max dZ in MakeSeeds - static const Float_t fgkMaxSeedDeltaZ12; // max abs(z1-z2) in MakeSeeds - static const Float_t fgkMaxSeedC; // max initial curvature in MakeSeeds - static const Float_t fgkMaxSeedTan; // max initial Tangens(lambda) in MakeSeeds - static const Float_t fgkMaxSeedVertexZ; // max vertex Z in MakeSeeds - static const Double_t fgkSeedErrorSY; // sy parameter in MakeSeeds - static const Double_t fgkSeedErrorSY3; // sy3 parameter in MakeSeeds - static const Double_t fgkSeedErrorSZ; // sz parameter in MakeSeeds - static const Float_t fgkLabelFraction; // min fraction of same label - static const Float_t fgkWideRoad; // max road width in FindProlongation - // - static const Double_t fgkDriftCorrection; // correction coeficients for drift velocity - static const Double_t fgkOffset; // correction coeficients - static const Double_t fgkOffsetX; // correction coeficients offset in X - static const Double_t fgkCoef; // correction coeficients - static const Double_t fgkMean; // correction coeficients - static const Double_t fgkExB; // correction coeficients - - Bool_t fVocal; // Whatever... - Bool_t fAddTRDseeds; // Something else - - Bool_t fNoTilt; // No tilt, or what? - Bool_t fHoles[5][18]; // holes + Int_t fN; // total number of layers + AliTRDgeometry *fGeom; // geometry + AliTRDpropagationLayer *fLayers[kMaxLayersPerSector]; // layers + Int_t fTimeBinIndex[kMaxTimeBinIndex]; // time bin index + Int_t fGeomSector; // sector # in AliTRDgeometry + + }; + + AliTRDgeometry *fGeom; // Pointer to TRD geometry + AliTRDtrackingSector *fTrSec[kTrackingSectors]; // array of tracking sectors; + Int_t fNclusters; // Number of clusters in TRD + TObjArray *fClusters; // List of clusters for all sectors + Int_t fNseeds; // Number of track seeds + TObjArray *fSeeds; // List of track seeds + Int_t fNtracks; // Number of reconstructed tracks + TObjArray *fTracks; // List of reconstructed tracks + Int_t fTimeBinsPerPlane; // timebins per plane in track prolongation + + static const Double_t fgkMaxChi2; // max increment in track chi2 + static const Float_t fgkMinClustersInTrack; // min number of clusters in track + static const Float_t fgkLabelFraction; // min fraction of same label + static const Double_t fgkMaxSnp; // maximal snp for tracking + static const Double_t fgkMaxStep; // maximal step for tracking + + Bool_t fAddTRDseeds; // Something else + Bool_t fNoTilt; // No tilt, or what? + Bool_t fHoles[5][18]; // holes - Bool_t AdjustSector(AliTRDtrack *track); + Bool_t AdjustSector(AliTRDtrack *track); - - // Barrel tracks [SR, 03.04.2003] + private: - static const Int_t fgkFirstPlane; // Id of the first (innermost) reference plane - static const Int_t fgkLastPlane; // Id of the last (outermost) reference plane + AliTRDtrack *RegisterSeed(AliTRDseed *seeds, Double_t *params); + void MakeSeedsMI(Int_t inner, Int_t outer, AliESD *esd=0); - private: - AliTRDtrack * RegisterSeed(AliTRDseed * seeds, Double_t *params); - virtual void MakeSeeds(Int_t inner, Int_t outer, Int_t turn); - void MakeSeedsMI(Int_t inner, Int_t outer, AliESD *esd=0); - - Int_t FollowProlongation(AliTRDtrack& t, Int_t rf); - Int_t FollowProlongationG(AliTRDtrack& t, Int_t rf); - Int_t FollowBackProlongation(AliTRDtrack& t); - Int_t FollowBackProlongationG(AliTRDtrack& t); - Int_t Refit(AliTRDtrack& t, Int_t rf); - void CookdEdxTimBin(AliTRDtrack& t); - - Int_t PropagateToTPC(AliTRDtrack& t); - Int_t PropagateToOuterPlane(AliTRDtrack& t, Double_t x); - - void SetSY2corr(Float_t w) {fSY2corr = w;} - void SetSZ2corr(Float_t w) {fSZ2corr = w;} - Double_t ExpectedSigmaY2(Double_t r, Double_t tgl, Double_t pt) const; - Double_t ExpectedSigmaZ2(Double_t r, Double_t tgl) const; - TTreeSRedirector *fDebugStreamer; //!debug streamer - ClassDef(AliTRDtracker,2) // manager base class + Int_t FollowBackProlongation(AliTRDtrack& t); + Int_t FollowProlongation(AliTRDtrack& t); + void CookdEdxTimBin(AliTRDtrack& t); + Int_t PropagateToX(AliTRDtrack& t, Double_t xToGo, Double_t maxStep); + Double_t ExpectedSigmaY2(Double_t r, Double_t tgl, Double_t pt) const; + Double_t ExpectedSigmaZ2(Double_t r, Double_t tgl) const; -}; + TTreeSRedirector *fDebugStreamer; //!debug streamer + ClassDef(AliTRDtracker,2) // TRD tracker + +}; #endif