geant4/source/geometry/solids/usolids/include/UUtils.hh

//
// ********************************************************************
// * This Software is part of the AIDA Unified Solids Library package *
// * See: https://aidasoft.web.cern.ch/USolids                        *
// ********************************************************************
//
// $Id:$
//
// --------------------------------------------------------------------
//
// UUtils
//
// Description:
//
//   Utility namespace providing common constants and mathematical utilities.
//
// 19.10.12 Marek Gayer
// --------------------------------------------------------------------

#ifndef USOLIDS_UUtils
#define USOLIDS_UUtils

#include <iostream>
#include <fstream>
#include <limits>
#include <cmath>
#include <cfloat>
#include <vector>
#include <algorithm>

#include "UVector3.hh"

class UTransform3D;

enum ExceptionSeverity
{ FatalError, FatalErrorInArguments, Error, Warning, Info };

namespace UUtils
{

  // Sign
  inline short  Sign(short a, short b);
  inline int    Sign(int a, int b);
  inline long   Sign(long a, long b);
  inline float  Sign(float a, float b);
  inline double Sign(double a, double b);

  // Trigonometric
  static const double kPi       = 3.14159265358979323846;
  static const double kTwoPi    = 2.0 * kPi;
  static const double kRadToDeg = 180.0 / kPi;
  static const double kDegToRad = kPi / 180.0;
  static const double kSqrt2    = 1.4142135623730950488016887242097;
  static const double kInfinity = DBL_MAX;

  static const double kMeshAngleDefault = (kPi / 4); // Angle for mesh `wedges' in rads
  static const int kMinMeshSections = 3; // Min wedges+1 to make
  static const int kMaxMeshSections = 37; // max wedges+1 to make

  inline double Infinity();

  inline double ASin(double);
  inline double ACos(double);
  inline double ATan(double);
  inline double ATan2(double, double);

  //Warnings and Errors Messages
  void Exception(const char* originOfException,
                 const char* exceptionCode,
                 ExceptionSeverity severity,
                 int level,
                 const char* description);


  // Comparing floating points
  inline bool AreEqualAbs(double af, double bf, double epsilon)
  {
    //return true if absolute difference between af and bf is less than epsilon
    return std::abs(af - bf) < epsilon;
  }
  inline bool AreEqualRel(double af, double bf, double relPrec)
  {
    //return true if relative difference between af and bf is less than relPrec
    return std::abs(af - bf) <= 0.5 * relPrec * (std::abs(af) + std::abs(bf));
  }

  // Locate Min, Max element number in an array
  long  LocMin(long n, const double* a);
  long  LocMax(long n, const double* a);

  // TransformLimits: Use the transformation to convert the local limits defined
  // by min/max vectors to the master frame. Returns modified limits.
  void TransformLimits(UVector3& min, UVector3& max, const UTransform3D& transformation);

  double Random(double min = 0.0, double max = 1.0);

  // Templates:
  template<typename T>
  struct CompareDesc
  {

    CompareDesc(T d) : fData(d) {}

    template<typename Index>
    bool operator()(Index i1, Index i2)
    {
      return *(fData + i1) > *(fData + i2);
    }

    T fData;
  };

  template<typename T>
  struct CompareAsc
  {

    CompareAsc(T d) : fData(d) {}

    template<typename Index>
    bool operator()(Index i1, Index i2)
    {
      return *(fData + i1) < *(fData + i2);
    }

    T fData;
  };

  int SaveVectorToExternalFile(const std::vector<double>& vector, const std::string& filename);
  int SaveVectorToExternalFile(const std::vector<UVector3>& vector, const std::string& filename);
  int SaveVectorToExternalFile(const std::vector<int>& vector, const std::string& filename);

  std::string ToString(int number);
  std::string ToString(double number);

  int FileSize(const std::string& filePath);

  int StrPos(const std::string& haystack, const std::string& needle);

  inline double GetRadiusInRing(double rmin, double rmax);

  template <class T>
  inline T sqr(const T& x)
  {
    return x * x;
  }

  inline bool StrEnds(std::string const& fullString, std::string const& ending)
  {
    if (fullString.length() >= ending.length())
    {
      return (0 == fullString.compare(fullString.length() - ending.length(), ending.length(), ending));
    }
    else
    {
      return false;
    }
  }
}

inline double UUtils::GetRadiusInRing(double rmin, double rmax)
{
  // Generate radius in annular ring according to uniform area
  //
  if (rmin <= 0.)
  {
    return rmax * std::sqrt(Random());
  }
  if (rmin != rmax)
  {
    return std::sqrt(Random()
                     * (sqr(rmax) - sqr(rmin)) + sqr(rmin));
  }
  return rmin;
}

//____________________________________________________________________________
inline double UUtils::Infinity()
{
  // returns an infinity as defined by the IEEE standard
  return std::numeric_limits<double>::infinity();
}

//---- Sign --------------------------------------------------------------------
inline short UUtils::Sign(short a, short b)
{
  return (b >= 0) ? std::abs(a) : -std::abs(a);
}

inline int UUtils::Sign(int a, int b)
{
  return (b >= 0) ? std::abs(a) : -std::abs(a);
}

inline long UUtils::Sign(long a, long b)
{
  return (b >= 0) ? std::abs(a) : -std::abs(a);
}

inline float UUtils::Sign(float a, float b)
{
  return (b >= 0) ? std::abs(a) : -std::abs(a);
}

inline double UUtils::Sign(double a, double b)
{
  return (b >= 0) ? std::abs(a) : -std::abs(a);
}


//---- Trigonometric------------------------------------------------------------
inline double UUtils::ASin(double x)
{
  if (x < -1.) return -kPi / 2;
  if (x >  1.) return  kPi / 2;
  return std::asin(x);
}

inline double UUtils::ACos(double x)
{
  if (x < -1.) return kPi;
  if (x >  1.) return 0;
  return std::acos(x);
}


inline double UUtils::ATan2(double y, double x)
{
  if (x != 0) return  std::atan2(y, x);
  if (y == 0) return  0;
  if (y >  0) return  kPi / 2;
  else        return -kPi / 2;
}

#endif