2 // Macro designed for use with the AliDptDptInMC task.
4 // Author: Prabhat Pujahari & Claude Pruneau, Wayne State
6 // system: 0: PbPb 1: pPb
7 // singlesOnly: 0: full correlations 1: singles only
8 // useWeights: 0: no 1: yes
9 // centralityMethod: 3: track count 4: V0 centrality 7: V0A centrality for pPb
10 // chargeSet: 0: ++ 1: +- 2: -+ 3: --
11 /////////////////////////////////////////////////////////////////////////////////
13 AliDptDptInMC *AddTaskCorrMC
16 int centralityMethod = 4,
18 int trackFilterBit = 128,
21 double dcaZMin = -3.0,
23 double dcaXYMin = -2.4,
24 double dcaXYMax = 2.4,
26 TString anadata = "MCAODreco",
27 Bool_t NoResonances = kTRUE,
28 Bool_t NoElectron = kTRUE,
29 const char* taskname = "MCTruth",
30 char *inputHistogramFileName = "alien:///alice/cern.ch/user/p/prabhat/MCCalib/MCTruthCaalib.root")
34 // Set Default Configuration of this analysis
35 // ==========================================
38 int rejectPairConversion = 1;
43 double minCentrality[10];
44 double maxCentrality[10];
47 if (centralityMethod == 4)
50 minCentrality[0] = 0.0; maxCentrality[0] = 10.0;
51 minCentrality[1] = 10.0; maxCentrality[1] = 20.0;
52 minCentrality[2] = 20.0; maxCentrality[2] = 30.0;
53 minCentrality[3] = 30.0; maxCentrality[3] = 40.0;
54 minCentrality[4] = 60.0; maxCentrality[4] = 70.0;
59 cout << "-F- AddTaskDptDptCorrelations() system:" << system << ". centralityMethod:" << centralityMethod << " Option NOT AVAILABLE. ABORT."
69 double dedxMax = 20000.0;
70 int requestedCharge1 = 1; //default
71 int requestedCharge2 = -1; //default
74 // Get the pointer to the existing analysis manager via the static access method.
75 // ==============================================================================
76 AliAnalysisManager *analysisManager = AliAnalysisManager::GetAnalysisManager();
80 ::Error("AddTaskDptDptCorrelations", "No analysis manager to connect to.");
87 TString prefixName = "Corr_";
88 TString pileupRejecSuffix = "_PileupRejec";
89 TString pairRejecSuffix = "_PairRejec";
90 TString calibSuffix = "_calib";
91 TString singlesOnlySuffix = "_SO";
94 TString inputPath = ".";
95 TString outputPath = ".";
99 //TString inputHistogramFileName;
100 TString outputHistogramFileName;
102 // Create the task and add subtask.
103 // ===========================================================================
104 int iTask = 0; // task counter
105 AliAnalysisDataContainer *taskInputContainer;
106 AliAnalysisDataContainer *taskOutputContainer;
109 for (int iCentrality=0; iCentrality < nCentrality; ++iCentrality)
113 case 0: part1Name = "P_"; part2Name = "P_"; requestedCharge1 = 1; requestedCharge2 = 1; sameFilter = 1; break;
114 case 1: part1Name = "P_"; part2Name = "M_"; requestedCharge1 = 1; requestedCharge2 = -1; sameFilter = 0; break;
115 case 2: part1Name = "M_"; part2Name = "P_"; requestedCharge1 = -1; requestedCharge2 = 1; sameFilter = 0; break;
116 case 3: part1Name = "M_"; part2Name = "M_"; requestedCharge1 = -1; requestedCharge2 = -1; sameFilter = 1; break;
118 //part1Name += int(1000*etaMin);
120 part1Name += int(1000*etaMax);
122 part1Name += int(1000*ptMin);
124 part1Name += int(1000*ptMax);
126 part1Name += int(1000*dcaZMin);
128 part1Name += int(1000*dcaZMax);
132 //part2Name += int(1000*etaMin);
134 part2Name += int(1000*etaMax);
136 part2Name += int(1000*ptMin);
138 part2Name += int(1000*ptMax);
140 part2Name += int(1000*dcaZMin);
142 part2Name += int(1000*dcaZMax);
146 eventName += int(10.*minCentrality[iCentrality] );
148 eventName += int(10.*maxCentrality[iCentrality] );
151 baseName = prefixName;
152 baseName += part1Name;
153 baseName += part2Name;
154 baseName += eventName;
159 outputHistogramFileName = baseName;
160 if (singlesOnly) outputHistogramFileName += singlesOnlySuffix;
161 outputHistogramFileName += ".root";
164 TFile * inputFile = 0;
165 TList * histoList = 0;
170 TGrid::Connect("alien:");
171 inputFile = TFile::Open(inputHistogramFileName,"OLD");
174 cout << "Requested file:" << inputHistogramFileName << " was not opened. ABORT." << endl;
177 TString nameHistoBase = "correction_";
179 nameHistoBase += eventName;
180 if (requestedCharge1 == 1)
182 nameHisto = nameHistoBase + "_p";
183 cout << "Input Histogram named: " << nameHisto << endl;
184 weight_1 = (TH3F *) inputFile->Get(nameHisto);
188 nameHisto = nameHistoBase + "_m";
189 cout << "Input Histogram named: " << nameHisto << endl;
190 weight_1 = (TH3F *) inputFile->Get(nameHisto);
194 cout << "Requested histogram 'correction_p/m' was not found. ABORT." << endl;
201 if (requestedCharge2 == 1)
203 nameHisto = nameHistoBase + "_p";
204 cout << "Input Histogram named: " << nameHisto << endl;
205 weight_2 = (TH3F *) inputFile->Get(nameHisto);
209 nameHisto = nameHistoBase + "_m";
210 cout << "Input Histogram named: " << nameHisto << endl;
211 weight_2 = (TH3F *) inputFile->Get(nameHisto);
215 cout << "Requested histogram 'correction_p/m' was not found. ABORT." << endl;
221 task = new AliDptDptInMC(taskName);
223 task->SetDebugLevel( debugLevel );
224 task->SetSameFilter( sameFilter );
225 task->SetSinglesOnly( singlesOnly );
226 task->SetUseWeights( useWeights );
227 task->SetRejectPileup( rejectPileup );
228 task->SetRejectPairConversion(rejectPairConversion);
229 task->SetVertexZMin( zMin );
230 task->SetVertexZMax( zMax );
231 task->SetVertexXYMin( -1. );
232 task->SetVertexXYMax( 1. );
233 task->SetCentralityMethod( centralityMethod);
234 task->SetCentrality( minCentrality[iCentrality], maxCentrality[iCentrality]);
235 task->SetPtMin1( ptMin );
236 task->SetPtMax1( ptMax );
237 task->SetEtaMin1( etaMin );
238 task->SetEtaMax1( etaMax );
239 task->SetPtMin2( ptMin );
240 task->SetPtMax2( ptMax );
241 task->SetEtaMin2( etaMin );
242 task->SetEtaMax2( etaMax );
243 task->SetDcaZMin( dcaZMin );
244 task->SetDcaZMax( dcaZMax );
245 task->SetDcaXYMin( dcaXYMin );
246 task->SetDcaXYMax( dcaXYMax );
247 task->SetDedxMin( dedxMin );
248 task->SetDedxMax( dedxMax );
249 //task->SetNClusterMin( nClusterMin );
250 task->SetTrackFilterBit( trackFilterBit );
251 task->SetRequestedCharge_1( requestedCharge1);
252 task->SetRequestedCharge_2( requestedCharge2);
253 task->SetWeigth_1( weight_1 );
254 task->SetWeigth_2( weight_2 );
255 task->SetAnalysisType(anadata);
257 task->SelectCollisionCandidates(AliVEvent::kMB);
259 task->SetResonancesCut(NoResonances);
260 task->SetElectronCut(NoElectron);
262 cout << "Creating task output container" << endl;
264 taskOutputContainer = analysisManager->CreateContainer(listName,
266 AliAnalysisManager::kOutputContainer,
267 Form("%s:%s", AliAnalysisManager::GetCommonFileName(),taskname));
268 cout << "Add task to analysis manager and connect it to input and output containers" << endl;
271 analysisManager->AddTask(task);
272 analysisManager->ConnectInput( task, 0, analysisManager->GetCommonInputContainer());
273 analysisManager->ConnectOutput(task, 0, taskOutputContainer );
274 cout << "Task added ...." << endl;