[u/mrichter/AliRoot.git] / PWGGA / EMCALTasks / AliCaloTrackParticle.h

#ifndef ALICALOTRACKPARTICLE_H
#define ALICALOTRACKPARTICLE_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */
/* $Id: $ */
 
//_________________________________________________________________________
// Class to fill photon and track for correlation 

#include "TLorentzVector.h"

class AliCaloTrackParticle :public TLorentzVector{
  
 public:
  
  AliCaloTrackParticle() ;
  AliCaloTrackParticle(Double_t px,Double_t py,Double_t pz,Double_t E) ; 
  ~AliCaloTrackParticle(){} 

   Int_t    GetChargedSign() const { return fChargedSign ;}

   Float_t  GetLambda0(void)        const {return fL0      ;}
   Float_t  GetLambda1(void)        const {return fL1      ;}
   Int_t    GetModule(void)         const {return fModule  ;}
   Double_t GetDistBad(void)        const {return fBadDist ;}
   Int_t    GetNCells(void)         const {return fNCells  ;}
   Double_t GetTOF(void)            const {return fClusterTime;}
   Int_t    GetClusterID(void)      const {return fClusterID;}
   Int_t    GetAODClusterID(void)   const {return fAODClusterID;}

   Float_t  GetPhotonPairDTime(void)    const {return fPhotonPairDTime   ;} 
   Int_t    GetPhotonPairDModule(void)  const {return fPhotonPairDModule ;} 
   Float_t  GetPhotonPairAsy(void)      const {return fPhotonPairAsy     ;}
   Float_t  GetPhotonPairAngle(void)    const {return fPhotonPairAngle   ;}
   Int_t    GetPhotonPairID(Int_t i)    const {if(i ==0 ) return fPhotonPairID0;   else if(i ==1) return fPhotonPairID1; else return -999;}
   Int_t    GetAODPhotonPairID(Int_t i)    const {if(i ==0 ) return fAODPhotonPairID0;   else if(i ==1) return fAODPhotonPairID1; else return -999;}

   Bool_t   IsInSSA(void)          const {return kInSSA           ;}
   Bool_t   IsInTrackMatched(void) const {return kInTrackMatched  ;}
   Bool_t   IsInTOF(void)          const {return kInTOF           ;}
   Bool_t   IsLeading(void)        const {return kIsLeading       ;}
   Bool_t   IsIsolated(void)       const {return kIsIsolated      ;}

   Bool_t   IsPIDOK(const Int_t fPid) const ;

   void SetChargedSign(Int_t n){fChargedSign = n;}
   
   void SetLambdas(Float_t l0,Float_t l1){fL0=l0,fL1=l1;}
   void SetModule(Int_t mod){fModule = mod;}
   void SetDistBad(Double_t nbad){fBadDist = nbad;}
   void SetNCells(Int_t ncell){fNCells = ncell;}
   void SetTOF(Double_t time){fClusterTime = time;}
   void SetClusterID(Int_t id){fClusterID = id;}
   void SetAODClusterID(Int_t aodid){fAODClusterID = aodid;}
   void SetSSABit(Bool_t ssa){kInSSA = ssa;} 
   void SetTOFBit(Bool_t tof){kInTOF = tof;} 
   void SetTrackMatchedBit(Bool_t matched){kInTrackMatched = matched;}
   void SetDistToBad(Int_t dist){fBadDist=dist;} 
   void SetPhotonPairDTime(Float_t pairtime){fPhotonPairDTime = pairtime;}
   void SetPhotonPairDModule(Int_t pairmodule){fPhotonPairDModule = pairmodule;}
   void SetPhotonPairAsy(Float_t asy){fPhotonPairAsy = asy;}
   void SetPhotonPairAngle(Float_t angle){fPhotonPairAngle = angle;}
   void SetPhotonPairID(Int_t id0, Int_t id1){fPhotonPairID0 =id0, fPhotonPairID1=id1 ; }
   void SetAODPhotonPairID(Int_t aodid0, Int_t aodid1){fAODPhotonPairID0 =aodid0, fAODPhotonPairID1=aodid1 ; }
   void SetIsLeading(Bool_t leading){kIsLeading = leading;}
   void SetIsolated(Bool_t iso){kIsIsolated = iso;}
private:

  Int_t     fChargedSign ;   //sign of charged track

  Float_t   fL0               ;
  Float_t   fL1               ;
  Int_t     fModule           ; 
  Int_t     fBadDist          ;
  Int_t     fNCells           ;
  Double_t  fClusterTime      ;
  Int_t     fClusterID        ;
  Int_t     fAODClusterID        ;
  Bool_t    kInSSA            ;   
  Bool_t    kInTOF            ;    
  Bool_t    kInTrackMatched   ;   
  Double_t  fPhotonPairDTime  ;
  Int_t     fPhotonPairDModule;
  Float_t   fPhotonPairAsy    ;
  Float_t   fPhotonPairAngle  ;
  Int_t     fPhotonPairID0    ;
  Int_t     fAODPhotonPairID0    ;
  Int_t     fPhotonPairID1    ;
  Int_t     fAODPhotonPairID1    ;
  Bool_t    kIsLeading        ;
  Bool_t    kIsIsolated       ;

  ClassDef(AliCaloTrackParticle,2)

};

#endif // #ifdef ALICALOTRACKPARTICLE_H
Commit	Line	Data
9c1fcdc6	1	#ifndef ALICALOTRACKPARTICLE_H
9c1fcdc6	2	#define ALICALOTRACKPARTICLE_H
364c8fcb	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5	/* $Id: $ */
	6
	7	//_________________________________________________________________________
	8	// Class to fill photon and track for correlation
	9
	10	#include "TLorentzVector.h"
	11
