rounder.h

Go to the documentation of this file.

#ifndef _RHEOLEF_ROUNDER_H
#define _RHEOLEF_ROUNDER_H
#include "rheolef/compiler.h"
#include <cmath>

namespace rheolef {

template <class T>
struct rounder_type : std::unary_function<T,T> {
  rounder_type (const T& prec) : _prec(prec) {}
  T operator() (const T& x) const {
     T value = _prec*std::floor(x/_prec + 0.5);
     if (1+value == 1) value = 0;
     return value;
  }
  T _prec;
};
template <class T>
struct floorer_type : std::unary_function<T,T> {
  floorer_type (const T& prec) : _prec(prec) {}
  T operator() (const T& x) const { 
     T value = _prec*std::floor(x/_prec);
     if (1+value == 1) value = 0;
     return value;
  }
  T _prec;
};
template <class T>
struct ceiler_type : std::unary_function<T,T> {
  ceiler_type (const T& prec) : _prec(prec) {}
  T operator() (const T& x) const { 
     T value = _prec*std::ceil(x/_prec);
     if (1+value == 1) value = 0;
     return value;
  }
  T _prec;
};

template <class T> rounder_type<T> rounder (const T& prec) { return rounder_type<T>(prec); }
template <class T> floorer_type<T> floorer (const T& prec) { return floorer_type<T>(prec); }
template <class T> ceiler_type<T>  ceiler  (const T& prec) { return ceiler_type<T>(prec); }

template<class T, class M>
field_basic<T,M>
round (
 const field_basic<T,M>& uh,
 const T&                prec)
{
  field_basic<T,M> vh (uh.get_space());
  std::transform (uh.begin_dof(), uh.end_dof(), vh.begin_dof(), rounder(prec));
  return vh;
}
template<class Expr, class T2>
field_basic<
  typename field_expr<Expr>::scalar_type
 ,typename field_expr<Expr>::memory_type
>
round (
 const field_expr<Expr>& expr,
 const T2&               prec)
{
  typedef typename field_expr<Expr>::scalar_type T;
  typedef typename field_expr<Expr>::memory_type M;
  return round (field_basic<T,M>(expr), prec);
}

}//namespace rheolef
#endif // _RHEOLEF_ROUNDER_H