[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisEventCuts.cxx

/**************************************************************************
 * Author: Panos Christakoglou.                                           *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

//-----------------------------------------------------------------
//           AliAnalysisEventCuts class
//   This is the class to deal with the event and track level cuts
//   Origin: Panos Christakoglou, UOA-CERN, Panos.Christakoglou@cern.ch
//-----------------------------------------------------------------


//ROOT
#include <TPaveText.h>
#include <TText.h>
#include <TLine.h>
#include <TCanvas.h>

#include "AliLog.h"
#include "AliESD.h"

#include "AliAnalysisEventCuts.h"

ClassImp(AliAnalysisEventCuts)

//----------------------------------------//
AliAnalysisEventCuts::AliAnalysisEventCuts()
{
  //Default constructor which calls the Reset method.
  Reset();
}

//----------------------------------------//
AliAnalysisEventCuts::~AliAnalysisEventCuts()
{
  //Defaut destructor.
}

//----------------------------------------//
void AliAnalysisEventCuts::Reset()
{
  fVerboseOff = kFALSE;

  //Sets dummy values to every private member.
  fVxMin = -1000.0;
  fVxMax = 1000.0; 
  fVyMin = -1000.0;
  fVyMax = 1000.0;  
  fVzMin = -1000.0;
  fVzMax = 1000.0;
  fMultMin = 0;
  fMultMax = 100000;
  fVzFlagType = "default";

  fMult = 0;  
  fVx = 0;  
  fVy = 0; 
  fVz = 0; 
  fVzFlag = 0; 
  fTotalEvents = 0; 
  fAcceptedEvents = 0; 

  fFlagMult = 0;  
  fFlagVx = 0;  
  fFlagVy = 0; 
  fFlagVz = 0; 
  fFlagVzType = 0; 
}

//----------------------------------------//
void AliAnalysisEventCuts::SetPrimaryVertexXRange(Float_t r1, Float_t r2)
{
  //Sets the primary vertex x range.
  fVxMin = r1;
  fVxMax = r2; 
  fFlagVx = 1;  
}

//----------------------------------------//
void AliAnalysisEventCuts::SetPrimaryVertexYRange(Float_t r1, Float_t r2)
{
  //Sets the primary vertex y range.
  fVyMin = r1;
  fVyMax = r2; 
  fFlagVy = 1;
}

//----------------------------------------//
void AliAnalysisEventCuts::SetPrimaryVertexZRange(Float_t r1, Float_t r2)
{
  //Sets the primary vertex z range.
  fVzMin = r1;
  fVzMax = r2; 
  fFlagVz = 1;
}

//----------------------------------------//
void AliAnalysisEventCuts::SetMultiplicityRange(Int_t n1, Int_t n2)
{
  //Sets the multiplicity range.
  fMultMin = n1;
  fMultMax = n2; 
  fFlagMult = 1;
}


//----------------------------------------//
Bool_t AliAnalysisEventCuts::IsAccepted(AliESD *esd)
{
  //Returns true if the events is accepted otherwise false.
  fTotalEvents++;
  if((esd->GetNumberOfTracks() < fMultMin) || (esd->GetNumberOfTracks() > fMultMax)) {
    fMult++;
    if(!fVerboseOff)
      AliInfo(Form("Event rejected due to multiplicity cut"));
    return kFALSE;
  }
  const AliESDVertex *esdvertex = esd->GetVertex();
  TString vertexname = esdvertex->GetName();
  if((esdvertex->GetXv() < fVxMin) || (esdvertex->GetXv() > fVxMax)) {
    fVx++;
    if(!fVerboseOff)
      AliInfo(Form("Event rejected due to Vx cut"));
    return kFALSE;
  }
  if((esdvertex->GetYv() < fVyMin) || (esdvertex->GetYv() > fVyMax)) {
    fVy++;
    if(!fVerboseOff)
      AliInfo(Form("Event rejected due to Vy cut"));
    return kFALSE;
  }
 if((esdvertex->GetZv() < fVzMin) || (esdvertex->GetZv() > fVzMax)) {
   fVz++;
   if(!fVerboseOff)
     AliInfo(Form("Event rejected due to Vz cut"));
   return kFALSE;
 }
 if((fFlagVzType == 1)&&(vertexname == "default")) {
   fVzFlag++;
   if(!fVerboseOff)
     AliInfo(Form("Event rejected due to Vz flag cut"));
   return kFALSE;
 }
 fAcceptedEvents++;

 return kTRUE;
}


//----------------------------------------//
TPaveText *AliAnalysisEventCuts::GetEventCuts()
{
  //Shows a TPaveText with all the event cut stats.
  TCanvas *ccuts = new TCanvas("ccuts","Event cuts",10,10,400,400);
  ccuts->SetFillColor(10);
  ccuts->SetHighLightColor(10);

  TPaveText *pave = new TPaveText(0.01,0.01,0.98,0.98);
  pave->SetFillColor(5);
  Char_t cutName[256];
 
  TLine *l1 = pave->AddLine(0,0.78,1,0.78);
  l1->SetLineWidth(2);
  TLine *l2 = pave->AddLine(0,0.58,1,0.58);
  l2->SetLineWidth(2);
  TLine *l3 = pave->AddLine(0,0.38,1,0.38);
  l3->SetLineWidth(2);
  TLine *l4 = pave->AddLine(0,0.18,1,0.18);
  l4->SetLineWidth(2);
 
  sprintf(cutName,"Total number of events: %d",fTotalEvents);
  TText *t1 = pave->AddText(cutName);
  t1->SetTextColor(4);
  t1->SetTextSize(0.04);
  t1->SetTextAlign(11);
 
  sprintf(cutName,"Total number of accepted events: %d",fAcceptedEvents);
  t1 = pave->AddText(cutName);
  t1->SetTextColor(4);
  t1->SetTextSize(0.04);
  t1->SetTextAlign(11);
 
  sprintf(cutName,"Multiplicity range: [%d,%d]",fMultMin,fMultMax);
  t1 = pave->AddText(cutName);
  t1->SetTextColor(4);
  t1->SetTextSize(0.04);
  t1->SetTextAlign(11);
  sprintf(cutName,"Events rejected: %d",fMult);
  t1 = pave->AddText(cutName);
  t1->SetTextColor(4);
  t1->SetTextSize(0.04);
  t1->SetTextAlign(11);
 
  sprintf(cutName,"Vx range: [%f,%f]",fVxMin,fVxMax);
  t1 = pave->AddText(cutName);
  t1->SetTextColor(4);
  t1->SetTextSize(0.04);
  t1->SetTextAlign(11);
  sprintf(cutName,"Events rejected: %d",fVx);
  t1 = pave->AddText(cutName);
  t1->SetTextColor(4);
  t1->SetTextSize(0.04);
  t1->SetTextAlign(11);
 
  sprintf(cutName,"Vy range: [%f,%f]",fVyMin,fVyMax);
  t1 = pave->AddText(cutName);
  t1->SetTextColor(4);
  t1->SetTextSize(0.04);
  t1->SetTextAlign(11);
  sprintf(cutName,"Events rejected: %d",fVy);
  t1 = pave->AddText(cutName);
  t1->SetTextColor(4);
  t1->SetTextSize(0.04);
  t1->SetTextAlign(11);
 
  sprintf(cutName,"Vz range: [%f,%f]",fVzMin,fVzMax);
  t1 = pave->AddText(cutName);
  t1->SetTextColor(4);
  t1->SetTextSize(0.04);
  t1->SetTextAlign(11);
  sprintf(cutName,"Events rejected: %d",fVz);
  t1 = pave->AddText(cutName);
  t1->SetTextColor(4);
  t1->SetTextSize(0.04);
  t1->SetTextAlign(11);
 
  return pave;
}

//----------------------------------------//
void AliAnalysisEventCuts::GetEventStats()
{
  //Returns the total event stats.
  //fTotalEvents is the total number of events.
  //fAcceptedEvents is the total number of accepted events.  
  AliInfo(Form("Total number of events: %d",fTotalEvents));
  AliInfo(Form("Total number of accepted events: %d",fAcceptedEvents)); 
}

//----------------------------------------//
void AliAnalysisEventCuts::GetMultStats()
{
  //Gets the multiplicity statistics.
  //Prints the percentage of events rejected due to this cut. 
  AliInfo(Form("Multiplicity range: [%d,%d]",fMultMin,fMultMax));
  AliInfo(Form("Events rejected: %d",fMult));
}

//----------------------------------------//
void AliAnalysisEventCuts::GetVxStats()
{
  //Gets the Vx statistics.
  //Prints the percentage of events rejected due to this cut. 
  AliInfo(Form("Vx range: [%f,%f]",fVxMin,fVxMax));
  AliInfo(Form("Events rejected: %d",fVx));
}

//----------------------------------------//
void AliAnalysisEventCuts::GetVyStats()
{
  //Gets the Vy statistics.
  //Prints the percentage of events rejected due to this cut. 
  AliInfo(Form("Vy range: [%f,%f]",fVyMin,fVyMax));
  AliInfo(Form("Events rejected: %d",fVy));
}

//----------------------------------------//
void AliAnalysisEventCuts::GetVzStats()
{
  //Gets the Vz statistics.
  //Prints the percentage of events rejected due to this cut. 
  AliInfo(Form("Vz range: [%f,%f]",fVzMin,fVzMax));
  AliInfo(Form("Events rejected: %d",fVz));
  AliInfo(Form("Events rejected (Vz flag): %d",fVzFlag));
}

//----------------------------------------//
void AliAnalysisEventCuts::PrintEventCuts()
{
  //Prints the event stats.
  //GetEventCuts()->Draw();  

  AliInfo(Form("**************EVENT CUTS**************"));
  GetEventStats();
  if(fFlagMult) GetMultStats();
  if(fFlagVx) GetVxStats();
  if(fFlagVy) GetVyStats();
  if((fFlagVz)||(fFlagVzType)) GetVzStats();
  AliInfo(Form("**************************************"));
}
Commit	Line	Data
cd286c84	1	/**************************************************************************
	2	* Author: Panos Christakoglou. *
	3	* Contributors are mentioned in the code where appropriate. *
	4	* *
	5	* Permission to use, copy, modify and distribute this software and its *
	6	* documentation strictly for non-commercial purposes is hereby granted *
	7	* without fee, provided that the above copyright notice appears in all *
	8	* copies and that both the copyright notice and this permission notice *
	9	* appear in the supporting documentation. The authors make no claims *
	10	* about the suitability of this software for any purpose. It is *
	11	* provided "as is" without express or implied warranty. *
	12	**************************************************************************/
	13
	14	/* $Id$ */
	15
	16	//-----------------------------------------------------------------
	17	// AliAnalysisEventCuts class
	18	// This is the class to deal with the event and track level cuts
	19	// Origin: Panos Christakoglou, UOA-CERN, Panos.Christakoglou@cern.ch
	20	//-----------------------------------------------------------------
	21
	22
	23
	24	//ROOT
cd286c84	25	#include <TPaveText.h>
	26	#include <TText.h>
	27	#include <TLine.h>
	28	#include <TCanvas.h>
	29
cd286c84	30	#include "AliLog.h"
	31	#include "AliESD.h"
	32
	33	#include "AliAnalysisEventCuts.h"
	34
	35	ClassImp(AliAnalysisEventCuts)
	36
	37	//----------------------------------------//
	38	AliAnalysisEventCuts::AliAnalysisEventCuts()
	39	{
7fe612db	40	//Default constructor which calls the Reset method.
cd286c84	41	Reset();
	42	}
	43
	44	//----------------------------------------//
	45	AliAnalysisEventCuts::~AliAnalysisEventCuts()
	46	{
7fe612db	47	//Defaut destructor.
cd286c84	48	}
	49
	50	//----------------------------------------//
	51	void AliAnalysisEventCuts::Reset()
	52	{
bb2f4e93	53	fVerboseOff = kFALSE;
bb2f4e93	54
7fe612db	55	//Sets dummy values to every private member.
cd286c84	56	fVxMin = -1000.0;
	57	fVxMax = 1000.0;
	58	fVyMin = -1000.0;
	59	fVyMax = 1000.0;
	60	fVzMin = -1000.0;
	61	fVzMax = 1000.0;
	62	fMultMin = 0;
	63	fMultMax = 100000;
bb2f4e93	64	fVzFlagType = "default";
bb2f4e93	65
cd286c84	66	fMult = 0;
	67	fVx = 0;
	68	fVy = 0;
	69	fVz = 0;
bb2f4e93	70	fVzFlag = 0;
cd286c84	71	fTotalEvents = 0;
	72	fAcceptedEvents = 0;
	73
	74	fFlagMult = 0;
	75	fFlagVx = 0;
	76	fFlagVy = 0;
	77	fFlagVz = 0;
bb2f4e93	78	fFlagVzType = 0;
cd286c84	79	}
	80
	81	//----------------------------------------//
	82	void AliAnalysisEventCuts::SetPrimaryVertexXRange(Float_t r1, Float_t r2)
	83	{
7fe612db	84	//Sets the primary vertex x range.
cd286c84	85	fVxMin = r1;
	86	fVxMax = r2;
	87	fFlagVx = 1;
	88	}
	89
	90	//----------------------------------------//
	91	void AliAnalysisEventCuts::SetPrimaryVertexYRange(Float_t r1, Float_t r2)
	92	{
7fe612db	93	//Sets the primary vertex y range.
cd286c84	94	fVyMin = r1;
	95	fVyMax = r2;
	96	fFlagVy = 1;
	97	}
	98
	99	//----------------------------------------//
	100	void AliAnalysisEventCuts::SetPrimaryVertexZRange(Float_t r1, Float_t r2)
	101	{
7fe612db	102	//Sets the primary vertex z range.
cd286c84	103	fVzMin = r1;
	104	fVzMax = r2;
	105	fFlagVz = 1;
	106	}
	107
	108	//----------------------------------------//
	109	void AliAnalysisEventCuts::SetMultiplicityRange(Int_t n1, Int_t n2)
	110	{
7fe612db	111	//Sets the multiplicity range.
cd286c84	112	fMultMin = n1;
	113	fMultMax = n2;
	114	fFlagMult = 1;
	115	}
	116
	117
	118	//----------------------------------------//
	119	Bool_t AliAnalysisEventCuts::IsAccepted(AliESD *esd)
	120	{
7fe612db	121	//Returns true if the events is accepted otherwise false.
cd286c84	122	fTotalEvents++;
bb2f4e93	123	if((esd->GetNumberOfTracks() < fMultMin) \|\| (esd->GetNumberOfTracks() > fMultMax)) {
	124	fMult++;
	125	if(!fVerboseOff)
cd286c84	126	AliInfo(Form("Event rejected due to multiplicity cut"));
bb2f4e93	127	return kFALSE;
bb2f4e93	128	}
cd286c84	129	const AliESDVertex *esdvertex = esd->GetVertex();
bb2f4e93	130	TString vertexname = esdvertex->GetName();
	131	if((esdvertex->GetXv() < fVxMin) \|\| (esdvertex->GetXv() > fVxMax)) {
	132	fVx++;
	133	if(!fVerboseOff)
cd286c84	134	AliInfo(Form("Event rejected due to Vx cut"));
bb2f4e93	135	return kFALSE;
	136	}
	137	if((esdvertex->GetYv() < fVyMin) \|\| (esdvertex->GetYv() > fVyMax)) {
	138	fVy++;
	139	if(!fVerboseOff)
cd286c84	140	AliInfo(Form("Event rejected due to Vy cut"));
bb2f4e93	141	return kFALSE;
	142	}
	143	if((esdvertex->GetZv() < fVzMin) \|\| (esdvertex->GetZv() > fVzMax)) {
	144	fVz++;
	145	if(!fVerboseOff)
	146	AliInfo(Form("Event rejected due to Vz cut"));
	147	return kFALSE;
	148	}
	149	if((fFlagVzType == 1)&&(vertexname == "default")) {
	150	fVzFlag++;
	151	if(!fVerboseOff)
	152	AliInfo(Form("Event rejected due to Vz flag cut"));
	153	return kFALSE;
	154	}
cd286c84	155	fAcceptedEvents++;
	156
	157	return kTRUE;
	158	}
	159
	160
	161	//----------------------------------------//
	162	TPaveText *AliAnalysisEventCuts::GetEventCuts()
	163	{
7fe612db	164	//Shows a TPaveText with all the event cut stats.
cd286c84	165	TCanvas *ccuts = new TCanvas("ccuts","Event cuts",10,10,400,400);
	166	ccuts->SetFillColor(10);
	167	ccuts->SetHighLightColor(10);
	168
	169	TPaveText *pave = new TPaveText(0.01,0.01,0.98,0.98);
	170	pave->SetFillColor(5);
7fe612db	171	Char_t cutName[256];
cd286c84	172
	173	TLine *l1 = pave->AddLine(0,0.78,1,0.78);
	174	l1->SetLineWidth(2);
	175	TLine *l2 = pave->AddLine(0,0.58,1,0.58);
	176	l2->SetLineWidth(2);
	177	TLine *l3 = pave->AddLine(0,0.38,1,0.38);
	178	l3->SetLineWidth(2);
	179	TLine *l4 = pave->AddLine(0,0.18,1,0.18);
	180	l4->SetLineWidth(2);
	181
7fe612db	182	sprintf(cutName,"Total number of events: %d",fTotalEvents);
7fe612db	183	TText *t1 = pave->AddText(cutName);
cd286c84	184	t1->SetTextColor(4);
	185	t1->SetTextSize(0.04);
	186	t1->SetTextAlign(11);
	187
7fe612db	188	sprintf(cutName,"Total number of accepted events: %d",fAcceptedEvents);
7fe612db	189	t1 = pave->AddText(cutName);
cd286c84	190	t1->SetTextColor(4);
	191	t1->SetTextSize(0.04);
	192	t1->SetTextAlign(11);
	193
7fe612db	194	sprintf(cutName,"Multiplicity range: [%d,%d]",fMultMin,fMultMax);
7fe612db	195	t1 = pave->AddText(cutName);
cd286c84	196	t1->SetTextColor(4);
	197	t1->SetTextSize(0.04);
	198	t1->SetTextAlign(11);
7fe612db	199	sprintf(cutName,"Events rejected: %d",fMult);
7fe612db	200	t1 = pave->AddText(cutName);
cd286c84	201	t1->SetTextColor(4);
	202	t1->SetTextSize(0.04);
	203	t1->SetTextAlign(11);
	204
7fe612db	205	sprintf(cutName,"Vx range: [%f,%f]",fVxMin,fVxMax);
7fe612db	206	t1 = pave->AddText(cutName);
cd286c84	207	t1->SetTextColor(4);
	208	t1->SetTextSize(0.04);
	209	t1->SetTextAlign(11);
7fe612db	210	sprintf(cutName,"Events rejected: %d",fVx);
7fe612db	211	t1 = pave->AddText(cutName);
cd286c84	212	t1->SetTextColor(4);
	213	t1->SetTextSize(0.04);
	214	t1->SetTextAlign(11);
	215
7fe612db	216	sprintf(cutName,"Vy range: [%f,%f]",fVyMin,fVyMax);
7fe612db	217	t1 = pave->AddText(cutName);
cd286c84	218	t1->SetTextColor(4);
	219	t1->SetTextSize(0.04);
	220	t1->SetTextAlign(11);
7fe612db	221	sprintf(cutName,"Events rejected: %d",fVy);
7fe612db	222	t1 = pave->AddText(cutName);
cd286c84	223	t1->SetTextColor(4);
	224	t1->SetTextSize(0.04);
	225	t1->SetTextAlign(11);
	226
7fe612db	227	sprintf(cutName,"Vz range: [%f,%f]",fVzMin,fVzMax);
7fe612db	228	t1 = pave->AddText(cutName);
cd286c84	229	t1->SetTextColor(4);
	230	t1->SetTextSize(0.04);
	231	t1->SetTextAlign(11);
7fe612db	232	sprintf(cutName,"Events rejected: %d",fVz);
7fe612db	233	t1 = pave->AddText(cutName);
cd286c84	234	t1->SetTextColor(4);
	235	t1->SetTextSize(0.04);
	236	t1->SetTextAlign(11);
	237
	238	return pave;
	239	}
	240
	241	//----------------------------------------//
	242	void AliAnalysisEventCuts::GetEventStats()
	243	{
7fe612db	244	//Returns the total event stats.
	245	//fTotalEvents is the total number of events.
	246	//fAcceptedEvents is the total number of accepted events.
cd286c84	247	AliInfo(Form("Total number of events: %d",fTotalEvents));
	248	AliInfo(Form("Total number of accepted events: %d",fAcceptedEvents));
	249	}
	250
	251	//----------------------------------------//
	252	void AliAnalysisEventCuts::GetMultStats()
	253	{
7fe612db	254	//Gets the multiplicity statistics.
7fe612db	255	//Prints the percentage of events rejected due to this cut.
cd286c84	256	AliInfo(Form("Multiplicity range: [%d,%d]",fMultMin,fMultMax));
	257	AliInfo(Form("Events rejected: %d",fMult));
	258	}
	259
	260	//----------------------------------------//
	261	void AliAnalysisEventCuts::GetVxStats()
	262	{
7fe612db	263	//Gets the Vx statistics.
7fe612db	264	//Prints the percentage of events rejected due to this cut.
cd286c84	265	AliInfo(Form("Vx range: [%f,%f]",fVxMin,fVxMax));
	266	AliInfo(Form("Events rejected: %d",fVx));
	267	}
	268
	269	//----------------------------------------//
	270	void AliAnalysisEventCuts::GetVyStats()
	271	{
7fe612db	272	//Gets the Vy statistics.
7fe612db	273	//Prints the percentage of events rejected due to this cut.
cd286c84	274	AliInfo(Form("Vy range: [%f,%f]",fVyMin,fVyMax));
	275	AliInfo(Form("Events rejected: %d",fVy));
	276	}
	277
	278	//----------------------------------------//
	279	void AliAnalysisEventCuts::GetVzStats()
	280	{
7fe612db	281	//Gets the Vz statistics.
7fe612db	282	//Prints the percentage of events rejected due to this cut.
cd286c84	283	AliInfo(Form("Vz range: [%f,%f]",fVzMin,fVzMax));
cd286c84	284	AliInfo(Form("Events rejected: %d",fVz));
bb2f4e93	285	AliInfo(Form("Events rejected (Vz flag): %d",fVzFlag));
cd286c84	286	}
	287
	288	//----------------------------------------//
	289	void AliAnalysisEventCuts::PrintEventCuts()
	290	{
7fe612db	291	//Prints the event stats.
cd286c84	292	//GetEventCuts()->Draw();
	293
	294	AliInfo(Form("************EVENT CUTS************"));
	295	GetEventStats();
bb2f4e93	296	if(fFlagMult) GetMultStats();
	297	if(fFlagVx) GetVxStats();
	298	if(fFlagVy) GetVyStats();
	299	if((fFlagVz)\|\|(fFlagVzType)) GetVzStats();
cd286c84	300	AliInfo(Form("**************************************"));
	301	}
	302
	303
	304