]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRDtracker.h
Classes moved to STEERBase.
[u/mrichter/AliRoot.git] / TRD / AliTRDtracker.h
index 63a65dcbf6ab50d575577fb02620180f14589735..89a0fba69c2d075264b4836b7bac15f35e8dfc5f 100644 (file)
@@ -4,19 +4,30 @@
 /* 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 TH1D;
+class TH2D;
+class TParticle;
+class TParticlePDG;
 
 class AliTRDgeometry;
 class AliTRDtrack;
 class AliTRDtracklet;
 class AliTRDcluster;
 class AliTRDseed;
-class AliESD;
+class AliESDEvent;
 class TTreeSRedirector;
 
 ///////////////////////////////////////////////////////////////////////////////
@@ -28,200 +39,239 @@ class TTreeSRedirector;
 
 class AliTRDtracker : public AliTracker { 
 
+  // Histograms
+  TH1D *fHBackfit;  // Histogram for back propagation
+  TH1D *fHClSearch; // Cluster search
+  TH1D *fHRefit;
+  
+  TH1D *fHX;
+  TH1D *fHNCl;
+  TH1D *fHNClTrack;
+  TH1D *fHFindCl[4];
+  TH1D *fHMinYPos;
+  TH1D *fHMinYNeg; 
+  TH1D *fHMinZ;
+
+  TH2D *fHMinD;
+  TH1D *fHDeltaX;
+
+  TH1D *fHXCl;
+
+  void InitLogHists();
+  void SaveLogHists();
+
  public:
 
   enum { kMaxLayersPerSector   = 1000
-       , kMaxTimeBinIndex      = 216
-       , kMaxClusterPerTimeBin = 7000
-       , kZones                = 5
-       , kTrackingSectors      = 18   };
-
+        , kMaxTimeBinIndex      = 216
+        , kMaxClusterPerTimeBin = 2300
+        , kZones                = 5
+        , kTrackingSectors      = 18   };
+  
   AliTRDtracker();
+  AliTRDtracker(const AliTRDtracker &t);
   AliTRDtracker(const TFile *in);
-  virtual ~AliTRDtracker(); 
-
-  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;  
-  
-  void          SetAddTRDseeds() { fAddTRDseeds = kTRUE; }
-  void          SetNoTilt() { fNoTilt = kTRUE; }
-
-  Double_t      GetTiltFactor(const AliTRDcluster* c);
-
-  Int_t         ReadClusters(TObjArray *array, TTree *in) const;
-  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);
-
-  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); }
+  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); }
-
+    { 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(AliESDEvent *event);
+  Int_t    PropagateBack(AliESDEvent *event);
+  Int_t    RefitInward(AliESDEvent *event);
+  
+  virtual void     CookLabel(AliKalmanTrack *t, Float_t wrong) const;
+  
+  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 {
-
-   // *****************  internal class *******************
-   public: 
-
-     AliTRDpropagationLayer(Double_t x, Double_t dx, Double_t rho, 
-                            Double_t x0, Int_t tbIndex, Int_t plane); 
-
-     ~AliTRDpropagationLayer() { if(fTimeBinIndex >= 0) { delete[] fClusters; delete[] fIndex; }}
-
-     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; }
-     Int_t          GetTimeBinIndex() const { return fTimeBinIndex; }     
-     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 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);
-     void           SetHoles(Bool_t* holes);
-     void           SetYmax(Double_t w, Double_t wsensitive) { fYmax = w; fYmaxSensitive = wsensitive; }
-     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]; }
-     
-     void           SetHole(Double_t Zmax, Double_t Ymax,
-                            Double_t rho = 1.29e-3, Double_t x0 = 36.66,
-                            Double_t Yc = 0, Double_t Zc = 0);
-                            
-     Bool_t         IsSensitive() const {return (fTimeBinIndex>=0)? kTRUE: kFALSE;}
-                       
-     void           Clear() {for(Int_t i=0; i<fN; i++) fClusters[i] = NULL; fN = 0;}
-     Bool_t         IsHole(Int_t zone) const  { return fIsHole[zone];}              
-
-   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 
-
+    
+  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 SetYmax(Double_t w, Double_t wsensitive) {
+      fYmax          = w;
+      fYmaxSensitive = wsensitive;
+    }
+
+    void SetZ(Double_t* center, Double_t *w, Double_t *wsensitive);
+    void SetHoles(Bool_t* holes);
+    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;
+    void SetX(Double_t x) {fX = x;}
+
+    
+  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() { for(Int_t i=0; i<fN; i++) delete fLayers[i]; }
-
-     Int_t    GetNumberOfLayers() const { return fN; }
-     Int_t    GetNumberOfTimeBins() const;
-     Double_t GetX(Int_t pl) const { return fLayers[pl]->GetX(); }
-     void     MapTimeBinLayers();
-     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];}
-     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
-     AliTRDpropagationLayer   *fLayers[kMaxLayersPerSector];    // layers   
-     Int_t                     fTimeBinIndex[kMaxTimeBinIndex]; // time bin index
-     Int_t                     fGeomSector;                     // sector # in AliTRDgeometry
-
+    
+    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); 
+  
+  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
-
-};
+  
+  AliTRDtrack *RegisterSeed(AliTRDseed *seeds, Double_t *params);
+  void     MakeSeedsMI(Int_t inner, Int_t outer, AliESDEvent *esd = 0);
+  
+  Int_t    FollowBackProlongation(AliTRDtrack &t);
+  Int_t    FollowProlongation(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