]>
Commit | Line | Data |
---|---|---|
a1d44e31 | 1 | AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec = "AOD",char* bGen = "",UInt_t filterMask = 16, UInt_t iPhysicsSelectionFlag = AliVEvent::kMB,Char_t *jf = "KT", Float_t radius = 0.4,Int_t nSkip = 0,Int_t kWriteAOD = kFALSE,char* deltaFile = "",Float_t ptTrackCut = 0.15); |
2 | ||
bbb42949 | 3 | Int_t kBackgroundMode = 0; |
4 | Float_t kPtTrackCut = 0.15; | |
a1d44e31 | 5 | |
bbb42949 | 6 | AliAnalysisTaskJetCluster *AddTaskJetClusterDelta(UInt_t filterMask = 16,Bool_t kUseAODMC = kFALSE,UInt_t iPhysicsSelectionFlag = AliVEvent::kMB,Char_t *jf = "KT", UInt_t iFlag){ |
a1d44e31 | 7 | AliAnalysisTaskJetCluster *js = 0; |
8 | if(kUseAODMC&&false){// do not use the MC info yet | |
9 | if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.00001); | |
10 | if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.1); | |
11 | if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.2); | |
12 | if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.4); | |
13 | if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.6); | |
14 | if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.8); | |
15 | if(iFlag&(1<<10))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,1.0); | |
16 | } | |
17 | else{ | |
18 | if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.00001); | |
19 | if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.1); | |
20 | if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.2); | |
21 | if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.4); | |
22 | if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.6); | |
23 | if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.8); | |
24 | if(iFlag&(1<<10))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,1.0); | |
25 | } | |
26 | ||
27 | return js; | |
28 | } | |
29 | ||
30 | ||
31 | AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec,char* bGen ,UInt_t filterMask,UInt_t iPhysicsSelectionFlag,Char_t *jf,Float_t radius,Int_t nSkip,Int_t kWriteAOD,char *deltaFile,Float_t ptTrackCut) | |
32 | { | |
33 | // Creates a jet fider task, configures it and adds it to the analysis manager. | |
34 | kPtTrackCut = ptTrackCut; | |
35 | ||
36 | TString outputFile(deltaFile); | |
37 | // Get the pointer to the existing analysis manager via the static access method. | |
38 | //============================================================================== | |
39 | AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager(); | |
40 | if (!mgr) { | |
41 | ::Error("AddTaskJetCluster", "No analysis manager to connect to."); | |
42 | return NULL; | |
43 | } | |
44 | ||
45 | // Check the analysis type using the event handlers connected to the analysis manager. | |
46 | //============================================================================== | |
47 | if (!mgr->GetInputEventHandler()) { | |
48 | ::Error("AddTaskJetCluster", "This task requires an input event handler"); | |
49 | return NULL; | |
50 | } | |
51 | ||
52 | TString type = mgr->GetInputEventHandler()->GetDataType(); | |
53 | TString typeRec(bRec); | |
54 | TString typeGen(bGen); | |
55 | if(!typeRec.Contains("AODextra")) { | |
56 | typeGen.ToUpper(); | |
57 | typeRec.ToUpper(); | |
58 | } | |
59 | cout << "typeRec: " << typeRec << endl; | |
60 | // Create the task and configure it. | |
61 | //=========================================================================== | |
62 | ||
63 | ||
64 | ||
65 | ||
66 | TString cAdd = ""; | |
67 | cAdd += Form("%02d_",(int)((radius+0.01)*10.)); | |
68 | cAdd += Form("B%d",(int)kBackgroundMode); | |
69 | cAdd += Form("_Filter%05d",filterMask); | |
70 | cAdd += Form("_Cut%05d",(int)(1000.*kPtTrackCut)); | |
71 | cAdd += Form("_Skip%02d",nSkip); | |
72 | Printf("%s %E",cAdd.Data(),kPtTrackCut); | |
73 | AliAnalysisTaskJetCluster* pwg4spec = new AliAnalysisTaskJetCluster(Form("JetCluster%s_%s%s",bRec,jf,cAdd.Data())); | |
74 | ||
75 | // or a config file | |
76 | // pwg4spec->SetAnalysisType(AliAnalysisTaskJetCluster::kAnaMC); | |
77 | // if(iAODanalysis)pwg4spec->SetAODInput(kTRUE); | |
78 | // pwg4spec->SetDebugLevel(11); | |
79 | pwg4spec->SetFilterMask(filterMask); | |
80 | // pwg4spec->SetUseGlobalSelection(kTRUE); | |
81 | ||
82 | if(type == "AOD"){ | |
83 | // Assume all jet are produced already | |
84 | pwg4spec->SetAODTrackInput(kTRUE); | |
85 | pwg4spec->SetAODMCInput(kTRUE); | |
86 | } | |
87 | ||
88 | if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD | |
89 | pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCChargedAcceptance); | |
90 | } | |
91 | else if (typeRec.Contains("AODMC2")){ | |
92 | pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCCharged); | |
93 | } | |
94 | else if (typeRec.Contains("AODMC")){ | |
95 | pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCAll); | |
96 | } | |
97 | else if (typeRec.Contains("AODextraonly")) { | |
98 | pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextraonly); | |
99 | pwg4spec->SetTrackPtCut(kPtTrackCut); | |
100 | } | |
101 | else if (typeRec.Contains("AODextra")) { | |
102 | cout << "AliAnalysisTaskJetCluster::kTrackAODextra: " << AliAnalysisTaskJetCluster::kTrackAODextra << endl; | |
103 | pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextra); | |
104 | pwg4spec->SetTrackPtCut(kPtTrackCut); | |
105 | } | |
106 | else if (typeRec.Contains("AOD")) { | |
107 | pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAOD); | |
108 | pwg4spec->SetTrackPtCut(kPtTrackCut); | |
109 | } | |
110 | ||
111 | ||
112 | if(typeGen.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD | |
113 | pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetCluster::kTrackAODMCChargedAcceptance); | |
114 | } | |
115 | else if (typeGen.Contains("AODMC2")){ | |
116 | pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetCluster::kTrackAODMCCharged); | |
117 | } | |
118 | else if (typeGen.Contains("AODMC")){ | |
119 | pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetCluster::kTrackAODMCAll); | |
120 | } | |
121 | else if (typeGen.Contains("AOD")) { | |
122 | pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetCluster::kTrackAOD); | |
123 | } | |
124 | ||
125 | ||
126 | ||
127 | pwg4spec->SetRparam(radius); | |
128 | ||
129 | switch (jf) { | |
130 | case "ANTIKT": | |
131 | pwg4spec->SetAlgorithm(2); // antikt from fastjet/JetDefinition.hh | |
132 | break; | |
133 | case "CA": | |
134 | pwg4spec->SetAlgorithm(1); // CA from fastjet/JetDefinition.hh | |
135 | break; | |
136 | case "KT": | |
137 | pwg4spec->SetAlgorithm(0); // kt from fastjet/JetDefinition.hh | |
138 | break; | |
139 | default: | |
140 | ::Error("AddTaskJetCluster", "Wrong jet finder selected\n"); | |
141 | return 0; | |
142 | } | |
143 | ||
144 | ||
145 | if(kWriteAOD){ | |
146 | if(outputFile.Length())pwg4spec->SetJetOutputFile(outputFile); | |
147 | pwg4spec->SetJetOutputBranch(Form("clusters%s_%s%s",bRec,jf,cAdd.Data())); | |
148 | pwg4spec->SetJetOutputMinPt(0); // store only jets / clusters above a certain threshold | |
149 | } | |
150 | ||
151 | pwg4spec->SetNSkipLeadingRan(nSkip); | |
152 | ||
153 | if(iPhysicsSelectionFlag)pwg4spec->SelectCollisionCandidates(iPhysicsSelectionFlag); | |
154 | ||
155 | mgr->AddTask(pwg4spec); | |
156 | ||
157 | // Create ONLY the output containers for the data produced by the task. | |
158 | // Get and connect other common input/output containers via the manager as below | |
159 | //============================================================================== | |
160 | AliAnalysisDataContainer *coutput1_Spec = mgr->CreateContainer(Form("pwg4cluster_%s_%s_%s%s",bRec,bGen,jf,cAdd.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_cluster_%s_%s_%s%s",AliAnalysisManager::GetCommonFileName(),bRec,bGen,jf,cAdd.Data())); | |
161 | ||
162 | mgr->ConnectInput (pwg4spec, 0, mgr->GetCommonInputContainer()); | |
163 | mgr->ConnectOutput (pwg4spec, 0, mgr->GetCommonOutputContainer()); | |
164 | mgr->ConnectOutput (pwg4spec, 1, coutput1_Spec ); | |
165 | ||
166 | return pwg4spec; | |
167 | } |