[u/mrichter/AliRoot.git] / PWG / EMCAL / AliPicoTrack.h

#ifndef AliPicoTrack_H
#define AliPicoTrack_H

#include <TMath.h>

#include "AliVTrack.h"
#include "AliAODMCParticle.h"

class AliVCluster;

class AliPicoTrack: public AliVTrack {
 public:
  AliPicoTrack();
  AliPicoTrack(Double_t pt, Double_t eta, Double_t phi, Byte_t q, Int_t label, Byte_t type,
               Double_t etaemc=0, Double_t phiemc=0, Double_t ptemc=0, Bool_t ise=0, Double_t mass=0.13957);
  AliPicoTrack(const AliPicoTrack &pc); 
  AliPicoTrack &operator=(const AliPicoTrack &pc);

  Double_t        Px()                        const { return fPt*TMath::Cos(fPhi);  }
  Double_t        Py()                        const { return fPt*TMath::Sin(fPhi);  }
  Double_t        Pz()                        const { return fPt*TMath::SinH(fEta); }
  Double_t        Pt()                        const { return fPt;                   }
  Double_t        P()                         const { return fPt*TMath::CosH(fEta); }
  
  Bool_t          PxPyPz(Double_t p[3])       const { p[0]=Px();p[1]=Py();p[2]=Pz(); return 1;         }
  Bool_t          GetPxPyPz(Double_t p[3])    const { p[0]=Px();p[1]=Py();p[2]=Pz(); return 1;         }
  Double_t        Xv()                        const { return 0.;      }
  Double_t        Yv()                        const { return 0.;      }
  Double_t        Zv()                        const { return 0.;      }
  Bool_t          XvYvZv(Double_t x[3])       const { x[0]=0;x[1]=0;x[2]=0; return 1;                  }
  Double_t        OneOverPt()                 const { return 1./fPt;  }
  Double_t        Phi()                       const { return fPhi;    }
  Double_t        Theta()                     const { return 2*TMath::ATan(TMath::Exp(-fEta));         }
  Double_t        E()                         const { Double_t p=P(); return TMath::Sqrt(M()*M()+p*p); }
  Double_t        M()                         const { return fM; }
  Double_t        Eta()                       const { return fEta;    }
  Double_t        Y()                         const { return 0.5*TMath::Log((E()+Pz())/(E()-Pz()));  }
  Short_t         Charge()                    const { return (char)fQ;}
  Int_t           GetLabel()                  const { return fLabel;  }
  void            SetLabel(Int_t label)             { fLabel = label; }
  Byte_t          GetTrackType()              const { return fTrackType;}
  void            SetTrackType(Byte_t type)         { fTrackType = type;}
  Int_t           PdgCode()                   const { return 0;       }
  const Double_t *PID()                const { return 0;       }
  Int_t           GetID()                     const { return 0;       }
  UChar_t         GetITSClusterMap()          const { return 0;       }
  Int_t           GetEMCALcluster()           const { return fClusId; }
  void            SetEMCALcluster(Int_t id)         { fClusId = id;   }
  Double_t        GetEtaEmc()                 const { return GetTrackEtaOnEMCal(); }
  Double_t        GetPhiEmc()                 const { return GetTrackPhiOnEMCal(); }
  Bool_t          IsEMCAL()                   const { return fEmcal;  }
  ULong_t         GetStatus()                 const { return 0;       }
  Bool_t          GetXYZ(Double_t *v)         const { v[0]=0; v[1]=0; v[2]=0; return 0; }
  using AliVTrack::GetXYZ;
  Double_t        GetBz()                     const { return 0;       }
  void            GetBxByBz(Double_t b[3])    const { b[0]=0;b[1]=0;b[2]=0; }
  Bool_t          GetCovarianceXYZPxPyPz(Double_t /*cv*/[21]) const { return 0; }
  AliVTrack      *GetTrack()                  const { return fOrig;   }
  void            SetTrack(AliVTrack *t)            { fOrig = t;      }
  Double_t        GetTrackPhiOnEMCal()        const { return fPhiEmc ; }
  Double_t        GetTrackEtaOnEMCal()        const { return fEtaEmc ; }
  Double_t        GetTrackPtOnEMCal()         const { return fPtEmc  ; }
  Double_t        GetTrackPOnEMCal()          const { return fPtEmc*TMath::CosH(fEtaEmc);}
  void            SetTrackPhiEtaPtOnEMCal(Double_t eta, Double_t phi, Double_t pt) { fEtaEmc = eta; fPhiEmc = phi; fPtEmc = pt;}
  void            Clear(Option_t * /*option*/ ="")  { fClusId = -1; fOrig = 0; } 
  Int_t           Compare(const TObject* obj) const;
  Bool_t          PropagateToDCA(const AliVVertex *, Double_t, Double_t, Double_t *, Double_t *) { return 0; }
  void            SetFlag(UInt_t flag) {fFlag = flag;}
  UInt_t          GetFlag() const {return fFlag;}
  void SetPrimary(Bool_t b = kTRUE){
    if(b)fFlag |= AliAODMCParticle::kPrimary;
    else fFlag &= ~AliAODMCParticle::kPrimary;
  }
  Bool_t IsPrimary() const {return ((fFlag&AliAODMCParticle::kPrimary)==AliAODMCParticle::kPrimary);} 
  
