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                                            Bool_t useCentralityTask=kFALSE,
   6                                            Int_t tofTimeZeroType=AliESDpid::kTOF_T0,
   7                                            Bool_t enableTPCOnlyAODTracks=kFALSE)
   8 {
   9 // Creates a filter task and adds it to the analysis manager.
  10
  11    // Get the pointer to the existing analysis manager via the static access method.
  12    //==============================================================================
  13    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  14    if (!mgr) {
  15       ::Error("AddTaskESDFilter", "No analysis manager to connect to.");
  16       return NULL;
  17    }
  18
  19    // This task requires an ESD input handler and an AOD output handler.
  20    // Check this using the analysis manager.
  21    //===============================================================================
  22    TString type = mgr->GetInputEventHandler()->GetDataType();
  23    if (!type.Contains("ESD")) {
  24       ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an ESD input handler.");
  25       return NULL;
  26    }
  27    // Check if AOD output handler exist.
  28    AliAODHandler *aod_h = (AliAODHandler*)mgr->GetOutputEventHandler();
  29    if (!aod_h) {
  30       ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an AOD output handler.");
  31       return NULL;
  32    }
  33    // Check if MC handler is connected in case kine filter requested
  34    AliMCEventHandler *mcH = (AliMCEventHandler*)mgr->GetMCtruthEventHandler();
  35    if (!mcH && useKineFilter) {
  36       ::Error("AddTaskESDFilter", "No MC handler connected while kine filtering requested");
  37       return NULL;
  38    }
  39
  40    // Create the task, add it to the manager and configure it.
  41    //===========================================================================
  42    // Barrel tracks filter
  43    AliAnalysisTaskESDfilter *esdfilter = new AliAnalysisTaskESDfilter("ESD Filter");
  44    esdfilter->SetTimeZeroType(tofTimeZeroType);
  45    esdfilter->DisableCascades();
  46    esdfilter->DisableKinks();
  47
  48    mgr->AddTask(esdfilter);
  49
  50    // Muons
  51    Bool_t onlyMuon=kTRUE;
  52    Bool_t keepAllEvents=kTRUE;
  53    Int_t mcMode=(useKineFilter ? 2 : 0); // use 1 instead of 2 to get all MC information instead of just ancestors of mu tracks
  54    AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter",onlyMuon,keepAllEvents,mcMode);
  55    mgr->AddTask(esdmuonfilter);
  56    if(usePhysicsSelection){
  57      esdfilter->SelectCollisionCandidates(AliVEvent::kAny);
  58      esdmuonfilter->SelectCollisionCandidates(AliVEvent::kAny);
  59    }
  60
  61    // Filtering of MC particles (decays conversions etc)
  62    // this task has to go AFTER all other filter tasks
  63    // since it fills the AODMC array with all
  64    // selected MC Particles, only this way we have the
  65    // AODMCparticle information available for following tasks
  66    AliAnalysisTaskMCParticleFilter *kinefilter = 0;
  67    if (useKineFilter) {
  68       kinefilter = new AliAnalysisTaskMCParticleFilter("Particle Kine Filter");
  69       mgr->AddTask(kinefilter);
  70    }
  71
  72    AliAnalysisTaskAODCentralityMaker* ctask = 0;
  73
  74    if (useCentralityTask) {
  75        ctask = new AliAnalysisTaskAODCentralityMaker("AODCentralityMaker");
  76        ctask->SetDeltaAODFileName("AliAODCentrality.root");
  77        mgr->AddTask(ctask);
  78    }
  79
  80    // Cuts on primary tracks
  81    AliESDtrackCuts* esdTrackCutsL = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
  82
  83    // ITS stand-alone tracks
  84    AliESDtrackCuts* esdTrackCutsITSsa = new AliESDtrackCuts("ITS stand-alone Track Cuts", "ESD Track Cuts");
  85    esdTrackCutsITSsa->SetRequireITSStandAlone(kTRUE);
  86
  87    // Pixel OR necessary for the electrons
  88    AliESDtrackCuts *itsStrong = new AliESDtrackCuts("ITSorSPD", "pixel requirement for ITS");
  89    itsStrong->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny);
  90
  91
  92    // PID for the electrons
  93    AliESDpidCuts *electronID = new AliESDpidCuts("Electrons", "Electron PID cuts");
  94    electronID->SetTPCnSigmaCut(AliPID::kElectron, 3.);
  95
  96    // standard cuts with very loose DCA
  97    AliESDtrackCuts* esdTrackCutsH = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kFALSE);
  98    esdTrackCutsH->SetMaxDCAToVertexXY(2.4);
  99    esdTrackCutsH->SetMaxDCAToVertexZ(3.2);
 100    esdTrackCutsH->SetDCAToVertex2D(kTRUE);
 101
 102    // standard cuts with tight DCA cut
 103    AliESDtrackCuts* esdTrackCutsH2 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010();
 104
 105    // standard cuts with tight DCA but with requiring the first SDD cluster instead of an SPD cluster
 106    // tracks selected by this cut are exclusive to those selected by the previous cut
 107    AliESDtrackCuts* esdTrackCutsH3 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010();
 108    esdTrackCutsH3->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone);
 109    esdTrackCutsH3->SetClusterRequirementITS(AliESDtrackCuts::kSDD, AliESDtrackCuts::kFirst);
 110
 111    // TPC only tracks: Optionally enable the writing of TPConly information
 112    // constrained to SPD vertex in the filter below
 113    AliESDtrackCuts* esdTrackCutsTPCOnly = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
 114    esdTrackCutsTPCOnly->SetMinNClustersTPC(70);
 115
 116    // Compose the filter
 117    AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter");
 118    // 1
 119    trackFilter->AddCuts(esdTrackCutsL);
 120    // 2
 121    trackFilter->AddCuts(esdTrackCutsITSsa);
 122    // 4
 123    trackFilter->AddCuts(itsStrong);
 124    itsStrong->SetFilterMask(1);        // AND with Standard track cuts
 125    // 8
 126    trackFilter->AddCuts(electronID);
 127    electronID->SetFilterMask(4);       // AND with Pixel Cuts
 128    // 16
 129    trackFilter->AddCuts(esdTrackCutsH);
 130    // 32
 131    trackFilter->AddCuts(esdTrackCutsH2);
 132    // 64
 133    trackFilter->AddCuts(esdTrackCutsH3);
 134    // 128 , 1 << 7
 135    trackFilter->AddCuts(esdTrackCutsTPCOnly);
 136    if(enableTPCOnlyAODTracks)esdfilter->SetTPCOnlyFilterMask(128);
 137
 138    // Filter with cuts on V0s
 139    AliESDv0Cuts*   esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts");
 140    esdV0Cuts->SetMinRadius(0.2);
 141    esdV0Cuts->SetMaxRadius(200);
 142    esdV0Cuts->SetMinDcaPosToVertex(0.05);
 143    esdV0Cuts->SetMinDcaNegToVertex(0.05);
 144    esdV0Cuts->SetMaxDcaV0Daughters(1.5);
 145    esdV0Cuts->SetMinCosinePointingAngle(0.99);
 146    AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter");
 147    v0Filter->AddCuts(esdV0Cuts);
 148
 149    esdfilter->SetTrackFilter(trackFilter);
 150    esdfilter->SetV0Filter(v0Filter);
 151
 152    // Enable writing of Muon AODs
 153    esdmuonfilter->SetWriteMuonAOD(writeMuonAOD);
 154
 155    // Enable writing of Dimuon AODs
 156    esdmuonfilter->SetWriteDimuonAOD(writeDimuonAOD);
 157
 158    // Create ONLY the output containers for the data produced by the task.
 159    // Get and connect other common input/output containers via the manager as below
 160    //==============================================================================
 161    mgr->ConnectInput  (esdfilter,  0, mgr->GetCommonInputContainer());
 162    mgr->ConnectOutput (esdfilter,  0, mgr->GetCommonOutputContainer());
 163    mgr->ConnectInput  (esdmuonfilter, 0, mgr->GetCommonInputContainer());
 164    if (useKineFilter) {
 165       mgr->ConnectInput  (kinefilter,  0, mgr->GetCommonInputContainer());
 166       mgr->ConnectOutput (kinefilter,  0, mgr->GetCommonOutputContainer());
 167       AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(),
 168                                                                    AliAnalysisManager::kOutputContainer,"pyxsec_hists.root");
 169       mgr->ConnectOutput (kinefilter,  1,coutputEx);
 170    }
 171    if (useCentralityTask) {
 172        mgr->ConnectInput (ctask, 0, mgr->GetCommonInputContainer());
 173        mgr->ConnectOutput(ctask, 0, mgr->GetCommonOutputContainer());
 174    }
 175    return esdfilter;
 176  }
 177