PWG/EMCAL/macros/AddTaskEmcalEsdTrackFilter.C

   1 // $Id$
   2
   3 AliEmcalEsdTrackFilterTask* AddTaskEmcalEsdTrackFilter(
   4   const char *name              = "TrackFilter",
   5   const char *trackCuts         = "Hybrid_LHC11h",
   6   const char *taskName          = "AliEmcalEsdTrackFilterTask"
   7 )
   8 {
   9   enum CutsType {
  10     kHybrid  = 0,
  11     kTpcOnly = 1
  12   };
  13
  14   enum DataSet {
  15     kLHC10h  = 0,
  16     kLHC11a  = 1,
  17     kLHC11c  = 3,
  18     kLHC11d  = 3,
  19     kLHC11h  = 3
  20   };
  21
  22   CutsType cutsType = kHybrid;
  23   DataSet  dataSet  = kLHC11h;
  24
  25   TString cutsLabel("hybrid tracks");
  26   TString dataSetLabel("LHC11h");
  27
  28   TString strTrackCuts(trackCuts);
  29   strTrackCuts.ToLower();
  30
  31   if (strTrackCuts.Contains("hybrid")) {
  32     cutsType = kHybrid;
  33   } else if (strTrackCuts.Contains("tpconly")) {
  34     cutsType = kTpcOnly;
  35     cutsLabel = "TPC only constrained tracks";
  36   } else {
  37     ::Warning("AddTaskEmcalEsdTpcTrack", "Cuts type not recognized, will assume Hybrid");
  38   }
  39
  40   if (strTrackCuts.Contains("lhc10h")) {
  41     dataSet = kLHC10h;
  42   } else if (strTrackCuts.Contains("lhc11a") || strTrackCuts.Contains("lhc12a15a")) {
  43     dataSet = kLHC11a;
  44     dataSetLabel = "LHC11a";
  45   } else if (strTrackCuts.Contains("lhc10e")   ||  strTrackCuts.Contains("lhc10d")   ||
  46              strTrackCuts.Contains("lhc10e20") ||  strTrackCuts.Contains("lhc10f6a") ||
  47              strTrackCuts.Contains("lhc11a1a") ||  strTrackCuts.Contains("lhc11a1b") ||
  48              strTrackCuts.Contains("lhc11a1c") ||  strTrackCuts.Contains("lhc11a1d") ||
  49              strTrackCuts.Contains("lhc11a1e") ||  strTrackCuts.Contains("lhc11a1f") ||
  50              strTrackCuts.Contains("lhc11a1g") ||  strTrackCuts.Contains("lhc11a1h") ||
  51              strTrackCuts.Contains("lhc11a1i") ||  strTrackCuts.Contains("lhc11a1j")) {
  52     dataSet = kLHC11a;
  53     dataSetLabel = "LHC10e";
  54   } else if (strTrackCuts.Contains("lhc11c")) {
  55     dataSet = kLHC11c;
  56     dataSetLabel = "LHC11c";
  57   } else if (strTrackCuts.Contains("lhc11d")) {
  58     dataSet = kLHC11d;
  59     dataSetLabel = "LHC11d";
  60   } else if (strTrackCuts.Contains("lhc11h") || strTrackCuts.Contains("lhc12a15e")) {
  61     dataSet = kLHC11h;
  62     dataSetLabel = "LHC11h";
  63   } else if (strTrackCuts.Contains("lhc12g")) {
  64     dataSet = kLHC11h;
  65     dataSetLabel = "LHC12g";
  66   } else if (strTrackCuts.Contains("lhc13b")) {
  67     dataSet = kLHC11h;
  68     dataSetLabel = "LHC13b";
  69   } else if (strTrackCuts.Contains("lhc13c")) {
  70     dataSet = kLHC11h;
  71     dataSetLabel = "LHC13c";
  72   } else if (strTrackCuts.Contains("lhc13d")) {
  73     dataSet = kLHC11h;
  74     dataSetLabel = "LHC13d";
  75   } else if (strTrackCuts.Contains("lhc13e")) {
  76     dataSet = kLHC11h;
  77     dataSetLabel = "LHC13e";
  78   } else if (strTrackCuts.Contains("lhc13f")) {
  79     dataSet = kLHC11h;
  80     dataSetLabel = "LHC13f";
  81   } else if (strTrackCuts.Contains("lhc13g")) {
  82     dataSet = kLHC11h;
  83     dataSetLabel = "LHC13g";
  84   } else if (strTrackCuts.Contains("lhc12a15f")) {
  85     dataSet = kLHC11h;
  86     dataSetLabel = "LHC12a15f";
  87   } else if (strTrackCuts.Contains("lhc13b4")) {
  88     dataSet = kLHC11h;
  89     dataSetLabel = "LHC13b4";
  90   } else if (strTrackCuts.Contains("lhc12a15g")) {
  91     dataSet = kLHC11d;
  92     dataSetLabel = "LHC12a15g";
  93   } else if (strTrackCuts.Contains("lhc12f2a")) {
  94     dataSet = kLHC11d;
  95     dataSetLabel = "LHC12f2a";
  96   } else {
  97     ::Warning("AddTaskEmcalEsdTpcTrack", "Dataset not recognized, will assume LHC11h");
  98   }
  99
 100   // Get the pointer to the existing analysis manager via the static access method.
 101   //==============================================================================
 102   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
 103   if (!mgr) {
 104     ::Error("AddTaskEmcalEsdTpcTrack", "No analysis manager to connect to.");
 105     return NULL;
 106   }
 107
 108   // Check the analysis type using the event handlers connected to the analysis manager.
 109   //==============================================================================
 110   AliVEventHandler *evhand = mgr->GetInputEventHandler();
 111   if (!evhand) {
 112     ::Error("AddTaskEmcalEsdTpcTrack", "This task requires an input event handler");
 113     return NULL;
 114   }
 115
 116   if (!evhand->InheritsFrom("AliESDInputHandler")) {
 117     ::Info("AddTaskEmcalEsdTpcTrack", "This task is only needed for ESD analysis. No task added.");
 118     return NULL;
 119   }
 120
 121   //-------------------------------------------------------
 122   // Init the task and do settings
 123   //-------------------------------------------------------
 124
 125   gROOT->LoadMacro("$ALICE_ROOT/PWGJE/macros/CreateTrackCutsPWGJE.C");
 126   AliEmcalEsdTrackFilterTask *eTask = new AliEmcalEsdTrackFilterTask(taskName); // default is no cut
 127   Bool_t includeNoITS = kFALSE;
 128   if ((dataSet == kLHC11c && cutsType == kHybrid) ||
 129       (dataSet == kLHC11d && cutsType == kHybrid) ||
 130       (dataSet == kLHC11h && cutsType == kHybrid)) {
 131     /* hybrid track cuts*/
 132     AliESDtrackCuts *cutsp = CreateTrackCutsPWGJE(10001008);       //1000 adds SPD any requirement
 133     eTask->SetTrackCuts(cutsp);
 134     AliESDtrackCuts *hybsp = CreateTrackCutsPWGJE(10041008);       //1004 removes ITSrefit requirement from standard set
 135     hybsp->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kOff);
 136     eTask->SetHybridTrackCuts(hybsp);
 137   } else if ((dataSet == kLHC10h && cutsType == kHybrid) ||
 138              (dataSet == kLHC11a && cutsType == kHybrid)) {
 139     /* hybrid track cuts*/
 140     AliESDtrackCuts *cutsp = CreateTrackCutsPWGJE(10001006);       //1000 adds SPD any requirement
 141     eTask->SetTrackCuts(cutsp);
 142     AliESDtrackCuts *hybsp = CreateTrackCutsPWGJE(10041006);       //1004 removes ITSrefit requirement from standard set
 143     hybsp->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kOff);
 144     eTask->SetHybridTrackCuts(hybsp);
 145     includeNoITS = kTRUE;
 146   }
 147   eTask->SetTracksName(name);
 148   eTask->SetIncludeNoITS(includeNoITS);
 149
 150   cout << " *** Track selector task configured to select " << cutsLabel  << " in dataset "<< dataSetLabel << " *** " << endl;
 151
 152   //-------------------------------------------------------
 153   // Final settings, pass to manager and set the containers
 154   //-------------------------------------------------------
 155   mgr->AddTask(eTask);
 156
 157   // Create containers for input/output
 158   AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();
 159   mgr->ConnectInput(eTask, 0, cinput1);
 160
 161   return eTask;
 162 }