Conding conventions violation and Doxygen comments (Philippe Pillot)
[u/mrichter/AliRoot.git] / MUON / AliMUONEventRecoCombi.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, 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
16 /* $Id$ */
17
18 // -------------------------------------
19 // Class AliMUONEventRecoCombi
20 // -------------------------------------
21 // Steering class for the combined cluster / track reconstructor
22 // Author: Alexander Zinchenko, JINR Dubna
23
24 #include "AliMUONEventRecoCombi.h"
25
26 #include "AliMUONData.h"
27 #include "AliMUONDetElement.h"
28 #include "AliMUONDigit.h"
29 #include "AliMUONHitForRec.h"
30 #include "AliMUONRawCluster.h"
31 #include "AliMUONTrackK.h"
32 #include "AliMUONTrackReconstructorK.h"
33 #include "AliMUONConstants.h"
34
35 #include "AliLoader.h"
36
37 #include <Riostream.h>
38 #include <TClonesArray.h>
39 #include <TArrayS.h>
40 #include <TArrayD.h>
41 #include "AliLog.h"
42
43 AliMUONEventRecoCombi* AliMUONEventRecoCombi::fgRecoCombi = 0; 
44
45 ClassImp(AliMUONEventRecoCombi)
46
47 //_________________________________________________________________________
48   AliMUONEventRecoCombi::AliMUONEventRecoCombi() 
49     : TObject(),
50       fDetElems(0x0),
51       fZ(0x0),
52       fNZ(0),
53       fDEvsZ(0x0)
54 {
55   // Ctor
56
57   fDetElems = new TClonesArray("AliMUONDetElement", 20);
58   fZ = new TArrayD(20);
59 }
60
61 //_________________________________________________________________________
62 AliMUONEventRecoCombi* AliMUONEventRecoCombi::Instance()
63 {
64 // return pointer to the singleton instance
65
66   if (fgRecoCombi == 0) {
67     fgRecoCombi = new AliMUONEventRecoCombi();
68   }
69   //fDetElems = new TClonesArray("AliMUONDetElement", 20);
70   //fZ = new TArrayD(20);
71   //fNZ = 0;
72   return fgRecoCombi;
73 }
74
75 //_________________________________________________________________________
76 AliMUONEventRecoCombi::~AliMUONEventRecoCombi()
77 {
78   // Destructor
79   delete fDetElems;
80   delete fZ;
81   delete [] fDEvsZ;
82 }
83
84 //_________________________________________________________________________
85 void AliMUONEventRecoCombi::FillEvent(AliMUONData *data, AliMUONClusterFinderAZ *recModel)
86 {
87   // Fill event information
88
89   // Clear previous event
90   fDetElems->Delete();
91   for (Int_t i = 0; i < fNZ; i++) delete [] fDEvsZ[i];
92   delete [] fDEvsZ; fDEvsZ = NULL;
93   fNZ = -1;
94
95   Int_t nDetElem = 0;
96   for (Int_t ich = 0; ich < 6; ich++) {
97     // loop over chambers 0-5
98     TClonesArray *digs = data->Digits(ich);
99     //cout << ich << " " << digs << " " << digs->GetEntriesFast() << endl;
100     Int_t idDE = -1;
101     for (Int_t i = 0; i < digs->GetEntriesFast(); i++) {
102       AliMUONDigit *dig = (AliMUONDigit*) digs->UncheckedAt(i);
103       if (dig->DetElemId() != idDE) {
104         idDE = dig->DetElemId();
105         new ((*fDetElems)[nDetElem++]) AliMUONDetElement(idDE, dig, recModel);
106       }
107       else ((AliMUONDetElement*)fDetElems->UncheckedAt(nDetElem-1))->AddDigit(dig);
108     }
109   }
110
111   // Sort according to Z
112   fDetElems->Sort();
113   //cout << nDetElem << endl;
114   // Fill det. elems. position index in the container
115   for (Int_t i = 0; i < nDetElem; i++) 
116     ((AliMUONDetElement*)fDetElems->UncheckedAt(i))->SetIndex(i);
117
118   // Find groups of det. elements with the same Z
119   Double_t z0 = -99999;
120   TArrayS *nPerZ = new TArrayS(20);
121   for (Int_t i = 0; i < nDetElem; i++) {
122     AliMUONDetElement *detElem = (AliMUONDetElement*) fDetElems->UncheckedAt(i);
123     detElem->Fill(data);
124     //cout << i << " " << detElem->Z() << endl;
125     if (detElem->Z() - z0 < 0.5) { 
126       // the same Z
127       (*nPerZ)[fNZ]++;
128     } else {
129       if (fZ->GetSize() <= fNZ) fZ->Set(fZ->GetSize()+10);
130       if (nPerZ->GetSize() <= fNZ) nPerZ->Set(nPerZ->GetSize()+10);
131       (*fZ)[++fNZ] = detElem->Z();
132       z0 = detElem->Z();
133       (*nPerZ)[fNZ]++;
134     }
135   }
136   fNZ++;
137   /*
138   cout << fNZ << endl;
139   for (Int_t i = 0; i < 7; i++) {
140     cout << i << " " << data->RawClusters(i)->GetEntriesFast() << endl;
141   }
142   */
143
144   // Build list of DE locations vs Z
145   fDEvsZ = new Int_t* [fNZ];
146   Int_t iPos = 0;
147   for (Int_t i = 0; i < fNZ; i++) {
148     Int_t *idPerZ = new Int_t[(*nPerZ)[i]+1]; 
149     for (Int_t j = 1; j < (*nPerZ)[i]+1; j++) idPerZ[j] = iPos++;
150     idPerZ[0] = (*nPerZ)[i]; // number of DE's as first element of array
151     fDEvsZ[i] = idPerZ;
152     //cout << (*nPerZ)[i] << " ";
153   }
154   //cout << endl;
155   delete nPerZ;
156
157   // Fill rec. point container for stations 4 and 5
158   //cout << data->TreeR() << endl;
159   //data->MakeBranch("RC");
160   data->SetTreeAddress("RCC");
161   for (Int_t ch = 6; ch < 10; ch++) {
162     TClonesArray *raw = data->RawClusters(ch);
163     //cout << raw->GetEntriesFast() << " " << data->RawClusters(ch) << endl;
164     for (Int_t i = 0; i < raw->GetEntriesFast(); i++) {
165       AliMUONRawCluster *clus = (AliMUONRawCluster*) raw->UncheckedAt(i);
166       data->AddRawCluster(ch, *clus);
167     }
168   }
169   //data->SetTreeAddress("RC");
170 }
171
172 //_________________________________________________________________________
173 void AliMUONEventRecoCombi::FillRecP(AliMUONData *dataCluster, AliMUONTrackReconstructorK *recoTrack) const
174 {
175   // Fill rec. points used for tracking from det. elems
176
177   TClonesArray *tracks = recoTrack->GetRecTracksPtr();
178   for (Int_t i = 0; i < recoTrack->GetNRecTracks(); i++) {
179     AliMUONTrackK *track = (AliMUONTrackK*) tracks->UncheckedAt(i);
180     TObjArray *hits = track->GetTrackHits();
181     for (Int_t j = 0; j < track->GetNTrackHits(); j++) {
182       AliMUONHitForRec *hit = (AliMUONHitForRec*) hits->UncheckedAt(j);
183       if (hit->GetHitNumber() >= 0) continue;
184       // Combined cluster / track finder
185       Int_t index = -hit->GetHitNumber() / 100000;
186       Int_t iPos = -hit->GetHitNumber() - index * 100000;
187       AliMUONRawCluster *clus = (AliMUONRawCluster*) DetElem(index-1)->RawClusters()->UncheckedAt(iPos);
188       //cout << j << " " << iPos << " " << clus << " " << index << " " << DetElem(index-1)->Chamber() << endl; 
189       dataCluster->AddRawCluster(DetElem(index-1)->Chamber(), *clus);
190     }
191   }
192   /*
193   for (Int_t ch = 0; ch < 10; ch++) {
194     TClonesArray *raw = dataCluster->RawClusters(ch);
195     cout << ch << " " << raw->GetEntriesFast() << endl;
196   }
197   */
198   // Reset raw cluster tree
199   /*
200   char branchname[30];
201   TBranch * branch = 0x0;
202   if ( dataCluster->TreeR()) {
203     if ( dataCluster->IsTriggerBranchesInTree() ) {
204       // Branch per branch filling
205       for (int i=0; i<AliMUONConstants::NTrackingCh(); i++) {
206         sprintf(branchname,"%sRawClusters%d",dataCluster->GetName(),i+1);
207         branch = dataCluster->TreeR()->GetBranch(branchname);
208         //branch->Fill();
209         //branch->Reset();
210         //branch->Clear();
211         branch->Delete();
212       }
213     }
214     //else  TreeR()->Fill();
215     else  dataCluster->TreeR()->Reset();
216   }
217   */
218 }
219
220 //_________________________________________________________________________
221 Int_t AliMUONEventRecoCombi::IZfromHit(AliMUONHitForRec *hit) const
222 {
223   // Get Iz of det. elem. from the hit
224
225   Int_t index = -hit->GetHitNumber() / 100000 - 1, iz0 = -1;
226   for (Int_t iz = 0; iz < fNZ; iz++) {
227     Int_t *pDEatZ = DEatZ(iz);
228     Int_t nDetElem = pDEatZ[-1];
229     for (Int_t j = 0; j < nDetElem; j++) {
230       if (pDEatZ[j] != index) continue;
231       iz0 = iz;
232       break;
233     }
234     if (iz0 >= 0) break;
235   }
236   return iz0;
237 }