Vector3D.h

Go to the documentation of this file.

//==========================================================================
//  AIDA Detector description implementation 
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : F.Gaede
//
//==========================================================================
#ifndef DDREC_VECTOR3D_H
#define DDREC_VECTOR3D_H 1
 
#include <cmath>
#include <iostream> 
#include <cassert>
 
 
namespace dd4hep{ namespace rec {
 
  class Vector3D{
    
  public:
    
    Vector3D() : _x(0.0),_y(0.0),_z(0.0) {}
    
    
    Vector3D(const Vector3D& v) : _x(v[0]),_y(v[1]),_z(v[2]) {}
 
    Vector3D(const float* v) : _x(v[0]),_y(v[1]),_z(v[2]) {}
    
    Vector3D(const double* v) : _x(v[0]),_y(v[1]),_z(v[2]) {}
    
    
    template <class T>
    Vector3D( double x,double y, double z , T(&)() ) ;
    
    
    Vector3D( double x_val,double y_val, double z_val ) :
      _x(x_val),
      _y(y_val),
      _z(z_val) {
    }
    
    // ---- this causes all sorts of template lookup errors ...    
    // /** Copy c'tor for three vectors from other packages - requires T::x(),T::y(), T::z().
    //  */
    // template <class T>
    // Vector3D( const T& t) :
    //   _x( t.x() ) , 
    //   _y( t.y() ) , 
    //   _z( t.z() ){
    // }
    
    //assignment operator
    Vector3D& operator=(const Vector3D& v) {
      _x = v[0] ;
      _y = v[1] ;
      _z = v[2] ;
      return *this ; 
    }
 
    template <class T>
    inline const Vector3D& fill(  const T& v  ) {    
 
      _x = v[0] ; _y = v[1] ; _z = v[2] ; 
      return *this ;
    }  
 
    inline const Vector3D& fill( const double* v) {    
 
      _x = v[0] ; _y = v[1] ; _z = v[2] ; 
      return *this ;
    }  
 
    inline const Vector3D& fill( double x_val, double y_val, double z_val) {    
      _x = x_val ; _y = y_val ; _z = z_val ; 
      return *this ;
    }  
 
 
    inline double x() const { return  _x ; }
    
    inline double y() const { return  _y ; }
    
    inline double z() const { return  _z ; }
    
    inline double& x() { return  _x ; }
    
    inline double& y() { return  _y ; }
    
    inline double& z() { return  _z ; }
    
    
    inline double operator[](int i) const {
      switch(i) {
      case 0: return _x ; break ;
      case 1: return _y ; break ;
      case 2: return _z ; break ;
      }
      return 0.0 ;
    }
    inline double& operator[](int i) {
      switch(i) {
      case 0: return _x ; break ;
      case 1: return _y ; break ;
      case 2: return _z ; break ;
      }
      static double dummy(0.0) ;
      return dummy ;
    }
    
    inline double phi() const {
      
      return _x == 0.0 && _y == 0.0 ? 0.0 : atan2(_y,_x);
    }
    
    inline double rho() const {
      
      return trans() ;
    }
    
    inline double trans() const {
      
      return sqrt( _x*_x + _y*_y ) ; 
    }
    
    inline double trans2() const {
      
      return  _x*_x + _y*_y  ;
    }
    
    inline double r() const {
      
      return sqrt( _x*_x + _y*_y + _z*_z ) ; 
    }
    
    
    inline double r2() const {
      
      return  _x*_x + _y*_y + _z*_z  ; 
    }
    
    inline double theta() const {
      
      return _x == 0.0 && _y == 0.0 && _z == 0.0 ? 0.0 : atan2( rho(),_z) ;
    }
    
    inline double dot( const Vector3D& v) const { 
      return _x * v.x() + _y * v.y() + _z * v.z() ;
    }
    
 
    inline Vector3D cross( const Vector3D& v) const { 
      
      return Vector3D( _y * v.z() - _z * v.y() ,
               _z * v.x() - _x * v.z() ,
               _x * v.y() - _y * v.x() )  ;
    }
    
    inline Vector3D unit() const { 
      
      double n = r() ;
      return Vector3D( _x / n , _y / n , _z / n ) ;
    }
    
    
    inline operator const double*() const {
      return &_x ;
    }
    inline const double* const_array() const {
      return &_x ;
    }
 
