#ifndef ALITRDTRACKER_H #define ALITRDTRACKER_H /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * * See cxx source for full Copyright notice */ /* $Id$ */ //////////////////////////////////////////////////////////////////////////// // // // The TRD tracker // // // //////////////////////////////////////////////////////////////////////////// #include "AliTracker.h" #include "TObjArray.h" class TFile; class TTree; class TParticle; class TParticlePDG; class AliTRDgeometry; class AliTRDtrack; class AliTRDtracklet; class AliTRDcluster; class AliTRDseed; class AliESD; class TTreeSRedirector; /////////////////////////////////////////////////////////////////////////////// // // // The standard TRD tracker // // Based on Kalman filltering approach // // // /////////////////////////////////////////////////////////////////////////////// class AliTRDtracker : public AliTracker { public: enum { kMaxLayersPerSector = 1000 , kMaxTimeBinIndex = 216 , kMaxClusterPerTimeBin = 2300 , kZones = 5 , kTrackingSectors = 18 }; AliTRDtracker(); AliTRDtracker(const AliTRDtracker &t); AliTRDtracker(const TFile *in); virtual ~AliTRDtracker(); AliTRDtracker &operator=(const AliTRDtracker &/*t*/) { return *this; } void SetAddTRDseeds() { fAddTRDseeds = kTRUE; } void SetNoTilt() { fNoTilt = kTRUE; } Int_t GetTimeBinsPerPlane() const { return fTimeBinsPerPlane; } Double_t GetMaxChi2() const { return fgkMaxChi2; } Float_t GetLabelFraction() const { return fgkLabelFraction; } Float_t GetMinClustersInTrack() const { return fgkMinClustersInTrack; } Int_t GetLastPlane(AliTRDtrack *track); Double_t GetTiltFactor(const AliTRDcluster *c); Bool_t GetTrackPoint(Int_t index, AliTrackPoint& p) const; 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); } AliCluster *GetCluster(Int_t index) const { if (index >= fNclusters) return NULL; return (AliCluster *) fClusters->UncheckedAt(index); } 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); virtual void CookLabel(AliKalmanTrack *t, Float_t wrong) const; Int_t CookSectorIndex(Int_t gs) const { return kTrackingSectors - 1 - gs; } Int_t LocalToGlobalID(Int_t lid); Int_t GlobalToLocalID(Int_t gid); Bool_t Transform(AliTRDcluster *cluster); Int_t LoadClusters(TTree *cTree); void UnloadClusters(); virtual void UseClusters(const AliKalmanTrack *t, Int_t from = 0) const; Int_t ReadClusters(TObjArray *array, TTree *in) const; AliTRDcluster *GetCluster(AliTRDtrack *track, Int_t plane, Int_t timebin, UInt_t &index); Int_t FindClusters(Int_t sector, Int_t t0, Int_t t1, AliTRDtrack *track , Int_t *clusters, AliTRDtracklet &tracklet); protected: class AliTRDpropagationLayer { public: AliTRDpropagationLayer(Double_t x, Double_t dx, Double_t rho , Double_t x0, Int_t tbIndex, Int_t plane); AliTRDpropagationLayer(const AliTRDpropagationLayer &/*p*/); ~AliTRDpropagationLayer() { if (fTimeBinIndex >= 0) { delete[] fClusters; delete[] fIndex; } } AliTRDpropagationLayer &operator=(const AliTRDpropagationLayer &/*p*/) { return *this; } operator Int_t() const { return fN; } AliTRDcluster *operator[](Int_t i) { return fClusters[i]; } 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); void SetHoles(Bool_t* holes); void SetYmax(Double_t w, Double_t wsensitive) { fYmax = w; fYmaxSensitive = wsensitive; } void SetHole(Double_t Zmax, Double_t Ymax , Double_t rho = 1.29e-3, Double_t x0 = 36.66 , Double_t Yc = 0.0, Double_t Zc = 0.0); Double_t GetYmax() const { return fYmax; } Double_t GetZmax(Int_t c) const { return fZmax[c]; } Double_t GetZc(Int_t c) const { return fZc[c]; } UInt_t GetIndex(Int_t i) const { return fIndex[i]; } Double_t GetX() const { return fX; } Double_t GetdX() const { return fdX; } Int_t GetTimeBinIndex() const { return fTimeBinIndex; } Int_t GetPlane() const { return fPlane; } Bool_t IsHole(Int_t zone) const { return fIsHole[zone]; } Bool_t IsSensitive() const { return (fTimeBinIndex >= 0) ? kTRUE : kFALSE; } void Clear() { for (Int_t i = 0; i < fN; i++) fClusters[i] = NULL; fN = 0; } void InsertCluster(AliTRDcluster *c, UInt_t index); Int_t Find(Float_t y) const; Int_t FindNearestCluster(Float_t y, Float_t z, Float_t maxroad, Float_t maxroadz) const; private: Int_t fN; // This is fN Int_t fSec; // Sector mumber AliTRDcluster **fClusters; // Array of pointers to clusters UInt_t *fIndex; // Array of cluster indexes Double_t fX; // X coordinate of the middle plane Double_t fdX; // Radial thickness of the time bin Double_t fRho; // Default density of the material Double_t fX0; // Default radiation length Int_t fTimeBinIndex; // Plane * F(local_tb) Int_t fPlane; // Plane number Double_t fZc[kZones]; // Z position of the center for 5 active areas Double_t fZmax[kZones]; // Half of active area length in Z Double_t fZmaxSensitive[kZones]; // Sensitive area for detection Z Bool_t fIsHole[kZones]; // Is hole in given sector Double_t fYmax; // Half of active area length in Y Double_t fYmaxSensitive; // Half of active area length in Y Bool_t fHole; // kTRUE if there is a hole in the layer Double_t fHoleZc; // Z of the center of the hole Double_t fHoleZmax; // Half of the hole length in Z Double_t fHoleYc; // Y of the center of the hole Double_t fHoleYmax; // Half of the hole length in Y Double_t fHoleRho; // Density of the gas in the hole Double_t fHoleX0; // Radiation length of the gas in the hole }; class AliTRDtrackingSector { public: AliTRDtrackingSector(AliTRDgeometry* geo, Int_t gs); AliTRDtrackingSector(const AliTRDtrackingSector &/*t*/); ~AliTRDtrackingSector() { for (Int_t i = 0; i < fN; i++) delete fLayers[i]; } AliTRDtrackingSector &operator=(const AliTRDtrackingSector &/*t*/){ return *this; } Int_t GetNumberOfLayers() const { return fN; } Int_t GetNumberOfTimeBins() const; Int_t GetLayerNumber(Double_t x) const; Int_t GetInnerTimeBin() const; Int_t GetOuterTimeBin() const; Int_t GetLayerNumber(Int_t tb) const { return fTimeBinIndex[tb]; } Double_t GetX(Int_t pl) const { return fLayers[pl]->GetX(); } AliTRDpropagationLayer* GetLayer(Int_t i) { return fLayers[i]; } void MapTimeBinLayers(); Int_t Find(Double_t x) const; void InsertLayer(AliTRDpropagationLayer *pl); Int_t CookTimeBinIndex(Int_t plane, Int_t localTB) const; private: 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); private: AliTRDtrack *RegisterSeed(AliTRDseed *seeds, Double_t *params); void MakeSeedsMI(Int_t inner, Int_t outer, AliESD *esd = 0); 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