cover case for AOD analysis

[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskJetSpectrum2.C

diff --git a/PWG4/macros/AddTaskJetSpectrum2.C b/PWG4/macros/AddTaskJetSpectrum2.C
index 49478237b5af0099bb5124731292313f2dd8b17a..6ba46d5ebdf457680baf92d2a812c03b213f0b02 100644 (file)
--- a/PWG4/macros/AddTaskJetSpectrum2.C
+++ b/PWG4/macros/AddTaskJetSpectrum2.C
@@ -1,4 +1,4 @@
-AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2()\r
+AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(char* bRec = "jets",char* bGen = "jetsAODMC_UA104",UInt_t filterMask = 16)\r
 {\r
 // Creates a jet fider task, configures it and adds it to the analysis manager.\r
 \r
@@ -13,30 +13,40 @@ AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2()
    // Check the analysis type using the event handlers connected to the analysis manager.\r
    //==============================================================================\r
    if (!mgr->GetInputEventHandler()) {\r
-      ::Error("AddTaskJetSpectrum2", "This task requires an input event handler");\r
+     ::Error("AddTaskJetSpectrum2", "This task requires an input event handler");\r
       return NULL;\r
    }\r
 \r
+   TString type = mgr->GetInputEventHandler()->GetDataType();\r
+   TString typeMC(bGen);\r
    // Create the task and configure it.\r
    //===========================================================================\r
    \r
-   AliAnalysisTaskJetSpectrum2* pwg4spec = new  AliAnalysisTaskJetSpectrum2("Jet Spectrum");\r
+   AliAnalysisTaskJetSpectrum2* pwg4spec = new  AliAnalysisTaskJetSpectrum2(Form("Jet Spectrum %s %s",bRec,bGen));\r
       \r
    // or a config file\r
    pwg4spec->SetAnalysisType(AliAnalysisTaskJetSpectrum2::kAnaMC);\r
-   //      if(iAODanalysis)pwg4spec->SetAODInput(kTRUE);\r
+   // if(iAODanalysis)pwg4spec->SetAODInput(kTRUE);\r
    // pwg4spec->SetDebugLevel(11); \r
-   pwg4spec->SetBranchGen("jetsMC"); \r
-   pwg4spec->SetBranchRec("jetsAOD"); \r
-   mgr->AddTask(pwg4spec);\r
-\r
+   pwg4spec->SetBranchRec(bRec); \r
+   pwg4spec->SetBranchGen(bGen); \r
+   pwg4spec->SetFilterMask(filterMask); \r
 \r
+   if(type == "AOD"){\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODIn);\r
+     pwg4spec->SetAODInput(kTRUE);\r
+   }\r
+   else pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODOut);\r
 \r
-      \r
+   if(typeMC.Contains("AODMC2"))pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCCharged);\r
+   else pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCAll);\r
+   \r
+   mgr->AddTask(pwg4spec);\r
+     \r
    // Create ONLY the output containers for the data produced by the task.\r
    // Get and connect other common input/output containers via the manager as below\r
    //==============================================================================\r
-   AliAnalysisDataContainer *coutput1_Spec = mgr->CreateContainer("pwg4spec2", TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_spec2",AliAnalysisManager::GetCommonFileName()));\r
+   AliAnalysisDataContainer *coutput1_Spec = mgr->CreateContainer("pwg4spec2", TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_spec2_%s_%s",AliAnalysisManager::GetCommonFileName(),bRec,bGen));\r
 \r
    mgr->ConnectInput  (pwg4spec, 0, mgr->GetCommonInputContainer());\r
    mgr->ConnectOutput (pwg4spec, 0, mgr->GetCommonOutputContainer());\r