Changing once more (hopefully we get it correct this time...) the logic to trig the...
[u/mrichter/AliRoot.git] / ITS / AliITSMultReconstructor.h
index ea2b1f3..da7d201 100644 (file)
 
 /* $Id$ */
 
-/////////////////////////////////////////////////////////////////////////
+//_________________________________________________________________________
 // 
-// AliITSMultReconstructor - find clusters in the pixels (theta and
-// phi) and tracklets.
-// 
-// These can be used to extract charged particles multiplcicity from the ITS.
+//        Implementation of the ITS-SPD trackleter class
 //
-// A tracklet consist of two ITS clusters, one in the first pixel
-// layer and one in the second. The clusters are associates if the 
-// differencies in Phi (azimuth) and Zeta (longitudinal) are inside 
-// a fiducial volume. In case of multiple candidates it is selected the
-// candidate with minimum distance in Phi. 
-// The boolean fOnlyOneTrackletPerC2 allows to control if two clusters 
-// in layer 2 can be associated to the same cluster in layer 1 or not.
+// It retrieves clusters in the pixels (theta and phi) and finds tracklets.
+// These can be used to extract charged particle multiplicity from the ITS.
 //
-/////////////////////////////////////////////////////////////////////////
+// A tracklet consists of two ITS clusters, one in the first pixel layer and 
+// one in the second. The clusters are associated if the differences in 
+// Phi (azimuth) and Theta (polar angle) are within fiducial windows.
+// In case of multiple candidates the candidate with minimum
+// distance is selected. 
+//_________________________________________________________________________
 
 #include "TObject.h"
 
 class TTree;
 class TH1F;
