[u/mrichter/AliRoot.git] / ANALYSIS / AliAODParticle.h

#ifndef ALIAODPARTICLE_H
#define ALIAODPARTICLE_H
//___________________________________________________________
/////////////////////////////////////////////////////////////
//
// class AliAODParticle
//
// Ali HBT Particle: simplified class TParticle
// Simplified in order to minimize the size of the object
//  - we want to keep a lot of such a objects in memory
// Additionaly adjusted for HBT Analysies purposes
// + pointer to Track Points
// + pointer to Cluster Map(s)
//
// Piotr.Skowronski@cern.ch
//
/////////////////////////////////////////////////////////////

#include "AliVAODParticle.h"
//#include <TLorentzVector.h>
//#include <TMath.h>
#include <TDatabasePDG.h>


class TParticle;
class AliTrackPoints;
class AliClusterMap;

class AliAODParticle: public AliVAODParticle
{
public:
                                // ****** constructors and destructor
  AliAODParticle();
  AliAODParticle(const AliAODParticle& in); 
  AliAODParticle(const AliVAODParticle& in); 
 
  AliAODParticle(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);

  AliAODParticle(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);

  AliAODParticle(const TParticle& p, Int_t idx);

  virtual ~AliAODParticle();
  
  AliAODParticle& operator=(const AliAODParticle& in); 
  
  void           Clear(Option_t*);//Must be implemented in order to store this object in Clones Array
  
  TLorentzVector FourMomentum() const {TLorentzVector v(fPx,fPy,fPz,fE);return v;}
  TVector3       ProductionVertex() const {TVector3 v(fVx,fVy,fVz);return v;}
  
  void           SetPIDprobability(Int_t pdg, Float_t prob = 1.0);
  Float_t        GetPIDprobability(Int_t pdg) const;
  Double_t       GetProbability(Int_t pdg) const {return GetPIDprobability(pdg);}
  
  Int_t          GetMostProbable() const { return (fPids)?fPids[0]:0;}
  
  Int_t          GetPdgCode      () const { return (fPids)?fPids[fPdgIdx]:0;}

  Double_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       Mass            () const { return (GetPDG())?GetPDG()->Mass():-1.;}
  
  
  TParticlePDG*  GetPDG          () const {return TDatabasePDG::Instance()->GetParticle(GetPdgCode());}
  Double_t       Charge          () const { return (GetPDG())?GetPDG()->Charge():1.e8;}
  
  Int_t          Beauty          ()  { return GetPDG()->Beauty(); }
  Int_t          Charm           ()  { return GetPDG()->Charm(); }
  Int_t          Strangeness     ()  { return GetPDG()->Strangeness();}
  void ProductionVertex(TLorentzVector &v) const { 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) const { v.SetPxPyPzE(fPx,fPy,fPz,fE);}
    
  Double_t         Pt    () const  //transverse momentum
    { return TMath::Sqrt(fPx*fPx+fPy*fPy); }
  Double_t         E() 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 TMath::Pi()+TMath::ATan2(-fPy,-fPx); }

  Double_t         Theta () const { return (fPz==0)?TMath::PiOver2():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;}
  
  void            SetUID(Int_t id){fIdxInEvent = id;}
  
  const Char_t*   GetName() const; 
  void            Print() const;
  
  void            SetTPCTrackPoints(AliTrackPoints* tpts){fTPCTrackPoints = tpts;}
  AliTrackPoints* GetTPCTrackPoints() const {return fTPCTrackPoints;}
  void            SetITSTrackPoints(AliTrackPoints* tpts){fITSTrackPoints = tpts;}
  AliTrackPoints* GetITSTrackPoints() const {return fITSTrackPoints;}
  void            SetClusterMap(AliClusterMap* cm){fClusterMap = cm;}
  AliClusterMap*  GetClusterMap() const {return fClusterMap;}
  
  
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

  AliTrackPoints* fTPCTrackPoints;      // track positions along trajectory - used by anti-merging cut
  AliTrackPoints* fITSTrackPoints;      // track positions along trajectory - used by anti-merging cut
  AliClusterMap*  fClusterMap;       // bit map of cluters occupation; 1 if has cluter on given layer/padrow/...
    
  ClassDef(AliAODParticle,3)  // TParticle vertex particle information
};

