]> git.uio.no Git - u/mrichter/AliRoot.git/blame - MFT/AliMFTClusterFinder.cxx
Restoring the functionallity of the PAR files: partial fix
[u/mrichter/AliRoot.git] / MFT / AliMFTClusterFinder.cxx
CommitLineData
820b4d9e 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//====================================================================================================================================================
17//
18// Class for finding and building the clusters of the ALICE Muon Forward Tracker
19//
20// Contact author: antonio.uras@cern.ch
21//
22//====================================================================================================================================================
23
24#include "AliLog.h"
25#include "TObjArray.h"
26#include "TClonesArray.h"
27#include "AliMFTDigit.h"
28#include "AliMFTCluster.h"
29#include "AliMFTSegmentation.h"
30#include "TTree.h"
31#include "TMath.h"
d4643a10 32#include "AliMFTConstants.h"
820b4d9e 33#include "AliMFTClusterFinder.h"
34
d4643a10 35const Double_t AliMFTClusterFinder::fCutForAvailableDigits = AliMFTConstants::fCutForAvailableDigits;
36const Double_t AliMFTClusterFinder::fCutForAttachingDigits = AliMFTConstants::fCutForAttachingDigits;
37
38
820b4d9e 39ClassImp(AliMFTClusterFinder)
40
41//====================================================================================================================================================
42
43AliMFTClusterFinder::AliMFTClusterFinder() :
44 TObject(),
45 fDigitsInCluster(0),
d4643a10 46 fCurrentDigit(0),
47 fCurrentCluster(0),
820b4d9e 48 fSegmentation(0),
49 fNPlanes(0)
50{
51
52 // Default constructor
53
54 for (Int_t iPlane=0; iPlane<fNMaxPlanes; iPlane++) fClustersPerPlane[iPlane] = NULL;
55 fDigitsInCluster = new TClonesArray("AliMFTDigit", fNMaxDigitsPerCluster);
56
57}
58
59//====================================================================================================================================================
60
61AliMFTClusterFinder::~AliMFTClusterFinder() {
62
63 AliDebug(1, "Deleting AliMFTClusterFinder...");
64
65 for (Int_t iPlane=0; iPlane<fNMaxPlanes; iPlane++) delete fClustersPerPlane[iPlane];
66
67 AliDebug(1, "... done!");
68
69}
70
71//====================================================================================================================================================
72
9cc245f6 73void AliMFTClusterFinder::Init(const Char_t *nameGeomFile) {
820b4d9e 74
75 fSegmentation = new AliMFTSegmentation(nameGeomFile);
76 fNPlanes = fSegmentation -> GetNPlanes();
77
78}
79
80//====================================================================================================================================================
81
82void AliMFTClusterFinder::StartEvent() {
83
84 // Cleaning up and preparation for the clustering procedure
85
86 AliDebug(1, "Starting Event...");
87
88 for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
89 fClustersPerPlane[iPlane]->Clear();
90 }
91
92 AliDebug(1, "... done!");
93
94}
95
96//====================================================================================================================================================
97
98void AliMFTClusterFinder::DigitsToClusters(const TObjArray *pDigitList) {
99
100 // where the clusterization is performed
101
102 AliInfo("Starting Clusterization for MFT");
103 AliDebug(1, Form("nPlanes = %d", fNPlanes));
104
105 StartEvent();
d4643a10 106 Bool_t isDigAvailableForNewCluster = kTRUE;
820b4d9e 107
108 for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
109
110 AliDebug(1, Form("Plane %02d", iPlane));
111
112 TClonesArray *myDigitList = (TClonesArray*) pDigitList->At(iPlane);
820b4d9e 113
114 AliDebug(1, Form("myDigitList->GetEntries() = %d", myDigitList->GetEntries()));
115
d4643a10 116 Int_t cycleOverDigits = 0;
117 Double_t myCutForAvailableDigits = fCutForAvailableDigits;
7230691d 118
d4643a10 119 while (myDigitList->GetEntries()) {
7230691d 120
d4643a10 121 for (Int_t iDig=0; iDig<myDigitList->GetEntries(); iDig++) {
7230691d 122
d4643a10 123 AliDebug(1, Form("Check %d: Digit %5d of %5d", cycleOverDigits, iDig, myDigitList->GetEntries()));
7230691d 124
d4643a10 125 fCurrentDigit = (AliMFTDigit*) myDigitList->At(iDig);
126 isDigAvailableForNewCluster = kTRUE;
7230691d 127
d4643a10 128 for (Int_t iCluster=0; iCluster<fClustersPerPlane[iPlane]->GetEntries(); iCluster++) {
129 fCurrentCluster = (AliMFTCluster*) fClustersPerPlane[iPlane]->At(iCluster);
130 AliDebug(2, Form("Distance between cluster and digit = %f",fCurrentCluster->GetDistanceFromPixel(fCurrentDigit)));
131 if (fCurrentCluster->GetDistanceFromPixel(fCurrentDigit) < fCutForAttachingDigits) {
132 fCurrentCluster->AddPixel(fCurrentDigit);
133 myDigitList->Remove(fCurrentDigit);
134 myDigitList->Compress();
135 iDig--;
136 isDigAvailableForNewCluster = kFALSE;
137 break;
138 }
139 if (fCurrentCluster->GetDistanceFromPixel(fCurrentDigit) < myCutForAvailableDigits) isDigAvailableForNewCluster=kFALSE;
140 }
7230691d 141
d4643a10 142 if (isDigAvailableForNewCluster) {
143 AliMFTCluster *newCluster = new AliMFTCluster();
144 newCluster->AddPixel(fCurrentDigit);
145 myDigitList->Remove(fCurrentDigit);
146 myDigitList->Compress();
147 iDig--;
148 new ((*fClustersPerPlane[iPlane])[fClustersPerPlane[iPlane]->GetEntries()]) AliMFTCluster(*newCluster);
149 }
7230691d 150
d4643a10 151 } // end of cycle over the digits
7230691d 152
d4643a10 153 if (cycleOverDigits) myCutForAvailableDigits -= 0.5;
154 cycleOverDigits++;
7230691d 155
d4643a10 156 } // no more digits to check in current plane!
7230691d 157
d4643a10 158 for (Int_t iCluster=0; iCluster<fClustersPerPlane[iPlane]->GetEntries(); iCluster++) {
159 fCurrentCluster = (AliMFTCluster*) fClustersPerPlane[iPlane]->At(iCluster);
160 fCurrentCluster -> TerminateCluster();
161 }
7230691d 162
d4643a10 163 AliDebug(1, Form("Found %d clusters in plane %02d", fClustersPerPlane[iPlane]->GetEntries(), iPlane));
820b4d9e 164
d4643a10 165 } // end of cycle over the planes
820b4d9e 166
167}
168
169//====================================================================================================================================================
170
171void AliMFTClusterFinder::MakeClusterBranch(TTree *treeCluster) {
172
173 // Creating the cluster branches, one for each plane (see AliMFTReconstructor::Reconstruct)
174
175 AliDebug(1, "Making Cluster Branch");
176
177 CreateClusters();
178
179 if (treeCluster) {
180 for(Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
181 AliDebug(1, Form("Setting Branch Plane_%02d for treeCluster",iPlane));
d6080682 182 if (treeCluster->GetBranch(Form("Plane_%02d",iPlane))) continue;
820b4d9e 183 AliDebug(1, Form("Branch Plane_%02d does not exist, creating!",iPlane));
d6080682 184 treeCluster->Branch(Form("Plane_%02d",iPlane), &(fClustersPerPlane[iPlane]));
820b4d9e 185 }
186 }
187
188}
189
190//====================================================================================================================================================
191
192void AliMFTClusterFinder::SetClusterTreeAddress(TTree *treeCluster) {
193
194 // Addressing the cluster branches, one for each plane (see AliMFTReconstructor::Reconstruct)
195
196 if (treeCluster && treeCluster->GetBranch("Plane_00")) {
197 CreateClusters();
198 for(Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
199 if (treeCluster->GetBranch(Form("Plane_%02d",iPlane))) {
200 treeCluster->SetBranchAddress(Form("Plane_%02d",iPlane), &(fClustersPerPlane[iPlane]));
201 }
202 else AliError(Form("No branch available with name Plane_%02d", iPlane));
203 }
204 }
205
206}
207
208//====================================================================================================================================================
209
210void AliMFTClusterFinder::CreateClusters() {
211
212 // create cluster list
213
214 AliDebug(1, Form("Creating clusters list: nPlanes = %d",fNPlanes));
215
216 if (fClustersPerPlane[0]) return;
217
218 for(Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
219 AliDebug(1, Form("plane %02d", iPlane));
220 fClustersPerPlane[iPlane] = new TClonesArray("AliMFTCluster");
221 }
222
223}
224
225//====================================================================================================================================================