[u/mrichter/AliRoot.git] / PWG2 / FEMTOSCOPY / AliFemto / AliFemtoCutMonitorEventVertex.cxx

////////////////////////////////////////////////////////////////////////////////
//                                                                            //
// AliFemtoCutMonitorEventVertex - the cut monitor for events to study        //
// the distribution and error of the primary vertex                           //
//                                                                            //
////////////////////////////////////////////////////////////////////////////////
#include "AliFemtoCutMonitorEventVertex.h"
#include "AliFemtoModelHiddenInfo.h"
#include "AliFemtoEvent.h"
#include <TH1D.h>
#include <TH2D.h>
#include <TList.h>
#include <TMath.h>

AliFemtoCutMonitorEventVertex::AliFemtoCutMonitorEventVertex():
  AliFemtoCutMonitor(),
  fEvVertRad(0),
  fEvVertXY(0),
  fEvVertSigXY(0),
  fEvVertZ(0),
  fEvVertSigZ(0)
{
  // Default constructor
  fEvVertRad   = new TH1D("EvVertRad",   "Vertex position radial", 200, 0.0, 0.2);
  fEvVertXY    = new TH2D("EvVertXY",    "Vertex position xy plane", 200, -0.2, 0.2, 200, -0.2, 0.2);
  fEvVertSigXY = new TH1D("EvVertSigXY", "Vertex error in xy plane", 200, 0.0, 0.2);
  fEvVertZ     = new TH1D("EvVertZ",     "Vertex position in z", 500, -50.0, 50.0);
  fEvVertSigZ  = new TH1D("EvVertSigZ",  "Vertex error in z", 100, 0.0, 0.2);
}

AliFemtoCutMonitorEventVertex::AliFemtoCutMonitorEventVertex(const char *aName):
  AliFemtoCutMonitor(),
  fEvVertRad(0),
  fEvVertXY(0),
  fEvVertSigXY(0),
  fEvVertZ(0),
  fEvVertSigZ(0)
{
  // Normal constructor
  char name[200];
  snprintf(name, 200, "EvVertRad%s", aName);
  fEvVertRad   = new TH1D(name,   "Vertex position radial", 200, 0.0, 0.2);
  snprintf(name, 200, "EvVertXY%s", aName);
  fEvVertXY    = new TH2D(name,    "Vertex position xy plane", 200, -0.2, 0.2, 200, -0.2, 0.2);
  snprintf(name, 200, "EvVertSigXY%s", aName);
  fEvVertSigXY = new TH1D(name, "Vertex error in xy plane", 200, 0.0, 0.2);
  snprintf(name, 200, "EvVertZ%s", aName);
  fEvVertZ     = new TH1D(name,     "Vertex position in z", 500, -50.0, 50.0);
  snprintf(name, 200, "EvVertSigZ%s", aName);
  fEvVertSigZ  = new TH1D(name,  "Vertex error in z", 100, 0.0, 0.2);
}

AliFemtoCutMonitorEventVertex::AliFemtoCutMonitorEventVertex(const AliFemtoCutMonitorEventVertex &aCut):
  AliFemtoCutMonitor(),
  fEvVertRad(0),
  fEvVertXY(0),
  fEvVertSigXY(0),
  fEvVertZ(0),
  fEvVertSigZ(0)
{
  // copy constructor
  if (fEvVertRad) delete fEvVertRad;
  fEvVertRad = new TH1D(*aCut.fEvVertRad);
  if (fEvVertXY) delete fEvVertXY;
  fEvVertXY = new TH2D(*aCut.fEvVertXY);
  if (fEvVertSigXY) delete fEvVertSigXY;
  fEvVertSigXY = new TH1D(*aCut.fEvVertSigXY);
  if (fEvVertZ) delete fEvVertZ;
  fEvVertZ = new TH1D(*aCut.fEvVertZ);
  if (fEvVertSigZ) delete fEvVertSigZ;
  fEvVertSigZ = new TH1D(*aCut.fEvVertSigZ);
}

AliFemtoCutMonitorEventVertex::~AliFemtoCutMonitorEventVertex()
{
  // Destructor
  delete fEvVertRad;
  delete fEvVertXY;
  delete fEvVertSigXY;
  delete fEvVertZ;
  delete fEvVertSigZ;
}

