#include <string.h>
#include "AliMultiplicity.h"
+#include "AliLog.h"
ClassImp(AliMultiplicity)
AliMultiplicity::AliMultiplicity():
TObject(),
fNtracks(0),
+ fNsingle(0),
+ fLabels(0),
+ fLabelsL2(0),
+ fUsedClusS(0),
+ fUsedClusT(0),
fTh(0),
fPhi(0),
+ fDeltTh(0),
fDeltPhi(0),
- fNsingle(0),
fThsingle(0),
- fPhisingle(0)
-
+ fPhisingle(0),
+ fLabelssingle(0),
+ fFastOrFiredChips(1200),
+ fClusterFiredChips(1200)
{
// Default Constructor
+ fFiredChips[0] = 0;
+ fFiredChips[1] = 0;
+ for(Int_t ilayer = 0; ilayer < 6; ilayer++)fITSClusters[ilayer] = 0;
}
//______________________________________________________________________
-AliMultiplicity::AliMultiplicity(Int_t ntr, Float_t *t, Float_t *ph, Float_t *df, Int_t ns, Float_t *ts, Float_t *ps):
+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),
+ fLabels(0),
+ fLabelsL2(0),
+ fUsedClusS(0),
+ fUsedClusT(0),
fTh(0),
fPhi(0),
+ fDeltTh(0),
fDeltPhi(0),
- fNsingle(ns),
fThsingle(0),
- fPhisingle(0)
+ fPhisingle(0),
+ fLabelssingle(0),
+ fFastOrFiredChips(1200),
+ fClusterFiredChips(1200)
{
// Standard constructor
if(ntr>0){
- fTh = new Float_t [ntr];
- fPhi = new Float_t [ntr];
- fDeltPhi = new Float_t [ntr];
+ 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;i<fNtracks;i++){
- fTh[i]=t[i];
+ fTh[i]=th[i];
fPhi[i]=ph[i];
- fDeltPhi[i]=df[i];
+ fDeltTh[i]=dth[i];
+ fDeltPhi[i]=dph[i];
+ fLabels[i] = labels[i];
+ fLabelsL2[i] = labelsL2[i];
}
}
if(ns>0){
- fThsingle = new Float_t [ns];
- fPhisingle = new Float_t [ns];
+ fThsingle = new Double_t [ns];
+ fPhisingle = new Double_t [ns];
+ fLabelssingle = new Int_t [ns];
for(Int_t i=0;i<fNsingle;i++){
fThsingle[i]=ts[i];
fPhisingle[i]=ps[i];
+ fLabelssingle[i]=labelss[i];
+ }
+ }
+ fFiredChips[0] = nfcL1;
+ fFiredChips[1] = nfcL2;
+ fFastOrFiredChips = fFastOr;
+ for(Int_t ilayer = 0; ilayer < 6; ilayer++)fITSClusters[ilayer] = 0;
+}
+
+//______________________________________________________________________
+AliMultiplicity::AliMultiplicity(Int_t ntr, Int_t ns, Short_t nfcL1, Short_t nfcL2, const TBits & fFastOr) :
+ TObject(),
+ fNtracks(ntr),
+ fNsingle(ns),
+ 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 to create the arrays w/o filling
+ 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];
+ fUsedClusT = new ULong64_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];
+ fUsedClusS = new UInt_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),
+ fLabels(0),
+ fLabelsL2(0),
+ fUsedClusS(0),
+ fUsedClusT(0),
fTh(0),
fPhi(0),
+ fDeltTh(0),
fDeltPhi(0),
- fNsingle(m.fNsingle),
fThsingle(0),
- fPhisingle(0)
+ fPhisingle(0),
+ fLabelssingle(0),
+ fFastOrFiredChips(1200),
+ fClusterFiredChips(1200)
{
// copy constructor
-
Duplicate(m);
-
}
//______________________________________________________________________
if(this == &m)return *this;
((TObject *)this)->operator=(m);
- if (fTh)delete [] fTh;
- if(fPhi)delete [] fPhi;
- if(fDeltPhi)delete [] fDeltPhi;
- if(fThsingle)delete [] fThsingle;
- if(fPhisingle)delete [] fPhisingle;
+ 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;
+
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<AliMultiplicity*>(&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 Float_t [fNtracks];
- fPhi = new Float_t [fNtracks];
- fDeltPhi = new Float_t [fNtracks];
+ 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));
}
else {
fTh = 0;
fPhi = 0;
+ fDeltTh = 0;
fDeltPhi = 0;
+ fLabels = 0;
+ fLabelsL2 = 0;
}
fNsingle = m.fNsingle;
if(fNsingle>0){
- fThsingle = new Float_t [fNsingle];
- fPhisingle = new Float_t [fNsingle];
+ 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));
}
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];
}
- memcpy(fTh,m.fTh,fNtracks*sizeof(Float_t));
- memcpy(fPhi,m.fPhi,fNtracks*sizeof(Float_t));
- memcpy(fDeltPhi,m.fDeltPhi,fNtracks*sizeof(Float_t));
- memcpy(fThsingle,m.fThsingle,fNsingle*sizeof(Float_t));
- memcpy(fPhisingle,m.fPhisingle,fNsingle*sizeof(Float_t));
+
+ fFastOrFiredChips = m.fFastOrFiredChips;
+ fClusterFiredChips = m.fClusterFiredChips;
}
//______________________________________________________________________
AliMultiplicity::~AliMultiplicity(){
// Destructor
- if (fTh)delete [] fTh;
- if(fPhi)delete [] fPhi;
- if(fDeltPhi)delete [] fDeltPhi;
- if(fThsingle)delete [] fThsingle;
- if(fPhisingle)delete [] fPhisingle;
+ 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;
+}
+
+//______________________________________________________________________
+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;
+ 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<fNtracks) {
+ if (layer == 0) {
+ fLabels[i] = label;
+ return;
+ } else if (layer == 1) {
+ if (fLabelsL2) {
+ fLabelsL2[i] = label;
+ return;
+ }
+ }
+ }
+ Error("SetLabel","Invalid track number %d or layer %d",i,layer);
+}
+
+//______________________________________________________________________
+void AliMultiplicity::SetLabelSingle(Int_t i, Int_t label)
+{
+ if(i>=0 && i<fNsingle) {
+ if (fLabelssingle) {
+ fLabelssingle[i] = label;
+ return;
+ }
+ }
+ Error("SetLabelSingle","Invalid single cluster number %d",i);
+}
+
+//______________________________________________________________________
+UInt_t AliMultiplicity::GetNumberOfITSClusters(Int_t layMin, Int_t layMax) const {
+
+ if(layMax < layMin) {
+ AliError("layer min > 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]);
+ 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]);
+ 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 (!fUsedClusT || mode<0 || mode>1 || i<0 || i>fNtracks) 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)]
+ if (!fUsedClusT || mode<0 || mode>1 || i<0 || i>fNtracks) {spd1 = spd2 = -1; 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);
+}
+
+//______________________________________________________________________
+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 (!fUsedClusS || mode<0 || mode>1 || i<0 || i>fNsingle) 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
+ //
+ // note: stored value: [(idSAPure+1)<<16+(idTPCITS/SA+1)]
+ if (!fUsedClusS || mode<0 || mode>1 || i<0 || i>fNsingle) {tr = -1; return kFALSE;}
+ tr = fUsedClusS[i];
+ if (mode) tr >>= 16;
+ else tr &= 0xffff;
+ return (--tr)>=0; // we are storing id+1
+}
+
+//______________________________________________________________________
+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\n",fNtracks,fNsingle);
+ TString opts = opt; opts.ToLower();
+ int t0spd1=-1,t1spd1=-1,t0spd2=-1,t1spd2=-1;
+ //
+ if (opts.Contains("t")) {
+ for (int i=0;i<fNtracks;i++) {
+ GetTrackletTrackIDs(i,0,t0spd1,t0spd2);
+ GetTrackletTrackIDs(i,1,t1spd1,t1spd2);
+ printf("T#%3d| Eta:%+5.2f Th:%+6.3f Phi:%+6.3f DTh:%+6.3f DPhi:%+6.3f L1:%6d L2:%6d U:L1[%5d/%5d] L2[%5d/%5d]\n",
+ i,GetEta(i),fTh[i],fPhi[i],fDeltTh[i],fDeltPhi[i],fLabels[i],fLabelsL2[i],t0spd1,t1spd1,t0spd2,t1spd2);
+ }
+ }
+ if (opts.Contains("s")) {
+ for (int i=0;i<fNsingle;i++) {
+ GetSingleClusterTrackID(i,0,t0spd1);
+ GetSingleClusterTrackID(i,1,t1spd1);
+ printf("S#%3d| Th:%+6.3f Phi:%+6.3f L:%6d U:[%5d/%5d]\n",
+ i,fThsingle[i],fPhisingle[i],fLabelssingle[i], t0spd1,t1spd1);
+ }
+ }
+ //
+}