CapillaryWave.h

Go to the documentation of this file.

#pragma once
#include "Node.h"
#include "Topology/HeightField.h"
 
namespace dyno
{
    template<typename TDataType>
    class CapillaryWave : public Node
    {
        DECLARE_TCLASS(CapillaryWave, TDataType)
    public:
        typedef typename TDataType::Real Real;
        typedef typename ::dyno::Vector<Real, 2> Coord2D;
        typedef typename ::dyno::Vector<Real, 3> Coord3D;
        typedef typename ::dyno::Vector<Real, 4> Coord4D;
 
        CapillaryWave();
        ~CapillaryWave() override;
 
        std::string getNodeType() override { return "Height Fields"; }
 
    public:
        DEF_VAR(Real, WaterLevel, 2, "");
 
        DEF_VAR(uint, Resolution, 512, "");
 
        DEF_VAR(Real, Length, 512.0f, "The simulated region size in meters");
 
        DEF_VAR(Real, Viscosity, 0.001, "Fluid viscosity");
 
    public:
        DEF_ARRAY2D_STATE(Coord4D, Height, DeviceType::GPU, "");
 
        DEF_INSTANCE_STATE(HeightField<TDataType>, HeightField, "Height field");
 
    public:
        void setOriginX(int x) { mOriginX = x; }
        void setOriginY(int y) { mOriginY = y; }
 
        int getOriginX() { return mOriginX; }
        int getOriginZ() { return mOriginY; }
 
        Real getRealGridSize() { return mRealGridSize; }
 
        Coord2D getOrigin() { return Coord2D(mOriginX * mRealGridSize, mOriginY * mRealGridSize); }
 
        //TODO: make improvements
        void moveDynamicRegion(int nx, int ny);
 
    protected:
        void resetStates() override;
 
        void updateStates() override;
 
    protected:
        DArray2D<Coord4D> mDeviceGrid;
        DArray2D<Coord4D> mDeviceGridNext;
        DArray2D<Coord4D> mDeviceGridOld;
 
    private:
        Real mRealGridSize;
 
        int mOriginX = 0;
        int mOriginY = 0;
    };
 
    IMPLEMENT_TCLASS(CapillaryWave, TDataType)
}