2 #include "AliMultiplicity.h"
4 #include "AliRefArray.h"
6 ClassImp(AliMultiplicity)
8 //______________________________________________________________________
9 AliMultiplicity::AliMultiplicity():
14 fDPhiWindow2(0.08*0.08),
15 fDThetaWindow2(0.025*0.025),
30 fFastOrFiredChips(1200),
31 fClusterFiredChips(1200)
33 // Default Constructor
36 for (int il=2;il--;) fSCl2Tracks[il] = fTCl2Tracks[il][0] = fTCl2Tracks[il][1] = 0;
37 for(Int_t ilayer = 0; ilayer < 6; ilayer++)fITSClusters[ilayer] = 0;
40 //______________________________________________________________________
41 AliMultiplicity::AliMultiplicity(Int_t ntr, Float_t *th, Float_t *ph, Float_t *dth, Float_t *dph, Int_t *labels, Int_t* labelsL2, Int_t ns, Float_t *ts, Float_t *ps, Int_t *labelss, Short_t nfcL1, Short_t nfcL2, const TBits & fFastOr):
46 fDPhiWindow2(0.08*0.08),
47 fDThetaWindow2(0.025*0.025),
62 fFastOrFiredChips(1200),
63 fClusterFiredChips(1200)
65 // Standard constructor
66 for (int il=2;il--;) fSCl2Tracks[il] = fTCl2Tracks[il][0] = fTCl2Tracks[il][1] = 0;
68 fLabels = new Int_t[ntr];
69 fLabelsL2 = new Int_t[ntr];
70 fTh = new Double_t [ntr];
71 fPhi = new Double_t [ntr];
72 fDeltTh = new Double_t [ntr];
73 fDeltPhi = new Double_t [ntr];
74 for(Int_t i=0;i<fNtracks;i++){
79 fLabels[i] = labels[i];
80 fLabelsL2[i] = labelsL2[i];
84 fThsingle = new Double_t [ns];
85 fPhisingle = new Double_t [ns];
86 fLabelssingle = new Int_t [ns];
87 for(Int_t i=0;i<fNsingle;i++){
90 fLabelssingle[i]=labelss[i];
93 fFiredChips[0] = nfcL1;
94 fFiredChips[1] = nfcL2;
95 fFastOrFiredChips = fFastOr;
96 for(Int_t ilayer = 0; ilayer < 6; ilayer++)fITSClusters[ilayer] = 0;
99 //______________________________________________________________________
100 AliMultiplicity::AliMultiplicity(Int_t ntr, Int_t ns, Short_t nfcL1, Short_t nfcL2, const TBits & fFastOr) :
105 fDPhiWindow2(0.08*0.08),
106 fDThetaWindow2(0.025*0.025),
121 fFastOrFiredChips(1200),
122 fClusterFiredChips(1200)
124 // Standard constructor to create the arrays w/o filling
125 for (int il=2;il--;) fSCl2Tracks[il] = fTCl2Tracks[il][0] = fTCl2Tracks[il][1] = 0;
127 fLabels = new Int_t[ntr];
128 fLabelsL2 = new Int_t[ntr];
129 fTh = new Double_t [ntr];
130 fPhi = new Double_t [ntr];
131 fDeltTh = new Double_t [ntr];
132 fDeltPhi = new Double_t [ntr];
133 for(Int_t i=fNtracks;i--;){
134 fTh[i]=fPhi[i]=fDeltTh[i]=fDeltPhi[i] = 0;
135 fLabels[i] = fLabelsL2[i] = 0;
139 fThsingle = new Double_t [ns];
140 fPhisingle = new Double_t [ns];
141 fLabelssingle = new Int_t [ns];
142 for(Int_t i=fNsingle;i--;) fThsingle[i] = fPhisingle[i] = fLabelssingle[i] = 0;
144 fFiredChips[0] = nfcL1;
145 fFiredChips[1] = nfcL2;
146 fFastOrFiredChips = fFastOr;
147 for(Int_t ilayer=6;ilayer--;) fITSClusters[ilayer] = 0;
150 //______________________________________________________________________
151 AliMultiplicity::AliMultiplicity(const AliMultiplicity& m):
153 fNtracks(m.fNtracks),
154 fNsingle(m.fNsingle),
156 fDPhiWindow2(0.08*0.08),
157 fDThetaWindow2(0.025*0.025),
172 fFastOrFiredChips(1200),
173 fClusterFiredChips(1200)
176 for (int il=2;il--;) fSCl2Tracks[il] = fTCl2Tracks[il][0] = fTCl2Tracks[il][1] = 0;
180 //______________________________________________________________________
181 AliMultiplicity &AliMultiplicity::operator=(const AliMultiplicity& m){
182 // assignment operator
183 if(this == &m)return *this;
184 ((TObject *)this)->operator=(m);
186 if(fTh)delete [] fTh;fTh = 0;
187 if(fPhi)delete [] fPhi;fPhi = 0;
188 if(fDeltTh)delete [] fDeltTh;fDeltTh= 0;
189 if(fDeltPhi)delete [] fDeltPhi;fDeltPhi = 0;
190 if(fLabels)delete [] fLabels;fLabels = 0;
191 if(fLabelsL2)delete [] fLabelsL2;fLabelsL2 = 0;
192 if(fThsingle)delete [] fThsingle;fThsingle = 0;
193 if(fPhisingle)delete [] fPhisingle;fPhisingle = 0;
194 if(fLabelssingle)delete [] fLabelssingle;fLabelssingle = 0;
195 if(fUsedClusS) delete[] fUsedClusS; fUsedClusS = 0;
196 if(fUsedClusT) delete[] fUsedClusT; fUsedClusT = 0;
197 for (int il=2;il--;) {
198 if (fSCl2Tracks[il]) delete fSCl2Tracks[il]; fSCl2Tracks[il] = 0;
199 if (fTCl2Tracks[il][0]) delete fTCl2Tracks[il][0]; fTCl2Tracks[il][0] = 0;
200 if (fTCl2Tracks[il][1]) delete fTCl2Tracks[il][1]; fTCl2Tracks[il][1] = 0;
207 void AliMultiplicity::Copy(TObject &obj) const {
209 // this overwrites the virtual TOBject::Copy()
210 // to allow run time copying without casting
213 if(this==&obj)return;
214 AliMultiplicity *robj = dynamic_cast<AliMultiplicity*>(&obj);
215 if(!robj)return; // not an AliMultiplicity
221 //______________________________________________________________________
222 void AliMultiplicity::Duplicate(const AliMultiplicity& m){
223 // used by copy constructor and assignment operator
224 fNtracks = m.fNtracks;
226 fTh = new Double_t[fNtracks];
227 fPhi = new Double_t[fNtracks];
228 fDeltTh = new Double_t[fNtracks];
229 fDeltPhi = new Double_t[fNtracks];
230 fLabels = new Int_t[fNtracks];
231 fLabelsL2 = new Int_t[fNtracks];
232 if (m.fUsedClusT) fUsedClusT = new ULong64_t[fNtracks]; else fUsedClusT = 0;
233 if(m.fTh)memcpy(fTh,m.fTh,fNtracks*sizeof(Double_t));
234 if(m.fPhi)memcpy(fPhi,m.fPhi,fNtracks*sizeof(Double_t));
235 if(m.fDeltTh)memcpy(fDeltTh,m.fDeltTh,fNtracks*sizeof(Double_t));
236 if(m.fDeltPhi)memcpy(fDeltPhi,m.fDeltPhi,fNtracks*sizeof(Double_t));
237 if(m.fLabels)memcpy(fLabels,m.fLabels,fNtracks*sizeof(Int_t));
238 if(m.fLabelsL2)memcpy(fLabelsL2,m.fLabelsL2,fNtracks*sizeof(Int_t));
239 if(fUsedClusT) memcpy(fUsedClusT,m.fUsedClusT,fNtracks*sizeof(ULong64_t));
240 for (int i=2;i--;) for (int j=2;j--;) if (m.fTCl2Tracks[i][j]) fTCl2Tracks[i][j] = new AliRefArray(*m.fTCl2Tracks[i][j]);
250 fNsingle = m.fNsingle;
252 fThsingle = new Double_t[fNsingle];
253 fPhisingle = new Double_t[fNsingle];
254 fLabelssingle = new Int_t[fNsingle];
255 if (m.fUsedClusS) fUsedClusS = new UInt_t[fNsingle];
257 if(m.fThsingle)memcpy(fThsingle,m.fThsingle,fNsingle*sizeof(Double_t));
258 if(m.fPhisingle)memcpy(fPhisingle,m.fPhisingle,fNsingle*sizeof(Double_t));
259 if(m.fLabelssingle)memcpy(fLabelssingle,m.fLabelssingle,fNsingle*sizeof(Int_t));
260 if(fUsedClusS) memcpy(fUsedClusS,m.fUsedClusS,fNsingle*sizeof(UInt_t));
261 for (int i=2;i--;) if (m.fSCl2Tracks[i]) fSCl2Tracks[i] = new AliRefArray(*m.fSCl2Tracks[i]);
269 fFiredChips[0] = m.fFiredChips[0];
270 fFiredChips[1] = m.fFiredChips[1];
271 for(Int_t ilayer = 0; ilayer < 6; ilayer++){
272 fITSClusters[ilayer] = m.fITSClusters[ilayer];
274 fDPhiWindow2 = m.fDPhiWindow2;
275 fDThetaWindow2 = m.fDThetaWindow2;
276 fDPhiShift = m.fDPhiShift;
277 fNStdDev = m.fNStdDev;
278 fFastOrFiredChips = m.fFastOrFiredChips;
279 fClusterFiredChips = m.fClusterFiredChips;
282 //______________________________________________________________________
283 AliMultiplicity::~AliMultiplicity(){
285 if(fTh)delete [] fTh;fTh = 0;
286 if(fPhi)delete [] fPhi;fPhi = 0;
287 if(fDeltTh)delete [] fDeltTh;fDeltTh = 0;
288 if(fDeltPhi)delete [] fDeltPhi;fDeltPhi = 0;
289 if(fLabels)delete [] fLabels;fLabels = 0;
290 if(fLabelsL2)delete [] fLabelsL2;fLabelsL2 = 0;
291 if(fThsingle)delete [] fThsingle;fThsingle = 0;
292 if(fPhisingle)delete [] fPhisingle;fPhisingle = 0;
293 if(fLabelssingle)delete [] fLabelssingle;fLabelssingle = 0;
294 if(fUsedClusS) delete[] fUsedClusS; fUsedClusS = 0;
295 if(fUsedClusT) delete[] fUsedClusT; fUsedClusT = 0;
296 for (int il=2;il--;) {
297 if (fSCl2Tracks[il]) delete fSCl2Tracks[il]; fSCl2Tracks[il] = 0;
298 if (fTCl2Tracks[il][0]) delete fTCl2Tracks[il][0]; fTCl2Tracks[il][0] = 0;
299 if (fTCl2Tracks[il][1]) delete fTCl2Tracks[il][1]; fTCl2Tracks[il][1] = 0;
303 //______________________________________________________________________
304 void AliMultiplicity::Clear(Option_t*)
308 if(fTh)delete [] fTh;fTh = 0;
309 if(fPhi)delete [] fPhi;fPhi = 0;
310 if(fDeltTh)delete [] fDeltTh;fDeltTh = 0;
311 if(fDeltPhi)delete [] fDeltPhi;fDeltPhi = 0;
312 if(fLabels)delete [] fLabels;fLabels = 0;
313 if(fLabelsL2)delete [] fLabelsL2;fLabelsL2 = 0;
314 if(fThsingle)delete [] fThsingle;fThsingle = 0;
315 if(fPhisingle)delete [] fPhisingle;fPhisingle = 0;
316 if(fLabelssingle)delete [] fLabelssingle;fLabelssingle = 0;
317 if(fUsedClusS) delete[] fUsedClusS; fUsedClusS = 0;
318 if(fUsedClusT) delete[] fUsedClusT; fUsedClusT = 0;
319 for (int il=2;il--;) {
320 if (fSCl2Tracks[il]) delete fSCl2Tracks[il]; fSCl2Tracks[il] = 0;
321 if (fTCl2Tracks[il][0]) delete fTCl2Tracks[il][0]; fTCl2Tracks[il][0] = 0;
322 if (fTCl2Tracks[il][1]) delete fTCl2Tracks[il][1]; fTCl2Tracks[il][1] = 0;
324 fNtracks = fNsingle = 0;
325 for (int i=6;i--;) fITSClusters[0] = 0;
326 fFiredChips[0] = fFiredChips[1] = 0;
327 fFastOrFiredChips.ResetAllBits(kTRUE);
328 fClusterFiredChips.ResetAllBits(kTRUE);
332 //______________________________________________________________________
333 void AliMultiplicity::SetLabel(Int_t i, Int_t layer, Int_t label)
335 if(i>=0 && i<fNtracks) {
339 } else if (layer == 1) {
341 fLabelsL2[i] = label;
346 Error("SetLabel","Invalid track number %d or layer %d",i,layer);
349 //______________________________________________________________________
350 void AliMultiplicity::SetLabelSingle(Int_t i, Int_t label)
352 if(i>=0 && i<fNsingle) {
354 fLabelssingle[i] = label;
358 Error("SetLabelSingle","Invalid single cluster number %d",i);
361 //______________________________________________________________________
362 UInt_t AliMultiplicity::GetNumberOfITSClusters(Int_t layMin, Int_t layMax) const {
364 if(layMax < layMin) {
365 AliError("layer min > layer max");
369 AliError("layer min < 0");
373 AliError("layer max > 0");
378 for (Int_t i=layMin; i<=layMax; i++) sum+=fITSClusters[i];
383 //______________________________________________________________________
384 void AliMultiplicity::SetTrackletData(Int_t id, const Float_t* tlet, UInt_t trSPD1, UInt_t trSPD2)
386 // fill tracklet data
387 if (id>=fNtracks) {AliError(Form("Number of declared tracklets %d < %d",fNtracks,id)); return;}
390 fDeltPhi[id] = tlet[2];
391 fDeltTh[id] = tlet[3];
392 fLabels[id] = Int_t(tlet[4]);
393 fLabelsL2[id] = Int_t(tlet[5]);
394 if (!GetMultTrackRefs()) {
395 if (!fUsedClusT) {fUsedClusT = new ULong64_t[fNtracks]; memset(fUsedClusT,0,fNtracks*sizeof(ULong64_t));}
396 fUsedClusT[id] = (((ULong64_t)trSPD2)<<32) + trSPD1;
401 //______________________________________________________________________
402 void AliMultiplicity::SetSingleClusterData(Int_t id, const Float_t* scl, UInt_t tr)
404 // fill single cluster data
405 if (id>=fNsingle) {AliError(Form("Number of declared singles %d < %d",fNsingle,id)); return;}
406 fThsingle[id] = scl[0];
407 fPhisingle[id] = scl[1];
408 fLabelssingle[id] = Int_t(scl[2]);
409 if (!GetMultTrackRefs()) {
410 if (!fUsedClusS) {fUsedClusS = new UInt_t[fNsingle]; memset(fUsedClusS,0,fNsingle*sizeof(UInt_t));}
416 //______________________________________________________________________
417 Bool_t AliMultiplicity::FreeClustersTracklet(Int_t i, Int_t mode) const
419 // return kTRUE if the tracklet was not used by the track (on any of layers) of type mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure
420 if (mode<0 || mode>1 || i<0 || i>fNtracks) return kFALSE;
421 if (GetMultTrackRefs()) { // new format allows multiple references
422 return !((fTCl2Tracks[0][mode] && fTCl2Tracks[0][mode]->HasReference(i)) ||
423 (fTCl2Tracks[1][mode] && fTCl2Tracks[1][mode]->HasReference(i)));
426 if (!fUsedClusT) return kFALSE;
427 const ULong64_t kMask0 = 0x0000ffff0000ffffLL;
428 const ULong64_t kMask1 = 0xffff0000ffff0000LL;
429 return (fUsedClusT[i]&(mode ? kMask1:kMask0)) == 0;
432 //______________________________________________________________________
433 Bool_t AliMultiplicity::GetTrackletTrackIDs(Int_t i, Int_t mode, Int_t &spd1, Int_t &spd2) const
435 // set spd1 and spd2 to ID's of the tracks using the clusters of the tracklet (-1 if not used)
436 // Mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure tracks
437 // return false if the neither of clusters is used
438 // note: stored value: [(idSAPureSPD2+1)<<16+(idTPCITS/SA_SPD2+1)]<<32 + [(idSAPureSPD1+1)<<16+(idTPCITS/SA_SPD1+1)]
439 // Attention: new format allows references to multiple tracks, here only the 1st will be returned
441 if ( mode<0 || mode>1 || i<0 || i>fNtracks ) return kFALSE;
442 if (GetMultTrackRefs()) {
443 if (fTCl2Tracks[0][mode]) spd1 = fTCl2Tracks[0][mode]->GetReference(i,0);
444 if (fTCl2Tracks[1][mode]) spd2 = fTCl2Tracks[1][mode]->GetReference(i,0);
447 if (!fUsedClusT) return kFALSE;
448 spd1 = (fUsedClusT[i]&0xffffffffLL);
449 spd2 = (fUsedClusT[i]>>32);
450 if (mode) { spd1 >>= 16; spd2 >>= 16;}
451 else { spd1 &= 0xffff; spd2 &= 0xffff;}
452 spd1--; // we are storing id+1
455 return !(spd1<0&&spd2<0);
458 //______________________________________________________________________
459 Int_t AliMultiplicity::GetTrackletTrackIDsLay(Int_t lr,Int_t i, Int_t mode, UInt_t* refs, UInt_t maxRef) const
461 // fill array refs with maximum maxRef references on tracks used by the cluster of layer lr of tracklet i.
462 // return number of filled references
463 // Mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure tracks
466 if ( mode<0 || mode>1 || i<0 || i>fNtracks || lr<0||lr>1) return nrefs;
467 if (GetMultTrackRefs()) {
468 if (fTCl2Tracks[lr][mode]) nrefs = fTCl2Tracks[lr][mode]->GetReferences(i,refs, maxRef);
471 if (!fUsedClusT || maxRef<1) return nrefs;
472 int tr = (fUsedClusT[i]&0xffffffffLL);
473 if (mode) { lr==0 ? tr >>= 16 : tr >>= 16;}
474 else { lr==0 ? tr &= 0xffff : tr &= 0xffff;}
475 refs[0] = tr--; // we are storing id+1
481 //______________________________________________________________________
482 Int_t AliMultiplicity::GetSingleClusterTrackIDs(Int_t i, Int_t mode, UInt_t* refs, UInt_t maxRef) const
484 // fill array refs with maximum maxRef references on tracks used by the single cluster i of layer lr
485 // return number of filled references
486 // Mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure tracks
489 if ( mode<0 || mode>1 || i<0 || i>fNtracks) return nrefs;
490 if (GetMultTrackRefs()) {
491 if (fSCl2Tracks[mode]) nrefs = fSCl2Tracks[mode]->GetReferences(i,refs, maxRef);
494 if (!fUsedClusS || maxRef<1) return nrefs;
495 int tr = fUsedClusS[i];
498 refs[0] = tr--; // we are storing id+1
504 //______________________________________________________________________
505 Bool_t AliMultiplicity::FreeSingleCluster(Int_t i, Int_t mode) const
507 // return kTRUE if the cluster was not used by the track of type mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure
508 if (mode<0 || mode>1 || i<0 || i>fNsingle) return kFALSE;
509 if (GetMultTrackRefs()) { // new format allows multiple references
510 return !(fSCl2Tracks[mode] && fSCl2Tracks[mode]->HasReference(i));
512 if (!fUsedClusS) return kFALSE;
513 const UInt_t kMask0 = 0x0000ffff;
514 const UInt_t kMask1 = 0xffff0000;
515 return (fUsedClusS[i]&(mode ? kMask1:kMask0)) == 0;
518 //______________________________________________________________________
519 Bool_t AliMultiplicity::GetSingleClusterTrackID(Int_t i, Int_t mode, Int_t &tr) const
521 // set tr to id of the track using the single clusters (-1 if not used)
522 // Mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure tracks
523 // return false if the cluster is not used
525 // Attention: new format allows references to multiple tracks, here only the 1st will be returned
527 if (mode<0 || mode>1 || i<0 || i>fNsingle) return kFALSE;
528 if (GetMultTrackRefs()) { if (fSCl2Tracks[mode]) tr = fSCl2Tracks[mode]->GetReference(i,0);}
530 if (!fUsedClusS) return kFALSE;
532 if (mode) tr >>= 16; else tr &= 0xffff;
538 //______________________________________________________________________
539 void AliMultiplicity::CompactBits()
541 // sqeeze bit contrainers to minimum
542 fFastOrFiredChips.Compact();
543 fClusterFiredChips.Compact();
546 //______________________________________________________________________
547 void AliMultiplicity::Print(Option_t *opt) const
550 printf("N.tracklets: %4d N.singles: %4d, Multiple cluster->track refs:%s\n"
551 "Used: DPhiShift: %.3e Sig^2: dPhi:%.3e dTht:%.3e NStdDev:%.2f ScaleDThtSin2T:%s\n",
552 fNtracks,fNsingle,GetMultTrackRefs() ? "ON":"OFF",
553 fDPhiShift,fDPhiWindow2,fDThetaWindow2,fNStdDev,GetScaleDThetaBySin2T() ? "ON":"OFF");
554 TString opts = opt; opts.ToLower();
555 int t0spd1=-1,t1spd1=-1,t0spd2=-1,t1spd2=-1,nt[2][2]={{0}};
556 UInt_t t[2][2][10]={{{0}}};
558 if (opts.Contains("t")) {
559 for (int i=0;i<fNtracks;i++) {
560 if (GetMultTrackRefs()) for (int il=2;il--;)for(int it=2;it--;) nt[il][it] = GetTrackletTrackIDsLay(il,i,it,t[il][it],10);
562 GetTrackletTrackIDs(i,0,t0spd1,t0spd2);
563 GetTrackletTrackIDs(i,1,t1spd1,t1spd2);
565 printf("T#%3d| Eta:%+5.2f Th:%+6.3f Phi:%+6.3f DTh:%+6.3f DPhi:%+6.3f L1:%5d L2:%5d ",
566 i,GetEta(i),fTh[i],fPhi[i],fDeltTh[i],fDeltPhi[i],fLabels[i],fLabelsL2[i]);
567 if (!GetMultTrackRefs()) printf("U:L1[%4d/%4d] L2[%4d/%4d]\n",t0spd1,t1spd1,t0spd2,t1spd2);
570 if (!nt[0][0]) printf("%4d ",-1); else for(int j=0;j<nt[0][0];j++) printf("%4d ",t[0][0][j]); printf("/");
571 if (!nt[0][1]) printf("%4d ",-1); else for(int j=0;j<nt[0][1];j++) printf("%4d ",t[0][1][j]); printf("]");
574 if (!nt[1][0]) printf("%4d ",-1); else for(int j=0;j<nt[1][0];j++) printf("%4d ",t[1][0][j]); printf("/");
575 if (!nt[1][1]) printf("%4d ",-1); else for(int j=0;j<nt[1][1];j++) printf("%4d ",t[1][1][j]); printf("]\n");
579 if (opts.Contains("s")) {
580 for (int i=0;i<fNsingle;i++) {
581 if (GetMultTrackRefs()) for(int it=2;it--;) nt[0][it] = GetSingleClusterTrackIDs(i,it,t[0][it],10);
583 GetSingleClusterTrackID(i,0,t0spd1);
584 GetSingleClusterTrackID(i,1,t1spd1);
586 printf("S#%3d| Th:%+6.3f Phi:%+6.3f L:%6d ",i,fThsingle[i],fPhisingle[i],fLabelssingle[i]);
587 if (!GetMultTrackRefs()) printf("U:[%4d/%4d]\n", t0spd1,t1spd1);
590 if (!nt[0][0]) printf("%4d ",-1); else for(int j=0;j<nt[0][0];j++) printf("%4d ",t[0][0][j]); printf("/");
591 if (!nt[0][1]) printf("%4d ",-1); else for(int j=0;j<nt[0][1];j++) printf("%4d ",t[0][1][j]); printf("]\n");