Merge branch 'feature-movesplit'

[u/mrichter/AliRoot.git] / PWG / EMCAL / macros / AddTaskEmcalEsdTrackFilter.C

diff --git a/PWG/EMCAL/macros/AddTaskEmcalEsdTrackFilter.C b/PWG/EMCAL/macros/AddTaskEmcalEsdTrackFilter.C
index 2f25017836fefaeb2b5a70712747485190561c88..6eff15abc02fef4e2b474e4c824df1bfa3c06c69 100644 (file)
--- a/PWG/EMCAL/macros/AddTaskEmcalEsdTrackFilter.C
+++ b/PWG/EMCAL/macros/AddTaskEmcalEsdTrackFilter.C
@@ -63,6 +63,9 @@ AliEmcalEsdTrackFilterTask* AddTaskEmcalEsdTrackFilter(
   } else if (strTrackCuts.Contains("lhc12g")) {
     dataSet = kLHC11h;
     dataSetLabel = "LHC12g";
+  } else if (strTrackCuts.Contains("lhc12")) {
+    dataSet = kLHC11h;
+    dataSetLabel = "LHC12";
   } else if (strTrackCuts.Contains("lhc13b")) {
     dataSet = kLHC11h;
     dataSetLabel = "LHC13b";
@@ -93,6 +96,12 @@ AliEmcalEsdTrackFilterTask* AddTaskEmcalEsdTrackFilter(
   } else if (strTrackCuts.Contains("lhc12f2a")) {
     dataSet = kLHC11d;
     dataSetLabel = "LHC12f2a";
+  } else if (strTrackCuts.Contains("lhc12a17")) {
+    dataSet = kLHC11h;
+    dataSetLabel = "LHC12a17";
+  } else if (strTrackCuts.Contains("lhc14a1")) {
+    dataSet = kLHC11h;
+    dataSetLabel = "LHC14a1";
   } else {
     ::Warning("AddTaskEmcalEsdTpcTrack", "Dataset not recognized, will assume LHC11h");
   }
@@ -124,7 +133,6 @@ AliEmcalEsdTrackFilterTask* AddTaskEmcalEsdTrackFilter(
 
   gROOT->LoadMacro("$ALICE_ROOT/PWGJE/macros/CreateTrackCutsPWGJE.C");
   AliEmcalEsdTrackFilterTask *eTask = new AliEmcalEsdTrackFilterTask(taskName); // default is no cut
-  Bool_t includeNoITS = kFALSE;
   if ((dataSet == kLHC11c && cutsType == kHybrid) ||
       (dataSet == kLHC11d && cutsType == kHybrid) ||
       (dataSet == kLHC11h && cutsType == kHybrid)) {
@@ -134,6 +142,7 @@ AliEmcalEsdTrackFilterTask* AddTaskEmcalEsdTrackFilter(
     AliESDtrackCuts *hybsp = CreateTrackCutsPWGJE(10041008);       //1004 removes ITSrefit requirement from standard set   
     hybsp->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kOff);
     eTask->SetHybridTrackCuts(hybsp);
+    eTask->SetIncludeNoITS(kFALSE);
   } else if ((dataSet == kLHC10h && cutsType == kHybrid) ||
             (dataSet == kLHC11a && cutsType == kHybrid)) {
     /* hybrid track cuts*/
@@ -142,10 +151,18 @@ AliEmcalEsdTrackFilterTask* AddTaskEmcalEsdTrackFilter(
     AliESDtrackCuts *hybsp = CreateTrackCutsPWGJE(10041006);       //1004 removes ITSrefit requirement from standard set    
     hybsp->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kOff);
     eTask->SetHybridTrackCuts(hybsp);
-    includeNoITS = kTRUE;
+    eTask->SetIncludeNoITS(kTRUE);
+  }
+  else if (dataSet == kLHC11h && cutsType == kTpcOnly) {
+    /* TPC-only constrained track cuts*/
+    AliESDtrackCuts *cutsp = CreateTrackCutsPWGJE(2001);       //TPC-only loose track cuts
+    eTask->SetTrackCuts(cutsp);
+    eTask->SetHybridTrackCuts(0);
+  }
+  else {
+    ::Error("AddTaskEmcalEsdTrackFilter","Track cuts type / period not recognized! Undefined beahviour will follow!");
   }
   eTask->SetTracksName(name);
-  eTask->SetIncludeNoITS(includeNoITS);
 
   cout << " *** Track selector task configured to select " << cutsLabel  << " in dataset "<< dataSetLabel << " *** " << endl;