1 /**************************************************************************
2 * Copyright(c) 1998-1999, 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 // -------------------------------------
19 // Class AliMUONEventRecoCombi
20 // -------------------------------------
21 // Steering class for the combined cluster / track reconstructor
22 // Author: Alexander Zinchenko, JINR Dubna
24 #include "AliMUONEventRecoCombi.h"
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"
35 #include "AliLoader.h"
38 #include <Riostream.h>
39 #include <TClonesArray.h>
44 ClassImp(AliMUONEventRecoCombi)
47 AliMUONEventRecoCombi* AliMUONEventRecoCombi::fgRecoCombi = 0;
49 //_________________________________________________________________________
50 AliMUONEventRecoCombi::AliMUONEventRecoCombi(AliMUONSegmentation* segmentation)
52 fSegmentation(segmentation),
60 fDetElems = new TClonesArray("AliMUONDetElement", 20);
63 //_________________________________________________________________________
64 AliMUONEventRecoCombi* AliMUONEventRecoCombi::Instance(AliMUONSegmentation* segmentation)
66 /// Return pointer to the singleton instance
68 if (fgRecoCombi == 0) {
69 fgRecoCombi = new AliMUONEventRecoCombi(segmentation);
74 //_________________________________________________________________________
75 AliMUONEventRecoCombi::~AliMUONEventRecoCombi()
83 //_________________________________________________________________________
84 void AliMUONEventRecoCombi::FillEvent(AliMUONData *data, AliMUONClusterFinderAZ *recModel)
86 /// Fill event information
88 // Clear previous event
90 for (Int_t i = 0; i < fNZ; i++) delete [] fDEvsZ[i];
91 delete [] fDEvsZ; fDEvsZ = NULL;
95 for (Int_t ich = 0; ich < 6; ich++) {
96 // loop over chambers 0-5
97 TClonesArray *digs = data->Digits(ich);
99 //cout << ich << " " << digs << " " << digs->GetEntriesFast() << endl;
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, fSegmentation);
107 else ((AliMUONDetElement*)fDetElems->UncheckedAt(nDetElem-1))->AddDigit(dig);
111 // Compute average Z-position
112 for (Int_t i = 0; i < nDetElem; i++) {
113 AliMUONDetElement *detElem = (AliMUONDetElement*) fDetElems->UncheckedAt(i);
114 Int_t nDigs = detElem->Digits(0)->GetEntriesFast() +
115 detElem->Digits(1)->GetEntriesFast();
116 detElem->SetZ(detElem->Z() / nDigs);
119 // Sort according to Z
121 //cout << nDetElem << endl;
122 // Fill det. elems. position index in the container
123 for (Int_t i = 0; i < nDetElem; i++)
124 ((AliMUONDetElement*)fDetElems->UncheckedAt(i))->SetIndex(i);
126 // Find groups of det. elements with the same Z
127 Double_t z0 = -99999;
128 TArrayS *nPerZ = new TArrayS(20);
129 for (Int_t i = 0; i < nDetElem; i++) {
130 AliMUONDetElement *detElem = (AliMUONDetElement*) fDetElems->UncheckedAt(i);
132 //cout << i << " " << detElem->IdDE() << " " << detElem->Z() << endl;
133 if (detElem->Z() - z0 < 0.05) {
137 if (fZ->GetSize() <= fNZ+1) fZ->Set(fZ->GetSize()+10);
138 if (nPerZ->GetSize() <= fNZ+1) nPerZ->Set(nPerZ->GetSize()+10);
139 (*fZ)[++fNZ] = detElem->Z();
147 for (Int_t i = 0; i < 7; i++) {
148 cout << i << " " << data->RawClusters(i)->GetEntriesFast() << endl;
152 // Build list of DE locations vs Z
153 fDEvsZ = new Int_t* [fNZ];
155 for (Int_t i = 0; i < fNZ; i++) {
156 Int_t *idPerZ = new Int_t[(*nPerZ)[i]+1];
157 for (Int_t j = 1; j < (*nPerZ)[i]+1; j++) idPerZ[j] = iPos++;
158 idPerZ[0] = (*nPerZ)[i]; // number of DE's as first element of array
160 //cout << (*nPerZ)[i] << " ";
165 // Fill rec. point container for stations 4 and 5
166 //cout << data->TreeR() << endl;
167 //data->MakeBranch("RC");
169 data->SetTreeAddress("RCC");
170 for (Int_t ch = 6; ch < 10; ch++) {
171 TClonesArray *raw = data->RawClusters(ch);
172 cout << ch << " " << raw->GetEntriesFast() << " " << data->RawClusters(ch) << endl;
173 //for (Int_t i = 0; i < raw->GetEntriesFast(); i++) {
174 for (Int_t i = 0; i < TMath::Min(raw->GetEntriesFast(),1000); i++) {
175 AliMUONRawCluster *clus = (AliMUONRawCluster*) raw->UncheckedAt(i);
176 data->AddRawCluster(ch, *clus);
177 cout << i << " " << raw->GetEntriesFast() << endl;
181 //data->SetTreeAddress("RC");
184 //_________________________________________________________________________
185 void AliMUONEventRecoCombi::FillRecP(AliMUONData *dataCluster, AliMUONTrackReconstructorK *recoTrack) const
187 /// Fill rec. points used for tracking from det. elems
189 TClonesArray *tracks = recoTrack->GetRecTracksPtr();
190 for (Int_t i = 0; i < recoTrack->GetNRecTracks(); i++) {
191 AliMUONTrackK *track = (AliMUONTrackK*) tracks->UncheckedAt(i);
192 TObjArray *hits = track->GetTrackHits();
193 for (Int_t j = 0; j < track->GetNTrackHits(); j++) {
194 AliMUONHitForRec *hit = (AliMUONHitForRec*) hits->UncheckedAt(j);
195 if (hit->GetHitNumber() >= 0) continue;
196 // Combined cluster / track finder
197 Int_t index = -hit->GetHitNumber() / 100000;
198 Int_t iPos = -hit->GetHitNumber() - index * 100000;
199 AliMUONRawCluster *clus = (AliMUONRawCluster*) DetElem(index-1)->RawClusters()->UncheckedAt(iPos);
200 //cout << j << " " << iPos << " " << clus << " " << index << " " << DetElem(index-1)->Chamber() << endl;
201 dataCluster->AddRawCluster(DetElem(index-1)->Chamber(), *clus);
205 for (Int_t ch = 0; ch < 10; ch++) {
206 TClonesArray *raw = dataCluster->RawClusters(ch);
207 cout << ch << " " << raw->GetEntriesFast() << endl;
210 // Reset raw cluster tree
213 TBranch * branch = 0x0;
214 if ( dataCluster->TreeR()) {
215 if ( dataCluster->IsTriggerBranchesInTree() ) {
216 // Branch per branch filling
217 for (int i=0; i<AliMUONConstants::NTrackingCh(); i++) {
218 sprintf(branchname,"%sRawClusters%d",dataCluster->GetName(),i+1);
219 branch = dataCluster->TreeR()->GetBranch(branchname);
226 //else TreeR()->Fill();
227 else dataCluster->TreeR()->Reset();
232 //_________________________________________________________________________
233 Int_t AliMUONEventRecoCombi::IZfromHit(AliMUONHitForRec *hit) const
235 /// Get Iz of det. elem. from the hit
237 Int_t index = -hit->GetHitNumber() / 100000 - 1, iz0 = -1;
238 for (Int_t iz = 0; iz < fNZ; iz++) {
239 Int_t *pDEatZ = DEatZ(iz);
240 Int_t nDetElem = pDEatZ[-1];
241 for (Int_t j = 0; j < nDetElem; j++) {
242 if (pDEatZ[j] != index) continue;