]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWG1/TRD/AliTRDinfoGen.cxx
Fixed compilation
[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDinfoGen.cxx
index 4883768cdebc76264d6273c18a21b8ab1eae8bc1..5f5603e4d57ae10f2e1ddb243b7006966318cd73 100644 (file)
@@ -34,7 +34,9 @@
 #include <TClonesArray.h>
 #include <TObjArray.h>
 #include <TObject.h>
-#include <TH1F.h>
+#include <TString.h>
+#include <TH1S.h>
+#include <TPad.h>
 #include <TFile.h>
 #include <TTree.h>
 #include <TROOT.h>
@@ -43,6 +45,7 @@
 
 #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"
 #include "info/AliTRDv0Info.h"
+#include "info/AliTRDeventCuts.h"
+#include "macros/AliTRDperformanceTrain.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  = 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;
 
 //____________________________________________________________________
-AliTRDinfoGen::AliTRDinfoGen():
-  AliTRDrecoTask("infoGen", "MC-REC TRD-track list generator")
+AliTRDinfoGen::AliTRDinfoGen()
+  :AliAnalysisTaskSE()
+  ,fEvTrigger(NULL)
   ,fESDev(NULL)
   ,fMCev(NULL)
-  ,fESDfriend(NULL)
+  ,fEventCut(NULL)
+  ,fTrackCut(NULL)
+  ,fV0Cut(NULL)
+  ,fOCDB("local://$ALICE_ROOT/OCDB")
   ,fTrackInfo(NULL)
   ,fEventInfo(NULL)
-  ,fV0container(NULL)
   ,fV0Info(NULL)
+  ,fTracksBarrel(NULL)
+  ,fTracksSA(NULL)
+  ,fTracksKink(NULL)
+  ,fV0List(NULL)
+  ,fContainer(NULL)
+  ,fDebugStream(NULL)
 {
   //
   // Default constructor
   //
+  SetNameTitle("TRDinfoGen", "MC-REC TRD-track list generator");
+}
 
-  DefineInput(0, TChain::Class());
-  DefineOutput(0, TObjArray::Class());
-  DefineOutput(1, AliTRDeventInfo::Class());
-  DefineOutput(2, TObjArray::Class());
+//____________________________________________________________________
+AliTRDinfoGen::AliTRDinfoGen(char* name)
+  :AliAnalysisTaskSE(name)
+  ,fEvTrigger(NULL)
+  ,fESDev(NULL)
+  ,fMCev(NULL)
+  ,fEventCut(NULL)
+  ,fTrackCut(NULL)
+  ,fV0Cut(NULL)
+  ,fOCDB("local://$ALICE_ROOT/OCDB")
+  ,fTrackInfo(NULL)
+  ,fEventInfo(NULL)
+  ,fV0Info(NULL)
+  ,fTracksBarrel(NULL)
+  ,fTracksSA(NULL)
+  ,fTracksKink(NULL)
+  ,fV0List(NULL)
+  ,fContainer(NULL)
+  ,fDebugStream(NULL)
+{
+  //
+  // Default constructor
+  //
+  SetTitle("MC-REC TRD-track list generator");
+  DefineOutput(kTracksBarrel, TObjArray::Class());
+  DefineOutput(kTracksSA, TObjArray::Class());
+  DefineOutput(kTracksKink, TObjArray::Class());
+  DefineOutput(kEventInfo, AliTRDeventInfo::Class());
+  DefineOutput(kV0List, TObjArray::Class());
+  DefineOutput(kMonitor, TObjArray::Class()); // histogram list
 }
 
 //____________________________________________________________________
 AliTRDinfoGen::~AliTRDinfoGen()
 {
 // Destructor
-
-  printf("AliTRDinfoGen::~AliTRDinfoGen() START\n");
+  
+  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;
-  printf("fTrackInfo\n");
   if(fEventInfo) delete fEventInfo; fEventInfo = NULL;
-  printf("fEventInfo\n");
   if(fV0Info) delete fV0Info; fV0Info = NULL;
-  printf("fV0Info\n");
+  if(fTracksBarrel){ 
+    fTracksBarrel->Delete(); delete fTracksBarrel;
+    fTracksBarrel = NULL;
+  }
+  if(fTracksSA){ 
+    fTracksSA->Delete(); delete fTracksSA;
+    fTracksSA = NULL;
+  }
+  if(fTracksKink){ 
+    fTracksKink->Delete(); delete fTracksKink;
+    fTracksKink = NULL;
+  }
+  if(fV0List){ 
+    fV0List->Delete(); 
+    delete fV0List;
+    fV0List = NULL;
+  }
   if(fContainer){ 
-    fContainer->Delete(); delete fContainer;
+    fContainer->Delete(); 
+    delete fContainer;
     fContainer = NULL;
   }
-  printf("fContainer\n");
-  if(fV0container){ 
-    fV0container->Delete(); delete fV0container;
-    fV0container = NULL;
-  }
-  printf("fV0container\n");
-  printf("AliTRDinfoGen::~AliTRDinfoGen() STOP\n");
 }
 
 //____________________________________________________________________
-void AliTRDinfoGen::ConnectInputData(Option_t *)
+Bool_t AliTRDinfoGen::GetRefFigure(Int_t)
 {
-  //
-  // Link the Input Data
-  //
-  TTree *tree = dynamic_cast<TChain*>(GetInputData(0));
-  if(!tree){
-    AliError("ESD event not found");
-  } else {
-    tree->SetBranchStatus("Tracks", 1);
-    tree->SetBranchStatus("ESDfriend*",1);
-  }
-
-  AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*>(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
-  if(!esdH){
-    printf("ERROR - ESD input handler not found");
-  } else {
-    fESDev = esdH->GetEvent();
-    if(!fESDev){
-      printf("ERROR - ESD event not found");
-    } else {
-      esdH->SetActiveBranches("ESDfriend*");
-      fESDfriend = (AliESDfriend *)fESDev->FindListObject("AliESDfriend");
-    }
-  }
-  if(HasMCdata()){
-    AliMCEventHandler *mcH = dynamic_cast<AliMCEventHandler*>(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
-    if(!mcH){ 
-      AliError("MC input handler not found");
-    } else {
-      fMCev = mcH->MCEvent();
-    }
+  if(!gPad){
+    AliWarning("Please provide a canvas to draw results.");
+    return kFALSE;
   }
+  fContainer->At(0)->Draw("bar");
+  return kTRUE;
 }
 
 //____________________________________________________________________
-void AliTRDinfoGen::CreateOutputObjects()
+void AliTRDinfoGen::UserCreateOutputObjects()
 {      
   //
   // Create Output Containers (TObjectArray containing 1D histograms)
   //
   fTrackInfo = new AliTRDtrackInfo();
   fEventInfo = new AliTRDeventInfo();
   fV0Info    = new AliTRDv0Info();
-  fContainer = new TObjArray(1000);
-  fContainer->SetOwner(kTRUE);
-  fV0container = new TObjArray(50);
-  fV0container->SetOwner(kTRUE);
+  fTracksBarrel = new TObjArray(200); fTracksBarrel->SetOwner(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);
 }
 
 //____________________________________________________________________
-void AliTRDinfoGen::Exec(Option_t *){
+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;
+}
+
+//____________________________________________________________________
+void AliTRDinfoGen::UserExec(Option_t *){
   //
   // Run the Analysis
   //
+
+  fESDev = dynamic_cast<AliESDEvent*>(InputEvent());
   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){ 
+    Bool_t kTRIGGERED(kFALSE);
+    const TObjArray *trig = fEvTrigger->Tokenize(" ");
+    for(Int_t itrig=trig->GetEntriesFast(); itrig--;){
+      const Char_t *trigClass(((TObjString*)(*trig)[itrig])->GetName());
+      if(fESDev->IsTriggerClassFired(trigClass)) {
+        AliDebug(2, Form("Ev[%4d] Trigger[%s]", fESDev->GetEventNumberInFile(), trigClass));
+        kTRIGGERED = kTRUE;
+        break; 
+      }
+    }
+    if(!kTRIGGERED){ 
+      AliDebug(2, Form("Reject Ev[%4d] Trigger", fESDev->GetEventNumberInFile()));
+      return;
+    }
+    // select only physical events
+    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(UseLocalEvSelection() && IsCollision()){
+    const AliESDVertex *vertex = fESDev->GetPrimaryVertex();
+    if(TMath::Abs(vertex->GetZv())<1.e-10 || 
+       TMath::Abs(vertex->GetZv())>fgkEvVertexZ || 
+       vertex->GetNContributors()<fgkEvVertexN) {
+      AliDebug(2, Form("Reject Ev[%4d] Vertex Zv[%f] Nv[%d]", fESDev->GetEventNumberInFile(), TMath::Abs(vertex->GetZv()), vertex->GetNContributors()));
+      return;
+    }
+  }
+
+  if(fEventCut && !fEventCut->IsSelected(fESDev, IsCollision())) return;
+
   if(!fESDfriend){
     AliError("Failed retrieving ESD friend event");
     return;
@@ -192,10 +331,12 @@ void AliTRDinfoGen::Exec(Option_t *){
     AliError("Failed retrieving MC event");
     return;
   }
-  fContainer->Delete();
-  fV0container->Delete();
+
+  fTracksBarrel->Delete();
+  fTracksSA->Delete();
+  fTracksKink->Delete();
+  fV0List->Delete();
   fEventInfo->Delete("");
-  fESDev->SetESDfriend(fESDfriend);
   new(fEventInfo)AliTRDeventInfo(fESDev->GetHeader(), const_cast<AliESDRun *>(fESDev->GetESDRun()));
   
   Bool_t *trackMap(NULL);
@@ -212,17 +353,22 @@ void AliTRDinfoGen::Exec(Option_t *){
   }
   
   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)));
+
     if(esdTrack->GetStatus()&AliESDtrack::kTPCout) nTPC++;
     if(esdTrack->GetStatus()&AliESDtrack::kTRDout) nTRDout++;
     if(esdTrack->GetStatus()&AliESDtrack::kTRDin) nTRDin++;
@@ -263,21 +409,19 @@ void AliTRDinfoGen::Exec(Option_t *){
         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
@@ -294,7 +438,6 @@ void AliTRDinfoGen::Exec(Option_t *){
         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));
@@ -330,19 +473,92 @@ void AliTRDinfoGen::Exec(Option_t *){
       << "\n";
       info.Delete("");
     }
-  
-    fContainer->Add(new AliTRDtrackInfo(*fTrackInfo));
+
+    ULong_t status(esdTrack->GetStatus());
+    if((status&AliESDtrack::kTPCout)){
+      if(!esdTrack->GetKinkIndex(0)){ // Barrel  Track Selection
+        Bool_t selected(kTRUE);
+        if(UseLocalTrkSelection()){
+          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(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));
+          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", (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), nTPC, nTRDin, nTRDout));
 
-//   AliESDv0 *v0 = NULL;
-//   for(Int_t iv0=0; iv0<fESD->GetNumberOfV0s(); iv0++){
-//     if(!(v0 = fESD->GetV0(iv0))) continue;
-//     fV0container->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;
 
-  // Insert also MC tracks which are passing TRD where the track is not reconstructed
+    // 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;
+  }
+
+  // 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"));
@@ -355,11 +571,12 @@ void AliTRDinfoGen::Exec(Option_t *){
       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;
         }
@@ -382,13 +599,80 @@ void AliTRDinfoGen::Exec(Option_t *){
         info.Delete("");
       }
       AliDebug(3, Form("Add MC track @ label[%d] nTRDrefs[%d].", itk, nRefsTRD));
-      fContainer->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;
   }
-  PostData(0, fContainer);
-  PostData(1, fEventInfo);
-  PostData(2, fV0container);
+  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(kEventInfo, fEventInfo);
+  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)
+{
+  if(!fEvTrigger) fEvTrigger = new TString(trigger);
+  else (*fEvTrigger) = trigger;
+}
+
+//____________________________________________________________________
+TTreeSRedirector* AliTRDinfoGen::DebugStream()
+{
+  if(!fDebugStream){
+    TDirectory *savedir = gDirectory;
+    fDebugStream = new TTreeSRedirector("TRD.DebugInfoGen.root");
+    savedir->cd();
+  }
+  return fDebugStream;
 }
 
+
+