Teuchos_ScalarTraits.hpp

Go to the documentation of this file.

// @HEADER
// ***********************************************************************
// 
//                    Teuchos: Common Tools Package
//                 Copyright (2004) Sandia Corporation
// 
// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
// license for use of this work by or on behalf of the U.S. Government.
// 
// This library is free software; you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation; either version 2.1 of the
// License, or (at your option) any later version.
//  
// This library is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//  
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
// USA
// Questions? Contact Michael A. Heroux (maherou@sandia.gov) 
// 
// ***********************************************************************
// @HEADER

// Kris
// 06.18.03 -- Minor formatting changes
//          -- Changed calls to LAPACK objects to use new <OType, SType> templates
// 07.08.03 -- Move into Teuchos package/namespace
// 07.11.03 -- Added ScalarTraits for ARPREC::mp_real
// 07.14.03 -- Fixed int rand() function (was set up to return a floating-point style random number)
// 07.17.03 -- Added squareroot() function

#ifndef _TEUCHOS_SCALARTRAITS_HPP_
#define _TEUCHOS_SCALARTRAITS_HPP_

#include "Teuchos_LAPACK.hpp"

#ifdef HAVE_TEUCHOS_ARPREC
#include "mp/mpreal.h"
#endif

#ifdef HAVE_TEUCHOS_GNU_MP
#include "gmp.h"
#include "gmpxx.h"
#endif

namespace Teuchos {
  template<class T>
  struct UndefinedScalarTraits
  {
    static inline T notDefined() { return T::this_type_is_missing_a_specialization(); };
  };

  template<class T>
  struct ScalarTraits
  {
    typedef T magnitudeType;
    static inline bool haveMachineParameters() { return false; };
    static inline magnitudeType eps()   { return UndefinedScalarTraits<T>::notDefined(); };
    static inline magnitudeType sfmin() { return UndefinedScalarTraits<T>::notDefined(); };
    static inline magnitudeType base()  { return UndefinedScalarTraits<T>::notDefined(); };
    static inline magnitudeType prec()  { return UndefinedScalarTraits<T>::notDefined(); };
    static inline magnitudeType t()     { return UndefinedScalarTraits<T>::notDefined(); };
    static inline magnitudeType rnd()   { return UndefinedScalarTraits<T>::notDefined(); };
    static inline magnitudeType emin()  { return UndefinedScalarTraits<T>::notDefined(); };
    static inline magnitudeType rmin()  { return UndefinedScalarTraits<T>::notDefined(); };
    static inline magnitudeType emax()  { return UndefinedScalarTraits<T>::notDefined(); };
    static inline magnitudeType rmax()  { return UndefinedScalarTraits<T>::notDefined(); };
    static inline magnitudeType magnitude(T a) { return UndefinedScalarTraits<T>::notDefined(); };
    static inline T zero()                     { return UndefinedScalarTraits<T>::notDefined(); };
    static inline T one()                      { return UndefinedScalarTraits<T>::notDefined(); };
    static inline T conjugate(T a) { return UndefinedScalarTraits<T>::notDefined(); };
    static inline T nan()                      { return UndefinedScalarTraits<T>::notDefined(); };
    static inline bool isnaninf(const T& x)     { return UndefinedScalarTraits<T>::notDefined(); };
    static inline void seedrandom(unsigned int s) { int i; T t = &i; };
    static inline T random()                   { return UndefinedScalarTraits<T>::notDefined(); };
    static inline std::string name()           { (void)UndefinedScalarTraits<T>::notDefined(); return 0; };
    static inline magnitudeType squareroot(T x) { return UndefinedScalarTraits<T>::notDefined(); };
  };
  
#ifndef DOXYGEN_SHOULD_SKIP_THIS