AliFemtoCutMonitorEventVertex& AliFemtoCutMonitorEventVertex::operator=(const AliFemtoCutMonitorEventVertex& aCut)
{
  // assignment operator
  if (this == &aCut) 
    return *this;

  if (fEvVertRad) delete fEvVertRad;
  fEvVertRad = new TH1D(*aCut.fEvVertRad);
  if (fEvVertXY) delete fEvVertXY;
  fEvVertXY = new TH2D(*aCut.fEvVertXY);
  if (fEvVertSigXY) delete fEvVertSigXY;
  fEvVertSigXY = new TH1D(*aCut.fEvVertSigXY);
  if (fEvVertZ) delete fEvVertZ;
  fEvVertZ = new TH1D(*aCut.fEvVertZ);
  if (fEvVertSigZ) delete fEvVertSigZ;
  fEvVertSigZ = new TH1D(*aCut.fEvVertSigZ);
  
  return *this;
}

AliFemtoString AliFemtoCutMonitorEventVertex::Report(){ 
  // Prepare report from the execution
  string stemp = "*** AliFemtoCutMonitorEventVertex report"; 
  AliFemtoString returnThis = stemp;
  return returnThis; 
}

void AliFemtoCutMonitorEventVertex::Fill(const AliFemtoEvent* aEvent)
{
  // Fill in the monitor histograms with the values from the current track
  fEvVertRad->Fill(TMath::Hypot(aEvent->PrimVertPos().x(), aEvent->PrimVertPos().y()));
  fEvVertXY->Fill(aEvent->PrimVertPos().x(), aEvent->PrimVertPos().y());
  fEvVertSigXY->Fill(TMath::Sqrt(aEvent->PrimVertCov()[0]+aEvent->PrimVertCov()[2]));
  fEvVertZ->Fill(aEvent->PrimVertPos().z());
  fEvVertSigZ->Fill(TMath::Sqrt(aEvent->PrimVertCov()[5]));
}

void AliFemtoCutMonitorEventVertex::Write()
{
  // Write out the relevant histograms
  fEvVertRad->Write();
  fEvVertXY->Write();
  fEvVertSigXY->Write();
  fEvVertZ->Write();
  fEvVertSigZ->Write();
}

