]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGJE/EMCALJetTasks/macros/AddTaskEmcalJetSpectraMECpA.C
from Megan
[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / macros / AddTaskEmcalJetSpectraMECpA.C
1 // $Id$
2
3 AliAnalysisTaskEmcalJetSpectraMECpA* AddTaskEmcalJetSpectraMECpA(
4    const char *outfilename    = "AnalysisOutput.root",
5    UInt_t type                = AliAnalysisTaskEmcal::kTPC,
6    const char *nRhosCh        = "rhoCh",
7    const char *nRhosChEm      = "rhoChEm",
8    const char *nRhosEm      = "rhoEm",
9    TF1 *sfunc                 = 0,
10    const Double_t radius      = 0.2,
11    const Double_t minPhi      = 1.8,
12    const Double_t maxPhi      = 2.74,
13    const Double_t minEta      = -0.3,
14    const Double_t maxEta      = 0.3,
15    const char* usedTracks     = "PicoTracks",
16    const char* outClusName    = "CaloClustersCorr",
17    const Double_t minTrackPt  = 0.15,
18    const Double_t minClusterPt = 0.30,
19    const char *CentEst         = "V0A"
20    )
21 {  
22   // Get the pointer to the existing analysis manager via the static access method.
23   //==============================================================================
24   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
25   if (!mgr)
26   {
27     ::Error("AddTasEmcalJetSpectraMECpA", "No analysis manager to connect to.");
28     return NULL;
29   }  
30   
31   // Check the analysis type using the event handlers connected to the analysis manager.
32   //==============================================================================
33   if (!mgr->GetInputEventHandler())
34   {
35     ::Error("AddTaskEmcalJetSpectraMECpA", "This task requires an input event handler");
36     return NULL;
37   }
38   
39   //Run the jet finder and rho tasks first
40
41   // Some constants for the jet finders
42   const Int_t cKT                 = 0;
43   const Int_t cANTIKT             = 1;
44   const Int_t cFULLJETS           = 0;
45   const Int_t cCHARGEDJETS        = 1;
46   const Int_t cNEUTRALJETS        = 2;
47     
48   float AreaCut = 0.6*radius*radius*TMath::Pi();
49
50   sfunc=new TF1("sfunc","[0]*x*x+[1]*x+[2]",-1,100);
51   sfunc->SetParameter(0,0.0);
52   sfunc->SetParameter(1,0.0);
53   sfunc->SetParameter(2,1.5);
54
55
56   gROOT->LoadMacro("$ALICE_ROOT/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJet.C");
57   gROOT->LoadMacro("$ALICE_ROOT/PWGJE/EMCALJetTasks/macros/AddTaskRhoSparse.C");
58
59   //const char *nJets;
60   TString nJets("");
61
62   TString scaledname(Form("%s_Scaled", nRhosCh));
63   TString newrhoname(Form("%s_All", nRhosCh));
64
65   
66   if(!(usedTracks=="")){
67     cout << "USEDTRACKS EXISTS" << usedTracks <<endl;
68     AliEmcalJetTask* jetFinderTaskChBack = AddTaskEmcalJet(usedTracks,"",cKT,radius,cCHARGEDJETS,minTrackPt, minClusterPt);
69
70     AliEmcalJetTask* jetFinderTaskChBackall = AddTaskEmcalJet(usedTracks,"",cKT,radius,cCHARGEDJETS,minTrackPt, minClusterPt,0.01,"Jets_allpt");
71     jetFinderTaskChBackall->SetMinJetPt(0);
72
73     AliEmcalJetTask* jetFinderTaskChSig = AddTaskEmcalJet(usedTracks,"",cANTIKT,radius,cCHARGEDJETS,minTrackPt, minClusterPt);
74
75     AliEmcalJetTask* jetFinderTaskChEmBack = AddTaskEmcalJet(usedTracks,outClusName,cKT,radius,cFULLJETS,minTrackPt, minClusterPt);
76     jetFinderTaskChEmBack->SetMinJetPt(0);
77
78   AliEmcalJetTask* jetFinderTask = AddTaskEmcalJet(usedTracks,outClusName,cANTIKT,radius, cFULLJETS,minTrackPt,minClusterPt);
79
80   AliAnalysisTaskRhoSparse *rhochtask = AddTaskRhoSparse(jetFinderTaskChBack->GetName(),jetFinderTaskChSig->GetName(),usedTracks,outClusName,nRhosCh,radius,0,0.01,0,0,sfunc,2,kTRUE,nRhosCh);
81   rhochtask->SetCentralityEstimator(CentEst);
82
83   AliAnalysisTaskRhoSparse *rhochalltask = AddTaskRhoSparse(jetFinderTaskChBackall->GetName(),jetFinderTaskChSig->GetName(),usedTracks,outClusName,newrhoname,radius,0,0.0,0,0,sfunc,2,kTRUE,newrhoname);
84   rhochtask->SetCentralityEstimator(CentEst);
85
86
87   AliAnalysisTaskRhoSparse *rhochemtask = AddTaskRhoSparse(jetFinderTaskChEmBack->GetName(),jetFinderTask->GetName(),usedTracks,outClusName,nRhosChEm,radius,0,0.01,0,0,0,1,kTRUE,nRhosChEm);
88   rhochemtask->SetCentralityEstimator(CentEst);
89
90   //nJets=jetFinderTask->GetName();
91   nJets+=jetFinderTask->GetName();
92
93   gROOT->LoadMacro("$ALICE_ROOT/PWGJE/EMCALJetTasks/macros/AddTaskDeltaPt.C");
94
95   TString deltaname(Form("DeltaPt_%s_Scaled", nRhosCh));
96   AliAnalysisTaskDeltaPt* deltapt = AddTaskDeltaPt(usedTracks,outClusName,nJets,"","","","","",scaledname,radius,1,AreaCut,minTrackPt,minClusterPt,AliAnalysisTaskEmcal::kEMCAL,deltaname);
97   deltapt->SetCentralityEstimator(CentEst);
98
99   TString emcdeltaname(Form("DeltaPt_%s", nRhosChEm));
100   AliAnalysisTaskDeltaPt* deltaptEMC = AddTaskDeltaPt(usedTracks,outClusName,nJets,"","","","","",nRhosChEm,radius,1,AreaCut,minTrackPt,minClusterPt,AliAnalysisTaskEmcal::kEMCAL,emcdeltaname);
101   deltapt->SetCentralityEstimator(CentEst);
102
103   gROOT->LoadMacro("$ALICE_ROOT/PWGJE/EMCALJetTasks/macros/AddTaskScale.C");
104
105
106   Int_t radlabel=(Int_t)floor(radius*100+0.5);
107   Int_t mincluslabel=(Int_t)floor(minClusterPt*1000+0.5);
108   Int_t mintracklabel=(Int_t)floor(minTrackPt*1000+0.5);
109   TString scalename(Form("Scale_R0%d", radlabel));
110
111   AliAnalysisTaskScale* scaletask = AddTaskScale(usedTracks,outClusName,minTrackPt,minClusterPt,scalename);
112   scaletask->SetCentralityEstimator(CentEst);
113   scaletask->SetScaleFunction(sfunc);
114
115   
116   }
117
118   cout << "Running non charged jet finders..." <<endl;
119   AliEmcalJetTask* jetFinderTaskEm = AddTaskEmcalJet("",outClusName,cANTIKT,radius,cNEUTRALJETS,minTrackPt,minClusterPt);
120
121   AliEmcalJetTask* jetFinderTaskEmBack = AddTaskEmcalJet("",outClusName,cKT,radius,cNEUTRALJETS,minTrackPt, minClusterPt);
122
123   cout << "Running non charged rho task..." <<endl;
124
125   AliAnalysisTaskRhoSparse *rhoemtask = AddTaskRhoSparse(jetFinderTaskEmBack->GetName(),jetFinderTaskEm->GetName(),usedTracks,outClusName,nRhosEm,radius,0,0.01,0,0,0,1,kTRUE,nRhosEm);
126   rhoemtask->SetCentralityEstimator(CentEst);
127
128   if(usedTracks=="") nJets +=jetFinderTaskEm->GetName();
129
130
131   //-------------------------------------------------------
132   // Init the task and do settings
133   //-------------------------------------------------------
134
135
136   cout << "Ready to run my task..." << nJets <<endl;
137
138   TString name(Form("SpectraMECpA_%s", nJets.Data()));
139   AliAnalysisTaskEmcalJetSpectraMECpA *spectratask = new AliAnalysisTaskEmcalJetSpectraMECpA(name);
140   spectratask->SetJetsName(nJets.Data());
141   spectratask->SetCentralityEstimator(CentEst);
142   spectratask->SetAnaType(type);
143   if(!(usedTracks=="")) spectratask->SetRhoName(nRhosCh);
144   else spectratask->SetRhoName(nRhosEm);
145   spectratask->SetJetPhiLimits(minPhi,maxPhi);
146   spectratask->SetJetEtaLimits(minEta,maxEta);
147   spectratask->SetJetAreaCut(AreaCut);
148   spectratask->SetTracksName(usedTracks);
149
150   //-------------------------------------------------------
151   // Final settings, pass to manager and set the containers
152   //-------------------------------------------------------
153
154   mgr->AddTask(spectratask);
155
156   // Create containers for input/output
157   mgr->ConnectInput (spectratask, 0, mgr->GetCommonInputContainer() );
158   AliAnalysisDataContainer *cospectra = mgr->CreateContainer(name,
159                                                            TList::Class(),
160                                                            AliAnalysisManager::kOutputContainer,
161                                                            outfilename);
162   mgr->ConnectOutput(spectratask,1,cospectra);
163
164   return spectratask;
165 }