SnowSimulation/ParticleSystem.h at master · FZzzz/SnowSimulation · 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
112
113
114
115
116
117
118
119
120
121
122
123
124
#ifndef _PARTICLE_SYSTEM_H_
#define _PARTICLE_SYSTEM_H_

#include <GL/glew.h>
#include <vector>
#include <memory>
#include "common.h"
#include "Particle.h"
#include <cuda_gl_interop.h>

class ParticleSystem
{
public:

	ParticleSystem();
	~ParticleSystem();

	void Initialize();
	void InitializeCUDA();
	void Update();
	void UpdateCUDA();
	void Release();

	ParticleSet* AllocateSPHParticles(size_t n, float particle_mass);
	ParticleSet* AllocateDEMParticles(size_t n, float particle_mass);
	ParticleSet* AllocateBoundaryParticles();

	//void setParticles(std::vector<std::shared_ptr<Particle>> particles);
	void setSPHInitialVelocity(glm::vec3 vel);
	void setDEMInitialVelocity(glm::vec3 vel);
	void setParticleRadius(float particle_radius);
	void setHottestTemperature(float value) { m_hottest_temperature = value; };
	void setCoolestTemperature(float value) { m_coolest_temperature = value; };
	void setMaximumConnection(uint max_connection) { m_maximum_connection = max_connection; };
	inline ParticleSet* getSPHParticles() { return m_sph_particles; };
	inline ParticleSet* getDEMParticles() { return m_dem_particles; };
	inline ParticleSet* getBoundaryParticles() { return m_boundary_particles; };
	inline ParticleDeviceData* getBufferParticleDeviceData() { return m_buffer_device_data; };
//	inline std::vector<std::shared_ptr<Particle>>& getParticles() { return m_particles; };
	inline GLuint getSPH_VAO() { return m_sph_vao; };
	inline GLuint getSPH_VBO_0() { return m_sph_vbo[0]; };
	inline GLuint getSPH_VBO_1() { return m_sph_vbo[1]; };
	inline GLuint getSPH_EBO() { return m_sph_ebo; };

	inline GLuint getDEMVAO() { return m_dem_vao; };
	inline GLuint getDEM_VBO_0() { return m_dem_vbo[0]; };
	inline GLuint getDEM_VBO_1() { return m_dem_vbo[1]; };
	inline GLuint getDEMEBO() { return m_dem_ebo; };

	inline GLuint getBoundaryVAO() { return m_boundary_vao; };
	inline GLuint getBoundaryVBO() { return m_boundary_vbo; };
	inline GLuint getBoundaryEBO() { return m_boundary_ebo; };

	inline cudaGraphicsResource** getSPHCUDAGraphicsResource() { return m_sph_cuda_vbo_resource; };
	inline cudaGraphicsResource** getDEMCUDAGraphicsResource() { return m_dem_cuda_vbo_resource; };
	inline cudaGraphicsResource** getBoundaryCUDAGraphicsResource() { return &m_boundary_cuda_vbo_resource; };


	inline double& getUpdateTime() { return m_update_elased_time; };

	inline float getParticleRadius() { return m_particle_radius; };
	inline float getHottestTemperature() { return m_hottest_temperature; };
	inline float getCoolestTemperature() { return m_coolest_temperature; };
	inline uint getMaximumConnection() { return m_maximum_connection; };

private:

	//inner function
	void SetupCUDAMemory();
	void RegisterCUDAVBO();
	void GenerateGLBuffers();

	void UpdateGLBUfferData();

	ParticleSet* m_sph_particles;
	ParticleSet* m_dem_particles;
	ParticleSet* m_boundary_particles;

	ParticleDeviceData* m_buffer_device_data;

	/* OpenGL members */
	//std::vector<std::shared_ptr<Particle>> m_particles;
	std::vector<unsigned int> m_particle_indices;
	std::vector<unsigned int> m_dem_particle_indices;
	std::vector<unsigned int> m_boundary_particle_indices;

	float m_particle_radius;

	GLuint m_sph_vao;
	GLuint m_sph_vbo[2];
	GLuint m_sph_ebo;

	GLuint m_dem_vao;
	GLuint m_dem_vbo[2];
	//GLuint m_dem_vbo;
	GLuint m_dem_ebo;

	GLuint m_boundary_vao;
	GLuint m_boundary_vbo;
	GLuint m_boundary_ebo;

	GLuint m_sph_cuda_vbo[2];
	GLuint m_dem_cuda_vbo[2];
	//GLuint m_dem_cuda_vbo;
	GLuint m_boundary_cuda_vbo;

	struct cudaGraphicsResource* m_sph_cuda_vbo_resource[2];
	struct cudaGraphicsResource* m_dem_cuda_vbo_resource[2];
	struct cudaGraphicsResource* m_boundary_cuda_vbo_resource;

	//struct cudaGraphicsResource* m_dem_color_vbo_res;

	float m_hottest_temperature;
	float m_coolest_temperature;

	glm::vec3 m_sph_init_vel;
	glm::vec3 m_dem_init_vel;

	uint m_maximum_connection;

	double m_update_elased_time;
};

#endif