[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / AliFemto / AliFemtoKTPairCut.cxx

/////////////////////////////////////////////////////////////////////////////
//                                                                         //
// AliFemtoKTPairCut - a pair cut which selects pairs based on their       //
// transverse momentum kT                                                  //
//                                                                         //
/////////////////////////////////////////////////////////////////////////////
/***************************************************************************
 *
 * $Id: AliFemtoKTPairCut.cxx,v 1.1.2.2 2007/11/09 11:20:35 akisiel Exp $
 *
 * Author: Adam Kisiel, Ohio State, kisiel@mps.ohio-state.edu
 ***************************************************************************
 *
 * Description: part of STAR HBT Framework: AliFemtoMaker package
 *   a cut to remove "shared" and "split" pairs
 *
 ***************************************************************************
 *
 *
 **************************************************************************/
#include "AliFemtoKTPairCut.h"
#include <string>
#include <cstdio>
#include <TMath.h>

#ifdef __ROOT__
ClassImp(AliFemtoKTPairCut)
#endif

//__________________
AliFemtoKTPairCut::AliFemtoKTPairCut():
  AliFemtoPairCut(),
  fKTMin(0),
  fKTMax(1.0e6),
  fPhiMin(0),
  fPhiMax(360.0),
  fPtMin(0.0),
  fPtMax(1000.0)
{
  fKTMin = 0;
   fKTMax = 1.0e6;
}
//__________________
AliFemtoKTPairCut::AliFemtoKTPairCut(double lo, double hi) :
  AliFemtoPairCut(),
  fKTMin(lo),
  fKTMax(hi),
  fPhiMin(0),
  fPhiMax(360),
  fPtMin(0.0),
  fPtMax(1000.0)
{
}
//__________________
AliFemtoKTPairCut::AliFemtoKTPairCut(const AliFemtoKTPairCut& c) : 
  AliFemtoPairCut(c),
  fKTMin(0),
  fKTMax(1.0e6),
  fPhiMin(0),
  fPhiMax(360),
  fPtMin(0.0),
  fPtMax(1000.0)
{ 
  fKTMin = c.fKTMin;
  fKTMax = c.fKTMax;
  fPhiMin = c.fPhiMin;
  fPhiMax = c.fPhiMax;
  fPtMin = c.fPtMin;
  fPtMax = c.fPtMax;
}

//__________________
AliFemtoKTPairCut::~AliFemtoKTPairCut(){
  /* no-op */
}
AliFemtoKTPairCut& AliFemtoKTPairCut::operator=(const AliFemtoKTPairCut& c)
{
  if (this != &c) {
    fKTMin = c.fKTMin;
    fKTMax = c.fKTMax;
    fPhiMin = c.fPhiMin;
    fPhiMax = c.fPhiMax;
    fPtMin = c.fPtMin;
    fPtMax = c.fPtMax;
  }

  return *this;
}
//__________________
/*bool AliFemtoKTPairCut::Pass(const AliFemtoPair* pair){
  bool temp = true;
  
  if (pair->KT() < fKTMin)
    temp = false;

  if (pair->KT() > fKTMax)
    temp = false;

  return temp;
}*/
//__________________
AliFemtoString AliFemtoKTPairCut::Report(){
  // Prepare a report from the execution
  string stemp = "AliFemtoKT Pair Cut \n";  char ctemp[100];
  snprintf(ctemp , 100, "Accept pair with kT in range %f , %f",fKTMin,fKTMax);
  snprintf(ctemp , 100, "Accept pair with angle in range %f , %f",fPhiMin,fPhiMax);
  stemp += ctemp;
  AliFemtoString returnThis = stemp;
  return returnThis;}
//__________________

TList *AliFemtoKTPairCut::ListSettings()
{
  // return a list of settings in a writable form
  TList *tListSetttings =  new TList();
  char buf[200];
  snprintf(buf, 200, "AliFemtoKTPairCut.ktmax=%f", fKTMax);
  tListSetttings->AddLast(new TObjString(buf));
  snprintf(buf, 200, "AliFemtoKTPairCut.ktmin=%f", fKTMin);
  tListSetttings->AddLast(new TObjString(buf));
  snprintf(buf, 200, "AliFemtoKTPairCut.phimax=%f", fPhiMax);
  tListSetttings->AddLast(new TObjString(buf));
  snprintf(buf, 200, "AliFemtoKTPairCut.phimin=%f", fPhiMin);
  tListSetttings->AddLast(new TObjString(buf));
  snprintf(buf, 200, "AliFemtoKTPairCut.ptmin=%f", fPtMin);
  tListSetttings->AddLast(new TObjString(buf));
  snprintf(buf, 200, "AliFemtoKTPairCut.ptmax=%f", fPtMax);
  tListSetttings->AddLast(new TObjString(buf));

  return tListSetttings;
}

void AliFemtoKTPairCut::SetKTRange(double ktmin, double ktmax)
{
  fKTMin = ktmin;
  fKTMax = ktmax;
}

void AliFemtoKTPairCut::SetPhiRange(double phimin, double phimax)
{
  fPhiMin = phimin;
  fPhiMax = phimax;
}

void AliFemtoKTPairCut::SetPTMin(double ptmin, double ptmax)
{
  fPtMin = ptmin;
  fPtMax = ptmax;
}

//______________________________________________________
bool AliFemtoKTPairCut::Pass(const AliFemtoPair* pair)
{
  bool temp = true;

//Taking care of the Kt cut
  if (pair->KT() < fKTMin)
    temp = false;

  if (pair->KT() > fKTMax)
    temp = false;

  if (!temp) return temp;

  if ((fPtMin > 0.0) || (fPtMax<1000.0)) {
//     double px1 = pair->Track1()->Track()->P().x();
//     double py1 = pair->Track1()->Track()->P().y();

//     double px2 = pair->Track2()->Track()->P().x();
//     double py2 = pair->Track2()->Track()->P().y();
    
//     double pt1 = TMath::Hypot(px1, py1);
//     double pt2 = TMath::Hypot(px2, py2);
    
//     if ((pt1<fPtMin) || (pt1>fPtMax)) return false;
//     if ((pt2<fPtMin) || (pt2>fPtMax)) return false;
    if ((pair->Track1()->Track()->Pt()<fPtMin) || (pair->Track1()->Track()->Pt()>fPtMax)) return false;
    if ((pair->Track2()->Track()->Pt()<fPtMin) || (pair->Track2()->Track()->Pt()>fPtMax)) return false;
  }

//Taking care of the Phi cut
//   double rpangle = (pair->GetPairAngleEP())*180/TMath::Pi();
  double rpangle = pair->GetPairAngleEP();

  if (rpangle > 180.0) rpangle -= 180.0;
  if (rpangle < 0.0) rpangle += 180.0;
  
  if (fPhiMin < 0) {
    if ((rpangle > fPhiMax) && (rpangle < 180.0+fPhiMin)) 
      temp = false;
  }
  else {
    if ((rpangle < fPhiMin) || (rpangle > fPhiMax))
      temp = false;
  }
  return temp;
}

//_____________________________________
bool AliFemtoKTPairCut::Pass(const AliFemtoPair* pair, double aRPAngle)
{
//The same as above, but it is defined with RP Angle as input in all the Correlatin function classes.

  bool temp = (aRPAngle > 0.);
  aRPAngle = true;
   
  if (!Pass(pair))
	temp = false;

  return temp;
}
Commit	Line	Data
76ce4b5b	1	/////////////////////////////////////////////////////////////////////////////
	2	// //
	3	// AliFemtoKTPairCut - a pair cut which selects pairs based on their //
	4	// transverse momentum kT //
	5	// //
	6	/////////////////////////////////////////////////////////////////////////////
	7	/***************************************************************************
	8	*
	9	* $Id: AliFemtoKTPairCut.cxx,v 1.1.2.2 2007/11/09 11:20:35 akisiel Exp $
	10	*
	11	* Author: Adam Kisiel, Ohio State, kisiel@mps.ohio-state.edu
	12	***************************************************************************
	13	*
	14	* Description: part of STAR HBT Framework: AliFemtoMaker package
	15	* a cut to remove "shared" and "split" pairs
	16	*
	17	***************************************************************************
	18	*
	19	*
	20	**************************************************************************/
	21	#include "AliFemtoKTPairCut.h"
	22	#include <string>
	23	#include <cstdio>
	24	#include <TMath.h>
	25
	26	#ifdef __ROOT__
	27	ClassImp(AliFemtoKTPairCut)
	28	#endif
	29
	30	//__________________
	31	AliFemtoKTPairCut::AliFemtoKTPairCut():
	32	AliFemtoPairCut(),
	33	fKTMin(0),
	34	fKTMax(1.0e6),
	35	fPhiMin(0),
	36	fPhiMax(360.0),
	37	fPtMin(0.0),
	38	fPtMax(1000.0)
	39	{
	40	fKTMin = 0;
	41	fKTMax = 1.0e6;
	42	}
	43	//__________________
	44	AliFemtoKTPairCut::AliFemtoKTPairCut(double lo, double hi) :
	45	AliFemtoPairCut(),
	46	fKTMin(lo),
	47	fKTMax(hi),
	48	fPhiMin(0),
	49	fPhiMax(360),
	50	fPtMin(0.0),
	51	fPtMax(1000.0)
	52	{
	53	}
	54	//__________________
	55	AliFemtoKTPairCut::AliFemtoKTPairCut(const AliFemtoKTPairCut& c) :
	56	AliFemtoPairCut(c),
	57	fKTMin(0),
	58	fKTMax(1.0e6),
	59	fPhiMin(0),
	60	fPhiMax(360),
	61	fPtMin(0.0),
	62	fPtMax(1000.0)
	63	{
	64	fKTMin = c.fKTMin;
65	fKTMax = c.fKTMax;
66	fPhiMin = c.fPhiMin;
67	fPhiMax = c.fPhiMax;
68	fPtMin = c.fPtMin;
69	fPtMax = c.fPtMax;
70	}
71
72	//__________________
73	AliFemtoKTPairCut::~AliFemtoKTPairCut(){
74	/* no-op */
75	}
76	AliFemtoKTPairCut& AliFemtoKTPairCut::operator=(const AliFemtoKTPairCut& c)
77	{
78	if (this != &c) {
79	fKTMin = c.fKTMin;
80	fKTMax = c.fKTMax;
81	fPhiMin = c.fPhiMin;
82	fPhiMax = c.fPhiMax;
83	fPtMin = c.fPtMin;
84	fPtMax = c.fPtMax;
85	}
86
87	return *this;
88	}
89	//__________________
90	/bool AliFemtoKTPairCut::Pass(const AliFemtoPair pair){
91	bool temp = true;
92
93	if (pair->KT() < fKTMin)
94	temp = false;
95
96	if (pair->KT() > fKTMax)
97	temp = false;
98
99	return temp;
100	}*/
101	//__________________
102	AliFemtoString AliFemtoKTPairCut::Report(){
103	// Prepare a report from the execution
104	string stemp = "AliFemtoKT Pair Cut \n"; char ctemp[100];
105	snprintf(ctemp , 100, "Accept pair with kT in range %f , %f",fKTMin,fKTMax);
106	snprintf(ctemp , 100, "Accept pair with angle in range %f , %f",fPhiMin,fPhiMax);
107	stemp += ctemp;
108	AliFemtoString returnThis = stemp;
109	return returnThis;}
110	//__________________
111
112	TList *AliFemtoKTPairCut::ListSettings()
113	{
114	// return a list of settings in a writable form
115	TList *tListSetttings = new TList();
116	char buf[200];
117	snprintf(buf, 200, "AliFemtoKTPairCut.ktmax=%f", fKTMax);
118	tListSetttings->AddLast(new TObjString(buf));
119	snprintf(buf, 200, "AliFemtoKTPairCut.ktmin=%f", fKTMin);
120	tListSetttings->AddLast(new TObjString(buf));
121	snprintf(buf, 200, "AliFemtoKTPairCut.phimax=%f", fPhiMax);
122	tListSetttings->AddLast(new TObjString(buf));
123	snprintf(buf, 200, "AliFemtoKTPairCut.phimin=%f", fPhiMin);
124	tListSetttings->AddLast(new TObjString(buf));
125	snprintf(buf, 200, "AliFemtoKTPairCut.ptmin=%f", fPtMin);
126	tListSetttings->AddLast(new TObjString(buf));
127	snprintf(buf, 200, "AliFemtoKTPairCut.ptmax=%f", fPtMax);
128	tListSetttings->AddLast(new TObjString(buf));
129
130	return tListSetttings;
131	}
132
133	void AliFemtoKTPairCut::SetKTRange(double ktmin, double ktmax)
134	{
135	fKTMin = ktmin;
136	fKTMax = ktmax;
137	}
138
139	void AliFemtoKTPairCut::SetPhiRange(double phimin, double phimax)
140	{
141	fPhiMin = phimin;
142	fPhiMax = phimax;
143	}
144
145	void AliFemtoKTPairCut::SetPTMin(double ptmin, double ptmax)
146	{
147	fPtMin = ptmin;
148	fPtMax = ptmax;
149	}
150
151	//______________________________________________________
152	bool AliFemtoKTPairCut::Pass(const AliFemtoPair* pair)
153	{
154	bool temp = true;
155
156	//Taking care of the Kt cut
157	if (pair->KT() < fKTMin)
158	temp = false;
159
160	if (pair->KT() > fKTMax)
161	temp = false;
162
163	if (!temp) return temp;
164
165	if ((fPtMin > 0.0) \|\| (fPtMax<1000.0)) {
166	// double px1 = pair->Track1()->Track()->P().x();
167	// double py1 = pair->Track1()->Track()->P().y();
168
169	// double px2 = pair->Track2()->Track()->P().x();
170	// double py2 = pair->Track2()->Track()->P().y();
171
172	// double pt1 = TMath::Hypot(px1, py1);
173	// double pt2 = TMath::Hypot(px2, py2);
174
175	// if ((pt1<fPtMin) \|\| (pt1>fPtMax)) return false;
176	// if ((pt2<fPtMin) \|\| (pt2>fPtMax)) return false;
177	if ((pair->Track1()->Track()->Pt()<fPtMin) \|\| (pair->Track1()->Track()->Pt()>fPtMax)) return false;
178	if ((pair->Track2()->Track()->Pt()<fPtMin) \|\| (pair->Track2()->Track()->Pt()>fPtMax)) return false;
179	}
180
181	//Taking care of the Phi cut
182	// double rpangle = (pair->GetPairAngleEP())*180/TMath::Pi();
183	double rpangle = pair->GetPairAngleEP();
184
185	if (rpangle > 180.0) rpangle -= 180.0;
186	if (rpangle < 0.0) rpangle += 180.0;
187
188	if (fPhiMin < 0) {
189	if ((rpangle > fPhiMax) && (rpangle < 180.0+fPhiMin))
190	temp = false;
191	}
192	else {
193	if ((rpangle < fPhiMin) \|\| (rpangle > fPhiMax))
194	temp = false;
195	}
196	return temp;
197	}
198
199	//_____________________________________
200	bool AliFemtoKTPairCut::Pass(const AliFemtoPair* pair, double aRPAngle)
201	{
202	//The same as above, but it is defined with RP Angle as input in all the Correlatin function classes.
203
204	bool temp = (aRPAngle > 0.);
205	aRPAngle = true;
206
207	if (!Pass(pair))
208	temp = false;
209
210	return temp;
211	}