#include #include "AliMultiplicity.h" #include "AliLog.h" #include "AliRefArray.h" ClassImp(AliMultiplicity) //______________________________________________________________________ AliMultiplicity::AliMultiplicity(): TObject(), fNtracks(0), fNsingle(0), // fDPhiWindow2(0.08*0.08), fDThetaWindow2(0.025*0.025), fDPhiShift(0.0045), fNStdDev(1.0), // fLabels(0), fLabelsL2(0), fUsedClusS(0), fUsedClusT(0), fTh(0), fPhi(0), fDeltTh(0), fDeltPhi(0), fThsingle(0), fPhisingle(0), fLabelssingle(0), fFastOrFiredChips(1200), fClusterFiredChips(1200) { // Default Constructor fFiredChips[0] = 0; fFiredChips[1] = 0; for (int il=2;il--;) fSCl2Tracks[il] = fTCl2Tracks[il][0] = fTCl2Tracks[il][1] = 0; for(Int_t ilayer = 0; ilayer < 6; ilayer++)fITSClusters[ilayer] = 0; } //______________________________________________________________________ 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): TObject(), fNtracks(ntr), fNsingle(ns), // fDPhiWindow2(0.08*0.08), fDThetaWindow2(0.025*0.025), fDPhiShift(0.0045), fNStdDev(1.0), // fLabels(0), fLabelsL2(0), fUsedClusS(0), fUsedClusT(0), fTh(0), fPhi(0), fDeltTh(0), fDeltPhi(0), fThsingle(0), fPhisingle(0), fLabelssingle(0), fFastOrFiredChips(1200), fClusterFiredChips(1200) { // Standard constructor for (int il=2;il--;) fSCl2Tracks[il] = fTCl2Tracks[il][0] = fTCl2Tracks[il][1] = 0; if(ntr>0){ fLabels = new Int_t[ntr]; fLabelsL2 = new Int_t[ntr]; fTh = new Double_t [ntr]; fPhi = new Double_t [ntr]; fDeltTh = new Double_t [ntr]; fDeltPhi = new Double_t [ntr]; for(Int_t i=0;i0){ fThsingle = new Double_t [ns]; fPhisingle = new Double_t [ns]; fLabelssingle = new Int_t [ns]; for(Int_t i=0;i0){ fLabels = new Int_t[ntr]; fLabelsL2 = new Int_t[ntr]; fTh = new Double_t [ntr]; fPhi = new Double_t [ntr]; fDeltTh = new Double_t [ntr]; fDeltPhi = new Double_t [ntr]; for(Int_t i=fNtracks;i--;){ fTh[i]=fPhi[i]=fDeltTh[i]=fDeltPhi[i] = 0; fLabels[i] = fLabelsL2[i] = 0; } } if(ns>0){ fThsingle = new Double_t [ns]; fPhisingle = new Double_t [ns]; fLabelssingle = new Int_t [ns]; for(Int_t i=fNsingle;i--;) fThsingle[i] = fPhisingle[i] = fLabelssingle[i] = 0; } fFiredChips[0] = nfcL1; fFiredChips[1] = nfcL2; fFastOrFiredChips = fFastOr; for(Int_t ilayer=6;ilayer--;) fITSClusters[ilayer] = 0; } //______________________________________________________________________ AliMultiplicity::AliMultiplicity(const AliMultiplicity& m): TObject(m), fNtracks(m.fNtracks), fNsingle(m.fNsingle), // fDPhiWindow2(0.08*0.08), fDThetaWindow2(0.025*0.025), fDPhiShift(0.0045), fNStdDev(1.0), // fLabels(0), fLabelsL2(0), fUsedClusS(0), fUsedClusT(0), fTh(0), fPhi(0), fDeltTh(0), fDeltPhi(0), fThsingle(0), fPhisingle(0), fLabelssingle(0), fFastOrFiredChips(1200), fClusterFiredChips(1200) { // copy constructor for (int il=2;il--;) fSCl2Tracks[il] = fTCl2Tracks[il][0] = fTCl2Tracks[il][1] = 0; Duplicate(m); } //______________________________________________________________________ AliMultiplicity &AliMultiplicity::operator=(const AliMultiplicity& m){ // assignment operator if(this == &m)return *this; ((TObject *)this)->operator=(m); if(fTh)delete [] fTh;fTh = 0; if(fPhi)delete [] fPhi;fPhi = 0; if(fDeltTh)delete [] fDeltTh;fDeltTh= 0; if(fDeltPhi)delete [] fDeltPhi;fDeltPhi = 0; if(fLabels)delete [] fLabels;fLabels = 0; if(fLabelsL2)delete [] fLabelsL2;fLabelsL2 = 0; if(fThsingle)delete [] fThsingle;fThsingle = 0; if(fPhisingle)delete [] fPhisingle;fPhisingle = 0; if(fLabelssingle)delete [] fLabelssingle;fLabelssingle = 0; if(fUsedClusS) delete[] fUsedClusS; fUsedClusS = 0; if(fUsedClusT) delete[] fUsedClusT; fUsedClusT = 0; for (int il=2;il--;) { if (fSCl2Tracks[il]) delete fSCl2Tracks[il]; fSCl2Tracks[il] = 0; if (fTCl2Tracks[il][0]) delete fTCl2Tracks[il][0]; fTCl2Tracks[il][0] = 0; if (fTCl2Tracks[il][1]) delete fTCl2Tracks[il][1]; fTCl2Tracks[il][1] = 0; } Duplicate(m); // return *this; } void AliMultiplicity::Copy(TObject &obj) const { // this overwrites the virtual TOBject::Copy() // to allow run time copying without casting // in AliESDEvent if(this==&obj)return; AliMultiplicity *robj = dynamic_cast(&obj); if(!robj)return; // not an AliMultiplicity *robj = *this; } //______________________________________________________________________ void AliMultiplicity::Duplicate(const AliMultiplicity& m){ // used by copy constructor and assignment operator fNtracks = m.fNtracks; if(fNtracks>0){ fTh = new Double_t[fNtracks]; fPhi = new Double_t[fNtracks]; fDeltTh = new Double_t[fNtracks]; fDeltPhi = new Double_t[fNtracks]; fLabels = new Int_t[fNtracks]; fLabelsL2 = new Int_t[fNtracks]; if (m.fUsedClusT) fUsedClusT = new ULong64_t[fNtracks]; else fUsedClusT = 0; if(m.fTh)memcpy(fTh,m.fTh,fNtracks*sizeof(Double_t)); if(m.fPhi)memcpy(fPhi,m.fPhi,fNtracks*sizeof(Double_t)); if(m.fDeltTh)memcpy(fDeltTh,m.fDeltTh,fNtracks*sizeof(Double_t)); if(m.fDeltPhi)memcpy(fDeltPhi,m.fDeltPhi,fNtracks*sizeof(Double_t)); if(m.fLabels)memcpy(fLabels,m.fLabels,fNtracks*sizeof(Int_t)); if(m.fLabelsL2)memcpy(fLabelsL2,m.fLabelsL2,fNtracks*sizeof(Int_t)); if(fUsedClusT) memcpy(fUsedClusT,m.fUsedClusT,fNtracks*sizeof(ULong64_t)); for (int i=2;i--;) for (int j=2;j--;) if (m.fTCl2Tracks[i][j]) fTCl2Tracks[i][j] = new AliRefArray(*m.fTCl2Tracks[i][j]); } else { fTh = 0; fPhi = 0; fDeltTh = 0; fDeltPhi = 0; fLabels = 0; fLabelsL2 = 0; } fNsingle = m.fNsingle; if(fNsingle>0){ fThsingle = new Double_t[fNsingle]; fPhisingle = new Double_t[fNsingle]; fLabelssingle = new Int_t[fNsingle]; if (m.fUsedClusS) fUsedClusS = new UInt_t[fNsingle]; else fUsedClusS = 0; if(m.fThsingle)memcpy(fThsingle,m.fThsingle,fNsingle*sizeof(Double_t)); if(m.fPhisingle)memcpy(fPhisingle,m.fPhisingle,fNsingle*sizeof(Double_t)); if(m.fLabelssingle)memcpy(fLabelssingle,m.fLabelssingle,fNsingle*sizeof(Int_t)); if(fUsedClusS) memcpy(fUsedClusS,m.fUsedClusS,fNsingle*sizeof(UInt_t)); for (int i=2;i--;) if (m.fSCl2Tracks[i]) fSCl2Tracks[i] = new AliRefArray(*m.fSCl2Tracks[i]); } else { fThsingle = 0; fPhisingle = 0; fLabelssingle = 0; } fFiredChips[0] = m.fFiredChips[0]; fFiredChips[1] = m.fFiredChips[1]; for(Int_t ilayer = 0; ilayer < 6; ilayer++){ fITSClusters[ilayer] = m.fITSClusters[ilayer]; } fDPhiWindow2 = m.fDPhiWindow2; fDThetaWindow2 = m.fDThetaWindow2; fDPhiShift = fDPhiShift; fNStdDev = fNStdDev; fFastOrFiredChips = m.fFastOrFiredChips; fClusterFiredChips = m.fClusterFiredChips; } //______________________________________________________________________ AliMultiplicity::~AliMultiplicity(){ // Destructor if(fTh)delete [] fTh;fTh = 0; if(fPhi)delete [] fPhi;fPhi = 0; if(fDeltTh)delete [] fDeltTh;fDeltTh = 0; if(fDeltPhi)delete [] fDeltPhi;fDeltPhi = 0; if(fLabels)delete [] fLabels;fLabels = 0; if(fLabelsL2)delete [] fLabelsL2;fLabelsL2 = 0; if(fThsingle)delete [] fThsingle;fThsingle = 0; if(fPhisingle)delete [] fPhisingle;fPhisingle = 0; if(fLabelssingle)delete [] fLabelssingle;fLabelssingle = 0; if(fUsedClusS) delete[] fUsedClusS; fUsedClusS = 0; if(fUsedClusT) delete[] fUsedClusT; fUsedClusT = 0; for (int il=2;il--;) { if (fSCl2Tracks[il]) delete fSCl2Tracks[il]; fSCl2Tracks[il] = 0; if (fTCl2Tracks[il][0]) delete fTCl2Tracks[il][0]; fTCl2Tracks[il][0] = 0; if (fTCl2Tracks[il][1]) delete fTCl2Tracks[il][1]; fTCl2Tracks[il][1] = 0; } } //______________________________________________________________________ void AliMultiplicity::Clear(Option_t*) { // reset all TObject::Clear(); if(fTh)delete [] fTh;fTh = 0; if(fPhi)delete [] fPhi;fPhi = 0; if(fDeltTh)delete [] fDeltTh;fDeltTh = 0; if(fDeltPhi)delete [] fDeltPhi;fDeltPhi = 0; if(fLabels)delete [] fLabels;fLabels = 0; if(fLabelsL2)delete [] fLabelsL2;fLabelsL2 = 0; if(fThsingle)delete [] fThsingle;fThsingle = 0; if(fPhisingle)delete [] fPhisingle;fPhisingle = 0; if(fLabelssingle)delete [] fLabelssingle;fLabelssingle = 0; if(fUsedClusS) delete[] fUsedClusS; fUsedClusS = 0; if(fUsedClusT) delete[] fUsedClusT; fUsedClusT = 0; for (int il=2;il--;) { if (fSCl2Tracks[il]) delete fSCl2Tracks[il]; fSCl2Tracks[il] = 0; if (fTCl2Tracks[il][0]) delete fTCl2Tracks[il][0]; fTCl2Tracks[il][0] = 0; if (fTCl2Tracks[il][1]) delete fTCl2Tracks[il][1]; fTCl2Tracks[il][1] = 0; } fNtracks = fNsingle = 0; for (int i=6;i--;) fITSClusters[0] = 0; fFiredChips[0] = fFiredChips[1] = 0; fFastOrFiredChips.ResetAllBits(kTRUE); fClusterFiredChips.ResetAllBits(kTRUE); // } //______________________________________________________________________ void AliMultiplicity::SetLabel(Int_t i, Int_t layer, Int_t label) { if(i>=0 && i=0 && i layer max"); return 0; } if(layMin < 0) { AliError("layer min < 0"); return 0; } if(layMax < 0) { AliError("layer max > 0"); return 0; } Int_t sum=0; for (Int_t i=layMin; i<=layMax; i++) sum+=fITSClusters[i]; return sum; } //______________________________________________________________________ void AliMultiplicity::SetTrackletData(Int_t id, const Float_t* tlet, UInt_t trSPD1, UInt_t trSPD2) { // fill tracklet data if (id>=fNtracks) {AliError(Form("Number of declared tracklets %d < %d",fNtracks,id)); return;} fTh[id] = tlet[0]; fPhi[id] = tlet[1]; fDeltPhi[id] = tlet[2]; fDeltTh[id] = tlet[3]; fLabels[id] = Int_t(tlet[4]); fLabelsL2[id] = Int_t(tlet[5]); if (!GetMultTrackRefs()) { if (!fUsedClusT) {fUsedClusT = new ULong64_t[fNtracks]; memset(fUsedClusT,0,fNtracks*sizeof(ULong64_t));} fUsedClusT[id] = (((ULong64_t)trSPD2)<<32) + trSPD1; } // } //______________________________________________________________________ void AliMultiplicity::SetSingleClusterData(Int_t id, const Float_t* scl, UInt_t tr) { // fill single cluster data if (id>=fNsingle) {AliError(Form("Number of declared singles %d < %d",fNsingle,id)); return;} fThsingle[id] = scl[0]; fPhisingle[id] = scl[1]; fLabelssingle[id] = Int_t(scl[2]); if (!GetMultTrackRefs()) { if (!fUsedClusS) {fUsedClusS = new UInt_t[fNsingle]; memset(fUsedClusS,0,fNsingle*sizeof(UInt_t));} fUsedClusS[id] = tr; } // } //______________________________________________________________________ Bool_t AliMultiplicity::FreeClustersTracklet(Int_t i, Int_t mode) const { // 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 if (mode<0 || mode>1 || i<0 || i>fNtracks) return kFALSE; if (GetMultTrackRefs()) { // new format allows multiple references return !((fTCl2Tracks[0][mode] && fTCl2Tracks[0][mode]->HasReference(i)) || (fTCl2Tracks[1][mode] && fTCl2Tracks[1][mode]->HasReference(i))); } // if (!fUsedClusT) return kFALSE; const ULong64_t kMask0 = 0x0000ffff0000ffffLL; const ULong64_t kMask1 = 0xffff0000ffff0000LL; return (fUsedClusT[i]&(mode ? kMask1:kMask0)) == 0; } //______________________________________________________________________ Bool_t AliMultiplicity::GetTrackletTrackIDs(Int_t i, Int_t mode, Int_t &spd1, Int_t &spd2) const { // set spd1 and spd2 to ID's of the tracks using the clusters of the tracklet (-1 if not used) // Mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure tracks // return false if the neither of clusters is used // note: stored value: [(idSAPureSPD2+1)<<16+(idTPCITS/SA_SPD2+1)]<<32 + [(idSAPureSPD1+1)<<16+(idTPCITS/SA_SPD1+1)] // Attention: new format allows references to multiple tracks, here only the 1st will be returned spd1 = spd2 = -1; if ( mode<0 || mode>1 || i<0 || i>fNtracks ) return kFALSE; if (GetMultTrackRefs()) { if (fTCl2Tracks[0][mode]) spd1 = fTCl2Tracks[0][mode]->GetReference(i,0); if (fTCl2Tracks[1][mode]) spd2 = fTCl2Tracks[1][mode]->GetReference(i,0); } else { if (!fUsedClusT) return kFALSE; spd1 = (fUsedClusT[i]&0xffffffffLL); spd2 = (fUsedClusT[i]>>32); if (mode) { spd1 >>= 16; spd2 >>= 16;} else { spd1 &= 0xffff; spd2 &= 0xffff;} spd1--; // we are storing id+1 spd2--; } return !(spd1<0&&spd2<0); } //______________________________________________________________________ Int_t AliMultiplicity::GetTrackletTrackIDsLay(Int_t lr,Int_t i, Int_t mode, UInt_t* refs, UInt_t maxRef) const { // fill array refs with maximum maxRef references on tracks used by the cluster of layer lr of tracklet i. // return number of filled references // Mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure tracks // int nrefs = 0; if ( mode<0 || mode>1 || i<0 || i>fNtracks || lr<0||lr>1) return nrefs; if (GetMultTrackRefs()) { if (fTCl2Tracks[lr][mode]) nrefs = fTCl2Tracks[lr][mode]->GetReferences(i,refs, maxRef); } else { if (!fUsedClusT || maxRef<1) return nrefs; int tr = (fUsedClusT[i]&0xffffffffLL); if (mode) { lr==0 ? tr >>= 16 : tr >>= 16;} else { lr==0 ? tr &= 0xffff : tr &= 0xffff;} refs[0] = tr--; // we are storing id+1 nrefs = 1; } return nrefs; } //______________________________________________________________________ Int_t AliMultiplicity::GetSingleClusterTrackIDs(Int_t i, Int_t mode, UInt_t* refs, UInt_t maxRef) const { // fill array refs with maximum maxRef references on tracks used by the single cluster i of layer lr // return number of filled references // Mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure tracks // int nrefs = 0; if ( mode<0 || mode>1 || i<0 || i>fNtracks) return nrefs; if (GetMultTrackRefs()) { if (fSCl2Tracks[mode]) nrefs = fSCl2Tracks[mode]->GetReferences(i,refs, maxRef); } else { if (!fUsedClusS || maxRef<1) return nrefs; int tr = fUsedClusS[i]; if (mode) tr >>= 16; else tr &= 0xffff; refs[0] = tr--; // we are storing id+1 nrefs = 1; } return nrefs; } //______________________________________________________________________ Bool_t AliMultiplicity::FreeSingleCluster(Int_t i, Int_t mode) const { // return kTRUE if the cluster was not used by the track of type mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure if (mode<0 || mode>1 || i<0 || i>fNsingle) return kFALSE; if (GetMultTrackRefs()) { // new format allows multiple references return !(fSCl2Tracks[mode] && fSCl2Tracks[mode]->HasReference(i)); } if (!fUsedClusS) return kFALSE; const UInt_t kMask0 = 0x0000ffff; const UInt_t kMask1 = 0xffff0000; return (fUsedClusS[i]&(mode ? kMask1:kMask0)) == 0; } //______________________________________________________________________ Bool_t AliMultiplicity::GetSingleClusterTrackID(Int_t i, Int_t mode, Int_t &tr) const { // set tr to id of the track using the single clusters (-1 if not used) // Mode: 0=TPC/ITS or ITS_SA, 1=ITS_SA_Pure tracks // return false if the cluster is not used // // Attention: new format allows references to multiple tracks, here only the 1st will be returned tr = -1; if (mode<0 || mode>1 || i<0 || i>fNsingle) return kFALSE; if (GetMultTrackRefs()) { if (fSCl2Tracks[mode]) tr = fSCl2Tracks[mode]->GetReference(i,0);} else { if (!fUsedClusS) return kFALSE; tr = fUsedClusS[i]; if (mode) tr >>= 16; else tr &= 0xffff; tr--; } return tr>=0; } //______________________________________________________________________ void AliMultiplicity::CompactBits() { // sqeeze bit contrainers to minimum fFastOrFiredChips.Compact(); fClusterFiredChips.Compact(); } //______________________________________________________________________ void AliMultiplicity::Print(Option_t *opt) const { // print printf("N.tracklets: %4d N.singles: %4d, Multiple cluster->track refs:%s\n" "Used: DPhiShift: %.3e Sig^2: dPhi:%.3e dTht:%.3e NStdDev:%.2f ScaleDThtSin2T:%s\n", fNtracks,fNsingle,GetMultTrackRefs() ? "ON":"OFF", fDPhiShift,fDPhiWindow2,fDThetaWindow2,fNStdDev,GetScaleDThetaBySin2T() ? "ON":"OFF"); TString opts = opt; opts.ToLower(); int t0spd1=-1,t1spd1=-1,t0spd2=-1,t1spd2=-1,nt[2][2]={{0}}; UInt_t t[2][2][10]={{{0}}}; // if (opts.Contains("t")) { for (int i=0;i