]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TPC/AliTPCtrackerMI.h
Fix for negative sigmas. Memory leaks (Marco)
[u/mrichter/AliRoot.git] / TPC / AliTPCtrackerMI.h
index 17cc3c59c82dc75369cd447ad3061971a331cacd..9925a00ca72b5489deab9aeca773f5b6e7f30a38 100644 (file)
 #include "AliTracker.h"
 #include "AliTPCreco.h"
 #include "AliPID.h"
+#include "AliTPCclusterMI.h"
 
 
 class TFile;
 class AliTPCParam;
 class AliTPCseed;
-class AliTPCclusterMI;
 class AliTPCTrackerPoint;
-class AliESD;   
+class AliESDEvent;   
 class TTree;
 class AliESDkink;
 class TTreeSRedirector;
-
+class AliTrackPoint;
 
 class AliTPCtrackerMI : public AliTracker {
 public:
-  AliTPCtrackerMI():AliTracker(),fkNIS(0),fkNOS(0) {
-    fInnerSec=fOuterSec=0; fSeeds=0; 
-  }
+  AliTPCtrackerMI();
   AliTPCtrackerMI(const AliTPCParam *par); 
-  AliTPCtrackerMI(const AliTPCtrackerMI& r);           //dummy copy constructor
-  AliTPCtrackerMI &operator=(const AliTPCtrackerMI& r);//dummy assignment operator
   virtual ~AliTPCtrackerMI();
   //
   void SetIteration(Int_t iteration){fIteration = iteration;}
-  virtual Int_t Clusters2Tracks (AliESD *esd);
-  virtual Int_t RefitInward (AliESD *esd);
+  virtual Int_t Clusters2Tracks (AliESDEvent *esd);
+  virtual Int_t RefitInward (AliESDEvent *esd);
   virtual Int_t LoadClusters (TTree * tree);
   Int_t  LoadClusters();
   void   UnloadClusters();
-
+  void   Transform(AliTPCclusterMI * cluster);
   //
   void SetIO();  //set default IO from folders
-  void SetIO(TTree * input, TTree * output, AliESD * event);
+  void SetIO(TTree * input, TTree * output, AliESDEvent * event);
   void FillESD(TObjArray* arr);
   void WriteTracks();
   void WriteTracks(TTree * tree);  
   void DeleteSeeds();
   void SetDebug(Int_t debug){ fDebug = debug;}
-  void FindKinks(TObjArray * array, AliESD * esd);
-  void FindV0s(TObjArray * array, AliESD * esd);
+  void FindKinks(TObjArray * array, AliESDEvent * esd);
+  void FindV0s(TObjArray * array, AliESDEvent * esd);
   void UpdateKinkQualityM(AliTPCseed * seed);
   void UpdateKinkQualityD(AliTPCseed * seed);
   Int_t CheckKinkPoint(AliTPCseed*seed, AliTPCseed &mother, AliTPCseed &daughter, AliESDkink &kink);
@@ -64,7 +60,7 @@ public:
    Int_t ReadSeeds(const TFile *in);
    TObjArray * GetSeeds(){return fSeeds;}
    //   
-   AliCluster * GetCluster (int) const {return 0;}
+   AliCluster * GetCluster(Int_t index) const {return (AliCluster*)GetClusterMI(index);}
    AliTPCclusterMI *GetClusterMI(Int_t index) const;
    Int_t Clusters2Tracks();
    virtual void  CookLabel(AliKalmanTrack *tk,Float_t wrong) const; 
@@ -75,7 +71,8 @@ public:
    
    Int_t FollowProlongation(AliTPCseed& t, Int_t rf=0, Int_t step=1);
    Int_t FollowProlongationFast(AliTPCseed& t, Int_t rf=0, Int_t step=1);
-   
+   Bool_t GetTrackPoint(Int_t index, AliTrackPoint &p ) const; 
+
    Int_t FollowBackProlongation(AliTPCseed& t, Int_t rf);
    Int_t FollowToNext(AliTPCseed& t, Int_t nr);
    Int_t FollowToNextFast(AliTPCseed& t, Int_t nr);
@@ -83,7 +80,7 @@ public:
    Int_t FollowToNextCluster( AliTPCseed& t, Int_t nr);
 
    Int_t PropagateBack(TObjArray * arr);
-   Int_t PropagateBack(AliESD * event);
+   Int_t PropagateBack(AliESDEvent * event);
    Int_t PropagateBack(AliTPCseed *pt, Int_t row0, Int_t row1);   
    Int_t PropagateForward();
    Int_t PropagateForward2(TObjArray * arr);
@@ -108,8 +105,6 @@ public:
 //**************** Internal tracker class ********************** 
    class AliTPCSector;
    class AliTPCRow {
-     friend class AliTPCtrackerMI;
-     friend class AliTPCtrackerMI::AliTPCSector;
    public:
      AliTPCRow();
      ~AliTPCRow();
@@ -126,6 +121,23 @@ public:
      
      void SetX(Double_t x) {fX=x;}
      Double_t GetX() const {return fX;}
+     Float_t GetDeadZone() const {return fDeadZone;}
+     void SetDeadZone(Float_t d) {fDeadZone=d;}
+     Int_t GetN1() const {return fN1;}
+     void SetN1(Int_t n) {fN1=n;}
+     Int_t GetN2() const {return fN2;}
+     void SetN2(Int_t n) {fN2=n;}
+     AliTPCclusterMI* GetClusters1() const {return fClusters1;}
+     AliTPCclusterMI* GetClusters2() const {return fClusters2;}
+     void SetClusters1(AliTPCclusterMI* cl) {fClusters1=cl;}
+     void SetClusters2(AliTPCclusterMI* cl) {fClusters2=cl;}
+     void SetCluster1(Int_t i, const AliTPCclusterMI &cl) {fClusters1[i]=cl;}
+     void SetCluster2(Int_t i, const AliTPCclusterMI &cl) {fClusters2[i]=cl;}
+     AliTPCclusterMI* GetCluster1(Int_t i) const {return &fClusters1[i];}
+     AliTPCclusterMI* GetCluster2(Int_t i) const {return &fClusters2[i];}
+     Short_t GetFastCluster(Int_t i) const {return fFastCluster[i];}
+     void SetFastCluster(Int_t i, Short_t cl) {fFastCluster[i]=cl;}
+
 private:  
      AliTPCRow & operator=(const AliTPCRow & );
      AliTPCRow(const AliTPCRow& /*r*/);           //dummy copy constructor
@@ -146,9 +158,16 @@ private:
 
 //**************** Internal tracker class ********************** 
    class AliTPCSector {
-     friend class AliTPCtrackerMI;
    public:
-     AliTPCSector() { fN=0; fRow = 0; }
+     AliTPCSector():
+       fN(0),
+       fRow(0),
+       fAlpha(0.),
+       fAlphaShift(0.),
+       fPadPitchWidth(0.),
+       fPadPitchLength(0.),
+       f1PadPitchLength(0.),
+       f2PadPitchLength(0.){}
     ~AliTPCSector() { delete[] fRow; }
     AliTPCRow& operator[](Int_t i) const { return *(fRow+i); }
     Int_t GetNRows() const { return fN; }
@@ -193,7 +212,10 @@ private:
    Int_t AcceptCluster(AliTPCseed * seed, AliTPCclusterMI * cluster, Float_t factor, Float_t cory=1., Float_t corz=1.);
 
 private:
+  AliTPCtrackerMI(const AliTPCtrackerMI& r);           //dummy copy constructor
+  AliTPCtrackerMI &operator=(const AliTPCtrackerMI& r);//dummy assignment operator
    inline AliTPCRow &GetRow(Int_t sec, Int_t row);
+   inline Bool_t     IsActive(Int_t sec, Int_t row);
    inline Double_t  GetXrow(Int_t row) const;
    inline Double_t  GetMaxY(Int_t row) const;
    inline Int_t GetRowNumber(Double_t x) const;
@@ -205,7 +227,7 @@ private:
    Float_t  GetSigmaZ(AliTPCseed * seed);
    void GetShape(AliTPCseed * seed, Int_t row);
  
-   void ReadSeeds(AliESD *event, Int_t direction);  //read seeds from the event
+   void ReadSeeds(AliESDEvent *event, Int_t direction);  //read seeds from the event
 
    void MakeSeeds3(TObjArray * arr, Int_t sec, Int_t i1, Int_t i2, Float_t cuts[4], Float_t deltay = -1, Int_t ddsec=0); 
    void MakeSeeds5(TObjArray * arr, Int_t sec, Int_t i1, Int_t i2, Float_t cuts[4], Float_t deltay = -1);
@@ -229,13 +251,15 @@ private:
    void Tracking(TObjArray * arr);
    TObjArray * Tracking(Int_t seedtype, Int_t i1, Int_t i2, Float_t cuts[4], Float_t dy=-1, Int_t dsec=0);
    TObjArray * Tracking();
-   void SumTracks(TObjArray *arr1,TObjArray *arr2) const;
+   TObjArray * TrackingSpecial();
+   void SumTracks(TObjArray *arr1,TObjArray *&arr2) const;
    void PrepareForBackProlongation(TObjArray * arr, Float_t fac) const;
    void PrepareForProlongation(TObjArray * arr, Float_t fac) const;
 
    void SetSampledEdx(AliTPCseed */*t*/, Float_t /*q*/, Int_t /*i*/) {;}
    Int_t UpdateTrack(AliTPCseed *t, Int_t accept); //update trackinfo
 
+   void MakeBitmaps(AliTPCseed *t);
 
    const Int_t fkNIS;        //number of inner sectors
    AliTPCSector *fInnerSec;  //array of inner sectors;
@@ -249,7 +273,7 @@ private:
    TTree * fOutput;      // output tree with tracks
    TTree * fSeedTree;    // output tree with seeds - filled in debug mode 1
    TTree * fTreeDebug;   // output with a debug information about track
-   AliESD * fEvent;      // output with esd tracks
+   AliESDEvent * fEvent;      // output with esd tracks
    Int_t    fDebug;      // debug option        
    Bool_t   fNewIO;      // indicated if we have data using New IO 
    Int_t fNtracks;                     //current number of tracks
@@ -262,7 +286,7 @@ private:
    Double_t fPadLength[200];                // max y for given pad row
    const AliTPCParam *fParam;          //pointer to the parameters
    TTreeSRedirector *fDebugStreamer;     //!debug streamer
-   ClassDef(AliTPCtrackerMI,1
+   ClassDef(AliTPCtrackerMI,2
 };
 
 
@@ -272,6 +296,15 @@ AliTPCtrackerMI::AliTPCRow & AliTPCtrackerMI::GetRow(Int_t sec, Int_t row)
   return (row>=fInnerSec->GetNRows()) ? fOuterSec[sec][row-fInnerSec->GetNRows()]:fInnerSec[sec][row];
 }
 
+Bool_t   AliTPCtrackerMI::IsActive(Int_t sec, Int_t row)
+{
+  //
+  // check if the given sector row is active 
+  //
+  return (row>=fInnerSec->GetNRows()) ? fOuterSec[sec][row-fInnerSec->GetNRows()].GetN()>0:fInnerSec[sec][row].GetN()>0;
+}
+
+
 Double_t  AliTPCtrackerMI::GetXrow(Int_t row) const {
   //  return (row>=fInnerSec->GetNRows()) ? fOuterSec->GetX(row-fInnerSec->GetNRows()):fInnerSec->GetX(row);
   return fXRow[row];