Various fixes, and possibility to set trigger efficiency
[u/mrichter/AliRoot.git] / PWGLF / FORWARD / analysis2 / AddTaskCentraldNdeta.C
1 /**
2  * @file   AddTaskCentraldNdeta.C
3  * @author Christian Holm Christensen <cholm@nbi.dk>
4  * @date   Fri Jan 28 10:22:26 2011
5  * 
6  * @brief Script to add a multiplicity task for the central
7  *        @f$\eta@f$ region
8  * 
9  * 
10  * @ingroup pwglf_forward_scripts_tasks
11  */
12 /** 
13  * Create the central @f$ dN/d\eta@f$ analysis task 
14  * 
15  * @param trig      Trigger to use 
16  * @param vzMin     Smallest @f$ v_z@f$
17  * @param vzMax     Biggest @f$ v_z@f$
18  * @param useCent   Whether to use the centrality or not
19  * @param scheme    Normalisation scheme
20  * @param cutEdges  Whether to cut edges when rebinning 
21  * @param mcanalysisfilename Take final MC corrections from this - if present
22  * 
23  * @return Newly created and configured task
24  *
25  * @ingroup pwglf_forward_dndeta
26  */
27 AliAnalysisTask*
28 AddTaskCentraldNdeta(const char* trig     = "INEL", 
29                      Double_t    vzMin    = -10, 
30                      Double_t    vzMax    = +10, 
31                      Bool_t      useCent  = false,
32                      const char* scheme   = 0,
33                      Bool_t      cutEdges = false,
34                      Double_t    trigEff  = 1, 
35                      Double_t    trigEff0 = 1,
36                      const char* mcanalysisfilename = "none")
37 {
38   // --- Load libraries ----------------------------------------------
39   gROOT->LoadClass("AliAODForwardMult", "libPWGLFforward2");
40
41   // --- Analysis manager --------------------------------------------
42   AliAnalysisManager* mgr = AliAnalysisManager::GetAnalysisManager();
43
44   // --- Check that we have an AOD input handler ---------------------
45   UShort_t aodInput = 0;
46   if (!(aodInput = AliForwardUtil::CheckForAOD())) 
47     Fatal("","Cannot proceed without and AOD handler");
48   if (aodInput == 2 &&
49       !AliForwardUtil::CheckForTask("AliCentralMultiplicityTask")) 
50     Fatal("","The relevant task wasn't added to the train");
51
52   // --- Make our object ---------------------------------------------
53   AliCentraldNdetaTask* task = new AliCentraldNdetaTask("Central");
54   task->SetMCFinalCorrFilename(mcanalysisfilename);
55   
56   // Set the vertex range to use 
57   task->SetVertexRange(vzMin, vzMax);
58   // Set the trigger mask to use (INEL,INEL>0,NSD)
59   task->SetTriggerMask(trig);
60   task->SetTriggerEff(trigEff); // 0.997535);
61   task->SetTriggerEff0(trigEff0); 
62   // Whether to cut edges when re-binning 
63   task->SetCutEdges(cutEdges);
64   // Whether to correct for empty bins when projecting 
65   task->SetCorrEmpty(false);
66   // Whether to use TH2::ProjectionX 
67   task->SetUseROOTProjectX(false);
68   // Bit mask of 
69   // 
70   //    kNone               Normalise to accepted events 
71   //    kEventLevel         Normalise to all events in selected range 
72   //    kBackground         Also correct for background triggers 
73   //    kTriggerEfficiency  Correct for trigger efficiency 
74   //    kShape              Correct shape 
75   // 
76   // kNone and kEventLevel are mutually exclusive.  If kEventLevel is
77   // not specified, then kNone is assumed.  kBackground only makes
78   // sense with kEventLevel. Furthermore, there
79   // are some constants that encode the common cases
80   //     
81   //    kFull    = kEventLevel |  kBackground | kShape | kTriggerEfficiency
82   // 
83   // Default is kFull
84   task->SetNormalizationScheme(AliBasedNdetaTask::kFull);
85   if (scheme) task->SetNormalizationScheme(scheme);
86   // Set the centrality bins to use.  These are mutually exclusive.
87   // Note, that a bin specified as a-b, covers the interval from a,
88   // inclusive to b exclusive.  An upper bound of 100 is treated
89   // especially, and the upper bound is inclusive in that case .
90   if (useCent) {
91     Short_t bins[] = { 0, 5, 10, 20, 30, 40, 50, 60, 70, 80, 90, 100 };
92     task->SetCentralityAxis(11, bins);
93   }
94   mgr->AddTask(task);
95
96   // --- create containers for input/output --------------------------
97   AliAnalysisDataContainer *sums = 
98     mgr->CreateContainer("CentralSums", TList::Class(), 
99                          AliAnalysisManager::kOutputContainer, 
100                          AliAnalysisManager::GetCommonFileName());
101   AliAnalysisDataContainer *output = 
102     mgr->CreateContainer("CentralResults", TList::Class(), 
103                          AliAnalysisManager::kParamContainer, 
104                          AliAnalysisManager::GetCommonFileName());
105   
106   // --- connect input/output ----------------------------------------
107   mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer());
108   mgr->ConnectOutput(task, 1, sums);
109   mgr->ConnectOutput(task, 2, output);
110   
111   return task;
112 }
113
114   
115 //________________________________________________________________________
116 //
117 // EOF
118 //