296dd262 |
1 | /* $Id: AliEventStatsTask.cxx 35782 2009-10-22 11:54:31Z jgrosseo $ */ |
2 | |
3 | #include "AliEventStatsTask.h" |
4 | |
296dd262 |
5 | #include <TFile.h> |
296dd262 |
6 | #include <TH1F.h> |
7 | #include <TH2F.h> |
296dd262 |
8 | |
9 | #include <AliLog.h> |
10 | #include <AliESDEvent.h> |
11 | #include <AliHeader.h> |
296dd262 |
12 | |
13 | #include "AliPhysicsSelection.h" |
b43e01ed |
14 | //#include "AliBackgroundSelection.h" |
296dd262 |
15 | |
16 | ClassImp(AliEventStatsTask) |
17 | |
18 | AliEventStatsTask::AliEventStatsTask(const char* opt) : |
19 | AliAnalysisTaskSE("AliEventStatsTask"), |
20 | fOutput(0), |
21 | fOption(opt), |
22 | fPhysicsSelection(0) |
23 | { |
24 | // |
25 | // Constructor. Initialization of pointers |
26 | // |
27 | |
28 | // Define input and output slots here |
29 | DefineOutput(1, TList::Class()); |
30 | |
31 | AliLog::SetClassDebugLevel("AliEventStatsTask", AliLog::kWarning); |
32 | } |
33 | |
34 | AliEventStatsTask::~AliEventStatsTask() |
35 | { |
36 | // |
37 | // Destructor |
38 | // |
39 | |
40 | // histograms are in the output list and deleted when the output |
41 | // list is deleted by the TSelector dtor |
42 | |
43 | if (fOutput) { |
44 | delete fOutput; |
45 | fOutput = 0; |
46 | } |
47 | } |
48 | |
49 | void AliEventStatsTask::UserCreateOutputObjects() |
50 | { |
51 | // create result objects and add to output list |
52 | |
53 | Printf("AliEventStatsTask::CreateOutputObjects"); |
54 | |
55 | fOutput = new TList; |
56 | fOutput->SetOwner(); |
57 | |
b43e01ed |
58 | if (!fPhysicsSelection) |
b43e01ed |
59 | fPhysicsSelection = new AliPhysicsSelection; |
b43e01ed |
60 | |
296dd262 |
61 | fOutput->Add(fPhysicsSelection); |
62 | } |
63 | |
64 | void AliEventStatsTask::UserExec(Option_t*) |
65 | { |
66 | // process the event |
67 | |
68 | // post the data already here |
69 | PostData(1, fOutput); |
70 | |
71 | AliESDEvent* esd = dynamic_cast<AliESDEvent*> (InputEvent()); |
72 | |
73 | if (!esd) |
74 | { |
75 | AliError("ESD branch not available"); |
76 | return; |
77 | } |
78 | |
79 | fPhysicsSelection->IsCollisionCandidate(esd); |
80 | } |
81 | |
82 | void AliEventStatsTask::Terminate(Option_t *) |
83 | { |
84 | // The Terminate() function is the last function to be called during |
85 | // a query. It always runs on the client, it can be used to present |
86 | // the results graphically or save the results to file. |
87 | |
88 | fOutput = dynamic_cast<TList*> (GetOutputData(1)); |
89 | if (!fOutput) |
90 | Printf("ERROR: fOutput not available"); |
91 | |
92 | if (fOutput) |
93 | { |
94 | fPhysicsSelection = dynamic_cast<AliPhysicsSelection*> (fOutput->FindObject("AliPhysicsSelection")); |
95 | } |
96 | |
97 | TFile* fout = new TFile("event_stat.root", "RECREATE"); |
98 | |
99 | if (fPhysicsSelection) |
100 | { |
101 | fPhysicsSelection->Print(); |
102 | fPhysicsSelection->SaveHistograms("physics_selection"); |
103 | } |
104 | |
105 | fout->Write(); |
106 | fout->Close(); |
107 | |
108 | Printf("Writting result to event_stat.root"); |
109 | } |