3 #include "AliPhysicsSelectionTask.h"
10 #include <AliESDEvent.h>
11 #include <AliHeader.h>
13 #include "AliPhysicsSelection.h"
14 #include "AliAnalysisManager.h"
15 #include "AliInputEventHandler.h"
17 //#include "AliBackgroundSelection.h"
19 ClassImp(AliPhysicsSelectionTask)
21 AliPhysicsSelectionTask::AliPhysicsSelectionTask() :
22 AliAnalysisTaskSE("AliPhysicsSelectionTask"),
28 // Default event handler
32 AliPhysicsSelectionTask::AliPhysicsSelectionTask(const char* opt) :
33 AliAnalysisTaskSE("AliPhysicsSelectionTask"),
36 fPhysicsSelection(new AliPhysicsSelection())
39 // Constructor. Initialization of pointers
41 AliInputEventHandler* handler = dynamic_cast<AliInputEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
43 handler->SetEventSelection(fPhysicsSelection);
44 AliInfo("Physics Event Selection enabled.");
46 AliError("No input event handler connected to analysis manager. No Physics Event Selection.");
48 // Define input and output slots here
49 DefineOutput(1, TList::Class());
50 fBranchNames = "ESD:AliESDRun.,AliESDHeader.,AliMultiplicity.,AliESDFMD.,AliESDVZERO.,AliESDZDC.,SPDVertex.,PrimaryVertex.";
52 AliLog::SetClassDebugLevel("AliPhysicsSelectionTask", AliLog::kWarning);
55 AliPhysicsSelectionTask::~AliPhysicsSelectionTask()
61 // histograms are in the output list and deleted when the output
62 // list is deleted by the TSelector dtor
64 if (fOutput && !AliAnalysisManager::GetAnalysisManager()->IsProofMode()) {
70 void AliPhysicsSelectionTask::UserCreateOutputObjects()
72 // create result objects and add to output list
74 Printf("AliPhysicsSelectionTask::CreateOutputObjects");
79 if (!fPhysicsSelection)
80 fPhysicsSelection = new AliPhysicsSelection;
82 fOutput->Add(fPhysicsSelection);
83 // All tasks must post data once for all outputs (AG)
87 void AliPhysicsSelectionTask::UserExec(Option_t*)
91 // AliPhysicsSelection::IsCollisionCandidate is called from the event handler
92 // post the data here anyway!
96 void AliPhysicsSelectionTask::FinishTaskOutput()
98 // This gets called at the end of the processing on the worker. It allows dumping
99 // statistics printed by the physics selection object to the statistics message
100 // handled by the analysis manager.
101 if (fPhysicsSelection) fPhysicsSelection->Print("STAT");
104 void AliPhysicsSelectionTask::Terminate(Option_t *)
106 // The Terminate() function is the last function to be called during
107 // a query. It always runs on the client, it can be used to present
108 // the results graphically or save the results to file.
110 fOutput = dynamic_cast<TList*> (GetOutputData(1));
112 Printf("ERROR: fOutput not available");
116 fPhysicsSelection = dynamic_cast<AliPhysicsSelection*> (fOutput->FindObject("AliPhysicsSelection"));
119 TFile* fout = new TFile("event_stat.root", "RECREATE");
121 if (fPhysicsSelection)
123 fPhysicsSelection->Print();
124 fPhysicsSelection->SaveHistograms();
130 Printf("Writting result to event_stat.root");