1 #ifndef ALIDNDPTTRACKDUMPTASK_H
2 #define ALIDNDPTTRACKDUMPTASK_H
4 //------------------------------------------------------------------------------
5 // Task to dump track information
6 // TPC constrained and TP+ITS combined
7 // for outliers analysis.
9 // Author: J.Otwinowski 19/06/2011
10 //------------------------------------------------------------------------------
14 class AlidNdPtEventCuts;
15 class AlidNdPtAcceptanceCuts;
16 class AliESDtrackCuts;
18 class AlidNdPtAnalysis;
19 class AlidNdPtCorrection;
28 class TTreeSRedirector;
30 #include "AliTriggerAnalysis.h"
31 #include "AliAnalysisTaskSE.h"
32 #include "AlidNdPtHelper.h"
34 class AlidNdPtTrackDumpTask : public AliAnalysisTaskSE {
38 AlidNdPtTrackDumpTask(const char *name = "AlidNdPtTrackDumpTask");
39 virtual ~AlidNdPtTrackDumpTask();
41 virtual void UserCreateOutputObjects();
42 virtual void UserExec(Option_t *option);
43 virtual void Terminate(Option_t *);
44 virtual Bool_t Notify();
45 virtual void FinishTaskOutput();
46 void SetUseMCInfo(Bool_t info) { fUseMCInfo = info; }
47 Bool_t IsUseMCInfo() const { return fUseMCInfo; }
50 virtual void ProcessAll(AliESDEvent *const esdEvent=0, AliMCEvent *const mcEvent=0, AliESDfriend *const esdFriend=0);
51 virtual void Process(AliESDEvent *const esdEvent=0, AliMCEvent *const mcEvent=0, AliESDfriend *const esdFriend=0);
52 virtual void ProcessV0(AliESDEvent *const esdEvent=0, AliMCEvent *const mcEvent=0, AliESDfriend *const esdFriend=0);
54 void SetEventCuts(AlidNdPtEventCuts* const cuts) { fdNdPtEventCuts = cuts; }
55 void SetAcceptanceCuts(AlidNdPtAcceptanceCuts* const cuts) { fdNdPtAcceptanceCuts = cuts; }
56 void SetRecAcceptanceCuts(AlidNdPtAcceptanceCuts* const cuts) { fdNdPtRecAcceptanceCuts = cuts; }
57 void SetTrackCuts(AliESDtrackCuts* const cuts) { fEsdTrackCuts = cuts; }
58 void SetTrigger(const AliTriggerAnalysis::Trigger trigger) { fTrigger = trigger; }
59 void SetAnalysisMode(const AlidNdPtHelper::AnalysisMode mode) { fAnalysisMode = mode; }
61 AlidNdPtEventCuts* GetEventCuts() const { return fdNdPtEventCuts; }
62 AlidNdPtAcceptanceCuts* GetAcceptanceCuts() const { return fdNdPtAcceptanceCuts; }
63 AlidNdPtAcceptanceCuts* GetRecAcceptanceCuts() const { return fdNdPtRecAcceptanceCuts; }
64 AliESDtrackCuts* GetTrackCuts() const { return fEsdTrackCuts; }
65 AliTriggerAnalysis::Trigger GetTrigger() const { return fTrigger; }
66 AlidNdPtHelper::AnalysisMode GetAnalysisMode() const { return fAnalysisMode; }
68 TString GetCentralityEstimator() const {return fCentralityEstimator; }
69 void SetCentralityEstimator(TString centEst="V0M") { fCentralityEstimator = centEst; }
71 Bool_t IsFromConversion(const Int_t label, AliStack *const stack);
72 Bool_t IsFromMaterial(const Int_t label, AliStack *const stack);
73 Bool_t IsFromStrangeness(const Int_t label, AliStack *const stack);
74 TParticle *GetMother(TParticle *const particle, AliStack *const stack);
76 Bool_t ConstrainTPCInner(AliExternalTrackParam *const tpcInnerC, const AliESDVertex* vtx, Double_t b[3]);
77 Bool_t ConstrainTrackInner(AliExternalTrackParam *const trackInnerC, const AliESDVertex* vtx, Double_t mass, Double_t b[3]);
80 Int_t GetKFParticle(AliESDv0 *const v0, AliESDEvent * const event, AliKFParticle & kfparticle);
81 Bool_t IsV0Downscaled(AliESDv0 *const v0);
82 Bool_t IsHighDeDxParticle(AliESDtrack * const track);
84 void SetLowPtTrackDownscaligF(Double_t fact) { fLowPtTrackDownscaligF = fact; }
85 void SetLowPtV0DownscaligF(Double_t fact) { fLowPtV0DownscaligF = fact; }
87 void SetProcessAll(Bool_t proc) { fProcessAll = proc; }
91 AliESDEvent *fESD; //! ESD event
92 AliMCEvent *fMC; //! MC event
93 AliESDfriend *fESDfriend; //! ESDfriend event
94 TList* fOutput; //! list send on output slot 0
95 TIterator *fPitList; //! iterator over the output objetcs
97 Bool_t fUseMCInfo; //! use MC information
99 AlidNdPtEventCuts *fdNdPtEventCuts; //! event cuts
100 AlidNdPtAcceptanceCuts *fdNdPtAcceptanceCuts; //! acceptance cuts
101 AlidNdPtAcceptanceCuts *fdNdPtRecAcceptanceCuts; //! additional recontruction acceptance cuts (not used for MC truth)
102 AliESDtrackCuts *fEsdTrackCuts; //! esd track cuts
103 AliTriggerAnalysis::Trigger fTrigger; //! trigger settings
104 AlidNdPtHelper::AnalysisMode fAnalysisMode; //! analysis mode TPC only, TPC + ITS
106 TTreeSRedirector* fTreeSRedirector; //! temp tree to dump output
108 TString fCentralityEstimator; //! use centrality can be "VOM" (default), "FMD", "TRK", "TKL", "CL0", "CL1", "V0MvsFMD", "TKLvsV0M", "ZEMvsZDC"
110 Double_t fLowPtTrackDownscaligF; // low pT track downscaling factor
111 Double_t fLowPtV0DownscaligF; // low pT V0 downscaling factor
112 Double_t fProcessAll; // Calculate all track properties including MC
114 AlidNdPtTrackDumpTask(const AlidNdPtTrackDumpTask&); // not implemented
115 AlidNdPtTrackDumpTask& operator=(const AlidNdPtTrackDumpTask&); // not implemented
117 ClassDef(AlidNdPtTrackDumpTask, 1); // example of analysis