sph-cpp/particle.h at master · bboys/sph-cpp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#ifndef SPH_PHYSICS_PARTICLE_H
#define SPH_PHYSICS_PARTICLE_H

#include "utils.h"
#include <vector>

namespace Physics {

class Bucket;
class Particle;
class Buckets;

extern float Acc_ext[3];
extern float gravity[3];
extern float gravity_0[3];

extern float stiffness;
extern float sigma;
extern float mu;
extern float r_0;
extern float radius_factor;
extern float mass_0;
extern float density_0;
extern float pressure_0;

extern float num_neighbour_particles;

class Particle
{
public:
    float position[3];
    float radius;
    float velocity[3];
    float normal[3];
    float acceleration[3];
    float mass;
    float density;
    float pressure;

    float effective_radius;
    float effective_radius2;

    float dt;
    float DDcs;

    float F_press[3];
    float F_vis[3];
    float F_surface[3];

protected:
    Bucket *d_bucket;
    Buckets *d_buckets;
    static std::vector<float> wall_dens_vec;
    static std::vector<float> wall_vis_vec;

public:
    Particle() = delete;
    ~Particle();
    Particle(float position[3], Buckets *buckets);
    Particle(const Particle &other);
    Particle(Particle &&other);

    Particle &operator=(Particle const &other);
    Particle &operator=(Particle &&other);

    static void init_wall();

    float update_density();
    void update(float tstep);
    void update_density(float tstep);
    void update_position(float tstep);

protected:
    void swap(Particle &other);

    void update_press();
    void update_forces();

    void update_F_press(Particle *particle_i, Difference const &diff);
    void update_F_vis(Particle *particle_i, Difference const &diff);
    void update_F_surface(Particle *particle_i, Difference const &diff);
    void update_normal(Particle *particle_i, Difference const &diff);

    float W(float plen2) const;
    float W(Difference const &diff) const;
    void DW(Difference const &diff, float out[3]) const;
    float DDW(Difference const &diff) const;
    void DW_press(Difference const &diff, float out[3]) const;
    float DDW_vis(Difference const &diff) const;

    void update_bucket(bool force = false);

    float Z_wall_dens(const float distance);
    float Z_wall_vis(const float distance);

    float wall_distance();
    void wall_vector(float out[3]);
    void get_normal_by_idx(float out[3], size_t idx);
};

class ParticlesBase : public std::vector<Particle>
{
public:
    ParticlesBase();
    virtual ~ParticlesBase();
    void update(float tstep);
};

}

#endif