pic_solver2.h

Go to the documentation of this file.

// Copyright (c) 2018 Doyub Kim
//
// I am making my contributions/submissions to this project solely in my
// personal capacity and am not conveying any rights to any intellectual
// property of any third parties.
 
#ifndef INCLUDE_JET_PIC_SOLVER2_H_
#define INCLUDE_JET_PIC_SOLVER2_H_
 
#include <jet/grid_fluid_solver2.h>
#include <jet/particle_emitter2.h>
#include <jet/particle_system_data2.h>
 
namespace jet {
 
class PicSolver2 : public GridFluidSolver2 {
 public:
    class Builder;
 
    PicSolver2();
 
    PicSolver2(
        const Size2& resolution,
        const Vector2D& gridSpacing,
        const Vector2D& gridOrigin);
 
    virtual ~PicSolver2();
 
    ScalarGrid2Ptr signedDistanceField() const;
 
    const ParticleSystemData2Ptr& particleSystemData() const;
 
    const ParticleEmitter2Ptr& particleEmitter() const;
 
    void setParticleEmitter(const ParticleEmitter2Ptr& newEmitter);
 
    static Builder builder();
 
 protected:
    Array2<char> _uMarkers;
    Array2<char> _vMarkers;
 
    void onInitialize() override;
 
    void onBeginAdvanceTimeStep(double timeIntervalInSeconds) override;
 
    void computeAdvection(double timeIntervalInSeconds) override;
 
    ScalarField2Ptr fluidSdf() const override;
 
    virtual void transferFromParticlesToGrids();
 
    virtual void transferFromGridsToParticles();
 
    virtual void moveParticles(double timeIntervalInSeconds);
 
 private:
    size_t _signedDistanceFieldId;
    ParticleSystemData2Ptr _particles;
    ParticleEmitter2Ptr _particleEmitter;
 
    void extrapolateVelocityToAir();
 
    void buildSignedDistanceField();
 
    void updateParticleEmitter(double timeIntervalInSeconds);
};
 
typedef std::shared_ptr<PicSolver2> PicSolver2Ptr;
 
 
class PicSolver2::Builder final
    : public GridFluidSolverBuilderBase2<PicSolver2::Builder> {
 public:
    PicSolver2 build() const;
 
    PicSolver2Ptr makeShared() const;
};
 
}  // namespace jet
 
#endif  // INCLUDE_JET_PIC_SOLVER2_H_