PWGLF/SPECTRA/PiKaPr/TestAOD/runGrid.C

   1 class  AliAnalysisManager;
   2 class  AliAnalysisAlien;
   3
   4 void runGrid(TString mode="test",Bool_t mc=0,Int_t day=19,Int_t month=7, Int_t year=2012)
   5 {
   6   //to be used with Aliroot > v5-03-32-AN
   7   AliLog::SetGlobalDebugLevel(100);
   8   // Load common libraries
   9   gEnv->SetValue("XSec.GSI.DelegProxy", "2");
  10
  11   // Load common libraries
  12   gSystem->Load("libCore");
  13   gSystem->Load("libTree");
  14   gSystem->Load("libGeom");
  15   gSystem->Load("libVMC");
  16   gSystem->Load("libPhysics");
  17   gSystem->Load("libMinuit");
  18   gSystem->Load("libGui");
  19   gSystem->Load("libXMLParser");
  20   gSystem->Load("libSTEERBase");
  21   gSystem->Load("libESD");
  22   gSystem->Load("libCDB");
  23   gSystem->Load("libAOD");
  24   gSystem->Load("libANALYSIS");
  25   gSystem->Load("libANALYSISalice");
  26   gSystem->Load("libCORRFW");
  27   gSystem->Load("libProof");
  28   gSystem->Load("libRAWDatabase");
  29   gSystem->Load("libSTEER");
  30   //__________________________________________________________________________
  31   // Use AliRoot includes to compile our task
  32   gROOT->ProcessLine(".include $ALICE_ROOT/include");
  33   gSystem->Load("libPWGLFspectra");
  34
  35   // Create and configure the alien handler plugin
  36   AliAnalysisGrid *alienHandler = CreateAlienHandler(mode,mc,day,month,year);
  37   if (!alienHandler) return;
  38   // Create the analysis manager
  39   AliAnalysisManager *mgr = new AliAnalysisManager("testAnalysis");
  40   // Connect plug-in to the analysis manager
  41   mgr->SetGridHandler(alienHandler);
  42   AliAODInputHandler* aodH = new AliAODInputHandler();
  43   mgr->SetInputEventHandler(aodH);
  44
  45   // Add PID task
  46   gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
  47   AliAnalysisTaskPIDResponse *taskPID=AddTaskPIDResponse(mc);
  48   // Printf("OADB PATH:::::%s",taskPID->GetOADBPath());
  49   // taskPID->SetOADBPath("alien:///alice/cern.ch/user/a/akalweit/ForLeornado/OADB");
  50   // Printf("OADB PATH:::::%s",taskPID->GetOADBPath());
  51
  52   gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskVZEROEPSelection.C");
  53   AliVZEROEPSelectionTask *selTask = AddTaskVZEROEPSelection();
  54
  55   gROOT->LoadMacro("$ALICE_ROOT/PWGLF/SPECTRA/PiKaPr/TestAOD/AddTaskSpectraAOD.C");
  56   //LOOP OVER SELECTION
  57   //                            0    1    2    3    4    5    6    7    8    9
  58   Double_t Nsigmapid=3.;
  59   Double_t pt=10000.;
  60   Double_t p=10000.;
  61   Double_t y=.5;
  62   Double_t ptTofMatch=.6;
  63   UInt_t trkbit=1024;
  64   UInt_t trkbitQVector=1;
  65   Bool_t UseCentPatchAOD049=kFALSE;
  66   Double_t DCA=100000;
  67   UInt_t minNclsTPC=70;
  68   Int_t nrebin=30;
  69   TString opt="";
  70   Int_t centcut=30;
  71
  72   for(Int_t icut=0;icut<1;icut++){
  73     //if(icut!=0)continue;
  74     //MB
  75     AliAnalysisTaskSpectraAOD *taskAOD =AddTaskSpectraAOD(mc,centcut,centcut+1,0,100,-0.8,0.8,Nsigmapid,pt,p,y,ptTofMatch,trkbit,trkbitQVector,UseCentPatchAOD049,DCA,minNclsTPC,nrebin,opt);
  76     taskAOD->GetOutputSlot(1)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(1)->GetContainer()->GetName(),taskAOD->GetName()));
  77     taskAOD->GetOutputSlot(2)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(2)->GetContainer()->GetName(),taskAOD->GetName()));
  78     taskAOD->GetOutputSlot(3)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(3)->GetContainer()->GetName(),taskAOD->GetName()));
  79     taskAOD->GetOutputSlot(4)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(4)->GetContainer()->GetName(),taskAOD->GetName()));
  80     //taskAOD->GetEventCuts()->SetMultiplicityCut(250,450);
  81     taskAOD->GetEventCuts()->PrintCuts();
  82     //smallq
  83     AliAnalysisTaskSpectraAOD *taskAOD =AddTaskSpectraAOD(mc,centcut,centcut+1,0,0.4,-0.8,0.8,Nsigmapid,pt,p,y,ptTofMatch,trkbit,trkbitQVector,UseCentPatchAOD049,DCA,minNclsTPC,nrebin,opt);
  84     taskAOD->GetOutputSlot(1)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(1)->GetContainer()->GetName(),taskAOD->GetName()));
  85     taskAOD->GetOutputSlot(2)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(2)->GetContainer()->GetName(),taskAOD->GetName()));
  86     taskAOD->GetOutputSlot(3)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(3)->GetContainer()->GetName(),taskAOD->GetName()));
  87     taskAOD->GetOutputSlot(4)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(4)->GetContainer()->GetName(),taskAOD->GetName()));
  88     //taskAOD->GetEventCuts()->SetMultiplicityCut(250,450);
  89     taskAOD->GetEventCuts()->PrintCuts();
  90     taskAOD->GetEventCuts()->SetCentFromV0(kTRUE);
  91     //largeq
  92     AliAnalysisTaskSpectraAOD *taskAOD =AddTaskSpectraAOD(mc,centcut,centcut+1,1.7,100,-0.8,0.8,Nsigmapid,pt,p,y,ptTofMatch,trkbit,trkbitQVector,UseCentPatchAOD049,DCA,minNclsTPC,nrebin,opt);
  93     taskAOD->GetOutputSlot(1)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(1)->GetContainer()->GetName(),taskAOD->GetName()));
  94     taskAOD->GetOutputSlot(2)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(2)->GetContainer()->GetName(),taskAOD->GetName()));
  95     taskAOD->GetOutputSlot(3)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(3)->GetContainer()->GetName(),taskAOD->GetName()));
  96     taskAOD->GetOutputSlot(4)->GetContainer()->SetName(Form("%s_%s",taskAOD->GetOutputSlot(4)->GetContainer()->GetName(),taskAOD->GetName()));
  97     //taskAOD->GetEventCuts()->SetMultiplicityCut(250,450);
  98     taskAOD->GetEventCuts()->PrintCuts();
  99     //increase centrality
 100     centcut++;
 101     }
 102
 103   mgr->SetDebugLevel(2);
 104
 105   //mgr->Init();
 106   if (!mgr->InitAnalysis())return;
 107   mgr->PrintStatus();
 108   // Start analysis in grid.
 109   mgr->StartAnalysis("grid");
 110 }
 111
 112
 113 AliAnalysisGrid* CreateAlienHandler(TString mode="test",Bool_t mc=1,Int_t day=0,Int_t month=0, Int_t year=2012)
 114 {
 115   AliAnalysisAlien *plugin = new AliAnalysisAlien();
 116   plugin->AddIncludePath("-I. -I$ROOTSYS/include -I$ALICE_ROOT/include -I$ALICE_ROOT/TOF");
 117   plugin->SetAdditionalLibs("libPWGLFspectra.so");
 118   // gROOT->LoadMacro("AliSpectraAODTrackCuts.cxx+g");
 119   // gROOT->LoadMacro("AliSpectraAODEventCuts.cxx+g");
 120   // gROOT->LoadMacro("AliSpectraAODHistoManager.cxx+g");
 121   // gROOT->LoadMacro("AliSpectraAODPID.cxx+g");
 122   // gROOT->LoadMacro("AliAnalysisTaskSpectraAOD.cxx+g");
 123
 124   //plugin->SetAnalysisSource("AliSpectraAODTrackCuts.cxx AliSpectraAODEventCuts.cxx AliSpectraAODHistoManager.cxx AliSpectraAODPID.cxx AliAnalysisTaskSpectraAOD.cxx");
 125   //plugin->SetAdditionalLibs("Histograms.h HistogramNames.h HistogramNames.cxx AliSpectraAODTrackCuts.cxx AliSpectraAODEventCuts.cxx AliSpectraAODHistoManager.cxx AliSpectraAODPID.cxx AliAnalysisTaskSpectraAOD.cxx AliSpectraAODTrackCuts.h AliSpectraAODEventCuts.h AliSpectraAODHistoManager.h AliSpectraAODPID.h AliAnalysisTaskSpectraAOD.h");
 126
 127   plugin->SetOverwriteMode();
 128   plugin->SetExecutableCommand("aliroot -q -b");
 129   plugin->SetRunMode(mode.Data());
 130   plugin->SetNtestFiles(1);
 131   //Set versions of used packages
 132   plugin->SetAPIVersion("V1.1x");
 133   plugin->SetROOTVersion("v5-33-02b");
 134   plugin->SetAliROOTVersion("v5-03-35-AN");
 135   // Declare input data to be processed.
 136   if(mc)
 137     {
 138       // plugin->SetGridDataDir("/alice/sim/LHC11a10a_bis");
 139       // plugin->SetDataPattern("AOD090/*AliAOD.root");
 140       // plugin->SetRunPrefix("");
 141       // plugin->SetAnalysisMacro("TaskAODPbPbMC.C");
 142       // plugin->SetExecutable("TaskAODPbPbMC.sh");
 143       // plugin->SetJDLName("TaskAODPbPbMC.jdl");
 144       // //plugin->SetSplitMaxInputFileNumber(500);
 145       // plugin->SetGridWorkingDir(Form("/AODPbPb%d%d%d/mc/",day,month,year));
 146
 147       plugin->SetGridDataDir(" /alice/sim/2012/LHC12a11e/");
 148       plugin->SetDataPattern("AOD081/*AliAOD.root");
 149       plugin->SetRunPrefix("");
 150       plugin->SetAnalysisMacro("TaskAODPbPbMC.C");
 151       plugin->SetExecutable("TaskAODPbPbMC.sh");
 152       plugin->SetJDLName("TaskAODPbPbMC.jdl");
 153       //plugin->SetSplitMaxInputFileNumber(200);
 154       plugin->SetGridWorkingDir(Form("/AODPbPb%d%d%d/mcAMPTNoPhysSel_RangeXLarger/",day,month,year));
 155     }
 156   else
 157     {
 158       plugin->SetGridDataDir("/alice/data/2010/LHC10h");
 159       plugin->SetDataPattern("ESDs/pass2/AOD086/*AliAOD.root");
 160       plugin->SetRunPrefix("000");
 161       plugin->SetAnalysisMacro("TaskAODPbPbdata.C");
 162       plugin->SetExecutable("TaskAODPbPbdata.sh");
 163       plugin->SetJDLName("TaskAODPbPbdata.jdl");
 164       //plugin->SetSplitMaxInputFileNumber(500);
 165       plugin->SetGridWorkingDir(Form("/AODPbPb%d%d%d/dataFinalPlot3040/",day,month,year));
 166     }
 167   FILE* listruns=fopen("RunListGrid-AOD086.txt","r");
 168   Int_t irun;
 169   while(!feof(listruns))
 170     {
 171       fscanf(listruns,"%d\n",&irun);
 172       plugin->AddRunNumber(irun);
 173     }
 174   // plugin->AddRunNumber(139437);
 175   // plugin->AddRunNumber(139038);
 176   // plugin->AddRunNumber(138653);
 177   // plugin->AddRunNumber(138534);
 178   // plugin->AddRunNumber(137686);
 179
 180   // Declare alien output directory. Relative to working directory.
 181     plugin->SetGridOutputDir("output"); // In this case will be $HOME/work/output
 182   //plugin->SetGridOutputDir("/alice/cern.ch/user/l/lmilano/AODPbPb472012/data/output/000/");
 183   plugin->SetNrunsPerMaster(24); //
 184
 185   // Declare the output file names separated by blancs.
 186   // (can be like: file.root or file.root@ALICE::Niham::File)
 187   //plugin->SetDefaultOutputs(0);
 188   //plugin->SetOutputFiles("AnalysisResults.root.root");
 189   plugin->SetMergeViaJDL(1);
 190   plugin->SetOneStageMerging(kFALSE);
 191   plugin->SetMaxMergeStages(2);
 192   //plugin->SetMaxMergeFiles(100);
 193   plugin->SetTTL(100000);
 194   // Optionally set input format (default xml-single)
 195   plugin->SetInputFormat("xml-single");
 196   // Optionally modify job price (default 1)
 197   plugin->SetPrice(1);
 198   // Optionally modify split mode (default 'se')
 199   //plugin->SetSplitMaxInputFileNumber();
 200   plugin->SetSplitMode("se");
 201   return plugin;
 202 }