ANALYSIS/macros/AddTaskESDFilter.C

   1 AliAnalysisTaskESDfilter *AddTaskESDFilter(Bool_t useKineFilter=kTRUE)
   2 {
   3 // Creates a filter task and adds it to the analysis manager.
   4
   5    // Get the pointer to the existing analysis manager via the static access method.
   6    //==============================================================================
   7    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   8    if (!mgr) {
   9       ::Error("AddTaskESDFilter", "No analysis manager to connect to.");
  10       return NULL;
  11    }
  12
  13    // This task requires an ESD input handler and an AOD output handler.
  14    // Check this using the analysis manager.
  15    //===============================================================================
  16    TString type = mgr->GetInputEventHandler()->GetDataType();
  17    if (!type.Contains("ESD")) {
  18       ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an ESD input handler.");
  19       return NULL;
  20    }
  21    // Check if AOD output handler exist.
  22    AliAODHandler *aod_h = (AliAODHandler*)mgr->GetOutputEventHandler();
  23    if (!aod_h) {
  24       ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an AOD output handler.");
  25       return NULL;
  26    }
  27    // Check if MC handler is connected in case kine filter requested
  28    AliMCEventHandler *mcH = (AliMCEventHandler*)mgr->GetMCtruthEventHandler();
  29    if (!mcH && useKineFilter) {
  30       ::Error("AddTaskESDFilter", "No MC handler connected while kine filtering requested");
  31       return NULL;
  32    }
  33
  34    // Filtering of MC particles (decays conversions etc)
  35    // this task is also needed to set the MCEventHandler
  36    // to the AODHandler, this will not be needed when
  37    // AODHandler goes to ANALYSISalice
  38    AliAnalysisTaskMCParticleFilter *kinefilter = 0;
  39    if (useKineFilter) {
  40       kinefilter = new AliAnalysisTaskMCParticleFilter("Particle Kine Filter");
  41       mgr->AddTask(kinefilter);
  42    }
  43
  44    // Create the task, add it to the manager and configure it.
  45    //===========================================================================
  46    // Barrel tracks filter
  47    AliAnalysisTaskESDfilter *esdfilter = new AliAnalysisTaskESDfilter("ESD Filter");
  48    mgr->AddTask(esdfilter);
  49    // Muons
  50    AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter");
  51    mgr->AddTask(esdmuonfilter);
  52
  53    // Cuts on primary tracks
  54    AliESDtrackCuts* esdTrackCutsL = new AliESDtrackCuts("Standard Track Cuts", "ESD Track Cuts");
  55    esdTrackCutsL->SetMinNClustersTPC(50);
  56    esdTrackCutsL->SetMaxChi2PerClusterTPC(3.5);
  57    esdTrackCutsL->SetRequireTPCRefit(kTRUE);
  58    esdTrackCutsL->SetMaxDCAToVertexXY(2.4);
  59    esdTrackCutsL->SetMaxDCAToVertexZ(3.2);
  60    esdTrackCutsL->SetDCAToVertex2D(kTRUE);
  61    esdTrackCutsL->SetRequireSigmaToVertex(kFALSE);
  62    esdTrackCutsL->SetAcceptKinkDaughters(kFALSE);
  63
  64    // ITS stand-alone tracks
  65    AliESDtrackCuts* esdTrackCutsITSsa = new AliESDtrackCuts("ITS stand-alone Track Cuts", "ESD Track Cuts");
  66    esdTrackCutsITSsa->SetRequireITSStandAlone(kTRUE);
  67
  68    // Pixel OR necessary for the electrons
  69    AliESDtrackCuts *itsStrong = new AliESDtrackCuts("ITSorSPD", "pixel requirement for ITS");
  70    itsStrong->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny);
  71
  72
  73    // PID for the electrons
  74    AliESDpidCuts *electronID = new AliESDpidCuts("Electrons", "Electron PID cuts");
  75    electronID->SetTPCnSigmaCut(AliPID::kElectron, 3.);
  76
  77
  78    // Compose the filter
  79    AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter");
  80    // 1
  81    trackFilter->AddCuts(esdTrackCutsL);
  82    // 2
  83    trackFilter->AddCuts(esdTrackCutsITSsa);
  84    // 4
  85    trackFilter->AddCuts(itsStrong);
  86    itsStrong->SetFilterMask(1);        // AND with Standard track cuts
  87    // 8
  88    trackFilter->AddCuts(electronID);
  89    electronID->SetFilterMask(4);       // AND with Pixel Cuts
  90
  91    // Filter with cuts on V0s
  92    AliESDv0Cuts*   esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts");
  93    esdV0Cuts->SetMinRadius(0.2);
  94    esdV0Cuts->SetMaxRadius(100);
  95    esdV0Cuts->SetMinDcaPosToVertex(0.05);
  96    esdV0Cuts->SetMinDcaNegToVertex(0.05);
  97    esdV0Cuts->SetMaxDcaV0Daughters(0.5);
  98    esdV0Cuts->SetMinCosinePointingAngle(0.99);
  99    AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter");
 100    v0Filter->AddCuts(esdV0Cuts);
 101
 102    esdfilter->SetTrackFilter(trackFilter);
 103    esdfilter->SetV0Filter(v0Filter);
 104
 105
 106    // Create ONLY the output containers for the data produced by the task.
 107    // Get and connect other common input/output containers via the manager as below
 108    //==============================================================================
 109    mgr->ConnectInput  (esdfilter,  0, mgr->GetCommonInputContainer());
 110    mgr->ConnectOutput (esdfilter,  0, mgr->GetCommonOutputContainer());
 111    mgr->ConnectInput  (esdmuonfilter, 0, mgr->GetCommonInputContainer());
 112    if (useKineFilter) {
 113       mgr->ConnectInput  (kinefilter,  0, mgr->GetCommonInputContainer());
 114       mgr->ConnectOutput (kinefilter,  0, mgr->GetCommonOutputContainer());
 115       AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(),
 116                                                                    AliAnalysisManager::kOutputContainer,"pyxsec_hists.root");
 117       mgr->ConnectOutput (kinefilter,  1,coutputEx);
 118    }
 119    return esdfilter;
 120 }