Small fix:

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

diff --git a/ANALYSIS/macros/AddTaskPhysicsSelection.C b/ANALYSIS/macros/AddTaskPhysicsSelection.C
index 5be0b3ae6af0d519a6e74f61a9afdcd726675aaf..57e6c90bde1bda6da078c897bd03eb02ab737986 100644 (file)
--- a/ANALYSIS/macros/AddTaskPhysicsSelection.C
+++ b/ANALYSIS/macros/AddTaskPhysicsSelection.C
@@ -1,43 +1,59 @@
-
-
-AliPhysicsSelectionTask* AddTaskPhysicsSelection() {
-    
-    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
-    if (!mgr) {
-       ::Error("AddTaskQAsym", "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("AddTasQAsym", "This task requires an input event handler");
-       return NULL;
+AliPhysicsSelectionTask* AddTaskPhysicsSelection(Bool_t mCAnalysisFlag = kFALSE, Bool_t deprecatedFlag = kTRUE, UInt_t computeBG = 0, Bool_t useSpecialOutput=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;
+  }
+
+  AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
+  
+  TString inputDataType = inputHandler->GetDataType(); // can be "ESD" or "AOD"
+
+  // Configure analysis
+  //===========================================================================
+  AliPhysicsSelectionTask *task = new AliPhysicsSelectionTask("");
+  task->SetUseSpecialOutput(useSpecialOutput); // RS: optionally use special output
+  // this makes physics selection to work using AliMultiInputEventHandler
+  if (inputHandler && (inputHandler->IsA() == AliMultiInputEventHandler::Class())) {
+    AliMultiInputEventHandler *multiInputHandler=(AliMultiInputEventHandler*)inputHandler;
+    AliInputEventHandler *ih = multiInputHandler->GetFirstInputEventHandler();
+    if (!ih) {
+      ::Error("AddTaskPhysicsSelection","ESD or AOD input handler is missing");
+      return NULL;
     }
-    TString inputDataType = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
-    
-    // Configure analysis
-    //===========================================================================
-    
-    
-
-  AliPhysicsSelectionTask *task= new AliPhysicsSelectionTask("");
+    ih->SetEventSelection(multiInputHandler->GetEventSelection());
+    inputDataType = ih->GetDataType(); // can be "ESD" or "AOD"
+  }
+  
   mgr->AddTask(task);
+  
+  AliPhysicsSelection* physSel = task->GetPhysicsSelection();
+  if (mCAnalysisFlag)      
+    physSel->SetAnalyzeMC();
+  if (computeBG)
+    physSel->SetComputeBG(computeBG);
 
-
+  if(!deprecatedFlag) 
+    AliFatal("The BG ID flag is deprecated. Please use the OADB to configure the cuts");
 
   AliAnalysisDataContainer *cinput0 = mgr->GetCommonInputContainer();
   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("cstatsout",
-                                                           TList::Class(),
-                                                           AliAnalysisManager::kOutputContainer,
-                                                           "EventStat.root");
-
-
-
+                TList::Class(),
+                AliAnalysisManager::kOutputContainer,
+                "EventStat_temp.root");
+  //           
+  if (useSpecialOutput) coutput1->SetSpecialOutput(); // RS: optionally use special output
+  //
   mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer());
   mgr->ConnectOutput(task,1,coutput1);
 
   return task;
-  
 }   
-//_____________________________________________________________________________