  template<>
  struct ScalarTraits<int>
  {
    typedef int magnitudeType;
    static inline bool haveMachineParameters() { return false; };
    // Not defined: eps(), sfmin(), base(), prec(), t(), rnd(), emin(), rmin(), emax(), rmax()
    static inline magnitudeType magnitude(int a) { return ::abs(a); };
    static inline int zero()  { return 0; };
    static inline int one()   { return 1; };
    static inline int conjugate(int x) { return x; };
    static inline void seedrandom(unsigned int s) { srand(s); };
    //static inline int random() { return (-1 + 2*rand()); };  // RAB: This version should be used to be consistent with others
    static inline int random() { return rand(); };             // RAB: This version should be used for an unsigned int, not int
    static inline std::string name() { return "int"; };
    static inline int squareroot(int x) { return (int) ::sqrt((double) x); };
  };

  extern float flt_nan;
 
  template<>
  struct ScalarTraits<float>
  {
    typedef float magnitudeType;
    static inline bool haveMachineParameters() { return true; };
    static inline float eps()   {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<float>::epsilon();
#else
      LAPACK<int, float> lp; return lp.LAMCH('E');
#endif
    }
    static inline float sfmin() {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<float>::min();
#else
      LAPACK<int, float> lp; return lp.LAMCH('S');
#endif
    };
    static inline float base()  {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<float>::radix;
#else
      LAPACK<int, float> lp; return lp.LAMCH('B');
#endif
    };
    static inline float prec()  {
#ifdef HAVE_NUMERIC_LIMITS
      return eps()*base();
#else
      LAPACK<int, float> lp; return lp.LAMCH('P');
#endif
    };
    static inline float t()     {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<float>::digits;
#else
      LAPACK<int, float> lp; return lp.LAMCH('N');
#endif
    };
    static inline float rnd()   {
#ifdef HAVE_NUMERIC_LIMITS
      return ( std::numeric_limits<float>::round_style == std::round_to_nearest ? float(1.0) : float(0.0) );
#else
      LAPACK<int, float> lp; return lp.LAMCH('R');
#endif
    };
    static inline float emin()  {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<float>::min_exponent;
#else
      LAPACK<int, float> lp; return lp.LAMCH('M');
#endif
    };
    static inline float rmin()  {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<float>::min();
#else
      LAPACK<int, float> lp; return lp.LAMCH('U');
#endif
    };
    static inline float emax()  {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<float>::max_exponent;
#else
      LAPACK<int, float> lp; return lp.LAMCH('L');
#endif
    };
    static inline float rmax()  {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<float>::max();
#else
      LAPACK<int, float> lp; return lp.LAMCH('O');
#endif
    };
    static inline magnitudeType magnitude(float a) { return fabs(a); };    
    static inline float zero()  { return(0.0); };
    static inline float one()   { return(1.0); };    
    static inline float conjugate(float x)   { return(x); };    
    static inline float nan()   { return flt_nan; };
    static inline bool isnaninf(float x) { // RAB: 2004/05/28: Taken from NOX_StatusTest_FiniteValue.C
      const float tol = 1e-6; // Any (bounded) number should do!
      if( !(x <= tol) && !(x > tol) ) return true;                 // IEEE says this should fail for NaN
      float z=0.0*x; if( !(z <= tol) && !(z > tol) ) return true;  // Use fact that Inf*0 = NaN
      return false;
    }
    static inline void seedrandom(unsigned int s) { srand(s); };
    static inline float random() { float rnd = (float) rand() / RAND_MAX; return (float)(-1.0 + 2.0 * rnd); };
    static inline std::string name() { return "float"; };
    static inline float squareroot(float x) { return sqrt(x); };
  };

  extern double dbl_nan;
 
