Parametrize lattice resolution

1 files changed, 2 insertions, 1 deletions

diff --git a/src/shader/wrap/compute_shader.h b/src/shader/wrap/compute_shader.h
index 74c8270..816b925 100644
--- a/src/shader/wrap/compute_shader.h
+++ b/src/shader/wrap/compute_shader.h
@@ -26,7 +26,8 @@ public:
 	bool isGood() const;
 
 	GLuint setUniform(const std::string& name, float x, float y) const;
+	GLuint setUniform(const std::string& name, unsigned int value) const;
 
 	void workOn(GLuint bufferA, GLuint bufferB, GLuint bufferC) const;
-	void dispatch() const;
+	void dispatch(GLuint nX, GLuint nY) const;
 };