[u/mrichter/AliRoot.git] / PWG0 / eventStats / AliEventStatsTask.cxx

/* $Id: AliEventStatsTask.cxx 35782 2009-10-22 11:54:31Z jgrosseo $ */

#include "AliEventStatsTask.h"

#include <TCanvas.h>
#include <TFile.h>
#include <TChain.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TH3F.h>

#include <AliLog.h>
#include <AliESDEvent.h>
#include <AliHeader.h>
#include <AliAnalysisManager.h>
#include <AliESDInputHandler.h>
#include <AliESDHeader.h>
#include <AliTriggerAnalysis.h>

#include "AliPhysicsSelection.h"
#include "AliBackgroundSelection.h"

ClassImp(AliEventStatsTask)

AliEventStatsTask::AliEventStatsTask(const char* opt) :
  AliAnalysisTaskSE("AliEventStatsTask"),
  fOutput(0),
  fOption(opt),
  fPhysicsSelection(0)
{
  //
  // Constructor. Initialization of pointers
  //

  // Define input and output slots here
  DefineOutput(1, TList::Class());
  
  AliLog::SetClassDebugLevel("AliEventStatsTask", AliLog::kWarning);
}

AliEventStatsTask::~AliEventStatsTask()
{
  //
  // Destructor
  //

  // histograms are in the output list and deleted when the output
  // list is deleted by the TSelector dtor

  if (fOutput) {
    delete fOutput;
    fOutput = 0;
  }
}

void AliEventStatsTask::UserCreateOutputObjects()
{
  // create result objects and add to output list

  Printf("AliEventStatsTask::CreateOutputObjects");

  fOutput = new TList;
  fOutput->SetOwner();
  
  fPhysicsSelection = new AliPhysicsSelection;
  AliBackgroundSelection* background = new AliBackgroundSelection("AliBackgroundSelection", "AliBackgroundSelection");
  background->Init();
  //fPhysicsSelection->AddBackgroundIdentification(background);
  //AliLog::SetClassDebugLevel("AliPhysicsSelection", AliLog::kDebug);

  fOutput->Add(fPhysicsSelection);
}

void AliEventStatsTask::UserExec(Option_t*)
{
  // process the event

  // post the data already here
  PostData(1, fOutput);

  AliESDEvent* esd = dynamic_cast<AliESDEvent*> (InputEvent());

  if (!esd)
  {
    AliError("ESD branch not available");
    return;
  }
  
  fPhysicsSelection->IsCollisionCandidate(esd);
}

void AliEventStatsTask::Terminate(Option_t *)
{
  // The Terminate() function is the last function to be called during
  // a query. It always runs on the client, it can be used to present
  // the results graphically or save the results to file.

  fOutput = dynamic_cast<TList*> (GetOutputData(1));
  if (!fOutput)
    Printf("ERROR: fOutput not available");
    
  if (fOutput)
  {
    fPhysicsSelection = dynamic_cast<AliPhysicsSelection*> (fOutput->FindObject("AliPhysicsSelection"));
  }

  TFile* fout = new TFile("event_stat.root", "RECREATE");

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