Add the connected pad info to the simulation (Raphaelle)

[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskJetSpectrum2.C

diff --git a/PWG4/macros/AddTaskJetSpectrum2.C b/PWG4/macros/AddTaskJetSpectrum2.C
index 438a7f8d7c6a9c3a4862a3f9fb715a033fa1ddf1..af9e36490831c8e947fc3d0fd91f10b04e1992d5 100644 (file)
--- a/PWG4/macros/AddTaskJetSpectrum2.C
+++ b/PWG4/macros/AddTaskJetSpectrum2.C
@@ -1,34 +1,39 @@
-AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(char* bRec = "jets",char* bGen = "jetsAODMC_UA104",UInt_t filterMask = 16);\r
+AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(const char* bRec = "jets",const char* bGen = "jetsAODMC_UA104",UInt_t filterMask = 16, Int_t iPhysicsSelection = 1);\r
 \r
 \r
-AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2Delta(UInt_t filterMask = 16,Bool_t kUseAODMC = kFALSE){\r
+AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2Delta(UInt_t filterMask = 16,Bool_t kUseAODMC = kFALSE,Int_t iPhysicsSelection = 1,UInt_t iFlag){\r
   AliAnalysisTaskJetSpectrum2 *js = 0;\r
   if(kUseAODMC){\r
-  js = AddTaskJetSpectrum2("jets","jetsAODMC_UA104",filterMask);\r
-    js = AddTaskJetSpectrum2("jets","jetsAODMC2_UA104",filterMask);\r
+    if(iFlag&(1<<0))js = AddTaskJetSpectrum2("jets","jetsAODMC_UA104",filterMask,iPhysicsSelection);\r
+    if(iFlag&(1<<1))js = AddTaskJetSpectrum2("jets","jetsAODMC2_UA104",filterMask,iPhysicsSelection);\r
 \r
-    js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC_FASTJET04",filterMask);\r
-    js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC2_FASTJET04",filterMask);\r
+    if(iFlag&(1<<2))js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC_FASTJET04",filterMask,iPhysicsSelection);\r
+    if(iFlag&(1<<3))js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","jetsAODMC2_FASTJET04",filterMask,iPhysicsSelection);\r
 \r
-    js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC_FASTKT04",filterMask);\r
-    js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC2_FASTKT04",filterMask);\r
-\r
-    js = AddTaskJetSpectrum2("jetsAOD_UA107","jetsAODMC_UA107",filterMask);\r
+    if(iFlag&(1<<4)){\r
+      js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC_FASTKT04",filterMask,iPhysicsSelection);\r
+    }\r
+    if(iFlag&(1<<5))js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","jetsAODMC2_FASTKT04",filterMask,iPhysicsSelection);\r
+    if(iFlag&(1<<6))js = AddTaskJetSpectrum2("jetsAOD_UA107","jetsAODMC_UA107",filterMask,iPhysicsSelection);\r
   }\r
-  js = AddTaskJetSpectrum2("jets","jetsAOD_FASTJET04",filterMask);\r
-  js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","",filterMask);\r
-  js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","",filterMask);\r
-  js = AddTaskJetSpectrum2("jetsAOD_UA107","",filterMask);\r
-  js->SetRecEtaWindow(0.2);\r
 \r
+  if(iFlag&(1<<7))js = AddTaskJetSpectrum2("jets","jetsAOD_FASTJET04",filterMask,iPhysicsSelection);\r
+\r
+  if(iFlag&(1<<8))js = AddTaskJetSpectrum2("jetsAOD_FASTJET04","",filterMask,iPhysicsSelection);\r
+  if(iFlag&(1<<9))js = AddTaskJetSpectrum2("jetsAOD_FASTKT04","",filterMask,iPhysicsSelection);\r
+  if(iFlag&(1<<10))js = AddTaskJetSpectrum2("jetsAOD_SISCONE04","",filterMask,iPhysicsSelection);\r
+  \r
+  if(iFlag&(1<<11)){\r
+    js = AddTaskJetSpectrum2("jetsAOD_UA107","",filterMask,iPhysicsSelection);\r
+    js->SetRecEtaWindow(0.2);\r
+  }\r
   return js;\r
 }\r
 \r
 \r
-AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(char* bRec,char* bGen ,UInt_t filterMask)\r
+AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(const char* bRec,const char* bGen ,UInt_t filterMask,Int_t iPhysicsSelection)\r
 {\r
 // Creates a jet fider task, configures it and adds it to the analysis manager.\r
-\r
    // Get the pointer to the existing analysis manager via the static access method.\r
    //==============================================================================\r
    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
@@ -45,43 +50,63 @@ AliAnalysisTaskJetSpectrum2 *AddTaskJetSpectrum2(char* bRec,char* bGen ,UInt_t f
    }\r
 \r
    TString type = mgr->GetInputEventHandler()->GetDataType();\r
-   TString typeMC(bGen);\r
-   typeMC.ToUpper();\r
+   TString typeRec(bRec);\r
+   TString typeGen(bGen);\r
+   typeGen.ToUpper();\r
+   typeRec.ToUpper();\r
    // Create the task and configure it.\r
    //===========================================================================\r
    \r
-   AliAnalysisTaskJetSpectrum2* pwg4spec = new  AliAnalysisTaskJetSpectrum2(Form("Jet Spectrum %s %s",bRec,bGen));\r
+   AliAnalysisTaskJetSpectrum2* pwg4spec = new  AliAnalysisTaskJetSpectrum2(Form("JetSpectrum2%s%s",bRec,bGen));\r
       \r
    // or a config file\r
    // pwg4spec->SetAnalysisType(AliAnalysisTaskJetSpectrum2::kAnaMC);\r
    // if(iAODanalysis)pwg4spec->SetAODInput(kTRUE);\r
    // pwg4spec->SetDebugLevel(11); \r
-   pwg4spec->SetBranchRec(bRec); \r
+\r
    pwg4spec->SetBranchGen(bGen); \r
+   pwg4spec->SetBranchRec(bRec); \r
+\r
+\r
    pwg4spec->SetFilterMask(filterMask); \r
-   //   pwg4spec->SetUseGlobalSelection(kTRUE); \r
+   pwg4spec->SetUseGlobalSelection(kTRUE); \r
+   pwg4spec->SetMinJetPt(5.);\r
+\r
 \r
    if(type == "AOD"){\r
-     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODIn);\r
-     pwg4spec->SetAODInput(kTRUE);\r
+     // Assume all jet are produced already\r
+     pwg4spec->SetAODJetInput(kTRUE);\r
+     pwg4spec->SetAODTrackInput(kTRUE);\r
+     pwg4spec->SetAODMCInput(kTRUE);\r
+   }\r
+\r
+   if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCChargedAcceptance);\r
+   }\r
+   else if (typeRec.Contains("AODMC2")){\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCCharged);\r
+   }\r
+   else if (typeRec.Contains("AODMC")){\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODMCAll);\r
+   }\r
+   else { // catch akk use AOD\r
+     pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAOD);\r
    }\r
-   else pwg4spec->SetTrackTypeRec(AliAnalysisTaskJetSpectrum2::kTrackAODOut);\r
 \r
-   if(typeMC.Contains("AODMC2")){\r
+   if(typeGen.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD\r
+     pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCChargedAcceptance);\r
+   }\r
+   else if (typeGen.Contains("AODMC2")){\r
      pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCCharged);\r
    }\r
-   else if (typeMC.Contains("AODMC2")){\r
+   else if (typeGen.Contains("AODMC")){\r
      pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODMCAll);\r
    }\r
-   else if (typeMC.Contains("AOD")) {\r
-     if(type == "AOD"){\r
-       pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODIn);\r
-     }\r
-     else{\r
-       pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAODOut);\r
-     }\r
+   else if (typeGen.Length()>0){ // catch all use AOD\r
+     pwg4spec->SetTrackTypeGen(AliAnalysisTaskJetSpectrum2::kTrackAOD);\r
    }\r
-   pwg4spec->SelectCollisionCandidates();\r
+\r
+   if(iPhysicsSelection)pwg4spec->SelectCollisionCandidates();\r
 \r
    mgr->AddTask(pwg4spec);\r
      \r