-class TH2F;
+class TH2F; 
+
 class AliITSgeom;
 
 class AliITSMultReconstructor : public TObject 
 {
 public:
   AliITSMultReconstructor();
-
-  void SetGeometry(AliITSgeom* geo) {fGeometry = geo;}
+  virtual ~AliITSMultReconstructor();
 
   void Reconstruct(TTree* tree, Float_t* vtx, Float_t* vtxRes);
+  void LoadClusterFiredChips(TTree* tree);
+  void FlagClustersInOverlapRegions(Int_t ic1,Int_t ic2);
 
+  // Following members are set via AliITSRecoParam
   void SetPhiWindow(Float_t w=0.08) {fPhiWindow=w;}
-  void SetZetaWindow(Float_t w=0.1) {fZetaWindow=w;}
-  void SetOnlyOneTrackletPerC2(Float_t b = kFALSE) {fOnlyOneTrackletPerC2 = b;}
-  
+  void SetThetaWindow(Float_t w=0.025) {fThetaWindow=w;}
+  void SetPhiShift(Float_t w=0.0045) {fPhiShift=w;}
+  void SetRemoveClustersFromOverlaps(Bool_t b = kFALSE) {fRemoveClustersFromOverlaps = b;}
+  void SetPhiOverlapCut(Float_t w=0.005) {fPhiOverlapCut=w;}
+  void SetZetaOverlapCut(Float_t w=0.05) {fZetaOverlapCut=w;}
+
   Int_t GetNClustersLayer1() const {return fNClustersLay1;}
   Int_t GetNClustersLayer2() const {return fNClustersLay2;}
   Int_t GetNTracklets() const {return fNTracklets;}
+  Int_t GetNSingleClusters() const {return fNSingleCluster;}
+  Short_t GetNFiredChips(Int_t layer) const {return fNFiredChips[layer];}
 
   Float_t* GetClusterLayer1(Int_t n) {return fClustersLay1[n];}
   Float_t* GetClusterLayer2(Int_t n) {return fClustersLay2[n];}
   Float_t* GetTracklet(Int_t n) {return fTracklets[n];}
+  Float_t* GetCluster(Int_t n) {return fSClusters[n];}
 
   void SetHistOn(Bool_t b=kFALSE) {fHistOn=b;}
   void SaveHists();
+
 protected:
+  AliITSMultReconstructor(const AliITSMultReconstructor& mr);
+  AliITSMultReconstructor& operator=(const AliITSMultReconstructor& mr);
 
-  AliITSgeom*   fGeometry;            // ITS geometry
   
-  Float_t**     fClustersLay1;        // clusters in the 1st layer of ITS 
-  Float_t**     fClustersLay2;        // clusters in the 2nd layer of ITS 
-  Float_t**     fTracklets;           // tracklets 
-  Bool_t*       fAssociationFlag;     // flag for the associations 
-  
-  Int_t         fNClustersLay1; // Number of clusters (Layer1)
-  Int_t         fNClustersLay2; // Number of clusters (Layer2)
-  Int_t         fNTracklets;    // Number of tracklets
+  Float_t**     fClustersLay1;               // clusters in the 1st layer of ITS 
+  Float_t**     fClustersLay2;               // clusters in the 2nd layer of ITS 
+  Int_t*        fDetectorIndexClustersLay1;  // module index for clusters 1st ITS layer
+  Int_t*        fDetectorIndexClustersLay2;  // module index for clusters 2nd ITS layer
+  Bool_t*       fOverlapFlagClustersLay1;    // flag for clusters in the overlap regions 1st ITS layer
+  Bool_t*       fOverlapFlagClustersLay2;    // flag for clusters in the overlap regions 2nd ITS layer 
 
-  Float_t       fPhiWindow;     // Search window in phi
-  Float_t       fZetaWindow;    // SEarch window in eta
 
-  Bool_t        fOnlyOneTrackletPerC2; // only one tracklet per cluster in layer 2?
+  Float_t**     fTracklets;            // tracklets 
+  Float_t**     fSClusters;            // single clusters (unassociated)
   
-  Bool_t        fHistOn;
-
-  TH1F*         fhClustersDPhi;
-  TH1F*         fhClustersDTheta;
-  TH1F*         fhClustersDZeta;
+  Int_t         fNClustersLay1;        // Number of clusters (Layer1)
+  Int_t         fNClustersLay2;        // Number of clusters (Layer2)
+  Int_t         fNTracklets;           // Number of tracklets
+  Int_t         fNSingleCluster;       // Number of unassociated clusters
+  Short_t       fNFiredChips[2];       // Number of fired chips in the two SPD layers
+  // Following members are set via AliITSRecoParam
+  Float_t       fPhiWindow;                    // Search window in phi
+  Float_t       fThetaWindow;                  // Search window in theta
+  Float_t       fPhiShift;                     // Phi shift reference value (at 0.5 T) 
+  Bool_t        fRemoveClustersFromOverlaps;   // Option to skip clusters in the overlaps
+  Float_t       fPhiOverlapCut;                // Fiducial window in phi for overlap cut
+  Float_t       fZetaOverlapCut;               // Fiducial window in eta for overlap cut
+
+  Bool_t        fHistOn;               // Option to define and fill the histograms 
+
+  TH1F*         fhClustersDPhiAcc;     // Phi2 - Phi1 for tracklets 
+  TH1F*         fhClustersDThetaAcc;   // Theta2 - Theta1 for tracklets 
+  TH1F*         fhClustersDPhiAll;     // Phi2 - Phi1 all the combinations 
+  TH1F*         fhClustersDThetaAll;   // Theta2 - Theta1 all the combinations
  
-  TH2F*         fhDPhiVsDThetaAll;
-  TH2F*         fhDPhiVsDThetaAcc;
+  TH2F*         fhDPhiVsDThetaAll;     // 2D plot for all the combinations  
+  TH2F*         fhDPhiVsDThetaAcc;     // same plot for tracklets 
+
+  TH1F*         fhetaTracklets;        // Pseudorapidity distr. for tracklets 
+  TH1F*         fhphiTracklets;        // Azimuthal (Phi) distr. for tracklets  
+  TH1F*         fhetaClustersLay1;     // Pseudorapidity distr. for Clusters L. 1
+  TH1F*         fhphiClustersLay1;     // Azimuthal (Phi) distr. for Clusters L. 1 
+
 
   void LoadClusterArrays(TTree* tree);
 
-  ClassDef(AliITSMultReconstructor,0)
+  ClassDef(AliITSMultReconstructor,6)
 };
 
 #endif