CFD_SMOKE2D_DIVMLS_GPU/DIVMLS.h at main · crocusia/CFD_SMOKE2D_DIVMLS_GPU · 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
#pragma once
#include<cuda_runtime.h>
#include<helper_cuda.h>
#include<memory>

#include"TArray.h"

#define RES_SMOKE	128

struct NODE {
	int		_neighbor[9];
	int		_constrain[4];
	double	_div[9];
	double	_P_Vec[(2 + 1) * 4];				//[(dim + divCoeffCount) * constrainCount]
	double	_B_Vec[(2 + 1) * 4][2 * 6];			//[(dim + divCoeffCount) * constrainCount][dim * coeffCount]
	double	_Bt_Vec[2 * 6][(2 + 1) * 4];					//[dim * coeffCount][(dim + divCoeffCount) * constrainCount]
	double	_W_Vec[(2 + 1) * 4][(2 + 1) * 4];	//[(dim + divCoeffCount) * constrainCount][(dim + divCoeffCount) * constrainCount]
	double	_W_Mat[2 * 6][(2 + 1) * 4];					//[dim * coeffCount][(dim + divCoeffCount) * constrainCount]
	double	_B_Mat[2][2 * 6];						//[dim][dim * coeffCount]
};

struct SVD {
	double _A[12 * 12];
	double _b[12];
	double _V[12 * 12];
	double _S[12];
	double _x[12];
};

class DIVMLS {
public:
	DIVMLS();
	~DIVMLS();

public:
	int _sizeNoEdge = RES_SMOKE * RES_SMOKE;
	int _size = (RES_SMOKE + 2) * (RES_SMOKE + 2);
	int _dim = 2;
	int _coeffCount = 6;
	int _divCoeffCount = 1;
	int _constrainCount = 4;
	double _dt;
	double3 _scale;

public:
	//Host Data
	double3*				_velocityH;
	int3*					_indexH;
	//Device Data
	NODE*					_nodes;
	SVD*					_svd;
	double3*				_velocityD;
	int3*					_indexD;
	double3*				_posD;
	int3*					_btIndexD;
	double3*				_btPosD;
	double3*				_resultD;

public:
	void initMLS();
	void freeMLS();
	void zeroMLS();
	void copyHtoD(double* ux, double* uy);
	void copyDtoH(double* dx, double* dy);
};