TList *AliFemtoCutMonitorEventVertex::GetOutputList()
{
  TList *tOutputList = new TList();
  tOutputList->Add(fEvVertRad);
  tOutputList->Add(fEvVertXY);
  tOutputList->Add(fEvVertSigXY);
  tOutputList->Add(fEvVertZ);
  tOutputList->Add(fEvVertSigZ);

  return tOutputList;
}
Commit	Line	Data
dc37df7f	1	////////////////////////////////////////////////////////////////////////////////
	2	// //
	3	// AliFemtoCutMonitorEventVertex - the cut monitor for events to study //
	4	// the distribution and error of the primary vertex //
	5	// //
	6	////////////////////////////////////////////////////////////////////////////////
	7	#include "AliFemtoCutMonitorEventVertex.h"
	8	#include "AliFemtoModelHiddenInfo.h"
	9	#include "AliFemtoEvent.h"
	10	#include <TH1D.h>
	11	#include <TH2D.h>
	12	#include <TList.h>
	13	#include <TMath.h>
	14
	15	AliFemtoCutMonitorEventVertex::AliFemtoCutMonitorEventVertex():
	16	AliFemtoCutMonitor(),
	17	fEvVertRad(0),
	18	fEvVertXY(0),
	19	fEvVertSigXY(0),
	20	fEvVertZ(0),
	21	fEvVertSigZ(0)
	22	{
	23	// Default constructor
	24	fEvVertRad = new TH1D("EvVertRad", "Vertex position radial", 200, 0.0, 0.2);
	25	fEvVertXY = new TH2D("EvVertXY", "Vertex position xy plane", 200, -0.2, 0.2, 200, -0.2, 0.2);
	26	fEvVertSigXY = new TH1D("EvVertSigXY", "Vertex error in xy plane", 200, 0.0, 0.2);
	27	fEvVertZ = new TH1D("EvVertZ", "Vertex position in z", 500, -50.0, 50.0);
	28	fEvVertSigZ = new TH1D("EvVertSigZ", "Vertex error in z", 100, 0.0, 0.2);
	29	}
	30
	31	AliFemtoCutMonitorEventVertex::AliFemtoCutMonitorEventVertex(const char *aName):
	32	AliFemtoCutMonitor(),
	33	fEvVertRad(0),
	34	fEvVertXY(0),
	35	fEvVertSigXY(0),
	36	fEvVertZ(0),
	37	fEvVertSigZ(0)
	38	{
	39	// Normal constructor
	40	char name[200];
	41	snprintf(name, 200, "EvVertRad%s", aName);
	42	fEvVertRad = new TH1D(name, "Vertex position radial", 200, 0.0, 0.2);
	43	snprintf(name, 200, "EvVertXY%s", aName);
	44	fEvVertXY = new TH2D(name, "Vertex position xy plane", 200, -0.2, 0.2, 200, -0.2, 0.2);
	45	snprintf(name, 200, "EvVertSigXY%s", aName);
	46	fEvVertSigXY = new TH1D(name, "Vertex error in xy plane", 200, 0.0, 0.2);
	47	snprintf(name, 200, "EvVertZ%s", aName);
	48	fEvVertZ = new TH1D(name, "Vertex position in z", 500, -50.0, 50.0);
	49	snprintf(name, 200, "EvVertSigZ%s", aName);
	50	fEvVertSigZ = new TH1D(name, "Vertex error in z", 100, 0.0, 0.2);
	51	}
	52
	53	AliFemtoCutMonitorEventVertex::AliFemtoCutMonitorEventVertex(const AliFemtoCutMonitorEventVertex &aCut):
	54	AliFemtoCutMonitor(),
	55	fEvVertRad(0),
	56	fEvVertXY(0),
	57	fEvVertSigXY(0),
	58	fEvVertZ(0),
	59	fEvVertSigZ(0)
	60	{
	61	// copy constructor
	62	if (fEvVertRad) delete fEvVertRad;
	63	fEvVertRad = new TH1D(*aCut.fEvVertRad);
	64	if (fEvVertXY) delete fEvVertXY;
65	fEvVertXY = new TH2D(*aCut.fEvVertXY);
66	if (fEvVertSigXY) delete fEvVertSigXY;
67	fEvVertSigXY = new TH1D(*aCut.fEvVertSigXY);
68	if (fEvVertZ) delete fEvVertZ;
69	fEvVertZ = new TH1D(*aCut.fEvVertZ);
70	if (fEvVertSigZ) delete fEvVertSigZ;
71	fEvVertSigZ = new TH1D(*aCut.fEvVertSigZ);
72	}
73
74	AliFemtoCutMonitorEventVertex::~AliFemtoCutMonitorEventVertex()
75	{
76	// Destructor
77	delete fEvVertRad;
78	delete fEvVertXY;
79	delete fEvVertSigXY;
80	delete fEvVertZ;
81	delete fEvVertSigZ;
82	}
83
84	AliFemtoCutMonitorEventVertex& AliFemtoCutMonitorEventVertex::operator=(const AliFemtoCutMonitorEventVertex& aCut)
85	{
86	// assignment operator
87	if (this == &aCut)
88	return *this;
89
90	if (fEvVertRad) delete fEvVertRad;
91	fEvVertRad = new TH1D(*aCut.fEvVertRad);
92	if (fEvVertXY) delete fEvVertXY;
93	fEvVertXY = new TH2D(*aCut.fEvVertXY);
94	if (fEvVertSigXY) delete fEvVertSigXY;
95	fEvVertSigXY = new TH1D(*aCut.fEvVertSigXY);
96	if (fEvVertZ) delete fEvVertZ;
97	fEvVertZ = new TH1D(*aCut.fEvVertZ);
98	if (fEvVertSigZ) delete fEvVertSigZ;
99	fEvVertSigZ = new TH1D(*aCut.fEvVertSigZ);
100
101	return *this;
102	}
103
104	AliFemtoString AliFemtoCutMonitorEventVertex::Report(){
105	// Prepare report from the execution
106	string stemp = "*** AliFemtoCutMonitorEventVertex report";
107	AliFemtoString returnThis = stemp;
108	return returnThis;
109	}
110
111	void AliFemtoCutMonitorEventVertex::Fill(const AliFemtoEvent* aEvent)
112	{
113	// Fill in the monitor histograms with the values from the current track
114	fEvVertRad->Fill(TMath::Hypot(aEvent->PrimVertPos().x(), aEvent->PrimVertPos().y()));
115	fEvVertXY->Fill(aEvent->PrimVertPos().x(), aEvent->PrimVertPos().y());
116	fEvVertSigXY->Fill(TMath::Sqrt(aEvent->PrimVertCov()[0]+aEvent->PrimVertCov()[2]));
117	fEvVertZ->Fill(aEvent->PrimVertPos().z());
118	fEvVertSigZ->Fill(TMath::Sqrt(aEvent->PrimVertCov()[5]));
119	}
120
121	void AliFemtoCutMonitorEventVertex::Write()
122	{
123	// Write out the relevant histograms
124	fEvVertRad->Write();
125	fEvVertXY->Write();
126	fEvVertSigXY->Write();
127	fEvVertZ->Write();
128	fEvVertSigZ->Write();
129	}
130
131	TList *AliFemtoCutMonitorEventVertex::GetOutputList()
132	{
133	TList *tOutputList = new TList();
134	tOutputList->Add(fEvVertRad);
135	tOutputList->Add(fEvVertXY);
136	tOutputList->Add(fEvVertSigXY);
137	tOutputList->Add(fEvVertZ);
138	tOutputList->Add(fEvVertSigZ);
139
140	return tOutputList;
141	}