Patch for jet analysis
[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / macros / AddTaskJetResponseMaker.C
CommitLineData
5193b38c 1// $Id$
2
3AliJetResponseMaker* AddTaskJetResponseMaker(
5193b38c 4 const char *ntracks = "Tracks",
5 const char *nclusters = "CaloClusters",
6 const char *njets = "Jets",
7 const char *nmcjets = "MCJets",
8 const char *nmctracks = "MCParticles",
9 Double_t jetradius = 0.4,
10 Double_t jetptcut = 1,
ceefbfbc 11 Double_t jetareacut = 0.8,
5193b38c 12 Double_t ptcut = 0.15,
13 Double_t jetBiasTrack = 10,
14 Double_t jetBiasClus = 10,
2bddb6ae 15 Double_t maxDistance = 0.25,
4643d2e8 16 UInt_t type = AliAnalysisTaskEmcal::kTPC,
17 Int_t ptHardBin = -999,
53dca394 18 const char *taskname = "AliJetResponseMaker",
19 AliJetResponseMaker* address = 0
5193b38c 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("AddTaskJetResponseMaker", "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("AddTaskJetResponseMaker", "This task requires an input event handler");
36 return NULL;
37 }
38
39 //-------------------------------------------------------
40 // Init the task and do settings
41 //-------------------------------------------------------
42
4643d2e8 43 TString name(taskname);
44 name += "_";
45 name += njets;
46 name += "_Track";
47 name += jetBiasTrack;
48 name += "_Clus";
49 name += jetBiasClus;
50 name += "_R0";
51 name += floor(jetradius*10+0.5);
52 name += "_PtCut";
53 name += floor(ptcut*1000+0.5);
54 name += "_";
55 if (type == AliAnalysisTaskEmcal::kTPC)
56 name += "TPC";
57 else if (type == AliAnalysisTaskEmcal::kEMCAL)
58 name += "EMCAL";
59 else if (type == AliAnalysisTaskEmcal::kTPCSmall)
60 name += "TPCSmall";
61 else if (type == AliAnalysisTaskEmcal::kEMCALOnly)
62 name += "EMCALOnly";
63 if (ptHardBin != -999) {
64 name += "_PtHard";
65 name += ptHardBin;
66 }
67
53dca394 68 AliJetResponseMaker* jetTask = address;
69 if (jetTask)
70 new (jetTask) AliJetResponseMaker(name);
71 else
72 jetTask = new AliJetResponseMaker(name);
73
5193b38c 74 jetTask->SetAnaType(type);
75 jetTask->SetTracksName(ntracks);
76 jetTask->SetClusName(nclusters);
77 jetTask->SetJetsName(njets);
78 jetTask->SetMCJetsName(nmcjets);
79 jetTask->SetMCTracksName(nmctracks);
80 jetTask->SetPtCut(ptcut);
81 jetTask->SetJetRadius(jetradius);
82 jetTask->SetJetPtCut(jetptcut);
ceefbfbc 83 jetTask->SetPercAreaCut(jetareacut);
5193b38c 84 jetTask->SetPtBiasJetTrack(jetBiasTrack);
85 jetTask->SetPtBiasJetClus(jetBiasClus);
2bddb6ae 86 jetTask->SetMaxDistance(maxDistance);
4643d2e8 87 jetTask->SetVzRange(-10,10);
88 jetTask->SetPtHardBin(ptHardBin);
5193b38c 89
90 //-------------------------------------------------------
91 // Final settings, pass to manager and set the containers
92 //-------------------------------------------------------
93
94 mgr->AddTask(jetTask);
95
96 // Create containers for input/output
97 AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer() ;
4643d2e8 98 TString contname(name);
5193b38c 99 contname += "_histos";
100 AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(contname.Data(),
101 TList::Class(),AliAnalysisManager::kOutputContainer,
102 Form("%s", AliAnalysisManager::GetCommonFileName()));
103 mgr->ConnectInput (jetTask, 0, cinput1 );
104 mgr->ConnectOutput (jetTask, 1, coutput1 );
105
106 return jetTask;
107}