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 **************************************************************************/
16 //-----------------------------------------------------//
18 // Date : March 25 2004 //
19 // This reads the file PMD.RecPoints.root(TreeR), //
20 // calls the Clustering algorithm and stores the //
21 // clustering output in PMD.RecPoints.root(TreeR) //
23 //-----------------------------------------------------//
25 #include <Riostream.h>
28 #include <TObjArray.h>
29 #include <TClonesArray.h>
33 #include <TParticle.h>
35 #include "AliPMDcluster.h"
36 #include "AliPMDclupid.h"
37 #include "AliPMDrecpoint1.h"
38 #include "AliPMDrecdata.h"
39 #include "AliPMDrechit.h"
40 #include "AliPMDUtility.h"
41 #include "AliPMDDiscriminator.h"
42 #include "AliPMDEmpDiscriminator.h"
43 #include "AliPMDtracker.h"
45 #include "AliESDPmdTrack.h"
46 #include "AliESDEvent.h"
49 ClassImp(AliPMDtracker)
51 AliPMDtracker::AliPMDtracker():
53 fRecpoints(new TClonesArray("AliPMDrecpoint1", 10)),
54 fRechits(new TClonesArray("AliPMDrechit", 10)),
55 fPMDcontin(new TObjArray()),
56 fPMDcontout(new TObjArray()),
57 fPMDutil(new AliPMDUtility()),
69 // Default Constructor
72 //--------------------------------------------------------------------//
73 AliPMDtracker:: AliPMDtracker(const AliPMDtracker & /* tracker */):
74 TObject(/* tracker */),
92 AliError("Copy constructor not allowed");
95 //--------------------------------------------------------------------//
96 AliPMDtracker& AliPMDtracker::operator=(const AliPMDtracker & /* tracker */)
98 // assignment operator
99 AliError("Assignment operator not allowed");
103 //--------------------------------------------------------------------//
104 AliPMDtracker::~AliPMDtracker()
118 fPMDcontin->Delete();
125 fPMDcontout->Delete();
132 //--------------------------------------------------------------------//
133 void AliPMDtracker::LoadClusters(TTree *treein)
135 // Load the Reconstructed tree
138 //--------------------------------------------------------------------//
139 void AliPMDtracker::Clusters2Tracks(AliESDEvent *event)
141 // Converts digits to recpoints after running clustering
142 // algorithm on CPV plane and PREshower plane
147 Int_t trackno, trackpid;
156 AliPMDrechit *rechit = 0x0;
158 TBranch *branch = fTreeR->GetBranch("PMDRecpoint");
161 AliError("PMDRecpoint branch not found");
164 branch->SetAddress(&fRecpoints);
166 TBranch *branch1 = fTreeR->GetBranch("PMDRechit");
169 AliError("PMDRechit branch not found");
172 branch1->SetAddress(&fRechits);
175 Int_t nmodules = (Int_t) branch->GetEntries();
177 AliDebug(1,Form("Number of modules filled in treeR = %d",nmodules));
178 for (Int_t imodule = 0; imodule < nmodules; imodule++)
180 branch->GetEntry(imodule);
181 Int_t nentries = fRecpoints->GetLast();
182 AliDebug(2,Form("Number of clusters per modules filled in treeR = %d"
185 for(Int_t ient = 0; ient < nentries+1; ient++)
187 fPMDrecpoint = (AliPMDrecpoint1*)fRecpoints->UncheckedAt(ient);
188 idet = fPMDrecpoint->GetDetector();
189 ismn = fPMDrecpoint->GetSMNumber();
190 clusdata[0] = fPMDrecpoint->GetClusX();
191 clusdata[1] = fPMDrecpoint->GetClusY();
192 clusdata[2] = fPMDrecpoint->GetClusADC();
193 clusdata[3] = fPMDrecpoint->GetClusCells();
194 clusdata[4] = fPMDrecpoint->GetClusSigmaX();
195 clusdata[5] = fPMDrecpoint->GetClusSigmaY();
197 if (clusdata[4] >= 0. && clusdata[5] >= 0.)
199 // extract the associated cell information
200 branch1->GetEntry(ncrhit);
201 Int_t nenbr1 = fRechits->GetLast() + 1;
203 irow = new Int_t[nenbr1];
204 icol = new Int_t[nenbr1];
205 itra = new Int_t[nenbr1];
206 ipid = new Int_t[nenbr1];
207 cadc = new Float_t[nenbr1];
209 for (Int_t ient1 = 0; ient1 < nenbr1; ient1++)
211 rechit = (AliPMDrechit*)fRechits->UncheckedAt(ient1);
212 //irow[ient1] = rechit->GetCellX();
213 //icol[ient1] = rechit->GetCellY();
214 itra[ient1] = rechit->GetCellTrack();
215 ipid[ient1] = rechit->GetCellPid();
216 cadc[ient1] = rechit->GetCellAdc();
220 AssignTrPidToCluster(nenbr1, itra, ipid, cadc,
235 fPMDclin = new AliPMDrecdata(idet,ismn,trackno,trackpid,clusdata);
236 fPMDcontin->Add(fPMDclin);
243 AliPMDDiscriminator *pmddiscriminator = new AliPMDEmpDiscriminator();
244 pmddiscriminator->Discrimination(fPMDcontin,fPMDcontout);
246 const Float_t kzpos = 361.5; // middle of the PMD
248 Int_t det,smn,trno,trpid,mstat;
250 Float_t adc, ncell, radx, rady;
251 Float_t xglobal = 0., yglobal = 0., zglobal = 0;
255 Int_t nentries2 = fPMDcontout->GetEntries();
256 AliDebug(1,Form("Number of clusters coming after discrimination = %d"
258 for (Int_t ient1 = 0; ient1 < nentries2; ient1++)
260 fPMDclout = (AliPMDclupid*)fPMDcontout->UncheckedAt(ient1);
262 det = fPMDclout->GetDetector();
263 smn = fPMDclout->GetSMN();
264 trno = fPMDclout->GetClusTrackNo();
265 trpid = fPMDclout->GetClusTrackPid();
266 mstat = fPMDclout->GetClusMatching();
267 xpos = fPMDclout->GetClusX();
268 ypos = fPMDclout->GetClusY();
269 adc = fPMDclout->GetClusADC();
270 ncell = fPMDclout->GetClusCells();
271 radx = fPMDclout->GetClusSigmaX();
272 rady = fPMDclout->GetClusSigmaY();
273 pid = fPMDclout->GetClusPID();
276 /**********************************************************************
277 * det : Detector, 0: PRE & 1:CPV *
278 * smn : Serial Module Number 0 to 23 for each plane *
279 * xpos : x-position of the cluster *
280 * ypos : y-position of the cluster *
281 * THESE xpos & ypos are not the true xpos and ypos *
282 * for some of the unit modules. They are rotated. *
283 * adc : ADC contained in the cluster *
284 * ncell : Number of cells contained in the cluster *
285 * rad : radius of the cluster (1d fit) *
286 **********************************************************************/
289 fPMDutil->RectGeomCellPos(smn,xpos,ypos,xglobal,yglobal);
293 zglobal = kzpos + 1.6; // PREshower plane
297 zglobal = kzpos - 1.7; // CPV plane
302 AliESDPmdTrack *esdpmdtr = new AliESDPmdTrack();
304 esdpmdtr->SetDetector(det);
305 esdpmdtr->SetSmn(smn);
306 esdpmdtr->SetClusterTrackNo(trno);
307 esdpmdtr->SetClusterTrackPid(trpid);
308 esdpmdtr->SetClusterMatching(mstat);
310 esdpmdtr->SetClusterX(xglobal);
311 esdpmdtr->SetClusterY(yglobal);
312 esdpmdtr->SetClusterZ(zglobal);
313 esdpmdtr->SetClusterADC(adc);
314 esdpmdtr->SetClusterCells(ncell);
315 esdpmdtr->SetClusterPID(pid);
316 esdpmdtr->SetClusterSigmaX(radx);
317 esdpmdtr->SetClusterSigmaY(rady);
319 event->AddPmdTrack(esdpmdtr);
322 fPMDcontin->Delete();
323 fPMDcontout->Delete();
326 //--------------------------------------------------------------------//
327 void AliPMDtracker::AssignTrPidToCluster(Int_t nentry, Int_t *itra,
328 Int_t *ipid, Float_t *cadc,
329 Int_t &trackno, Int_t &trackpid)
331 // assign the track number and the corresponding pid to a cluster
332 // split cluster part will be done at the time of calculating eff/pur
334 Int_t *phentry = new Int_t [nentry];
335 Int_t *hadentry = new Int_t [nentry];
336 Int_t *trenergy = 0x0;
338 Int_t *sortcoord = 0x0;
342 for (Int_t i = 0; i < nentry; i++)
349 phentry[ngtrack] = i;
352 else if (ipid[i] != 22)
354 hadentry[nhtrack] = i;
359 Int_t nghadtrack = ngtrack + nhtrack;
364 // no need of track number, set to -1
368 else if (ngtrack >= 1)
370 // one or more than one photon track + charged track
371 // find out which track deposits maximum energy and
372 // assign that track number and track pid
374 trenergy = new Int_t [nghadtrack];
375 trpid = new Int_t [nghadtrack];
376 sortcoord = new Int_t [nghadtrack];
377 for (Int_t i = 0; i < ngtrack; i++)
381 for (Int_t j = 0; j < nentry; j++)
383 if (ipid[j] == 22 && itra[j] == itra[phentry[i]])
385 trenergy[i] += cadc[j];
390 for (Int_t i = ngtrack; i < nghadtrack; i++)
394 for (Int_t j = 0; j < nentry; j++)
396 if (ipid[j] != 22 && itra[j] == itra[hadentry[i-ngtrack]])
398 trenergy[i] += cadc[j];
405 TMath::Sort(nghadtrack,trenergy,sortcoord,jsort);
407 Int_t gtr = sortcoord[0];
408 if (trpid[gtr] == 22)
411 trackno = itra[phentry[gtr]]; // highest adc track
423 } // end of ngtrack >= 1
426 //--------------------------------------------------------------------//
427 void AliPMDtracker::SetVertex(Double_t vtx[3], Double_t evtx[3])
436 //--------------------------------------------------------------------//
437 void AliPMDtracker::ResetClusters()
439 if (fRecpoints) fRecpoints->Clear();
441 //--------------------------------------------------------------------//