]> git.uio.no Git - u/mrichter/AliRoot.git/blame - MUON/AliMUONMerger.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
2D Invariant Mass plots
[u/mrichter/AliRoot.git] / MUON / AliMUONMerger.cxx
CommitLineData
66f93042 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 **************************************************************************/
15
88cb7938 16/* $Id$ */
ba030c0e 17
88cb7938 18#include <Riostream.h>
19#include <TTree.h>
116cbefd 20#include <TObjArray.h>
88cb7938 21#include <TFile.h>
22#include <TDirectory.h>
116cbefd 23#include <TPDGCode.h>
66f93042 24
66f93042 25#include "AliHitMap.h"
66f93042 26#include "AliMUON.h"
116cbefd 27#include "AliMUONChamber.h"
28#include "AliMUONConstants.h"
29#include "AliMUONDigit.h"
66f93042 30#include "AliMUONHit.h"
116cbefd 31#include "AliMUONHitMapA1.h"
32#include "AliMUONMerger.h"
66f93042 33#include "AliMUONPadHit.h"
66f93042 34#include "AliMUONTransientDigit.h"
35#include "AliRun.h"
66f93042 36
37ClassImp(AliMUONMerger)
38
11ca64ac 39//----------------------------------------------------------------------
66f93042 40AliMUONMerger::AliMUONMerger()
41{
42// Default constructor
43 fEvNrSig = 0;
44 fEvNrBgr = 0;
45 fMerge = kDigitize;
46 fFnBgr = 0;
47 fHitMap = 0;
48 fList = 0;
49 fTrH1 = 0;
50 fHitsBgr = 0;
51 fPadHitsBgr = 0;
52 fHitMap = 0;
53 fList = 0;
77ccf5b0 54 fBgrFile = 0;
ba030c0e 55 fDebug = 0;
66f93042 56}
57
11ca64ac 58//----------------------------------------------------------------------
59AliMUONMerger::AliMUONMerger(const AliMUONMerger&)
60{
61// Protected copy constructor
62
63 Fatal("AliMUONMergerModule", "Not implemented.");
64}
65
66f93042 66//------------------------------------------------------------------------
67AliMUONMerger::~AliMUONMerger()
68{
69// Destructor
70 if (fTrH1) delete fTrH1;
71 if (fHitsBgr) delete fHitsBgr;
72 if (fPadHitsBgr) delete fPadHitsBgr;
3ec837bb 73 if (fHitMap) delete [] fHitMap;
66f93042 74 if (fList) delete fList;
77ccf5b0 75 if (fBgrFile) delete fBgrFile;
66f93042 76}
77
11ca64ac 78//----------------------------------------------------------------------
79AliMUONMerger& AliMUONMerger::operator=(const AliMUONMerger& rhs)
80{
81// Protected assignement operator
82
83 if (this == &rhs) return *this;
84
85 Fatal("operator=", "Not implemented.");
86
87 return *this;
88}
89
66f93042 90//------------------------------------------------------------------------
c7955c78 91Bool_t AliMUONMerger::Exists(const AliMUONPadHit *padhit) const
66f93042 92{
3ec837bb 93// test if the given padhit was already fired
66f93042 94 return (fHitMap[fNch]->TestHit(padhit->PadX(),padhit->PadY()));
95}
96
97//------------------------------------------------------------------------
8f36c696 98void AliMUONMerger::Update(AliMUONPadHit *padhit)
66f93042 99{
3ec837bb 100// add new contribution to the fired padhit
8f36c696 101 AliMUONTransientDigit *pdigit =
102 static_cast<AliMUONTransientDigit*>(
103 fHitMap[fNch]->GetHit(padhit->PadX(),padhit->PadY()));
66f93042 104
66f93042 105 // update charge
106 //
8f36c696 107 Int_t iqpad = padhit->QPad(); // charge per pad
108 pdigit->AddSignal(iqpad);
109 pdigit->AddPhysicsSignal(iqpad);
110
66f93042 111 // update list of tracks
112 //
3351e14b 113 Int_t track, charge;
8f36c696 114 if (fSignal) {
115 track = fTrack;
116 charge = iqpad;
66f93042 117 } else {
8f36c696 118 track = kBgTag;
119 charge = kBgTag;
66f93042 120 }
8f36c696 121 pdigit->UpdateTrackList(track,charge);
66f93042 122}
123
124//------------------------------------------------------------------------
8f36c696 125void AliMUONMerger::CreateNew(AliMUONPadHit *padhit)
66f93042 126{
3ec837bb 127// add new transient digit to the list, update hit map
66f93042 128 fList->AddAtAndExpand(
129 new AliMUONTransientDigit(fNch,fDigits),fCounter);
8f36c696 130 fHitMap[fNch]->SetHit(padhit->PadX(),padhit->PadY(),fCounter);
131 AliMUONTransientDigit* pdigit =
c7955c78 132 static_cast<AliMUONTransientDigit*>(fList->Last());
66f93042 133 // list of tracks
8f36c696 134 Int_t track, charge;
135 if (fSignal) {
136 track = fTrack;
137 charge = padhit->QPad();
138 } else {
139 track = kBgTag;
140 charge = kBgTag;
141 }
142 pdigit->AddToTrackList(track,charge);
143 fCounter++;
66f93042 144}
145
146
147//------------------------------------------------------------------------
148void AliMUONMerger::Init()
149{
150// Initialisation
77ccf5b0 151 // open only once the background file !!
152 if (fMerge && !fBgrFile) fBgrFile = InitBgr();
66f93042 153}
154
155
156
157//------------------------------------------------------------------------
158TFile* AliMUONMerger::InitBgr()
159{
160// Initialise background event
161 TFile *file = new TFile(fFnBgr);
162// add error checking later
163 printf("\n AliMUONMerger has opened %s file with background event \n", fFnBgr);
164 fHitsBgr = new TClonesArray("AliMUONHit",1000);
165 fPadHitsBgr = new TClonesArray("AliMUONPadHit",1000);
166 return file;
167}
168
169//------------------------------------------------------------------------
170void AliMUONMerger::Digitise()
171{
172
173 // keep galice.root for signal and name differently the file for
174 // background when add! otherwise the track info for signal will be lost !
ce3f5e87 175 // Obsolete sep 2003 Gines MARTINEZ
176 // AliMUONChamber* iChamber;
177// AliSegmentation* segmentation;
178
179// fList = new TObjArray;
180
181// AliMUON *pMUON = (AliMUON *) gAlice->GetModule("MUON");
182// fHitMap= new AliHitMap* [AliMUONConstants::NCh()];
183// if (fMerge ) {
184// fBgrFile->cd();
185// //
186// // Get Hits Tree header from file
187// //if(fHitsBgr) fHitsBgr->Clear(); // Useless because line 327
188// //if(fPadHitsBgr) fPadHitsBgr->Clear(); // Useless because line 328
189// if(fTrH1) delete fTrH1;
190// fTrH1 = 0;
66f93042 191
ce3f5e87 192// char treeName[20];
193// sprintf(treeName,"TreeH%d",fEvNrBgr);
194// fTrH1 = (TTree*)gDirectory->Get(treeName);
195// if (!fTrH1) {
196// printf("ERROR: cannot find Hits Tree for event:%d\n",fEvNrBgr);
197// }
198// //
199// // Set branch addresses
200// TBranch *branch;
201// char branchname[20];
202// sprintf(branchname,"%s",pMUON->GetName());
203// if (fTrH1 && fHitsBgr) {
204// branch = fTrH1->GetBranch(branchname);
205// if (branch) branch->SetAddress(&fHitsBgr);
206// }
207// if (fTrH1 && fPadHitsBgr) {
208// branch = fTrH1->GetBranch("MUONCluster");
209// if (branch) branch->SetAddress(&fPadHitsBgr);
210// }
211// }
212// //
213// // loop over cathodes
214// //
215// fSignal = kTRUE;
216// for (int icat = 0; icat < 2; icat++) {
217// fCounter = 0;
218// for (Int_t i = 0; i < AliMUONConstants::NCh(); i++) {
219// iChamber = &(pMUON->Chamber(i));
220// if (iChamber->Nsec() == 1 && icat == 1) {
221// continue;
222// } else {
223// segmentation = iChamber->SegmentationModel(icat+1);
224// }
225// fHitMap[i] = new AliMUONHitMapA1(segmentation, fList);
226// }
227
228// //
229// // Loop over tracks
230// //
231
232// /******************************************************************/
233// TTree* treeH = pMUON->TreeH();
234// if (treeH == 0x0)
235// {
236// cerr<<"AliMUONMerger::Exec: Can not get TreeH"<<endl;
237// return;
238// }
239// /******************************************************************/
88cb7938 240
241
242
ce3f5e87 243// Int_t ntracks = (Int_t) treeH->GetEntries();
244// treeH->SetBranchStatus("*",0); // switch off all branches
245// treeH->SetBranchStatus("MUON*",1); // switch on only MUON
246
247// for (fTrack = 0; fTrack < ntracks; fTrack++) {
248// gAlice->ResetHits();
249// treeH->GetEntry(fTrack,0);
250// //
251// // Loop over hits
252// for(AliMUONHit* mHit = (AliMUONHit*)pMUON->FirstHit(-1);
253// mHit;
254// mHit = (AliMUONHit*)pMUON->NextHit())
255// {
256// fNch = mHit->Chamber()-1; // chamber number
257// if (fNch > AliMUONConstants::NCh()-1) continue;
258// iChamber = &(pMUON->Chamber(fNch));
66f93042 259
ce3f5e87 260// //
261// // Loop over pad hits
262// for (AliMUONPadHit* mPad =
263// (AliMUONPadHit*)pMUON->FirstPad(mHit,pMUON->PadHits());
264// mPad;
265// mPad = (AliMUONPadHit*)pMUON->NextPad(pMUON->PadHits()))
266// {
267// Int_t cathode = mPad->Cathode(); // cathode number
268// if (cathode != (icat+1)) continue;
269// Int_t iqpad = Int_t(mPad->QPad()); // charge per pad
270// // segmentation = iChamber->SegmentationModel(cathode);
271// fDigits[0] = mPad->PadX();
272// fDigits[1] = mPad->PadY();
273// if (!(fHitMap[fNch]->ValidateHit(fDigits[0], fDigits[1]))) continue;
274// fDigits[2] = icat;
275// fDigits[3] = iqpad;
276// fDigits[4] = iqpad;
277// if (mHit->Particle() == kMuonPlus ||
278// mHit->Particle() == kMuonMinus) {
279// fDigits[5] = mPad->HitNumber();
280// } else fDigits[5] = -1;
281
282// // build the list of fired pads and update the info
283
284// if (!Exists(mPad)) {
285// CreateNew(mPad);
286// } else {
287// Update(mPad);
288// } // end if pdigit
289// } //end loop over clusters
290// } // hit loop
291// } // track loop
292
293// // open the file with background
66f93042 294
ce3f5e87 295// if (fMerge) {
296// fSignal = kFALSE;
297// ntracks = (Int_t)fTrH1->GetEntries();
298// //
299// // Loop over tracks
300// //
301// for (fTrack = 0; fTrack < ntracks; fTrack++) {
302
303// if (fHitsBgr) fHitsBgr->Clear();
304// if (fPadHitsBgr) fPadHitsBgr->Clear();
305
306// fTrH1->GetEvent(fTrack);
307// //
308// // Loop over hits
309// AliMUONHit* mHit;
310// for(Int_t i = 0; i < fHitsBgr->GetEntriesFast(); ++i)
311// {
312// mHit = (AliMUONHit*) (*fHitsBgr)[i];
313// fNch = mHit->Chamber()-1; // chamber number
314// iChamber = &(pMUON->Chamber(fNch));
315// //
316// // Loop over pad hits
317// for (AliMUONPadHit* mPad =
318// (AliMUONPadHit*)pMUON->FirstPad(mHit,fPadHitsBgr);
319// mPad;
320// mPad = (AliMUONPadHit*)pMUON->NextPad(fPadHitsBgr))
321// {
322// Int_t cathode = mPad->Cathode(); // cathode number
323// Int_t ipx = mPad->PadX(); // pad number on X
324// Int_t ipy = mPad->PadY(); // pad number on Y
325// Int_t iqpad = Int_t(mPad->QPad()); // charge per pad
326// if (!(fHitMap[fNch]->ValidateHit(ipx, ipy))) continue;
327
328// if (cathode != (icat+1)) continue;
329// fDigits[0] = ipx;
330// fDigits[1] = ipy;
331// fDigits[2] = icat;
332// fDigits[3] = iqpad;
333// fDigits[4] = 0;
334// fDigits[5] = -1;
66f93042 335
ce3f5e87 336// // build the list of fired pads and update the info
337// if (!Exists(mPad)) {
338// CreateNew(mPad);
339// } else {
340// Update(mPad);
341// } // end if !Exists
342// } //end loop over clusters
343// } // hit loop
344// } // track loop
345
346// TTree *treeK = gAlice->TreeK();
347// TFile *file = NULL;
66f93042 348
ce3f5e87 349// if (treeK) file = treeK->GetCurrentFile();
350// file->cd();
351// } // if fMerge
66f93042 352
ce3f5e87 353// Int_t tracks[kMAXTRACKS];
354// Int_t charges[kMAXTRACKS];
355// Int_t nentries = fList->GetEntriesFast();
66f93042 356
ce3f5e87 357// for (Int_t nent = 0; nent < nentries; nent++) {
358// AliMUONTransientDigit *address = (AliMUONTransientDigit*)fList->At(nent);
359// if (address == 0) continue;
360// Int_t ich = address->Chamber();
361// Int_t q = address->Signal();
362// iChamber = &(pMUON->Chamber(ich));
363// //
364// // Digit Response (noise, threshold, saturation, ...)
365// AliMUONResponse * response = iChamber->ResponseModel();
366// q = response->DigitResponse(q,address);
66f93042 367
ce3f5e87 368// if (!q) continue;
66f93042 369
ce3f5e87 370// fDigits[0] = address->PadX();
371// fDigits[1] = address->PadY();
372// fDigits[2] = address->Cathode();
373// fDigits[3] = q;
374// fDigits[4] = address->Physics();
375// fDigits[5] = address->Hit();
66f93042 376
ce3f5e87 377// Int_t nptracks = address->GetNTracks();
378
379// if (nptracks > kMAXTRACKS) {
380// if (fDebug>0)
381// printf("\n Attention - nptracks > kMAXTRACKS %d \n", nptracks);
382// nptracks = kMAXTRACKS;
383// }
384// if (nptracks > 2) {
385// if (fDebug>0) {
386// printf("Attention - nptracks > 2 %d \n",nptracks);
387// printf("cat,ich,ix,iy,q %d %d %d %d %d \n",icat,ich,fDigits[0],fDigits[1],q);
388// }
389// }
390// for (Int_t tr = 0; tr < nptracks; tr++) {
391// tracks[tr] = address->GetTrack(tr);
392// charges[tr] = address->GetCharge(tr);
393// } //end loop over list of tracks for one pad
394// // Sort list of tracks according to charge
395// if (nptracks > 1) {
396// SortTracks(tracks,charges,nptracks);
397// }
398// if (nptracks < kMAXTRACKS ) {
399// for (Int_t i = nptracks; i < kMAXTRACKS; i++) {
400// tracks[i] = 0;
401// charges[i] = 0;
402// }
403// }
66f93042 404
ce3f5e87 405// // fill digits
406// pMUON->AddDigits(ich,tracks,charges,fDigits);
407// }
408// gAlice->TreeD()->Fill();
409// pMUON->ResetDigits();
410// fList->Delete();
66f93042 411
412
ce3f5e87 413// for(Int_t ii = 0; ii < AliMUONConstants::NCh(); ++ii) {
414// if (fHitMap[ii]) {
415// delete fHitMap[ii];
416// fHitMap[ii] = 0;
417// }
418// }
419// } //end loop over cathodes
420// delete [] fHitMap;
421// delete fList;
66f93042 422
77ccf5b0 423// no need to delete ... and it makes a crash also
424// if (fHitsBgr) fHitsBgr->Delete();
425// if (fPadHitsBgr) fPadHitsBgr->Delete();
66f93042 426 // gObjectTable->Print();
427}
428
429
430
a3f472a9 431void AliMUONMerger::SortTracks(Int_t *tracks,Int_t *charges,Int_t ntr)
432{
433 //
434 // Sort the list of tracks contributing to a given digit
435 // Only the 3 most significant tracks are acctually sorted
436 //
437
438 //
439 // Loop over signals, only 3 times
440 //
441
442 Int_t qmax;
443 Int_t jmax;
444 Int_t idx[3] = {-2,-2,-2};
445 Int_t jch[3] = {-2,-2,-2};
446 Int_t jtr[3] = {-2,-2,-2};
447 Int_t i,j,imax;
448
449 if (ntr<3) imax=ntr;
450 else imax=3;
451 for(i=0;i<imax;i++){
452 qmax=0;
453 jmax=0;
454
455 for(j=0;j<ntr;j++){
456
457 if((i == 1 && j == idx[i-1])
458 ||(i == 2 && (j == idx[i-1] || j == idx[i-2]))) continue;
459
460 if(charges[j] > qmax) {
461 qmax = charges[j];
462 jmax=j;
463 }
464 }
465
466 if(qmax > 0) {
467 idx[i]=jmax;
468 jch[i]=charges[jmax];
469 jtr[i]=tracks[jmax];
470 }
471
472 }
473
474 for(i=0;i<3;i++){
475 if (jtr[i] == -2) {
476 charges[i]=0;
477 tracks[i]=0;
478 } else {
479 charges[i]=jch[i];
480 tracks[i]=jtr[i];
481 }
482 }
483}
484
66f93042 485
486