PWG2/RESONANCES/macros/test/runRsnAnalysisSE.C

   1 void runRsnAnalysisSE(AliLog::EType_t type=AliLog::kInfo,Bool_t useKine = kTRUE)
   2 {
   3
   4   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   5   AliAnalysisDataContainer *cin;
   6   AliAnalysisDataContainer *cout;
   7
   8   cin = SetCorrectHandlersAndReturnInput(mgr,useKine);
   9
  10   AliRsnAnalysisSE *task = new AliRsnAnalysisSE("AliRsnAnalysisSE");
  11   task->SetLogType(type,"AliRsnAnalysisManager:AliRsnPairManager:AliRsnPairManager:AliRsnPair");
  12   task->SetPrintInfoNumber(10);
  13
  14   task->SetPriorProbability(AliRsnDaughter::kElectron, 0.02);
  15   task->SetPriorProbability(AliRsnDaughter::kMuon,     0.02);
  16   task->SetPriorProbability(AliRsnDaughter::kPion,     0.83);
  17   task->SetPriorProbability(AliRsnDaughter::kKaon,     0.07);
  18   task->SetPriorProbability(AliRsnDaughter::kProton,   0.06);
  19   task->DumpPriors();
  20
  21   AliRsnAnalysisManager *analMgr = task->GetAnalysisManager("MyAnalysisSE");
  22   Int_t i=0;
  23   Int_t analNum=1;
  24   for (i=0;i<analNum;i++) {
  25 //     analMgr->AddConfig("RsnConfigTest.C",Form("PHI%d",i),);
  26     analMgr->AddConfig("RsnConfig.C",Form("PHI%d",i),"RsnConfig_PHI");
  27     analMgr->AddConfig("RsnConfig.C",Form("KSTAR%d",i),"RsnConfig_KSTAR");
  28   }
  29
  30   AliAnalysisDataContainer *output = mgr->CreateContainer("RSNSE", TList::Class(), AliAnalysisManager::kOutputContainer, "RSNAnalysis.root");
  31
  32   mgr->AddTask(task);
  33
  34   mgr->ConnectInput(task, 0, cin);
  35   mgr->ConnectOutput(task, 1, output);
  36
  37 }
  38
  39 AliAnalysisDataContainer *SetCorrectHandlersAndReturnInput(AliAnalysisManager *mgr,Bool_t useKine = kTRUE)
  40 {
  41
  42   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  43
  44   AliAnalysisDataContainer *cin;
  45
  46   if (useKine) {
  47     AliMCEventHandler* mcInputHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
  48     if (!mcInputHandler) {
  49       Info("","Creating mcInputHandler ...");
  50       AliMCEventHandler* mcInputHandler = new AliMCEventHandler();
  51       mgr->SetMCtruthEventHandler(mcInputHandler);
  52     }
  53   }
  54
  55   AliESDInputHandler *esdInputHandler = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
  56   if (!esdInputHandler) {
  57     Info("","Creating esdInputHandler ...");
  58     esdInputHandler = new AliESDInputHandler();
  59     mgr->SetInputEventHandler  (esdInputHandler);
  60     cin = mgr->GetCommonInputContainer();
  61   } else {
  62     cin = mgr->GetCommonOutputContainer();
  63   }
  64
  65   return cin;
  66 }