CG.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 CG_H
#define CG_H
 
#include "CG/BaseCG.h"
#include <vector>
 
class CGHandler;
 
class Function;
 
class BaseParticle;
 
class BaseInteraction;
 
#include "CG/CGPoint.h"
#include "DPMBase.h"
#include "CG/CGHandler.h"
#include "Species/ParticleSpecies.h"
#include "Logger.h"
#include "CG/Functions/Lucy.h"
#include "CG/Functions/Heaviside.h"
 
template<class Coordinates = CGCoordinates::O,
        template<class> class BaseFunction=CGFunctions::Lucy,
        class Fields=CGFields::StandardFields>
class CG : public BaseCG
{
public:
    typedef BaseFunction<Coordinates> Function;
    typedef CGPoint<Coordinates, Fields> Point;
    
    CG() = default;
    
    CG(Mdouble width, unsigned n);
    
    CG(const CG& p) = default;
    
    virtual ~CG() = default;
    
    CG<Coordinates, BaseFunction, Fields>* copy() const override;
    
    void write(std::ostream& os) const override;
    
    void writeAll(std::ostream& os) const;
    
    std::string getName() const override;
    
    void initialise() override;
    
    virtual void createMesh();
    
    void evaluate() override;
    
    Point evaluateAverage();
    
    Point evaluateTotal();
    
    void evaluateParticleAtPoint(Fields& currentInteraction, const BaseParticle& p, Point& r);
    
    void evaluateParticle(const BaseParticle& p);
    
    void evaluateContact(const BaseInteraction& i);
    
    IntegralType getIntegralType(const BaseInteraction& c);
    
    void finish() override;
    
    const Point& getPoint(size_t i) const
    {
        return points_[i];
    }
    
    const std::vector<Point>& getPoints() const
    {
        return points_;
    }
    
    Function& getFunction()
    {
        return function_;
    }
    
    void setStandardDeviation(Mdouble std) override
    {
        function_.setStandardDeviation(std);
    }
    
    void setRadius(Mdouble radius) override
    {
        if (std::is_base_of<CGCoordinates::Base_X_Y_Z, Coordinates>::value)
        {
            function_.setStandardDeviation(radius * sqrt(.2));
        }
        else if (std::is_base_of<CGCoordinates::Base_XY_XZ_YZ, Coordinates>::value)
        {
            function_.setStandardDeviation(radius * sqrt(.4));
        }
        else /*XYZ*/ {
            function_.setStandardDeviation(radius * sqrt(.6));
        }
    }
    
    void setWidth(Mdouble width) override
    {
        function_.setWidth(width);
    }
    
    Mdouble getWidth() const override
    {
        return function_.getWidth();
    }
 
protected:
    
    void resetVariables();
    
    void volumeAverageVariables();
    
    void writeVariables();
    
    void evaluateCommon();
    
    void outputSumOfVariables();
 
protected:
    
    std::vector<Point> points_;
    
    Function function_;
    
    
};
 
#include "CG/CG.hcc"
 
#endif