#include <TObjArray.h>
#include <TObject.h>
#include <TString.h>
-#include <TH1F.h>
+#include <TH1S.h>
+#include <TPad.h>
#include <TFile.h>
#include <TTree.h>
#include <TROOT.h>
#include "AliLog.h"
#include "AliAnalysisManager.h"
+#include "AliCDBManager.h"
#include "AliESDEvent.h"
#include "AliMCEvent.h"
#include "AliESDInputHandler.h"
#include "AliESDfriendTrack.h"
#include "AliESDHeader.h"
#include "AliESDtrack.h"
+#include "AliESDv0.h"
#include "AliESDtrackCuts.h"
#include "AliMCParticle.h"
#include "AliPID.h"
#include "AliStack.h"
-#include "AliTRDtrackV1.h"
#include "AliTrackReference.h"
-#include "AliTRDgeometry.h"
-#include "AliTRDcluster.h"
-#include "AliTRDseedV1.h"
#include "TTreeStream.h"
#include <cstdio>
#include <cstring>
#include <iostream>
+#include "AliTRDcalibDB.h"
+#include "AliTRDtrackerV1.h"
+#include "AliTRDgeometry.h"
+#include "AliTRDtrackV1.h"
+#include "AliTRDseedV1.h"
+#include "AliTRDcluster.h"
#include "AliTRDinfoGen.h"
#include "info/AliTRDtrackInfo.h"
#include "info/AliTRDeventInfo.h"
ClassImp(AliTRDinfoGen)
+const Float_t AliTRDinfoGen::fgkITS = 100.; // to be checked
const Float_t AliTRDinfoGen::fgkTPC = 290.;
-const Float_t AliTRDinfoGen::fgkTOF = 365.;
+const Float_t AliTRDinfoGen::fgkTRD = 365.;
const Float_t AliTRDinfoGen::fgkEvVertexZ = 15.;
const Int_t AliTRDinfoGen::fgkEvVertexN = 1;
-const Float_t AliTRDinfoGen::fgkTrkDCAxy = 40.;
-const Float_t AliTRDinfoGen::fgkTrkDCAz = 15.;
-const Int_t AliTRDinfoGen::fgkNclTPC = 10;
+const Float_t AliTRDinfoGen::fgkTrkDCAxy = 3.;
+const Float_t AliTRDinfoGen::fgkTrkDCAz = 10.;
+const Int_t AliTRDinfoGen::fgkNclTPC = 70;
const Float_t AliTRDinfoGen::fgkPt = 0.2;
const Float_t AliTRDinfoGen::fgkEta = 0.9;
,fMCev(NULL)
,fEventCut(NULL)
,fTrackCut(NULL)
+ ,fV0Cut(NULL)
+ ,fOCDB("local://$ALICE_ROOT/OCDB")
,fTrackInfo(NULL)
,fEventInfo(NULL)
,fV0Info(NULL)
,fTracksSA(NULL)
,fTracksKink(NULL)
,fV0List(NULL)
+ ,fContainer(NULL)
,fDebugStream(NULL)
{
//
// Default constructor
//
- SetNameTitle("infoGen", "MC-REC TRD-track list generator");
+ SetNameTitle("TRDinfoGen", "MC-REC TRD-track list generator");
}
//____________________________________________________________________
,fMCev(NULL)
,fEventCut(NULL)
,fTrackCut(NULL)
+ ,fV0Cut(NULL)
+ ,fOCDB("local://$ALICE_ROOT/OCDB")
,fTrackInfo(NULL)
,fEventInfo(NULL)
,fV0Info(NULL)
,fTracksSA(NULL)
,fTracksKink(NULL)
,fV0List(NULL)
+ ,fContainer(NULL)
,fDebugStream(NULL)
{
//
DefineOutput(kTracksKink, TObjArray::Class());
DefineOutput(kEventInfo, AliTRDeventInfo::Class());
DefineOutput(kV0List, TObjArray::Class());
+ DefineOutput(kMonitor, TObjArray::Class()); // histogram list
}
//____________________________________________________________________
{
// Destructor
+ if(fDebugStream) delete fDebugStream;
if(fEvTrigger) delete fEvTrigger;
+ if(fV0Cut) delete fV0Cut;
if(fTrackCut) delete fTrackCut;
if(fEventCut) delete fEventCut;
if(fTrackInfo) delete fTrackInfo; fTrackInfo = NULL;
fTracksKink = NULL;
}
if(fV0List){
- fV0List->Delete(); delete fV0List;
+ fV0List->Delete();
+ delete fV0List;
fV0List = NULL;
}
+ if(fContainer){
+ fContainer->Delete();
+ delete fContainer;
+ fContainer = NULL;
+ }
+}
+
+//____________________________________________________________________
+Bool_t AliTRDinfoGen::GetRefFigure(Int_t)
+{
+ if(!gPad){
+ AliWarning("Please provide a canvas to draw results.");
+ return kFALSE;
+ }
+ fContainer->At(0)->Draw("bar");
+ return kTRUE;
}
//____________________________________________________________________
fTracksSA = new TObjArray(20); fTracksSA->SetOwner(kTRUE);
fTracksKink = new TObjArray(20); fTracksKink->SetOwner(kTRUE);
fV0List = new TObjArray(10); fV0List->SetOwner(kTRUE);
+
+ // define general monitor
+ fContainer = new TObjArray(1); fContainer->SetOwner(kTRUE);
+ TH1 *h=new TH1S("hStat", "Run statistics;Observable;Entries", 12, -0.5, 11.5);
+ TAxis *ax(h->GetXaxis());
+ ax->SetBinLabel( 1, "ESD");
+ ax->SetBinLabel( 2, "MC");
+ ax->SetBinLabel( 3, "V0");
+ ax->SetBinLabel( 4, "TPC");
+ ax->SetBinLabel( 5, "TRDin");
+ ax->SetBinLabel( 6, "TRDout");
+ ax->SetBinLabel( 7, "Barrel");
+ ax->SetBinLabel( 8, "BarrelMC");
+ ax->SetBinLabel( 9, "SA");
+ ax->SetBinLabel(10, "SAMC");
+ ax->SetBinLabel(11, "Kink");
+ ax->SetBinLabel(12, "KinkMC");
+ fContainer->AddAt(h, 0);
+ PostData(kMonitor, fContainer);
+}
+
+//____________________________________________________________________
+Bool_t AliTRDinfoGen::Load(const Char_t *file, const Char_t *dir, const Char_t *name)
+{
+// Load data from performance file
+
+ if(!TFile::Open(file)){
+ AliWarning(Form("Couldn't open file %s.", file));
+ return kFALSE;
+ }
+ if(dir){
+ if(!gFile->cd(dir)){
+ AliWarning(Form("Couldn't cd to %s in %s.", dir, file));
+ return kFALSE;
+ }
+ }
+ TObjArray *o(NULL);
+ const Char_t *tn=(name ? name : GetName());
+ if(!(o = (TObjArray*)gDirectory->Get(tn))){
+ AliWarning(Form("Missing histogram container %s.", tn));
+ return kFALSE;
+ }
+ fContainer = (TObjArray*)o->Clone(GetName());
+ gFile->Close();
+ return kTRUE;
}
//____________________________________________________________________
//
fESDev = dynamic_cast<AliESDEvent*>(InputEvent());
- fMCev = MCEvent();
-
if(!fESDev){
AliError("Failed retrieving ESD event");
return;
}
+ if(!IsInitOCDB()){
+ // prepare OCDB access
+ AliCDBManager* ocdb = AliCDBManager::Instance();
+ ocdb->SetDefaultStorage(fOCDB.Data());
+ ocdb->SetRun(fESDev->GetRunNumber());
+ AliTRDtrackerV1::SetNTimeBins(AliTRDcalibDB::Instance()->GetNumberOfTimeBinsDCS());
+ AliInfo(Form("OCDB : Loc[%s] Run[%d] TB[%d]", fOCDB.Data(), ocdb->GetRun(), AliTRDtrackerV1::GetNTimeBins()));
+ SetInitOCDB();
+ }
+
+ // link MC if available
+ fMCev = MCEvent();
// event selection : trigger cut
if(UseLocalEvSelection() && fEvTrigger){
break;
}
}
- if(!kTRIGGERED) return;
-
+ if(!kTRIGGERED){
+ AliDebug(2, Form("Reject Ev[%4d] Trigger", fESDev->GetEventNumberInFile()));
+ return;
+ }
// select only physical events
- if(fESDev->GetEventType() != 7) return;
+ if(fESDev->GetEventType() != 7){
+ AliDebug(2, Form("Reject Ev[%4d] EvType[%d]", fESDev->GetEventNumberInFile(), fESDev->GetEventType()));
+ return;
+ }
}
// if the required trigger is a collision trigger then apply event vertex cut
if(TMath::Abs(vertex->GetZv())<1.e-10 ||
TMath::Abs(vertex->GetZv())>fgkEvVertexZ ||
vertex->GetNContributors()<fgkEvVertexN) {
- //PostData(0, fOutStorage);
+ AliDebug(2, Form("Reject Ev[%4d] Vertex Zv[%f] Nv[%d]", fESDev->GetEventNumberInFile(), TMath::Abs(vertex->GetZv()), vertex->GetNContributors()));
return;
}
}
}
Double32_t dedx[100]; Int_t nSlices(0);
- Int_t nTRDout(0), nTRDin(0), nTPC(0), nclsTrklt;
- AliDebug(2, Form("Entry[%3d] Tracks: ESD[%d] MC[%d]\n", (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), nTracksESD, nTracksMC));
+ Int_t nTRDout(0), nTRDin(0), nTPC(0)
+ ,nclsTrklt
+ ,nBarrel(0), nSA(0), nKink(0)
+ ,nBarrelMC(0), nSAMC(0), nKinkMC(0);
AliESDtrack *esdTrack = NULL;
AliESDfriendTrack *esdFriendTrack = NULL;
TObject *calObject = NULL;
AliTRDtrackV1 *track = NULL;
AliTRDseedV1 *tracklet = NULL;
AliTRDcluster *cl = NULL;
+ // LOOP 0 - over ESD tracks
for(Int_t itrk = 0; itrk < nTracksESD; itrk++){
+ new(fTrackInfo) AliTRDtrackInfo();
esdTrack = fESDev->GetTrack(itrk);
AliDebug(3, Form("\n%3d ITS[%d] TPC[%d] TRD[%d]\n", itrk, esdTrack->GetNcls(0), esdTrack->GetNcls(1), esdTrack->GetNcls(2)));
iref++;
}
- new(fTrackInfo) AliTRDtrackInfo();
+ fTrackInfo->SetMC();
fTrackInfo->SetPDG(fPdg);
fTrackInfo->SetPrimary(mcParticle->Particle()->IsPrimary());
+ fTrackInfo->SetLabel(label);
Int_t jref = iref;//, kref = 0;
while(jref<nRefs){
ref = mcParticle->GetTrackReference(jref);
- if(ref->LocalX() > fgkTOF) break;
+ if(ref->LocalX() > fgkTRD) break;
AliDebug(4, Form(" trackRef[%2d (%2d)] @ %7.3f OK", jref-iref, jref, ref->LocalX()));
fTrackInfo->AddTrackRef(ref);
jref++;
}
AliDebug(3, Form("NtrackRefs[%d(%d)]", fTrackInfo->GetNTrackRefs(), nRefs));
- } else {
- new (fTrackInfo) AliTRDtrackInfo();
- fTrackInfo->SetPDG(fPdg);
}
// copy some relevant info to TRD track info
dedx[in++]=esdTrack->GetTRDslice(il, is);
for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++) dedx[in++]=esdTrack->GetTRDmomentum(il);
fTrackInfo->SetSlices(in, dedx);
- fTrackInfo->SetLabel(label);
fTrackInfo->SetNumberOfClustersRefit(esdTrack->GetNcls(2));
// some other Informations which we may wish to store in order to find problematic cases
fTrackInfo->SetKinkIndex(esdTrack->GetKinkIndex(0));
if(!esdTrack->GetKinkIndex(0)){ // Barrel Track Selection
Bool_t selected(kTRUE);
if(UseLocalTrkSelection()){
- if(esdTrack->Pt() < fgkPt) selected = kFALSE;
- if(TMath::Abs(esdTrack->Eta()) > fgkEta) selected = kFALSE;
- if(esdTrack->GetTPCNcls() < fgkNclTPC) selected = kFALSE;
+ if(esdTrack->Pt() < fgkPt){
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] Pt[%5.2f]", fESDev->GetEventNumberInFile(), itrk, esdTrack->Pt()));
+ selected = kFALSE;
+ }
+ if(selected && TMath::Abs(esdTrack->Eta()) > fgkEta){
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] Eta[%5.2f]", fESDev->GetEventNumberInFile(), itrk, TMath::Abs(esdTrack->Eta())));
+ selected = kFALSE;
+ }
+ if(selected && esdTrack->GetTPCNcls() < fgkNclTPC){
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] NclTPC[%d]", fESDev->GetEventNumberInFile(), itrk, esdTrack->GetTPCNcls()));
+ selected = kFALSE;
+ }
Float_t par[2], cov[3];
esdTrack->GetImpactParameters(par, cov);
if(IsCollision()){ // cuts on DCA
- if(TMath::Abs(par[0]) > fgkTrkDCAxy) selected = kFALSE;
- if(TMath::Abs(par[1]) > fgkTrkDCAz) selected = kFALSE;
+ if(selected && TMath::Abs(par[0]) > fgkTrkDCAxy){
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] DCAxy[%f]", fESDev->GetEventNumberInFile(), itrk, TMath::Abs(par[0])));
+ selected = kFALSE;
+ }
+ if(selected && TMath::Abs(par[1]) > fgkTrkDCAz){
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] DCAz[%f]", fESDev->GetEventNumberInFile(), itrk, TMath::Abs(par[1])));
+ selected = kFALSE;
+ }
+ } else if(selected && fMCev && !fMCev->IsPhysicalPrimary(alab)){;
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] Primary", fESDev->GetEventNumberInFile(), itrk));
+ selected = kFALSE;
}
}
if(fTrackCut && !fTrackCut->IsSelected(esdTrack)) selected = kFALSE;
- if(selected) fTracksBarrel->Add(new AliTRDtrackInfo(*fTrackInfo));
- } else fTracksKink->Add(new AliTRDtrackInfo(*fTrackInfo));
- } else if((status&AliESDtrack::kTRDout) && !(status&AliESDtrack::kTRDin)) fTracksSA->Add(new AliTRDtrackInfo(*fTrackInfo));
+ if(selected){
+ fTracksBarrel->Add(new AliTRDtrackInfo(*fTrackInfo));
+ nBarrel++;
+ }
+ } else {
+ fTracksKink->Add(new AliTRDtrackInfo(*fTrackInfo));
+ nKink++;
+ }
+ } else if((status&AliESDtrack::kTRDout) && !(status&AliESDtrack::kTRDin)){
+ fTracksSA->Add(new AliTRDtrackInfo(*fTrackInfo));
+ nSA++;
+ }
fTrackInfo->Delete("");
}
- AliDebug(2, Form(
- "%3d Tracks: TPCout[%d] TRDin[%d] TRDout[%d]\n"
- " Barrel[%d] SA[%d] Kink[%d]"
- ,(Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry()
- , nTPC, nTRDin, nTRDout
- ,fTracksBarrel->GetEntriesFast()
- ,fTracksSA->GetEntriesFast()
- ,fTracksKink->GetEntriesFast()
- ));
-
-// AliESDv0 *v0 = NULL;
-// for(Int_t iv0=0; iv0<fESD->GetNumberOfV0s(); iv0++){
-// if(!(v0 = fESD->GetV0(iv0))) continue;
-// fV0List->Add(new AliTRDv0Info(v0));
-// }
+ // LOOP 1 - over ESD v0s
+ Bool_t kFOUND(kFALSE);
+ Float_t bField(fESDev->GetMagneticField());
+ AliESDv0 *v0(NULL);
+ for(Int_t iv0(0); iv0<fESDev->GetNumberOfV0s(); iv0++){
+ if(!(v0 = fESDev->GetV0(iv0))) continue;
+
+ // purge V0 irelevant for TRD
+ Int_t jb(0); AliTRDtrackInfo *ti(NULL);
+ for(Int_t ib(0); ib<nBarrel; ib++){
+ ti = (AliTRDtrackInfo*)fTracksBarrel->At(ib);
+ Int_t id(ti->GetTrackId());
+ if(id!=v0->GetPindex() && id!=v0->GetNindex()) continue;
+ kFOUND=kTRUE; ti->SetV0(); jb=ib+1;
+ break;
+ }
+ if(!kFOUND) continue;
+
+ // register v0
+ if(fV0Cut) new(fV0Info) AliTRDv0Info(*fV0Cut);
+ else new(fV0Info) AliTRDv0Info();
+ fV0Info->SetMagField(bField);
+ fV0Info->SetV0tracks(fESDev->GetTrack(v0->GetPindex()), fESDev->GetTrack(v0->GetNindex()));
+ fV0Info->SetV0Info(v0);
+ // mark the other leg too
+ for(Int_t ib(jb); ib<nBarrel; ib++){
+ ti = (AliTRDtrackInfo*)fTracksBarrel->At(ib);
+ if(!fV0Info->HasTrack(ti)) continue;
+ ti->SetV0();
+ break;
+ }
+ //fV0Info->Print("a");
+ fV0List->Add(new AliTRDv0Info(*fV0Info)); kFOUND=kFALSE;
+ }
- // Insert also MC tracks which are passing TRD where the track is not reconstructed
+ // LOOP 2 - over MC tracks which are passing TRD where the track is not reconstructed
if(HasMCdata()){
AliDebug(10, "Output of the MC track map:");
for(Int_t itk = 0; itk < nTracksMC; itk++) AliDebug(10, Form("trackMap[%d] = %s", itk, trackMap[itk] == kTRUE ? "TRUE" : "kFALSE"));
Int_t iref = 0; AliTrackReference *ref = NULL;
Int_t nRefsTRD = 0;
new(fTrackInfo) AliTRDtrackInfo();
+ fTrackInfo->SetMC();
fTrackInfo->SetPDG(fPdg);
- while(iref<nRefs){
+ while(iref<nRefs){ // count TRD TR
Bool_t kIN(kFALSE);
ref = mcParticle->GetTrackReference(iref);
- if(ref->LocalX() > fgkTPC && ref->LocalX() < fgkTOF){
+ if(ref->LocalX() > fgkTPC && ref->LocalX() < fgkTRD){
fTrackInfo->AddTrackRef(ref);
nRefsTRD++;kIN=kTRUE;
}
info.Delete("");
}
AliDebug(3, Form("Add MC track @ label[%d] nTRDrefs[%d].", itk, nRefsTRD));
- fTracksBarrel->Add(new AliTRDtrackInfo(*fTrackInfo));
+ // check where the track starts
+ ref = mcParticle->GetTrackReference(0);
+ if(ref->LocalX() < fgkITS){
+ fTracksBarrel->Add(new AliTRDtrackInfo(*fTrackInfo));
+ nBarrelMC++;
+ } else if(ref->LocalX() < fgkTPC) {
+ fTracksKink->Add(new AliTRDtrackInfo(*fTrackInfo));
+ nKinkMC++;
+ } else if(nRefsTRD>6){
+ fTracksSA->Add(new AliTRDtrackInfo(*fTrackInfo));
+ nSAMC++;
+ }
fTrackInfo->Delete("");
}
delete[] trackMap;
}
+ AliDebug(1, Form(
+ "\nEv[%3d] Tracks: ESD[%d] MC[%d] V0[%d]\n"
+ " TPCout[%d] TRDin[%d] TRDout[%d]\n"
+ " Barrel[%3d+%3d=%3d] SA[%2d+%2d=%2d] Kink[%2d+%2d=%2d]"
+ ,(Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), nTracksESD, nTracksMC, fV0List->GetEntries()
+ , nTPC, nTRDin, nTRDout
+ ,nBarrel, nBarrelMC, fTracksBarrel->GetEntries()
+ ,nSA, nSAMC, fTracksSA->GetEntries()
+ ,nKink, nKinkMC, fTracksKink->GetEntries()
+ ));
+ // save track statistics
+ TH1 *h((TH1S*)fContainer->At(0));
+ h->Fill( 0., nTracksESD);
+ h->Fill( 1., nTracksMC);
+ h->Fill( 2., fV0List->GetEntries());
+ h->Fill( 3., nTPC);
+ h->Fill( 4., nTRDin);
+ h->Fill( 5., nTRDout);
+ h->Fill( 6., nBarrel);
+ h->Fill( 7., nBarrelMC);
+ h->Fill( 8., nSA);
+ h->Fill( 9., nSAMC);
+ h->Fill(10., nKink);
+ h->Fill(11., nKinkMC);
+
PostData(kTracksBarrel, fTracksBarrel);
PostData(kTracksSA, fTracksSA);
PostData(kTracksKink, fTracksKink);
PostData(kV0List, fV0List);
}
+//____________________________________________________________________
+void AliTRDinfoGen::SetLocalV0Selection(AliTRDv0Info *v0)
+{
+// Set V0 cuts from outside
+
+ if(!fV0Cut) fV0Cut = new AliTRDv0Info(*v0);
+ else new(fV0Cut) AliTRDv0Info(*v0);
+}
+
//____________________________________________________________________
void AliTRDinfoGen::SetTrigger(const Char_t *trigger)
{
}
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff