HMPID module
[u/mrichter/AliRoot.git] / HMPID / HMPIDsim / AliHMPIDDigitizer.cxx
CommitLineData
d3da6dc4 1/**************************************************************************
2 * Copyright(c) 1998-2000, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
1b10372c 15//.
16//.
17//.
18//.
19//.
20#include <AliRun.h>
d3da6dc4 21#include <AliRunLoader.h>
f21fc003 22#include "AliDigitizationInput.h"
d3da6dc4 23#include <AliLoader.h>
6f842d2a 24#include "AliLog.h"
23ba1e93 25#include <AliCDBEntry.h>
26#include <AliCDBManager.h>
1b10372c 27#include "AliHMPIDDigitizer.h"
23ba1e93 28#include "AliHMPIDReconstructor.h"
1b10372c 29#include "AliHMPIDDigit.h"
30#include "AliHMPID.h"
ae5a42aa 31#include "AliHMPIDParam.h"
1b10372c 32#include <TRandom.h>
23ba1e93 33#include <TMath.h>
c93255fe 34#include <TTree.h>
23ba1e93 35#include <TObjArray.h>
d3da6dc4 36
37ClassImp(AliHMPIDDigitizer)
38
23ba1e93 39Bool_t AliHMPIDDigitizer::fgDoNoise=kTRUE;
d3da6dc4 40//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
f21fc003 41void AliHMPIDDigitizer::Digitize(Option_t*)
d3da6dc4 42{
43// This methode is responsible for merging sdigits to a list of digits
44//Disintegration leeds to the fact that one hit affects several neighbouring pads, which means that the same pad might be affected by few hits.
f21fc003 45 AliDebug(1,Form("Start with %i input(s) for event %i",fDigInput->GetNinputs(),fDigInput->GetOutputEventNr()));
d3da6dc4 46//First we read all sdigits from all inputs
47 AliRunLoader *pInRunLoader=0;//in and out Run loaders
48 AliLoader *pInRichLoader=0;//in and out HMPID loaders
d6183365 49 static TClonesArray sdigs("AliHMPIDDigit");//tmp storage for sdigits sum up from all input files
d3da6dc4 50 Int_t total=0;
f21fc003 51 for(Int_t inFileN=0;inFileN<fDigInput->GetNinputs();inFileN++){//files loop
52 pInRunLoader = AliRunLoader::GetRunLoader(fDigInput->GetInputFolderName(inFileN)); //get run loader from current input
d3da6dc4 53 pInRichLoader = pInRunLoader->GetLoader("HMPIDLoader"); if(pInRichLoader==0) continue; //no HMPID in this input, check the next input
54 if (!pInRunLoader->GetAliRun()) pInRunLoader->LoadgAlice();
55 AliHMPID* pInRich=(AliHMPID*)pInRunLoader->GetAliRun()->GetDetector("HMPID"); //take HMPID from current input
56 pInRichLoader->LoadSDigits(); pInRichLoader->TreeS()->GetEntry(0); //take list of HMPID sdigits from current input
57 AliDebug(1,Form("input %i has %i sdigits",inFileN,pInRich->SdiLst()->GetEntries()));
58 for(Int_t i=0;i<pInRich->SdiLst()->GetEntries();i++){ //collect sdigits from current input
59 AliHMPIDDigit *pSDig=(AliHMPIDDigit*)pInRich->SdiLst()->At(i);
f21fc003 60 pSDig->AddTidOffset(fDigInput->GetMask(inFileN)); //apply TID shift since all inputs count tracks independently starting from 0
d3da6dc4 61 new(sdigs[total++]) AliHMPIDDigit(*pSDig);
62 }
63 pInRichLoader->UnloadSDigits(); pInRich->SdiReset(); //close current input and reset
64 }//files loop
65
66 //PH if(sdigs.GetEntries()==0) return; //no sdigits collected, nothing to convert
67
f21fc003 68 AliRunLoader *pOutRunLoader = AliRunLoader::GetRunLoader(fDigInput->GetOutputFolderName()); //open output stream (only 1 possible)
d3da6dc4 69 AliLoader *pOutRichLoader = pOutRunLoader->GetLoader("HMPIDLoader"); //take output HMPID loader
1b10372c 70 AliRun *pArun = pOutRunLoader->GetAliRun();
71 AliHMPID *pOutRich = (AliHMPID*)pArun->GetDetector("HMPID"); //take output HMPID
d3da6dc4 72 pOutRichLoader->MakeTree("D"); pOutRich->MakeBranch("D"); //create TreeD in output stream
73
74 Sdi2Dig(&sdigs,pOutRich->DigLst());
75
76 pOutRichLoader->TreeD()->Fill(); //fill the output tree with the list of digits
77 pOutRichLoader->WriteDigits("OVERWRITE"); //serialize them to file
78
79 sdigs.Clear(); //remove all tmp sdigits
80 pOutRichLoader->UnloadDigits(); pOutRich->DigReset();
81}//Exec()
82//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
83void AliHMPIDDigitizer::Sdi2Dig(TClonesArray *pSdiLst,TObjArray *pDigLst)
84{
85// Converts list of sdigits to 7 lists of digits, one per each chamber
86// Arguments: pSDigLst - list of all sdigits
87// pDigLst - list of 7 lists of digits
88// Returns: none
89
90 TClonesArray *pLst[7]; Int_t iCnt[7];
1b10372c 91
d3da6dc4 92 for(Int_t i=0;i<7;i++){
93 pLst[i]=(TClonesArray*)(*pDigLst)[i];
3c6274c1 94 iCnt[i]=0; if(pLst[i]->GetEntries()!=0) AliErrorClass("Some of digits lists is not empty"); //in principle those lists should be empty
d3da6dc4 95 }
6f842d2a 96
c5c19d6a 97 // make noise array
cd2a74bf 98 Float_t arrNoise[7][6][80][48], arrSigmaPed[7][6][80][48];
94b1fbfa 99 if(fgDoNoise) {
6f842d2a 100 for (Int_t iCh=AliHMPIDParam::kMinCh;iCh<=AliHMPIDParam::kMaxCh;iCh++)
ae5a42aa 101 for (Int_t iPc=AliHMPIDParam::kMinPc;iPc<=AliHMPIDParam::kMaxPc;iPc++)
102 for(Int_t iPx=AliHMPIDParam::kMinPx;iPx<=AliHMPIDParam::kMaxPx;iPx++)
23ba1e93 103 for(Int_t iPy=AliHMPIDParam::kMinPy;iPy<=AliHMPIDParam::kMaxPy;iPy++){
cd2a74bf 104 arrNoise[iCh][iPc][iPx][iPy] = gRandom->Gaus(0,1.);
105 arrSigmaPed[iCh][iPc][iPx][iPy] = 1.;
106 }
107
108 AliCDBEntry *pDaqSigEnt = AliCDBManager::Instance()->Get("HMPID/Calib/DaqSig"); //contains TObjArray of TObjArray 14 TMatrixF sigmas values for pads
109
110 if(pDaqSigEnt){
111 TObjArray *pDaqSig = (TObjArray*)pDaqSigEnt->GetObject();
112 for(Int_t iCh=AliHMPIDParam::kMinCh;iCh<=AliHMPIDParam::kMaxCh;iCh++){ //chambers loop
113 TMatrixF *pM = (TMatrixF*)pDaqSig->At(iCh);
114 for (Int_t iPc=AliHMPIDParam::kMinPc;iPc<=AliHMPIDParam::kMaxPc;iPc++)
115 for(Int_t iPx=AliHMPIDParam::kMinPx;iPx<=AliHMPIDParam::kMaxPx;iPx++)
116 for(Int_t iPy=AliHMPIDParam::kMinPy;iPy<=AliHMPIDParam::kMaxPy;iPy++){
117 Int_t padX = (iPc%2)*AliHMPIDParam::kPadPcX+iPx;
118 Int_t padY = (iPc/2)*AliHMPIDParam::kPadPcY+iPy;
119 if((*pM)(padX,padY)>0.){
120 arrNoise[iCh][iPc][iPx][iPy] = gRandom->Gaus(0,(*pM)(padX,padY));
121 arrSigmaPed[iCh][iPc][iPx][iPy] = (*pM)(padX,padY);}
122 else{
123 arrNoise[iCh][iPc][iPx][iPy] = gRandom->Gaus(0,1.);
124 arrSigmaPed[iCh][iPc][iPx][iPy] = 1.;}
125 }
126 }
127 }
c5c19d6a 128 }
d3da6dc4 129
130 pSdiLst->Sort();
131
132 Int_t iPad=-1,iCh=-1,iNdigPad=-1,aTids[3]={-1,-1,-1}; Float_t q=-1;
133 for(Int_t i=0;i<pSdiLst->GetEntries();i++){ //sdigits loop (sorted)
23ba1e93 134 AliHMPIDDigit *pSdig=(AliHMPIDDigit*)pSdiLst->At(i); //take current sdigit
d3da6dc4 135 if(pSdig->Pad()==iPad){ //if the same pad
136 q+=pSdig->Q(); //sum up charge
137 iNdigPad++; if(iNdigPad<=3) aTids[iNdigPad-1]=pSdig->GetTrack(0); //collect TID
138 continue;
139 }
23ba1e93 140 if(i!=0 && iCh>=AliHMPIDParam::kMinCh && iCh<=AliHMPIDParam::kMaxCh){
cd2a74bf 141 AliHMPIDParam::Instance()->SetThreshold((TMath::Nint(arrSigmaPed[iCh][pSdig->Pc()][pSdig->PadPcX()][pSdig->PadPcY()])*AliHMPIDParam::Nsig()));
23ba1e93 142 if(AliHMPIDParam::IsOverTh(q)) new((*pLst[iCh])[iCnt[iCh]++]) AliHMPIDDigit(iPad,(Int_t)q,aTids);} //do not create digit for the very first sdigit
aa8db9d6 143
ae5a42aa 144 iPad=pSdig->Pad(); iCh=AliHMPIDParam::A2C(iPad); //new sdigit comes, reset collectors
d3da6dc4 145 iNdigPad=1;
146 aTids[0]=pSdig->GetTrack(0);aTids[1]=aTids[2]=-1;
c5c19d6a 147 q=pSdig->Q();
94b1fbfa 148 if(fgDoNoise) q+=arrNoise[iCh][pSdig->Pc()][pSdig->PadPcX()][pSdig->PadPcY()];
c5c19d6a 149 arrNoise[iCh][pSdig->Pc()][pSdig->PadPcX()][pSdig->PadPcY()]=0;
d3da6dc4 150 }//sdigits loop (sorted)
151
23ba1e93 152 if(iCh>=AliHMPIDParam::kMinCh && iCh<=AliHMPIDParam::kMaxCh){
153 Int_t pc = AliHMPIDParam::A2P(iPad);
cd2a74bf 154 Int_t px = AliHMPIDParam::A2X(iPad);
155 Int_t py = AliHMPIDParam::A2Y(iPad);
156 AliHMPIDParam::Instance()->SetThreshold((TMath::Nint(arrSigmaPed[iCh][pc][px][py])*AliHMPIDParam::Nsig()));
157 if(AliHMPIDParam::IsOverTh(q)) new((*pLst[iCh])[iCnt[iCh]++]) AliHMPIDDigit(iPad,(Int_t)q,aTids);
158 } //add the last one, in case of empty sdigits list q=-1
aa8db9d6 159
c5c19d6a 160// add noise pad above threshold with no signal merged...if any
94b1fbfa 161 if(!fgDoNoise) return;
23ba1e93 162
c5c19d6a 163 aTids[0]=aTids[1]=aTids[2]=-1;
23ba1e93 164 for (Int_t iChCurr=AliHMPIDParam::kMinCh;iChCurr<=AliHMPIDParam::kMaxCh;iChCurr++){
ae5a42aa 165 for (Int_t iPc=AliHMPIDParam::kMinPc;iPc<=AliHMPIDParam::kMaxPc;iPc++)
166 for(Int_t iPx=AliHMPIDParam::kMinPx;iPx<=AliHMPIDParam::kMaxPx;iPx++)
167 for(Int_t iPy=AliHMPIDParam::kMinPy;iPy<=AliHMPIDParam::kMaxPy;iPy++) {
ec575c20 168 Float_t qNoise = arrNoise[iChCurr][iPc][iPx][iPy];
cd2a74bf 169 AliHMPIDParam::Instance()->SetThreshold((TMath::Nint(arrSigmaPed[iChCurr][iPc][iPx][iPy])*AliHMPIDParam::Nsig()));
ec575c20 170 if(AliHMPIDParam::IsOverTh(qNoise)) new((*pLst[iChCurr])[iCnt[iChCurr]++]) AliHMPIDDigit(AliHMPIDParam::Abs(iChCurr,iPc,iPx,iPy),(Int_t)qNoise,aTids);
c5c19d6a 171 }
23ba1e93 172 }
d3da6dc4 173}//Sdi2Dig()
174//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++