I improved (by a factor 2.5) the speed of the MatchToMC method

[u/mrichter/AliRoot.git] / STEER / AliTracker.h

diff --git a/STEER/AliTracker.h b/STEER/AliTracker.h
index a8562acab14fa90b4f59256a352d7505d34dfabf..ca5c8d5ac8bb56f096ad9eaf2f5847e90166affd 100644 (file)
--- a/STEER/AliTracker.h
+++ b/STEER/AliTracker.h
@@ -10,31 +10,30 @@
 //   that is the base for AliTPCtracker, AliITStrackerV2 and AliTRDtracker
 //       Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
 //-------------------------------------------------------------------------
+
+#include <TGeoGlobalMagField.h>
 #include <TObject.h>
-#include <TClass.h>
-#include "AliKalmanTrack.h"
+
 #include "AliMagF.h"
-#include "AliLog.h"
+#include "AliRecoParam.h"
+#include "AliPlaneEff.h"
 
-class AliCluster;
 class TTree;
-class AliESD;
+class AliCluster;
+class AliESDEvent;
+class AliESDtrack;
+class AliExternalTrackParam;
 class AliTrackPoint;
-
+class AliKalmanTrack;
+class AliEventInfo;
 class AliTracker : public TObject {
 public:
-
-  enum {kTrackInward, kTrackBack, kTrackRefit} Propagation_t; //type of propagation
-  
   AliTracker();
-  AliTracker(const AliTracker &atr): TObject(atr)
-    {Fatal("Copy ctor","Not Implemented!\n");}
-  // AliTracker & operator=(const AliTracker &)
-  //  {Fatal("= op","Not Implemented\n");return *this;}
   virtual ~AliTracker(){}
-  virtual Int_t Clusters2Tracks(AliESD *event)=0;
-  virtual Int_t PropagateBack(AliESD *event)=0;
-  virtual Int_t RefitInward(AliESD *event)=0;
+  virtual Int_t Clusters2Tracks(AliESDEvent *event)=0;
+  virtual Int_t PropagateBack(AliESDEvent *event)=0;
+  virtual Int_t RefitInward(AliESDEvent *event)=0;
+  virtual Int_t PostProcess(AliESDEvent */*event*/) {return 0;}
   void SetVertex(const Double_t *xyz, const Double_t *ers=0) { 
      fX=xyz[0]; fY=xyz[1]; fZ=xyz[2];
      if (ers) { fSigmaX=ers[0]; fSigmaY=ers[1]; fSigmaZ=ers[2]; } 
@@ -43,9 +42,12 @@ public:
 //protected:
   virtual Int_t LoadClusters(TTree *)=0;
   virtual void UnloadClusters()=0;
+  virtual void FillClusterArray(TObjArray* array) const;
   virtual AliCluster *GetCluster(Int_t index) const=0;
-  //  virtual UShort_t GetVolumeID(Int_t index) {return 0;}
+  virtual AliPlaneEff *GetPlaneEff() {return NULL;}
   virtual Bool_t GetTrackPoint(Int_t /* index */ , AliTrackPoint& /* p */) const { return kFALSE;}
+  virtual Bool_t GetTrackPointTrackingError(Int_t /* index */, 
+          AliTrackPoint& /* p */, const AliESDtrack* /* t */) { return kFALSE;}
   virtual void  UseClusters(const AliKalmanTrack *t, Int_t from=0) const;
   virtual void  CookLabel(AliKalmanTrack *t,Float_t wrong) const; 
   Double_t GetX() const {return fX;}
@@ -55,38 +57,58 @@ public:
   Double_t GetSigmaY() const {return fSigmaY;}
   Double_t GetSigmaZ() const {return fSigmaZ;}
 
-  static void SetFieldMap(const AliMagF* map, Bool_t uni);
-  static const AliMagF *GetFieldMap() {return fgkFieldMap;}
-
+  static 
+  Double_t MeanMaterialBudget(const Double_t *start, const Double_t *end, Double_t *mparam);
+  static
+  Bool_t PropagateTrackTo(AliExternalTrackParam *track, Double_t x, Double_t m,
+        Double_t maxStep, Bool_t rotateTo=kTRUE, Double_t maxSnp=0.8);  
+  Bool_t PropagateTrackToBxByBz(AliExternalTrackParam *track, Double_t x, 
+         Double_t m,
+        Double_t maxStep, Bool_t rotateTo=kTRUE, Double_t maxSnp=0.8);  
+  //
+  static Double_t GetBz(const Double_t *r);
+  static void GetBxByBz(const Double_t r[3], Double_t b[3]);
+  static Double_t GetBz();
+  static Bool_t   UniformField();
+  //
+  static void FillResiduals(const AliExternalTrackParam *t,
+                          Double_t *p, Double_t *cov, 
+                           UShort_t id, Bool_t updated=kTRUE);
+  static void FillResiduals(const AliExternalTrackParam *t,
+                            const AliCluster *c, Bool_t updated=kTRUE);
+  static void SetFillResiduals(AliRecoParam::EventSpecie_t es, Bool_t flag=kTRUE) { fFillResiduals=flag; fEventSpecie = es ;}
+  static void SetResidualsArray(TObjArray **arr) { fResiduals=arr; }
+  static TObjArray ** GetResidualsArray() { return fResiduals; }
+
+  void                SetEventInfo(AliEventInfo *evInfo) {fEventInfo = evInfo;}
+  const AliEventInfo* GetEventInfo() const {return fEventInfo;}
+
+protected:
+  AliTracker(const AliTracker &atr);
 private:
-
   AliTracker & operator=(const AliTracker & atr);
-
-  static const AliMagF *fgkFieldMap; //field map
+  static Bool_t fFillResiduals;       // Fill residuals flag
+  static TObjArray **fResiduals;    //! Array of histograms with residuals
 
   Double_t fX;  //X-coordinate of the primary vertex
   Double_t fY;  //Y-coordinate of the primary vertex
   Double_t fZ;  //Z-coordinate of the primary vertex
-
+ 
   Double_t fSigmaX; // error of the primary vertex position in X
   Double_t fSigmaY; // error of the primary vertex position in Y
   Double_t fSigmaZ; // error of the primary vertex position in Z
-
-  ClassDef(AliTracker,2) //abstract tracker
+  
+  static AliRecoParam::EventSpecie_t fEventSpecie ; //! event specie, see AliRecoParam
+  AliEventInfo*                      fEventInfo;    //! pointer to the event info object
+  
+  ClassDef(AliTracker,5) //abstract tracker
 };
 
-inline void AliTracker::SetFieldMap(const AliMagF* map, Bool_t uni) {
-  //--------------------------------------------------------------------
-  //This passes the field map to the reconstruction.
-  //--------------------------------------------------------------------
-  if (map==0) AliFatalClass("Can't access the field map !");
-  AliKalmanTrack::SetFieldMap(map);
-  if (uni) AliKalmanTrack::SetUniformFieldTracking();
-  fgkFieldMap=map;
+//__________________________________________________________________________
+inline Bool_t AliTracker::UniformField()
+{
+  AliMagF* fld = (AliMagF*)TGeoGlobalMagField::Instance()->GetField();
+  return fld ? fld->IsUniform():kTRUE;
 }
 
-
-
 #endif
-
-