]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWG1/TRD/AliTRDinfoGen.cxx
Fixed compilation
[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDinfoGen.cxx
index 276c20d9fce91591acf56d797a51248f2b434c0a..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  = 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;
 
 //____________________________________________________________________
 AliTRDinfoGen::AliTRDinfoGen()
-  :AliTRDrecoTask()
-  ,fEvTrigger("CINT1B-ABCE-NOPF-ALL")
+  :AliAnalysisTaskSE()
+  ,fEvTrigger(NULL)
   ,fESDev(NULL)
   ,fMCev(NULL)
+  ,fEventCut(NULL)
+  ,fTrackCut(NULL)
+  ,fV0Cut(NULL)
+  ,fOCDB("local://$ALICE_ROOT/OCDB")
   ,fTrackInfo(NULL)
   ,fEventInfo(NULL)
-  ,fV0container(NULL)
   ,fV0Info(NULL)
-  ,fEventCut(NULL)
-  ,fTrackCut(NULL)
+  ,fTracksBarrel(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");
 }
 
 //____________________________________________________________________
 AliTRDinfoGen::AliTRDinfoGen(char* name)
-  :AliTRDrecoTask(name)
-  ,fEvTrigger("")
+  :AliAnalysisTaskSE(name)
+  ,fEvTrigger(NULL)
   ,fESDev(NULL)
   ,fMCev(NULL)
+  ,fEventCut(NULL)
+  ,fTrackCut(NULL)
+  ,fV0Cut(NULL)
+  ,fOCDB("local://$ALICE_ROOT/OCDB")
   ,fTrackInfo(NULL)
   ,fEventInfo(NULL)
-  ,fV0container(NULL)
   ,fV0Info(NULL)
-  ,fEventCut(NULL)
-  ,fTrackCut(NULL)
+  ,fTracksBarrel(NULL)
+  ,fTracksSA(NULL)
+  ,fTracksKink(NULL)
+  ,fV0List(NULL)
+  ,fContainer(NULL)
+  ,fDebugStream(NULL)
 {
   //
   // Default constructor
   //
-    DefineOutput(2, AliTRDeventInfo::Class());   // -> TRDeventInfo
-    DefineOutput(3, TObjArray::Class());         // -> TObjArray
+  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
-
+  
+  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;
   if(fEventInfo) delete fEventInfo; fEventInfo = NULL;
   if(fV0Info) delete fV0Info; fV0Info = NULL;
+  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;
   }
-  if(fV0container){ 
-    fV0container->Delete(); delete fV0container;
-    fV0container = 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;
 }
 
 //____________________________________________________________________
@@ -157,10 +212,55 @@ void AliTRDinfoGen::UserCreateOutputObjects()
   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);
+}
+
+//____________________________________________________________________
+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;
 }
 
 //____________________________________________________________________
@@ -168,22 +268,29 @@ void AliTRDinfoGen::UserExec(Option_t *){
   //
   // Run the Analysis
   //
-  printf("AliTRDinfoGen::UserExec : ESD[%p] MC[%p]\n", (void*)fESDev, (void*)fMCev);
 
   fESDev = dynamic_cast<AliESDEvent*>(InputEvent());
-  fMCev = MCEvent();
-  AliInfo(Form("ESD[%p] MC[%p]", (void*)fESDev, (void*)fMCev));
-
   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.Data()[0]!='\0'){ 
+  if(UseLocalEvSelection() && fEvTrigger){ 
     Bool_t kTRIGGERED(kFALSE);
-    const TObjArray *trig = fEvTrigger.Tokenize(" ");
+    const TObjArray *trig = fEvTrigger->Tokenize(" ");
     for(Int_t itrig=trig->GetEntriesFast(); itrig--;){
       const Char_t *trigClass(((TObjString*)(*trig)[itrig])->GetName());
       if(fESDev->IsTriggerClassFired(trigClass)) {
@@ -192,10 +299,15 @@ void AliTRDinfoGen::UserExec(Option_t *){
         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
@@ -204,7 +316,7 @@ void AliTRDinfoGen::UserExec(Option_t *){
     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;
     }
   }
@@ -220,8 +332,10 @@ void AliTRDinfoGen::UserExec(Option_t *){
     return;
   }
 
-  fContainer->Delete();
-  fV0container->Delete();
+  fTracksBarrel->Delete();
+  fTracksSA->Delete();
+  fTracksKink->Delete();
+  fV0List->Delete();
   fEventInfo->Delete("");
   new(fEventInfo)AliTRDeventInfo(fESDev->GetHeader(), const_cast<AliESDRun *>(fESDev->GetESDRun()));
   
@@ -239,32 +353,22 @@ void AliTRDinfoGen::UserExec(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)));
 
-    // Track Selection
-    if(UseLocalTrkSelection()){
-      if(esdTrack->Pt() < fgkPt) continue;
-      if(TMath::Abs(esdTrack->Eta()) > fgkEta) continue;
-      if(esdTrack->GetTPCNcls() < fgkNclTPC) continue;
-      Float_t par[2], cov[3];
-      esdTrack->GetImpactParameters(par, cov);
-      if(IsCollision()){ // cuts on DCA
-        if(TMath::Abs(par[0]) > fgkTrkDCAxy) continue;
-        if(TMath::Abs(par[1]) > fgkTrkDCAz) continue;
-      }
-    }
-    if(fTrackCut && !fTrackCut->IsSelected(esdTrack)) continue;
-
     if(esdTrack->GetStatus()&AliESDtrack::kTPCout) nTPC++;
     if(esdTrack->GetStatus()&AliESDtrack::kTRDout) nTRDout++;
     if(esdTrack->GetStatus()&AliESDtrack::kTRDin) nTRDin++;
@@ -305,21 +409,19 @@ void AliTRDinfoGen::UserExec(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
@@ -336,7 +438,6 @@ void AliTRDinfoGen::UserExec(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));
@@ -372,20 +473,92 @@ void AliTRDinfoGen::UserExec(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;
+
+    // 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"));
@@ -398,11 +571,12 @@ void AliTRDinfoGen::UserExec(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;
         }
@@ -425,13 +599,80 @@ void AliTRDinfoGen::UserExec(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(1, fContainer);
-  PostData(2, fEventInfo);
-  PostData(3, 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;
+}
+
+
+