9c1fcdc6	12	class AliCaloTrackParticle :public TLorentzVector{
364c8fcb	13
	14	public:
	15
9c1fcdc6	16	AliCaloTrackParticle() ;
	17	AliCaloTrackParticle(Double_t px,Double_t py,Double_t pz,Double_t E) ;
	18	~AliCaloTrackParticle(){}
364c8fcb	19
	20	Int_t GetChargedSign() const { return fChargedSign ;}
	21
	22	Float_t GetLambda0(void) const {return fL0 ;}
	23	Float_t GetLambda1(void) const {return fL1 ;}
	24	Int_t GetModule(void) const {return fModule ;}
89d9553e	25	Double_t GetDistBad(void) const {return fBadDist ;}
364c8fcb	26	Int_t GetNCells(void) const {return fNCells ;}
	27	Double_t GetTOF(void) const {return fClusterTime;}
	28	Int_t GetClusterID(void) const {return fClusterID;}
	29	Int_t GetAODClusterID(void) const {return fAODClusterID;}
	30
	31	Float_t GetPhotonPairDTime(void) const {return fPhotonPairDTime ;}
	32	Int_t GetPhotonPairDModule(void) const {return fPhotonPairDModule ;}
	33	Float_t GetPhotonPairAsy(void) const {return fPhotonPairAsy ;}
	34	Float_t GetPhotonPairAngle(void) const {return fPhotonPairAngle ;}
	35	Int_t GetPhotonPairID(Int_t i) const {if(i ==0 ) return fPhotonPairID0; else if(i ==1) return fPhotonPairID1; else return -999;}
	36	Int_t GetAODPhotonPairID(Int_t i) const {if(i ==0 ) return fAODPhotonPairID0; else if(i ==1) return fAODPhotonPairID1; else return -999;}
	37
	38	Bool_t IsInSSA(void) const {return kInSSA ;}
	39	Bool_t IsInTrackMatched(void) const {return kInTrackMatched ;}
	40	Bool_t IsInTOF(void) const {return kInTOF ;}
	41	Bool_t IsLeading(void) const {return kIsLeading ;}
	42	Bool_t IsIsolated(void) const {return kIsIsolated ;}
	43
	44	Bool_t IsPIDOK(const Int_t fPid) const ;
	45
	46	void SetChargedSign(Int_t n){fChargedSign = n;}
	47
	48	void SetLambdas(Float_t l0,Float_t l1){fL0=l0,fL1=l1;}
	49	void SetModule(Int_t mod){fModule = mod;}
89d9553e	50	void SetDistBad(Double_t nbad){fBadDist = nbad;}
364c8fcb	51	void SetNCells(Int_t ncell){fNCells = ncell;}
	52	void SetTOF(Double_t time){fClusterTime = time;}
	53	void SetClusterID(Int_t id){fClusterID = id;}
	54	void SetAODClusterID(Int_t aodid){fAODClusterID = aodid;}
	55	void SetSSABit(Bool_t ssa){kInSSA = ssa;}
	56	void SetTOFBit(Bool_t tof){kInTOF = tof;}
	57	void SetTrackMatchedBit(Bool_t matched){kInTrackMatched = matched;}
	58	void SetDistToBad(Int_t dist){fBadDist=dist;}
	59	void SetPhotonPairDTime(Float_t pairtime){fPhotonPairDTime = pairtime;}
	60	void SetPhotonPairDModule(Int_t pairmodule){fPhotonPairDModule = pairmodule;}
	61	void SetPhotonPairAsy(Float_t asy){fPhotonPairAsy = asy;}
	62	void SetPhotonPairAngle(Float_t angle){fPhotonPairAngle = angle;}
	63	void SetPhotonPairID(Int_t id0, Int_t id1){fPhotonPairID0 =id0, fPhotonPairID1=id1 ; }
	64	void SetAODPhotonPairID(Int_t aodid0, Int_t aodid1){fAODPhotonPairID0 =aodid0, fAODPhotonPairID1=aodid1 ; }
	65	void SetIsLeading(Bool_t leading){kIsLeading = leading;}
	66	void SetIsolated(Bool_t iso){kIsIsolated = iso;}
	67	private:
	68
	69	Int_t fChargedSign ; //sign of charged track
	70
	71	Float_t fL0 ;
	72	Float_t fL1 ;
	73	Int_t fModule ;
	74	Int_t fBadDist ;
	75	Int_t fNCells ;
	76	Double_t fClusterTime ;
	77	Int_t fClusterID ;
	78	Int_t fAODClusterID ;
	79	Bool_t kInSSA ;
	80	Bool_t kInTOF ;
	81	Bool_t kInTrackMatched ;
	82	Double_t fPhotonPairDTime ;
	83	Int_t fPhotonPairDModule;
	84	Float_t fPhotonPairAsy ;
	85	Float_t fPhotonPairAngle ;
	86	Int_t fPhotonPairID0 ;
	87	Int_t fAODPhotonPairID0 ;
	88	Int_t fPhotonPairID1 ;
	89	Int_t fAODPhotonPairID1 ;
	90	Bool_t kIsLeading ;
	91	Bool_t kIsIsolated ;
	92
9c1fcdc6	93	ClassDef(AliCaloTrackParticle,2)
364c8fcb	94
	95	};
	96
9c1fcdc6	97	#endif // #ifdef ALICALOTRACKPARTICLE_H
364c8fcb	98
364c8fcb	99