#ifndef ROOT_Math_GenVector_PxPyPzM4D
#define ROOT_Math_GenVector_PxPyPzM4D 1
#include "Math/GenVector/etaMax.h"
#include "Math/GenVector/GenVector_exception.h"
#if defined(__MAKECINT__) || defined(G__DICTIONARY)
#include "Math/GenVector/PtEtaPhiE4D.h"
#include "Math/GenVector/PxPyPzE4D.h"
#endif
#include <cmath>
namespace ROOT {
namespace Math {
template <class ScalarType = double>
class PxPyPzM4D {
public :
typedef ScalarType Scalar;
PxPyPzM4D() : fX(0), fY(0), fZ(0), fM(0) { }
PxPyPzM4D(Scalar x, Scalar y, Scalar z, Scalar m) :
fX(x), fY(y), fZ(z), fM(m) { }
template <class CoordSystem>
explicit PxPyPzM4D(const CoordSystem & v) :
fX( v.x() ), fY( v.y() ), fZ( v.z() ), fM(0)
{
fM = std::sqrt( v.t()*v.t() - P2() );
}
void SetCoordinates( const Scalar src[] )
{ fX=src[0]; fY=src[1]; fZ=src[2]; fM=src[3]; }
void GetCoordinates( Scalar dest[] ) const
{ dest[0] = fX; dest[1] = fY; dest[2] = fZ; dest[3] = fM; }
void SetCoordinates(Scalar x, Scalar y, Scalar z, Scalar m)
{ fX=x; fY=y; fZ=z; fM=m;}
void GetCoordinates(Scalar& x, Scalar& y, Scalar& z, Scalar& m) const
{ x=fX; y=fY; z=fZ; m=fM;}
Scalar Px() const { return fX;}
Scalar Py() const { return fY;}
Scalar Pz() const { return fZ;}
Scalar M() const { return fM; }
Scalar X() const { return fX;}
Scalar Y() const { return fY;}
Scalar Z() const { return fZ;}
Scalar E() const { return std::sqrt( P2() + M2() );}
Scalar T() const { return E();}
Scalar P2() const { return fX*fX + fY*fY + fZ*fZ; }
Scalar P() const { return std::sqrt(P2()); }
Scalar R() const { return P(); }
Scalar M2() const { return fM*fM;}
Scalar Mag2() const { return M2(); }
Scalar Mag() const { return M(); }
Scalar E2() const { return P2() + M2(); }
Scalar Pt2() const { return fX*fX + fY*fY;}
Scalar Perp2() const { return Pt2();}
Scalar Pt() const { return std::sqrt(Perp2());}
Scalar Perp() const { return Pt();}
Scalar Rho() const { return Pt();}
Scalar Mt2() const { return E2() - fZ*fZ; }
Scalar Mt() const {
Scalar mm = Mt2();
if (mm >= 0) {
return std::sqrt(mm);
} else {
GenVector_exception e ("PxPyPzM4D::Mt() - Tachyonic:\n"
" Pz^2 > E^2 so the transverse mass would be imaginary");
Throw(e);
return -std::sqrt(-mm);
}
}
Scalar Et2() const {
Scalar pt2 = Pt2();
return pt2 == 0 ? 0 : E2() * pt2/( pt2 + fZ*fZ );
}
Scalar Et() const {
Scalar etet = Et2();
return fM < 0.0 ? -std::sqrt(etet) : std::sqrt(etet);
}
Scalar Phi() const {
return (fX == 0.0 && fY == 0.0) ? 0.0 : std::atan2(fY,fX);
}
Scalar Theta() const {
return (fX == 0.0 && fY == 0.0 && fZ == 0.0) ? 0.0 : std::atan2(Pt(),fZ);
}
Scalar Eta() const {
Scalar rho = Pt();
if (rho > 0) {
Scalar z_scaled(fZ/rho);
return std::log(z_scaled+std::sqrt(z_scaled*z_scaled+1));
} else if (fZ==0) {
return 0;
} else if (fZ>0) {
return fZ + etaMax<Scalar>( );
} else {
return fZ - etaMax<Scalar>( );
}
}
void SetPx( Scalar x) {
fX = x;
}
void SetPy( Scalar y) {
fY = y;
}
void SetPz( Scalar z) {
fZ = z;
}
void SetM( Scalar m) {
fM = m;
}
void Negate( ) { fX = -fX; fY = -fY; fZ = -fZ; fM = -fM;}
void Scale( const Scalar & a) {
fX *= a;
fY *= a;
fZ *= a;
fM *= a;
}
template <class AnyCoordSystem>
PxPyPzM4D & operator = (const AnyCoordSystem & v) {
fX = v.x();
fY = v.y();
fZ = v.z();
fM = std::sqrt ( v.t()*v.t() - P2() );
return *this;
}
bool operator == (const PxPyPzM4D & rhs) const {
return fX == rhs.fX && fY == rhs.fY && fZ == rhs.fZ && fM == rhs.fM;
}
bool operator != (const PxPyPzM4D & rhs) const {return !(operator==(rhs));}
Scalar x() const { return X(); }
Scalar y() const { return Y(); }
Scalar z() const { return Z(); }
Scalar t() const { return E(); }
#if defined(__MAKECINT__) || defined(G__DICTIONARY)
void SetPt(Scalar pt) {
GenVector_exception e("PxPyPzM4D::SetPt() is not supposed to be called");
Throw(e);
PtEtaPhiE4D<Scalar> v(*this); v.SetPt(pt); *this = PxPyPzM4D<Scalar>(v);
}
void SetEta(Scalar eta) {
GenVector_exception e("PxPyPzM4D::SetEta() is not supposed to be called");
Throw(e);
PtEtaPhiE4D<Scalar> v(*this); v.SetEta(eta); *this = PxPyPzM4D<Scalar>(v);
}
void SetPhi(Scalar phi) {
GenVector_exception e("PxPyPzM4D::SetPhi() is not supposed to be called");
Throw(e);
PtEtaPhiE4D<Scalar> v(*this); v.SetPhi(phi); *this = PxPyPzM4D<Scalar>(v);
}
void SetE(Scalar t) {
GenVector_exception e("PxPyPzM4D::SetM() is not supposed to be called");
Throw(e);
PxPyPzE4D<Scalar> v(*this); v.SetE(t);
*this = PxPyPzM4D<Scalar>(v);
}
#endif
private:
Scalar fX;
Scalar fY;
Scalar fZ;
Scalar fM;
};
}
}
#endif // ROOT_Math_GenVector_PxPyPzM4D
ROOT page - Class index - Class Hierarchy - Top of the page
This page has been automatically generated. If you have any comments or suggestions about the page layout send a mail to ROOT support, or contact the developers with any questions or problems regarding ROOT.