]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGDQ/dielectron/macrosLMEE/AddTask_Remi_LMEEPbPb2011AOD_Peri.C
Merge branch 'feature-movesplit'
[u/mrichter/AliRoot.git] / PWGDQ / dielectron / macrosLMEE / AddTask_Remi_LMEEPbPb2011AOD_Peri.C
1 AliAnalysisTask *AddTask_Remi_LMEEPbPb2011AOD_Peri(Char_t* outputFileName="LMEEoutput.root", Bool_t runAll=kFALSE,Bool_t setMC=kFALSE,Bool_t getFromAlien=kFALSE, Bool_t PIDbaseline=kFALSE, Bool_t rejOnly=kTRUE) {
2   
3   Bool_t bESDANA=kFALSE; //Autodetect via InputHandler
4   //get the current analysis manager
5   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
6   if (!mgr) {
7     Error("AddTask_Remi_LMEEPbPb2011", "No analysis manager found.");
8     return 0;
9   }
10   
11   
12   //  create task and add it to the manager
13   //    gSystem->AddIncludePath("$ALICE_ROOT/PWGDQ/dielectron/macrosLMEE");
14   
15   
16   TString configBasePath("$TRAIN_ROOT/cbaumann_dielectron/");
17   TString trainRoot=gSystem->Getenv("TRAIN_ROOT");
18   //  if (trainRoot.IsNull()) configBasePath= "/home/tanizaki/nfs/LMee_Deflection/ver1/";
19   if (trainRoot.IsNull()) configBasePath= "$ALICE_ROOT/PWGDQ/dielectron/macrosLMEE/";
20   
21   
22   if (getFromAlien &&
23       (!gSystem->Exec("alien_cp alien:///alice/cern.ch/user/c/cbaumann/PWGDQ/dielectron/macrosLMEE/ConfigRemiLMEEPbPb2011AOD.C .")) &&
24       (!gSystem->Exec("alien_cp alien:///alice/cern.ch/user/c/cbaumann/PWGDQ/dielectron/macrosLMEE/LMEECutLibRemi.C ."))
25       ) {
26     configBasePath=Form("%s/",gSystem->pwd());
27   }
28   
29   
30   TString configFile("ConfigRemiLMEEPbPb2011AOD.C");
31   TString configLMEECutLib("LMEECutLibRemi.C");
32   
33   TString configFilePath(configBasePath+configFile);
34   TString configLMEECutLibPath(configBasePath+configLMEECutLib);
35   
36   //AOD Usage currently tested with separate task, to be merged
37   if (mgr->GetInputEventHandler()->IsA()==AliAODInputHandler::Class()){
38     ::Info("AddTaskLMEEPbPb2011", "no dedicated AOD configuration");
39   }
40   else if (mgr->GetInputEventHandler()->IsA()==AliESDInputHandler::Class()){
41     ::Info("AddTaskLMEEPbPb2011AOD","switching on ESD specific code");
42     bESDANA=kTRUE;
43   }
44   
45   
46   //Do we have an MC handler?
47   Bool_t hasMC=setMC;
48   if (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0)
49     hasMC=kTRUE;
50   
51   
52   
53   if (!gROOT->GetListOfGlobalFunctions()->FindObject(configLMEECutLib.Data()))
54     gROOT->LoadMacro(configLMEECutLibPath.Data());
55   if (!gROOT->GetListOfGlobalFunctions()->FindObject(configFile.Data()))
56     gROOT->LoadMacro(configFilePath.Data());
57   
58   
59   LMEECutLibRemi* cutlib = new LMEECutLibRemi();
60   AliAnalysisTaskMultiDielectron *task=new AliAnalysisTaskMultiDielectron("MultiDiEData");
61   if (!hasMC) task->UsePhysicsSelection();
62   task->SetTriggerMask(AliVEvent::kMB+AliVEvent::kCentral+AliVEvent::kSemiCentral);
63   //  task->SelectCollisionCandidates(AliVEvent::kMB+AliVEvent::kCentral+AliVEvent::kSemiCentral);
64   //  task->SetRejectPileup();
65   task->SelectCollisionCandidates(AliVEvent::kAny);  
66   task->SetEventFilter(cutlib->GetEventCuts(LMEECutLibRemi::kPbPb2011pidITSTPCTOF)); //
67   
68   //load dielectron configuration file
69   
70   //add dielectron analysis with different cuts to the task
71   
72   
73   
74     AliDielectron *lowmass4=ConfigRemiLMEEPbPb2011AOD(4,hasMC,bESDANA);
75     task->AddDielectron(lowmass4);
76     printf("add: %s\n",lowmass4->GetName());
77     AliDielectron *lowmass8=ConfigRemiLMEEPbPb2011AOD(8,hasMC,bESDANA);
78     task->AddDielectron(lowmass8);
79     printf("add: %s\n",lowmass8->GetName());
80     AliDielectron *lowmass12=ConfigRemiLMEEPbPb2011AOD(12,hasMC,bESDANA);
81     task->AddDielectron(lowmass12);
82     printf("add: %s\n",lowmass12->GetName());
83     AliDielectron *lowmass16=ConfigRemiLMEEPbPb2011AOD(16,hasMC,bESDANA);
84     task->AddDielectron(lowmass16);
85     printf("add: %s\n",lowmass16->GetName());
86
87     AliDielectron *lowmass20=ConfigRemiLMEEPbPb2011AOD(20,hasMC,bESDANA);
88     task->AddDielectron(lowmass20);
89     printf("add: %s\n",lowmass20->GetName());
90     AliDielectron *lowmass24=ConfigRemiLMEEPbPb2011AOD(24,hasMC,bESDANA);
91     task->AddDielectron(lowmass24);
92     printf("add: %s\n",lowmass24->GetName());
93     AliDielectron *lowmass28=ConfigRemiLMEEPbPb2011AOD(28,hasMC,bESDANA);
94     task->AddDielectron(lowmass28);
95     printf("add: %s\n",lowmass28->GetName());
96
97   
98   mgr->AddTask(task);
99
100   //create output container
101   AliAnalysisDataContainer *coutput1 =
102         mgr->CreateContainer("Remi_LMEEPbPb2011Peri_tree",
103                 TTree::Class(),
104                 AliAnalysisManager::kExchangeContainer,
105                 outputFileName);
106
107   AliAnalysisDataContainer *cOutputHist1 =
108         mgr->CreateContainer("Remi_LMEEPbPb2011Peri_out",
109                 TList::Class(),
110                 AliAnalysisManager::kOutputContainer,
111                 outputFileName);
112
113   AliAnalysisDataContainer *cOutputHist2 =
114         mgr->CreateContainer("Remi_LMEEPbPb2011Peri_CF",
115                 TList::Class(),
116                 AliAnalysisManager::kOutputContainer,
117                 outputFileName);
118
119   AliAnalysisDataContainer *cOutputHist3 =
120         mgr->CreateContainer("Remi_EventStatPbPb2011Peri",
121                 TH1D::Class(),
122                 AliAnalysisManager::kOutputContainer,
123                 outputFileName);
124
125
126   mgr->ConnectInput(task,  0, mgr->GetCommonInputContainer());
127   mgr->ConnectOutput(task, 0, coutput1 );
128   mgr->ConnectOutput(task, 1, cOutputHist1);
129   mgr->ConnectOutput(task, 2, cOutputHist2);
130   mgr->ConnectOutput(task, 3, cOutputHist3);
131
132   return task;
133 }