Transition to NewIO
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALClusterizerv1.h
1 #ifndef ALIEMCALCLUSTERIZERV1_H
2 #define ALIEMCALCLUSTERIZERV1_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 //_________________________________________________________________________
9 //  Implementation version 1 of the clusterization algorithm                     
10 //  Performs clusterization (collects neighbouring active cells) and 
11 //  unfolding of the clusters with several local maxima.  
12 //  results are stored in TreeR#, branches PHOSEmcRP (EMC recPoints),
13 //  PHOSCpvRP (CPV RecPoints) and AliPHOSClusterizer
14 //
15 //*-- Author: Yves Schutz (SUBATECH)
16 // Modif: 
17 //  August 2002 Yves Schutz: clone PHOS as closely as possible and intoduction
18 //                           of new  IO (à la PHOS)
19
20 // --- ROOT system ---
21
22 // --- Standard library ---
23
24 // --- AliRoot header files ---
25
26 #include "AliEMCALClusterizer.h"
27 class AliEMCALTowerRecPoint ; 
28 class AliEMCALDigit ;
29 class AliEMCALDigitizer ;
30 class AliEMCALGeometry ;
31
32
33 class AliEMCALClusterizerv1 : public AliEMCALClusterizer {
34   
35 public:
36   
37   AliEMCALClusterizerv1() ;         
38   AliEMCALClusterizerv1(const TString alirunFileNameFile, const TString eventFolderName = AliConfig::fgkDefaultEventFolderName);
39   virtual ~AliEMCALClusterizerv1()  ;
40   
41   virtual Int_t   AreNeighbours(AliEMCALDigit * d1, AliEMCALDigit * d2)const ; 
42                                // Checks if digits are in neighbour cells 
43
44   virtual Float_t Calibrate(Int_t amp, Int_t where)const ;  // Tranforms Amp to energy 
45
46   virtual void    GetNumberOfClustersFound(int * numb )const{ numb[0] = fNumberOfPREClusters ; 
47                                                               numb[1] = fNumberOfECAClusters ; 
48                                                               numb[2] = fNumberOfHCAClusters ; }
49
50   virtual Float_t GetPREClusteringThreshold()const{ return fPREClusteringThreshold;  } 
51   virtual Float_t GetECAClusteringThreshold()const{ return fECAClusteringThreshold;}
52   virtual Float_t GetHCAClusteringThreshold()const{ return fHCAClusteringThreshold;}
53
54   virtual Float_t GetPRELocalMaxCut()const       { return fPRELocMaxCut;} 
55   virtual Float_t GetPREShoLogWeight()const      { return fPREW0;}  
56   virtual Float_t GetECALocalMaxCut()const       { return fECALocMaxCut;} 
57   virtual Float_t GetECALogWeight()const         { return fECAW0;}  
58   virtual Float_t GetHCALocalMaxCut()const       { return fHCALocMaxCut;} 
59   virtual Float_t GetHCALogWeight()const         { return fHCAW0;}  
60
61   virtual Float_t GetTimeGate() const            { return fTimeGate ; }
62   virtual const char *  GetRecPointsBranch() const{ return GetName() ;}
63   virtual const Int_t GetRecPointsInRun() const   {return fRecPointsInRun ;} 
64
65   void    Exec(Option_t *option);                // Does the job
66
67   virtual void Print(Option_t * option)const ;
68
69   virtual void SetECAClusteringThreshold(Float_t cluth)  { fECAClusteringThreshold = cluth ; }
70   virtual void SetECALocalMaxCut(Float_t cut)            { fECALocMaxCut = cut ; }
71   virtual void SetECALogWeight(Float_t w)                { fECAW0 = w ; }
72   virtual void SetHCAClusteringThreshold(Float_t cluth)  { fHCAClusteringThreshold = cluth ; }
73   virtual void SetHCALocalMaxCut(Float_t cut)            { fHCALocMaxCut = cut ; }
74   virtual void SetHCALogWeight(Float_t w)                { fHCAW0 = w ; }
75   virtual void SetTimeGate(Float_t gate)                 { fTimeGate = gate ;}
76   virtual void SetPREClusteringThreshold(Float_t cluth)  { fPREClusteringThreshold = cluth ; }
77   virtual void SetPRELocalMaxCut(Float_t cut)            { fPRELocMaxCut = cut ; }
78   virtual void SetPRELogWeight(Float_t w)                { fPREW0 = w ; }
79   virtual void SetUnfolding(Bool_t toUnfold = kTRUE )    {fToUnfold = toUnfold ;}  
80   static Double_t ShowerShape(Double_t r) ; // Shape of EM shower used in unfolding; 
81                                             //class member function (not object member function)
82   static void UnfoldingChiSquare(Int_t & nPar, Double_t * Grad, Double_t & fret, Double_t * x, Int_t iflag)  ;
83                                             // Chi^2 of the fit. Should be static to be passes to MINUIT
84   void Unload() ; 
85   virtual const char * Version() const { return "clu-v1" ; }  
86
87 protected:
88
89   void           WriteRecPoints(Int_t event) ;
90   virtual void   MakeClusters( ) ;            
91   
92 private:
93
94   const TString BranchName() const ; 
95   void    GetCalibrationParameters(void) ;
96   
97   Bool_t  FindFit(AliEMCALTowerRecPoint * emcRP, AliEMCALDigit ** MaxAt, Float_t * maxAtEnergy, 
98                   Int_t NPar, Float_t * FitParametres) const; //Used in UnfoldClusters, calls TMinuit
99   void Init() ;
100   void InitParameters() ;
101
102   virtual void   MakeUnfolding() ;
103   void           UnfoldCluster(AliEMCALTowerRecPoint * iniEmc,Int_t Nmax, 
104                        AliEMCALDigit ** maxAt,Float_t * maxAtEnergy ) ; //Unfolds cluster using TMinuit package
105   void           PrintRecPoints(Option_t * option) ;
106
107 private:
108
109   Bool_t  fDefaultInit;              //! Says if the task was created by defaut ctor (only parameters are initialized)
110
111   Int_t   fNTowers ;                 // number of Towers in EMCAL
112
113   Bool_t  fToUnfold ;                // To perform unfolding 
114
115   Int_t   fNumberOfPREClusters ;     // number of clusters found in PRE section 
116   Int_t   fNumberOfECAClusters ;     // number of clusters found in EC section
117   Int_t   fNumberOfHCAClusters ;     // number of clusters found in HC section
118   
119   //Calibration parameters... to be replaced by database 
120   Float_t fADCchannelPRE ;          // width of one ADC channel for PRE section (GeV)
121   Float_t fADCpedestalPRE ;         // pedestal of ADC for PRE section (GeV)
122   Float_t fADCchannelECA ;          // width of one ADC channel for EC section (GeV)
123   Float_t fADCpedestalECA ;         // pedestal of ADC for EC section (GeV) 
124   Float_t fADCchannelHCA ;          // width of one ADC channel for HC section (GeV)
125   Float_t fADCpedestalHCA ;         // pedestal of ADC for HC section (GeV) 
126  
127   Float_t fECAClusteringThreshold ;  // minimum energy to include a EC digit in a cluster
128   Float_t fHCAClusteringThreshold ;  // minimum energy to include a HC digit in a cluster
129   Float_t fPREClusteringThreshold ;  // minimum energy to include a PRE digit in a cluster
130   Float_t fECALocMaxCut ;            // minimum energy difference to distinguish local maxima in a cluster
131   Float_t fECAW0 ;                   // logarithmic weight for the cluster center of gravity calculation
132   Float_t fHCALocMaxCut ;            // minimum energy difference to distinguish local maxima in a cluster
133   Float_t fHCAW0 ;                   // logarithmic weight for the cluster center of gravity calculation
134   Float_t fPRELocMaxCut ;            //  minimum energy difference to distinguish local maxima in a CPV cluster
135   Float_t fPREW0 ;                   // logarithmic weight for the CPV cluster center of gravity calculation
136   Int_t fRecPointsInRun ;            //! Total number of recpoints in one run
137   Float_t fTimeGate ;                // Maximum time difference between the digits in ont EMC cluster
138     
139   ClassDef(AliEMCALClusterizerv1,2)   // Clusterizer implementation version 1
140
141 };
142
143 #endif // AliEMCALCLUSTERIZERV1_H