Fixes for memory leaks (Christian)
[u/mrichter/AliRoot.git] / HMPID / AliHMPIDReconstructor.h
1 #ifndef AliHMPIDReconstructor_h
2 #define AliHMPIDReconstructor_h
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5 //.
6 // HMPID base class to reconstruct an event
7 //.
8 #include <AliReconstructor.h>        //base class
9 #include "AliHMPIDTracker.h"         //CreateTracker()
10 #include "AliHMPIDDigit.h"           //Dig2Clu(), UseDig()
11 #include <TMatrixF.h>                //UseDig()
12 #include <TClonesArray.h>            //UseDig()
13 #include <TObjArray.h>               //SigConv()
14 class AliRawReader;                  //Reconstruct() with raw data   
15 class AliHMPIDCluster;               //Dig2Clu()
16
17 class AliHMPIDReconstructor: public AliReconstructor 
18 {
19 public:
20            AliHMPIDReconstructor();              
21   virtual ~AliHMPIDReconstructor()                                  {delete fDig;delete fClu;delete [] fUserCut;}//dtor  
22 //framework part  
23   AliTracker*  CreateTracker         (AliRunLoader*                      )const{return new AliHMPIDTracker;}            //from AliReconstructor for clusters->PID
24   void         ConvertDigits         (AliRawReader *pRR, TTree *pDigTree) const;                                        //from AliReconstruction for raw->digit
25   Bool_t       HasDigitConversion()   const {return kTRUE;}                                                             //HMPID digits converted with ConvertDigits 
26   void         Reconstruct           (TTree* digitsTree, TTree* clustersTree) const;                                    //from AliReconstruction for digit->cluster
27   void         Reconstruct           (AliRunLoader *pAL,AliRawReader* pRR)const;                                        //from AliReconstruction for raw->cluster with Digits on fly
28   Bool_t       HasLocalReconstruction() const {return kTRUE;}                                                           // HMPID has local reconstruction algorithm
29   void         FillESD               (AliRunLoader* pAL,AliESD *pESD)const;                                             //calculate pid for HMPID
30   
31   using AliReconstructor::FillESD;                                                                                      //
32   using AliReconstructor::Reconstruct;                                                                                  // 
33
34   //private part  
35   static        void           Dig2Clu (TObjArray *pDigLst,TObjArray *pCluLst,Bool_t isUnfold=kTRUE                      );//digits->clusters
36   static        void           FormClu (AliHMPIDCluster *pClu,AliHMPIDDigit *pDig,TClonesArray *pDigLst,TMatrixF *pPadMap);//cluster formation recursive algorithm
37   static inline AliHMPIDDigit* UseDig  (Int_t padX,Int_t padY,                    TClonesArray *pDigLst,TMatrixF *pDigMap);//use this pad's digit to form a cluster
38   inline Bool_t                IsDigSurvive(AliHMPIDDigit *pDig                                                     )const;//check for sigma cut
39   protected:
40   Int_t     *fUserCut;                 // n sigmas for pedestals decided by the User for each chamber(if in OCDB)
41   TObjArray *fDaqSig;                  // container for the pad pedestal sigmas
42   TObjArray *fDig;                     // tmp list of digits
43   TObjArray *fClu;                     // tmp list of clusters
44   ClassDef(AliHMPIDReconstructor, 0)   // class for the HMPID reconstruction
45 };
46
47 //__________________________________________________________________________________________________
48 AliHMPIDDigit* AliHMPIDReconstructor::UseDig(Int_t padX,Int_t padY,TClonesArray *pDigLst,TMatrixF *pPadMap)
49 {
50 //Digit map contains a matrix if digit numbers.
51 //Main operation in forming initial cluster is done here. Requested digit pointer is returned and this digit marked as taken.
52 //Arguments: padX,padY - pad number
53 //           pDigLst   - list of digits for one sector
54 //           pDigMap   - map of those digits
55 //  Returns: pointer to digit if not yet used or 0 if used
56   Int_t iDig=(Int_t)(*pPadMap)(padX,padY);(*pPadMap)(padX,padY)=-1;//take digit number from the map and reset this map cell to -1
57   if(iDig!=-1)    return (AliHMPIDDigit*)pDigLst->At(iDig);        //digit pointer
58   else            return 0;
59 }
60 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
61 Bool_t AliHMPIDReconstructor::IsDigSurvive(AliHMPIDDigit *pDig)const
62 {
63 //Check if the current digit survive to a riapllied sigma cut
64 //Arguments: pDig pointer to the current digit
65 //  Returns: kTRUE if charge > mean+n*sigma
66   Int_t iCh = pDig->Ch();
67   Int_t iDaqSigCut =(Int_t)fDaqSig->At(iCh)->GetUniqueID(); 
68   if(fUserCut[iCh]<=iDaqSigCut) return kTRUE;
69   TMatrixF *pM = (TMatrixF*)fDaqSig->At(pDig->Ch());
70   Float_t sig = (*pM)(pDig->PadChX(),pDig->PadChY());
71   if(pDig->Q()>fUserCut[iCh]*sig) return kTRUE;
72   else return kFALSE;
73 }
74
75 #endif