Extending the functionality of the physics selection. Instead of providing a yes...

[u/mrichter/AliRoot.git] / ANALYSIS / macros / AddTaskPhysicsSelection.C

diff --git a/ANALYSIS/macros/AddTaskPhysicsSelection.C b/ANALYSIS/macros/AddTaskPhysicsSelection.C
index d3f560f7962a0d77dcda01192b11fe60b263b4ae..65694ce8ff51275fcb6cd0e448daeeca77cc0bb0 100644 (file)
--- a/ANALYSIS/macros/AddTaskPhysicsSelection.C
+++ b/ANALYSIS/macros/AddTaskPhysicsSelection.C
@@ -1,43 +1,43 @@
-
-
-AliPhysicsSelectionTask* AddTaskPhysicsSelection() {
-    
-    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
-    if (!mgr) {
-       ::Error("AddTaskPhysicsSelection", "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("AddTaskPhysicsSelection", "This task requires an input event handler");
-       return NULL;
-    }
-    TString inputDataType = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
-    
-    // Configure analysis
-    //===========================================================================
+AliPhysicsSelectionTask* AddTaskPhysicsSelection(Bool_t mCAnalysisFlag = kFALSE, Bool_t withBckgndRejection = kTRUE, Bool_t computeBG = kFALSE) 
+{
+  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
+  if (!mgr) {
+    ::Error("AddTaskPhysicsSelection", "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("AddTaskPhysicsSelection", "This task requires an input event handler");
+    return NULL;
+  }
+  TString inputDataType = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
+  
+  // Configure analysis
+  //===========================================================================
     
     
 
-  AliPhysicsSelectionTask *task= new AliPhysicsSelectionTask("");
+  AliPhysicsSelectionTask *task = new AliPhysicsSelectionTask("");
   mgr->AddTask(task);
-
-
+  
+  AliPhysicsSelection* physSel = task->GetPhysicsSelection();
+  if (withBckgndRejection) 
+    physSel->AddBackgroundIdentification(new AliBackgroundSelection());
+  if (mCAnalysisFlag)      
+    physSel->SetAnalyzeMC();
+  if (computeBG)
+    physSel->SetComputeBG();
 
   AliAnalysisDataContainer *cinput0 = mgr->GetCommonInputContainer();
   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("cstatsout",
-                                                           TList::Class(),
-                                                           AliAnalysisManager::kOutputContainer,
-                                                           "EventStat_temp.root");
-
-
+                TList::Class(),
+                AliAnalysisManager::kOutputContainer,
+                "EventStat_temp.root");
 
   mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer());
   mgr->ConnectOutput(task,1,coutput1);
 
   return task;
-  
 }   
-//_____________________________________________________________________________