[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliEmcalJet.h

#ifndef AliEmcalJet_H
#define AliEmcalJet_H

// $Id$

#include <TArrayS.h>
#include <TLorentzVector.h>
#include <TMath.h>
#include <TClonesArray.h>

#include "AliVParticle.h"
#include "AliVCluster.h"

class AliEmcalJet : public AliVParticle
{
 public:
  AliEmcalJet();
  AliEmcalJet(Double_t px, Double_t py, Double_t pz);
  AliEmcalJet(Double_t pt, Double_t eta, Double_t phi, Double_t m);
  AliEmcalJet(const AliEmcalJet &jet); 
  AliEmcalJet& operator=(const AliEmcalJet &jet);

  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;         }
  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 0.13957; }
  Double_t          Eta()                        const { return fEta;    }
  Double_t          Y()                          const { return 0.5*TMath::Log((E()+Pz())/(E()-Pz()));    }
  Short_t           Charge()                     const { return 0;       }
  Int_t             GetLabel()                   const { return -1;      }
  Int_t             PdgCode()                    const { return 0;       }
  const Double_t   *PID()                        const { return 0;       }
  void              GetMom(TLorentzVector &vec)  const;
  void              Print(Option_t* option = "") const;

  Double_t          Area()                       const { return fArea;                     }
  Double_t          AreaPt()                     const { return fArea;                     }
  Double_t          AreaEta()                    const { return fAreaEta;                  }
  Double_t          AreaPhi()                    const { return fAreaPhi;                  }
  Double_t          AreaEmc()                    const { return fAreaEmc;                  }
  Bool_t            AxisInEmcal()                const { return fAxisInEmcal;              }
  Int_t             Compare(const TObject* obj)  const;
  AliEmcalJet*      ClosestJet()                 const { return fClosestJets[0];           }
  Double_t          ClosestJetDistance()         const { return fClosestJetsDist[0];       }
  Short_t           ClusterAt(Int_t idx)         const { return fClusterIDs.At(idx);       }
  AliVCluster      *ClusterAt(Int_t idx, TClonesArray *ca)  const { if (!ca) return 0; return dynamic_cast<AliVCluster*>(ca->At(ClusterAt(idx))); }
  UShort_t          GetNumberOfClusters()        const { return fClusterIDs.GetSize();     }
  UShort_t          GetNumberOfTracks()          const { return fTrackIDs.GetSize();       }
  UShort_t          GetNumberOfConstituents()    const { return GetNumberOfClusters()+GetNumberOfTracks();       }
  Double_t          FracEmcalArea()              const { return fAreaEmc/fArea;            }
  Bool_t            IsInsideEmcal()              const { return (fAreaEmc/fArea>0.999);    }
  Bool_t            IsInEmcal()                  const { return (fAreaEmc>0);              }
  Bool_t            IsMC()                       const { return (Bool_t)(MCPt() > 0);      }
  Bool_t            IsSortable()                 const { return kTRUE;                     }
  Double_t          MaxNeutralPt()               const { return fMaxNPt;                   }
  Double_t          MaxChargedPt()               const { return fMaxCPt;                   }
  Double_t          NEF()                        const { return fNEF;                      }
  UShort_t          Nn()                         const { return fNn;                       }
  UShort_t          Nch()                        const { return fNch;                      }
  UShort_t          N()                          const { return Nch()+Nn();                }
  Int_t             NEmc()                       const { return fNEmc;                     }
  Double_t          MCPt()                       const { return fMCPt;                     }
  AliEmcalJet*      MatchedJet()                 const { return fMatched < 2 ? fClosestJets[fMatched] : 0; }
  Double_t          MaxClusterPt()               const { return MaxNeutralPt();            }
  Double_t          MaxTrackPt()                 const { return MaxChargedPt();            }
  Double_t          MaxPartPt()                  const { return fMaxCPt < fMaxNPt ? fMaxNPt : fMaxCPt;     }
  Double_t          PtEmc()                      const { return fPtEmc;                    }
  Double_t          PtSub()                      const { return fPtSub;                    }
  Double_t          PtSub(Double_t rho)          const { return fPt - fArea*rho;           }
  Double_t          PtSubVect(Double_t rho)      const;
  AliEmcalJet*      SecondClosestJet()           const { return fClosestJets[1];           }
  Double_t          SecondClosestJetDistance()   const { return fClosestJetsDist[1];       }
  Short_t           TrackAt(Int_t idx)           const { return fTrackIDs.At(idx);         }
  AliVParticle     *TrackAt(Int_t idx, TClonesArray *ta)   const { if (!ta) return 0; return dynamic_cast<AliVParticle*>(ta->At(TrackAt(idx))); } 

  void              AddClusterAt(Int_t clus, Int_t idx){ fClusterIDs.AddAt(clus, idx);     }
  void              AddTrackAt(Int_t track, Int_t idx) { fTrackIDs.AddAt(track, idx);      }
  void              Clear(Option_t */*option*/="")     { fClusterIDs.Set(0); fTrackIDs.Set(0); fClosestJets[0] = 0; fClosestJets[1] = 0; 
                                                         fClosestJetsDist[0] = 0; fClosestJetsDist[1] = 0; fMatched = 0; fPtSub = 0; }
  void              SetArea(Double_t a)                { fArea    = a;                     }
  void              SetAreaEta(Double_t a)             { fAreaEta = a;                     }
  void              SetAreaPhi(Double_t a)             { fAreaPhi = a;                     }
  void              SetAreaEmc(Double_t a)             { fAreaEmc = a;                     }
  void              SetAxisInEmcal(Bool_t b)           { fAxisInEmcal = b;                 }
  void              SetMaxNeutralPt(Double32_t t)      { fMaxNPt  = t;                     }
  void              SetMaxChargedPt(Double32_t t)      { fMaxCPt  = t;                     }
  void              SetNEF(Double_t nef)               { fNEF     = nef;                   }
  void              SetNumberOfClusters(Int_t n)       { fClusterIDs.Set(n);               }
  void              SetNumberOfTracks(Int_t n)         { fTrackIDs.Set(n);                 }
  void              SetNumberOfCharged(Int_t n)        { fNch = n;                         }
  void              SetNumberOfNeutrals(Int_t n)       { fNn = n;                          }
  void              SetMCPt(Double_t p)                { fMCPt = p;                        }
  void              SortConstituents();
  void              SetClosestJet(AliEmcalJet *j, Double_t d)       { fClosestJets[0] = j; fClosestJetsDist[0] = d; }
  void              SetSecondClosestJet(AliEmcalJet *j, Double_t d) { fClosestJets[1] = j; fClosestJetsDist[1] = d; }
  void              SetMatchedToClosest()                           { fMatched   = 0;      }
  void              SetMatchedToSecondClosest()                     { fMatched   = 1;      }
  void              SetNEmc(Int_t n)                                { fNEmc      = n;      }
  void              SetPtEmc(Double_t pt)                           { fPtEmc     = pt;     }
  void              SetPtSub(Double_t ps)                           { fPtSub     = ps;     } 
  void              SetPtSubVect(Double_t ps)                       { fPtVectSub = ps;     } 

 protected:
  Double32_t        fPt;                  //[0,0,12]   pt 
  Double32_t        fEta;                 //[-1,1,12]  eta
  Double32_t        fPhi;                 //[0,6.3,12] phi
  Double32_t        fM;                   //[0,0,8]    mass
  Double32_t        fNEF;                 //[0,1,8]    neutral energy fraction
  Double32_t        fArea;                //[0,0,12]   area
  Double32_t        fAreaEta;             //[0,0,12]   area eta
  Double32_t        fAreaPhi;             //[0,0,12]   area phi
  Double32_t        fAreaEmc;             //[0,0,12]   area on EMCAL surface (determined from ghosts)
  Bool_t            fAxisInEmcal;         //           =true if jet axis inside EMCAL acceptance
  Double32_t        fMaxCPt;              //[0,0,12]   pt of maximum charged constituent
  Double32_t        fMaxNPt;              //[0,0,12]   pt of maximum neutral constituent
  Double32_t        fMCPt;                //           pt from MC particles contributing to the jet
  Int_t             fNn;                  //           number of neutral constituents
  Int_t             fNch;                 //           number of charged constituents
  Double32_t        fPtEmc;               //[0,0,12]   pt in EMCAL acceptance
  Int_t             fNEmc;                //           number of constituents in EMCAL acceptance
  TArrayS           fClusterIDs;          //           array of cluster constituents  
  TArrayS           fTrackIDs;            //           array of track constituents   
  AliEmcalJet      *fClosestJets[2];      //!          if this is MC it contains the two closest detector level jets in order of distance and viceversa
  Double32_t        fClosestJetsDist[2];  //!          distance to closest jets (see above)
  UShort_t          fMatched;             //!          0,1 if it is matched with one of the closest jets; 2 if it is not matched
  Double_t          fPtSub;               //!          background subtracted pt (not stored set from outside) 
  Double_t          fPtVectSub;           //!          background vector subtracted pt (not stored set from outside) 

  ClassDef(AliEmcalJet,9) // Emcal jet class in cylindrical coordinates
};
#endif
Commit	Line	Data
914d486c	1	#ifndef AliEmcalJet_H
914d486c	2	#define AliEmcalJet_H
7df864a3	3
688670de	4	// $Id$
688670de	5
f8087a81	6	#include <TArrayS.h>
7efbea04	7	#include <TLorentzVector.h>
7efbea04	8	#include <TMath.h>
a55e4f1d	9	#include <TClonesArray.h>
a55e4f1d	10
f8087a81	11	#include "AliVParticle.h"
a55e4f1d	12	#include "AliVCluster.h"
7efbea04	13
914d486c	14	class AliEmcalJet : public AliVParticle
7efbea04	15	{
7df864a3	16	public:
96919f91	17	AliEmcalJet();
914d486c	18	AliEmcalJet(Double_t px, Double_t py, Double_t pz);
101cefde	19	AliEmcalJet(Double_t pt, Double_t eta, Double_t phi, Double_t m);
914d486c	20	AliEmcalJet(const AliEmcalJet &jet);
914d486c	21	AliEmcalJet& operator=(const AliEmcalJet &jet);
7efbea04	22
a55e4f1d	23	Double_t Px() const { return fPt*TMath::Cos(fPhi); }
da1bd0e7	24	Double_t Py() const { return fPt*TMath::Sin(fPhi); }
a55e4f1d	25	Double_t Pz() const { return fPt*TMath::SinH(fEta); }
	26	Double_t Pt() const { return fPt; }
	27	Double_t P() const { return fPt*TMath::CosH(fEta); }
	28	Bool_t PxPyPz(Double_t p[3]) const { p[0]=Px();p[1]=Py();p[2]=Pz(); return 1; }
	29	Double_t Xv() const { return 0.; }
	30	Double_t Yv() const { return 0.; }
	31	Double_t Zv() const { return 0.; }
	32	Bool_t XvYvZv(Double_t x[3]) const { x[0]=0;x[1]=0;x[2]=0; return 1; }
	33	Double_t OneOverPt() const { return 1./fPt; }
	34	Double_t Phi() const { return fPhi; }
	35	Double_t Theta() const { return 2*TMath::ATan(TMath::Exp(-fEta)); }
	36	Double_t E() const { Double_t p=P(); return TMath::Sqrt(M()M()+pp); }
	37	Double_t M() const { return 0.13957; }
	38	Double_t Eta() const { return fEta; }
	39	Double_t Y() const { return 0.5*TMath::Log((E()+Pz())/(E()-Pz())); }
	40	Short_t Charge() const { return 0; }
	41	Int_t GetLabel() const { return -1; }
	42	Int_t PdgCode() const { return 0; }
	43	const Double_t *PID() const { return 0; }
	44	void GetMom(TLorentzVector &vec) const;
	45	void Print(Option_t* option = "") const;
688670de	46
a55e4f1d	47	Double_t Area() const { return fArea; }
db8a0184	48	Double_t AreaPt() const { return fArea; }
	49	Double_t AreaEta() const { return fAreaEta; }
	50	Double_t AreaPhi() const { return fAreaPhi; }
101cefde	51	Double_t AreaEmc() const { return fAreaEmc; }
101cefde	52	Bool_t AxisInEmcal() const { return fAxisInEmcal; }
db8a0184	53	Int_t Compare(const TObject* obj) const;
	54	AliEmcalJet* ClosestJet() const { return fClosestJets[0]; }
	55	Double_t ClosestJetDistance() const { return fClosestJetsDist[0]; }
a55e4f1d	56	Short_t ClusterAt(Int_t idx) const { return fClusterIDs.At(idx); }
8895e61b	57	AliVCluster ClusterAt(Int_t idx, TClonesArray ca) const { if (!ca) return 0; return dynamic_cast<AliVCluster*>(ca->At(ClusterAt(idx))); }
db8a0184	58	UShort_t GetNumberOfClusters() const { return fClusterIDs.GetSize(); }
96919f91	59	UShort_t GetNumberOfTracks() const { return fTrackIDs.GetSize(); }
624bef5b	60	UShort_t GetNumberOfConstituents() const { return GetNumberOfClusters()+GetNumberOfTracks(); }
101cefde	61	Double_t FracEmcalArea() const { return fAreaEmc/fArea; }
101cefde	62	Bool_t IsInsideEmcal() const { return (fAreaEmc/fArea>0.999); }
101cefde	63	Bool_t IsInEmcal() const { return (fAreaEmc>0); }
db8a0184	64	Bool_t IsMC() const { return (Bool_t)(MCPt() > 0); }
db8a0184	65	Bool_t IsSortable() const { return kTRUE; }
96919f91	66	Double_t MaxNeutralPt() const { return fMaxNPt; }
96919f91	67	Double_t MaxChargedPt() const { return fMaxCPt; }
a55e4f1d	68	Double_t NEF() const { return fNEF; }
96919f91	69	UShort_t Nn() const { return fNn; }
96919f91	70	UShort_t Nch() const { return fNch; }
a55e4f1d	71	UShort_t N() const { return Nch()+Nn(); }
8895e61b	72	Int_t NEmc() const { return fNEmc; }
96919f91	73	Double_t MCPt() const { return fMCPt; }
96919f91	74	AliEmcalJet* MatchedJet() const { return fMatched < 2 ? fClosestJets[fMatched] : 0; }
96919f91	75	Double_t MaxClusterPt() const { return MaxNeutralPt(); }
96919f91	76	Double_t MaxTrackPt() const { return MaxChargedPt(); }
e44e8726	77	Double_t MaxPartPt() const { return fMaxCPt < fMaxNPt ? fMaxNPt : fMaxCPt; }
8895e61b	78	Double_t PtEmc() const { return fPtEmc; }
	79	Double_t PtSub() const { return fPtSub; }
	80	Double_t PtSub(Double_t rho) const { return fPt - fArea*rho; }
db8a0184	81	Double_t PtSubVect(Double_t rho) const;
	82	AliEmcalJet* SecondClosestJet() const { return fClosestJets[1]; }
	83	Double_t SecondClosestJetDistance() const { return fClosestJetsDist[1]; }
	84	Short_t TrackAt(Int_t idx) const { return fTrackIDs.At(idx); }
	85	AliVParticle TrackAt(Int_t idx, TClonesArray ta) const { if (!ta) return 0; return dynamic_cast<AliVParticle*>(ta->At(TrackAt(idx))); }
35789a2d	86
a55e4f1d	87	void AddClusterAt(Int_t clus, Int_t idx){ fClusterIDs.AddAt(clus, idx); }
a55e4f1d	88	void AddTrackAt(Int_t track, Int_t idx) { fTrackIDs.AddAt(track, idx); }
038f0049	89	void Clear(Option_t /option*/="") { fClusterIDs.Set(0); fTrackIDs.Set(0); fClosestJets[0] = 0; fClosestJets[1] = 0;
038f0049	90	fClosestJetsDist[0] = 0; fClosestJetsDist[1] = 0; fMatched = 0; fPtSub = 0; }
101cefde	91	void SetArea(Double_t a) { fArea = a; }
db8a0184	92	void SetAreaEta(Double_t a) { fAreaEta = a; }
db8a0184	93	void SetAreaPhi(Double_t a) { fAreaPhi = a; }
101cefde	94	void SetAreaEmc(Double_t a) { fAreaEmc = a; }
101cefde	95	void SetAxisInEmcal(Bool_t b) { fAxisInEmcal = b; }
96919f91	96	void SetMaxNeutralPt(Double32_t t) { fMaxNPt = t; }
96919f91	97	void SetMaxChargedPt(Double32_t t) { fMaxCPt = t; }
101cefde	98	void SetNEF(Double_t nef) { fNEF = nef; }
a55e4f1d	99	void SetNumberOfClusters(Int_t n) { fClusterIDs.Set(n); }
a55e4f1d	100	void SetNumberOfTracks(Int_t n) { fTrackIDs.Set(n); }
96919f91	101	void SetNumberOfCharged(Int_t n) { fNch = n; }
	102	void SetNumberOfNeutrals(Int_t n) { fNn = n; }
	103	void SetMCPt(Double_t p) { fMCPt = p; }
a55e4f1d	104	void SortConstituents();
96919f91	105	void SetClosestJet(AliEmcalJet *j, Double_t d) { fClosestJets[0] = j; fClosestJetsDist[0] = d; }
96919f91	106	void SetSecondClosestJet(AliEmcalJet *j, Double_t d) { fClosestJets[1] = j; fClosestJetsDist[1] = d; }
db8a0184	107	void SetMatchedToClosest() { fMatched = 0; }
	108	void SetMatchedToSecondClosest() { fMatched = 1; }
	109	void SetNEmc(Int_t n) { fNEmc = n; }
	110	void SetPtEmc(Double_t pt) { fPtEmc = pt; }
	111	void SetPtSub(Double_t ps) { fPtSub = ps; }
	112	void SetPtSubVect(Double_t ps) { fPtVectSub = ps; }
629e03fc	113
7df864a3	114	protected:
43a9dcd0	115	Double32_t fPt; //[0,0,12] pt
	116	Double32_t fEta; //[-1,1,12] eta
	117	Double32_t fPhi; //[0,6.3,12] phi
	118	Double32_t fM; //[0,0,8] mass
	119	Double32_t fNEF; //[0,1,8] neutral energy fraction
	120	Double32_t fArea; //[0,0,12] area
db8a0184	121	Double32_t fAreaEta; //[0,0,12] area eta
db8a0184	122	Double32_t fAreaPhi; //[0,0,12] area phi
43a9dcd0	123	Double32_t fAreaEmc; //[0,0,12] area on EMCAL surface (determined from ghosts)
	124	Bool_t fAxisInEmcal; // =true if jet axis inside EMCAL acceptance
	125	Double32_t fMaxCPt; //[0,0,12] pt of maximum charged constituent
	126	Double32_t fMaxNPt; //[0,0,12] pt of maximum neutral constituent
	127	Double32_t fMCPt; // pt from MC particles contributing to the jet
	128	Int_t fNn; // number of neutral constituents
	129	Int_t fNch; // number of charged constituents
8895e61b	130	Double32_t fPtEmc; //[0,0,12] pt in EMCAL acceptance
8895e61b	131	Int_t fNEmc; // number of constituents in EMCAL acceptance
43a9dcd0	132	TArrayS fClusterIDs; // array of cluster constituents
	133	TArrayS fTrackIDs; // array of track constituents
	134	AliEmcalJet *fClosestJets[2]; //! if this is MC it contains the two closest detector level jets in order of distance and viceversa
	135	Double32_t fClosestJetsDist[2]; //! distance to closest jets (see above)
	136	UShort_t fMatched; //! 0,1 if it is matched with one of the closest jets; 2 if it is not matched
8895e61b	137	Double_t fPtSub; //! background subtracted pt (not stored set from outside)
db8a0184	138	Double_t fPtVectSub; //! background vector subtracted pt (not stored set from outside)
7df864a3	139
624bef5b	140	ClassDef(AliEmcalJet,9) // Emcal jet class in cylindrical coordinates
7df864a3	141	};
f472e642	142	#endif