// $Id$

#include "AliESDJet.h"

ClassImp(AliESDJet)

//__________________________________________________________________________________________________
AliESDJet::AliESDJet(Double_t px, Double_t py, Double_t pz) 
  : AliVParticle(), 
    fPt(TMath::Sqrt(px*px+py*py)), 
    fEta(TMath::ASinH(pz/fPt)),
    fPhi(0), fM(0), fNEF(0), fArea(0), fNch(0), fNn(0),
    fMaxCPt(0), fMaxNPt(0)
{    
  // Constructor.

  if (fPt != 0) {
    fPhi = TMath::ATan2(py, px);
    if (fPhi<0.) 
      fPhi += 2. * TMath::Pi();
  }
}

//_________________________________________________________________________________________________
AliESDJet::AliESDJet(Double_t pt, Double_t eta, Double_t phi, Double_t m) :
  AliVParticle(), 
  fPt(pt), fEta(eta), fPhi(phi), 
  fM(m), fNEF(0), fArea(0), 
  fNch(0), fNn(0), 
  fMaxCPt(0), fMaxNPt(0) 
{
  // Constructor.

 if (fPhi<0.) 
   fPhi += 2. * TMath::Pi();
}

//_________________________________________________________________________________________________
AliESDJet::AliESDJet(const AliESDJet &jet) :
  AliVParticle(jet),
  fPt(jet.fPt), fEta(jet.fEta), fPhi(jet.fPhi), 
  fM(jet.fM), fNEF(jet.fNEF), fArea(jet.fArea), 
  fNch(jet.fNch), fNn(jet.fNn),
  fMaxCPt(jet.fMaxCPt), fMaxNPt(jet.fMaxNPt)
{
  // Constructor.
}

//_________________________________________________________________________________________________
AliESDJet &AliESDJet::operator=(const AliESDJet &jet)
{
  // Assignment operator.

  if (this!=&jet) {
    AliVParticle::operator=(jet);
    fPt     = jet.fPt;
    fEta    = jet.fEta;
    fPhi    = jet.fPhi;
    fM      = jet.fM; 
    fNEF    = jet.fNEF;
    fArea   = jet.fArea; 
    fNch    = jet.fNch; 
    fNn     = jet.fNn;
    fMaxCPt = jet.fMaxCPt; 
    fMaxNPt = jet.fMaxNPt;
  }

  return *this;
}

//__________________________________________________________________________________________________
void AliESDJet::GetMom(TLorentzVector &vec) const
{
  // Return momentum as four vector.

  Double_t p = fPt *TMath::CosH(fEta);
  vec.SetPtEtaPhiE(fPt,fEta,fPhi,TMath::Sqrt(p*p+fM*fM));
}

//__________________________________________________________________________________________________
void AliESDJet::Print(Option_t* /*option*/) const
{
  // Print jet information.

  printf("Jet pt=%.2f, eta=%.2f, phi=%.2f, area=%.2f, NEF=%.2f\n", fPt, fEta, fPhi, fArea, fNEF);
}