Bastian Bathen:
[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskFragmentationFunction.C
index 6b9a5d7fad1c097565d868fc4c1a0ab94dcea4da..be833604fdc43d89cd082b85d5f15fb7eb40bc3b 100644 (file)
@@ -30,12 +30,14 @@ AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(UInt_t iFlag=
 
         // only reconstructed (default)
        if(iFlag&(1<<0)) ff = AddTaskFragmentationFunction("jets", "", "", "", filterMask);
-        // MC tracks in acceptance, MC jets in acceptance
-       if(iFlag&(1<<1)) ff = AddTaskFragmentationFunction("jets", "AODMC2b", "AODMCb", "AODMCb", filterMask);
+        // charged MC tracks and jets
+       if(iFlag&(1<<1)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA04", "AODMC", "AODMC2", filterMask);
+        // charged MC tracks and jets with acceptance cuts
+       if(iFlag&(1<<2)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA04", "AODMCb", "AODMC2b", filterMask);
         // kine tracks in acceptance, pythia jets in acceptance
-       if(iFlag&(1<<2)) ff = AddTaskFragmentationFunction("jets", "", "KINEb", "KINEb", filterMask);
+       if(iFlag&(1<<3)) ff = AddTaskFragmentationFunction("jets", "", "KINEb", "KINEb", filterMask);
         // reconstructed charged tracks after cuts, MC jets in acceptance 
-       if(iFlag&(1<<3)) ff = AddTaskFragmentationFunction("jets", "AODMC2b", "AODMCb", "AOD2b", filterMask);
+       if(iFlag&(1<<4)) ff = AddTaskFragmentationFunction("jets", "jetsMC2b", "AODMCb", "AOD2b", filterMask);
        
        return ff;
 }
@@ -130,6 +132,10 @@ AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(
    else if(typeJets.Contains("jetTypeUndef")) task->SetJetTypeGen(0); // undefined
    else Printf("jetType %s not found", typeJets.Data());
    
+   if(typeJets.Contains("AODMCb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance); // kJetsRecAcceptance
+   else if(typeJets.Contains("AODb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance); 
+   else task->SetJetTypeRecEff(0);
+
    task->SetFilterMask(filterMask);
   
    // Set default parameters