influenceFn.h

Go to the documentation of this file.

/*
 * -------------------------------------------
 * Copyright (c) 2021 - 2024 Prashant K. Jha
 * -------------------------------------------
 * PeriDEM https://github.com/prashjha/PeriDEM
 *
 * Distributed under the Boost Software License, Version 1.0. (See accompanying
 * file LICENSE)
 */
 
#ifndef MATERIAL_PD_INFLUENCEFN_H
#define MATERIAL_PD_INFLUENCEFN_H
 
#include "util/io.h"
#include <cstring>
#include <iostream>
#include <vector>
 
namespace material {
 
class BaseInfluenceFn {
 
public:
  BaseInfluenceFn() = default;
 
  virtual double getInfFn(const double &r) const = 0;
 
  virtual double getMoment(const size_t &i) const = 0;
 
  virtual std::string printStr(int nt, int lvl) const {
 
    auto tabS = util::io::getTabS(nt);
    std::ostringstream oss;
    oss << tabS << "------- BaseInfluenceFn --------" << std::endl << std::endl;
    oss << tabS << "Provides abstraction for different influence function "
                   "types" << std::endl;
    oss << tabS << std::endl;
 
    return oss.str();
  }
 
  virtual void print(int nt, int lvl) const { std::cout << printStr(nt, lvl); }
 
  virtual void print() const { print(0, 0); }
};
 
class ConstInfluenceFn : public BaseInfluenceFn {
 
public:
  ConstInfluenceFn(const std::vector<double> &params, const size_t &dim);
 
  double getInfFn(const double &r) const override;
 
  double getMoment(const size_t &i) const override;
 
  std::string printStr(int nt, int lvl) const override {
 
    auto tabS = util::io::getTabS(nt);
    std::ostringstream oss;
    oss << tabS << "------- ConstInfluenceFn --------" << std::endl << std::endl;
    oss << tabS << "Constant function with constant = " << d_a0 << std::endl;
    oss << tabS << "First moment = " << getMoment(1)
              << ", second moment = " << getMoment(2)
              << ", third moment = " << getMoment(3) << std::endl;
    oss << tabS << std::endl;
 
    return oss.str();
  }
 
  void print(int nt, int lvl) const override {
    std::cout << printStr(nt, lvl);
  }
 
  void print() const override { print(0, 0); }
 
private:
  double d_a0;
};
 
class LinearInfluenceFn : public BaseInfluenceFn {
 
public:
  LinearInfluenceFn(const std::vector<double> &params, const size_t &dim);
 
  double getInfFn(const double &r) const override;
 
  double getMoment(const size_t &i) const override;
 
  std::string printStr(int nt, int lvl) const override {
 
    auto tabS = util::io::getTabS(nt);
    std::ostringstream oss;
    oss << tabS << "------- LinearInfluenceFn --------" << std::endl << std::endl;
    oss << tabS << "Linear function a0 + a1*r with constants: a0 = "
                << d_a0 << ", a1 = " << d_a1 << std::endl;
    oss << tabS << "First moment = " << getMoment(1)
        << ", second moment = " << getMoment(2)
        << ", third moment = " << getMoment(3) << std::endl;
    oss << tabS << std::endl;
 
    return oss.str();
  }
 
  void print(int nt, int lvl) const override {
    std::cout << printStr(nt, lvl);
  }
 
  void print() const override { print(0, 0); }
 
private:
  double d_a0;
 
  double d_a1;
};
 
class GaussianInfluenceFn : public BaseInfluenceFn {
 
public:
  GaussianInfluenceFn(const std::vector<double> &params, const size_t &dim);
 
  double getInfFn(const double &r) const override;
 
  double getMoment(const size_t &i) const override;
 
  std::string printStr(int nt, int lvl) const override {
 
    auto tabS = util::io::getTabS(nt);
    std::ostringstream oss;
    oss << tabS << "------- GaussianInfluenceFn --------" << std::endl << std::endl;
    oss << tabS << "Gaussian function a0 * exp(-r*r / a1) with constants: a0 = "
                << d_alpha << ", a1 = " << d_beta << std::endl;
    oss << tabS << "First moment = " << getMoment(1)
        << ", second moment = " << getMoment(2)
        << ", third moment = " << getMoment(3) << std::endl;
    oss << tabS << std::endl;
 
    return oss.str();
  }
 
  void print(int nt, int lvl) const override {
    std::cout << printStr(nt, lvl);
  }
 
  void print() const override { print(0, 0); }
 
private:
  double d_alpha;
 
  double d_beta;
};
 
} // namespace material
 
#endif // MATERIAL_PD_INFLUENCEFN_H