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::GetStandardITSTPCTrackCuts2011(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::GetStandardITSTPCTrackCuts2011();
  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::GetStandardITSTPCTrackCuts2011();
 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    // Extra cuts for hybrids
 111    // first the global tracks we want to take
 112    AliESDtrackCuts* esdTrackCutsHTG = AliESDtrackCuts::GetStandardITSTPCTrackCuts2011(kFALSE);
 113    esdTrackCutsHTG->SetName("Global Hybrid tracks, loose DCA");
 114    esdTrackCutsHTG->SetMaxDCAToVertexXY(2.4);
 115    esdTrackCutsHTG->SetMaxDCAToVertexZ(3.2);
 116    esdTrackCutsHTG->SetDCAToVertex2D(kTRUE);
 117    esdTrackCutsHTG->SetMaxChi2TPCConstrainedGlobal(36);
 118
 119    // Than the complementary tracks which will be stored as global
 120    // constraint, complement is done in the ESDFilter task
 121    AliESDtrackCuts* esdTrackCutsHTGC = new AliESDtrackCuts(*esdTrackCutsHTG);
 122    esdTrackCutsHTGC->SetName("Global Constraint Hybrid tracks, loose DCA no it requirement");
 123    esdTrackCutsHTGC->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kOff);
 124    //   esdTrackCutsHTGC->SetRequireITSRefit(kFALSE)
 125
 126    // Compose the filter
 127    AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter");
 128    // 1, 1<<0
 129    trackFilter->AddCuts(esdTrackCutsL);
 130    // 2, 1<<1
 131    trackFilter->AddCuts(esdTrackCutsITSsa);
 132    // 4, 1<<2
 133    trackFilter->AddCuts(itsStrong);
 134    itsStrong->SetFilterMask(1);        // AND with Standard track cuts
 135    // 8, 1<<3
 136    trackFilter->AddCuts(electronID);
 137    electronID->SetFilterMask(4);       // AND with Pixel Cuts
 138    // 16, 1<<4
 139    trackFilter->AddCuts(esdTrackCutsH);
 140    // 32, 1<<5
 141    trackFilter->AddCuts(esdTrackCutsH2);
 142    // 64, 1<<6
 143    trackFilter->AddCuts(esdTrackCutsH3);
 144    // 128 , 1 << 7
 145    trackFilter->AddCuts(esdTrackCutsTPCOnly);
 146    if(enableTPCOnlyAODTracks)esdfilter->SetTPCOnlyFilterMask(128);
 147    // 256, 1 << 8 Global Hybrids
 148    trackFilter->AddCuts(esdTrackCutsHTG);
 149    esdfilter->SetHybridFilterMaskGlobalConstrainedGlobal((1<<8)); // these normal global tracks will be marked as hybrid
 150    // 512, 1<< 9 GlobalConstraint Hybrids
 151    trackFilter->AddCuts(esdTrackCutsHTGC);
 152    esdfilter->SetGlobalConstrainedFilterMask(1<<9); // these tracks are written out as global constrained tracks
 153    esdfilter->SetWriteHybridGlobalConstrainedOnly(kTRUE); // write only the complement
 154
 155    // Filter with cuts on V0s
 156    AliESDv0Cuts*   esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts");
 157    esdV0Cuts->SetMinRadius(0.2);
 158    esdV0Cuts->SetMaxRadius(200);
 159    esdV0Cuts->SetMinDcaPosToVertex(0.05);
 160    esdV0Cuts->SetMinDcaNegToVertex(0.05);
 161    esdV0Cuts->SetMaxDcaV0Daughters(1.5);
 162    esdV0Cuts->SetMinCosinePointingAngle(0.99);
 163    AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter");
 164    v0Filter->AddCuts(esdV0Cuts);
 165
 166    esdfilter->SetTrackFilter(trackFilter);
 167    esdfilter->SetV0Filter(v0Filter);
 168
 169    // Enable writing of Muon AODs
 170    esdmuonfilter->SetWriteMuonAOD(writeMuonAOD);
 171
 172    // Enable writing of Dimuon AODs
 173    esdmuonfilter->SetWriteDimuonAOD(writeDimuonAOD);
 174
 175    // Create ONLY the output containers for the data produced by the task.
 176    // Get and connect other common input/output containers via the manager as below
 177    //==============================================================================
 178    mgr->ConnectInput  (esdfilter,  0, mgr->GetCommonInputContainer());
 179    mgr->ConnectOutput (esdfilter,  0, mgr->GetCommonOutputContainer());
 180    mgr->ConnectInput  (esdmuonfilter, 0, mgr->GetCommonInputContainer());
 181    if (useKineFilter) {
 182       mgr->ConnectInput  (kinefilter,  0, mgr->GetCommonInputContainer());
 183       mgr->ConnectOutput (kinefilter,  0, mgr->GetCommonOutputContainer());
 184       AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(),
 185                                                                    AliAnalysisManager::kOutputContainer,"pyxsec_hists.root");
 186       mgr->ConnectOutput (kinefilter,  1,coutputEx);
 187    }
 188    return esdfilter;
 189  }
 190