Fix for CMake

[u/mrichter/AliRoot.git] / PWG0 / multiplicity / run.C

diff --git a/PWG0/multiplicity/run.C b/PWG0/multiplicity/run.C
index 1e3305d581b6b1d77f0d1f3b4f73eb116bea9d06..08b8ff08b93c14480cc716c461ccce681444ffbc 100644 (file)
--- a/PWG0/multiplicity/run.C
+++ b/PWG0/multiplicity/run.C
@@ -1,22 +1,39 @@
-void run(Char_t* data, Int_t nRuns=20, Int_t offset=0, Bool_t aDebug = kFALSE, Bool_t aProof = kFALSE, const char* option = "")
+void run(Char_t* data, Long64_t nRuns = -1, Long64_t offset = 0, Bool_t aDebug = kFALSE, Int_t aProof = 0, Bool_t mc = kTRUE, const char* option = "")
 {
+  // aProof option: 0 no proof
+  //                1 proof with chain
+  //                2 proof with dataset
+
+  if (nRuns < 0)
+    nRuns = 1234567890;
+
   if (aProof)
   {
-    TProof::Open("lxb6046");
+    TProof::Open("alicecaf");
+    //gProof->SetParallel(1);
 
     // Enable the needed package
-    gProof->UploadPackage("STEERBase");
-    gProof->EnablePackage("STEERBase");
-    gProof->UploadPackage("ESD");
-    gProof->EnablePackage("ESD");
-    gProof->UploadPackage("AOD");
-    gProof->EnablePackage("AOD");
-    gProof->UploadPackage("ANALYSIS");
-    gProof->EnablePackage("ANALYSIS");
-    gProof->UploadPackage("PWG0base");
-    gProof->EnablePackage("PWG0base");
-
-    //gProof->AddInput(new TNamed("PROOF_Packetizer", "TPacketizer"));
+    if (1)
+    {
+      gProof->UploadPackage("$ALICE_ROOT/STEERBase");
+      gProof->EnablePackage("$ALICE_ROOT/STEERBase");
+      gProof->UploadPackage("$ALICE_ROOT/ESD");
+      gProof->EnablePackage("$ALICE_ROOT/ESD");
+      gProof->UploadPackage("$ALICE_ROOT/AOD");
+      gProof->EnablePackage("$ALICE_ROOT/AOD");
+      gProof->UploadPackage("$ALICE_ROOT/ANALYSIS");
+      gProof->EnablePackage("$ALICE_ROOT/ANALYSIS");
+      gProof->UploadPackage("$ALICE_ROOT/ANALYSISalice");
+      gProof->EnablePackage("$ALICE_ROOT/ANALYSISalice");
+    }
+    else
+    {
+      gProof->UploadPackage("$ALICE_ROOT/AF-v4-16");
+      gProof->EnablePackage("$ALICE_ROOT/AF-v4-16");
+    }
+
+    gProof->UploadPackage("$ALICE_ROOT/PWG0base");
+    gProof->EnablePackage("$ALICE_ROOT/PWG0base");
   }
   else
   {
@@ -26,50 +43,88 @@ void run(Char_t* data, Int_t nRuns=20, Int_t offset=0, Bool_t aDebug = kFALSE, B
     gSystem->Load("libESD");
     gSystem->Load("libAOD");
     gSystem->Load("libANALYSIS");
+    gSystem->Load("libANALYSISalice");
     gSystem->Load("libPWG0base");
   }
 
-  // Create chain of input files
-  gROOT->LoadMacro("../CreateESDChain.C");
-  chain = CreateESDChain(data, nRuns, offset);
-
   // Create the analysis manager
-  mgr = new AliAnalysisManager("testAnalysis");
+  mgr = new AliAnalysisManager;
 
-  // selection of esd tracks
-  gROOT->ProcessLine(".L CreateCuts.C");
-  AliESDtrackCuts* esdTrackCuts = CreateTrackCuts();
-  if (!esdTrackCuts)
-  {
-    printf("ERROR: esdTrackCuts could not be created\n");
-    return;
-  }
+  AliPWG0Helper::AnalysisMode analysisMode = AliPWG0Helper::kSPD;
+  AliPWG0Helper::Trigger      trigger      = AliPWG0Helper::kMB1;
+
+  AliPWG0Helper::PrintConf(analysisMode, trigger);
 
   TString taskName("AliMultiplicityTask.cxx+");
   if (aDebug)
     taskName += "+g";
 
   // Create, add task
-  if (aProof) {
+  if (aProof > 0) {
     gProof->Load(taskName);
   } else
     gROOT->Macro(taskName);
 
-  task = new AliMultiplicityTask;
-  task->SetTrackCuts(esdTrackCuts);
+  task = new AliMultiplicityTask(option);
+
+  if (analysisMode != AliPWG0Helper::kSPD)
+  {
+    // selection of esd tracks
+    gROOT->ProcessLine(".L ../CreateStandardCuts.C");
+    AliESDtrackCuts* esdTrackCuts = CreateTrackCuts(analysisMode);
+    if (!esdTrackCuts)
+    {
+      printf("ERROR: esdTrackCuts could not be created\n");
+      return;
+    }
+
+    task->SetTrackCuts(esdTrackCuts);
+  }
+  else
+    task->SetDeltaPhiCut(0.05);
+
+  task->SetAnalysisMode(analysisMode);
+  task->SetTrigger(trigger);
+
+  if (mc)
+    task->SetReadMC();
+
+  //task->SetUseMCVertex();
+
   mgr->AddTask(task);
 
-  // Enable MC event handler
-  AliMCEventHandler* handler = new AliMCEventHandler;
-  handler->SetReadTR(kFALSE);
-  //mgr->SetMCtruthEventHandler(handler);
+  TString optionStr(option);
+  
+  if (mc) {
+    // Enable MC event handler
+    AliMCEventHandler* handler = new AliMCEventHandler;
+    if (!optionStr.Contains("particle-efficiency"))
+      handler->SetReadTR(kFALSE);
+    mgr->SetMCtruthEventHandler(handler);
+  }
+
+  // pt study
+  if (optionStr.Contains("pt-spectrum-func"))
+  {
+    TF1* func = new TF1("func", "1", 0, 0.2);
+    //TF1* func = new TF1("func", "1.5 - x / 0.2 * 0.5", 0, 0.2);
+    //TF1* func = new TF1("func", "1.25 - x / 0.2 * 0.25", 0, 0.2);
+    //TF1* func = new TF1("func", "0.75 + x / 0.2 * 0.25", 0, 0.2);
+    hist = func->GetHistogram();
+    //new TCanvas; func->Draw();
+    //inputList.Add(func->GetHistogram()->Clone("pt-spectrum"));
+
+    new TCanvas; hist->Draw();
+    task->SetPtSpectrum((TH1D*) hist->Clone("pt-spectrum"));
+  }
 
   // Add ESD handler
   AliESDInputHandler* esdH = new AliESDInputHandler;
+  esdH->SetInactiveBranches("AliESDACORDE FMD ALIESDTZERO ALIESDZDC AliRawDataErrorLogs CaloClusters Cascades EMCALCells EMCALTrigger ESDfriend Kinks AliESDTZERO ALIESDACORDE MuonTracks TrdTracks");
   mgr->SetInputEventHandler(esdH);
 
   // Attach input
-  cInput  = mgr->CreateContainer("cInput", TChain::Class(), AliAnalysisManager::kInputContainer);
+  cInput  = mgr->GetCommonInputContainer();
   mgr->ConnectInput(task, 0, cInput);
 
   // Attach output
@@ -84,13 +139,26 @@ void run(Char_t* data, Int_t nRuns=20, Int_t offset=0, Bool_t aDebug = kFALSE, B
   // Run analysis
   mgr->InitAnalysis();
   mgr->PrintStatus();
-  mgr->StartAnalysis((aProof) ? "proof" : "local", chain);
-}
 
-void runAll()
-{
-  run("part1.txt", 1000000, 0, kFALSE, kTRUE);
-  gSystem->Exec("mv multiplicityMC.root multiplicityMC_1.root");
-  run("part2.txt", 1000000, 0, kFALSE, kTRUE);
-  gSystem->Exec("mv multiplicityMC.root multiplicityMC_2.root");
+  if (aProof == 2)
+  {
+    // process dataset
+
+    mgr->StartAnalysis("proof", data, nRuns, offset);
+  }
+  else if (aProof == 3)
+  {
+    gROOT->ProcessLine(".L CreateChainFromDataSet.C");
+    ds = gProof->GetDataSet(data)->GetStagedSubset();
+    chain = CreateChainFromDataSet(ds);
+    mgr->StartAnalysis("local", chain, nRuns, offset);
+  }
+  else
+  {
+    // Create chain of input files
+    gROOT->LoadMacro("../CreateESDChain.C");
+    chain = CreateESDChain(data, nRuns, offset);
+
+    mgr->StartAnalysis((aProof > 0) ? "proof" : "local", chain);
+  }
 }