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