PWGGA/EMCALTasks/AliEmcalJet.h

   1 #ifndef AliEmcalJet_H
   2 #define AliEmcalJet_H
   3
   4 // $Id$
   5
   6 #include <TArrayS.h>
   7 #include <TLorentzVector.h>
   8 #include <TMath.h>
   9 #include "AliVParticle.h"
  10
  11 class AliEmcalJet : public AliVParticle
  12 {
  13  public:
  14   AliEmcalJet() : AliVParticle(), fPt(0), fEta(0), fPhi(0), fM(0), fNEF(0), fArea(0),
  15                   fMaxCPt(0), fMaxNPt(0), fClusterIDs(), fTrackIDs() {;}
  16   AliEmcalJet(Double_t pt, Double_t eta, Double_t phi, Double_t m);
  17   AliEmcalJet(Double_t px, Double_t py, Double_t pz);
  18   AliEmcalJet(const AliEmcalJet &jet);
  19   AliEmcalJet& operator=(const AliEmcalJet &jet);
  20
  21   Double_t    Px()                         const { return fPt*TMath::Cos(fPhi);  }
  22   Double_t    Py()                         const { return fPt*TMath::Cos(fPhi);  }
  23   Double_t    Pz()                         const { return fPt*TMath::SinH(fEta); }
  24   Double_t    Pt()                         const { return fPt;                   }
  25   Double_t    P()                          const { return fPt*TMath::CosH(fEta); }
  26   Bool_t      PxPyPz(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()+p*p); }
  35   Double_t    M()                          const { return 0.13957; }
  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 0;       }
  39   Int_t       GetLabel()                   const { return -1;      }
  40   Int_t       PdgCode()                    const { return 0;       }
  41   const Double_t *PID()                    const { return 0;       }
  42   void        GetMom(TLorentzVector &vec)  const;
  43   void        Print(Option_t* option = "") const;
  44
  45   Double_t    Area()                       const { return fArea;                     }
  46   Short_t     ClusterAt(Int_t idx)         const { return fClusterIDs.At(idx);       }
  47   UShort_t    GetNumberOfClusters()        const { return Nn();                      }
  48   UShort_t    GetNumberOfTracks()          const { return Nch();                     }
  49   Double_t    MaxClusterPt()               const { return fMaxNPt;                   }
  50   Double_t    MaxTrackPt()                 const { return fMaxCPt;                   }
  51   Double_t    NEF()                        const { return fNEF;                      }
  52   UShort_t    Nn()                         const { return fClusterIDs.GetSize();     }
  53   UShort_t    Nch()                        const { return fTrackIDs.GetSize();       }
  54   UShort_t    N()                          const { return Nch()+Nn();                }
  55   Short_t     TrackAt(Int_t idx)           const { return fTrackIDs.At(idx);         }
  56
  57   void        AddClusterAt(Int_t clus, Int_t idx){ fClusterIDs.AddAt(clus, idx);     }
  58   void        AddTrackAt(Int_t track, Int_t idx) { fTrackIDs.AddAt(track, idx);      }
  59   void        Clear(Option_t */*option*/="")     { fClusterIDs.Set(0);
  60                                                    fTrackIDs.Set(0);                 }
  61   void        SetArea(Double_t a)                { fArea   = a;                      }
  62   void        SetMaxClusterPt(Double32_t t)      { fMaxNPt = t;                      }
  63   void        SetMaxTrackPt(Double32_t t)        { fMaxCPt = t;                      }
  64   void        SetNEF(Double_t nef)               { fNEF    = nef;                    }
  65   void        SetNumberOfClusters(Int_t n)       { fClusterIDs.Set(n);               }
  66   void        SetNumberOfTracks(Int_t n)         { fTrackIDs.Set(n);                 }
  67   void        SortConstituents();
  68
  69  protected:
  70   Double32_t  fPt;           //[0,0,12]   pt
  71   Double32_t  fEta;          //[-1,1,12]  eta
  72   Double32_t  fPhi;          //[0,6.3,12] phi
  73   Double32_t  fM;            //[0,0,8]    mass
  74   Double32_t  fNEF;          //[0,1,8]    neutral energy fraction
  75   Double32_t  fArea;         //[0,0,12]   area
  76   Double32_t  fMaxCPt;       //[0,0,12]   pt of maximum track
  77   Double32_t  fMaxNPt;       //[0,0,12]   pt of maximum cluster
  78   TArrayS     fClusterIDs;   //           array of cluster constituents
  79   TArrayS     fTrackIDs;     //           array of track constituents
  80
  81   ClassDef(AliEmcalJet,2) // ESD jet class in cylindrical coordinates
  82 };
  83 #endif