// @(#)root/mathcore:$Id: 2fd203872f434b1e4e74933903abb3429494ea6f $ // Authors: W. Brown, M. Fischler, L. Moneta 2005 /********************************************************************** * * * Copyright (c) 2005 , LCG ROOT MathLib Team * * & FNAL LCG ROOT Mathlib Team * * * * * **********************************************************************/ // Header file for class Cartesian3D // // Created by: Lorenzo Moneta at Mon May 30 11:16:56 2005 // Major revamp: M. FIschler at Wed Jun 8 2005 // // Last update: $ID: $ // #ifndef ROOT_Math_GenVector_Cartesian3D #define ROOT_Math_GenVector_Cartesian3D 1 #ifndef ROOT_Math_GenVector_Polar3Dfwd #include "Math/GenVector/Polar3Dfwd.h" #endif #ifndef ROOT_Math_Math #include "Math/Math.h" #endif #include #ifndef ROOT_Math_GenVector_eta #include "Math/GenVector/eta.h" #endif namespace ROOT { namespace Math { //__________________________________________________________________________________________ /** Class describing a 3D cartesian coordinate system (x, y, z coordinates) @ingroup GenVector */ template class Cartesian3D { public : typedef T Scalar; /** Default constructor with x=y=z=0 */ Cartesian3D() : fX(0.0), fY(0.0), fZ(0.0) { } /** Constructor from x,y,z coordinates */ Cartesian3D(Scalar xx, Scalar yy, Scalar zz) : fX(xx), fY(yy), fZ(zz) { } /** Construct from any Vector or coordinate system implementing X(), Y() and Z() */ template explicit Cartesian3D(const CoordSystem & v) : fX(v.X()), fY(v.Y()), fZ(v.Z()) { } // for g++ 3.2 and 3.4 on 32 bits found that the compiler generated copy ctor and assignment are much slower // re-implement them ( there is no no need to have them with g++4) /** copy constructor */ Cartesian3D(const Cartesian3D & v) : fX(v.X()), fY(v.Y()), fZ(v.Z()) { } /** assignment operator */ Cartesian3D & operator= (const Cartesian3D & v) { fX = v.x(); fY = v.y(); fZ = v.z(); return *this; } /** Set internal data based on an array of 3 Scalar numbers */ void SetCoordinates( const Scalar src[] ) { fX=src[0]; fY=src[1]; fZ=src[2]; } /** get internal data into an array of 3 Scalar numbers */ void GetCoordinates( Scalar dest[] ) const { dest[0] = fX; dest[1] = fY; dest[2] = fZ; } /** Set internal data based on 3 Scalar numbers */ void SetCoordinates(Scalar xx, Scalar yy, Scalar zz) { fX=xx; fY=yy; fZ=zz; } /** get internal data into 3 Scalar numbers */ void GetCoordinates(Scalar& xx, Scalar& yy, Scalar& zz) const {xx=fX; yy=fY; zz=fZ;} Scalar X() const { return fX;} Scalar Y() const { return fY;} Scalar Z() const { return fZ;} Scalar Mag2() const { return fX*fX + fY*fY + fZ*fZ;} Scalar Perp2() const { return fX*fX + fY*fY ;} Scalar Rho() const { return std::sqrt( Perp2());} Scalar R() const { return std::sqrt( Mag2());} Scalar Theta() const { return (fX==0 && fY==0 && fZ==0) ? 0 : atan2(Rho(),Z());} Scalar Phi() const { return (fX==0 && fY==0) ? 0 : atan2(fY,fX);} // pseudorapidity Scalar Eta() const { return Impl::Eta_FromRhoZ ( Rho(),fZ); } /** set the x coordinate value keeping y and z constant */ void SetX(Scalar xx) { fX = xx; } /** set the y coordinate value keeping x and z constant */ void SetY(Scalar yy) { fY = yy; } /** set the z coordinate value keeping x and y constant */ void SetZ(Scalar zz) { fZ = zz; } /** set all values using cartesian coordinates */ void SetXYZ(Scalar xx, Scalar yy, Scalar zz) { fX=xx; fY=yy; fZ=zz; } /** scale the vector by a scalar quantity a */ void Scale(Scalar a) { fX *= a; fY *= a; fZ *= a; } /** negate the vector */ void Negate() { fX = -fX; fY = -fY; fZ = -fZ; } /** Assignment from any class implementing x(),y() and z() (can assign from any coordinate system) */ template Cartesian3D & operator = (const CoordSystem & v) { fX = v.x(); fY = v.y(); fZ = v.z(); return *this; } /** Exact equality */ bool operator == (const Cartesian3D & rhs) const { return fX == rhs.fX && fY == rhs.fY && fZ == rhs.fZ; } bool operator != (const Cartesian3D & rhs) const {return !(operator==(rhs));} // ============= Compatibility section ================== // The following make this coordinate system look enough like a CLHEP // vector that an assignment member template can work with either T x() const { return X();} T y() const { return Y();} T z() const { return Z(); } // ============= Overloads for improved speed ================== template explicit Cartesian3D( const Polar3D & v ) : fZ (v.Z()) { T rho = v.Rho(); // re-using this instead of calling v.X() and v.Y() // is the speed improvement fX = rho * std::cos(v.Phi()); fY = rho * std::sin(v.Phi()); } // Technical note: This works even though only Polar3Dfwd.h is // included (and in fact, including Polar3D.h would cause circularity // problems). It works because any program **using** this ctor must itself // be including Polar3D.h. template Cartesian3D & operator = (const Polar3D & v) { T rho = v.Rho(); fX = rho * std::cos(v.Phi()); fY = rho * std::sin(v.Phi()); fZ = v.Z(); return *this; } #if defined(__MAKECINT__) || defined(G__DICTIONARY) // ====== Set member functions for coordinates in other systems ======= void SetR(Scalar r); void SetTheta(Scalar theta); void SetPhi(Scalar phi); void SetRho(Scalar rho); void SetEta(Scalar eta); #endif private: T fX; // x coordinate T fY; // y coordinate T fZ; // z coordinate }; } // end namespace Math } // end namespace ROOT #if defined(__MAKECINT__) || defined(G__DICTIONARY) // need to put here setter methods to resolve nasty cyclical dependencies // I need to include other coordinate systems only when Cartesian is already defined // since they depend on it #ifndef ROOT_Math_GenVector_GenVector_exception #include "Math/GenVector/GenVector_exception.h" #endif #ifndef ROOT_Math_GenVector_CylindricalEta3D #include "Math/GenVector/CylindricalEta3D.h" #endif #ifndef ROOT_Math_GenVector_Polar3D #include "Math/GenVector/Polar3D.h" #endif // ====== Set member functions for coordinates in other systems ======= namespace ROOT { namespace Math { template void Cartesian3D::SetR(Scalar r) { GenVector_exception e("Cartesian3D::SetR() is not supposed to be called"); throw e; Polar3D v(*this); v.SetR(r); *this = Cartesian3D(v); } template void Cartesian3D::SetTheta(Scalar theta) { GenVector_exception e("Cartesian3D::SetTheta() is not supposed to be called"); throw e; Polar3D v(*this); v.SetTheta(theta); *this = Cartesian3D(v); } template void Cartesian3D::SetPhi(Scalar phi) { GenVector_exception e("Cartesian3D::SetPhi() is not supposed to be called"); throw e; Polar3D v(*this); v.SetPhi(phi); *this = Cartesian3D(v); } template void Cartesian3D::SetRho(Scalar rho) { GenVector_exception e("Cartesian3D::SetRho() is not supposed to be called"); throw e; CylindricalEta3D v(*this); v.SetRho(rho); *this = Cartesian3D(v); } template void Cartesian3D::SetEta(Scalar eta) { GenVector_exception e("Cartesian3D::SetEta() is not supposed to be called"); throw e; CylindricalEta3D v(*this); v.SetEta(eta); *this = Cartesian3D(v); } } // end namespace Math } // end namespace ROOT #endif #endif /* ROOT_Math_GenVector_Cartesian3D */