  template<>
  struct ScalarTraits<double>
  {
    typedef double magnitudeType;
    static inline bool haveMachineParameters() { return true; };
    static inline double eps()   {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<double>::epsilon();
#else
      LAPACK<int, double> lp; return lp.LAMCH('E');
#endif
    }
    static inline double sfmin() {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<double>::min();
#else
      LAPACK<int, double> lp; return lp.LAMCH('S');
#endif
    };
    static inline double base()  {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<double>::radix;
#else
      LAPACK<int, double> lp; return lp.LAMCH('B');
#endif
    };
    static inline double prec()  {
#ifdef HAVE_NUMERIC_LIMITS
      return eps()*base();
#else
      LAPACK<int, double> lp; return lp.LAMCH('P');
#endif
    };
    static inline double t()     {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<double>::digits;
#else
      LAPACK<int, double> lp; return lp.LAMCH('N');
#endif
    };
    static inline double rnd()   {
#ifdef HAVE_NUMERIC_LIMITS
      return ( std::numeric_limits<double>::round_style == std::round_to_nearest ? double(1.0) : double(0.0) );
#else
      LAPACK<int, double> lp; return lp.LAMCH('R');
#endif
    };
    static inline double emin()  {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<double>::min_exponent;
#else
      LAPACK<int, double> lp; return lp.LAMCH('M');
#endif
    };
    static inline double rmin()  {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<double>::min();
#else
      LAPACK<int, double> lp; return lp.LAMCH('U');
#endif
    };
    static inline double emax()  {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<double>::max_exponent;
#else
      LAPACK<int, double> lp; return lp.LAMCH('L');
#endif
    };
    static inline double rmax()  {
#ifdef HAVE_NUMERIC_LIMITS
      return std::numeric_limits<double>::max();
#else
      LAPACK<int, double> lp; return lp.LAMCH('O');
#endif
    };
    static inline magnitudeType magnitude(double a) { return fabs(a); };
    static inline double zero()  { return 0.0; };
    static inline double one()   { return 1.0; };
    static inline double conjugate(double x)   { return(x); };    
    static inline double nan() { return dbl_nan; };
    static inline bool isnaninf(double x) { // RAB: 2004/05/28: Taken from NOX_StatusTest_FiniteValue.C
      const double tol = 1e-6; // Any (bounded) number should do!
      if( !(x <= tol) && !(x > tol) ) return true;                  // IEEE says this should fail for NaN
      double z=0.0*x; if( !(z <= tol) && !(z > tol) ) return true;  // Use fact that Inf*0 = NaN
      return false;
    }
    static inline void seedrandom(unsigned int s) { srand(s); };
    static inline double random() { double rnd = (double) rand() / RAND_MAX; return (double)(-1.0 + 2.0 * rnd); };
    static inline std::string name() { return "double"; };
    static inline double squareroot(double x) { return ::sqrt(x); };
  };

#ifdef HAVE_TEUCHOS_GNU_MP

  extern gmp_randclass gmp_rng; 

  template<>
  struct ScalarTraits<mpf_class>
  {
    typedef mpf_class magnitudeType;
    static inline bool haveMachineParameters() { return false; };
    // Not defined: eps(), sfmin(), base(), prec(), t(), rnd(), emin(), rmin(), emax(), rmax()
    static magnitudeType magnitude(mpf_class a) { return abs(a); };
    static inline mpf_class zero() { mpf_class zero = 0.0; return zero; };
    static inline mpf_class one() { mpf_class one = 1.0; return one; };    
    static inline mpf_class conjugate(mpf_class x) { return x; };
    static inline void seedrandom(unsigned int s) { 
      unsigned long int seedVal = static_cast<unsigned long int>(s);
      gmp_rng.seed( seedVal );  
    };
    static inline mpf_class random() { 
      return gmp_rng.get_f(); 
    };
    static inline std::string name() { return "mpf_class"; };
    static inline mpf_class squareroot(mpf_class x) { return sqrt(x); };
    // Todo: RAB: 2004/05/28: Add nan() and isnaninf() functions when needed!
  };

#endif  

#ifdef HAVE_TEUCHOS_ARPREC

  template<>
  struct ScalarTraits<mp_real>
  {
    typedef mp_real magnitudeType;
    static inline bool haveMachineParameters() { return false; };
    // Not defined: eps(), sfmin(), base(), prec(), t(), rnd(), emin(), rmin(), emax(), rmax()
    static magnitudeType magnitude(mp_real a) { return abs(a); };
    static inline mp_real zero() { mp_real zero = 0.0; return zero; };
    static inline mp_real one() { mp_real one = 1.0; return one; };    
    static inline mp_real conjugate(mp_real x) { return x; };
    static inline void seedrandom(unsigned int s) { 
      long int seedVal = static_cast<long int>(s);
      srand48(seedVal);
    };
    static inline mp_real random() { return mp_rand(); };
    static inline std::string name() { return "mp_real"; };
    static inline mp_real squareroot(mp_real x) { return sqrt(x); };
    // Todo: RAB: 2004/05/28: Add nan() and isnaninf() functions when needed!
  };
  
#endif // HAVE_TEUCHOS_ARPREC
 
#if ( defined(HAVE_COMPLEX) || defined(HAVE_COMPLEX_H) ) && defined(HAVE_TEUCHOS_COMPLEX)

