]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRDtracker.h
Coding rule violations
[u/mrichter/AliRoot.git] / TRD / AliTRDtracker.h
index 339e832e535f04cd227494cd9fbbb32b52668856..0067e56bf4421e6e023983c6f6bde3f8fcdc2993 100644 (file)
 //                                                                        //
 ////////////////////////////////////////////////////////////////////////////
 
-#include "AliTracker.h" 
 #include "TObjArray.h" 
 
+#include "AliTracker.h" 
+
 class TFile;
 class TTree;
+class TH1D;
+class TH2D;
 class TParticle;
 class TParticlePDG;
+class TTreeSRedirector;
 
 class AliTRDgeometry;
 class AliTRDtrack;
 class AliTRDtracklet;
 class AliTRDcluster;
 class AliTRDseed;
-class AliESD;
-class TTreeSRedirector;
+class AliESDEvent;
 
 ///////////////////////////////////////////////////////////////////////////////
 //                                                                           //
@@ -39,207 +42,227 @@ class AliTRDtracker : public AliTracker {
 
  public:
 
+  void InitLogHists();
+  void SaveLogHists();
+
   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);
-
+  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(AliESDEvent *event);
+  Int_t            PropagateBack(AliESDEvent *event);
+  Int_t            RefitInward(AliESDEvent *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();
+  Int_t            LocalToGlobalID(Int_t lid);
+  Int_t            GlobalToLocalID(Int_t gid);
+  
+  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            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); 
+                        , Double_t x0, Int_t tbIndex, Int_t plane); 
     AliTRDpropagationLayer(const AliTRDpropagationLayer &/*p*/);
-    ~AliTRDpropagationLayer()                                          { if (fTimeBinIndex >= 0) { 
-                                                                           delete[] fClusters; 
-                                                                           delete[] fIndex; 
-                                                                         }                              }
+    ~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;
-
+                                                              { 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 
-
+    
+    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;     
-
+    
+    AliTRDtrackingSector(AliTRDgeometry* geo, Int_t gs);
+    AliTRDtrackingSector(const AliTRDtrackingSector &/*t*/);
+    ~AliTRDtrackingSector();
+    
+    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
-
+    
+    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?
+
+  // Histograms
+  TH1D                    *fHBackfit;                      // QA histogram
+  TH1D                    *fHClSearch;                     // QA histogram
+  TH1D                    *fHRefit;                        // QA histogram
+  TH1D                    *fHX;                            // QA histogram
+  TH1D                    *fHNCl;                          // QA histogram
+  TH1D                    *fHNClTrack;                     // QA histogram
+  TH1D                    *fHFindCl[4];                    // QA histogram
+  TH1D                    *fHMinYPos;                      // QA histogram
+  TH1D                    *fHMinYNeg;                      // QA histogram
+  TH1D                    *fHMinZ;                         // QA histogram
+  TH2D                    *fHMinD;                         // QA histogram
+  TH1D                    *fHDeltaX;                       // QA histogram
+  TH1D                    *fHXCl;                          // QA histogram
+  
+  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,3)                                // TRD tracker
+    
 };
 #endif