Polynomial.h

Go to the documentation of this file.

// This file is part of the MercuryDPM project (https://www.mercurydpm.org).
// Copyright (c), The MercuryDPM Developers Team. All rights reserved.
// License: BSD 3-Clause License; see the LICENSE file in the root directory.
 
#ifndef Polynomial_H
#define Polynomial_H
 
#include <GeneralDefine.h>
#include "IntegralType.h"
#include <iostream>
#include <vector>
 
class BaseParticle;
 
class BaseInteraction;
 
class DPMBase;
 
class Coordinates;
 
namespace CGFunctions
{
 
enum class PolynomialType : unsigned char
{
    HEAVISIDE = 0,
    LINEAR = 1,
    LUCY = 2
};
 
std::ostream& operator<<(std::ostream& os, PolynomialType type);
 
std::istream& operator>>(std::istream& is, PolynomialType& type);
 
template<class Coordinates>
class Polynomial
{
public:
    
    typedef Coordinates CoordinatesType;
    
    Polynomial();
    
    Polynomial(const Polynomial& p) = default;
    
    ~Polynomial() = default;
    
    void write(std::ostream& os) const;
    
    void setPolynomialType(PolynomialType polynomialType);
    
    void setWidth(Mdouble width);
    
    Mdouble getWidth() const;
    
    void setStandardDeviation(Mdouble std);
    
    void setCutoff(Mdouble cutoff);
    
    Mdouble getCutoff() const;
    
    inline Vec3D getCutoffVector(Mdouble radius, Vec3D cellSize) const { return Vec3D(getCutoff(), getCutoff(), getCutoff()); }
    
    void computeCoefficients();
    
    Mdouble evaluateCGFunction(const Vec3D& position, const Coordinates& r);
    
    Mdouble evaluateCGFunction(const Vec3D& position, Mdouble rad, const Coordinates& r)
    {
        return evaluateCGFunction(position, r);
    }
    
    Mdouble
    evaluateCGIntegral(const BaseInteraction& i, const Coordinates& r, IntegralType type = IntegralType::I_TO_P);
    
    Mdouble
    evaluateCGIntegral1D(const BaseInteraction& i, const Coordinates& r, IntegralType type = IntegralType::I_TO_P);
    
    std::vector<Mdouble> getCoefficients();
    
    Vec3D evaluateCGFunctionDerivatives(const Vec3D& position, const Coordinates& r);
    
    Mdouble evaluateCGFunctionDerivativeWithFD(const Vec3D& position, const Coordinates& r, const int i);
    
    Mdouble getEps() const;
    
    void setEps(Mdouble eps);
 
protected:
    
    std::vector<Mdouble> coefficients_;
    
    PolynomialType polynomialType_;
    
    Mdouble normalLength_;
    
    Vec3D normal_;
    
    unsigned currentInteraction_;
    
    Mdouble cutoff_;
    
    Mdouble eps_;
    
};
    
} //namespace CGFunctions
#include "Polynomial.hcc"
 
#endif