    inline double* array() {
      return &_x ;
    }
 
 
      inline bool isEqual(  const Vector3D& b , double epsilon=1e-6) { 
      
      if( fabs( x() - b.x() ) < epsilon &&
      fabs( y() - b.y() ) < epsilon && 
      fabs( z() - b.z() ) < epsilon )
    return true;
      else
    return false;
  }
  
 
 
    // this causes template lookup errors on some machines :
    //   -> use explicit conversion with to<T>()
    // /** Implicit templated conversion to anything that has a c'tor T(x,y,z) 
    //  *  and accessor functions x(),y(),z(). For safety the result is checked which 
    //  *  causes a small performance penalty.
    //  *  @see to()
    //  *  
    //  */
    // template <class T>
    // inline operator T() const { 
 
    //   T t( _x, _y , _z ) ;
      
    //   assert( t.x()== _x && t.y()== _y && t.z()== _z ) ;
      
    //   return t ;
      
    //   //     return T( _x, _y, _z ) ; 
    // } 
    
    
    template <class T>
    inline T to() const { return T( _x, _y, _z ) ; } 
    
    
  protected:
    
    double _x,_y,_z ;
    
    
    // helper classes and function to allow 
    // different c'tors selected at compile time
  public:
    
    struct Cartesian   { } ;
    struct Cylindrical { } ;
    struct Spherical   { } ;
    
    static Cartesian   cartesian()  { return Cartesian() ; }
    static Cylindrical cylindrical(){ return Cylindrical() ;}
    static Spherical   spherical()  { return Spherical() ; } 
    
  } ;
  
  inline Vector3D operator+(  const Vector3D& a, const Vector3D& b ) { 
    
    return Vector3D( a.x() + b.x()  , a.y() + b.y(), a.z() + b.z()  ) ;
  }
  inline Vector3D operator-(  const Vector3D& a, const Vector3D& b ) { 
    
    return Vector3D( a.x() - b.x()  , a.y() - b.y(), a.z() - b.z()  ) ;
  }
  inline bool operator==(  const Vector3D& a, const Vector3D& b ) { 
    
    if( a.x() == b.x()  &&  a.y() == b.y() && a.z() == b.z()  ) 
      return true;
    else
      return false;
  }
 
  inline Vector3D operator*( double s , const Vector3D& v ) { 
    
    return Vector3D( s * v.x()  , s * v.y()  ,  s * v.z() ) ;
  }
  
  inline Vector3D operator-(  const Vector3D& v) { 
    
    return Vector3D( -v.x(), - v.y(), - v.z()  ) ;
  }
 
  inline double operator*( const Vector3D& v0, const Vector3D& v1 ){
    return v0.dot( v1 ) ;
  }
 
  
  // template specializations for constructors of  different coordinate systems
  
  template <>
  inline Vector3D::Vector3D( double x_val,double y_val, double z_val, Vector3D::Cartesian (&)() ) : 
    _x(x_val),
    _y(y_val),
    _z(z_val) {
  }
  
  template <>
  inline Vector3D::Vector3D( double rho_val,double phi_val, double z_val, Vector3D::Cylindrical (&)() ) : _z(z_val)  {
    
    _x = rho_val * cos( phi_val ) ;
    _y = rho_val * sin( phi_val ) ;
  }
  
  
  template <>
  inline Vector3D::Vector3D( double r_val,double phi_val, double theta_val, Vector3D::Spherical (&)() ) {
    double rst =  r_val * sin( theta_val ) ;
    _x = rst * cos( phi_val ) ;
    _y = rst * sin( phi_val ) ;
    _z = r_val * cos( theta_val ) ;
  }
  
  
  
  inline std::ostream & operator << (std::ostream & os, const Vector3D &v) {
 
    //    os << "( " << v[0] << ", " << v[1] << ", " << v[2] << " )" ;
    os << "  ( " << v[0] 
       << ", " << v[1]
       << ", " << v[2]
       << " ) -  [ phi: " << v.phi()
       << " , rho: " << v.rho() << " ] "  
       << "  [ theta: " << v.theta()
       << " , r: " << v.r() << " ] " ;
    
    return os ;
  }
 
  
 
}} // namespace
 
 
 
 
 
 
#endif