nicefunction.hh

/* -*- mode:c++; mode:semantic -*- */

#ifndef NICE_FUNCTION_HH
#define NICE_FUNCTION_HH

#include <algorithm>
#include <cmath>

#include <dune/common/function.hh>

namespace Dune {
class NiceFunction : public VirtualFunction<double, double> {
public:
  virtual ~NiceFunction() {}

  virtual double leftDifferential(double s) const = 0;
  virtual double rightDifferential(double s) const = 0;
};

class RuinaFunction : public NiceFunction {
public:
  RuinaFunction() {}

  RuinaFunction(double coefficient, double a, double mu, double eta,
                double normalStress)
      : coefficient(coefficient),
        a(a),
        mu(mu),
        rho(exp(-mu / a)),
        eta(eta),
        normalStress(normalStress) {}

  /*
    If mu and sigma_n denote the coefficient of friction and the
    normal stress, respectively, this function is given by

    1/eta sigma_n h(max(eta id, rho)) + a rho + mu

    with the constants a and b from Ruina's model and

    h(beta) = beta (a (log beta - 1) + mu)

    as well as

    rho = exp(-mu/a)
  */
  void virtual evaluate(double const &x, double &y) const {
    double const arg = std::max(eta * x, rho);
    double const h = arg * (a * (std::log(arg) - 1) + mu);
    y = 1 / eta * normalStress * h + a * rho + mu;
    y *= coefficient;
  }

  double virtual leftDifferential(double s) const {
    if (eta * s <= rho)
      return 0;

    return normalStress * (a * std::log(s) + mu);
  }

  double virtual rightDifferential(double s) const {
    if (eta * s <= rho)
      return 0;

    return normalStress * (a * std::log(s) + mu);
  }

private:
  double coefficient;
  double a;
  double mu;
  double eta;
  double normalStress;

  double rho;
};

class LinearFunction : public NiceFunction {
public:
  LinearFunction() {}

  LinearFunction(double a) : coefficient(a) {}

  void virtual evaluate(double const &x, double &y) const {
    y = coefficient * x;
  }

  double virtual leftDifferential(double s) const { return coefficient; }

  double virtual rightDifferential(double s) const { return coefficient; }

private:
  double coefficient;
};

template <int slope> class SampleFunction : public NiceFunction {
public:
  void virtual evaluate(double const &x, double &y) const {
    y = (x < 1) ? x : (slope * (x - 1) + 1);
  }

  double virtual leftDifferential(double s) const {
    return (s <= 1) ? 1 : slope;
  }

  double virtual rightDifferential(double s) const {
    return (s < 1) ? 1 : slope;
  }
};

class SteepFunction : public NiceFunction {
public:
  void virtual evaluate(double const &x, double &y) const { y = 100 * x; }

  double virtual leftDifferential(double s) const { return 100; }

  double virtual rightDifferential(double s) const { return 100; }
};

class TrivialFunction : public NiceFunction {
public:
  void virtual evaluate(double const &x, double &y) const { y = 0; }

  double virtual leftDifferential(double) const { return 0; }

  double virtual rightDifferential(double) const { return 0; }
};

// slope in [n-1,n] is n
class HorribleFunction : public NiceFunction {
public:
  void virtual evaluate(double const &x, double &y) const {
    double const fl = floor(x);
    double const sum = fl * (fl + 1) / 2;
    y = sum + (fl + 1) * (x - fl);
  }

  double virtual leftDifferential(double x) const {
    double const fl = floor(x);
    if (x - fl < 1e-14)
      return fl;
    else
      return fl + 1;
  }

  double virtual rightDifferential(double x) const {
    double const c = ceil(x);
    if (c - x < 1e-14)
      return c + 1;
    else
      return c;
  }
};

// slope in [n-1,n] is log(n+1)
class HorribleFunctionLogarithmic : public NiceFunction {
public:
  void virtual evaluate(double const &x, double &y) const {
    y = 0;
    size_t const fl = floor(x);
    for (size_t i = 1; i <= fl;)
      y += log(++i); // factorials grow to fast so we compute this incrementally

    y += log(fl + 2) * (x - fl);
  }

  double virtual leftDifferential(double x) const {
    double const fl = floor(x);
    if (x - fl < 1e-14)
      return log(fl + 1);
    else
      return log(fl + 2);
  }

  double virtual rightDifferential(double x) const {
    double const c = ceil(x);
    if (c - x < 1e-14)
      return log(c + 2);
    else
      return log(c + 1);
  }
};
}
#endif