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

////////////////////////////////////////////////////////////////////////////////
///                                                                          ///
/// AliFemtoModelWeightGenerator - abstract base class for femtoscopic       ///
/// weight generator                                                         ///
/// Authors: Adam Kisiel kisiel@mps.ohio-state.edu                           ///
///                                                                          ///
////////////////////////////////////////////////////////////////////////////////
#ifdef __ROOT__
  ClassImp(AliFemtoModelGausLCMSFreezeOutGenerator, 1)
#endif

#include "AliFemtoModelWeightGenerator.h"
#include "AliFemtoModelHiddenInfo.h"

const Int_t AliFemtoModelWeightGenerator::fgkPionPlusPionPlus = 1;
const Int_t AliFemtoModelWeightGenerator::fgkPionPlusPionMinus = 2;
const Int_t AliFemtoModelWeightGenerator::fgkKaonPlusKaonPlus = 3;
const Int_t AliFemtoModelWeightGenerator::fgkKaonPlusKaonMinus = 4;
const Int_t AliFemtoModelWeightGenerator::fgkProtonProton = 5;
const Int_t AliFemtoModelWeightGenerator::fgkProtonAntiproton = 6;
const Int_t AliFemtoModelWeightGenerator::fgkPionPlusKaonPlus = 7;
const Int_t AliFemtoModelWeightGenerator::fgkPionPlusKaonMinus = 8;
const Int_t AliFemtoModelWeightGenerator::fgkPionPlusProton = 9;
const Int_t AliFemtoModelWeightGenerator::fgkPionPlusAntiproton = 10;
const Int_t AliFemtoModelWeightGenerator::fgkKaonPlusProton = 11;
const Int_t AliFemtoModelWeightGenerator::fgkKaonPlusAntiproton = 12;

//_____________________________________________
AliFemtoModelWeightGenerator::AliFemtoModelWeightGenerator() :
  fPairType(0), 
  fKStarOut(0), fKStarSide(0), fKStarLong(0), fKStar(0), 
  fRStarOut(0), fRStarSide(0), fRStarLong(0), fRStar(0)
{}
//_____________________________________________
AliFemtoModelWeightGenerator::AliFemtoModelWeightGenerator(const AliFemtoModelWeightGenerator &aModel) :
  fPairType(0), 
  fKStarOut(0), fKStarSide(0), fKStarLong(0), fKStar(0), 
  fRStarOut(0), fRStarSide(0), fRStarLong(0), fRStar(0)
{
  fPairType = aModel.fPairType;
}
//_____________________________________________
AliFemtoModelWeightGenerator::~AliFemtoModelWeightGenerator(){/* no-op */}
//_____________________________________________
void     AliFemtoModelWeightGenerator::SetPairType(Int_t aPairType)
{
  fPairType = aPairType;
}

//_____________________________________________
Int_t    AliFemtoModelWeightGenerator::GetPairType() const
{
  return fPairType;
}

//_____________________________________________
void     AliFemtoModelWeightGenerator::SetPairTypeFromPair(AliFemtoPair *aPair)
{
  AliFemtoModelHiddenInfo *inf1 = ( AliFemtoModelHiddenInfo *) aPair->Track1()->HiddenInfo();
  AliFemtoModelHiddenInfo *inf2 = ( AliFemtoModelHiddenInfo *) aPair->Track2()->HiddenInfo();

  const Int_t ktPid1 = inf1->GetPDGPid();
  const Int_t ktPid2 = inf2->GetPDGPid();

  if      (((ktPid1 ==   211) && (ktPid2 ==   211)) ||
           ((ktPid1 ==  -211) && (ktPid2 ==  -211)))
    fPairType = fgkPionPlusPionPlus;
  else if (((ktPid1 ==  -211) && (ktPid2 ==   211)) ||
           ((ktPid1 ==   211) && (ktPid2 ==  -211)))
    fPairType = fgkPionPlusPionMinus;
  else if (((ktPid1 ==   321) && (ktPid2 ==   321)) ||
           ((ktPid1 ==  -321) && (ktPid2 ==  -321)))
    fPairType = fgkKaonPlusKaonPlus;
  else if (((ktPid1 ==  -321) && (ktPid2 ==   321)) ||
           ((ktPid1 ==   321) && (ktPid2 ==  -321)))
    fPairType = fgkKaonPlusKaonMinus;
  else if (((ktPid1 ==  2212) && (ktPid2 ==  2212)) ||
           ((ktPid1 == -2212) && (ktPid2 == -2212)))
    fPairType = fgkProtonProton;
  else if (((ktPid1 == -2212) && (ktPid2 ==  2212)) ||
           ((ktPid1 ==  2212) && (ktPid2 == -2212)))
    fPairType = fgkProtonAntiproton;
  else if (((ktPid1 ==   211) && (ktPid2 ==   321)) ||
           ((ktPid1 ==  -211) && (ktPid2 ==  -321)))
    fPairType = fgkPionPlusKaonPlus;
  else if (((ktPid1 ==  -211) && (ktPid2 ==   321)) ||
           ((ktPid1 ==   211) && (ktPid2 ==  -321)))
    fPairType = fgkPionPlusKaonMinus;
  else if (((ktPid1 ==   211) && (ktPid2 ==  2212)) ||
           ((ktPid1 ==  -211) && (ktPid2 == -2212)))
    fPairType = fgkPionPlusProton;
  else if (((ktPid1 ==  -211) && (ktPid2 ==  2212)) ||
           ((ktPid1 ==   211) && (ktPid2 == -2212)))
    fPairType = fgkPionPlusAntiproton;
  else if (((ktPid1 ==   321) && (ktPid2 ==  2212)) ||
           ((ktPid1 ==  -321) && (ktPid2 == -2212)))
    fPairType = fgkKaonPlusProton;
  else if (((ktPid1 ==  -321) && (ktPid2 ==  2212)) ||
           ((ktPid1 ==   321) && (ktPid2 == -2212)))
    fPairType = fgkKaonPlusAntiproton;
}

//_____________________________________________
AliFemtoModelWeightGenerator* AliFemtoModelWeightGenerator::Clone() const
{
  return 0;
}
Commit	Line	Data
d0e92d9a	1	////////////////////////////////////////////////////////////////////////////////
	2	/// ///
	3	/// AliFemtoModelWeightGenerator - abstract base class for femtoscopic ///
	4	/// weight generator ///
	5	/// Authors: Adam Kisiel kisiel@mps.ohio-state.edu ///
	6	/// ///
	7	////////////////////////////////////////////////////////////////////////////////
	8	#ifdef __ROOT__
	9	ClassImp(AliFemtoModelGausLCMSFreezeOutGenerator, 1)
	10	#endif
	11
	12	#include "AliFemtoModelWeightGenerator.h"
	13	#include "AliFemtoModelHiddenInfo.h"
	14
	15	const Int_t AliFemtoModelWeightGenerator::fgkPionPlusPionPlus = 1;
	16	const Int_t AliFemtoModelWeightGenerator::fgkPionPlusPionMinus = 2;
	17	const Int_t AliFemtoModelWeightGenerator::fgkKaonPlusKaonPlus = 3;
	18	const Int_t AliFemtoModelWeightGenerator::fgkKaonPlusKaonMinus = 4;
	19	const Int_t AliFemtoModelWeightGenerator::fgkProtonProton = 5;
	20	const Int_t AliFemtoModelWeightGenerator::fgkProtonAntiproton = 6;
	21	const Int_t AliFemtoModelWeightGenerator::fgkPionPlusKaonPlus = 7;
	22	const Int_t AliFemtoModelWeightGenerator::fgkPionPlusKaonMinus = 8;
	23	const Int_t AliFemtoModelWeightGenerator::fgkPionPlusProton = 9;
	24	const Int_t AliFemtoModelWeightGenerator::fgkPionPlusAntiproton = 10;
	25	const Int_t AliFemtoModelWeightGenerator::fgkKaonPlusProton = 11;
	26	const Int_t AliFemtoModelWeightGenerator::fgkKaonPlusAntiproton = 12;
	27
	28	//_____________________________________________
	29	AliFemtoModelWeightGenerator::AliFemtoModelWeightGenerator() :
	30	fPairType(0),
	31	fKStarOut(0), fKStarSide(0), fKStarLong(0), fKStar(0),
	32	fRStarOut(0), fRStarSide(0), fRStarLong(0), fRStar(0)
	33	{}
	34	//_____________________________________________
	35	AliFemtoModelWeightGenerator::AliFemtoModelWeightGenerator(const AliFemtoModelWeightGenerator &aModel) :
	36	fPairType(0),
	37	fKStarOut(0), fKStarSide(0), fKStarLong(0), fKStar(0),
	38	fRStarOut(0), fRStarSide(0), fRStarLong(0), fRStar(0)
	39	{
	40	fPairType = aModel.fPairType;
	41	}
	42	//_____________________________________________
	43	AliFemtoModelWeightGenerator::~AliFemtoModelWeightGenerator(){/* no-op */}
	44	//_____________________________________________
	45	void AliFemtoModelWeightGenerator::SetPairType(Int_t aPairType)
	46	{
	47	fPairType = aPairType;
	48	}
	49
	50	//_____________________________________________
	51	Int_t AliFemtoModelWeightGenerator::GetPairType() const
	52	{
	53	return fPairType;
	54	}
	55
	56	//_____________________________________________
	57	void AliFemtoModelWeightGenerator::SetPairTypeFromPair(AliFemtoPair *aPair)
	58	{
	59	AliFemtoModelHiddenInfo inf1 = ( AliFemtoModelHiddenInfo ) aPair->Track1()->HiddenInfo();
	60	AliFemtoModelHiddenInfo inf2 = ( AliFemtoModelHiddenInfo ) aPair->Track2()->HiddenInfo();
	61
	62	const Int_t ktPid1 = inf1->GetPDGPid();
	63	const Int_t ktPid2 = inf2->GetPDGPid();
	64
65	if (((ktPid1 == 211) && (ktPid2 == 211)) \|\|
66	((ktPid1 == -211) && (ktPid2 == -211)))
67	fPairType = fgkPionPlusPionPlus;
68	else if (((ktPid1 == -211) && (ktPid2 == 211)) \|\|
69	((ktPid1 == 211) && (ktPid2 == -211)))
70	fPairType = fgkPionPlusPionMinus;
71	else if (((ktPid1 == 321) && (ktPid2 == 321)) \|\|
72	((ktPid1 == -321) && (ktPid2 == -321)))
73	fPairType = fgkKaonPlusKaonPlus;
74	else if (((ktPid1 == -321) && (ktPid2 == 321)) \|\|
75	((ktPid1 == 321) && (ktPid2 == -321)))
76	fPairType = fgkKaonPlusKaonMinus;
77	else if (((ktPid1 == 2212) && (ktPid2 == 2212)) \|\|
78	((ktPid1 == -2212) && (ktPid2 == -2212)))
79	fPairType = fgkProtonProton;
80	else if (((ktPid1 == -2212) && (ktPid2 == 2212)) \|\|
81	((ktPid1 == 2212) && (ktPid2 == -2212)))
82	fPairType = fgkProtonAntiproton;
83	else if (((ktPid1 == 211) && (ktPid2 == 321)) \|\|
84	((ktPid1 == -211) && (ktPid2 == -321)))
85	fPairType = fgkPionPlusKaonPlus;
86	else if (((ktPid1 == -211) && (ktPid2 == 321)) \|\|
87	((ktPid1 == 211) && (ktPid2 == -321)))
88	fPairType = fgkPionPlusKaonMinus;
89	else if (((ktPid1 == 211) && (ktPid2 == 2212)) \|\|
90	((ktPid1 == -211) && (ktPid2 == -2212)))
91	fPairType = fgkPionPlusProton;
92	else if (((ktPid1 == -211) && (ktPid2 == 2212)) \|\|
93	((ktPid1 == 211) && (ktPid2 == -2212)))
94	fPairType = fgkPionPlusAntiproton;
95	else if (((ktPid1 == 321) && (ktPid2 == 2212)) \|\|
96	((ktPid1 == -321) && (ktPid2 == -2212)))
97	fPairType = fgkKaonPlusProton;
98	else if (((ktPid1 == -321) && (ktPid2 == 2212)) \|\|
99	((ktPid1 == 321) && (ktPid2 == -2212)))
100	fPairType = fgkKaonPlusAntiproton;
101	}
102
103	//_____________________________________________
104	AliFemtoModelWeightGenerator* AliFemtoModelWeightGenerator::Clone() const
105	{
106	return 0;
107	}