]>
Commit | Line | Data |
---|---|---|
1 | #if !defined(__CINT__) || defined(__MAKECINT__) | |
2 | #include <TChain.h> | |
3 | #include <TSystem.h> | |
4 | #include "AliAnalysisManager.h" | |
5 | #include "AliESDInputHandler.h" | |
6 | #include "AliAODHandler.h" | |
7 | #include "AliAnalysisTaskESDfilter.h" | |
8 | #include "AliAnalysisDataContainer.h" | |
9 | #endif | |
10 | ||
11 | void CreateAODfromESD(const char *inFileName = "AliESDs.root", | |
12 | const char *outFileName = "AliAOD.root", | |
13 | Bool_t bKineFilter = kTRUE) | |
14 | { | |
15 | ||
16 | gSystem->Load("libTree"); | |
17 | gSystem->Load("libGeom"); | |
18 | gSystem->Load("libPhysics"); | |
19 | gSystem->Load("libVMC"); | |
20 | gSystem->Load("libSTEERBase"); | |
21 | gSystem->Load("libESD"); | |
22 | gSystem->Load("libAOD"); | |
23 | ||
24 | gSystem->Load("libANALYSIS"); | |
25 | gSystem->Load("libANALYSISalice"); | |
26 | gSystem->Load("libCORRFW"); | |
27 | gSystem->Load("libPWG3base"); | |
28 | gSystem->Load("libPWG3muon"); | |
29 | ||
30 | TChain *chain = new TChain("esdTree"); | |
31 | // Steering input chain | |
32 | chain->Add(inFileName); | |
33 | AliAnalysisManager *mgr = new AliAnalysisManager("ESD to AOD", "Analysis Manager"); | |
34 | ||
35 | // Input | |
36 | AliESDInputHandler* inpHandler = new AliESDInputHandler(); | |
37 | inpHandler->SetReadFriends(kFALSE); | |
38 | inpHandler->SetReadTags(); | |
39 | mgr->SetInputEventHandler (inpHandler); | |
40 | // Output | |
41 | AliAODHandler* aodHandler = new AliAODHandler(); | |
42 | aodHandler->SetOutputFileName(outFileName); | |
43 | mgr->SetOutputEventHandler(aodHandler); | |
44 | ||
45 | // MC Truth | |
46 | if(bKineFilter){ | |
47 | AliMCEventHandler* mcHandler = new AliMCEventHandler(); | |
48 | mgr->SetMCtruthEventHandler(mcHandler); | |
49 | } | |
50 | ||
51 | ||
52 | // Tasks | |
53 | // Filtering of MC particles (decays conversions etc) | |
54 | // this task is also needed to set the MCEventHandler | |
55 | // to the AODHandler, this will not be needed when | |
56 | // AODHandler goes to ANALYSISalice | |
57 | AliAnalysisTaskMCParticleFilter *kinefilter = new AliAnalysisTaskMCParticleFilter("Particle Filter"); | |
58 | if (bKineFilter) mgr->AddTask(kinefilter); | |
59 | ||
60 | // Barrel Tracks | |
61 | AliAnalysisTaskESDfilter *filter = new AliAnalysisTaskESDfilter("Filter"); | |
62 | mgr->AddTask(filter); | |
63 | // Muons | |
64 | AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter"); | |
65 | mgr->AddTask(esdmuonfilter); | |
66 | ||
67 | // Cuts on primary tracks | |
68 | AliESDtrackCuts* esdTrackCutsL = new AliESDtrackCuts("AliESDtrackCuts", "Standard"); | |
69 | esdTrackCutsL->SetMinNClustersTPC(50); | |
70 | esdTrackCutsL->SetMaxChi2PerClusterTPC(3.5); | |
71 | esdTrackCutsL->SetMaxCovDiagonalElements(2, 2, 0.5, 0.5, 2); | |
72 | esdTrackCutsL->SetRequireTPCRefit(kTRUE); | |
73 | esdTrackCutsL->SetMaxDCAToVertexXY(3.0); | |
74 | esdTrackCutsL->SetMaxDCAToVertexZ(3.0); | |
75 | esdTrackCutsL->SetDCAToVertex2D(kTRUE); | |
76 | esdTrackCutsL->SetRequireSigmaToVertex(kFALSE); | |
77 | esdTrackCutsL->SetAcceptKinkDaughters(kFALSE); | |
78 | // ITS stand-alone tracks | |
79 | AliESDtrackCuts* esdTrackCutsITSsa = new AliESDtrackCuts("AliESDtrackCuts", "ITS stand-alone"); | |
80 | esdTrackCutsITSsa->SetRequireITSStandAlone(kTRUE); | |
81 | ||
82 | AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter"); | |
83 | trackFilter->AddCuts(esdTrackCutsL); | |
84 | trackFilter->AddCuts(esdTrackCutsITSsa); | |
85 | ||
86 | // Cuts on V0s | |
87 | AliESDv0Cuts* esdV0Cuts = new AliESDv0Cuts("AliESDv0Cuts", "Standard pp"); | |
88 | esdV0Cuts->SetMinRadius(0.2); | |
89 | esdV0Cuts->SetMaxRadius(200); | |
90 | esdV0Cuts->SetMinDcaPosToVertex(0.05); | |
91 | esdV0Cuts->SetMinDcaNegToVertex(0.05); | |
92 | esdV0Cuts->SetMaxDcaV0Daughters(1.0); | |
93 | esdV0Cuts->SetMinCosinePointingAngle(0.99); | |
94 | AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter"); | |
95 | v0Filter->AddCuts(esdV0Cuts); | |
96 | ||
97 | ||
98 | // | |
99 | filter->SetTrackFilter(trackFilter); | |
100 | filter->SetV0Filter(v0Filter); | |
101 | ||
102 | ||
103 | // Create AOD Tags | |
104 | AliAnalysisTaskTagCreator* tagTask = new AliAnalysisTaskTagCreator("AOD Tag Creator"); | |
105 | mgr->AddTask(tagTask); | |
106 | ||
107 | // Pipelining | |
108 | AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer(); | |
109 | AliAnalysisDataContainer *coutput1 = mgr->GetCommonOutputContainer(); | |
110 | ||
111 | ||
112 | AliAnalysisDataContainer *coutputT | |
113 | = mgr->CreateContainer("cTag", TTree::Class(), AliAnalysisManager::kOutputContainer, "AOD.tag.root"); | |
114 | ||
115 | coutput1->SetSpecialOutput(); | |
116 | coutputT->SetSpecialOutput(); | |
117 | ||
118 | if(bKineFilter) { | |
119 | mgr->ConnectInput (kinefilter, 0, cinput1 ); | |
120 | mgr->ConnectOutput (kinefilter, 0, coutput1 ); | |
121 | } | |
122 | ||
123 | mgr->ConnectInput (filter, 0, cinput1 ); | |
124 | mgr->ConnectOutput(filter, 0, coutput1); | |
125 | ||
126 | mgr->ConnectInput (esdmuonfilter, 0, cinput1 ); | |
127 | // mgr->ConnectOutput(esdmuonfilter, 0, coutput1); | |
128 | ||
129 | mgr->ConnectInput (tagTask, 0, cinput1); | |
130 | mgr->ConnectOutput(tagTask, 1, coutputT); | |
131 | ||
132 | // | |
133 | // Run the analysis | |
134 | // | |
135 | mgr->InitAnalysis(); | |
136 | mgr->PrintStatus(); | |
137 | mgr->StartAnalysis("local", chain); | |
138 | } | |
139 |