1 class AliAnalysisDataContainer;
2 class AliFlowEventCuts;
3 class AliFlowEventTrackCuts;
4 class AliFlowEventTrackSimpleCuts;
5 class AliRDHFCutsD0toKpi;
6 class AliRDHFCutsDStartoKpipi;
8 void AddTaskFlowD2H(TString fileNameCuts, TString folderName, Int_t nDmeson, Int_t myHarmonic,
9 Bool_t bDoQC, Bool_t bDoSPTPC, Bool_t bDoSPVZERO, Bool_t bDoEPTPC, Bool_t bDoEPVZERO,
10 Int_t ptBinWidth, Double_t gapTPC, Double_t etaVZERO1, Double_t etaVZERO2, Double_t etaVZERO3, Double_t etaVZERO4,
11 Bool_t bOldApproach=kFALSE, Bool_t shrinkSP=kFALSE ) {
12 TFile *filecuts = TFile::Open( fileNameCuts.Data() );
13 if( (!filecuts) || ( filecuts && !filecuts->IsOpen()) ){
14 AliFatal("Could not open cuts file.");
17 TString fileName = AliAnalysisManager::GetCommonFileName();
18 fileName.ReplaceAll(".root","");
20 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
21 AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();
23 TString thecuts = folderName;
24 TString sgn = DMesonName( nDmeson ) + Form("w%d",ptBinWidth);
26 //**********************************************************************
28 AliFlowTrackCuts* cutsRFPTPC = new AliFlowTrackCuts( "GlobalRFPTPC" );
29 cutsRFPTPC->SetParamType(AliFlowTrackCuts::kGlobal);
30 cutsRFPTPC->SetPtRange(0.2,5.);
31 cutsRFPTPC->SetEtaRange(-0.8,0.8);
32 cutsRFPTPC->SetMinNClustersTPC(70);
33 cutsRFPTPC->SetMinChi2PerClusterTPC(0.2);
34 cutsRFPTPC->SetMaxChi2PerClusterTPC(4.0);
35 cutsRFPTPC->SetAcceptKinkDaughters(kFALSE);
36 cutsRFPTPC->SetMinimalTPCdedx(10.);
37 cutsRFPTPC->SetAODfilterBit(1);
38 // cutsRFPTPC->SetQA(kTRUE);
39 AliFlowTrackCuts* cutsRFPVZE = new AliFlowTrackCuts( "GlobalRFPVZE" );
40 cutsRFPVZE->SetParamType(AliFlowTrackCuts::kV0);
41 cutsRFPVZE->SetEtaRange(-10,+10);
42 cutsRFPVZE->SetPhiMin(0);
43 cutsRFPVZE->SetPhiMax(TMath::TwoPi());
44 // cutsRFPVZE->SetQA(kTRUE);
45 AliFlowTrackCuts* cutsPOI = new AliFlowTrackCuts( "GlobalPOI" );
46 cutsPOI->SetParamType(AliFlowTrackCuts::kGlobal);
47 cutsPOI->SetPtRange(0.2,-5.);
48 cutsPOI->SetEtaRange(0.8,-0.8);
50 //**********************************************************************
52 AliFlowTrackSimpleCuts *filterPOIQC[50]; // MASS BANDS
53 AliFlowTrackSimpleCuts *filterPOISP[50][2]; // MASS BANDS || ETA || SUBEVENT GAP
54 int myMassBands = MassBands( nDmeson, bOldApproach );
55 for(int mb=0; mb!=myMassBands; ++mb) {
56 filterPOISP[mb][0] = new AliFlowTrackSimpleCuts( Form("FilterPOISP_MB%d_ETANEG",mb) );
57 filterPOISP[mb][0]->SetEtaMin( -0.8 );
58 filterPOISP[mb][0]->SetEtaMax( 0. );
59 filterPOISP[mb][0]->SetMassMin( MassBandLowEdge(nDmeson,mb,bOldApproach) );
60 filterPOISP[mb][0]->SetMassMax( MassBandLowEdge(nDmeson,mb+1,bOldApproach) );
61 filterPOISP[mb][1] = new AliFlowTrackSimpleCuts( Form("FilterPOISP_MB%d_ETAPOS",mb) );
62 filterPOISP[mb][1]->SetEtaMin( 0. );
63 filterPOISP[mb][1]->SetEtaMax( +0.8 );
64 filterPOISP[mb][1]->SetMassMin( MassBandLowEdge(nDmeson,mb,bOldApproach) );
65 filterPOISP[mb][1]->SetMassMax( MassBandLowEdge(nDmeson,mb+1,bOldApproach) );
66 filterPOIQC[mb] = new AliFlowTrackSimpleCuts( Form("FilterPOIQC_MB%d",mb) );
67 filterPOIQC[mb]->SetEtaMin( -0.8 );
68 filterPOIQC[mb]->SetEtaMax( +0.8 );
69 filterPOIQC[mb]->SetMassMin( MassBandLowEdge(nDmeson,mb,bOldApproach) );
70 filterPOIQC[mb]->SetMassMax( MassBandLowEdge(nDmeson,mb+1,bOldApproach) );
73 // * DMESON SELECTOR *************************************************
74 AliAnalysisTaskFlowD2H *taskSel;
76 case ( AliRDHFCuts::kD0toKpiCuts ):
77 AliRDHFCutsD0toKpi *myCutsD0 = (AliRDHFCutsD0toKpi*)filecuts->Get("D0toKpiCuts");
79 AliFatal("Problems reaching D0toKpiCuts");
81 taskSel = new AliAnalysisTaskFlowD2H( Form("TaskD0Selector_%s",thecuts.Data()),
82 cutsRFPTPC, cutsRFPVZE, myCutsD0, nDmeson );
84 case ( AliRDHFCuts::kDstarCuts ):
85 AliRDHFCutsDStartoKpipi *myCutsDStar = (AliRDHFCutsDStartoKpipi*)filecuts->Get("DStartoKpipiCuts");
87 AliFatal("Problems reaching DStarToKpipiCuts");
89 taskSel = new AliAnalysisTaskFlowD2H( Form("TaskDStarSelector_%s",thecuts.Data()),
90 cutsRFPTPC, cutsRFPVZE, myCutsDStar, nDmeson);
92 case (AliRDHFCuts::kDplusCuts):
93 AliRDHFCutsDplustoKpipi *myCutsDplus = (AliRDHFCutsDplustoKpipi*)filecuts->Get("AnalysisCuts");
95 AliFatal("Problems reaching AnalysisCuts");
97 taskSel = new AliAnalysisTaskFlowD2H( Form("TaskDplusSelector_%s",thecuts.Data()),
98 cutsRFPTPC, cutsRFPVZE, myCutsDplus, nDmeson );
101 taskSel->SetCommonConstants( MassBins(nDmeson), MinMass(nDmeson), MaxMass(nDmeson), ptBinWidth );
103 //taskSel->SelectCollisionCandidates(trigger);
104 //taskSel->SetDebug();
105 AliAnalysisDataContainer *coutHist = mgr->CreateContainer( Form("%sSelector_%s",sgn.Data(),thecuts.Data()),
106 TList::Class(),AliAnalysisManager::kOutputContainer,
107 Form("%s.root:FlowD2H_%s",fileName.Data(),thecuts.Data()) );
108 AliAnalysisDataContainer *exc_TPC = mgr->CreateContainer( Form("TPCEventWithCandidates_%s_%s",sgn.Data(),thecuts.Data()),
109 AliFlowEventSimple::Class(),
110 AliAnalysisManager::kExchangeContainer );
111 AliAnalysisDataContainer *exc_VZE = mgr->CreateContainer( Form("VZEEventWithCandidates_%s_%s",sgn.Data(),thecuts.Data()),
112 AliFlowEventSimple::Class(),
113 AliAnalysisManager::kExchangeContainer );
114 mgr->AddTask(taskSel);
115 mgr->ConnectOutput(taskSel,1,coutHist);
116 mgr->ConnectOutput(taskSel,2,exc_TPC);
117 mgr->ConnectOutput(taskSel,3,exc_VZE);
118 mgr->ConnectInput (taskSel,0,cinput1);
120 // * HANGING ANALYSIS TASKS ********************************************
121 int harm = myHarmonic;
122 for(int mb=0; mb!=myMassBands; ++mb) {
124 AddQCmethod( Form("%sQCTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), harm, exc_TPC, filterPOIQC[mb]);
127 AddSPmethod( Form("%sSPVZEMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), etaVZERO1, etaVZERO2, etaVZERO3, etaVZERO4,
128 "Qa", harm, exc_VZE, 0, filterPOIQC[mb], NULL, false, shrinkSP );
129 AddSPmethod( Form("%sSPVZEMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), etaVZERO1, etaVZERO2, etaVZERO3, etaVZERO4,
130 "Qb", harm, exc_VZE, 0, filterPOIQC[mb], NULL, false, shrinkSP );
133 AddSPmethod( Form("%sEPVZEMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), etaVZERO1, etaVZERO2, etaVZERO3, etaVZERO4,
134 "QaQb", harm, exc_VZE, 0, filterPOIQC[mb], NULL, true, shrinkSP );
137 for(int eg=0; eg!=2; ++eg) {
138 AddSPmethod( Form("%sSPTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), -0.8, -gapTPC*eg, +gapTPC*eg, +0.8,
139 "Qa", harm, exc_TPC, eg, filterPOISP[mb][1], NULL, false, shrinkSP );
140 AddSPmethod( Form("%sSPTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), -0.8, -gapTPC*eg, +gapTPC*eg, +0.8,
141 "Qb", harm, exc_TPC, eg, filterPOISP[mb][0], NULL, false, shrinkSP );
145 AddSPmethod( Form("%sEPTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), -0.8, -0.0, +0.0, +0.8,
146 "QaQb", harm, exc_TPC, 0, filterPOIQC[mb], NULL, true, shrinkSP );
151 void AddSPmethod(char *name, char *fileName, char *thecuts,
152 double minEtaA, double maxEtaA, double minEtaB, double maxEtaB,
153 char *Qvector, int harmonic, AliAnalysisDataContainer *flowEvent, int eg,
154 AliFlowTrackSimpleCuts *cutsPOI=NULL, AliFlowTrackSimpleCuts *cutsRFP=NULL,
155 bool bEP, bool shrink=false ) {
156 TString myFolder = Form("v%d_%s",harmonic,thecuts);
157 TString myNameSP = Form("%sSPv%d%sGAP%d_%s",name,harmonic,Qvector,eg,thecuts);
158 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
159 AliAnalysisDataContainer *flowEvent2 = mgr->CreateContainer( Form("Filter_%s", myNameSP.Data()),
160 AliFlowEventSimple::Class(),
161 AliAnalysisManager::kExchangeContainer );
162 AliAnalysisTaskFilterFE *tskFilter = new AliAnalysisTaskFilterFE( Form("TaskFilter_%s",myNameSP.Data()),
164 tskFilter->SetSubeventEtaRange( minEtaA, maxEtaA, minEtaB, maxEtaB );
165 mgr->AddTask(tskFilter);
166 mgr->ConnectInput( tskFilter,0,flowEvent);
167 mgr->ConnectOutput(tskFilter,1,flowEvent2);
168 AliAnalysisDataContainer *outSP = mgr->CreateContainer( myNameSP.Data(),
169 TList::Class(),AliAnalysisManager::kOutputContainer,
170 Form("%s.root:FlowD2H_SP_%s",fileName,myFolder.Data()) );
171 AliAnalysisTaskScalarProduct *tskSP = new AliAnalysisTaskScalarProduct( Form("TaskScalarProduct_%s",
172 myNameSP.Data()),kFALSE);
173 tskSP->SetApplyCorrectionForNUA(kTRUE);
174 tskSP->SetHarmonic(harmonic);
175 tskSP->SetTotalQvector(Qvector);
176 if(bEP) tskSP->SetBehaveAsEP();
177 if(shrink) tskSP->SetBookOnlyBasicCCH(kTRUE);
179 mgr->ConnectInput( tskSP,0,flowEvent2);
180 mgr->ConnectOutput(tskSP,1,outSP);
183 void AddQCmethod(char *name, char *fileName, char *thecuts,
184 int harmonic, AliAnalysisDataContainer *flowEvent,
185 AliFlowTrackSimpleCuts *cutsPOI=NULL, AliFlowTrackSimpleCuts *cutsRFP=NULL) {
186 TString myFolder = Form("v%d_%s",harmonic,thecuts);
187 TString myName = Form("%sv%d_%s",name,harmonic,thecuts);
188 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
189 AliAnalysisDataContainer *flowEvent2 = mgr->CreateContainer( Form("Filter_%s", myName.Data()),
190 AliFlowEventSimple::Class(),
191 AliAnalysisManager::kExchangeContainer );
192 AliAnalysisTaskFilterFE *tskFilter = new AliAnalysisTaskFilterFE( Form("TaskFilter_%s",myName.Data()),
194 mgr->AddTask(tskFilter);
195 mgr->ConnectInput( tskFilter,0,flowEvent);
196 mgr->ConnectOutput(tskFilter,1,flowEvent2);
197 AliAnalysisDataContainer *outQC = mgr->CreateContainer( myName.Data(),
198 TList::Class(),AliAnalysisManager::kOutputContainer,
199 Form("%s.root:FlowD2H_QC_%s",fileName,myFolder.Data()) );
200 AliAnalysisTaskQCumulants *tskQC = new AliAnalysisTaskQCumulants( Form("TaskQCumulants_%s",
201 myName.Data()),kFALSE);
202 tskQC->SetApplyCorrectionForNUA(kTRUE);
203 tskQC->SetHarmonic(harmonic);
204 tskQC->SetBookOnlyBasicCCH(kTRUE);
206 mgr->ConnectInput( tskQC,0,flowEvent2);
207 mgr->ConnectOutput(tskQC,1,outQC);
210 int MassBands( int nDmeson, bool bOldApproach=false ) {
212 case ( AliRDHFCuts::kD0toKpiCuts ):
213 if(bOldApproach) return 5;
215 case ( AliRDHFCuts::kDstarCuts ):
220 double MassBandLowEdge( int nDmeson, int mb, bool bOldApproach=false ) {
222 case ( AliRDHFCuts::kD0toKpiCuts ): // 2 + 20 + 4
223 double lowEdgeMinimal[5+1] = {1.75,1.80,1.83,1.90,1.93,2.03};
224 double lowEdge[26+1] = { 1.66, 1.71, 1.76, 1.77, 1.78, 1.79, 1.80, 1.81, 1.82, 1.83,
225 1.84, 1.85, 1.86, 1.87, 1.88, 1.89, 1.90, 1.91, 1.92, 1.93,
226 1.94, 1.95, 1.96, 2.01, 2.06, 2.11, 2.16 };
227 if(bOldApproach) return lowEdgeMinimal[mb];
228 else return lowEdge[mb];
229 case ( AliRDHFCuts::kDstarCuts ): // 2 + 10 + 3
230 double lowEdge[15+1] = { 0.138, 0.140, 0.142, 0.143, 0.144, 0.145, 0.146, 0.147, 1.148, 0.149,
231 0.150, 0.151, 0.152, 0.154, 0.156, 0.158 };
236 double MinMass( int nDmeson ) {
238 case ( AliRDHFCuts::kD0toKpiCuts ):
240 case ( AliRDHFCuts::kDstarCuts ):
245 double MaxMass( int nDmeson ) {
247 case ( AliRDHFCuts::kD0toKpiCuts ):
249 case ( AliRDHFCuts::kDstarCuts ):
254 int MassBins( int nDmeson ) {
256 case ( AliRDHFCuts::kD0toKpiCuts ):
258 case ( AliRDHFCuts::kDstarCuts ):
263 TString DMesonName( int nDmeson ) {
266 case ( AliRDHFCuts::kD0toKpiCuts ):
267 toReturn = "D0"; break;
268 case ( AliRDHFCuts::kDstarCuts ):
269 toReturn = "DStar"; break;