LevelSetWall.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 LevelSetWall_H
#define LevelSetWall_H
 
#include "BaseWall.h"
#include "Math/Vector.h"
 
class LevelSetWall final : public BaseWall
{
public:
    
    // a set of predefined shapes for which level set values can be set easily
    enum class Shape
    {
        Sphere,
        Cube,
        Diamond,
        FourSided,
        Cylinder
    };
    
    // Constructor; currently only allows predefined shapes
    LevelSetWall(Shape s, double radius, ParticleSpecies* sp = nullptr);
    
    ~LevelSetWall() override;
    
    LevelSetWall* copy() const override;
    
    using BaseWall::move;
 
//    /*!
//     * \brief Returns the distance of the wall to the particle.
//     */
//    Mdouble getDistance(Vec3D position) const;
    
    bool getDistanceAndNormal(const BaseParticle& p, Mdouble& distance, Vec3D& normal_return) const override;
    
    void read(std::istream& is) override;
    
    std::string getName() const override;
    
    void writeVTK(VTKContainer& vtk) const override;
    
    void writeToFile(int n, double radiusContact) const;
 
private:
    
    // Evaluates the distance and normal direction to the surface defined by the level set. Needed for contact detection.
    // If distance is bigger than radius_+radiusContact, no normal & distance is returned because no contact is possible.
    // Else, it returns the overlap (level set value) and normal direction (-gradient).
    bool getDistanceAndNormalLabCoordinates(Vec3D position, Mdouble interactionRadius, Mdouble& distance,
                                            Vec3D& normal) const;
    
    void setShapeSphere();
    
    void setShapeCube();
    
    //Better interpolated than a square as the edges of the level set align with the mesh.
    void setShapeDiamond();
 
    void setShapeCylinder();
 
    void setShapeFourSided();
    
    void createVTKSphere();
    
    void createVTKCube();
    
    void createVTKDiamond();
    
    void createVTK();
    
    
    // N determines the number of level set values (2N+1)^3;
    // \todo template the level set with N.
    static const int N = 10;
    
    // discrete set of level-set values; levelSet_[i,j,k] is the value of the level set at (x,y,z)=(i,j,k)/N*radius_.
    double levelSet_[2 * N + 1][2 * N + 1][2 * N + 1];
    
    // the radius of a sphere that envelopes the object
    double radius_ = 1;
    
    VTKContainer vtkLabFrame_;
};
 
#endif