  void SetPhysicalPrimary(Bool_t b = kTRUE){
    if(b)fFlag |= AliAODMCParticle::kPhysicalPrim;
    else fFlag &= ~AliAODMCParticle::kPhysicalPrim; 
  } 
  Bool_t IsPhysicalPrimary() const {return ((fFlag&AliAODMCParticle::kPhysicalPrim)==AliAODMCParticle::kPhysicalPrim);} 
  
  void SetSecondaryFromWeakDecay(Bool_t b = kTRUE){
    if(b)fFlag |= AliAODMCParticle::kSecondaryFromWeakDecay;
    else fFlag &= ~AliAODMCParticle::kSecondaryFromWeakDecay; 
  } 
  Bool_t IsSecondaryFromWeakDecay() const {return ((fFlag&AliAODMCParticle::kSecondaryFromWeakDecay)==AliAODMCParticle::kSecondaryFromWeakDecay);} 
  
  void SetSecondaryFromMaterial(Bool_t b = kTRUE){
    if(b)fFlag |= AliAODMCParticle::kSecondaryFromMaterial;
    else fFlag &= ~AliAODMCParticle::kSecondaryFromMaterial; 
  } 
  Bool_t IsSecondaryFromMaterial() const {return ((fFlag&AliAODMCParticle::kSecondaryFromMaterial)==AliAODMCParticle::kSecondaryFromMaterial);} 

  Bool_t  IsFromHIJING() const { return (fGeneratorIndex == 0); }
  void    SetGeneratorIndex(Short_t i) {fGeneratorIndex = i;}
  Short_t GetGeneratorIndex() const {return fGeneratorIndex;}

  static void     GetEtaPhiDiff(const AliVTrack *t, const AliVCluster *v, Double_t &phidiff, Double_t &etadiff);
  static Byte_t   GetTrackType(const AliVTrack *t);

 protected:
  Double32_t      fPt;             //[0,0,12]   pt at vertex
  Double32_t      fEta;            //[-1,1,12]  eta at vertex
  Double32_t      fPhi;            //[0,6.3,12] phi at vertex
  Double32_t      fM;              //           mass
  Byte_t          fQ;              //           charge
  Int_t           fLabel;          //           label  
  Byte_t          fTrackType;      //           0=global track; 1=w/o SPD, w/ ITS refit; 2=w/o SPD, w/o ITS refit
  Double32_t      fEtaEmc;         //[-1,1,12]  eta at emcal surface
  Double32_t      fPhiEmc;         //[0,6.3,12] phi at emcal surface
  Double32_t      fPtEmc;          //[0,0,12]   pt at emcal surface
  Bool_t          fEmcal;          //           is true if track propagated to emcal
  UInt_t          fFlag;           //!          Flag for indication of primary etc (MC)
  Short_t         fGeneratorIndex; //!          Index of generator in cocktail
  Short_t         fClusId;         //!          cluster id of matched cluster; -1 if not set
  AliVTrack      *fOrig;           //!          ptr to original track

  ClassDef(AliPicoTrack, 8) // Pico track class
};
#endif
Commit	Line	Data
dd233e4e	1	#ifndef AliPicoTrack_H
	2	#define AliPicoTrack_H
	3
