new class for omega measurement via 3 pion combinationomega3pi/

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

AliJetReader *CreateJetReader(Char_t *jr); // Common config\r
AliJetFinder *CreateJetFinder(Char_t *jf);\r
\r
AliAnalysisTaskJets *AddTaskJets(Char_t *jr, Char_t *jf); // for the new AF\r
AliAnalysisTaskJets *AddTaskJets(Char_t *jr, Char_t *jf,AliAnalysisManager* mgr,AliAnalysisDataContainer *cinput); // for the old AF\r
\r
AliAnalysisTaskJets *AddTaskJets(Char_t *jr, Char_t *jf)\r
{\r
  // Creates a jet finder 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
   if (!mgr) {\r
      ::Error("AddTaskJets", "No analysis manager to connect to.");\r
      return NULL;\r
   }  \r
   \r
   // Check the analysis type using the event handlers connected to the analysis manager.\r
   //==============================================================================\r
   if (!mgr->GetInputEventHandler()) {\r
      ::Error("AddTaskJets", "This task requires an input event handler");\r
      return NULL;\r
   }\r
\r
   // Create the task and configure it.\r
   //===========================================================================\r
   AliAnalysisTaskJets *jetana;\r
   AliJetReader *er = CreateJetReader(jr);\r
    // Define jet header and jet finder\r
   AliJetFinder *jetFinder = CreateJetFinder(jf);\r
\r
   if (jetFinder){\r
       if (er) jetFinder->SetJetReader(er);\r
   }\r
\r
   jetana = new AliAnalysisTaskJets(Form("JetAnalysis%s%s",jr,jf));\r
   AliAnalysisDataContainer *cout_jet = mgr->CreateContainer(Form("jethist%s%s",jr,jf), TList::Class(),\r
							      AliAnalysisManager::kOutputContainer, Form("jethist%s_%s.root",jr,jf));\r
   // Connect jet finder to task.\r
   jetana->SetJetFinder(jetFinder);\r
   jetana->SetConfigFile("");\r
   jetana->SetDebugLevel(10);\r
   mgr->AddTask(jetana);\r
\r
   // Create ONLY the output containers for the data produced by the task.\r
   // Get and connect other common input/output containers via the manager as below\r
   //==============================================================================\r
   mgr->ConnectInput  (jetana, 0, mgr->GetCommonInputContainer());\r
// AOD output slot will be used in a different way in future\r
   mgr->ConnectOutput (jetana, 0, mgr->GetCommonOutputContainer());\r
   mgr->ConnectOutput (jetana, 1, cout_jet);\r
   \r
   return jetana;\r
}\r
\r
AliAnalysisTaskJets *AddTaskJets(Char_t *jr, Char_t *jf,AliAnalysisManager* mgr,AliAnalysisDataContainer *cinput)\r
{\r
  // This is only for running on PROOF with the old root version 5-22-00 \r
  // and the older version of the AF\r
\r
  // Creates a jet fider task, configures it and adds it to the analysis manager.\r
  // Get the pointer to the existing analysis manager via the static access method.\r
  //==============================================================================\r
\r
  if (!mgr) {\r
    ::Error("AddTaskJets", "No analysis manager to connect to.");\r
    return NULL;\r
  }  \r
   \r
   // Check the analysis type using the event handlers connected to the analysis manager.\r
   //==============================================================================\r
   if (!mgr->GetInputEventHandler()) {\r
      ::Error("AddTaskJets", "This task requires an input event handler");\r
      return NULL;\r
   }\r
\r
   // Create the task and configure it.\r
   //===========================================================================\r
   AliAnalysisTaskJets *jetana;\r
   AliJetReader *er = CreateJetReader(jr);\r
\r
    // Define jet header and jet finder\r
   AliJetFinder *jetFinder = CreateJetFinder(jf);\r
\r
   if (jetFinder){\r
       if (er) jetFinder->SetJetReader(er);\r
   }\r
\r
   jetana = new AliAnalysisTaskJets(Form("JetAnalysis%s%s",jr,jf));\r
   AliAnalysisDataContainer *cout_jet = mgr->CreateContainer(Form("jethist%s%s",jr,jf), TList::Class(),AliAnalysisManager::kOutputContainer, Form("jethist%s_%s.root",jr,jf));\r
   // Connect jet finder to task.\r
   jetana->SetJetFinder(jetFinder);\r
   jetana->SetConfigFile("");\r
   jetana->SetDebugLevel(10);\r
   mgr->AddTask(jetana);\r
\r
   // Create the output containers for the data produced by the task.\r
   //==============================================================================\r
   // AOD output slot will be used in a different way in future\r
   // we need to have different names for each usage\r
   c_aodjet = mgr->CreateContainer(Form("cAODjet%s%s",jr,jf), TTree::Class(),\r
				   AliAnalysisManager::kExchangeContainer);\r
   // Connect to data containers\r
   mgr->ConnectInput (jetana,0, cinput );\r
   mgr->ConnectOutput (jetana,0, c_aodjet );\r
   mgr->ConnectOutput (jetana, 1, cout_jet);\r
   \r
   return jetana;\r
}\r
\r
AliJetFinder *CreateJetFinder(Char_t *jf){\r
\r
  switch (jf) {\r
  case "CDF":\r
    AliCdfJetHeader *jh = new AliCdfJetHeader();\r
    jh->SetRadius(0.7);\r
    \r
    jetFinder = new AliCdfJetFinder();\r
    jetFinder->SetOutputFile("jets.root");\r
    if (jh) jetFinder->SetJetHeader(jh);\r
    break;\r
\r
  case "DA":\r
    AliDAJetHeader *jh=new AliDAJetHeader();\r
    jh->SetComment("DA jet code with default parameters");\r
    jh->SelectJets(kTRUE);\r
    jh->SetNclust(10);\r
    \r
    jetFinder = new AliDAJetFinder();\r
    if (jh) jetFinder->SetJetHeader(jh);\r
    break;\r
\r
\r
  case "Fastjet":\r
    AliFastJetHeader *jh = new AliFastJetHeader();\r
    jh->SetRparam(0.7); // setup parameters                                  \r
\r
    jetFinder = new AliFastJetFinder();\r
    jetFinder->SetOutputFile("jets.root");\r
    if (jh) jetFinder->SetJetHeader(jh);\r
    break;\r
\r
  case "UA1":\r
    AliUA1JetHeaderV1 *jh=new AliUA1JetHeaderV1();\r
    jh->SetComment("UA1 jet code with default parameters");\r
    jh->BackgMode(0);\r
    jh->SetRadius(0.4);\r
    jh->SetEtSeed(4.);\r
    jh->SetLegoNbinPhi(432);\r
    jh->SetLegoNbinEta(274);\r
    jh->SetLegoEtaMin(-2);\r
    jh->SetLegoEtaMax(+2);\r
    jh->SetMinJetEt(10.);\r
    jh->SetJetEtaMax(1.5);\r
    jh->SetJetEtaMin(-1.5);\r
\r
    jetFinder = new AliUA1JetFinderV1();\r
    if (jh) jetFinder->SetJetHeader(jh);\r
    break;\r
\r
  case "UA1MC":\r
    AliUA1JetHeaderV1 *jh=new AliUA1JetHeaderV1();\r
    jh->SetComment("UA1 jet code with default MC parameters");\r
    jh->BackgMode(0);\r
    jh->SetRadius(1.0);\r
    jh->SetEtSeed(4.);\r
    jh->SetLegoNbinPhi(432);\r
    jh->SetLegoNbinEta(274);\r
    jh->SetLegoEtaMin(-2);\r
    jh->SetLegoEtaMax(+2);\r
    jh->SetMinJetEt(10.);\r
    jh->SetJetEtaMax(1.5);\r
    jh->SetJetEtaMin(-1.5);\r
\r
    jetFinder = new AliUA1JetFinderV1();\r
    if (jh) jetFinder->SetJetHeader(jh);\r
    break;\r
  case default:\r
    ::Error("AddTaskJets", "Wrong jet finder selected\n");\r
    return 0;\r
  }\r
\r
  return jetFinder;\r
\r
}\r
\r
AliJetReader *CreateJetReader(Char_t *jr){\r
  AliJetReader *er = 0;\r
\r
  switch (jr) {\r
  case "MC":\r
    AliJetKineReaderHeader *jrh = new AliJetKineReaderHeader();\r
    jrh->SetComment("MC full Kinematics");\r
    jrh->SetFastSimTPC(kFALSE);\r
    jrh->SetFastSimEMCAL(kFALSE);\r
    jrh->SetPtCut(0.);\r
    jrh->SetFiducialEta(-2.1,2.1); // to take all MC particles default is 0 .9                                                                             \r
    // Define reader and set its header                                     \r
    er = new AliJetKineReader();\r
    er->SetReaderHeader(jrh);\r
    break;\r
  case "MC2":\r
    AliJetKineReaderHeader *jrh = new AliJetKineReaderHeader();\r
    jrh->SetComment("MC full Kinematics spearate config");\r
    jrh->SetFastSimTPC(kFALSE);\r
    jrh->SetFastSimEMCAL(kFALSE);\r
    jrh->SetPtCut(0.);\r
    jrh->SetFiducialEta(-2.1,2.1); // to take all MC particles default is 0 .9                                                                             \r
    // Define reader and set its header                                     \r
    er = new AliJetKineReader();\r
    er->SetReaderHeader(jrh);\r
    break;\r
  case "ESD":\r
    AliJetESDReaderHeader *jrh = new AliJetESDReaderHeader();\r
    jrh->SetComment("Testing");\r
    jrh->SetFirstEvent(0);\r
    jrh->SetLastEvent(1000);\r
    jrh->SetPtCut(0.);\r
    jrh->SetReadSignalOnly(kFALSE);\r
    // Define reader and set its header                                     \r
    er = new AliJetESDReader();\r
    er->SetReaderHeader(jrh);\r
    break;\r
\r
  case "AOD":\r
    AliJetAODReaderHeader *jrh = new AliJetAODReaderHeader();\r
    jrh->SetComment("AOD Reader");\r
    jrh->SetPtCut(0.);\r
    jrh->SetTestFilterMask(1<<0);\r
    // Define reader and set its header\r
    er = new AliJetAODReader();\r
    er->SetReaderHeader(jrh);\r
    break;\r
\r
  default:\r
    ::Error("AddTaskJets", "Wrong jet reader selected\n");\r
    return 0;\r
  }\r
\r
  return er;\r
\r
}\r