Adding new task macros for Kink resonance analysis (by P.Ganoti, pganoti@phys.uoa.gr)

[u/mrichter/AliRoot.git] / PWG2 / KINK / macros / AddTaskKinkResonanceKstarMC.C

diff --git a/PWG2/KINK/macros/AddTaskKinkResonanceKstarMC.C b/PWG2/KINK/macros/AddTaskKinkResonanceKstarMC.C
new file mode 100644 (file)
index 0000000..c3d6a4e
--- /dev/null
+++ b/PWG2/KINK/macros/AddTaskKinkResonanceKstarMC.C
@@ -0,0 +1,66 @@
+AliAnalysisTaskKinkResonance *AddTaskKinkResonanceKstarMC(Short_t lCollidingSystems=0 /*0 = pp, 1 = AA*/)
+{
+// Creates, configures and attaches to the train a kink resonance task.
+// Get the pointer to the existing analysis manager via the static access method.
+//==============================================================================
+AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
+ if (!mgr) {
+ ::Error("AddTaskKinkResonanceKstarMC", "No analysis manager to connect to.");
+return NULL;
+}
+
+// Check the analysis type using the event handlers connected to the analysis manager.
+//==============================================================================
+if (!mgr->GetInputEventHandler()) {
+ ::Error("AddTaskKinkResonanceKstarMC", "This task requires an input event handler");
+ return NULL;
+}
+TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
+ if (type != "ESD") {
+ ::Error("AddTaskKinkResonanceKstarMC", "This task needs an ESD input handler");
+return NULL;
+}
+if (!mgr->GetMCtruthEventHandler()) {
+ ::Error("AddTaskKinkResonanceKstarMC", "This task needs an MC handler");
+ return NULL;
+}
+
+// Setup the analysis object
+  
+  AliResonanceKink  *kinkResonanceObjectKMC=new AliResonanceKink();
+  kinkResonanceObjectKMC->InitOutputHistograms(60, 0.6, 1.2);
+  kinkResonanceObjectKMC->SetPDGCodes(kKPlus, kPiPlus, AliResonanceKink::kKstar0); 
+  kinkResonanceObjectKMC->SetAnalysisType("MC"); // "ESD" or "MC"
+  kinkResonanceObjectKMC->SetMaxNsigmaToVertex(4.0);
+  kinkResonanceObjectKMC->SetMaxDCAxy(3.0);
+  kinkResonanceObjectKMC->SetMaxDCAzaxis(3.0);
+  kinkResonanceObjectKMC->SetPtTrackCut(0.25);
+  kinkResonanceObjectKMC->SetMinTPCclusters(50);
+  kinkResonanceObjectKMC->SetMaxChi2PerTPCcluster(3.5);
+  kinkResonanceObjectKMC->SetMaxCov0(2.0);
+  kinkResonanceObjectKMC->SetMaxCov2(2.0);
+  kinkResonanceObjectKMC->SetMaxCov5(0.5);
+  kinkResonanceObjectKMC->SetMaxCov9(0.5);
+  kinkResonanceObjectKMC->SetMaxCov14(2.0);
+
+// Create and configure the task
+AliAnalysisTaskKinkResonance *taskresonanceKstarMC = new AliAnalysisTaskKinkResonance("TaskResKstarMCKinkPID");
+taskresonanceKstarMC->SetAnalysisKinkObject(kinkResonanceObjectKMC);
+mgr->AddTask(taskresonanceKstarMC);
+
+// Create ONLY the output containers for the data produced by the task.
+// Get and connect other common input/output containers via the manager as below
+//==============================================================================
+TString outname = "PP";
+if (lCollidingSystems) outname = "AA";
+if (mgr->GetMCtruthEventHandler()) outname += "-MC-";
+outname += "KinkResonanceList.root";
+AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("KinkResKstarMC",
+                                                          TList::Class(),
+                                                          AliAnalysisManager::kOutputContainer,
+                                                          outname );
+
+mgr->ConnectInput(taskresonanceKstarMC, 0, mgr->GetCommonInputContainer());
+mgr->ConnectOutput(taskresonanceKstarMC, 1, coutput1);
+return taskresonanceKstarMC;
+}