#endif
Commit	Line	Data
a5556ea5	1	#ifndef ALIAODPARTICLE_H
a5556ea5	2	#define ALIAODPARTICLE_H
afa8b37b	3	//___________________________________________________________
a5556ea5	4	/////////////////////////////////////////////////////////////
a5556ea5	5	//
afa8b37b	6	// class AliAODParticle
	7	//
	8	// Ali HBT Particle: simplified class TParticle
	9	// Simplified in order to minimize the size of the object
	10	// - we want to keep a lot of such a objects in memory
	11	// Additionaly adjusted for HBT Analysies purposes
	12	// + pointer to Track Points
	13	// + pointer to Cluster Map(s)
	14	//
	15	// Piotr.Skowronski@cern.ch
a5556ea5	16	//
	17	/////////////////////////////////////////////////////////////
	18
afa8b37b	19	#include "AliVAODParticle.h"
	20	//#include <TLorentzVector.h>
	21	//#include <TMath.h>
	22	#include <TDatabasePDG.h>
a5556ea5	23
a5556ea5	24
afa8b37b	25	class TParticle;
a5556ea5	26	class AliTrackPoints;
	27	class AliClusterMap;
	28
afa8b37b	29	class AliAODParticle: public AliVAODParticle
afa8b37b	30	{
a5556ea5	31	public:
afa8b37b	32	// ****** constructors and destructor
	33	AliAODParticle();
	34	AliAODParticle(const AliAODParticle& in);
78d7c6d3	35	AliAODParticle(const AliVAODParticle& in);
afa8b37b	36
	37	AliAODParticle(Int_t pdg, Int_t idx, Double_t px, Double_t py, Double_t pz, Double_t etot,
	38	Double_t vx, Double_t vy, Double_t vz, Double_t time);
	39
	40	AliAODParticle(Int_t pdg, Float_t prob, Int_t idx, Double_t px, Double_t py, Double_t pz, Double_t etot,
	41	Double_t vx, Double_t vy, Double_t vz, Double_t time);
	42
	43	AliAODParticle(const TParticle& p, Int_t idx);
	44
	45	virtual ~AliAODParticle();
	46
	47	AliAODParticle& operator=(const AliAODParticle& in);
	48
295c08ac	49	void Clear(Option_t*);//Must be implemented in order to store this object in Clones Array
295c08ac	50
afa8b37b	51	TLorentzVector FourMomentum() const {TLorentzVector v(fPx,fPy,fPz,fE);return v;}
	52	TVector3 ProductionVertex() const {TVector3 v(fVx,fVy,fVz);return v;}
	53
	54	void SetPIDprobability(Int_t pdg, Float_t prob = 1.0);
	55	Float_t GetPIDprobability(Int_t pdg) const;
	56	Double_t GetProbability(Int_t pdg) const {return GetPIDprobability(pdg);}
78d7c6d3	57
afa8b37b	58	Int_t GetMostProbable() const { return (fPids)?fPids[0]:0;}
	59
	60	Int_t GetPdgCode () const { return (fPids)?fPids[fPdgIdx]:0;}
	61
78d7c6d3	62	Double_t GetPidProb () const { return (fPidProb)?fPidProb[fPdgIdx]:0;}
afa8b37b	63
	64	Int_t GetUID () const { return fIdxInEvent;}
	65	Int_t GetNumberOfPids () const { return fNPids;}
	66	Int_t GetNthPid (Int_t idx) const;
	67	Float_t GetNthPidProb (Int_t idx) const;
	68
	69	void SetPdgCode(Int_t pdg, Float_t prob = 1.0);
	70	Double_t GetCalcMass () const { return fCalcMass; }
	71	Double_t Mass () const { return (GetPDG())?GetPDG()->Mass():-1.;}
	72
	73
	74	TParticlePDG* GetPDG () const {return TDatabasePDG::Instance()->GetParticle(GetPdgCode());}
	75	Double_t Charge () const { return (GetPDG())?GetPDG()->Charge():1.e8;}
	76
	77	Int_t Beauty () { return GetPDG()->Beauty(); }
	78	Int_t Charm () { return GetPDG()->Charm(); }
	79	Int_t Strangeness () { return GetPDG()->Strangeness();}
	80	void ProductionVertex(TLorentzVector &v) const { v.SetXYZT(fVx,fVy,fVz,fVt);}
	81
	82
	83	Double_t Vx () const { return fVx;}
	84	Double_t Vy () const { return fVy;}
	85	Double_t Vz () const { return fVz;}
	86	Double_t T () const { return fVt;}
	87
	88	Double_t Px () const { return fPx; } //X coordinate of the momentum
	89	Double_t Py () const { return fPy; } //Y coordinate of the momentum
	90	Double_t Pz () const { return fPz; } //Z coordinate of the momentum
	91	Double_t P () const //momentum
	92	{ return TMath::Sqrt(fPxfPx+fPyfPy+fPz*fPz); }
	93
	94	void Momentum(TLorentzVector &v) const { v.SetPxPyPzE(fPx,fPy,fPz,fE);}
	95
	96	Double_t Pt () const //transverse momentum
	97	{ return TMath::Sqrt(fPxfPx+fPyfPy); }
b9eb3bc7	98	Double_t E() const { return fE; }
afa8b37b	99
	100	//Pseudo Rapidity
	101	Double_t Eta () const { if (P() != fPz) return 0.5*TMath::Log((P()+fPz)/(P()-fPz));
	102	else return 1.e30;}
	103
	104	//Rapidity
	105	Double_t Y () const { if (fE != fPz) return 0.5*TMath::Log((fE+fPz)/(fE-fPz));
	106	else return 1.e30;}
	107
	108	Double_t Phi () const { return TMath::Pi()+TMath::ATan2(-fPy,-fPx); }
	109
	110	Double_t Theta () const { return (fPz==0)?TMath::PiOver2():TMath::ACos(fPz/P()); }
	111
	112	// setters
	113
	114	void SetMomentum(Double_t px, Double_t py, Double_t pz, Double_t e)
	115	{fPx=px; fPy=py; fPz=pz; fE=e;}
	116	void SetMomentum(const TLorentzVector& p)
	117	{SetMomentum(p.Px(),p.Py(),p.Pz(),p.Energy());}
	118
	119	void SetProductionVertex(Double_t vx, Double_t vy, Double_t vz, Double_t t)
	120	{fVx=vx; fVy=vy; fVz=vz; fVt=t;}
	121	void SetProductionVertex(const TLorentzVector& v)
	122	{SetProductionVertex(v.X(),v.Y(),v.Z(),v.T());}
	123	void SetCalcMass(Double_t mass) {fCalcMass = mass;}
	124
78d7c6d3	125	void SetUID(Int_t id){fIdxInEvent = id;}
afa8b37b	126
78d7c6d3	127	const Char_t* GetName() const;
78d7c6d3	128	void Print() const;
afa8b37b	129
78d7c6d3	130	void SetTPCTrackPoints(AliTrackPoints* tpts){fTPCTrackPoints = tpts;}
	131	AliTrackPoints* GetTPCTrackPoints() const {return fTPCTrackPoints;}
	132	void SetITSTrackPoints(AliTrackPoints* tpts){fITSTrackPoints = tpts;}
	133	AliTrackPoints* GetITSTrackPoints() const {return fITSTrackPoints;}
	134	void SetClusterMap(AliClusterMap* cm){fClusterMap = cm;}
	135	AliClusterMap* GetClusterMap() const {return fClusterMap;}
afa8b37b	136
	137
	138	protected:
	139	Int_t GetPidSlot(Int_t pdg) const;//returns position of the given PID in fPids (and fPidProb) array.
a5556ea5	140
a5556ea5	141	private:
afa8b37b	142	Char_t fPdgIdx; // index of PDG code of the particle in fPids
	143	Int_t fIdxInEvent; // index of a particle: the same particle can appear in the event
	144	// many times with different pid's. Idx allows to check that they are the same particles
	145	Int_t fNPids; // number of non-zero proboble Pids
	146	Int_t *fPids; // [fNPids] Array with PIDs
	147	Float_t *fPidProb; // [fNPids] PIDs probabilities
	148	Double_t fCalcMass; // Calculated mass
	149
	150
	151	Double_t fPx; // x component of momentum
	152	Double_t fPy; // y component of momentum
	153	Double_t fPz; // z component of momentum
	154	Double_t fE; // Energy
	155
	156	Double_t fVx; // x of production vertex
	157	Double_t fVy; // y of production vertex
	158	Double_t fVz; // z of production vertex
	159	Double_t fVt; // t of production vertex
a5556ea5	160
78d7c6d3	161	AliTrackPoints* fTPCTrackPoints; // track positions along trajectory - used by anti-merging cut
78d7c6d3	162	AliTrackPoints* fITSTrackPoints; // track positions along trajectory - used by anti-merging cut
afa8b37b	163	AliClusterMap* fClusterMap; // bit map of cluters occupation; 1 if has cluter on given layer/padrow/...
	164
	165	ClassDef(AliAODParticle,3) // TParticle vertex particle information
a5556ea5	166	};
	167
	168	#endif