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
  43    // Make the AOD a little bit lighter and filtering faster
  44
  45    esdfilter->DisableCascades();
  46    //   esdfilter->DisableV0s();
  47    esdfilter->DisableKinks();
  48    //   esdfilter->DisableTracks();
  49    esdfilter->DisablePmdClusters();
  50    //   esdfilter->DisableCaloClusters();
  51    //   esdfilter->DisableCells();
  52    esdfilter->DisableTracklets();
  53
  54
  55    mgr->AddTask(esdfilter);
  56    // Muons
  57    //   AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter");
  58    //   mgr->AddTask(esdmuonfilter);
  59    if(usePhysicsSelection){
  60      esdfilter->SelectCollisionCandidates(AliVEvent::kAny);
  61      //     esdmuonfilter->SelectCollisionCandidates(AliVEvent::kAny);
  62    }
  63
  64    // Filtering of MC particles (decays conversions etc)
  65    // this task has to go AFTER all other filter tasks
  66    // since it fills the AODMC array with all
  67    // selected MC Particles, only this way we have the
  68    // AODMCparticle information available for following tasks
  69    AliAnalysisTaskMCParticleFilter *kinefilter = 0;
  70    if (useKineFilter) {
  71       kinefilter = new AliAnalysisTaskMCParticleFilter("Particle Kine Filter");
  72       if(usePhysicsSelection)kinefilter->SelectCollisionCandidates(AliVEvent::kAny);
  73       mgr->AddTask(kinefilter);
  74    }
  75
  76    // Cuts on primary tracks
  77    AliESDtrackCuts* esdTrackCutsL = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
  78
  79    // ITS stand-alone tracks
  80    AliESDtrackCuts* esdTrackCutsITSsa = new AliESDtrackCuts("ITS stand-alone Track Cuts", "ESD Track Cuts");
  81    esdTrackCutsITSsa->SetRequireITSStandAlone(kTRUE);
  82
  83    // Pixel OR necessary for the electrons
  84    AliESDtrackCuts *itsStrong = new AliESDtrackCuts("ITSorSPD", "pixel requirement for ITS");
  85    itsStrong->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny);
  86
  87
  88    // PID for the electrons
  89    AliESDpidCuts *electronID = new AliESDpidCuts("Electrons", "Electron PID cuts");
  90    electronID->SetTPCnSigmaCut(AliPID::kElectron, 3.);
  91
  92    // tighter cuts on primary particles for high pT tracks
  93    // take the standard cuts, which include already
  94    // ITSrefit and use only primaries...
  95
  96    // loose DCA cuts
  97    AliESDtrackCuts* esdTrackCutsH = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kFALSE);
  98    esdTrackCutsH->SetMaxDCAToVertexXY(2.4);
  99    esdTrackCutsH->SetMaxDCAToVertexZ(3.2);
 100    esdTrackCutsH->SetDCAToVertex2D(kTRUE);
 101    esdTrackCutsH->SetDCAToVertex2D(kTRUE);
 102    esdTrackCutsH->SetPtRange(0.15,1E10);
 103
 104    // standard cuts with tight DCA cut
 105    AliESDtrackCuts* esdTrackCutsH2 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010();
 106    esdTrackCutsH2->SetPtRange(0.15,1E10);
 107
 108    // standard cuts with tight DCA but with requiring the first SDD cluster instead of an SPD cluster
 109    // tracks selected by this cut are exclusive to those selected by the previous cut
 110    AliESDtrackCuts* esdTrackCutsH3 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010();
 111    esdTrackCutsH3->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone);
 112    esdTrackCutsH3->SetClusterRequirementITS(AliESDtrackCuts::kSDD, AliESDtrackCuts::kFirst);
 113
 114
 115    AliESDtrackCuts* esdTrackCutsH4 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kFALSE);
 116    esdTrackCutsH4->SetMaxDCAToVertexXY(2.4);
 117    esdTrackCutsH4->SetMaxDCAToVertexZ(3.2);
 118    esdTrackCutsH4->SetDCAToVertex2D(kTRUE);
 119    esdTrackCutsH4->SetDCAToVertex2D(kTRUE);
 120    esdTrackCutsH4->SetPtRange(0.15,1E10);
 121    esdTrackCutsH4->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone);
 122    esdTrackCutsH4->SetClusterRequirementITS(AliESDtrackCuts::kSDD, AliESDtrackCuts::kFirst);
 123
 124    // TPC only tracks
 125    AliESDtrackCuts* esdTrackCutsTPCOnly = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
 126    esdTrackCutsTPCOnly->SetMinNClustersTPC(70);
 127
 128    // Compose the filter
 129    AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter");
 130    // 1, 1<<0
 131    trackFilter->AddCuts(esdTrackCutsL);
 132    // 2 1<<1
 133    trackFilter->AddCuts(esdTrackCutsITSsa);
 134    // 4 1<<2
 135    trackFilter->AddCuts(itsStrong);
 136    itsStrong->SetFilterMask(1);        // AND with Standard track cuts
 137    // 8 1<<3
 138    trackFilter->AddCuts(electronID);
 139    electronID->SetFilterMask(4);       // AND with Pixel Cuts
 140    // 16 1<<4
 141    trackFilter->AddCuts(esdTrackCutsH);
 142    // 32 1<<5
 143    trackFilter->AddCuts(esdTrackCutsH2);
 144    // 64 1<<6
 145    trackFilter->AddCuts(esdTrackCutsH3);
 146    // 128 1<<7
 147    trackFilter->AddCuts(esdTrackCutsH4);
 148    // 256 1<<8
 149    trackFilter->AddCuts(esdTrackCutsTPCOnly);
 150    esdfilter->SetTPCOnlyFilterMask(1<<8);
 151    // Filter with cuts on V0s
 152    AliESDv0Cuts*   esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts");
 153    esdV0Cuts->SetMinRadius(0.2);
 154    esdV0Cuts->SetMaxRadius(200);
 155    esdV0Cuts->SetMinDcaPosToVertex(0.05);
 156    esdV0Cuts->SetMinDcaNegToVertex(0.05);
 157    esdV0Cuts->SetMaxDcaV0Daughters(1.5);
 158    esdV0Cuts->SetMinCosinePointingAngle(0.99);
 159    AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter");
 160    v0Filter->AddCuts(esdV0Cuts);
 161
 162    esdfilter->SetTrackFilter(trackFilter);
 163    esdfilter->SetV0Filter(v0Filter);
 164
 165    // Enable writing of Muon AODs
 166    //   esdmuonfilter->SetWriteMuonAOD(writeMuonAOD);
 167
 168    // Enable writing of Dimuon AODs
 169    //   esdmuonfilter->SetWriteDimuonAOD(writeDimuonAOD);
 170
 171    // Create ONLY the output containers for the data produced by the task.
 172    // Get and connect other common input/output containers via the manager as below
 173    //==============================================================================
 174
 175    mgr->ConnectInput  (esdfilter,  0, mgr->GetCommonInputContainer());
 176    mgr->ConnectOutput (esdfilter,  0, mgr->GetCommonOutputContainer());
 177
 178
 179    //   mgr->ConnectInput  (esdmuonfilter, 0, mgr->GetCommonInputContainer());
 180    if (useKineFilter) {
 181       mgr->ConnectInput  (kinefilter,  0, mgr->GetCommonInputContainer());
 182       mgr->ConnectOutput (kinefilter,  0, mgr->GetCommonOutputContainer());
 183       AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(),
 184                                                                    AliAnalysisManager::kOutputContainer,"pyxsec_hists.root");
 185       mgr->ConnectOutput (kinefilter,  1,coutputEx);
 186    }
 187    return esdfilter;
 188  }
 189