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