eb556d0c	4	#include <TMath.h>
dd233e4e	5
dd233e4e	6	#include "AliVTrack.h"
eb556d0c	7	#include "AliAODMCParticle.h"
eb556d0c	8
b723beb6	9	class AliVCluster;
dd233e4e	10
	11	class AliPicoTrack: public AliVTrack {
	12	public:
	13	AliPicoTrack();
39799f2a	14	AliPicoTrack(Double_t pt, Double_t eta, Double_t phi, Byte_t q, Int_t label, Byte_t type,
de1c0913	15	Double_t etaemc=0, Double_t phiemc=0, Double_t ptemc=0, Bool_t ise=0, Double_t mass=0.13957);
dd233e4e	16	AliPicoTrack(const AliPicoTrack &pc);
	17	AliPicoTrack &operator=(const AliPicoTrack &pc);
	18
fb4e4852	19	Double_t Px() const { return fPt*TMath::Cos(fPhi); }
	20	Double_t Py() const { return fPt*TMath::Sin(fPhi); }
	21	Double_t Pz() const { return fPt*TMath::SinH(fEta); }
	22	Double_t Pt() const { return fPt; }
	23	Double_t P() const { return fPt*TMath::CosH(fEta); }
61ac1db0	24
fb4e4852	25	Bool_t PxPyPz(Double_t p[3]) const { p[0]=Px();p[1]=Py();p[2]=Pz(); return 1; }
	26	Bool_t GetPxPyPz(Double_t p[3]) const { p[0]=Px();p[1]=Py();p[2]=Pz(); return 1; }
	27	Double_t Xv() const { return 0.; }
	28	Double_t Yv() const { return 0.; }
	29	Double_t Zv() const { return 0.; }
	30	Bool_t XvYvZv(Double_t x[3]) const { x[0]=0;x[1]=0;x[2]=0; return 1; }
	31	Double_t OneOverPt() const { return 1./fPt; }
	32	Double_t Phi() const { return fPhi; }
	33	Double_t Theta() const { return 2*TMath::ATan(TMath::Exp(-fEta)); }
	34	Double_t E() const { Double_t p=P(); return TMath::Sqrt(M()M()+pp); }
	35	Double_t M() const { return fM; }
	36	Double_t Eta() const { return fEta; }
	37	Double_t Y() const { return 0.5*TMath::Log((E()+Pz())/(E()-Pz())); }
	38	Short_t Charge() const { return (char)fQ;}
	39	Int_t GetLabel() const { return fLabel; }
	40	void SetLabel(Int_t label) { fLabel = label; }
	41	Byte_t GetTrackType() const { return fTrackType;}
	42	void SetTrackType(Byte_t type) { fTrackType = type;}
	43	Int_t PdgCode() const { return 0; }
dd233e4e	44	const Double_t *PID() const { return 0; }
fb4e4852	45	Int_t GetID() const { return 0; }
	46	UChar_t GetITSClusterMap() const { return 0; }
	47	Int_t GetEMCALcluster() const { return fClusId; }
	48	void SetEMCALcluster(Int_t id) { fClusId = id; }
	49	Double_t GetEtaEmc() const { return GetTrackEtaOnEMCal(); }
	50	Double_t GetPhiEmc() const { return GetTrackPhiOnEMCal(); }
	51	Bool_t IsEMCAL() const { return fEmcal; }
	52	ULong_t GetStatus() const { return 0; }
	53	Bool_t GetXYZ(Double_t *v) const { v[0]=0; v[1]=0; v[2]=0; return 0; }
6a33e0e9	54	using AliVTrack::GetXYZ;
fb4e4852	55	Double_t GetBz() const { return 0; }
	56	void GetBxByBz(Double_t b[3]) const { b[0]=0;b[1]=0;b[2]=0; }
	57	Bool_t GetCovarianceXYZPxPyPz(Double_t /cv/[21]) const { return 0; }
	58	AliVTrack *GetTrack() const { return fOrig; }
	59	void SetTrack(AliVTrack *t) { fOrig = t; }
	60	Double_t GetTrackPhiOnEMCal() const { return fPhiEmc ; }
	61	Double_t GetTrackEtaOnEMCal() const { return fEtaEmc ; }
	62	Double_t GetTrackPtOnEMCal() const { return fPtEmc ; }
	63	Double_t GetTrackPOnEMCal() const { return fPtEmc*TMath::CosH(fEtaEmc);}
	64	void SetTrackPhiEtaPtOnEMCal(Double_t eta, Double_t phi, Double_t pt) { fEtaEmc = eta; fPhiEmc = phi; fPtEmc = pt;}
	65	void Clear(Option_t * /option/ ="") { fClusId = -1; fOrig = 0; }
	66	Int_t Compare(const TObject* obj) const;
	67	Bool_t PropagateToDCA(const AliVVertex , Double_t, Double_t, Double_t , Double_t *) { return 0; }
eb556d0c	68	void SetFlag(UInt_t flag) {fFlag = flag;}
	69	UInt_t GetFlag() const {return fFlag;}
	70	void SetPrimary(Bool_t b = kTRUE){
	71	if(b)fFlag \|= AliAODMCParticle::kPrimary;
	72	else fFlag &= ~AliAODMCParticle::kPrimary;
	73	}
	74	Bool_t IsPrimary() const {return ((fFlag&AliAODMCParticle::kPrimary)==AliAODMCParticle::kPrimary);}
	75
	76	void SetPhysicalPrimary(Bool_t b = kTRUE){
	77	if(b)fFlag \|= AliAODMCParticle::kPhysicalPrim;
	78	else fFlag &= ~AliAODMCParticle::kPhysicalPrim;
	79	}
	80	Bool_t IsPhysicalPrimary() const {return ((fFlag&AliAODMCParticle::kPhysicalPrim)==AliAODMCParticle::kPhysicalPrim);}
	81
	82	void SetSecondaryFromWeakDecay(Bool_t b = kTRUE){
	83	if(b)fFlag \|= AliAODMCParticle::kSecondaryFromWeakDecay;
	84	else fFlag &= ~AliAODMCParticle::kSecondaryFromWeakDecay;
	85	}
	86	Bool_t IsSecondaryFromWeakDecay() const {return ((fFlag&AliAODMCParticle::kSecondaryFromWeakDecay)==AliAODMCParticle::kSecondaryFromWeakDecay);}
	87
	88	void SetSecondaryFromMaterial(Bool_t b = kTRUE){
	89	if(b)fFlag \|= AliAODMCParticle::kSecondaryFromMaterial;
	90	else fFlag &= ~AliAODMCParticle::kSecondaryFromMaterial;
	91	}
	92	Bool_t IsSecondaryFromMaterial() const {return ((fFlag&AliAODMCParticle::kSecondaryFromMaterial)==AliAODMCParticle::kSecondaryFromMaterial);}
	93
	94	Bool_t IsFromHIJING() const { return (fGeneratorIndex == 0); }
	95	void SetGeneratorIndex(Short_t i) {fGeneratorIndex = i;}
	96	Short_t GetGeneratorIndex() const {return fGeneratorIndex;}
b723beb6	97
fb4e4852	98	static void GetEtaPhiDiff(const AliVTrack t, const AliVCluster v, Double_t &phidiff, Double_t &etadiff);
fb4e4852	99	static Byte_t GetTrackType(const AliVTrack *t);
dd233e4e	100
dd233e4e	101	protected:
eb556d0c	102	Double32_t fPt; //[0,0,12] pt at vertex
	103	Double32_t fEta; //[-1,1,12] eta at vertex
	104	Double32_t fPhi; //[0,6.3,12] phi at vertex
	105	Double32_t fM; // mass
	106	Byte_t fQ; // charge
	107	Int_t fLabel; // label
	108	Byte_t fTrackType; // 0=global track; 1=w/o SPD, w/ ITS refit; 2=w/o SPD, w/o ITS refit
	109	Double32_t fEtaEmc; //[-1,1,12] eta at emcal surface
	110	Double32_t fPhiEmc; //[0,6.3,12] phi at emcal surface
	111	Double32_t fPtEmc; //[0,0,12] pt at emcal surface
	112	Bool_t fEmcal; // is true if track propagated to emcal
	113	UInt_t fFlag; //! Flag for indication of primary etc (MC)
	114	Short_t fGeneratorIndex; //! Index of generator in cocktail
	115	Short_t fClusId; //! cluster id of matched cluster; -1 if not set
	116	AliVTrack *fOrig; //! ptr to original track
dd233e4e	117
eb556d0c	118	ClassDef(AliPicoTrack, 8) // Pico track class
dd233e4e	119	};
dd233e4e	120	#endif