PWG4/macros/AddTaskESDFilterPWG4Train.C

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