+// @(#) $Id$
+
// Author: Anders Vestbo <mailto:vestbo@fi.uib.no>
-//*-- Copyright © ASV
+//*-- Copyright © ALICE HLT Group
+
+/** \class AliL3Evaluate
+<pre>
+//_____________________________________________________________
+// AliL3Evaluate
+//
+// Evaluation class for tracking; plots, efficiencies etc..
+//
+</pre>
+*/
-#include <stdio.h>
-#include <math.h>
-#include <fstream.h>
+#include <TObject.h>
#include <TFile.h>
#include <TH1.h>
#include <TParticle.h>
#include <TTree.h>
#include <TClonesArray.h>
+#include <TTree.h>
+#include <TNtuple.h>
-#include "AliRun.h"
-#include "AliSimDigits.h"
-#include "AliTPC.h"
-#include "AliTPCClustersArray.h"
-#include "AliTPCClustersRow.h"
-#include "AliTPCcluster.h"
-#include "AliTPCParam.h"
-#include "AliComplexCluster.h"
+#include <AliRun.h>
+#include <AliSimDigits.h>
+#include <AliTPC.h>
+#include <AliTPCcluster.h>
+#include <AliTPCClustersArray.h>
+#include <AliTPCClustersRow.h>
+#include <AliTPCParam.h>
+#include <AliComplexCluster.h>
+#include <AliStack.h>
+#include "AliL3StandardIncludes.h"
+#include "AliL3Logging.h"
#include "AliL3Transform.h"
#include "AliL3SpacePointData.h"
#include "AliL3Track.h"
#include "AliL3FileHandler.h"
#include "AliL3TrackArray.h"
#include "AliL3Evaluate.h"
-#include "AliL3Logging.h"
-
-//_____________________________________________________________
-// AliL3Evaluate
-//
-// Evaluation class for tracking. Plots efficiencies etc..
-//
+#if __GNUC__ == 3
+#include <iosfwd>
+using namespace std;
+#endif
ClassImp(AliL3Evaluate)
AliL3Evaluate::AliL3Evaluate()
-{
- fDigitsFile = NULL;
- fTracks = NULL;
- fMCclusterfile = NULL;
- fNFastPoints = 0;
- fMcIndex = 0;
- fMcId = 0;
- fMinSlice=0;
- fMaxSlice=0;
+{
+ //constructor
+ Clear();
}
-AliL3Evaluate::AliL3Evaluate(Char_t *mcfile,Int_t *slice)
-{
- //Normal constructor. Input are the rootfile containing the
- //original MC information of the simulated event.
+AliL3Evaluate::AliL3Evaluate(Char_t *datapath,Int_t minclusters,Int_t minhits,Double_t minpt,Double_t maxpt,Int_t *slice)
+{
+ //constructor
+ Clear();
- fEventFile = new TFile(mcfile,"READ");
-
- fParam = (AliTPCParam*)fEventFile->Get("75x40_100x60");
-
- fMinSlice = slice[0];
- fMaxSlice = slice[1];
- fMinGoodPt = 0.1;
- fNoOverlap = kFALSE;
-}
-
-AliL3Evaluate::AliL3Evaluate(Int_t *slice)
-{
- //ctor to use if you do not need any rootfile.
+ if(slice)
+ {
+ fMinSlice=slice[0];
+ fMaxSlice=slice[1];
+ }
+ else
+ {
+ fMinSlice=0;
+ fMaxSlice=35;
+ }
- fMinSlice = slice[0];
- fMaxSlice = slice[1];
+ sprintf(fPath,"%s",datapath);
+ fMinPointsOnTrack = minclusters;
+ fMinHitsFromParticle = minhits;
+ fMinGoodPt = minpt;
+ fMaxGoodPt = maxpt;
}
AliL3Evaluate::~AliL3Evaluate()
-{
- if(fDigitsTree) fDigitsTree->Delete();
- if(fDigitsFile) {
- fDigitsFile->Close();
- delete fDigitsFile;
- }
+{
+ //destructor
+ if(fGoodTracks) delete[] fGoodTracks;
if(fTracks) delete fTracks;
if(fPtRes) delete fPtRes;
if(fNGoodTracksPt) delete fNGoodTracksPt;
if(fMcIndex) delete [] fMcIndex;
if(fMcId) delete [] fMcId;
if(fNtuppel) delete fNtuppel;
+ if(fNtupleRes) delete fNtupleRes;
+
+ for(Int_t s=0; s<=35; s++)
+ for(Int_t p=0; p<6; p++)
+ {
+ //cout << s << " " << p <<endl;
+ if(fClustersFile[s][p]) delete fClustersFile[s][p];
+ fClusters[s][p]=0;
+ }
}
-void AliL3Evaluate::Setup(Char_t *trackfile,Char_t *path)
-{
- //Read in the hit and track information from produced files.
- AliL3Transform::Init(path);
-
+void AliL3Evaluate::Clear()
+{
+ //clear
+ fTracks = 0;
+ fMinSlice=0;
+ fMaxSlice=0;
+ fNFastPoints = 0;
+ fMcIndex = 0;
+ fMcId = 0;
fGoodFound = 0;
fGoodGen = 0;
- Char_t fname[256];
- AliL3FileHandler *clusterfile[36][6];
+ fMinGoodPt = 0;
+ fMaxGoodPt = 0;
+ fMinPointsOnTrack = 0;
+ fMinHitsFromParticle = 0;
+ fGoodTracks = 0;
+ fMaxFalseClusters = 0.1;
+
+ fNtuppel=0;
+ fPtRes=0;
+ fNGoodTracksPt = 0;
+ fNFoundTracksPt = 0;
+ fNFakeTracksPt = 0;
+ fTrackEffPt = 0;
+ fFakeTrackEffPt = 0;
+ fNGoodTracksEta = 0;
+ fNFoundTracksEta = 0;
+ fNFakeTracksEta = 0;
+ fTrackEffEta = 0;
+ fFakeTrackEffEta = 0;
+ fNtupleRes = 0;
+
+ fStandardComparison=kTRUE;
+ for(Int_t s=0; s<=35; s++)
+ for(Int_t p=0; p<6; p++){
+ fClusters[s][p]=0;
+ fClustersFile[s][p]=0;
+ }
+ sprintf(fPath,"./");
+}
+
+void AliL3Evaluate::LoadData(Int_t event,Bool_t sp)
+{
+ //load cluster points
+ Char_t fname[1024];
+
for(Int_t s=fMinSlice; s<=fMaxSlice; s++)
{
- for(Int_t p=0; p<6; p++)
+ for(Int_t p=0; p<AliL3Transform::GetNPatches(); p++)
{
+ Int_t patch;
+ if(sp==kTRUE)
+ patch=-1;
+ else
+ patch=p;
+
+ if(fClustersFile[s][p])
+ delete fClustersFile[s][p];
fClusters[s][p] = 0;
- clusterfile[s][p] = new AliL3FileHandler();
- sprintf(fname,"%s/points_%d_%d.raw",path,s,p);
- if(!clusterfile[s][p]->SetBinaryInput(fname))
+ fClustersFile[s][p] = new AliL3FileHandler();
+ if(event<0)
+ sprintf(fname,"%s/points_%d_%d.raw",fPath,s,patch);
+ else
+ sprintf(fname,"%s/points_%d_%d_%d.raw",fPath,event,s,patch);
+ if(!fClustersFile[s][p]->SetBinaryInput(fname))
{
LOG(AliL3Log::kError,"AliL3Evaluation::Setup","File Open")
<<"Inputfile "<<fname<<" does not exist"<<ENDLOG;
- delete clusterfile[s][p];
- clusterfile[s][p] = 0;
+ delete fClustersFile[s][p];
+ fClustersFile[s][p] = 0;
continue;
}
- fClusters[s][p] = (AliL3SpacePointData*)clusterfile[s][p]->Allocate();
- clusterfile[s][p]->Binary2Memory(fNcl[s][p],fClusters[s][p]);
- clusterfile[s][p]->CloseBinaryInput();
+ fClusters[s][p] = (AliL3SpacePointData*)fClustersFile[s][p]->Allocate();
+ fClustersFile[s][p]->Binary2Memory(fNcl[s][p],fClusters[s][p]);
+ fClustersFile[s][p]->CloseBinaryInput();
+ if(sp==kTRUE)
+ break;
}
}
-
-
+
+ sprintf(fname,"%s/tracks_%d.raw",fPath,event);
AliL3FileHandler *tfile = new AliL3FileHandler();
- if(!tfile->SetBinaryInput(trackfile)){
+ if(!tfile->SetBinaryInput(fname)){
LOG(AliL3Log::kError,"AliL3Evaluation::Setup","File Open")
- <<"Inputfile "<<trackfile<<" does not exist"<<ENDLOG;
+ <<"Inputfile "<<fname<<" does not exist"<<ENDLOG;
return;
}
+ if(fTracks)
+ delete fTracks;
fTracks = new AliL3TrackArray();
tfile->Binary2TrackArray(fTracks);
tfile->CloseBinaryInput();
+ fTracks->QSort();
delete tfile;
}
-void AliL3Evaluate::SetupSlow(Char_t *trackfile,Char_t *digitsfile,Char_t *path)
-{
- //Setup for using the slow simulator.
- AliL3Transform::Init(path);
-
- fDigitsFile = new TFile(digitsfile,"READ");
- if(!fDigitsFile->IsOpen())
- {
- LOG(AliL3Log::kError,"AliL3Evaluate::SetupSlow","File Open")
- <<"Inputfile "<<digitsfile<<" does not exist"<<ENDLOG;
- return;
- }
- fIsSlow = true;
- Setup(trackfile,path);
-
- if(!InitMC())
- LOG(AliL3Log::kError,"AliL3Evaluation::SetupSlow","Digits Tree")
- <<"Error setting up digits tree"<<ENDLOG;
-
-}
-
-void AliL3Evaluate::SetupFast(Char_t *trackfile,Char_t *mcClusterfile,Char_t *path)
-{
- //Setup for using the fast simulator.
- AliL3Transform::Init(path);
-
- fIsSlow = false;
- GetFastClusterIDs(path);
-
- fMCclusterfile = new TFile(mcClusterfile);
- if(!fMCclusterfile->IsOpen())
- LOG(AliL3Log::kError,"AliL3Evaluation::SetupFast","File Open")
- <<"Inputfile "<<mcClusterfile<<" does not exist"<<ENDLOG;
-
- Setup(trackfile,path);
- InitMC();
-}
-
-Bool_t AliL3Evaluate::InitMC()
-{
- if(fIsSlow)
- {
- fDigitsFile->cd();
- fDigitsTree = (TTree*)fDigitsFile->Get("TreeD_75x40_100x60_0");
- if(!fDigitsTree)
- {
- LOG(AliL3Log::kError,"AliL3Evaluate::InitMC","Digits Tree")
- <<AliL3Log::kHex<<"Error getting digitstree "<<(Int_t)fDigitsTree<<ENDLOG;
- return false;
- }
- fDigitsTree->GetBranch("Segment")->SetAddress(&fDigits);
- for(Int_t i=0; i<fDigitsTree->GetEntries(); i++)
- {
- if(!fDigitsTree->GetEvent(i)) continue;
- Int_t se,ro,slice,slicerow;
- fParam->AdjustSectorRow(fDigits->GetID(),se,ro);
- AliL3Transform::Sector2Slice(slice,slicerow,se,ro);
- fRowid[slice][slicerow] = i;
- }
- }
-
- fEventFile->cd();
- gAlice = (AliRun*)fEventFile->Get("gAlice");
- if (!gAlice)
- {
- LOG(AliL3Log::kError,"AliL3Evaluate::InitMC","gAlice")
- <<"AliRun object non existing on file"<<ENDLOG;
- return false;
- }
-
- gAlice->GetEvent(0);
-
- return true;
-
-}
-
-void AliL3Evaluate::DefineGoodTracks(Int_t *slice,Int_t *padrow,Int_t good_number,Char_t *fname)
-{
- //Loop over MC particles, and mark the good ones
- //(which the tracker should find...)
-
-
- ifstream in(fname);
- if(in)
- {
- LOG(AliL3Log::kInformational,"AliL3Evaluate::DefineGoodTracks","infile")
- <<"Reading good particles from file "<<fname<<ENDLOG;
- while (in>>fGoodTracks[fGoodGen].label>>fGoodTracks[fGoodGen].code>>
- fGoodTracks[fGoodGen].px>>fGoodTracks[fGoodGen].py>>fGoodTracks[fGoodGen].pz>>
- fGoodTracks[fGoodGen].pt)
- {
- fGoodGen++;
- if (fGoodGen==15000)
- {
- LOG(AliL3Log::kWarning,"AliL3Evaluate::DefineGoodTracks","fGoodGen")
- <<"Too many good tracks"<<ENDLOG;
- break;
- }
- }
- if (!in.eof()) cerr<<"Read error (good_tracks_tpc) !\n";
- return;
- }
-
-
- AliTPC *TPC = (AliTPC*)gAlice->GetDetector("TPC");
-
- TPC->SetParam(fParam);
-
- Int_t ver = TPC->IsVersion();
- LOG(AliL3Log::kInformational,"AliL3Evaluate::DefineGoodTracks","TPC version")
- <<"TPC version "<<ver<<" found on file"<<ENDLOG;
-
- Int_t zero=TPC->GetParam()->GetZeroSup();
-
- Int_t np = gAlice->GetNtrack();
- Int_t *good = new Int_t[np];
- for(Int_t ii=0; ii<np; ii++)
- good[ii] = 0;
-
- if(ver==1)
- {
- if(fIsSlow)
- LOG(AliL3Log::kError,"AliL3Evaluate::DefineGoodTracks","TPC version")
- <<"TPC version "<<ver<<" does not match."<<ENDLOG;
- fMCclusterfile->cd();
- AliTPCClustersArray carray;
- carray.Setup(fParam);
- carray.SetClusterType("AliTPCcluster");
- Char_t cname[100];
- Int_t eventn = 0;
- sprintf(cname,"TreeC_TPC_%d",eventn);
- Bool_t clusterok = carray.ConnectTree(cname);
- if(!clusterok)
- LOG(AliL3Log::kError,"AliL3Evaluate::DefineGoodTracks","Cluster Array")
- <<"Error loading clusters from rootfile"<<ENDLOG;
-
- for(Int_t i=0; i<carray.GetTree()->GetEntries(); i++)
- {
- Int_t sec,row,sl,lr;
- AliSegmentID *s = carray.LoadEntry(i);
- fParam->AdjustSectorRow(s->GetID(),sec,row);
- AliL3Transform::Sector2Slice(sl,lr,sec,row);
-
- if(sl != slice[0]) {carray.ClearRow(sec,row); continue;}
- if(lr < padrow[0]) {carray.ClearRow(sec,row); continue;}
- if(lr > padrow[1]) {carray.ClearRow(sec,row); continue;}
- AliTPCClustersRow *cRow = carray.GetRow(sec,row);
- for(Int_t j=0; j<cRow->GetArray()->GetEntriesFast(); j++)
- {
- AliTPCcluster *cluster=(AliTPCcluster*)(*cRow)[j];
- Int_t lab=cluster->GetLabel(0);
- if(lab<0) continue;
- lab=TMath::Abs(lab);
- good[lab]++;
- }
- if(carray.GetRow(sec,row))
- carray.ClearRow(sec,row);
- }
- }
- else if(ver==2)
- {
- if(!fIsSlow)
- LOG(AliL3Log::kError,"AliL3Evaluate::DefineGoodTracks","TPC version")
- <<"TPC version "<<ver<<" does not match."<<ENDLOG;
- Int_t *count = new Int_t[np]; //np number of particles.
- Int_t i;
- Float_t xyz[3];
- for (i=0; i<np; i++) count[i]=0;
- for(Int_t sl=slice[0]; sl<=slice[1]; sl++)
- {
- for (i=padrow[0]; i<=padrow[1]; i++) {
- Int_t index = fRowid[sl][i];
- if (!fDigitsTree->GetEvent(index)) continue;
- Int_t sec,row;
- fParam->AdjustSectorRow(fDigits->GetID(),sec,row);
- fDigits->First();
- do {
- Int_t it=fDigits->CurrentRow(), ip=fDigits->CurrentColumn();
- Short_t dig = fDigits->GetDigit(it,ip);
-
- if(dig<=fParam->GetZeroSup()) continue;
- if(it < fParam->GetMaxTBin()-1 && it > 0)
- if(fDigits->GetDigit(it+1,ip) <= fParam->GetZeroSup()
- && fDigits->GetDigit(it-1,ip) <= fParam->GetZeroSup())
- continue;
-
- AliL3Transform::Raw2Local(xyz,sec,row,ip,it);
- if(fParam->GetPadRowRadii(sec,row)<230./250.*fabs(xyz[2]))
- continue;
-
- Int_t idx0=fDigits->GetTrackID(it,ip,0);
- Int_t idx1=fDigits->GetTrackID(it,ip,1);
- Int_t idx2=fDigits->GetTrackID(it,ip,2);
-
- if (idx0>=0 && dig>=zero) count[idx0]+=1;
- if (idx1>=0 && dig>=zero) count[idx1]+=1;
- if (idx2>=0 && dig>=zero) count[idx2]+=1;
- } while (fDigits->Next());
-
- for (Int_t j=0; j<np; j++)
- {
- if (count[j]>1) //at least two digits at this padrow
- good[j]++;
-
- count[j]=0;
- }
- }
- }
- delete[] count;
- }
-
- else
+void AliL3Evaluate::AssignPIDs()
+{
+ //assign pid
+ if(!fTracks) return;
+ fTracks->QSort();
+ LOG(AliL3Log::kDebug,"AliL3Evaluate::AssignPIDs","Track Loop")
+ <<"Assigning pid to the found tracks...."<<ENDLOG;
+ for(Int_t i=0; i<fTracks->GetNTracks(); i++)
{
- LOG(AliL3Log::kError,"AliL3Evaluation::FillEffHistos","TPC version")
- <<"No valid TPC version found"<<ENDLOG;
- return;
- }
-
-
- //The following code has been taken from offlinemacro->AliTPCComparison.C
-
- TTree *TH=gAlice->TreeH();
- Int_t npart=(Int_t)TH->GetEntries();
- Int_t max = 15000;
- while (npart--) {
- AliTPChit *hit0=0;
-
- TPC->ResetHits();
- TH->GetEvent(npart);
- AliTPChit * hit = (AliTPChit*) TPC->FirstHit(-1);
- while (hit){
- if (hit->fQ==0.) break;
- hit = (AliTPChit*) TPC->NextHit();
- }
- if (hit) {
- hit0 = new AliTPChit(*hit); //Make copy of hit
- hit = hit0;
+ AliL3Track *track = (AliL3Track*)fTracks->GetCheckedTrack(i);
+ if(!track) continue;
+ if(track->GetNumberOfPoints() < fMinPointsOnTrack)
+ track->SetPID(0);
+ else {
+ Float_t pid = GetTrackPID(track);
+ track->SetPID(pid);
+ }
}
- else continue;
- AliTPChit *hit1=(AliTPChit*)TPC->NextHit();
- if (hit1==0) continue;
- if (hit1->fQ != 0.) continue;
- Int_t i=hit->Track();
- TParticle *p = (TParticle*)gAlice->Particle(i);
-
- //printf("Checking particle %d with code %d\n",i,p->GetPdgCode());
- if (p->GetFirstMother()>=0) continue; //secondary particle
- if (good[i] < good_number || good[i]>176) continue;
- if (p->Pt()<fMinGoodPt) continue;
- if (TMath::Abs(p->Pz()/p->Pt())>0.999) continue;
- printf("Found good particle %d, nHits %d\n",i,good[i]);
-
- fGoodTracks[fGoodGen].label=i;
- fGoodTracks[fGoodGen].code=p->GetPdgCode();
- //**** px py pz - in global coordinate system, x y z - in local !
- fGoodTracks[fGoodGen].px=hit->X(); fGoodTracks[fGoodGen].py=hit->Y(); fGoodTracks[fGoodGen].pz=hit->Z();
- fGoodTracks[fGoodGen].pt = p->Pt();
- fGoodGen++;
- //nt++;
- if (hit0) delete hit0;
- if (fGoodGen==max) {cerr<<"Too many good tracks !n"; break;}
- }
-
- delete [] good;
-
- LOG(AliL3Log::kInformational,"AliL3Evaluate::DefineGoodParticles","Eval")
- <<AliL3Log::kDec<<"Found "<<fGoodGen<<"good tracks"<<ENDLOG;
-
- ofstream out(fname);
- if(out)
- {
- for (Int_t ngd=0; ngd<fGoodGen; ngd++)
- out<<fGoodTracks[ngd].label<<' '<<fGoodTracks[ngd].code<<' '<<
- fGoodTracks[ngd].px<<' '<<fGoodTracks[ngd].py<<' '<<fGoodTracks[ngd].pz<<' '<<
- fGoodTracks[ngd].pt<<endl;
-
- }
- else
- LOG(AliL3Log::kError,"AliL3Evaluate::DefineGoodParticles","file")
- <<"Can not open file containing good tracks"<<ENDLOG;
-
- out.close();
-
-}
-
-void AliL3Evaluate::EvaluatePatch(Int_t slice,Int_t patch,Int_t min_points,Int_t good_number)
-{
- //Make efficiency plots for tracking on patch level (before any merging).
-
- Int_t row[6][2] = {{ 0, 45},{46,77},{78,109},{110,141},{142,175}};
- Int_t sl[2] ={slice,slice};
- DefineGoodTracks(sl,row[patch],good_number);
- SetMinPoints(min_points);
- AssignIDs();
- CreateHistos();
- FillEffHistos();
- CalcEffHistos();
-}
-
-void AliL3Evaluate::EvaluateSlice(Int_t slice,Int_t min_points,Int_t good_number)
-{
- //Make efficiency plots for tracking on a slice (after merging).
- //min_points = minimum points on track to be considered for evaluation
- //good_number = minimum hits (padrows) produced by simulated track for consideration.
-
- Int_t row[2] = {0,175};
- Int_t sl[2] ={slice,slice};
- DefineGoodTracks(sl,row,good_number);
-
- SetMinPoints(min_points);
-
- AssignIDs();
- CreateHistos();
- FillEffHistos();
- CalcEffHistos();
}
-
-void AliL3Evaluate::EvaluateGlobal(Int_t min_points,Int_t good_number,Char_t *fname)
-{
- //Make efficiency plots for tracking on several slices.
- Int_t row[2] = {0,175};
- SetMinPoints(min_points);
- Int_t slice[2] = {fMinSlice,fMaxSlice};
- DefineGoodTracks(slice,row,good_number,fname);
- AssignIDs();
- CreateHistos(5,0,2);
- printf("filling histos\n");
- FillEffHistos();
- printf("done fillling\n");
- CalcEffHistos();
-}
-
void AliL3Evaluate::AssignIDs()
-{
+{
//Assign MC id to the tracks.
-
+#ifndef do_mc
+ cerr<<"AliL3Evaluate::AssignIDs() : You need to compile with the do_mc flag!"<<endl;
+ return;
+#endif
+ if(!fTracks) return;
+ fGoodFound=0;
fTracks->QSort();
LOG(AliL3Log::kDebug,"AliL3Evaluate::AssignIDs","Track Loop")
<<"Assigning MC id to the found tracks...."<<ENDLOG;
fGoodFound++;
Int_t tID = GetMCTrackLabel(track);
track->SetMCid(tID);
- printf("track %i id %d nHits %d\n",i,tID,track->GetNumberOfPoints());
}
+ //cout<<"Found "<<fGoodFound<<" good tracks "<<endl;
}
+Float_t AliL3Evaluate::GetTrackPID(AliL3Track *track)
+{
+ //get track pid
+ track->CalculateHelix();
+ // Track dEdx
+ Int_t nc=track->GetNHits();
+ UInt_t *hits = track->GetHitNumbers();
+ Float_t sampleDEdx[159];
+ for (Int_t iHit = 0; iHit < nc; iHit++) {
+ UInt_t hitID = hits[iHit];
+ Int_t iSector = (hitID>>25) & 0x7f;
+ Int_t patch = (hitID>>22) & 0x7;
+ UInt_t position = hitID&0x3fffff;
+ AliL3SpacePointData *points = fClusters[iSector][patch];
+ if(!points) continue;
+ if(position>=fNcl[iSector][patch])
+ {
+ LOG(AliL3Log::kError,"AliL3Evaluate::GetMCTrackLabel","Clusterarray")
+ <<AliL3Log::kDec<<"ERROR"<<ENDLOG;
+ continue;
+ }
+ UChar_t padrow = points[position].fPadRow;
+ Float_t pWidth = AliL3Transform::GetPadPitchWidthLow();
+ if (padrow>63)
+ pWidth = AliL3Transform::GetPadPitchWidthUp();
+ Float_t corr=1.; if (padrow>63) corr=0.67;
+ sampleDEdx[iHit] = points[position].fCharge/pWidth*corr;
+ Double_t crossingangle = track->GetCrossingAngle(padrow,iSector);
+ Double_t s = sin(crossingangle);
+ Double_t t = track->GetTgl();
+ sampleDEdx[iHit] *= sqrt((1-s*s)/(1+t*t));
+ }
+
+ /* Cook dEdx */
+ Int_t i;
+ Int_t swap;//stupid sorting
+ do {
+ swap=0;
+ for (i=0; i<nc-1; i++) {
+ if (sampleDEdx[i]<=sampleDEdx[i+1]) continue;
+ Float_t tmp=sampleDEdx[i];
+ sampleDEdx[i]=sampleDEdx[i+1]; sampleDEdx[i+1]=tmp;
+ swap++;
+ }
+ } while (swap);
+
+ Double_t low=0.05; Double_t up=0.7;
+ Int_t nl=Int_t(low*nc), nu=Int_t(up*nc);
+ Float_t trackDEdx=0;
+ for (i=nl; i<=nu; i++) trackDEdx += sampleDEdx[i];
+ trackDEdx /= (nu-nl+1);
+
+ // cout<<" PID: "<<nc<<" "<<nl<<" "<<nu<<" "<<trackDEdx<<" "<<track->GetPt()<<endl;
+ return trackDEdx;
+}
-struct S {Int_t lab; Int_t max;};
-Int_t AliL3Evaluate::GetMCTrackLabel(AliL3Track *track){
+Int_t AliL3Evaluate::GetMCTrackLabel(AliL3Track *track)
+{
//Returns the MCtrackID of the belonging clusters.
//If MCLabel < 0, means that track is fake.
- //Fake track means that more than 10 percent of clusters are assigned incorrectly.
+ //Definitions are identical to offline.
+ //Fake track means:
+ // - more than 10 percent of clusters are assigned incorrectly
+ // - more than half of the innermost 10% of clusters were assigned incorrectly.
+
#ifdef do_mc
- Int_t num_of_clusters = track->GetNumberOfPoints();
- S *s=new S[num_of_clusters];
+ Int_t numofclusters = track->GetNumberOfPoints();
+ AliS *s=new AliS[numofclusters];
Int_t i;
- for (i=0; i<num_of_clusters; i++) s[i].lab=s[i].max=0;
+ for (i=0; i<numofclusters; i++) s[i].flab=s[i].fmax=0;
UInt_t *hitnum = track->GetHitNumbers();
UInt_t id;
- //Int_t **trackID = GetClusterIDs(track);
-
Int_t lab=123456789;
- for (i=0; i<num_of_clusters; i++)
+ for (i=0; i<numofclusters; i++)
{
//Tricks to get the clusters belonging to this track:
id = hitnum[i];
}
//Get the label of the cluster:
-
- lab=abs(points[pos].fTrackID[0]);
+ lab=points[pos].fTrackID[0];
+
Int_t j;
- for (j=0; j<num_of_clusters; j++)
- if (s[j].lab==lab || s[j].max==0) break;
- s[j].lab=lab;
- s[j].max++;
+ for (j=0; j<numofclusters; j++)
+ if (s[j].flab==lab || s[j].fmax==0) break;
+ s[j].flab=lab;
+ s[j].fmax++;
}
Int_t max=0;
- for (i=0; i<num_of_clusters; i++)
- if (s[i].max>max) {max=s[i].max; lab=s[i].lab;}
+ for (i=0; i<numofclusters; i++)
+ if (s[i].fmax>max) {max=s[i].fmax; lab=s[i].flab;}
+
+ if(lab == -1)
+ return -1; //If most clusters is -1, this is a noise track.
+ if(lab < 0)
+ cerr<<"AliL3Evaluate::GetMCTrackLabel : Track label negative :"<<lab<<endl;
delete[] s;
- for (i=0; i<num_of_clusters; i++)
+ for (i=0; i<numofclusters; i++)
{
id = hitnum[i];
Int_t slice = (id>>25) & 0x7f;
abs(points[pos].fTrackID[2]) == lab ) max++;
}
- //check if more than 10% of the clusters are incorrectly assigned (fake track):
- if (1.-Float_t(max)/num_of_clusters > 0.10)
+
+ //Check if more than 10% of the clusters were assigned incorrectly:
+ if (1.-Float_t(max)/numofclusters > fMaxFalseClusters)
{
return -lab;
}
-
- return lab;
-#else //If we are running with mc_ids or not
- return 0;
-#endif
-
-}
-
-Int_t **AliL3Evaluate::GetClusterIDs(AliL3Track *track)
-{
- //Return the MC information of all clusters belonging to track.
-
- Int_t num_of_clusters = track->GetNumberOfPoints();
- Int_t **trackID = new Int_t*[num_of_clusters];
-
- UInt_t *hitnum = track->GetHitNumbers();
- UInt_t id;
-
- Float_t xyz[3];
- Int_t padrow;
- for(Int_t i=0; i<num_of_clusters; i++)
+ else //Check if at least half of the 10% innermost clusters are assigned correctly.
{
- id = hitnum[i];
- Int_t slice = (id>>25) & 0x7f;
- Int_t patch = (id>>22) & 0x7;
- UInt_t pos = id&0x3fffff;
-
- AliL3SpacePointData *points = fClusters[slice][patch];
-
- if(!points)
- {
- LOG(AliL3Log::kError,"AliL3Evaluate::GetClusterIDs","Clusterarray")
- <<"No points at slice "<<slice<<" patch "<<patch<<" pos "<<pos<<ENDLOG;
- continue;
- }
- if(pos>=fNcl[slice][patch])
+ Int_t tail=Int_t(0.10*numofclusters);
+ max=0;
+ for (i=1; i<=tail; i++)
{
- LOG(AliL3Log::kError,"AliL3Evaluate::GetClusterIDs","Clusterarray")
- <<AliL3Log::kDec<<"ERROR"<<ENDLOG;
- continue;
- }
-
- xyz[0] = points[pos].fX;
- xyz[1] = points[pos].fY;
- xyz[2] = points[pos].fZ;
- //sector = points[pos].fSector;
- padrow = points[pos].fPadRow;
- Int_t se,ro;
- AliL3Transform::Slice2Sector(slice,padrow,se,ro);
- AliL3Transform::Global2Raw(xyz,se,ro);
-
- if(fIsSlow)
- {
- Int_t p = fRowid[slice][padrow];
- if(!fDigitsTree->GetEvent(p))
- LOG(AliL3Log::kError,"AliL3Evaluate::GetClusterIDs","Digits Tree")
- <<"Error reading digits tree"<<ENDLOG;
+ id = hitnum[numofclusters - i];
+ Int_t slice = (id>>25) & 0x7f;
+ Int_t patch = (id>>22) & 0x7;
+ UInt_t pos = id&0x3fffff;
- trackID[i] = new Int_t[3];
- trackID[i][0] = fDigits->GetTrackID((Int_t)rint(xyz[2]),(Int_t)rint(xyz[1]),0);
- trackID[i][1] = fDigits->GetTrackID((Int_t)rint(xyz[2]),(Int_t)rint(xyz[1]),1);
- trackID[i][2] = fDigits->GetTrackID((Int_t)rint(xyz[2]),(Int_t)rint(xyz[1]),2);
- //if(trackID[i][0]==6 || trackID[i][0]==32)
- //printf("trackID %d, padrow %d pad %d time %d\n",fDigits->GetTrackID((Int_t)rint(xyz[2]),(Int_t)rint(xyz[1]),0),padrow,(int)rint(xyz[1]),(int)rint(xyz[2]));
- /*if(trackID[i][0]<0)
- {
-
- printf("trackID %d, padrow %d pad %d time %d\n",trackID[i][0],padrow,(int)rint(xyz[1]),(int)rint(xyz[2]));
- printf("on the side %d %d %d %d\n",fDigits->GetTrackID(((int)rint(xyz[2])-1),((int)rint(xyz[1])),0),fDigits->GetTrackID(((int)rint(xyz[2])+1),((int)rint(xyz[1])),0),fDigits->GetTrackID(((int)rint(xyz[2])),((int)rint(xyz[1])-1),0),fDigits->GetTrackID(((int)rint(xyz[2])),((int)rint(xyz[1])+1),0));
- }*/
- }
- else
- {
- Int_t tmp_pid=0;
- for(Int_t ii=0; ii<fNFastPoints; ii++)
- {
- tmp_pid = fMcId[ii];
- if(fMcIndex[ii] == id) break;
- }
- trackID[i] = new Int_t[3];
- trackID[i][0] = tmp_pid;
- trackID[i][1] = -1;
- trackID[i][2] = -1;
+ AliL3SpacePointData *points = fClusters[slice][patch];
+ if(!points) continue;
+ if(lab == abs(points[pos].fTrackID[0]) ||
+ lab == abs(points[pos].fTrackID[1]) ||
+ lab == abs(points[pos].fTrackID[2])) max++;
}
+ if (max < Int_t(0.5*tail)) return -lab;
}
- return trackID;
+ return lab;
+#else //If we are running with mc_ids or not
+ return 0;
+#endif
+
}
void AliL3Evaluate::GetFastClusterIDs(Char_t *path)
{
//Get the MC id of space points in case of using the fast simulator.
- char fname[256];
- sprintf(fname,"%s/point_mc.dat",path);
- FILE *infile = fopen(fname,"r");
+ char name[256];
+ sprintf(name,"%s/point_mc.dat",path);
+ FILE *infile = fopen(name,"r");
if(!infile) return;
Int_t hitid,hitmc,i;
fclose(infile);
}
-void AliL3Evaluate::CreateHistos(Int_t nbin,Int_t xlow,Int_t xup)
+void AliL3Evaluate::CreateHistos(Int_t nbin,Float_t xlow,Float_t xup)
{
//Create the histograms
<<"Creating histograms..."<<ENDLOG;
fNtuppel = new TNtuple("fNtuppel","Pt resolution","pt_gen:pt_found:nHits");
-
+ fNtuppel->SetDirectory(0);
fPtRes = new TH1F("fPtRes","Relative Pt resolution",30,-10.,10.);
fNGoodTracksPt = new TH1F("fNGoodTracksPt","Good tracks vs pt",nbin,xlow,xup);
fNFoundTracksPt = new TH1F("fNFoundTracksPt","Found tracks vs pt",nbin,xlow,xup);
fNFakeTracksEta = new TH1F("fNFakeTracksEta","Fake tracks vs eta",20,-50,50);
fTrackEffEta = new TH1F("fTrackEffEta","Tracking efficienct vs eta",20,-50,50);
fFakeTrackEffEta = new TH1F("fFakeTrackEffEta","Efficiency for fake tracks vs eta",20,-50,50);
- printf("finished creating histos\n");
+
+}
+
+void AliL3Evaluate::GetGoodParticles(Char_t *path,Int_t event,Int_t *padrowrange)
+{
+ //Read the good particles from file. This file should already have been
+ //generated by macro AliTPCComparison.C.
+
+ Char_t filename[1024];
+ if(event<0 && !padrowrange)
+ sprintf(filename,"%s/good_tracks_tpc",path);
+ else if(event>=0 && !padrowrange)
+ sprintf(filename,"%s/good_tracks_tpc_%d",path,event);
+ else
+ sprintf(filename,"%s/good_tracks_tpc_%d_%d_%d",path,event,padrowrange[0],padrowrange[1]);
+ ifstream in(filename);
+ if(!in)
+ {
+ cerr<<"AliL3Evaluate::GetGoodParticles : Problems opening file :"<<filename<<endl;
+ return;
+ }
+ Int_t MaxTracks=20000;
+ if(fGoodTracks)
+ delete [] fGoodTracks;
+ fGoodGen=0;
+ fGoodTracks = new AliGoodTrack[MaxTracks];
+
+ if(fStandardComparison){
+ while (in>>fGoodTracks[fGoodGen].flabel>>fGoodTracks[fGoodGen].fcode>>
+ fGoodTracks[fGoodGen].fpx>>fGoodTracks[fGoodGen].fpy>>fGoodTracks[fGoodGen].fpz>>
+ fGoodTracks[fGoodGen].fx>>fGoodTracks[fGoodGen].fy>>fGoodTracks[fGoodGen].fz)
+ {
+ fGoodTracks[fGoodGen].fnhits=-1;
+ fGoodTracks[fGoodGen].fsector=-1;
+ fGoodGen++;
+ if (fGoodGen==MaxTracks)
+ {
+ cerr<<"AliL3Evaluate::GetGoodParticles : Too many good tracks !\n";
+ break;
+ }
+ }
+ } else {
+ while (in>>fGoodTracks[fGoodGen].flabel>>fGoodTracks[fGoodGen].fcode>>
+ fGoodTracks[fGoodGen].fpx>>fGoodTracks[fGoodGen].fpy>>fGoodTracks[fGoodGen].fpz>>
+ fGoodTracks[fGoodGen].fx>>fGoodTracks[fGoodGen].fy >>fGoodTracks[fGoodGen].fz>>
+ fGoodTracks[fGoodGen].fnhits>>fGoodTracks[fGoodGen].fsector)
+ {
+ fGoodGen++;
+ if (fGoodGen==MaxTracks)
+ {
+ cerr<<"AliL3Evaluate::GetGoodParticles : Too many good tracks !\n";
+ break;
+ }
+ }
+ }
}
-/*
void AliL3Evaluate::FillEffHistos()
-{
- //Fill the efficiency histograms.
-
+{
+ //has to be modified for fakes.
+
+ if(!fGoodTracks)
+ {
+ cerr<<"AliL3Evaluate::FillEffHistos : No good tracks"<<endl;
+ return;
+ }
+ if(!fTracks) return;
+
+ //cout<<"Comparing "<<fGoodGen<<" good tracks ..."<<endl;
for(Int_t i=0; i<fGoodGen; i++)
{
- Double_t ptg=fGoodTracks[i].pt,pzg=fGoodTracks[i].pz;
+ //cout<<"Checking particle "<<i<<endl;
+ if(!fStandardComparison)
+ if(fGoodTracks[i].fnhits < fMinHitsFromParticle) continue;
+ Float_t ptg = TMath::Sqrt(fGoodTracks[i].fpx*fGoodTracks[i].fpx + fGoodTracks[i].fpy*fGoodTracks[i].fpy);
+ if(ptg < fMinGoodPt || ptg > fMaxGoodPt) continue;
+ Float_t pzg=fGoodTracks[i].fpz;
Float_t dipangle=TMath::ATan2(pzg,ptg)*180./TMath::Pi();
- //printf("filling particle with pt %f and dipangle %f\n",ptg,dipangle);
+
+ //If we are only considering tracks on one side of the TPC:
+ if(fMaxSlice <= 17)
+ if(dipangle < 0)
+ continue;
+
fNGoodTracksPt->Fill(ptg);
fNGoodTracksEta->Fill(dipangle);
Int_t found = 0;
+
for(Int_t k=0; k<fTracks->GetNTracks(); k++)
{
AliL3Track *track = fTracks->GetCheckedTrack(k);
if(!track) continue;
Int_t nHits = track->GetNumberOfPoints();
if(nHits < fMinPointsOnTrack) break;
-
Int_t tracklabel;
tracklabel = track->GetMCid();
- //printf("evaluating track id %d\n",tracklabel);
- if(TMath::Abs(tracklabel) != fGoodTracks[i].label) continue;
+
+ if(TMath::Abs(tracklabel) != fGoodTracks[i].flabel) continue;
found=1;
- if(tracklabel == fGoodTracks[i].label) {fNFoundTracksPt->Fill(ptg); fNFoundTracksEta->Fill(dipangle);}
- else {fNFakeTracksPt->Fill(ptg); fNFakeTracksEta->Fill(dipangle);}
Float_t pt=track->GetPt();
- fPtRes->Fill((pt-ptg)/ptg*100.);
- fNtuppel->Fill(ptg,pt,nHits);
+ if(tracklabel == fGoodTracks[i].flabel)
+ {
+ fNFoundTracksPt->Fill(ptg);
+ fNFoundTracksEta->Fill(dipangle);
+ fNtuppel->Fill(ptg,pt,nHits);
+ fPtRes->Fill((pt-ptg)/ptg*100.);
+ }
+ else
+ {
+ fNFakeTracksPt->Fill(ptg);
+ fNFakeTracksEta->Fill(dipangle);
+ }
+ //fPtRes->Fill((pt-ptg)/ptg*100.);
+ //fNtuppel->Fill(ptg,pt,nHits);
break;
}
+ //if(!found)
+ //cout<<"Track "<<fGoodTracks[i].label<<" was not found"<<endl;
}
}
-*/
-void AliL3Evaluate::FillEffHistos()
+void AliL3Evaluate::FillEffHistosNAIVE()
{
//Fill the efficiency histograms.
cout<<endl<<"Note: Doing NAIVE evaluation "<<endl;
for(Int_t i=0; i<fGoodGen; i++)
{
- Double_t ptg=fGoodTracks[i].pt,pzg=fGoodTracks[i].pz;
+ if(!fStandardComparison)
+ if(fGoodTracks[i].fnhits < fMinHitsFromParticle) continue;
+ Double_t ptg=TMath::Sqrt(fGoodTracks[i].fpx*fGoodTracks[i].fpx + fGoodTracks[i].fpy*fGoodTracks[i].fpy);
+ if(ptg < fMinGoodPt || ptg > fMaxGoodPt) continue;
+ Double_t pzg=fGoodTracks[i].fpz;
Float_t dipangle=TMath::ATan2(pzg,ptg)*180./TMath::Pi();
//printf("filling particle with pt %f and dipangle %f\n",ptg,dipangle);
fNGoodTracksPt->Fill(ptg);
if(!track) continue;
Int_t nHits = track->GetNumberOfPoints();
if(nHits < fMinPointsOnTrack) break;
- if(track->GetPt()<fMinGoodPt) continue;
+ if(track->GetPt()<fMinGoodPt || track->GetPt() > fMaxGoodPt) continue;
if(fabs(track->GetPseudoRapidity())>0.9) continue;
fNFoundTracksPt->Fill(track->GetPt()); fNFoundTracksEta->Fill(track->GetPseudoRapidity());
//Float_t pt=track->GetPt();
//fPtRes->Fill((pt-ptg)/ptg*100.);
//fNtuppel->Fill(ptg,pt,nHits);
-
}
}
-void AliL3Evaluate::CalcEffHistos(){
-
+void AliL3Evaluate::CalcEffHistos()
+{
+ //calc eff histos
+
Stat_t ngood=fNGoodTracksPt->GetEntries();
Stat_t nfound=fNFoundTracksPt->GetEntries();
Stat_t nfake=fNFakeTracksPt->GetEntries();
-
+
LOG(AliL3Log::kInformational,"AliL3Evaluate::FillEffHistos","Efficiency")
<<AliL3Log::kDec<<"There was "<<ngood<<" generated good tracks"<<ENDLOG;
LOG(AliL3Log::kInformational,"AliL3Evaluate::FillEffHistos","Efficiency")
<<AliL3Log::kDec<<"Integral efficiency is about "<<nfound/ngood*100<<ENDLOG;
LOG(AliL3Log::kInformational,"AliL3Evaluate::FillEffHistos","Efficiency")
<<AliL3Log::kDec<<"Fake tracks relative is about "<<nfake/ngood*100<<ENDLOG;
-
- LOG(AliL3Log::kInformational,"AliL3Evaluate::FillEffHistos","Efficiency")
- <<AliL3Log::kDec<<"Naive efficiency "<<(Double_t)fGoodFound/(Double_t)fGoodGen<<ENDLOG;
+ //LOG(AliL3Log::kInformational,"AliL3Evaluate::FillEffHistos","Efficiency")
+ //<<AliL3Log::kDec<<"Naive efficiency "<<(Double_t)fGoodFound/(Double_t)fGoodGen<<ENDLOG;
-
fNFoundTracksPt->Sumw2(); fNGoodTracksPt->Sumw2();
fTrackEffPt->Divide(fNFoundTracksPt,fNGoodTracksPt,1,1.,"b");
fFakeTrackEffPt->Divide(fNFakeTracksPt,fNGoodTracksPt,1,1.,"b");
fFakeTrackEffPt->SetFillStyle(3013);
fTrackEffPt->SetLineColor(4);
fFakeTrackEffPt->SetFillColor(2);
-
+
fNFoundTracksEta->Sumw2(); fNGoodTracksEta->Sumw2();
fTrackEffEta->Divide(fNFoundTracksEta,fNGoodTracksEta,1,1.,"b");
fFakeTrackEffEta->Divide(fNFakeTracksEta,fNGoodTracksEta,1,1.,"b");
fFakeTrackEffEta->SetFillStyle(3013);
fTrackEffEta->SetLineColor(4);
fFakeTrackEffEta->SetFillColor(2);
-
}
void AliL3Evaluate::Write2File(Char_t *outputfile)
{
//Write histograms to file:
- TFile *of = new TFile(outputfile,"RECREATE");
+ TFile *of = TFile::Open(outputfile,"RECREATE");
if(!of->IsOpen())
{
LOG(AliL3Log::kError,"AliL3Evaluate::Write2File","File Open")
fFakeTrackEffEta->Write();
of->Close();
- delete of;
-
}
-TNtuple *AliL3Evaluate::CalculateResiduals()
-{
-
- TNtuple *ntuppel=new TNtuple("ntuppel","Residuals","residual_trans:residual_long:zHit:pt:dipangle:beta:padrow:nHits");
+TNtuple *AliL3Evaluate::GetNtuple()
+{
+ //get ntuple
+ if(!fNtupleRes)
+ {
+ fNtupleRes = new TNtuple("ntuppel","Residuals","residual_trans:residual_long:zHit:pt:dipangle:beta:padrow:nHits");
+ fNtupleRes->SetDirectory(0);//Bug in older version of root.
+ }
+ return fNtupleRes;
+}
- for(int f=fMinSlice; f<=fMaxSlice; f++)
+void AliL3Evaluate::CalculateResiduals()
+{
+ //calculate residuals
+ TNtuple *ntuppel = GetNtuple();
+
+ for(Int_t i=0; i<fTracks->GetNTracks(); i++)
{
- AliL3FileHandler *tfile = new AliL3FileHandler();
- char fname[256];
- sprintf(fname,"tracks_tr_%d_0.raw",f);
- if(!tfile->SetBinaryInput(fname)){
- LOG(AliL3Log::kError,"AliL3Evaluation::Setup","File Open")
- <<"Inputfile "<<fname<<" does not exist"<<ENDLOG;
- return 0;
- }
- fTracks = new AliL3TrackArray();
- tfile->Binary2TrackArray(fTracks);
- tfile->CloseBinaryInput();
- delete tfile;
- printf("Looking in slice %d\n",f);
- for(Int_t i=0; i<fTracks->GetNTracks(); i++)
+
+ AliL3Track *track = (AliL3Track*)fTracks->GetCheckedTrack(i);
+ if(!track) continue;
+ if(track->GetNHits() < fMinPointsOnTrack) break;
+
+ track->CalculateHelix();
+ UInt_t *hitnum = track->GetHitNumbers();
+ UInt_t id;
+
+ Float_t xyz[3];
+ Int_t padrow;
+ for(Int_t j=0; j<track->GetNumberOfPoints()-1; j++)
{
+ id = hitnum[j];
+ Int_t slice = (id>>25) & 0x7f;
+ Int_t patch = (id>>22) & 0x7;
+ UInt_t pos = id&0x3fffff;
+
+ //if(slice<18) continue;
- AliL3Track *track = (AliL3Track*)fTracks->GetCheckedTrack(i);
- if(!track) continue;
+ AliL3SpacePointData *points = fClusters[slice][patch];
+ if(!points)
+ {
+ LOG(AliL3Log::kError,"AliL3Evaluate::CalculateResiduals","Clusterarray")
+ <<"No points at slice "<<slice<<" patch "<<patch<<" pos "<<pos<<ENDLOG;
+ continue;
+ }
+ if(pos>=fNcl[slice][patch])
+ {
+ LOG(AliL3Log::kError,"AliL3Evaluate::CalculateResiduals","Clusterarray")
+ <<AliL3Log::kDec<<"ERROR"<<ENDLOG;
+ continue;
+ }
- track->CalculateHelix();
- UInt_t *hitnum = track->GetHitNumbers();
- UInt_t id;
+ xyz[0] = points[pos].fX;
+ xyz[1] = points[pos].fY;
+ xyz[2] = points[pos].fZ;
+ padrow = points[pos].fPadRow;
+ //AliL3Transform::Global2Local(xyz,slice,kTRUE);
+ AliL3Transform::Global2LocHLT(xyz,slice);
- Float_t xyz[3];
- Int_t padrow;
- for(Int_t j=0; j<track->GetNumberOfPoints()-1; j++)
+ Float_t angle = 0;
+ AliL3Transform::Local2GlobalAngle(&angle,slice);
+ if(!track->CalculateReferencePoint(angle,AliL3Transform::Row2X(padrow)))
{
- id = hitnum[j];
- Int_t slice = (id>>25) & 0x7f;
- Int_t patch = (id>>22) & 0x7;
- UInt_t pos = id&0x3fffff;
-
- //if(slice!=1) continue;
-
- AliL3SpacePointData *points = fClusters[slice][patch];
-
- if(!points)
- {
- LOG(AliL3Log::kError,"AliL3Evaluate::CalculateResiduals","Clusterarray")
- <<"No points at slice "<<slice<<" patch "<<patch<<" pos "<<pos<<ENDLOG;
- continue;
- }
- if(pos>=fNcl[slice][patch])
- {
- LOG(AliL3Log::kError,"AliL3Evaluate::CalculateResiduals","Clusterarray")
- <<AliL3Log::kDec<<"ERROR"<<ENDLOG;
- continue;
- }
-
- xyz[0] = points[pos].fX;
- xyz[1] = points[pos].fY;
- xyz[2] = points[pos].fZ;
- padrow = points[pos].fPadRow;
- //AliL3Transform::Global2Local(xyz,slice);
-
- Float_t xyz_cross[3];
- track->GetCrossingPoint(padrow,xyz_cross);
- Double_t beta = track->GetCrossingAngle(padrow);
-
- Double_t yres = xyz_cross[1] - xyz[1];
- Double_t zres = xyz_cross[2] - xyz[2];
-
- Double_t dipangle = atan(track->GetTgl());
- ntuppel->Fill(yres,zres,xyz_cross[2],track->GetPt(),dipangle,beta,padrow,track->GetNumberOfPoints());
-
+ LOG(AliL3Log::kError,"AliL3Evaluate::CalculateResiduals","Crossing point")
+ <<"Track does not crossing padrow "<<padrow<<" in slice "<<slice<<ENDLOG;
+ continue;
}
+
+ Float_t xyzcross[3] = {track->GetPointX(),track->GetPointY(),track->GetPointZ()};
+ //AliL3Transform::Global2Local(xyzcross,slice,kTRUE);
+ AliL3Transform::Global2LocHLT(xyzcross,slice);
+
+ Double_t beta = track->GetCrossingAngle(padrow,slice);
+
+ Double_t yres = xyzcross[1] - xyz[1];
+ Double_t zres = xyzcross[2] - xyz[2];
+ Double_t dipangle = atan(track->GetTgl());
+ ntuppel->Fill(yres,zres,xyzcross[2],track->GetPt(),dipangle,beta,padrow,track->GetNumberOfPoints());
+
}
- if(fTracks)
- delete fTracks;
}
- return ntuppel;
}
-TNtuple *AliL3Evaluate::EvaluatePoints(Char_t *rootfile)
+enum tagprimary {kPrimaryCharged = 0x4000};
+void AliL3Evaluate::EvaluatePoints(Char_t *rootfile,Char_t *exactfile,Char_t *tofile,Int_t nevent,Bool_t offline,Bool_t sp)
{
//Compare points to the exact crossing points of track and padrows.
//The input file to this function, contains the exact clusters calculated
//in AliTPC::Hits2ExactClusters.
-
+
+#ifndef do_mc
+ cerr<<"AliL3Evaluate::EvaluatePoints : Compile with do_mc flag!"<<endl;
+ return;
+#else
cout<<"Evaluating points"<<endl;
- TNtuple *ntuppel = new TNtuple("ntuppel","residuals","slice:padrow:resy:resz:zHit:pt");
+ TNtuple *ntuppel = new TNtuple("ntuppel_res","Cluster properties",
+ "slice:padrow:charge:resy:resz:zHit:pt:beta:sigmaY2:sigmaZ2:psigmaY2:psigmaZ2");
+ ntuppel->SetDirectory(0);
+ TNtuple *ntuppel2 = new TNtuple("ntuppel_eff","Efficiency","slice:padrow:nfound:ngen");
+ ntuppel2->SetDirectory(0);
+
TFile *exfile = TFile::Open(rootfile);
if(!exfile)
{
cerr<<"Error opening rootfile "<<rootfile<<endl;
- return 0;
+ return;
+ }
+ gAlice = (AliRun*)exfile->Get("gAlice");
+ if (!gAlice)
+ {
+ LOG(AliL3Log::kError,"AliL3Evaluate::InitMC","gAlice")
+ <<"AliRun object non existing on file"<<ENDLOG;
+ return;
}
- AliTPCParam *param = (AliTPCParam*)exfile->Get("75x40_100x60");
+ AliTPCParam *param = (AliTPCParam*)exfile->Get(AliL3Transform::GetParamName());
- //Get the exact clusters from file:
- AliTPCClustersArray *arr = new AliTPCClustersArray;
- arr->Setup(param);
- arr->SetClusterType("AliComplexCluster");
- char treeName[500];
- sprintf(treeName,"TreeCExact_%s",param->GetTitle());
- Bool_t clusterok = arr->ConnectTree(treeName);//Segment Tree (for offline clusters)
- if(!clusterok) {printf("AliL3Evaluate::EvaluatePoints : Error in clusterloading\n"); return 0;}
+ TFile *exact = TFile::Open(exactfile);
+ if(!exact)
+ {
+ cerr<<"AliL3Evaluate::EvaluatePoints : Problems opening file :"<<exactfile<<endl;
+ return;
+ }
- cout<<"Entering loop with "<<(Int_t)arr->GetTree()->GetEntries()<<endl;
- for(Int_t i=0; i<arr->GetTree()->GetEntries(); i++)
+ AliStack *astack=gAlice->Stack();
+
+ AliTPCClustersArray *arr=0;
+ for(Int_t event=0; event<nevent; event++)
{
- //Get the exact clusters for this row:
- Int_t cursec,currow;
- AliSegmentID *s = arr->LoadEntry(i);
- param->AdjustSectorRow(s->GetID(),cursec,currow);
+ LoadData(event,sp);
+ exfile->cd();
+ if(arr)
+ delete arr;
+ Int_t nparticles = gAlice->GetEvent(event);
+ Int_t nprimaries = 0;//FindPrimaries(nparticles);
+ cout<<"Event "<<event<<" had "<<nparticles<<" particles and "<<nprimaries<<" primaries"<<endl;
+ exact->cd();
- AliTPCClustersRow *ro = (AliTPCClustersRow *)arr->GetRow(cursec,currow);
- TClonesArray *clusters = ro->GetArray();
- int num_of_offline=clusters->GetEntriesFast();
+ //Get the exact clusters from file:
+ AliTPCClustersArray *arr = new AliTPCClustersArray;
+ arr->Setup(param);
+ arr->SetClusterType("AliComplexCluster");
+ char treeName[500];
+ sprintf(treeName,"TreeCExact_%s_%d",param->GetTitle(),event);
+ Bool_t clusterok = arr->ConnectTree(treeName);//Segment Tree (for offline clusters)
+ if(!clusterok) {printf("AliL3Evaluate::EvaluatePoints : Error in clusterloading\n"); return;}
- //Get the found clusters:
- Int_t slice,padrow;
- AliL3Transform::Sector2Slice(slice,padrow,cursec,currow);
- if(slice<fMinSlice || slice>fMaxSlice) continue;
- AliL3SpacePointData *points = fClusters[slice][0];
-
- Int_t index = fRowid[slice][padrow];
- if(!fDigitsTree->GetEvent(index))
- printf("AliL3Evaluate::EvaluatePoints : ERROR IN DIGITSTREE\n");
- printf("Checking slice %d padrow %d with %d clusters\n",slice,padrow,num_of_offline);
-
- for(UInt_t c=0; c<fNcl[slice][0]; c++)
+ //cout<<"Entering loop with "<<(Int_t)arr->GetTree()->GetEntries()<<endl;
+ for(Int_t i=0; i<arr->GetTree()->GetEntries(); i++)
{
- if(points[c].fPadRow!=padrow) continue;
- for(Int_t m=0; m<num_of_offline; m++)
+ //Get the exact clusters for this row:
+ Int_t cursec,currow;
+ AliSegmentID *s = arr->LoadEntry(i);
+ param->AdjustSectorRow(s->GetID(),cursec,currow);
+
+ AliTPCClustersRow *ro = (AliTPCClustersRow *)arr->GetRow(cursec,currow);
+ TClonesArray *clusters = ro->GetArray();
+ int numofoffline=clusters->GetEntriesFast();
+
+ //Get the found clusters:
+ Int_t slice,padrow;
+ AliL3Transform::Sector2Slice(slice,padrow,cursec,currow);
+ if(slice < fMinSlice) continue;
+ if(slice > fMaxSlice) break;
+
+ Int_t patch = AliL3Transform::GetPatch(padrow);
+ if(sp)
+ patch=0;
+ AliL3SpacePointData *points = fClusters[slice][patch];
+ if(!points)
+ continue;
+
+ //cout<<"Slice "<<slice<<" padrow "<<padrow<<" has "<<numofoffline<<" clusters "<<endl;
+ Int_t clustercount=0;
+ Int_t crosscount=0;
+ for(Int_t m=0; m<numofoffline; m++)
{
AliComplexCluster *cluster = (AliComplexCluster *)clusters->UncheckedAt(m);
+#ifdef use_newio
+ Int_t mcId = cluster->GetTrack(0);
+#else
Int_t mcId = cluster->fTracks[0];
- //Int_t mcId = cluster->GetLabel(0);
+#endif
if(mcId <0) continue;
- TParticle *part = gAlice->Particle(mcId);
+
+#ifdef use_newio
+ if(cluster->GetY() < 1 || cluster->GetY() > AliL3Transform::GetNPads(padrow) - 2 ||
+ cluster->GetX() < 1 || cluster->GetX() > AliL3Transform::GetNTimeBins() - 2)
+ continue;
+#else
+ if(cluster->fY < 1 || cluster->fY > AliL3Transform::GetNPads(padrow) - 2 ||
+ cluster->fX < 1 || cluster->fX > AliL3Transform::GetNTimeBins() - 2)
+ continue;
+#endif
+ Float_t xyzex[3];
+
+#ifdef use_newio
+ AliL3Transform::Raw2Local(xyzex,cursec,currow,cluster->GetY(),cluster->GetX());
+#else
+ AliL3Transform::Raw2Local(xyzex,cursec,currow,cluster->fY,cluster->fX);
+#endif
+ //In function AliTPC::Hits2ExactClusters the time offset is not included,
+ //so we have to substract it again here.
+ if(slice<18)
+ xyzex[2]-=AliL3Transform::GetZOffset();
+ else
+ xyzex[2]+=AliL3Transform::GetZOffset();
- Float_t xyz_cl[3] = {points[c].fX,points[c].fY,points[c].fZ};
- Float_t xyz_ex[3];
- AliL3Transform::Global2Raw(xyz_cl,cursec,currow);
- if(fDigits->GetTrackID((Int_t)rint(xyz_cl[2]),(Int_t)rint(xyz_cl[1]),0)!=mcId &&
- fDigits->GetTrackID((Int_t)rint(xyz_cl[2]),(Int_t)rint(xyz_cl[1]),1)!=mcId &&
- fDigits->GetTrackID((Int_t)rint(xyz_cl[2]),(Int_t)rint(xyz_cl[1]),2)!=mcId)
+ //Outside our cone:
+ if(param->GetPadRowRadii(cursec,currow)<230./250.*fabs(xyzex[2]))
continue;
- AliL3Transform::Raw2Local(xyz_ex,cursec,currow,cluster->fY,cluster->fX);
- AliL3Transform::Raw2Local(xyz_cl,cursec,currow,xyz_cl[1],xyz_cl[2]);
- Float_t resy = xyz_cl[1] - xyz_ex[1];//cluster->GetY()
- Float_t resz = xyz_cl[2] - xyz_ex[2];//cluster->GetZ()
- ntuppel->Fill(slice,padrow,resy,resz,xyz_ex[2],part->Pt());
+ TParticle *part = astack->Particle(mcId);
+ crosscount++;
+
+ if(part->Pt() < fMinGoodPt) continue;
+
+ //Dont take secondaries, because in width calculation we assume primaries:
+ //if(!(part->TestBit(kPrimaryCharged))) continue;
+ if(part->GetFirstMother()>=0) continue;
+
+ Int_t tempcount=0;
+ for(UInt_t c=0; c<fNcl[slice][patch]; c++)
+ {
+ if((Int_t)points[c].fPadRow!=padrow) continue;
+ Float_t xyzcl[3] = {points[c].fX,points[c].fY,points[c].fZ};
+
+ if(!offline)
+ AliL3Transform::Global2Local(xyzcl,cursec);
+ tempcount++;
+
+ if(points[c].fTrackID[0] != mcId &&
+ points[c].fTrackID[1] != mcId &&
+ points[c].fTrackID[2] != mcId)
+ continue;
+
+ //Residuals:
+ Float_t resy = xyzcl[1] - xyzex[1];
+ Float_t resz = xyzcl[2] - xyzex[2];
+
+ //Cluster shape
+ Int_t charge = (Int_t)points[c].fCharge;
+ Float_t beta = GetCrossingAngle(part,slice,padrow,xyzex);
+ Double_t tanl = xyzex[2]/sqrt(xyzex[0]*xyzex[0]+xyzex[1]*xyzex[1]);
+ Float_t psigmaY2 = AliL3Transform::GetParSigmaY2(padrow,xyzex[2],beta);
+ Float_t psigmaZ2 = AliL3Transform::GetParSigmaZ2(padrow,xyzex[2],tanl);
+ Float_t sigmaY2 = points[c].fSigmaY2;
+ Float_t sigmaZ2 = points[c].fSigmaZ2;
+ ntuppel->Fill(slice,padrow,charge,resy,resz,xyzex[2],part->Pt(),beta,sigmaY2,sigmaZ2,psigmaY2,psigmaZ2);
+ }
+ clustercount=tempcount;
}
- }
- arr->ClearRow(cursec,currow);
+ ntuppel2->Fill(slice,padrow,clustercount,crosscount);
+ arr->ClearRow(cursec,currow);
+ }
}
+ exfile->Close();
+ exact->Close();
- return ntuppel;
+ TFile *ofile = TFile::Open(tofile,"RECREATE");
+ ntuppel->Write();
+ ntuppel2->Write();
+ ofile->Close();
+#endif
}
-void AliL3Evaluate::GetCFeff(Char_t *outfile)
+void AliL3Evaluate::GetCFeff(Char_t *path,Char_t *outfile,Int_t nevent,Bool_t sp)
{
+ //Evaluate the cluster finder efficiency.
+
+#ifndef do_mc
+ cerr<<"AliL3Evaluate::GetCFeff : Compile with do_mc flag"<<endl;
+ return;
+#else
+ TNtuple *ntuppel = new TNtuple("ntuppel","Cluster finder efficiency","slice:row:ncrossings:nclusters");
+ ntuppel->SetDirectory(0);
+
+ Char_t filename[1024];
+ sprintf(filename,"%s/alirunfile.root",path);
+ TFile *rfile = TFile::Open(filename);
+ gAlice = (AliRun*)rfile->Get("gAlice");
+
+ AliStack *astack=gAlice->Stack();
- TNtuple *ntuppel = new TNtuple("ntuppel","Cluster finder efficiency","row:ncrossings:nclusters");
-
- AliTPC *TPC = (AliTPC*)gAlice->GetDetector("TPC");
-
- TPC->SetParam(fParam);
-
- Int_t ver = TPC->IsVersion();
- LOG(AliL3Log::kInformational,"AliL3Evaluate::GetCFeff","TPC version")
- <<"TPC version "<<ver<<" found on file"<<ENDLOG;
-
- Int_t zero=TPC->GetParam()->GetZeroSup();
-
- Int_t np = gAlice->GetNtrack();
+ AliTPCParam *param = (AliTPCParam*)rfile->Get(AliL3Transform::GetParamName());
+
+ Int_t zero=param->GetZeroSup();
+
+ sprintf(filename,"%s/digitfile.root",path);
+ TFile *dfile = TFile::Open(filename);
- Int_t crossed,recs;
- Int_t *count = new Int_t[np]; //np number of particles.
- Int_t i;
- Float_t xyz[3];
- for (i=0; i<np; i++) count[i]=0;
- for(Int_t sl=fMinSlice; sl<=fMaxSlice; sl++)
+ for(Int_t event=0; event<nevent; event++)
{
- for (i=0; i<=175; i++)
+ LoadData(event,sp);
+ rfile->cd();
+ gAlice->GetEvent(event);
+ Int_t np = astack->GetNtrack();
+ cout<<"Processing event "<<event<<" with "<<np<<" particles "<<endl;
+ dfile->cd();
+ sprintf(filename,"TreeD_75x40_100x60_150x60_%d",event);
+ TTree *TD=(TTree*)gDirectory->Get(filename);
+ AliSimDigits da, *digits=&da;
+ TD->GetBranch("Segment")->SetAddress(&digits);
+
+ Int_t crossed=0,recs=0;
+ Int_t *count = new Int_t[np]; //np number of particles.
+ Int_t i;
+ Float_t xyz[3];
+ for (i=0; i<np; i++) count[i]=0;
+
+
+ Int_t sec,row,sl,sr;
+ for(Int_t i=0; i<(Int_t)TD->GetEntries(); i++)
{
- crossed=0;
- recs=0;
- Int_t index = fRowid[sl][i];
- if (!fDigitsTree->GetEvent(index)) continue;
- Int_t sec,row;
- fParam->AdjustSectorRow(fDigits->GetID(),sec,row);
- fDigits->First();
+ crossed=recs=0;
+ if (!TD->GetEvent(i)) continue;
+ param->AdjustSectorRow(digits->GetID(),sec,row);
+ AliL3Transform::Sector2Slice(sl,sr,sec,row);
+ if(sl < fMinSlice) continue;
+ if(sl > fMaxSlice) break;
+ cout<<"Processing slice "<<sl<<" row "<<sr<<endl;
+ digits->First();
do {
- Int_t it=fDigits->CurrentRow(), ip=fDigits->CurrentColumn();
- Short_t dig = fDigits->GetDigit(it,ip);
-
- if(dig<=fParam->GetZeroSup()) continue;
- if(it < fParam->GetMaxTBin()-1 && it > 0)
- if(fDigits->GetDigit(it+1,ip) <= fParam->GetZeroSup()
- && fDigits->GetDigit(it-1,ip) <= fParam->GetZeroSup())
- continue;
+ Int_t it=digits->CurrentRow(), ip=digits->CurrentColumn();
+ Short_t dig = digits->GetDigit(it,ip);
+ if(dig<=param->GetZeroSup()) continue;
AliL3Transform::Raw2Local(xyz,sec,row,ip,it);
- if(fParam->GetPadRowRadii(sec,row)<230./250.*fabs(xyz[2]))
+ if(param->GetPadRowRadii(sec,row)<230./250.*fabs(xyz[2]))
continue;
-
- Int_t idx0=fDigits->GetTrackID(it,ip,0);
- Int_t idx1=fDigits->GetTrackID(it,ip,1);
- Int_t idx2=fDigits->GetTrackID(it,ip,2);
+ Int_t idx0=digits->GetTrackID(it,ip,0);
+ Int_t idx1=digits->GetTrackID(it,ip,1);
+ Int_t idx2=digits->GetTrackID(it,ip,2);
if (idx0>=0 && dig>=zero) count[idx0]+=1;
if (idx1>=0 && dig>=zero) count[idx1]+=1;
if (idx2>=0 && dig>=zero) count[idx2]+=1;
- } while (fDigits->Next());
-
+ } while (digits->Next());
for (Int_t j=0; j<np; j++)
{
+ TParticle *part = astack->Particle(j);
+ if(part->Pt() < fMinGoodPt) continue;
+ if(part->GetFirstMother() >= 0) continue;
if (count[j]>1) //at least two digits at this padrow
{
crossed++;
count[j]=0;
}
}
- AliL3SpacePointData *points = fClusters[sl][0];
- for(UInt_t k=0; k<fNcl[sl][0]; k++)
+
+ Int_t patch = AliL3Transform::GetPatch(sr);
+ if(sp==kTRUE)
+ patch=0;
+ AliL3SpacePointData *points = fClusters[sl][patch];
+ if(!points)
+ continue;
+ for(UInt_t k=0; k<fNcl[sl][patch]; k++)
{
- if(points[k].fPadRow!=i) continue;
+ if(points[k].fPadRow!=sr) continue;
recs++;
}
- ntuppel->Fill(i,crossed,recs);
+ ntuppel->Fill(sl,sr,crossed,recs);
}
+ TD->Delete();
+ delete[] count;
}
- delete[] count;
-
TFile *file = TFile::Open(outfile,"RECREATE");
ntuppel->Write();
file->Close();
+ rfile->Close();
+ dfile->Close();
+#endif
}
-Bool_t AliL3Evaluate::GetParticleCrossingPoint(TParticle *part,Int_t slice,Int_t padrow,Float_t *xyz)
+Float_t AliL3Evaluate::GetCrossingAngle(TParticle *part,Int_t slice,Int_t /*padrow*/,Float_t *xyz)
{
- //Calcluate the crossing point between a generated particle and given padrow.
+ //Calculate the padrow crossing angle of the particle
- Double_t Pi = AliL3Transform::Pi();
- Double_t kappa = AliL3Transform::GetBField()*0.0029980/part->Pt();
+ Double_t kappa = AliL3Transform::GetBField()*AliL3Transform::GetBFact()/part->Pt();
Double_t radius = 1/fabs(kappa);
if(part->GetPdgCode() > 0) kappa = -kappa;
-
- Int_t charg;
- if(kappa>0)
- charg=-1;
- else
- charg=1;
-
+
Float_t angl[1] = {part->Phi()};
AliL3Transform::Global2LocalAngle(angl,slice);
Double_t charge = -1.*kappa;
+
Double_t trackPhi0 = angl[0] + charge*0.5*AliL3Transform::Pi()/fabs(charge);
Double_t x0=0;
Double_t y0=0;
Double_t xc = x0 - radius * cos(trackPhi0);
Double_t yc = y0 - radius * sin(trackPhi0);
-
- //printf("radius %f xc %f yc %f\n",radius,xc,yc);
-
- Double_t xHit = AliL3Transform::Row2X(padrow);
- xyz[0] = xHit;
- Double_t aa = (xHit - xc)*(xHit - xc);
- Double_t r2 = radius*radius;
- if(aa > r2)
- return false;
- Double_t aa2 = sqrt(r2 - aa);
- Double_t y1 = yc + aa2;
- Double_t y2 = yc - aa2;
- xyz[1] = y1;
- if(fabs(y2) < fabs(y1)) xyz[1] = y2;
-
- Double_t tgl = part->Pz()/part->Pt();
+ Double_t tangent[2];
+ tangent[0] = -1.*(xyz[1] - yc)/radius;
+ tangent[1] = (xyz[0] - xc)/radius;
+
+ Double_t perppadrow[2] = {1,0}; //locally in slice
+
+ Double_t cosbeta = fabs(tangent[0]*perppadrow[0] + tangent[1]*perppadrow[1]);
+ if(cosbeta > 1) cosbeta=1;
+ return acos(cosbeta);
+}
- Double_t yHit = xyz[1];
- Double_t angle1 = atan2((yHit - yc),(xHit - xc));
- if(angle1 < 0) angle1 += 2.*AliL3Transform::Pi();
- Double_t angle2 = atan2((0 - yc),(0 - xc));
- if(angle2 < 0) angle2 += 2.*AliL3Transform::Pi();
- Double_t diff_angle = angle1 - angle2;
- diff_angle = fmod(diff_angle,2*AliL3Transform::Pi());
- if((charg*diff_angle) > 0) diff_angle = diff_angle - charg*2.*AliL3Transform::Pi();
- Double_t s_tot = fabs(diff_angle)*radius;
- Double_t zHit = 0 + s_tot*tgl;
- xyz[2] = zHit;
+Int_t AliL3Evaluate::FindPrimaries(Int_t nparticles)
+{
+ // cuts:
+ Double_t vertcut = 0.001;
+ Double_t decacut = 3.;
+ Double_t timecut = 0.;
+ Int_t nprch1=0;
+ AliStack *astack=gAlice->Stack();
+ TParticle * part = astack->Particle(0);
+ Double_t xori = part->Vx();
+ Double_t yori = part->Vy();
+ Double_t zori = part->Vz();
+ for(Int_t iprim = 0; iprim<nparticles; iprim++){ //loop on tracks
+
+ part = astack->Particle(iprim);
+ const char * xxx=strstr(part->GetName(),"XXX");
+ if(xxx)continue;
+
+ TParticlePDG *ppdg = part->GetPDG();
+ if(TMath::Abs(ppdg->Charge())!=3)continue; // only charged (no quarks)
+
+ Double_t dist=TMath::Sqrt((part->Vx()-xori)*(part->Vx()-xori)+(part->Vy()-yori)*(part->Vy()-yori)+(part->Vz()-zori)*(part->Vz()-zori));
+ if(dist>vertcut)continue; // cut on the vertex
+
+ if(part->T()>timecut)continue;
+
+ Double_t ptot=TMath::Sqrt(part->Px()*part->Px()+part->Py()*part->Py()+part->Pz()*part->Pz());
+ if(ptot==(TMath::Abs(part->Pz())))continue; // no beam particles
+
+ Bool_t prmch = kTRUE; // candidate primary track
+ Int_t fidau=part->GetFirstDaughter(); // cut on daughters
+ Int_t lasdau=0;
+ Int_t ndau=0;
+ if(fidau>=0){
+ lasdau=part->GetLastDaughter();
+ ndau=lasdau-fidau+1;
+ }
+ if(ndau>0){
+ for(Int_t j=fidau;j<=lasdau;j++){
+ TParticle *dau=astack->Particle(j);
+ Double_t distd=TMath::Sqrt((dau->Vx()-xori)*(dau->Vx()-xori)+(dau->Vy()-yori)*(dau->Vy()-yori)+(dau->Vz()-zori)*(dau->Vz()-zori));
+ if(distd<decacut)prmch=kFALSE; // eliminate if the decay is near the vertex
+ }
+ }
+
+ if(prmch){
+ nprch1++;
+ part->SetBit(kPrimaryCharged);
+ }
+ }
- return true;
+ return nprch1;
}
-
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff