[u/mrichter/AliRoot.git] / HBTAN / AliHBTParticle.h

#ifndef ALIHBTPARTICLE
#define ALIHBTPARTICLE

//Ali HBT Particle: simplified class TParticle
//Simplified in order to minimize the size of object
//  - we want to keep a lot of such a objects in memory

#include <TObject.h>
#include <TLorentzVector.h>
#include <TMath.h>
#include <TDatabasePDG.h>

#include "AliConst.h"

class TParticle;

class AliHBTParticle : public TObject
{

public:
                                // ****** constructors and destructor
  AliHBTParticle();
  AliHBTParticle(const AliHBTParticle& in); 
 
  AliHBTParticle(Int_t pdg, Int_t idx, Double_t px, Double_t py, Double_t pz, Double_t etot,
                 Double_t vx, Double_t vy, Double_t vz, Double_t time);

  AliHBTParticle(Int_t pdg, Float_t prob, Int_t idx, Double_t px, Double_t py, Double_t pz, Double_t etot,
                 Double_t vx, Double_t vy, Double_t vz, Double_t time);

  AliHBTParticle(const TParticle& p,Int_t idx);

  virtual ~AliHBTParticle();
  
  AliHBTParticle& operator=(const AliHBTParticle& in); 
  
  void           SetPIDprobability(Int_t pdg, Float_t prob = 1.0);
  Float_t        GetPIDprobability(Int_t pdg);
  
  Int_t          GetPdgCode      () const { return (fPids)?fPids[fPdgIdx]:0;}
  Int_t          GetPid          () const { return GetPdgCode();}
  Float_t        GetPidProb      () const { return (fPidProb)?fPidProb[fPdgIdx]:0;}
  
  Int_t          GetUID          () const { return fIdxInEvent;}
  Int_t          GetNumberOfPids () const { return fNPids;}
  Int_t          GetNthPid         (Int_t idx) const;
  Float_t        GetNthPidProb     (Int_t idx) const;
      
  void           SetPdgCode(Int_t pdg, Float_t prob = 1.0);
  Double_t       GetCalcMass     () const { return fCalcMass; }
  Double_t       GetMass         ()       { return (GetPDG())?GetPDG()->Mass():-1.;}

  TParticlePDG*  GetPDG          (){return TDatabasePDG::Instance()->GetParticle(GetPdgCode());}

  Int_t          Beauty          ()  { return GetPDG()->Beauty(); }
  Int_t          Charm           ()  { return GetPDG()->Charm(); }
  Int_t          Strangeness     ()  { return GetPDG()->Strangeness();}
  void ProductionVertex(TLorentzVector &v) { v.SetXYZT(fVx,fVy,fVz,fVt);}


  Double_t         Vx    () const { return fVx;}
  Double_t         Vy    () const { return fVy;}
  Double_t         Vz    () const { return fVz;}
  Double_t         T     () const { return fVt;}

  Double_t         Px    () const { return fPx; } //X coordinate of the momentum
  Double_t         Py    () const { return fPy; } //Y coordinate of the momentum
  Double_t         Pz    () const { return fPz; } //Z coordinate of the momentum
  Double_t         P     () const                 //momentum
    { return TMath::Sqrt(fPx*fPx+fPy*fPy+fPz*fPz); }
  
  void Momentum(TLorentzVector &v) { v.SetPxPyPzE(fPx,fPy,fPz,fE);}
    
  Double_t         Pt    () const  //transverse momentum
    { return TMath::Sqrt(fPx*fPx+fPy*fPy); }
  Double_t         Energy() const { return fE; }
  
                                   //Pseudo Rapidity
  Double_t         Eta   () const { if (P() != fPz) return 0.5*TMath::Log((P()+fPz)/(P()-fPz)); 
                                    else return 1.e30;}

                                   //Rapidity
  Double_t         Y     () const { if (fE  != fPz) return 0.5*TMath::Log((fE+fPz)/(fE-fPz));
                                    else return 1.e30;}

  Double_t         Phi   () const { return kPI+TMath::ATan2(-fPy,-fPx); }

  Double_t         Theta () const { return (fPz==0)?kPI/2:TMath::ACos(fPz/P()); }

                                // setters


  void           SetMomentum(Double_t px, Double_t py, Double_t pz, Double_t e)
                             {fPx=px; fPy=py; fPz=pz; fE=e;}
  void           SetMomentum(const TLorentzVector& p)
                             {SetMomentum(p.Px(),p.Py(),p.Pz(),p.Energy());}

  void           SetProductionVertex(Double_t vx, Double_t vy, Double_t vz, Double_t t)
                             {fVx=vx; fVy=vy; fVz=vz; fVt=t;}
  void           SetProductionVertex(const TLorentzVector& v)
                             {SetProductionVertex(v.X(),v.Y(),v.Z(),v.T());}
  void           SetCalcMass(Double_t mass) {fCalcMass = mass;}
  const Char_t*  GetName() const; 
  void           Print() const;

  static void    SetDebug(Int_t dbg=1){fgDebug=dbg;}
  static Int_t   GetDebug(){return fgDebug;}
  static Int_t   fgDebug; //debug printout level
  
protected:
  Int_t          GetPidSlot(Int_t pdg) const;//returns position of the given PID in fPids (and fPidProb) array.

private:
  Char_t         fPdgIdx;               // index of PDG code of the particle in fPids
  Int_t          fIdxInEvent;           // index of a particle: the same particle can appear in the event
                                        //  many times with different pid's. Idx allows to check that they are the same particles
  Int_t          fNPids;                // number of non-zero proboble Pids
  Int_t         *fPids;                 // [fNPids] Array with PIDs
  Float_t       *fPidProb;              // [fNPids] PIDs probabilities
  Double_t       fCalcMass;             // Calculated mass

  Double_t       fPx;                   // x component of momentum
  Double_t       fPy;                   // y component of momentum
  Double_t       fPz;                   // z component of momentum
  Double_t       fE;                    // Energy

  Double_t       fVx;                   // x of production vertex
  Double_t       fVy;                   // y of production vertex
  Double_t       fVz;                   // z of production vertex
  Double_t       fVt;                   // t of production vertex

  ClassDef(AliHBTParticle,2)  // TParticle vertex particle information
};

#endif
Commit	Line	Data
1b446896	1	#ifndef ALIHBTPARTICLE
	2	#define ALIHBTPARTICLE
	3
	4	//Ali HBT Particle: simplified class TParticle
	5	//Simplified in order to minimize the size of object
	6	// - we want to keep a lot of such a objects in memory
	7
	8	#include <TObject.h>
	9	#include <TLorentzVector.h>
	10	#include <TMath.h>
	11	#include <TDatabasePDG.h>
	12
829f5c7c	13	#include "AliConst.h"
829f5c7c	14
1b446896	15	class TParticle;
	16
	17	class AliHBTParticle : public TObject
	18	{
	19
	20	public:
	21	// ****** constructors and destructor
	22	AliHBTParticle();
8089a083	23	AliHBTParticle(const AliHBTParticle& in);
	24
	25	AliHBTParticle(Int_t pdg, Int_t idx, Double_t px, Double_t py, Double_t pz, Double_t etot,
	26	Double_t vx, Double_t vy, Double_t vz, Double_t time);
1b446896	27
8089a083	28	AliHBTParticle(Int_t pdg, Float_t prob, Int_t idx, Double_t px, Double_t py, Double_t pz, Double_t etot,
1b446896	29	Double_t vx, Double_t vy, Double_t vz, Double_t time);
1b446896	30
8089a083	31	AliHBTParticle(const TParticle& p,Int_t idx);
1b446896	32
bed069a4	33	virtual ~AliHBTParticle();
	34
	35	AliHBTParticle& operator=(const AliHBTParticle& in);
	36
8089a083	37	void SetPIDprobability(Int_t pdg, Float_t prob = 1.0);
	38	Float_t GetPIDprobability(Int_t pdg);
	39
	40	Int_t GetPdgCode () const { return (fPids)?fPids[fPdgIdx]:0;}
	41	Int_t GetPid () const { return GetPdgCode();}
	42	Float_t GetPidProb () const { return (fPidProb)?fPidProb[fPdgIdx]:0;}
	43
	44	Int_t GetUID () const { return fIdxInEvent;}
e665964c	45	Int_t GetNumberOfPids () const { return fNPids;}
	46	Int_t GetNthPid (Int_t idx) const;
	47	Float_t GetNthPidProb (Int_t idx) const;
	48
8089a083	49	void SetPdgCode(Int_t pdg, Float_t prob = 1.0);
1b446896	50	Double_t GetCalcMass () const { return fCalcMass; }
8089a083	51	Double_t GetMass () { return (GetPDG())?GetPDG()->Mass():-1.;}
1b446896	52
8089a083	53	TParticlePDG* GetPDG (){return TDatabasePDG::Instance()->GetParticle(GetPdgCode());}
1b446896	54
	55	Int_t Beauty () { return GetPDG()->Beauty(); }
	56	Int_t Charm () { return GetPDG()->Charm(); }
	57	Int_t Strangeness () { return GetPDG()->Strangeness();}
	58	void ProductionVertex(TLorentzVector &v) { v.SetXYZT(fVx,fVy,fVz,fVt);}
	59
	60
	61	Double_t Vx () const { return fVx;}
	62	Double_t Vy () const { return fVy;}
	63	Double_t Vz () const { return fVz;}
	64	Double_t T () const { return fVt;}
	65
	66	Double_t Px () const { return fPx; } //X coordinate of the momentum
	67	Double_t Py () const { return fPy; } //Y coordinate of the momentum
	68	Double_t Pz () const { return fPz; } //Z coordinate of the momentum
	69	Double_t P () const //momentum
	70	{ return TMath::Sqrt(fPxfPx+fPyfPy+fPz*fPz); }
	71
	72	void Momentum(TLorentzVector &v) { v.SetPxPyPzE(fPx,fPy,fPz,fE);}
	73
	74	Double_t Pt () const //transverse momentum
	75	{ return TMath::Sqrt(fPxfPx+fPyfPy); }
	76	Double_t Energy() const { return fE; }
	77
	78	//Pseudo Rapidity
	79	Double_t Eta () const { if (P() != fPz) return 0.5*TMath::Log((P()+fPz)/(P()-fPz));
	80	else return 1.e30;}
	81
	82	//Rapidity
	83	Double_t Y () const { if (fE != fPz) return 0.5*TMath::Log((fE+fPz)/(fE-fPz));
	84	else return 1.e30;}
	85
	86	Double_t Phi () const { return kPI+TMath::ATan2(-fPy,-fPx); }
	87
	88	Double_t Theta () const { return (fPz==0)?kPI/2:TMath::ACos(fPz/P()); }
	89
	90	// setters
	91
	92
	93	void SetMomentum(Double_t px, Double_t py, Double_t pz, Double_t e)
	94	{fPx=px; fPy=py; fPz=pz; fE=e;}
	95	void SetMomentum(const TLorentzVector& p)
	96	{SetMomentum(p.Px(),p.Py(),p.Pz(),p.Energy());}
	97
	98	void SetProductionVertex(Double_t vx, Double_t vy, Double_t vz, Double_t t)
	99	{fVx=vx; fVy=vy; fVz=vz; fVt=t;}
	100	void SetProductionVertex(const TLorentzVector& v)
	101	{SetProductionVertex(v.X(),v.Y(),v.Z(),v.T());}
4d1d0fe9	102	void SetCalcMass(Double_t mass) {fCalcMass = mass;}
637fd167	103	const Char_t* GetName() const;
8089a083	104	void Print() const;
	105
	106	static void SetDebug(Int_t dbg=1){fgDebug=dbg;}
	107	static Int_t GetDebug(){return fgDebug;}
	108	static Int_t fgDebug; //debug printout level
1b446896	109
1b446896	110	protected:
8089a083	111	Int_t GetPidSlot(Int_t pdg) const;//returns position of the given PID in fPids (and fPidProb) array.
	112
	113	private:
	114	Char_t fPdgIdx; // index of PDG code of the particle in fPids
	115	Int_t fIdxInEvent; // index of a particle: the same particle can appear in the event
	116	// many times with different pid's. Idx allows to check that they are the same particles
ff3d1148	117	Int_t fNPids; // number of non-zero proboble Pids
	118	Int_t *fPids; // [fNPids] Array with PIDs
	119	Float_t *fPidProb; // [fNPids] PIDs probabilities
1b446896	120	Double_t fCalcMass; // Calculated mass
	121
	122	Double_t fPx; // x component of momentum
	123	Double_t fPy; // y component of momentum
	124	Double_t fPz; // z component of momentum
	125	Double_t fE; // Energy
	126
	127	Double_t fVx; // x of production vertex
	128	Double_t fVy; // y of production vertex
	129	Double_t fVz; // z of production vertex
	130	Double_t fVt; // t of production vertex
	131
ff3d1148	132	ClassDef(AliHBTParticle,2) // TParticle vertex particle information
1b446896	133	};
	134
	135	#endif