  // Partial specialization for complex numbers templated on real type T
  template<class T> 
  struct ScalarTraits<
#if defined(HAVE_COMPLEX)
    std::complex<T>
#elif  defined(HAVE_COMPLEX_H)
    ::complex<T>
#endif
    >
  {
#if defined(HAVE_COMPLEX)
    typedef std::complex<T>  ComplexT;
#elif  defined(HAVE_COMPLEX_H)
    typedef ::complex<T>     ComplexT;
#endif
    typedef typename ScalarTraits<T>::magnitudeType magnitudeType;
    static inline bool haveMachineParameters() { return true; };
    static inline magnitudeType eps()          { return ScalarTraits<magnitudeType>::eps(); };
    static inline magnitudeType sfmin()        { return ScalarTraits<magnitudeType>::sfmin(); };
    static inline magnitudeType base()         { return ScalarTraits<magnitudeType>::base(); };
    static inline magnitudeType prec()         { return ScalarTraits<magnitudeType>::prec(); };
    static inline magnitudeType t()            { return ScalarTraits<magnitudeType>::t(); };
    static inline magnitudeType rnd()          { return ScalarTraits<magnitudeType>::rnd(); };
    static inline magnitudeType emin()         { return ScalarTraits<magnitudeType>::emin(); };
    static inline magnitudeType rmin()         { return ScalarTraits<magnitudeType>::rmin(); };
    static inline magnitudeType emax()         { return ScalarTraits<magnitudeType>::emax(); };
    static inline magnitudeType rmax()         { return ScalarTraits<magnitudeType>::rmax(); };
    static magnitudeType magnitude(ComplexT a) { return std::abs(a); };
    static inline ComplexT zero()              { return ComplexT(ScalarTraits<magnitudeType>::zero(),ScalarTraits<magnitudeType>::zero()); };
    static inline ComplexT one()               { return ComplexT(ScalarTraits<magnitudeType>::one(),ScalarTraits<magnitudeType>::zero()); };
    static inline ComplexT conjugate(ComplexT a){ return ComplexT(a.real(),-a.imag()); };
    static inline ComplexT nan()               { return ComplexT(ScalarTraits<magnitudeType>::nan(),ScalarTraits<magnitudeType>::nan()); };
    static inline bool isnaninf(ComplexT x)    { return ScalarTraits<magnitudeType>::isnaninf(x.real()) || ScalarTraits<magnitudeType>::isnaninf(x.imag()); };
    static inline void seedrandom(unsigned int s) { ScalarTraits<magnitudeType>::seedrandom(s); };
    static inline ComplexT random()
    {
      const T rnd1 = ScalarTraits<magnitudeType>::random();
      const T rnd2 = ScalarTraits<magnitudeType>::random();
      return ComplexT(rnd1,rnd2);
    };
    static inline std::string name() { return std::string("std::complex<")+std::string(ScalarTraits<magnitudeType>::name())+std::string(">"); };
    // This will only return one of the square roots of x, the other can be obtained by taking its conjugate
    static inline ComplexT squareroot(ComplexT x)
    {
      typedef ScalarTraits<magnitudeType>  STMT;
      const T r  = x.real(), i = x.imag();
      const T a  = STMT::squareroot((r*r)+(i*i));
      const T nr = STMT::squareroot((a+r)/2);
      const T ni = STMT::squareroot((a-r)/2);
      return ComplexT(nr,ni);
    };
  };

#endif //  HAVE_COMPLEX || HAVE_COMPLEX_H

#endif // DOXYGEN_SHOULD_SKIP_THIS

} // Teuchos namespace

#endif // _TEUCHOS_SCALARTRAITS_HPP_

---