1 /**************************************************************************
2 * Copyright(c) 1998-2000, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
18 #include <Riostream.h>
20 #include <TObjArray.h>
22 #include <TDirectory.h>
25 #include "AliHitMap.h"
27 #include "AliMUONChamber.h"
28 #include "AliMUONConstants.h"
29 #include "AliMUONDigit.h"
30 #include "AliMUONHit.h"
31 #include "AliMUONHitMapA1.h"
32 #include "AliMUONMerger.h"
33 #include "AliMUONPadHit.h"
34 #include "AliMUONTransientDigit.h"
37 ClassImp(AliMUONMerger)
39 //___________________________________________
40 AliMUONMerger::AliMUONMerger()
42 // Default constructor
58 //------------------------------------------------------------------------
59 AliMUONMerger::~AliMUONMerger()
62 if (fTrH1) delete fTrH1;
63 if (fHitsBgr) delete fHitsBgr;
64 if (fPadHitsBgr) delete fPadHitsBgr;
65 if (fHitMap) delete [] fHitMap;
66 if (fList) delete fList;
67 if (fBgrFile) delete fBgrFile;
70 //------------------------------------------------------------------------
71 Bool_t AliMUONMerger::Exists(const AliMUONPadHit *padhit) const
73 // test if the given padhit was already fired
74 return (fHitMap[fNch]->TestHit(padhit->PadX(),padhit->PadY()));
77 //------------------------------------------------------------------------
78 void AliMUONMerger::Update(AliMUONPadHit *padhit)
80 // add new contribution to the fired padhit
81 AliMUONTransientDigit *pdigit =
82 static_cast<AliMUONTransientDigit*>(
83 fHitMap[fNch]->GetHit(padhit->PadX(),padhit->PadY()));
87 Int_t iqpad = padhit->QPad(); // charge per pad
88 pdigit->AddSignal(iqpad);
89 pdigit->AddPhysicsSignal(iqpad);
91 // update list of tracks
101 pdigit->UpdateTrackList(track,charge);
104 //------------------------------------------------------------------------
105 void AliMUONMerger::CreateNew(AliMUONPadHit *padhit)
107 // add new transient digit to the list, update hit map
108 fList->AddAtAndExpand(
109 new AliMUONTransientDigit(fNch,fDigits),fCounter);
110 fHitMap[fNch]->SetHit(padhit->PadX(),padhit->PadY(),fCounter);
111 AliMUONTransientDigit* pdigit =
112 static_cast<AliMUONTransientDigit*>(fList->Last());
117 charge = padhit->QPad();
122 pdigit->AddToTrackList(track,charge);
127 //------------------------------------------------------------------------
128 void AliMUONMerger::Init()
131 // open only once the background file !!
132 if (fMerge && !fBgrFile) fBgrFile = InitBgr();
137 //------------------------------------------------------------------------
138 TFile* AliMUONMerger::InitBgr()
140 // Initialise background event
141 TFile *file = new TFile(fFnBgr);
142 // add error checking later
143 printf("\n AliMUONMerger has opened %s file with background event \n", fFnBgr);
144 fHitsBgr = new TClonesArray("AliMUONHit",1000);
145 fPadHitsBgr = new TClonesArray("AliMUONPadHit",1000);
149 //------------------------------------------------------------------------
150 void AliMUONMerger::Digitise()
153 // keep galice.root for signal and name differently the file for
154 // background when add! otherwise the track info for signal will be lost !
156 AliMUONChamber* iChamber;
157 AliSegmentation* segmentation;
159 fList = new TObjArray;
161 AliMUON *pMUON = (AliMUON *) gAlice->GetModule("MUON");
162 fHitMap= new AliHitMap* [AliMUONConstants::NCh()];
166 // Get Hits Tree header from file
167 //if(fHitsBgr) fHitsBgr->Clear(); // Useless because line 327
168 //if(fPadHitsBgr) fPadHitsBgr->Clear(); // Useless because line 328
169 if(fTrH1) delete fTrH1;
173 sprintf(treeName,"TreeH%d",fEvNrBgr);
174 fTrH1 = (TTree*)gDirectory->Get(treeName);
176 printf("ERROR: cannot find Hits Tree for event:%d\n",fEvNrBgr);
179 // Set branch addresses
182 sprintf(branchname,"%s",pMUON->GetName());
183 if (fTrH1 && fHitsBgr) {
184 branch = fTrH1->GetBranch(branchname);
185 if (branch) branch->SetAddress(&fHitsBgr);
187 if (fTrH1 && fPadHitsBgr) {
188 branch = fTrH1->GetBranch("MUONCluster");
189 if (branch) branch->SetAddress(&fPadHitsBgr);
193 // loop over cathodes
196 for (int icat = 0; icat < 2; icat++) {
198 for (Int_t i = 0; i < AliMUONConstants::NCh(); i++) {
199 iChamber = &(pMUON->Chamber(i));
200 if (iChamber->Nsec() == 1 && icat == 1) {
203 segmentation = iChamber->SegmentationModel(icat+1);
205 fHitMap[i] = new AliMUONHitMapA1(segmentation, fList);
212 /******************************************************************/
213 TTree* treeH = pMUON->TreeH();
216 cerr<<"AliMUONMerger::Exec: Can not get TreeH"<<endl;
219 /******************************************************************/
223 Int_t ntracks = (Int_t) treeH->GetEntries();
224 treeH->SetBranchStatus("*",0); // switch off all branches
225 treeH->SetBranchStatus("MUON*",1); // switch on only MUON
227 for (fTrack = 0; fTrack < ntracks; fTrack++) {
229 treeH->GetEntry(fTrack,0);
232 for(AliMUONHit* mHit = (AliMUONHit*)pMUON->FirstHit(-1);
234 mHit = (AliMUONHit*)pMUON->NextHit())
236 fNch = mHit->Chamber()-1; // chamber number
237 if (fNch > AliMUONConstants::NCh()-1) continue;
238 iChamber = &(pMUON->Chamber(fNch));
241 // Loop over pad hits
242 for (AliMUONPadHit* mPad =
243 (AliMUONPadHit*)pMUON->FirstPad(mHit,pMUON->PadHits());
245 mPad = (AliMUONPadHit*)pMUON->NextPad(pMUON->PadHits()))
247 Int_t cathode = mPad->Cathode(); // cathode number
248 if (cathode != (icat+1)) continue;
249 Int_t iqpad = Int_t(mPad->QPad()); // charge per pad
250 // segmentation = iChamber->SegmentationModel(cathode);
251 fDigits[0] = mPad->PadX();
252 fDigits[1] = mPad->PadY();
253 if (!(fHitMap[fNch]->ValidateHit(fDigits[0], fDigits[1]))) continue;
257 if (mHit->Particle() == kMuonPlus ||
258 mHit->Particle() == kMuonMinus) {
259 fDigits[5] = mPad->HitNumber();
260 } else fDigits[5] = -1;
262 // build the list of fired pads and update the info
269 } //end loop over clusters
273 // open the file with background
277 ntracks = (Int_t)fTrH1->GetEntries();
281 for (fTrack = 0; fTrack < ntracks; fTrack++) {
283 if (fHitsBgr) fHitsBgr->Clear();
284 if (fPadHitsBgr) fPadHitsBgr->Clear();
286 fTrH1->GetEvent(fTrack);
290 for(Int_t i = 0; i < fHitsBgr->GetEntriesFast(); ++i)
292 mHit = (AliMUONHit*) (*fHitsBgr)[i];
293 fNch = mHit->Chamber()-1; // chamber number
294 iChamber = &(pMUON->Chamber(fNch));
296 // Loop over pad hits
297 for (AliMUONPadHit* mPad =
298 (AliMUONPadHit*)pMUON->FirstPad(mHit,fPadHitsBgr);
300 mPad = (AliMUONPadHit*)pMUON->NextPad(fPadHitsBgr))
302 Int_t cathode = mPad->Cathode(); // cathode number
303 Int_t ipx = mPad->PadX(); // pad number on X
304 Int_t ipy = mPad->PadY(); // pad number on Y
305 Int_t iqpad = Int_t(mPad->QPad()); // charge per pad
306 if (!(fHitMap[fNch]->ValidateHit(ipx, ipy))) continue;
308 if (cathode != (icat+1)) continue;
316 // build the list of fired pads and update the info
322 } //end loop over clusters
326 TTree *treeK = gAlice->TreeK();
329 if (treeK) file = treeK->GetCurrentFile();
333 Int_t tracks[kMAXTRACKS];
334 Int_t charges[kMAXTRACKS];
335 Int_t nentries = fList->GetEntriesFast();
337 for (Int_t nent = 0; nent < nentries; nent++) {
338 AliMUONTransientDigit *address = (AliMUONTransientDigit*)fList->At(nent);
339 if (address == 0) continue;
340 Int_t ich = address->Chamber();
341 Int_t q = address->Signal();
342 iChamber = &(pMUON->Chamber(ich));
344 // Digit Response (noise, threshold, saturation, ...)
345 AliMUONResponse * response = iChamber->ResponseModel();
346 q = response->DigitResponse(q,address);
350 fDigits[0] = address->PadX();
351 fDigits[1] = address->PadY();
352 fDigits[2] = address->Cathode();
354 fDigits[4] = address->Physics();
355 fDigits[5] = address->Hit();
357 Int_t nptracks = address->GetNTracks();
359 if (nptracks > kMAXTRACKS) {
361 printf("\n Attention - nptracks > kMAXTRACKS %d \n", nptracks);
362 nptracks = kMAXTRACKS;
366 printf("Attention - nptracks > 2 %d \n",nptracks);
367 printf("cat,ich,ix,iy,q %d %d %d %d %d \n",icat,ich,fDigits[0],fDigits[1],q);
370 for (Int_t tr = 0; tr < nptracks; tr++) {
371 tracks[tr] = address->GetTrack(tr);
372 charges[tr] = address->GetCharge(tr);
373 } //end loop over list of tracks for one pad
374 // Sort list of tracks according to charge
376 SortTracks(tracks,charges,nptracks);
378 if (nptracks < kMAXTRACKS ) {
379 for (Int_t i = nptracks; i < kMAXTRACKS; i++) {
386 pMUON->AddDigits(ich,tracks,charges,fDigits);
388 gAlice->TreeD()->Fill();
389 pMUON->ResetDigits();
393 for(Int_t ii = 0; ii < AliMUONConstants::NCh(); ++ii) {
399 } //end loop over cathodes
403 // no need to delete ... and it makes a crash also
404 // if (fHitsBgr) fHitsBgr->Delete();
405 // if (fPadHitsBgr) fPadHitsBgr->Delete();
406 // gObjectTable->Print();
411 void AliMUONMerger::SortTracks(Int_t *tracks,Int_t *charges,Int_t ntr)
414 // Sort the list of tracks contributing to a given digit
415 // Only the 3 most significant tracks are acctually sorted
419 // Loop over signals, only 3 times
424 Int_t idx[3] = {-2,-2,-2};
425 Int_t jch[3] = {-2,-2,-2};
426 Int_t jtr[3] = {-2,-2,-2};
437 if((i == 1 && j == idx[i-1])
438 ||(i == 2 && (j == idx[i-1] || j == idx[i-2]))) continue;
440 if(charges[j] > qmax) {
448 jch[i]=charges[jmax];