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);
174 Int_t nmodules = (Int_t) branch->GetEntries();
176 AliDebug(1,Form("Number of modules filled in treeR = %d",nmodules));
177 for (Int_t imodule = 0; imodule < nmodules; imodule++)
179 branch->GetEntry(imodule);
180 Int_t nentries = fRecpoints->GetLast();
181 AliDebug(2,Form("Number of clusters per modules filled in treeR = %d"
186 for(Int_t ient = 0; ient < nentries+1; ient++)
188 fPMDrecpoint = (AliPMDrecpoint1*)fRecpoints->UncheckedAt(ient);
189 idet = fPMDrecpoint->GetDetector();
190 ismn = fPMDrecpoint->GetSMNumber();
191 clusdata[0] = fPMDrecpoint->GetClusX();
192 clusdata[1] = fPMDrecpoint->GetClusY();
193 clusdata[2] = fPMDrecpoint->GetClusADC();
194 clusdata[3] = fPMDrecpoint->GetClusCells();
195 clusdata[4] = fPMDrecpoint->GetClusSigmaX();
196 clusdata[5] = fPMDrecpoint->GetClusSigmaY();
198 if (clusdata[4] != -99. && clusdata[5] != -99.)
200 // extract the associated cell information
201 branch1->GetEntry(ncrhit);
202 Int_t nenbr1 = fRechits->GetLast() + 1;
204 irow = new Int_t[nenbr1];
205 icol = new Int_t[nenbr1];
206 itra = new Int_t[nenbr1];
207 ipid = new Int_t[nenbr1];
208 cadc = new Float_t[nenbr1];
210 for (Int_t ient1 = 0; ient1 < nenbr1; ient1++)
212 rechit = (AliPMDrechit*)fRechits->UncheckedAt(ient1);
213 //irow[ient1] = rechit->GetCellX();
214 //icol[ient1] = rechit->GetCellY();
215 itra[ient1] = rechit->GetCellTrack();
216 ipid[ient1] = rechit->GetCellPid();
217 cadc[ient1] = rechit->GetCellAdc();
219 AssignTrPidToCluster(nenbr1, itra, ipid, cadc,
228 fPMDclin = new AliPMDrecdata(idet,ismn,trackno,trackpid,clusdata);
229 fPMDcontin->Add(fPMDclin);
236 AliPMDDiscriminator *pmddiscriminator = new AliPMDEmpDiscriminator();
237 pmddiscriminator->Discrimination(fPMDcontin,fPMDcontout);
239 const Float_t kzpos = 361.5; // middle of the PMD
241 Int_t det,smn,trno,trpid,mstat;
243 Float_t adc, ncell, radx, rady;
244 Float_t xglobal = 0., yglobal = 0., zglobal = 0;
248 Int_t nentries2 = fPMDcontout->GetEntries();
249 AliDebug(1,Form("Number of clusters coming after discrimination = %d"
251 for (Int_t ient1 = 0; ient1 < nentries2; ient1++)
253 fPMDclout = (AliPMDclupid*)fPMDcontout->UncheckedAt(ient1);
255 det = fPMDclout->GetDetector();
256 smn = fPMDclout->GetSMN();
257 trno = fPMDclout->GetClusTrackNo();
258 trpid = fPMDclout->GetClusTrackPid();
259 mstat = fPMDclout->GetClusMatching();
260 xpos = fPMDclout->GetClusX();
261 ypos = fPMDclout->GetClusY();
262 adc = fPMDclout->GetClusADC();
263 ncell = fPMDclout->GetClusCells();
264 radx = fPMDclout->GetClusSigmaX();
265 rady = fPMDclout->GetClusSigmaY();
266 pid = fPMDclout->GetClusPID();
269 /**********************************************************************
270 * det : Detector, 0: PRE & 1:CPV *
271 * smn : Serial Module Number 0 to 23 for each plane *
272 * xpos : x-position of the cluster *
273 * ypos : y-position of the cluster *
274 * THESE xpos & ypos are not the true xpos and ypos *
275 * for some of the unit modules. They are rotated. *
276 * adc : ADC contained in the cluster *
277 * ncell : Number of cells contained in the cluster *
278 * rad : radius of the cluster (1d fit) *
279 **********************************************************************/
282 fPMDutil->RectGeomCellPos(smn,xpos,ypos,xglobal,yglobal);
286 zglobal = kzpos + 1.6; // PREshower plane
290 zglobal = kzpos - 1.7; // CPV plane
295 AliESDPmdTrack *esdpmdtr = new AliESDPmdTrack();
297 esdpmdtr->SetDetector(det);
298 esdpmdtr->SetSmn(smn);
299 esdpmdtr->SetClusterTrackNo(trno);
300 esdpmdtr->SetClusterTrackPid(trpid);
301 esdpmdtr->SetClusterMatching(mstat);
303 esdpmdtr->SetClusterX(xglobal);
304 esdpmdtr->SetClusterY(yglobal);
305 esdpmdtr->SetClusterZ(zglobal);
306 esdpmdtr->SetClusterADC(adc);
307 esdpmdtr->SetClusterCells(ncell);
308 esdpmdtr->SetClusterPID(pid);
309 esdpmdtr->SetClusterSigmaX(radx);
310 esdpmdtr->SetClusterSigmaY(rady);
312 event->AddPmdTrack(esdpmdtr);
315 fPMDcontin->Delete();
316 fPMDcontout->Delete();
319 //--------------------------------------------------------------------//
320 void AliPMDtracker::AssignTrPidToCluster(Int_t nentry, Int_t *itra,
321 Int_t *ipid, Float_t *cadc,
322 Int_t &trackno, Int_t &trackpid)
324 // assign the track number and the corresponding pid to a cluster
325 // split cluster part will be done at the time of calculating eff/pur
327 Int_t *phentry = new Int_t [nentry];
328 Int_t *trenergy = 0x0;
329 Int_t *sortcoord = 0x0;
332 for (Int_t i = 0; i < nentry; i++)
337 phentry[ngtrack] = i;
345 // no need of track number, set to -1
349 else if (ngtrack == 1)
351 // only one photon track
352 // track number set to photon track
354 trackno = itra[phentry[0]];
356 else if (ngtrack > 1)
358 // more than one photon track
360 trenergy = new Int_t [ngtrack];
361 sortcoord = new Int_t [ngtrack];
362 for (Int_t i = 0; i < ngtrack; i++)
365 for (Int_t j = 0; j < nentry; j++)
367 if (ipid[j] == 22 && itra[j] == itra[phentry[i]])
369 trenergy[i] += cadc[j];
375 TMath::Sort(ngtrack,trenergy,sortcoord,jsort);
377 Int_t gtr = sortcoord[0];
378 trackno = itra[phentry[gtr]]; // highest adc track
384 } // end of ngtrack > 1
389 //--------------------------------------------------------------------//
390 void AliPMDtracker::SetVertex(Double_t vtx[3], Double_t evtx[3])
399 //--------------------------------------------------------------------//
400 void AliPMDtracker::ResetClusters()
402 if (fRecpoints) fRecpoints->Clear();
404 //--------------------------------------------------------------------//