---

Basic3DVector.h

Go to the documentation of this file.

#ifndef GeometryVector_Basic3DVector_h
#define GeometryVector_Basic3DVector_h

#include "DataFormats/GeometryVector/interface/Basic2DVector.h"
#include "DataFormats/GeometryVector/interface/Theta.h"
#include "DataFormats/GeometryVector/interface/Phi.h"
#include "DataFormats/GeometryVector/interface/PreciseFloatType.h"
#include "DataFormats/GeometryVector/interface/CoordinateSets.h"
#include <iosfwd>
#include <iostream>
#include <cmath>

template < typename T> 
class Basic3DVector {
public:

  typedef T                                   ScalarType;
  typedef Geom::Cylindrical2Cartesian<T>      Cylindrical;
  typedef Geom::Spherical2Cartesian<T>        Spherical;
  typedef Spherical                           Polar; // synonym
    
  Basic3DVector() : theX( T()), theY( T()), theZ( T()) {}

  Basic3DVector( const Basic3DVector & p) : 
    theX(p.x()), theY(p.y()), theZ(p.z()) {}

  template <class U>
  Basic3DVector( const Basic3DVector<U> & p) : 
    theX(p.x()), theY(p.y()), theZ(p.z()) {}

  Basic3DVector( const Basic2DVector<T> & p) : 
    theX(p.x()), theY(p.y()), theZ(0) {}

  template <class OtherPoint> 
  explicit Basic3DVector( const OtherPoint& p) : 
    theX(p.x()), theY(p.y()), theZ(p.z()) {}

  Basic3DVector( const T& x, const T& y, const T& z) : 
    theX(x), theY(y), theZ(z) {}

  template <typename U>
  Basic3DVector( const Geom::Theta<U>& theta, 
                 const Geom::Phi<U>& phi, const T& r) {
    Polar p( theta.value(), phi.value(), r);
    theX = p.x(); theY = p.y(); theZ = p.z();
  }

  T x() const { return theX;}

  T y() const { return theY;}

  T z() const { return theZ;}

  T mag2() const { return theX*theX + theY*theY + theZ*theZ;}

  T mag() const  { return sqrt( mag2());}

  T perp2() const { return x()*x() + y()*y();}

  T perp() const { return sqrt( perp2());}

  T transverse() const { return perp();}

  T barePhi() const {return atan2(theY,theX);}
  Geom::Phi<T> phi() const {return Geom::Phi<T>(atan2(theY,theX));}

  T bareTheta() const {return atan2(perp(),z());}
  Geom::Theta<T> theta() const {return Geom::Theta<T>(atan2(perp(),z()));}

  // T eta() const { return -log( tan( theta()/2.));} 
  T eta() const { T x(z()/perp()); return log(x+sqrt(x*x+1));} // faster 

  Basic3DVector unit() const {
    T my_mag = mag();
    return my_mag == 0 ? *this : *this / my_mag;
  }

  template <class U> 
  Basic3DVector& operator+= ( const Basic3DVector<U>& p) {
    theX += p.x();
    theY += p.y();
    theZ += p.z();
    return *this;
  } 

  template <class U> 
  Basic3DVector& operator-= ( const Basic3DVector<U>& p) {
    theX -= p.x();
    theY -= p.y();
    theZ -= p.z();
    return *this;
  } 

  Basic3DVector operator-() const { return Basic3DVector(-x(),-y(),-z());}

  Basic3DVector& operator*= ( const T& t) {
    theX *= t;
    theY *= t;
    theZ *= t;
    return *this;
  } 

  Basic3DVector& operator/= ( const T& t) {
    theX /= t;
    theY /= t;
    theZ /= t;
    return *this;
  } 

  T dot( const Basic3DVector& v) const { 
    return x()*v.x() + y()*v.y() + z()*v.z();
  }

  template <class U> 
  typename PreciseFloatType<T,U>::Type dot( const Basic3DVector<U>& v) const { 
    return x()*v.x() + y()*v.y() + z()*v.z();
  }

  Basic3DVector cross( const Basic3DVector& v) const {
    return Basic3DVector( y()*v.z() - v.y()*z(), 
                          z()*v.x() - v.z()*x(), 
                          x()*v.y() - v.x()*y());
  }

  template <class U> 
  Basic3DVector<typename PreciseFloatType<T,U>::Type> 
  cross( const Basic3DVector<U>& v) const {
    return Basic3DVector<typename PreciseFloatType<T,U>::Type>( y()*v.z() - v.y()*z(), 
                                                                z()*v.x() - v.z()*x(), 
                                                                x()*v.y() - v.x()*y());
  }

private:
  T theX;
  T theY;
  T theZ;
};

template <class T>
inline std::ostream & operator<<( std::ostream& s, const Basic3DVector<T>& v) {
  return s << " (" << v.x() << ',' << v.y() << ',' << v.z() << ") ";
} 

template <class T, class U>
inline Basic3DVector<typename PreciseFloatType<T,U>::Type>
operator+( const Basic3DVector<T>& a, const Basic3DVector<U>& b) {
  typedef Basic3DVector<typename PreciseFloatType<T,U>::Type> RT;
  return RT(a.x()+b.x(), a.y()+b.y(), a.z()+b.z());
}

template <class T, class U>
inline Basic3DVector<typename PreciseFloatType<T,U>::Type>
operator-( const Basic3DVector<T>& a, const Basic3DVector<U>& b) {
  typedef Basic3DVector<typename PreciseFloatType<T,U>::Type> RT;
  return RT(a.x()-b.x(), a.y()-b.y(), a.z()-b.z());
}

template <class T>
inline T operator*( const Basic3DVector<T>& v1, const Basic3DVector<T>& v2) {
  return v1.dot(v2);
}

template <class T, class U>
inline typename PreciseFloatType<T,U>::Type operator*( const Basic3DVector<T>& v1, 
                                                       const Basic3DVector<U>& v2) {
  return v1.x()*v2.x() + v1.y()*v2.y() + v1.z()*v2.z();
}

template <class T, class Scalar>
inline Basic3DVector<T> operator*( const Basic3DVector<T>& v, const Scalar& s) {
  T t = static_cast<T>(s);
  return Basic3DVector<T>(v.x()*t, v.y()*t, v.z()*t);
}

template <class T, class Scalar>
inline Basic3DVector<T> operator*( const Scalar& s, const Basic3DVector<T>& v) {
  T t = static_cast<T>(s);
  return Basic3DVector<T>(v.x()*t, v.y()*t, v.z()*t);
}

template <class T, class Scalar>
inline Basic3DVector<T> operator/( const Basic3DVector<T>& v, const Scalar& s) {
  T t = static_cast<T>(s);
  return Basic3DVector<T>(v.x()/t, v.y()/t, v.z()/t);
}

#endif // GeometryVector_Basic3DVector_h

---