-/**************************************************************************
-* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
-* *
-* Author: The ALICE Off-line Project. *
-* Contributors are mentioned in the code where appropriate. *
-* *
-* Permission to use, copy, modify and distribute this software and its *
-* documentation strictly for non-commercial purposes is hereby granted *
-* without fee, provided that the above copyright notice appears in all *
-* copies and that both the copyright notice and this permission notice *
-* appear in the supporting documentation. The authors make no claims *
-* about the suitability of this software for any purpose. It is *
-* provided "as is" without express or implied warranty. *
-**************************************************************************/
-
-/* $Id: AliTRDinfoGen.cxx 27496 2008-07-22 08:35:45Z cblume $ */
-
-////////////////////////////////////////////////////////////////////////////
-//
-// Tender wagon for TRD performance/calibration train
-//
-// In this wagon the information from
-// - ESD
-// - Friends [if available]
-// - MC [if available]
-// are grouped into AliTRDtrackInfo objects and fed to worker tasks
-//
-// Authors:
-// Markus Fasel <M.Fasel@gsi.de>
-// Alexandru Bercuci <A.Bercuci@gsi.de>
-//
-////////////////////////////////////////////////////////////////////////////
-
-#include <TClonesArray.h>
-#include <TObjArray.h>
-#include <TObject.h>
-#include <TH1F.h>
-#include <TFile.h>
-#include <TTree.h>
-#include <TROOT.h>
-#include <TChain.h>
-#include <TParticle.h>
-
-#include "AliLog.h"
-#include "AliAnalysisManager.h"
-#include "AliESDEvent.h"
-#include "AliMCEvent.h"
-#include "AliESDInputHandler.h"
-#include "AliMCEventHandler.h"
-
-#include "AliESDfriend.h"
-#include "AliESDfriendTrack.h"
-#include "AliESDHeader.h"
-#include "AliESDtrack.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 <climits>
-#include <cstring>
-#include <iostream>
-
-#include "AliTRDinfoGen.h"
-#include "info/AliTRDtrackInfo.h"
-#include "info/AliTRDeventInfo.h"
-#include "info/AliTRDv0Info.h"
-
-ClassImp(AliTRDinfoGen)
-
-const Float_t AliTRDinfoGen::fgkTPC = 290.;
-const Float_t AliTRDinfoGen::fgkTOF = 365.;
-
-//____________________________________________________________________
-AliTRDinfoGen::AliTRDinfoGen():
- AliTRDrecoTask("infoGen", "MC-REC TRD-track list generator")
- ,fESDev(NULL)
- ,fMCev(NULL)
- ,fESDfriend(NULL)
- ,fTrackInfo(NULL)
- ,fEventInfo(NULL)
- ,fV0container(NULL)
- ,fV0Info(NULL)
-{
- //
- // Default constructor
- //
-
- DefineInput(0, TChain::Class());
- DefineOutput(0, TObjArray::Class());
- DefineOutput(1, AliTRDeventInfo::Class());
- DefineOutput(2, TObjArray::Class());
-}
-
-//____________________________________________________________________
-AliTRDinfoGen::~AliTRDinfoGen()
-{
-// Destructor
-
- printf("AliTRDinfoGen::~AliTRDinfoGen() START\n");
- 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(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 *)
-{
- //
- // 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();
- }
- }
-}
-
-//____________________________________________________________________
-void AliTRDinfoGen::CreateOutputObjects()
-{
- //
- // 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);
-
-}
-
-//____________________________________________________________________
-void AliTRDinfoGen::Exec(Option_t *){
- //
- // Run the Analysis
- //
- if(!fESDev){
- AliError("Failed retrieving ESD event");
- return;
- }
- if(!fESDfriend){
- AliError("Failed retrieving ESD friend event");
- return;
- }
- if(HasMCdata() && !fMCev){
- AliError("Failed retrieving MC event");
- return;
- }
- fContainer->Delete();
- fV0container->Delete();
- fEventInfo->Delete("");
- fESDev->SetESDfriend(fESDfriend);
- new(fEventInfo)AliTRDeventInfo(fESDev->GetHeader(), const_cast<AliESDRun *>(fESDev->GetESDRun()));
-
- Bool_t *trackMap(NULL);
- AliStack * mStack(NULL);
- Int_t nTracksMC = HasMCdata() ? fMCev->GetNumberOfTracks() : 0, nTracksESD = fESDev->GetNumberOfTracks();
- if(HasMCdata()){
- mStack = fMCev->Stack();
- if(!mStack){
- AliError("Failed retrieving MC Stack");
- return;
- }
- trackMap = new Bool_t[nTracksMC];
- memset(trackMap, 0, sizeof(Bool_t) * nTracksMC);
- }
-
- 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));
- AliESDtrack *esdTrack = NULL;
- AliESDfriendTrack *esdFriendTrack = NULL;
- TObject *calObject = NULL;
- AliTRDtrackV1 *track = NULL;
- AliTRDseedV1 *tracklet = NULL;
- AliTRDcluster *cl = NULL;
- for(Int_t itrk = 0; itrk < nTracksESD; itrk++){
- 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++;
-
- // look at external track param
- const AliExternalTrackParam *op = esdTrack->GetOuterParam();
- Double_t xyz[3];
- if(op){
- op->GetXYZ(xyz);
- op->Global2LocalPosition(xyz, op->GetAlpha());
- AliDebug(3, Form("op @ X[%7.3f]\n", xyz[0]));
- }
-
- // read MC info
- Int_t fPdg = -1;
- Int_t label = -1; UInt_t alab=UINT_MAX;
- if(HasMCdata()){
- label = esdTrack->GetLabel();
- alab = TMath::Abs(label);
- // register the track
- if(alab < UInt_t(nTracksMC)){
- trackMap[alab] = kTRUE;
- } else {
- AliError(Form("MC label[%d] outside scope for Ev[%d] Trk[%d].", label, (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), itrk));
- continue;
- }
- AliMCParticle *mcParticle = NULL;
- if(!(mcParticle = (AliMCParticle*) fMCev->GetTrack(alab))){
- AliError(Form("MC particle label[%d] missing for Ev[%d] Trk[%d].", label, (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), itrk));
- continue;
- }
- fPdg = mcParticle->Particle()->GetPdgCode();
- Int_t nRefs = mcParticle->GetNumberOfTrackReferences();
- Int_t iref = 0; AliTrackReference *ref = NULL;
- while(iref<nRefs){
- ref = mcParticle->GetTrackReference(iref);
- if(ref->LocalX() > fgkTPC) break;
- iref++;
- }
-
- new(fTrackInfo) AliTRDtrackInfo();
- fTrackInfo->SetPDG(fPdg);
- fTrackInfo->SetPrimary(mcParticle->Particle()->IsPrimary());
- Int_t jref = iref;//, kref = 0;
- while(jref<nRefs){
- ref = mcParticle->GetTrackReference(jref);
- if(ref->LocalX() > fgkTOF) 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
- fTrackInfo->SetStatus(esdTrack->GetStatus());
- fTrackInfo->SetTrackId(esdTrack->GetID());
- Double_t p[AliPID::kSPECIES]; esdTrack->GetTRDpid(p);
- fTrackInfo->SetESDpid(p);
- fTrackInfo->SetESDpidQuality(esdTrack->GetTRDntrackletsPID());
- if(!nSlices) nSlices = esdTrack->GetNumberOfTRDslices();
- memset(dedx, 0, 100*sizeof(Double32_t));
- Int_t in(0);
- for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++)
- for(Int_t is=0; is<nSlices; is++)
- dedx[in++]=esdTrack->GetTRDslice(il, is);
- for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++) dedx[in++]=esdTrack->GetTRDmomentum(il);
- printf("n[%d] slices[%d]\n", in, nSlices);
- 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));
- fTrackInfo->SetTPCncls(static_cast<UShort_t>(esdTrack->GetNcls(1)));
- nclsTrklt = 0;
-
-
- // read REC info
- esdFriendTrack = fESDfriend->GetTrack(itrk);
- if(esdFriendTrack){
- Int_t icalib = 0;
- while((calObject = esdFriendTrack->GetCalibObject(icalib++))){
- if(strcmp(calObject->IsA()->GetName(),"AliTRDtrackV1") != 0) continue; // Look for the TRDtrack
- if(!(track = dynamic_cast<AliTRDtrackV1*>(calObject))) break;
- AliDebug(4, Form("TRD track OK"));
- // Set the clusters to unused
- for(Int_t ipl = 0; ipl < AliTRDgeometry::kNlayer; ipl++){
- if(!(tracklet = track->GetTracklet(ipl))) continue;
- tracklet->ResetClusterIter();
- while((cl = tracklet->NextCluster())) cl->Use(0);
- }
- fTrackInfo->SetTrack(track);
- break;
- }
- AliDebug(3, Form("Ntracklets[%d]\n", fTrackInfo->GetNTracklets()));
- } else AliDebug(3, "No ESD friends");
- if(op) fTrackInfo->SetOuterParam(op);
-
- if(DebugLevel() >= 1){
- AliTRDtrackInfo info(*fTrackInfo);
- (*DebugStream()) << "trackInfo"
- << "TrackInfo.=" << &info
- << "\n";
- info.Delete("");
- }
-
- fContainer->Add(new AliTRDtrackInfo(*fTrackInfo));
- 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));
-// }
-
- // Insert also 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"));
-
- for(Int_t itk = 0; itk < nTracksMC; itk++){
- if(trackMap[itk]) continue;
- AliMCParticle *mcParticle = (AliMCParticle*) fMCev->GetTrack(TMath::Abs(itk));
- Int_t fPdg = mcParticle->Particle()->GetPdgCode();
- Int_t nRefs = mcParticle->GetNumberOfTrackReferences();
- Int_t iref = 0; AliTrackReference *ref = NULL;
- Int_t nRefsTRD = 0;
- new(fTrackInfo) AliTRDtrackInfo();
- fTrackInfo->SetPDG(fPdg);
- while(iref<nRefs){
- Bool_t kIN(kFALSE);
- ref = mcParticle->GetTrackReference(iref);
- if(ref->LocalX() > fgkTPC && ref->LocalX() < fgkTOF){
- fTrackInfo->AddTrackRef(ref);
- nRefsTRD++;kIN=kTRUE;
- }
- AliDebug(4, Form(" trackRef[%2d] @ x[%7.3f] %s", iref, ref->LocalX(), kIN?"IN":"OUT"));
- iref++;
- }
- if(!nRefsTRD){
- // In this stage we at least require 1 hit inside TRD. What will be done with this tracks is a task for the
- // analysis job
- fTrackInfo->Delete("");
- continue;
- }
- fTrackInfo->SetPrimary(mcParticle->Particle()->IsPrimary());
- fTrackInfo->SetLabel(itk);
- if(DebugLevel() >= 1){
- AliTRDtrackInfo info(*fTrackInfo);
- (*DebugStream()) << "trackInfo"
- << "TrackInfo.=" << &info
- << "\n";
- info.Delete("");
- }
- AliDebug(3, Form("Add MC track @ label[%d] nTRDrefs[%d].", itk, nRefsTRD));
- fContainer->Add(new AliTRDtrackInfo(*fTrackInfo));
- fTrackInfo->Delete("");
- }
- delete[] trackMap;
- }
- PostData(0, fContainer);
- PostData(1, fEventInfo);
- PostData(2, fV0container);
-}
-
+/**************************************************************************\r
+* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
+* *\r
+* Author: The ALICE Off-line Project. *\r
+* Contributors are mentioned in the code where appropriate. *\r
+* *\r
+* Permission to use, copy, modify and distribute this software and its *\r
+* documentation strictly for non-commercial purposes is hereby granted *\r
+* without fee, provided that the above copyright notice appears in all *\r
+* copies and that both the copyright notice and this permission notice *\r
+* appear in the supporting documentation. The authors make no claims *\r
+* about the suitability of this software for any purpose. It is *\r
+* provided "as is" without express or implied warranty. *\r
+**************************************************************************/\r
+\r
+/* $Id: AliTRDinfoGen.cxx 27496 2008-07-22 08:35:45Z cblume $ */\r
+\r
+////////////////////////////////////////////////////////////////////////////\r
+//\r
+// Tender wagon for TRD performance/calibration train\r
+//\r
+// In this wagon the information from\r
+// - ESD\r
+// - Friends [if available]\r
+// - MC [if available]\r
+// are grouped into AliTRDtrackInfo objects and fed to worker tasks\r
+//\r
+// Authors:\r
+// Markus Fasel <M.Fasel@gsi.de>\r
+// Alexandru Bercuci <A.Bercuci@gsi.de>\r
+//\r
+////////////////////////////////////////////////////////////////////////////\r
+\r
+#include <TClonesArray.h>\r
+#include <TObjArray.h>\r
+#include <TObject.h>\r
+#include <TString.h>\r
+#include <TH1S.h>\r
+#include <TPad.h>\r
+#include <TFile.h>\r
+#include <TTree.h>\r
+#include <TROOT.h>\r
+#include <TChain.h>\r
+#include <TParticle.h>\r
+\r
+#include "AliLog.h"\r
+#include "AliAnalysisManager.h"\r
+#include "AliGeomManager.h"\r
+#include "AliCDBManager.h"\r
+#include "AliCDBEntry.h"\r
+#include "AliCDBPath.h"\r
+#include "AliESDEvent.h"\r
+#include "AliMCEvent.h"\r
+#include "AliESDInputHandler.h"\r
+#include "AliMCEventHandler.h"\r
+\r
+#include "AliESDfriend.h"\r
+#include "AliESDfriendTrack.h"\r
+#include "AliESDHeader.h"\r
+#include "AliESDRun.h"\r
+#include "AliESDtrack.h"\r
+#include "AliESDv0.h"\r
+#include "AliESDtrackCuts.h"\r
+#include "AliMCParticle.h"\r
+#include "AliMultiplicity.h"\r
+#include "AliPID.h"\r
+#include "AliStack.h"\r
+#include "AliTrackReference.h"\r
+#include "TTreeStream.h"\r
+\r
+#include <cstdio>\r
+#include <climits>\r
+#include <cstring>\r
+#include <iostream>\r
+#include <memory>\r
+\r
+#include "AliTRDReconstructor.h"\r
+#include "AliTRDrecoParam.h"\r
+#include "AliTRDcalibDB.h"\r
+#include "AliTRDtrackerV1.h"\r
+#include "AliTRDgeometry.h"\r
+#include "AliTRDtrackV1.h"\r
+#include "AliTRDseedV1.h"\r
+#include "AliTRDcluster.h"\r
+#include "AliTRDinfoGen.h"\r
+#include "AliTRDpwg1Helper.h"\r
+#include "info/AliTRDtrackInfo.h"\r
+#include "info/AliTRDeventInfo.h"\r
+#include "info/AliTRDv0Info.h"\r
+#include "info/AliTRDeventCuts.h"\r
+\r
+ClassImp(AliTRDinfoGen)\r
+\r
+const Float_t AliTRDinfoGen::fgkITS = 100.; // to be checked\r
+const Float_t AliTRDinfoGen::fgkTPC = 290.;\r
+const Float_t AliTRDinfoGen::fgkTRD = 365.;\r
+\r
+const Float_t AliTRDinfoGen::fgkEvVertexZ = 15.;\r
+const Int_t AliTRDinfoGen::fgkEvVertexN = 1;\r
+\r
+const Float_t AliTRDinfoGen::fgkTrkDCAxy = 3.;\r
+const Float_t AliTRDinfoGen::fgkTrkDCAz = 10.;\r
+const Int_t AliTRDinfoGen::fgkNclTPC = 70;\r
+const Float_t AliTRDinfoGen::fgkPt = 0.2;\r
+const Float_t AliTRDinfoGen::fgkEta = 0.9;\r
+AliTRDReconstructor* AliTRDinfoGen::fgReconstructor(NULL);\r
+AliTRDgeometry* AliTRDinfoGen::fgGeo(NULL);\r
+//____________________________________________________________________\r
+AliTRDinfoGen::AliTRDinfoGen()\r
+ :AliAnalysisTaskSE()\r
+ ,fEvTrigger(NULL)\r
+ ,fESDev(NULL)\r
+ ,fMCev(NULL)\r
+ ,fEventCut(NULL)\r
+ ,fTrackCut(NULL)\r
+ ,fV0Cut(NULL)\r
+ ,fOCDB("local://$ALICE_ROOT/OCDB")\r
+ ,fTrackInfo(NULL)\r
+ ,fEventInfo(NULL)\r
+ ,fV0Info(NULL)\r
+ ,fTracksBarrel(NULL)\r
+ ,fTracksSA(NULL)\r
+ ,fTracksKink(NULL)\r
+ ,fV0List(NULL)\r
+ ,fContainer(NULL)\r
+ ,fDebugStream(NULL)\r
+{\r
+ //\r
+ // Default constructor\r
+ //\r
+ SetNameTitle("TRDinfoGen", "MC-REC TRD-track list generator");\r
+}\r
+\r
+//____________________________________________________________________\r
+AliTRDinfoGen::AliTRDinfoGen(char* name)\r
+ :AliAnalysisTaskSE(name)\r
+ ,fEvTrigger(NULL)\r
+ ,fESDev(NULL)\r
+ ,fMCev(NULL)\r
+ ,fEventCut(NULL)\r
+ ,fTrackCut(NULL)\r
+ ,fV0Cut(NULL)\r
+ ,fOCDB("local://$ALICE_ROOT/OCDB")\r
+ ,fTrackInfo(NULL)\r
+ ,fEventInfo(NULL)\r
+ ,fV0Info(NULL)\r
+ ,fTracksBarrel(NULL)\r
+ ,fTracksSA(NULL)\r
+ ,fTracksKink(NULL)\r
+ ,fV0List(NULL)\r
+ ,fContainer(NULL)\r
+ ,fDebugStream(NULL)\r
+{\r
+ //\r
+ // Default constructor\r
+ //\r
+ SetTitle("MC-REC TRD-track list generator");\r
+ DefineOutput(AliTRDpwg1Helper::kTracksBarrel, TObjArray::Class());\r
+ DefineOutput(AliTRDpwg1Helper::kTracksSA, TObjArray::Class());\r
+ DefineOutput(AliTRDpwg1Helper::kTracksKink, TObjArray::Class());\r
+ DefineOutput(AliTRDpwg1Helper::kEventInfo, AliTRDeventInfo::Class());\r
+ DefineOutput(AliTRDpwg1Helper::kV0List, TObjArray::Class());\r
+ DefineOutput(AliTRDpwg1Helper::kMonitor, TObjArray::Class()); // histogram list\r
+}\r
+\r
+//____________________________________________________________________\r
+AliTRDinfoGen::~AliTRDinfoGen()\r
+{\r
+// Destructor\r
+ if(fgGeo) delete fgGeo;\r
+ if(fgReconstructor) delete fgReconstructor;\r
+ if(fDebugStream) delete fDebugStream;\r
+ if(fEvTrigger) delete fEvTrigger;\r
+ if(fV0Cut) delete fV0Cut;\r
+ if(fTrackCut) delete fTrackCut;\r
+ if(fEventCut) delete fEventCut;\r
+ if(fTrackInfo) delete fTrackInfo; fTrackInfo = NULL;\r
+ if(fEventInfo) delete fEventInfo; fEventInfo = NULL;\r
+ if(fV0Info) delete fV0Info; fV0Info = NULL;\r
+ if(fTracksBarrel){ \r
+ fTracksBarrel->Delete(); delete fTracksBarrel;\r
+ fTracksBarrel = NULL;\r
+ }\r
+ if(fTracksSA){ \r
+ fTracksSA->Delete(); delete fTracksSA;\r
+ fTracksSA = NULL;\r
+ }\r
+ if(fTracksKink){ \r
+ fTracksKink->Delete(); delete fTracksKink;\r
+ fTracksKink = NULL;\r
+ }\r
+ if(fV0List){ \r
+ fV0List->Delete(); \r
+ delete fV0List;\r
+ fV0List = NULL;\r
+ }\r
+ if(fContainer){ \r
+ fContainer->Delete(); \r
+ delete fContainer;\r
+ fContainer = NULL;\r
+ }\r
+}\r
+\r
+//____________________________________________________________________\r
+Bool_t AliTRDinfoGen::GetRefFigure(Int_t)\r
+{\r
+// General graphs for PWG1/TRD train\r
+ if(!gPad){\r
+ AliWarning("Please provide a canvas to draw results.");\r
+ return kFALSE;\r
+ }\r
+ fContainer->At(0)->Draw("bar");\r
+ return kTRUE;\r
+}\r
+\r
+//____________________________________________________________________\r
+void AliTRDinfoGen::UserCreateOutputObjects()\r
+{ \r
+ //\r
+ // Create Output Containers (TObjectArray containing 1D histograms)\r
+ //\r
+ \r
+ fTrackInfo = new AliTRDtrackInfo();\r
+ fEventInfo = new AliTRDeventInfo();\r
+ fV0Info = new AliTRDv0Info();\r
+ fTracksBarrel = new TObjArray(200); fTracksBarrel->SetOwner(kTRUE);\r
+ fTracksSA = new TObjArray(20); fTracksSA->SetOwner(kTRUE);\r
+ fTracksKink = new TObjArray(20); fTracksKink->SetOwner(kTRUE);\r
+ fV0List = new TObjArray(10); fV0List->SetOwner(kTRUE);\r
+\r
+ // define general monitor\r
+ fContainer = new TObjArray(1); fContainer->SetOwner(kTRUE);\r
+ TH1 *h=new TH1I("hStat", "Run statistics;Observable;Entries", Int_t(kNObjects), -0.5, Float_t(kNObjects)-0.5);\r
+ TAxis *ax(h->GetXaxis());\r
+ ax->SetBinLabel(Int_t(kTracksESD) + 1, "ESD");\r
+ ax->SetBinLabel(Int_t(kTracksMC) + 1, "MC");\r
+ ax->SetBinLabel(Int_t(kV0) + 1, "V0");\r
+ ax->SetBinLabel(Int_t(kTPC) + 1, "TPC");\r
+ ax->SetBinLabel(Int_t(kTRDin) + 1, "TRDin");\r
+ ax->SetBinLabel(Int_t(kTRDout) + 1, "TRDout");\r
+ ax->SetBinLabel(Int_t(kBarrel) + 1, "Barrel");\r
+ ax->SetBinLabel(Int_t(kBarrelMC) + 1, "BarrelMC");\r
+ ax->SetBinLabel(Int_t(kSA) + 1, "SA");\r
+ ax->SetBinLabel(Int_t(kSAMC) + 1, "SAMC");\r
+ ax->SetBinLabel(Int_t(kKink) + 1, "Kink");\r
+ ax->SetBinLabel(Int_t(kKinkMC) + 1, "KinkMC");\r
+ ax->SetBinLabel(Int_t(kBarrelFriend) + 1, "BFriend");\r
+ ax->SetBinLabel(Int_t(kSAFriend) + 1, "SFriend");\r
+ fContainer->AddAt(h, 0);\r
+ PostData(AliTRDpwg1Helper::kMonitor, fContainer);\r
+}\r
+\r
+//____________________________________________________________________\r
+Bool_t AliTRDinfoGen::Load(const Char_t *file, const Char_t *dir, const Char_t *name)\r
+{\r
+// Load data from performance file\r
+\r
+ if(!TFile::Open(file)){\r
+ AliWarning(Form("Couldn't open file %s.", file));\r
+ return kFALSE;\r
+ }\r
+ if(dir){\r
+ if(!gFile->cd(dir)){\r
+ AliWarning(Form("Couldn't cd to %s in %s.", dir, file));\r
+ return kFALSE;\r
+ }\r
+ }\r
+ TObjArray *o(NULL);\r
+ const Char_t *tn=(name ? name : GetName());\r
+ if(!(o = (TObjArray*)gDirectory->Get(tn))){\r
+ AliWarning(Form("Missing histogram container %s.", tn));\r
+ return kFALSE;\r
+ }\r
+ fContainer = (TObjArray*)o->Clone(GetName());\r
+ gFile->Close();\r
+ return kTRUE;\r
+}\r
+\r
+//____________________________________________________________________\r
+void AliTRDinfoGen::UserExec(Option_t *){\r
+ //\r
+ // Run the Analysis\r
+ //\r
+\r
+ fTracksBarrel->Delete();\r
+ fTracksSA->Delete();\r
+ fTracksKink->Delete();\r
+ fV0List->Delete();\r
+ fEventInfo->Delete("");\r
+\r
+ fESDev = dynamic_cast<AliESDEvent*>(InputEvent());\r
+ if(!fESDev){\r
+ AliError("Failed retrieving ESD event");\r
+ return;\r
+ }\r
+ // WARNING\r
+ // This part may conflict with other detectors !!\r
+ if(!IsInitOCDB()){ \r
+ AliInfo("Initializing OCDB ...");\r
+ // prepare OCDB access\r
+ AliCDBManager* ocdb = AliCDBManager::Instance();\r
+ ocdb->SetDefaultStorage(fOCDB.Data());\r
+ ocdb->SetRun(fESDev->GetRunNumber());\r
+ // create geo manager\r
+ AliCDBEntry* obj(NULL);\r
+ if(!(obj = ocdb->Get(AliCDBPath("GRP", "Geometry", "Data")))){\r
+ AliError("GEOMETRY failed initialization.");\r
+ } else {\r
+ AliGeomManager::SetGeometry((TGeoManager*)obj->GetObject());\r
+ AliGeomManager::GetNalignable("TRD");\r
+ AliGeomManager::ApplyAlignObjsFromCDB("TRD");\r
+ }\r
+ fgGeo = new AliTRDgeometry;\r
+ fgGeo->CreateClusterMatrixArray();\r
+ //init magnetic field\r
+ if(!TGeoGlobalMagField::Instance()->IsLocked() &&\r
+ !fESDev->InitMagneticField()){\r
+ AliError("MAGNETIC FIELD failed initialization.");\r
+ }\r
+\r
+ // set no of time bins\r
+ AliTRDtrackerV1::SetNTimeBins(AliTRDcalibDB::Instance()->GetNumberOfTimeBinsDCS());\r
+ AliInfo(Form("OCDB : Loc[%s] Run[%d] TB[%d]", fOCDB.Data(), ocdb->GetRun(), AliTRDtrackerV1::GetNTimeBins()));\r
+\r
+ // set reco param valid for this run/event\r
+ AliInfo(Form("Initializing TRD reco params for EventSpecie[%d]...",\r
+ fESDev->GetEventSpecie()));\r
+ fgReconstructor = new AliTRDReconstructor();\r
+ if(!(obj = ocdb->Get(AliCDBPath("TRD", "Calib", "RecoParam")))){\r
+ AliError("RECO PARAM failed initialization.");\r
+ fgReconstructor->SetRecoParam(AliTRDrecoParam::GetLowFluxParam());\r
+ } else {\r
+ obj->PrintMetaData();\r
+ TObjArray *recos((TObjArray*)obj->GetObject());\r
+ for(Int_t ireco(0); ireco<recos->GetEntriesFast(); ireco++){\r
+ AliTRDrecoParam *reco((AliTRDrecoParam*)recos->At(ireco));\r
+ Int_t es(reco->GetEventSpecie());\r
+ if(!(es&fESDev->GetEventSpecie())) continue;\r
+ fgReconstructor->SetRecoParam(reco);\r
+ TString s;\r
+ if(es&AliRecoParam::kLowMult) s="LowMult";\r
+ else if(es&AliRecoParam::kHighMult) s="HighMult";\r
+ else if(es&AliRecoParam::kCosmic) s="Cosmic";\r
+ else if(es&AliRecoParam::kCalib) s="Calib";\r
+ else s="Unknown";\r
+ AliInfo(Form("Using reco params for %s", s.Data()));\r
+ break;\r
+ }\r
+ }\r
+ SetInitOCDB();\r
+ }\r
+\r
+ // link MC if available\r
+ fMCev = MCEvent();\r
+\r
+ // event selection : trigger cut\r
+ if(UseLocalEvSelection() && fEvTrigger){ \r
+ Bool_t kTRIGGERED(kFALSE);\r
+ std::auto_ptr<TObjArray> trig(fEvTrigger->Tokenize(" "));\r
+ for(Int_t itrig=trig->GetEntriesFast(); itrig--;){\r
+ const Char_t *trigClass(((TObjString*)(*trig)[itrig])->GetName());\r
+ if(fESDev->IsTriggerClassFired(trigClass)) {\r
+ AliDebug(2, Form("Ev[%4d] Trigger[%s]", fESDev->GetEventNumberInFile(), trigClass));\r
+ kTRIGGERED = kTRUE;\r
+ break; \r
+ }\r
+ }\r
+ if(!kTRIGGERED){ \r
+ AliDebug(2, Form("Reject Ev[%4d] Trigger", fESDev->GetEventNumberInFile()));\r
+ return;\r
+ }\r
+ // select only physical events\r
+ if(fESDev->GetEventType() != 7){ \r
+ AliDebug(2, Form("Reject Ev[%4d] EvType[%d]", fESDev->GetEventNumberInFile(), fESDev->GetEventType()));\r
+ return;\r
+ }\r
+ }\r
+\r
+ // if the required trigger is a collision trigger then apply event vertex cut\r
+ if(UseLocalEvSelection() && IsCollision()){\r
+ const AliESDVertex *vertex = fESDev->GetPrimaryVertex();\r
+ if(TMath::Abs(vertex->GetZv())<1.e-10 || \r
+ TMath::Abs(vertex->GetZv())>fgkEvVertexZ || \r
+ vertex->GetNContributors()<fgkEvVertexN) {\r
+ AliDebug(2, Form("Reject Ev[%4d] Vertex Zv[%f] Nv[%d]", fESDev->GetEventNumberInFile(), TMath::Abs(vertex->GetZv()), vertex->GetNContributors()));\r
+ return;\r
+ }\r
+ }\r
+\r
+ if(fEventCut && !fEventCut->IsSelected(fESDev, IsCollision())) return;\r
+\r
+ if(!fESDfriend){\r
+ AliError("Failed retrieving ESD friend event");\r
+ return;\r
+ }\r
+ if(HasMCdata() && !fMCev){\r
+ AliError("Failed retrieving MC event");\r
+ return;\r
+ }\r
+\r
+ new(fEventInfo)AliTRDeventInfo(fESDev->GetHeader(), const_cast<AliESDRun *>(fESDev->GetESDRun()));\r
+ // Determine centrality\r
+ // Author: Ionut Arsene <I.C.Arsene@gsi.de>\r
+ Int_t centralityBin = -1;\r
+ AliDebug(2, Form("Beam Type: %s", fESDev->GetESDRun()->GetBeamType()));\r
+ if(TString(fESDev->GetESDRun()->GetBeamType()).Contains("Pb-Pb")){\r
+ centralityBin = 4;\r
+ const AliMultiplicity *mult = fESDev->GetMultiplicity();\r
+ Double_t zdcNeutronEnergy = fESDev->GetZDCN1Energy()+fESDev->GetZDCN2Energy();\r
+ Double_t itsNTracklets = mult->GetNumberOfTracklets();\r
+ Double_t centralitySlopes[6] = {0.0, 4.0, 8.0, 20.0, 50.0, 1000000.};\r
+ AliDebug(1, Form("zdcNeutronEnergy: %f, itsNTracklets: %f\n", zdcNeutronEnergy, itsNTracklets));\r
+ for(Int_t iCent=1; iCent<=5; ++iCent) {\r
+ if(zdcNeutronEnergy>centralitySlopes[iCent-1]*itsNTracklets && zdcNeutronEnergy<centralitySlopes[iCent]*itsNTracklets)\r
+ centralityBin=iCent - 1;\r
+ }\r
+ AliDebug(1, Form("Centrality Class: %d", centralityBin));\r
+ }\r
+ fEventInfo->SetCentrality(centralityBin);\r
+\r
+ Bool_t *trackMap(NULL);\r
+ AliStack * mStack(NULL);\r
+ Int_t nTracksMC = HasMCdata() ? fMCev->GetNumberOfTracks() : 0, nTracksESD = fESDev->GetNumberOfTracks();\r
+ if(HasMCdata()){\r
+ mStack = fMCev->Stack();\r
+ if(!mStack){\r
+ AliError("Failed retrieving MC Stack");\r
+ return;\r
+ }\r
+ trackMap = new Bool_t[nTracksMC];\r
+ memset(trackMap, 0, sizeof(Bool_t) * nTracksMC);\r
+ }\r
+ \r
+ Double32_t dedx[100]; Int_t nSlices(0);\r
+ Int_t nTRDout(0), nTRDin(0), nTPC(0)\r
+ ,nclsTrklt\r
+ ,nBarrel(0), nSA(0), nKink(0)\r
+ ,nBarrelFriend(0), nSAFriend(0)\r
+ ,nBarrelMC(0), nSAMC(0), nKinkMC(0);\r
+ AliESDtrack *esdTrack = NULL;\r
+ AliESDfriendTrack *esdFriendTrack = NULL;\r
+ TObject *calObject = NULL;\r
+ AliTRDtrackV1 *track = NULL;\r
+ AliTRDseedV1 *tracklet = NULL;\r
+ AliTRDcluster *cl = NULL;\r
+\r
+\r
+ // LOOP 0 - over ESD v0s\r
+ Float_t bField(fESDev->GetMagneticField());\r
+ AliESDv0 *v0(NULL);\r
+ Int_t v0pid[AliPID::kSPECIES];\r
+ for(Int_t iv0(0); iv0<fESDev->GetNumberOfV0s(); iv0++){\r
+ if(!(v0 = fESDev->GetV0(iv0))) continue;\r
+ // register v0\r
+ if(fV0Cut) new(fV0Info) AliTRDv0Info(*fV0Cut);\r
+ else new(fV0Info) AliTRDv0Info();\r
+ fV0Info->SetMagField(bField);\r
+ fV0Info->SetV0tracks(fESDev->GetTrack(v0->GetPindex()), fESDev->GetTrack(v0->GetNindex()));\r
+ fV0Info->SetV0Info(v0);\r
+ fV0List->Add(new AliTRDv0Info(*fV0Info));// kFOUND=kFALSE;\r
+ }\r
+\r
+\r
+ // LOOP 1 - over ESD tracks\r
+ AliTRDv0Info *v0info=NULL;\r
+ for(Int_t itrk = 0; itrk < nTracksESD; itrk++){\r
+ new(fTrackInfo) AliTRDtrackInfo();\r
+ esdTrack = fESDev->GetTrack(itrk);\r
+ AliDebug(3, Form("\n%3d ITS[%d] TPC[%d] TRD[%d]\n", itrk, esdTrack->GetNcls(0), esdTrack->GetNcls(1), esdTrack->GetNcls(2)));\r
+\r
+ if(esdTrack->GetStatus()&AliESDtrack::kTPCout) nTPC++;\r
+ if(esdTrack->GetStatus()&AliESDtrack::kTRDout) nTRDout++;\r
+ if(esdTrack->GetStatus()&AliESDtrack::kTRDin) nTRDin++;\r
+\r
+ // look at external track param\r
+ const AliExternalTrackParam *op = esdTrack->GetOuterParam();\r
+ Double_t xyz[3];\r
+ if(op){\r
+ op->GetXYZ(xyz);\r
+ op->Global2LocalPosition(xyz, op->GetAlpha());\r
+ AliDebug(3, Form("op @ X[%7.3f]\n", xyz[0]));\r
+ }\r
+\r
+ // read MC info\r
+ Int_t fPdg = -1;\r
+ Int_t label = -1; UInt_t alab=UINT_MAX;\r
+ if(HasMCdata()){\r
+ label = esdTrack->GetLabel(); \r
+ alab = TMath::Abs(label);\r
+ // register the track\r
+ if(alab < UInt_t(nTracksMC)){ \r
+ trackMap[alab] = kTRUE; \r
+ } else { \r
+ AliError(Form("MC label[%d] outside scope for Ev[%d] Trk[%d].", label, (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), itrk));\r
+ continue; \r
+ }\r
+ AliMCParticle *mcParticle = NULL; \r
+ if(!(mcParticle = (AliMCParticle*) fMCev->GetTrack(alab))){\r
+ AliError(Form("MC particle label[%d] missing for Ev[%d] Trk[%d].", label, (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), itrk));\r
+ continue;\r
+ }\r
+ fPdg = mcParticle->Particle()->GetPdgCode();\r
+ Int_t nRefs = mcParticle->GetNumberOfTrackReferences();\r
+ Int_t iref = 0; AliTrackReference *ref = NULL; \r
+ while(iref<nRefs){\r
+ ref = mcParticle->GetTrackReference(iref);\r
+ if(ref->LocalX() > fgkTPC) break;\r
+ iref++;\r
+ }\r
+\r
+ fTrackInfo->SetMC();\r
+ fTrackInfo->SetPDG(fPdg);\r
+ fTrackInfo->SetPrimary(mcParticle->Particle()->IsPrimary());\r
+ fTrackInfo->SetLabel(label);\r
+ Int_t jref = iref;//, kref = 0;\r
+ while(jref<nRefs){\r
+ ref = mcParticle->GetTrackReference(jref);\r
+ if(ref->LocalX() > fgkTRD) break;\r
+ AliDebug(4, Form(" trackRef[%2d (%2d)] @ %7.3f OK", jref-iref, jref, ref->LocalX()));\r
+ fTrackInfo->AddTrackRef(ref);\r
+ jref++;\r
+ }\r
+ AliDebug(3, Form("NtrackRefs[%d(%d)]", fTrackInfo->GetNTrackRefs(), nRefs));\r
+ }\r
+\r
+ // copy some relevant info to TRD track info\r
+ fTrackInfo->SetStatus(esdTrack->GetStatus());\r
+ fTrackInfo->SetTrackId(esdTrack->GetID());\r
+ Double_t p[AliPID::kSPECIES]; esdTrack->GetTRDpid(p);\r
+ fTrackInfo->SetESDpid(p);\r
+ fTrackInfo->SetESDpidQuality(esdTrack->GetTRDntrackletsPID());\r
+ if(!nSlices) nSlices = esdTrack->GetNumberOfTRDslices();\r
+ memset(dedx, 0, 100*sizeof(Double32_t));\r
+ Int_t in(0);\r
+ for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++)\r
+ for(Int_t is=0; is<nSlices; is++) \r
+ dedx[in++]=esdTrack->GetTRDslice(il, is);\r
+ for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++) dedx[in++]=esdTrack->GetTRDmomentum(il);\r
+ fTrackInfo->SetSlices(in, dedx);\r
+ fTrackInfo->SetNumberOfClustersRefit(esdTrack->GetNcls(2));\r
+ // some other Informations which we may wish to store in order to find problematic cases\r
+ fTrackInfo->SetKinkIndex(esdTrack->GetKinkIndex(0));\r
+ fTrackInfo->SetTPCncls(static_cast<UShort_t>(esdTrack->GetNcls(1)));\r
+ nclsTrklt = 0;\r
+ \r
+ // set V0pid info\r
+ for(Int_t iv(0); iv<fV0List->GetEntriesFast(); iv++){\r
+ if(!(v0info = (AliTRDv0Info*)fV0List->At(iv))) continue;\r
+ if(!v0info->GetV0Daughter(1) && !v0info->GetV0Daughter(-1)) continue;\r
+ if(!v0info->HasTrack(fTrackInfo)) continue;\r
+ memset(v0pid, 0, AliPID::kSPECIES*sizeof(Int_t));\r
+ fTrackInfo->SetV0();\r
+ for(Int_t is=AliPID::kSPECIES; is--;){v0pid[is] = v0info->GetPID(is, fTrackInfo);}\r
+ fTrackInfo->SetV0pid(v0pid);\r
+ fTrackInfo->SetV0();\r
+ //const AliTRDtrackInfo::AliESDinfo *ei = fTrackInfo->GetESDinfo();\r
+ break;\r
+ }\r
+\r
+ // read REC info\r
+ esdFriendTrack = (fESDfriend->GetNumberOfTracks() > itrk) ? fESDfriend->GetTrack(itrk): NULL;\r
+\r
+ if(esdFriendTrack){\r
+ Int_t icalib = 0;\r
+ while((calObject = esdFriendTrack->GetCalibObject(icalib++))){\r
+ if(strcmp(calObject->IsA()->GetName(),"AliTRDtrackV1") != 0) continue; // Look for the TRDtrack\r
+ if(!(track = dynamic_cast<AliTRDtrackV1*>(calObject))) break;\r
+ AliDebug(4, Form("TRD track OK"));\r
+ // Set the clusters to unused\r
+ for(Int_t ipl = 0; ipl < AliTRDgeometry::kNlayer; ipl++){\r
+ if(!(tracklet = track->GetTracklet(ipl))) continue;\r
+ tracklet->ResetClusterIter();\r
+ while((cl = tracklet->NextCluster())) cl->Use(0);\r
+ }\r
+ fTrackInfo->SetTrack(track);\r
+ break;\r
+ }\r
+ AliDebug(3, Form("Ntracklets[%d]\n", fTrackInfo->GetNTracklets()));\r
+ } else AliDebug(3, "No ESD friends");\r
+ if(op) fTrackInfo->SetOuterParam(op);\r
+\r
+ if(DebugLevel() >= 1){\r
+ AliTRDtrackInfo info(*fTrackInfo);\r
+ (*DebugStream()) << "trackInfo"\r
+ << "TrackInfo.=" << &info\r
+ << "\n";\r
+ info.Delete("");\r
+ }\r
+\r
+ ULong_t status(esdTrack->GetStatus());\r
+ if((status&AliESDtrack::kTPCout)){\r
+ if(!esdTrack->GetKinkIndex(0)){ // Barrel Track Selection\r
+ Bool_t selected(kTRUE);\r
+ if(UseLocalTrkSelection()){\r
+ if(esdTrack->Pt() < fgkPt){ \r
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] Pt[%5.2f]", fESDev->GetEventNumberInFile(), itrk, esdTrack->Pt()));\r
+ selected = kFALSE;\r
+ }\r
+ if(selected && TMath::Abs(esdTrack->Eta()) > fgkEta){\r
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] Eta[%5.2f]", fESDev->GetEventNumberInFile(), itrk, TMath::Abs(esdTrack->Eta())));\r
+ selected = kFALSE;\r
+ }\r
+ if(selected && esdTrack->GetTPCNcls() < fgkNclTPC){ \r
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] NclTPC[%d]", fESDev->GetEventNumberInFile(), itrk, esdTrack->GetTPCNcls()));\r
+ selected = kFALSE;\r
+ }\r
+ Float_t par[2], cov[3];\r
+ esdTrack->GetImpactParameters(par, cov);\r
+ if(IsCollision()){ // cuts on DCA\r
+ if(selected && TMath::Abs(par[0]) > fgkTrkDCAxy){ \r
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] DCAxy[%f]", fESDev->GetEventNumberInFile(), itrk, TMath::Abs(par[0])));\r
+ selected = kFALSE;\r
+ }\r
+ if(selected && TMath::Abs(par[1]) > fgkTrkDCAz){ \r
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] DCAz[%f]", fESDev->GetEventNumberInFile(), itrk, TMath::Abs(par[1])));\r
+ selected = kFALSE;\r
+ }\r
+ } else if(selected && fMCev && !fMCev->IsPhysicalPrimary(alab)){;\r
+ AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] Primary", fESDev->GetEventNumberInFile(), itrk));\r
+ selected = kFALSE;\r
+ }\r
+ }\r
+ if(fTrackCut && !fTrackCut->IsSelected(esdTrack)) selected = kFALSE;\r
+ if(selected){ \r
+ fTracksBarrel->Add(new AliTRDtrackInfo(*fTrackInfo));\r
+ nBarrel++;\r
+ if(fTrackInfo->GetTrack()) \r
+ nBarrelFriend++;\r
+ }\r
+ } else {\r
+ fTracksKink->Add(new AliTRDtrackInfo(*fTrackInfo));\r
+ nKink++;\r
+ }\r
+ } else if((status&AliESDtrack::kTRDout) && !(status&AliESDtrack::kTRDin)){ \r
+ fTracksSA->Add(new AliTRDtrackInfo(*fTrackInfo));\r
+ nSA++;\r
+ if(fTrackInfo->GetTrack()) \r
+ nSAFriend++;\r
+ }\r
+ fTrackInfo->Delete("");\r
+ }\r
+\r
+ // LOOP 2 - over MC tracks which are passing TRD where the track is not reconstructed\r
+ if(HasMCdata()){\r
+ AliDebug(10, "Output of the MC track map:");\r
+ for(Int_t itk = 0; itk < nTracksMC; itk++) AliDebug(10, Form("trackMap[%d] = %s", itk, trackMap[itk] == kTRUE ? "TRUE" : "kFALSE"));\r
+ \r
+ for(Int_t itk = 0; itk < nTracksMC; itk++){\r
+ if(trackMap[itk]) continue;\r
+ AliMCParticle *mcParticle = (AliMCParticle*) fMCev->GetTrack(TMath::Abs(itk));\r
+ Int_t fPdg = mcParticle->Particle()->GetPdgCode();\r
+ Int_t nRefs = mcParticle->GetNumberOfTrackReferences();\r
+ Int_t iref = 0; AliTrackReference *ref = NULL; \r
+ Int_t nRefsTRD = 0;\r
+ new(fTrackInfo) AliTRDtrackInfo();\r
+ fTrackInfo->SetMC();\r
+ fTrackInfo->SetPDG(fPdg);\r
+ while(iref<nRefs){ // count TRD TR\r
+ Bool_t kIN(kFALSE);\r
+ ref = mcParticle->GetTrackReference(iref);\r
+ if(ref->LocalX() > fgkTPC && ref->LocalX() < fgkTRD){\r
+ fTrackInfo->AddTrackRef(ref);\r
+ nRefsTRD++;kIN=kTRUE;\r
+ }\r
+ AliDebug(4, Form(" trackRef[%2d] @ x[%7.3f] %s", iref, ref->LocalX(), kIN?"IN":"OUT"));\r
+ iref++;\r
+ }\r
+ if(!nRefsTRD){\r
+ // In this stage we at least require 1 hit inside TRD. What will be done with this tracks is a task for the \r
+ // analysis job\r
+ fTrackInfo->Delete("");\r
+ continue;\r
+ }\r
+ fTrackInfo->SetPrimary(mcParticle->Particle()->IsPrimary());\r
+ fTrackInfo->SetLabel(itk);\r
+ if(DebugLevel() >= 1){\r
+ AliTRDtrackInfo info(*fTrackInfo);\r
+ (*DebugStream()) << "trackInfo"\r
+ << "TrackInfo.=" << &info\r
+ << "\n";\r
+ info.Delete("");\r
+ }\r
+ AliDebug(3, Form("Add MC track @ label[%d] nTRDrefs[%d].", itk, nRefsTRD));\r
+ // check where the track starts\r
+ ref = mcParticle->GetTrackReference(0);\r
+ if(ref->LocalX() < fgkITS){ \r
+ fTracksBarrel->Add(new AliTRDtrackInfo(*fTrackInfo));\r
+ nBarrelMC++;\r
+ } else if(ref->LocalX() < fgkTPC) {\r
+ fTracksKink->Add(new AliTRDtrackInfo(*fTrackInfo));\r
+ nKinkMC++;\r
+ } else if(nRefsTRD>6){\r
+ fTracksSA->Add(new AliTRDtrackInfo(*fTrackInfo));\r
+ nSAMC++;\r
+ }\r
+ fTrackInfo->Delete("");\r
+ }\r
+ delete[] trackMap;\r
+ }\r
+ AliDebug(1, Form(\r
+ "\nEv[%3d] Tracks: ESD[%d] MC[%d] V0[%d]\n"\r
+ " TPCout[%d] TRDin[%d] TRDout[%d]\n"\r
+ " Barrel[%3d+%3d=%3d] SA[%2d+%2d=%2d] Kink[%2d+%2d=%2d]"\r
+ ,(Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), nTracksESD, nTracksMC, fV0List->GetEntries()\r
+ , nTPC, nTRDin, nTRDout\r
+ ,nBarrel, nBarrelMC, fTracksBarrel->GetEntries()\r
+ ,nSA, nSAMC, fTracksSA->GetEntries()\r
+ ,nKink, nKinkMC, fTracksKink->GetEntries()\r
+ ));\r
+ // save track statistics\r
+ TH1 *h((TH1S*)fContainer->At(0));\r
+ h->Fill(Float_t(kTracksESD), nTracksESD);\r
+ h->Fill(Float_t(kTracksMC), nTracksMC);\r
+ h->Fill(Float_t(kV0), fV0List->GetEntries());\r
+ h->Fill(Float_t(kTPC), nTPC);\r
+ h->Fill(Float_t(kTRDin), nTRDin);\r
+ h->Fill(Float_t(kTRDout), nTRDout);\r
+ h->Fill(Float_t(kBarrel), nBarrel);\r
+ h->Fill(Float_t(kBarrelMC), nBarrelMC);\r
+ h->Fill(Float_t(kSA), nSA);\r
+ h->Fill(Float_t(kSAMC), nSAMC);\r
+ h->Fill(Float_t(kKink), nKink);\r
+ h->Fill(Float_t(kKinkMC), nKinkMC);\r
+ h->Fill(Float_t(kBarrelFriend), nBarrelFriend);\r
+ h->Fill(Float_t(kSAFriend), nSAFriend);\r
+\r
+ PostData(AliTRDpwg1Helper::kTracksBarrel, fTracksBarrel);\r
+ PostData(AliTRDpwg1Helper::kTracksSA, fTracksSA);\r
+ PostData(AliTRDpwg1Helper::kTracksKink, fTracksKink);\r
+ PostData(AliTRDpwg1Helper::kEventInfo, fEventInfo);\r
+ PostData(AliTRDpwg1Helper::kV0List, fV0List);\r
+}\r
+\r
+//____________________________________________________________________\r
+void AliTRDinfoGen::SetLocalV0Selection(const AliTRDv0Info *v0)\r
+{\r
+// Set V0 cuts from outside\r
+\r
+ if(!fV0Cut) fV0Cut = new AliTRDv0Info(*v0);\r
+ else new(fV0Cut) AliTRDv0Info(*v0);\r
+}\r
+\r
+//____________________________________________________________________\r
+void AliTRDinfoGen::SetTrigger(const Char_t *trigger)\r
+{\r
+ if(!fEvTrigger) fEvTrigger = new TString(trigger);\r
+ else (*fEvTrigger) = trigger;\r
+}\r
+\r
+//____________________________________________________________________\r
+TTreeSRedirector* AliTRDinfoGen::DebugStream()\r
+{\r
+// Manage debug stream for task\r
+ if(!fDebugStream){\r
+ TDirectory *savedir = gDirectory;\r
+ fDebugStream = new TTreeSRedirector("TRD.DebugInfoGen.root");\r
+ savedir->cd();\r
+ }\r
+ return fDebugStream;\r
+}\r
+\r
+\r
+\r