FluidSimulation.h

浏览该文件的文档.

#pragma once
#include "common/Export.h"
 
 
#include "fluids/FluidMath.h"
#include "fluids/FluidSdf.h"
 
#include <glm/glm.hpp>
 
#include <vector>
 
namespace eve::fluids {
 
struct SimGrid {
    glm::vec3 origin{0.f};
    glm::ivec3 dims{0};
    float cellSize = 1.f;
 
    static SimGrid make(const MeshSdf& sdf, float cellSize);
 
    int cellCount() const;
 
    int cellIndex(int x, int y, int z) const;
 
    bool inBounds(int x, int y, int z) const;
 
    glm::ivec3 cellOf(const glm::vec3& p) const;
};
 
class EVENGINE_API_DOMAINS FluidSimulation {
public:
    explicit FluidSimulation(int maxParticles, const FluidParams& params = FluidParams{});
 
    void setSdf(const MeshSdf& sdf);
 
    const MeshSdf& sdf() const { return sdf_; }
 
    void clear();
 
    int spawnDrop(const glm::vec3& center, float radius, int count);
 
    void step(float dt);
    void step(float dt, int substeps);
 
    int particleCount() const { return count_; }
 
    int maxParticles() const { return maxParticles_; }
 
    const std::vector<FluidParticle>& particles() const { return particles_; }
 
    std::vector<FluidParticle>& particles() { return particles_; }
 
    const std::vector<float>& densities() const { return densities_; }
 
    std::vector<float>& densities() { return densities_; }
 
    FluidParams& params() { return params_; }
 
    const FluidParams& params() const { return params_; }
 
private:
    void rebuildGrid();
    void computeDensitiesAndGrads();
    void computeLambdas();
    void applyPositionCorrections();
    void integrate(float dt);
 
    FluidParams                params_;
    MeshSdf                    sdf_;
    SimGrid                    grid_;
    int                        maxParticles_ = 0;
    int                        count_        = 0;
    std::vector<FluidParticle> particles_;
    std::vector<float>         densities_;
    std::vector<float>         lambdas_;
    std::vector<glm::vec3>     gradSums_;
    std::vector<int>           cellHead_;
    std::vector<int>           cellNext_;
};
 
}  // namespace eve::fluids