PWG4/macros/AddTaskESDFilterPWG4Train.C

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