ANALYSIS/macros/AddTaskESDFilter.C

   1 AliAnalysisTaskESDfilter *AddTaskESDFilter(Bool_t useKineFilter=kTRUE,
   2                                            Bool_t writeMuonAOD=kFALSE,
   3                                            Bool_t writeDimuonAOD=kFALSE,
   4                                            Bool_t usePhysicsSelection=kFALSE)
   5 {
   6 // Creates a filter task and adds it to the analysis manager.
   7
   8    // Get the pointer to the existing analysis manager via the static access method.
   9    //==============================================================================
  10    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  11    if (!mgr) {
  12       ::Error("AddTaskESDFilter", "No analysis manager to connect to.");
  13       return NULL;
  14    }
  15
  16    // This task requires an ESD input handler and an AOD output handler.
  17    // Check this using the analysis manager.
  18    //===============================================================================
  19    TString type = mgr->GetInputEventHandler()->GetDataType();
  20    if (!type.Contains("ESD")) {
  21       ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an ESD input handler.");
  22       return NULL;
  23    }
  24    // Check if AOD output handler exist.
  25    AliAODHandler *aod_h = (AliAODHandler*)mgr->GetOutputEventHandler();
  26    if (!aod_h) {
  27       ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an AOD output handler.");
  28       return NULL;
  29    }
  30    // Check if MC handler is connected in case kine filter requested
  31    AliMCEventHandler *mcH = (AliMCEventHandler*)mgr->GetMCtruthEventHandler();
  32    if (!mcH && useKineFilter) {
  33       ::Error("AddTaskESDFilter", "No MC handler connected while kine filtering requested");
  34       return NULL;
  35    }
  36
  37    // Create the task, add it to the manager and configure it.
  38    //===========================================================================
  39    // Barrel tracks filter
  40    AliAnalysisTaskESDfilter *esdfilter = new AliAnalysisTaskESDfilter("ESD Filter");
  41    mgr->AddTask(esdfilter);
  42    // Muons
  43    AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter");
  44    mgr->AddTask(esdmuonfilter);
  45    if(usePhysicsSelection){
  46      esdfilter->SelectCollisionCandidates(AliVEvent::kAny);
  47      esdmuonfilter->SelectCollisionCandidates(AliVEvent::kAny);
  48    }
  49
  50    // Filtering of MC particles (decays conversions etc)
  51    // this task has to go AFTER all other filter tasks
  52    // since it fills the AODMC array with all
  53    // selected MC Particles, only this way we have the
  54    // AODMCparticle information available for following tasks
  55    AliAnalysisTaskMCParticleFilter *kinefilter = 0;
  56    if (useKineFilter) {
  57       kinefilter = new AliAnalysisTaskMCParticleFilter("Particle Kine Filter");
  58       mgr->AddTask(kinefilter);
  59    }
  60
  61    // Cuts on primary tracks
  62    AliESDtrackCuts* esdTrackCutsL = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
  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    // standard cuts with very loose DCA
  78    AliESDtrackCuts* esdTrackCutsH = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kFALSE);
  79    esdTrackCutsH->SetMaxDCAToVertexXY(2.4);
  80    esdTrackCutsH->SetMaxDCAToVertexZ(3.2);
  81    esdTrackCutsH->SetDCAToVertex2D(kTRUE);
  82
  83    // standard cuts with tight DCA cut
  84    AliESDtrackCuts* esdTrackCutsH2 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010();
  85
  86    // standard cuts with tight DCA but with requiring the first SDD cluster instead of an SPD cluster
  87    // tracks selected by this cut are exclusive to those selected by the previous cut
  88    AliESDtrackCuts* esdTrackCutsH3 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010();
  89    esdTrackCutsH3->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone);
  90    esdTrackCutsH3->SetClusterRequirementITS(AliESDtrackCuts::kSDD, AliESDtrackCuts::kFirst);
  91
  92
  93    // Compose the filter
  94    AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter");
  95    // 1
  96    trackFilter->AddCuts(esdTrackCutsL);
  97    // 2
  98    trackFilter->AddCuts(esdTrackCutsITSsa);
  99    // 4
 100    trackFilter->AddCuts(itsStrong);
 101    itsStrong->SetFilterMask(1);        // AND with Standard track cuts
 102    // 8
 103    trackFilter->AddCuts(electronID);
 104    electronID->SetFilterMask(4);       // AND with Pixel Cuts
 105    // 16
 106    trackFilter->AddCuts(esdTrackCutsH);
 107    // 32
 108    trackFilter->AddCuts(esdTrackCutsH2);
 109    // 64
 110    trackFilter->AddCuts(esdTrackCutsH3);
 111
 112    // Filter with cuts on V0s
 113    AliESDv0Cuts*   esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts");
 114    esdV0Cuts->SetMinRadius(0.2);
 115    esdV0Cuts->SetMaxRadius(200);
 116    esdV0Cuts->SetMinDcaPosToVertex(0.05);
 117    esdV0Cuts->SetMinDcaNegToVertex(0.05);
 118    esdV0Cuts->SetMaxDcaV0Daughters(1.5);
 119    esdV0Cuts->SetMinCosinePointingAngle(0.99);
 120    AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter");
 121    v0Filter->AddCuts(esdV0Cuts);
 122
 123    esdfilter->SetTrackFilter(trackFilter);
 124    esdfilter->SetV0Filter(v0Filter);
 125
 126    // Enable writing of Muon AODs
 127    esdmuonfilter->SetWriteMuonAOD(writeMuonAOD);
 128
 129    // Enable writing of Dimuon AODs
 130    esdmuonfilter->SetWriteDimuonAOD(writeDimuonAOD);
 131
 132    // Create ONLY the output containers for the data produced by the task.
 133    // Get and connect other common input/output containers via the manager as below
 134    //==============================================================================
 135    mgr->ConnectInput  (esdfilter,  0, mgr->GetCommonInputContainer());
 136    mgr->ConnectOutput (esdfilter,  0, mgr->GetCommonOutputContainer());
 137    mgr->ConnectInput  (esdmuonfilter, 0, mgr->GetCommonInputContainer());
 138    if (useKineFilter) {
 139       mgr->ConnectInput  (kinefilter,  0, mgr->GetCommonInputContainer());
 140       mgr->ConnectOutput (kinefilter,  0, mgr->GetCommonOutputContainer());
 141       AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(),
 142                                                                    AliAnalysisManager::kOutputContainer,"pyxsec_hists.root");
 143       mgr->ConnectOutput (kinefilter,  1,coutputEx);
 144    }
 145    return esdfilter;
 146  }
 147