PWG4/macros/AddTaskESDFilterPWG4Train.C

   1
   2
   3 AliAnalysisTaskESDfilter *AddTaskESDFilter(Bool_t useKineFilter=kTRUE,
   4                                            Bool_t writeMuonAOD=kFALSE,
   5                                            Bool_t writeDimuonAOD=kFALSE,
   6                                            Bool_t usePhysicsSelection=kFALSE)
   7 {
   8 // Creates a filter task and adds it to the analysis manager.
   9
  10    // Get the pointer to the existing analysis manager via the static access method.
  11    //==============================================================================
  12    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  13    if (!mgr) {
  14       ::Error("AddTaskESDFilter", "No analysis manager to connect to.");
  15       return NULL;
  16    }
  17
  18    // This task requires an ESD input handler and an AOD output handler.
  19    // Check this using the analysis manager.
  20    //===============================================================================
  21    TString type = mgr->GetInputEventHandler()->GetDataType();
  22    if (!type.Contains("ESD")) {
  23       ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an ESD input handler.");
  24       return NULL;
  25    }
  26    // Check if AOD output handler exist.
  27    AliAODHandler *aod_h = (AliAODHandler*)mgr->GetOutputEventHandler();
  28    if (!aod_h) {
  29       ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an AOD output handler.");
  30       return NULL;
  31    }
  32    // Check if MC handler is connected in case kine filter requested
  33    AliMCEventHandler *mcH = (AliMCEventHandler*)mgr->GetMCtruthEventHandler();
  34    if (!mcH && useKineFilter) {
  35       ::Error("AddTaskESDFilter", "No MC handler connected while kine filtering requested");
  36       return NULL;
  37    }
  38
  39    // Create the task, add it to the manager and configure it.
  40    //===========================================================================
  41    // Barrel tracks filter
  42    AliAnalysisTaskESDfilter *esdfilter = new AliAnalysisTaskESDfilter("ESD Filter");
  43
  44    // Make the AOD a little bit lighter and filtering faster
  45
  46    esdfilter->DisableCascades();
  47    //   esdfilter->DisableV0s();
  48    esdfilter->DisableKinks();
  49    //   esdfilter->DisableTracks();
  50    esdfilter->DisablePmdClusters();
  51    //   esdfilter->DisableCaloClusters();
  52    //   esdfilter->DisableCells();
  53    esdfilter->DisableTracklets();
  54
  55
  56    mgr->AddTask(esdfilter);
  57    // Muons
  58    //   AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter");
  59    //   mgr->AddTask(esdmuonfilter);
  60    if(usePhysicsSelection){
  61      esdfilter->SelectCollisionCandidates(AliVEvent::kAny);
  62      //     esdmuonfilter->SelectCollisionCandidates(AliVEvent::kAny);
  63    }
  64
  65    // Filtering of MC particles (decays conversions etc)
  66    // this task has to go AFTER all other filter tasks
  67    // since it fills the AODMC array with all
  68    // selected MC Particles, only this way we have the
  69    // AODMCparticle information available for following tasks
  70    AliAnalysisTaskMCParticleFilter *kinefilter = 0;
  71    if (useKineFilter) {
  72       kinefilter = new AliAnalysisTaskMCParticleFilter("Particle Kine Filter");
  73       if(usePhysicsSelection)kinefilter->SelectCollisionCandidates(AliVEvent::kAny);
  74       mgr->AddTask(kinefilter);
  75    }
  76
  77    // Cuts on primary tracks
  78    AliESDtrackCuts* esdTrackCutsL = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
  79
  80    // ITS stand-alone tracks
  81    AliESDtrackCuts* esdTrackCutsITSsa = new AliESDtrackCuts("ITS stand-alone Track Cuts", "ESD Track Cuts");
  82    esdTrackCutsITSsa->SetRequireITSStandAlone(kTRUE);
  83
  84    // Pixel OR necessary for the electrons
  85    AliESDtrackCuts *itsStrong = new AliESDtrackCuts("ITSorSPD", "pixel requirement for ITS");
  86    itsStrong->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny);
  87
  88
  89    // PID for the electrons
  90    AliESDpidCuts *electronID = new AliESDpidCuts("Electrons", "Electron PID cuts");
  91    electronID->SetTPCnSigmaCut(AliPID::kElectron, 3.);
  92
  93    // tighter cuts on primary particles for high pT tracks
  94    // take the standard cuts, which include already
  95    // ITSrefit and use only primaries...
  96
  97    // ITS cuts for new jet analysis
  98    AliESDtrackCuts* esdTrackCutsH0 = new AliESDtrackCuts("ITS+TPC cuts tight");
  99
 100    // ITS
 101    esdTrackCutsH0->SetRequireITSRefit(kTRUE);
 102
 103    // TPC
 104    esdTrackCutsH0->SetMinNClustersTPC(80);
 105    esdTrackCutsH0->SetMaxChi2PerClusterTPC(4);
 106    esdTrackCutsH0->SetAcceptKinkDaughters(kFALSE);
 107    esdTrackCutsH0->SetRequireTPCRefit(kTRUE);
 108    esdTrackCutsH0->SetMaxFractionSharedTPCClusters(0.4);
 109
 110    //
 111    esdTrackCutsH0->SetMaxDCAToVertexXY(2.4);
 112    esdTrackCutsH0->SetMaxDCAToVertexZ(3.2);
 113    esdTrackCutsH0->SetDCAToVertex2D(kTRUE);
 114    esdTrackCutsH0->SetMaxChi2PerClusterITS(36);
 115    esdTrackCutsH0->SetPtRange(0.15,1E10);
 116
 117
 118    // throw out tracks with too low number of clusters in
 119    // the first pass (be consistent with TPC only tracks)
 120    // N.B. the number off crossed rows still acts on the tracks after
 121    // all iterations if we require tpc standalone, number of clusters
 122    // and chi2 TPC cuts act on track after the first iteration
 123    //   esdTrackCutsH0->SetRequireTPCStandAlone(kTRUE);
 124    //   esdTrackCutsH0->SetMinNClustersTPC(80); // <--- first pass
 125
 126
 127    //
 128    AliESDtrackCuts* esdTrackCutsH1 = new AliESDtrackCuts(*esdTrackCutsH0);
 129    esdTrackCutsH1->SetName("loose ITS fake cuts");
 130    esdTrackCutsH1->SetMaxChi2PerClusterITS(1E10);
 131
 132
 133    // standard cuts also used in R_AA analysis
 134    gROOT->LoadMacro("$ALICE_ROOT/PWG0/dNdPt/macros/CreatedNdPtTrackCuts.C");
 135    AliESDtrackCuts* esdTrackCutsH2 = CreatedNdPtTrackCuts(200);
 136    esdTrackCutsH2->SetMaxChi2PerClusterITS(36.);
 137    esdTrackCutsH2->SetPtRange(0.15,1E10);
 138
 139    // TPC only tracks
 140    AliESDtrackCuts* esdTrackCutsTPCOnly = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
 141    esdTrackCutsTPCOnly->SetPtRange(0.15,1E10);
 142    esdTrackCutsTPCOnly->SetMaxFractionSharedTPCClusters(0.4);
 143    // throw out tracks with too low number of clusters in
 144    // the first pass
 145    // N.B. the number off crossed rows still acts on the tracks after
 146    // all iterations if we require tpc standalone, number of clusters
 147    // and chi2 TPC cuts act on track after the first iteration
 148    esdTrackCutsTPCOnly->SetRequireTPCStandAlone(kTRUE);
 149    esdTrackCutsTPCOnly->SetMinNClustersTPC(80); // <--- first pass
 150
 151    // Compose the filter
 152    AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter");
 153    // 1, 1<<0
 154    trackFilter->AddCuts(esdTrackCutsL);
 155    // 2 1<<1
 156    trackFilter->AddCuts(esdTrackCutsITSsa);
 157    // 4 1<<2
 158    trackFilter->AddCuts(itsStrong);
 159    itsStrong->SetFilterMask(1);        // AND with Standard track cuts
 160    // 8 1<<3
 161    trackFilter->AddCuts(electronID);
 162    electronID->SetFilterMask(4);       // AND with Pixel Cuts
 163    // 16 1<<4
 164    trackFilter->AddCuts(esdTrackCutsH0);
 165    // 32 1<<5
 166    trackFilter->AddCuts(esdTrackCutsH1);
 167    // 64 1<<6
 168    trackFilter->AddCuts(esdTrackCutsH2);
 169    // 128 1<<7
 170    trackFilter->AddCuts(esdTrackCutsH0); // add once more for tpc only tracks
 171    // 256 1<<8
 172    trackFilter->AddCuts(esdTrackCutsTPCOnly);
 173    // 512 1<<9
 174    trackFilter->AddCuts(esdTrackCutsH1); // add once more for tpc only tracks
 175
 176    esdfilter->SetTPCOnlyFilterMask((1<<7)|(1<<8)); // these tracks are written out as TPC only
 177    esdfilter->SetHybridFilterMaskITSTPC((1<<4)|(1<<5)); // these global tracks will be marked has hybrid
 178    esdfilter->SetHybridFilterMasksTPC(1<<7,1<<9);   // arg0, these tpc tracks will be marked as TPC  hybrid, in addtion to thos that fail arg1
 179
 180
 181    // Filter with cuts on V0s
 182    AliESDv0Cuts*   esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts");
 183    esdV0Cuts->SetMinRadius(0.2);
 184    esdV0Cuts->SetMaxRadius(200);
 185    esdV0Cuts->SetMinDcaPosToVertex(0.05);
 186    esdV0Cuts->SetMinDcaNegToVertex(0.05);
 187    esdV0Cuts->SetMaxDcaV0Daughters(1.5);
 188    esdV0Cuts->SetMinCosinePointingAngle(0.99);
 189    AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter");
 190    v0Filter->AddCuts(esdV0Cuts);
 191
 192    esdfilter->SetTrackFilter(trackFilter);
 193    esdfilter->SetV0Filter(v0Filter);
 194
 195    // Enable writing of Muon AODs
 196    //   esdmuonfilter->SetWriteMuonAOD(writeMuonAOD);
 197
 198    // Enable writing of Dimuon AODs
 199    //   esdmuonfilter->SetWriteDimuonAOD(writeDimuonAOD);
 200
 201    // Create ONLY the output containers for the data produced by the task.
 202    // Get and connect other common input/output containers via the manager as below
 203    //==============================================================================
 204
 205    mgr->ConnectInput  (esdfilter,  0, mgr->GetCommonInputContainer());
 206    mgr->ConnectOutput (esdfilter,  0, mgr->GetCommonOutputContainer());
 207
 208
 209    //   mgr->ConnectInput  (esdmuonfilter, 0, mgr->GetCommonInputContainer());
 210    if (useKineFilter) {
 211       mgr->ConnectInput  (kinefilter,  0, mgr->GetCommonInputContainer());
 212       mgr->ConnectOutput (kinefilter,  0, mgr->GetCommonOutputContainer());
 213       AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(),
 214                                                                    AliAnalysisManager::kOutputContainer,"pyxsec_hists.root");
 215       mgr->ConnectOutput (kinefilter,  1,coutputEx);
 216    }
 217    return esdfilter;
 218  }
 219