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

#ifndef AliPicoTrack_H
#define AliPicoTrack_H

// $Id$

#include "AliVTrack.h"
#include <TMath.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() {;}
  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; }
  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; }
  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; }
  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;}
  Int_t    Compare(const TObject* obj) const;
  Bool_t   PropagateToDCA(const AliVVertex *, Double_t, Double_t, Double_t *, Double_t *) { return 0; }
  void     SetEMCALcluster(Int_t id)         { fClusId = id;   }

  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
  Short_t          fClusId;   //!          cluster id of matched cluster; -1 if not set

  ClassDef(AliPicoTrack, 6) // Pico track class
};
#endif
Commit	Line	Data
dd233e4e	1	#ifndef AliPicoTrack_H
	2	#define AliPicoTrack_H
	3
59807387	4	// $Id$
dd233e4e	5
	6	#include "AliVTrack.h"
	7	#include <TMath.h>
b723beb6	8	class AliVCluster;
dd233e4e	9
	10	class AliPicoTrack: public AliVTrack {
	11	public:
	12	AliPicoTrack();
39799f2a	13	AliPicoTrack(Double_t pt, Double_t eta, Double_t phi, Byte_t q, Int_t label, Byte_t type,
de1c0913	14	Double_t etaemc=0, Double_t phiemc=0, Double_t ptemc=0, Bool_t ise=0, Double_t mass=0.13957);
dd233e4e	15	~AliPicoTrack() {;}
	16	AliPicoTrack(const AliPicoTrack &pc);
	17	AliPicoTrack &operator=(const AliPicoTrack &pc);
	18
	19	Double_t Px() const { return fPt*TMath::Cos(fPhi); }
675bb119	20	Double_t Py() const { return fPt*TMath::Sin(fPhi); }
dd233e4e	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); }
	24	Bool_t PxPyPz(Double_t p[3]) const { p[0]=Px();p[1]=Py();p[2]=Pz(); return 1; }
c92284d5	25	Bool_t GetPxPyPz(Double_t p[3]) const { p[0]=Px();p[1]=Py();p[2]=Pz(); return 1; }
dd233e4e	26	Double_t Xv() const { return 0.; }
	27	Double_t Yv() const { return 0.; }
	28	Double_t Zv() const { return 0.; }
	29	Bool_t XvYvZv(Double_t x[3]) const { x[0]=0;x[1]=0;x[2]=0; return 1; }
	30	Double_t OneOverPt() const { return 1./fPt; }
	31	Double_t Phi() const { return fPhi; }
	32	Double_t Theta() const { return 2*TMath::ATan(TMath::Exp(-fEta)); }
	33	Double_t E() const { Double_t p=P(); return TMath::Sqrt(M()M()+pp); }
de1c0913	34	Double_t M() const { return fM; }
dd233e4e	35	Double_t Eta() const { return fEta; }
dd233e4e	36	Double_t Y() const { return 0.5*TMath::Log((E()+Pz())/(E()-Pz())); }
dae383aa	37	Short_t Charge() const { return (char)fQ;}
dd233e4e	38	Int_t GetLabel() const { return fLabel; }
39799f2a	39	void SetLabel(Int_t label) { fLabel = label; }
1f12855f	40	Byte_t GetTrackType() const { return fTrackType;}
5be3857d	41	void SetTrackType(Byte_t type) { fTrackType = type;}
dd233e4e	42	Int_t PdgCode() const { return 0; }
	43	const Double_t *PID() const { return 0; }
	44	Int_t GetID() const { return 0; }
	45	UChar_t GetITSClusterMap() const { return 0; }
b723beb6	46	Int_t GetEMCALcluster() const { return fClusId; }
442d4a40	47	Double_t GetEtaEmc() const { return GetTrackEtaOnEMCal(); }
442d4a40	48	Double_t GetPhiEmc() const { return GetTrackPhiOnEMCal(); }
dd233e4e	49	Bool_t IsEMCAL() const { return fEmcal; }
b723beb6	50	ULong_t GetStatus() const { return 0; }
dd233e4e	51	Bool_t GetXYZ(Double_t *v) const { v[0]=0; v[1]=0; v[2]=0; return 0; }
b723beb6	52	Double_t GetBz() const { return 0; }
	53	void GetBxByBz(Double_t b[3]) const { b[0]=0;b[1]=0;b[2]=0; }
	54	Bool_t GetCovarianceXYZPxPyPz(Double_t /cv/[21]) const { return 0; }
213a3a51	55	Double_t GetTrackPhiOnEMCal() const { return fPhiEmc ; }
	56	Double_t GetTrackEtaOnEMCal() const { return fEtaEmc ; }
	57	Double_t GetTrackPtOnEMCal() const { return fPtEmc ; }
	58	Double_t GetTrackPOnEMCal() const { return fPtEmc*TMath::CosH(fEtaEmc);}
	59	void SetTrackPhiEtaPtOnEMCal(Double_t eta, Double_t phi, Double_t pt) { fEtaEmc = eta; fPhiEmc = phi; fPtEmc = pt;}
dd233e4e	60	Int_t Compare(const TObject* obj) const;
dd233e4e	61	Bool_t PropagateToDCA(const AliVVertex , Double_t, Double_t, Double_t , Double_t *) { return 0; }
b723beb6	62	void SetEMCALcluster(Int_t id) { fClusId = id; }
b723beb6	63
59807387	64	static void GetEtaPhiDiff(const AliVTrack t, const AliVCluster v, Double_t &phidiff, Double_t &etadiff);
59807387	65	static Byte_t GetTrackType(const AliVTrack *t);
dd233e4e	66
	67	protected:
	68	Double32_t fPt; //[0,0,12] pt at vertex
	69	Double32_t fEta; //[-1,1,12] eta at vertex
	70	Double32_t fPhi; //[0,6.3,12] phi at vertex
de1c0913	71	Double32_t fM; // mass
dd233e4e	72	Byte_t fQ; // charge
1f12855f	73	Int_t fLabel; // label
1f12855f	74	Byte_t fTrackType;// 0=global track; 1=w/o SPD, w/ ITS refit; 2=w/o SPD, w/o ITS refit
dd233e4e	75	Double32_t fEtaEmc; //[-1,1,12] eta at emcal surface
dd233e4e	76	Double32_t fPhiEmc; //[0,6.3,12] phi at emcal surface
56bd3193	77	Double32_t fPtEmc; //[0,0,12] pt at emcal surface
dd233e4e	78	Bool_t fEmcal; // is true if track propagated to emcal
b723beb6	79	Short_t fClusId; //! cluster id of matched cluster; -1 if not set
dd233e4e	80
de1c0913	81	ClassDef(AliPicoTrack, 6) // Pico track class
dd233e4e	82	};
dd233e4e	83	#endif