Merge streaming into collide shaderpattern/ab

i.e. implement the A-B pattern.

Dispatching only one compute shader per interaction-less simulation step
already yields very noticeable performance gains. All cell types are now
fully handled by the collide shader which further simplifies the code.

2 files changed, 28 insertions, 90 deletions

diff --git a/src/shader/code/collide.glsl b/src/shader/code/collide.glsl
index 673a285..4efd7fb 100644
--- a/src/shader/code/collide.glsl
+++ b/src/shader/code/collide.glsl
@@ -61,7 +61,7 @@ float get(uint x, uint y, int i, int j) {
 }
 
 void set(uint x, uint y, int i, int j, float v) {
-	collideCells[indexOfLatticeCell(x,y) + indexOfDirection(i,j)] = v;
+	streamCells[indexOfLatticeCell(x,y) + indexOfDirection(i,j)] = v;
 }
 
 void setFluid(uint x, uint y, vec2 v) {
@@ -105,6 +105,18 @@ bool isBulkFluidCell(int material) {
 	return material == 1 || material == 4 || material == 5 || material == 6;
 }
 
+bool isBounceBackCell(int material) {
+	return material == 2 || material == 3;
+}
+
+bool isInflowCell(int material) {
+	return material == 5;
+}
+
+bool isOutflowCell(int material) {
+	return material == 6;
+}
+
 float getExternalMassInflux(int material) {
 	if ( material == 4 ) {
 		return 1.5;
@@ -113,7 +125,7 @@ float getExternalMassInflux(int material) {
 	}
 };
 
-/// Actual collide kernel
+/// Actual collide&stream kernel
 
 void main() {
 	const uint x = gl_GlobalInvocationID.x;
@@ -125,14 +137,14 @@ void main() {
 
 	const int material = getMaterial(x,y);
 
-	if ( isBulkFluidCell(material) ) {
-		float d = max(density(x,y), getExternalMassInflux(material));
-		vec2  v = velocity(x,y,d);
+	float d = max(density(x,y), getExternalMassInflux(material));
+	vec2  v = velocity(x,y,d);
 
-		if ( material == 5 ) {
+	if ( isBulkFluidCell(material) ) {
+		if ( isInflowCell(material) ) {
 			v = vec2(0.1,0.0);
 		}
-		if ( material == 6 ) {
+		if ( isOutflowCell(material) ) {
 			d = 1.0;
 		}
 
@@ -140,7 +152,15 @@ void main() {
 
 		for ( int i = -1; i <= 1; ++i ) {
 			for ( int j = -1; j <= 1; ++j ) {
-				set(x,y,i,j, get(x,y,i,j) + omega * (equilibrium(d,v,i,j) - get(x,y,i,j)));
+				set(x+i,y+j,i,j, get(x,y,i,j) + omega * (equilibrium(d,v,i,j) - get(x,y,i,j)));
+			}
+		}
+	}
+
+	if ( isBounceBackCell(material) ) {
+		for ( int i = -1; i <= 1; ++i ) {
+			for ( int j = -1; j <= 1; ++j ) {
+				set(x+(-1)*i,y+(-1)*j,(-1)*i,(-1)*j, get(x,y,i,j) + omega * (equilibrium(d,v,i,j) - get(x,y,i,j)));
 			}
 		}
 	}
diff --git a/src/shader/code/stream.glsl b/src/shader/code/stream.glsl
deleted file mode 100644
index 59aed98..0000000
--- a/src/shader/code/stream.glsl
+++ /dev/null
@@ -1,82 +0,0 @@
-static const std::string STREAM_SHADER_CODE = R"(
-#version 430
-
-layout (local_size_x = 1, local_size_y = 1) in;
-
-layout (std430, binding=1) buffer bufferCollide  { float collideCells[];  };
-layout (std430, binding=2) buffer bufferStream   { float streamCells[];   };
-layout (std430, binding=3) buffer bufferFluid    { float fluidCells[];    };
-
-/// LBM constants
-
-uniform uint nX;
-uniform uint nY;
-
-const uint q = 9;
-
-// Array indexing
-
-uint indexOfDirection(int i, int j) {
-	return 3*(i+1) + (j+1);
-}
-
-uint indexOfLatticeCell(uint x, uint y) {
-	return q*nX*y + q*x;
-}
-
-uint indexOfFluidVertex(uint x, uint y) {
-	return 3*nX*y + 3*x;
-}
-
-/// Data access
-
-float get(uint x, uint y, int i, int j) {
-	return collideCells[indexOfLatticeCell(x,y) + indexOfDirection(i,j)];
-}
-
-void set(uint x, uint y, int i, int j, float v) {
-	streamCells[indexOfLatticeCell(x,y) + indexOfDirection(i,j)] = v;
-}
-
-int getMaterial(uint x, uint y) {
-	const uint idx = indexOfFluidVertex(x, y);
-	return int(fluidCells[idx + 2]);
-}
-
-/// Boundary conditions
-
-void bounceBack(uint x, uint y) {
-	for ( int i = -1; i <= 1; ++i ) {
-		for ( int j = -1; j <= 1; ++j ) {
-			set(x,y,i,j, get(x,y,(-1)*i,(-1)*j));
-		}
-	}
-}
-
-/// Actual stream kernel
-
-void main() {
-	const uint x = gl_GlobalInvocationID.x;
-	const uint y = gl_GlobalInvocationID.y;
-
-	if ( !(x < nX && y < nY) ) {
-		return;
-	}
-
-	const int material = getMaterial(x,y);
-
-	if ( material == 0 ) {
-		return;
-	}
-
-	if ( material == 2 || material == 3 ) {
-		bounceBack(x,y);
-	}
-
-	for ( int i = -1; i <= 1; ++i ) {
-		for ( int j = -1; j <= 1; ++j ) {
-			set(x+i,y+j,i,j, get(x,y,i,j));
-		}
-	}
-}
-)";