From 8f204f0ccf1131302adb55c90bb3c108ef82b183 Mon Sep 17 00:00:00 2001
From: Yang Gu <yang.gu@intel.com>
Date: Tue, 26 Sep 2017 18:40:13 +0800
Subject: [PATCH] Test shader storage buffer

---
 .../gles31/functional/es31fAtomicCounterTests.cpp  | 59 ++++++++++++++++++++++
 1 file changed, 59 insertions(+)

diff --git a/modules/gles31/functional/es31fAtomicCounterTests.cpp b/modules/gles31/functional/es31fAtomicCounterTests.cpp
index 878b2ae7..1cf91370 100644
--- a/modules/gles31/functional/es31fAtomicCounterTests.cpp
+++ b/modules/gles31/functional/es31fAtomicCounterTests.cpp
@@ -54,6 +54,18 @@ namespace Functional
 namespace
 {
 
+ static const std::string &csSource =
+        "#version 310 es\n"
+        "layout(local_size_x=1, local_size_y=1, local_size_z=1) in;\n"
+        "layout(std140, binding = 1) buffer blockName {\n"
+        "    uint data[2];\n"
+        "};\n"
+        "void main()\n"
+        "{\n"
+        "    data[0] = 3u;\n"
+        "    data[1] = 4u;\n"
+        "}\n";
+
 class AtomicCounterTest : public TestCase
 {
 public:
@@ -112,6 +124,7 @@ public:
 	void				init						(void);
 	void				deinit						(void);
 	IterateResult		iterate						(void);
+	void		testAtomicCounter						(void);
 
 private:
 	const TestSpec		m_spec;
@@ -912,8 +925,54 @@ bool AtomicCounterTest::checkAndLogCallValues (TestLog& log, const vector<deUint
 	return isOk;
 }
 
+void AtomicCounterTest::testAtomicCounter (void)
+{
+	const glw::Functions&		gl					= m_context.getRenderContext().getFunctions();
+	const glu::ShaderProgram	program		(m_context.getRenderContext(), glu::ProgramSources() << glu::ShaderSource(glu::SHADERTYPE_COMPUTE, csSource));
+	deUint32			programID		= program.getProgram();
+        if (!program.isOk())
+	{
+		tcu::print("------Compile failed.");
+		return;
+	}
+
+	gl.useProgram(programID);
+
+        unsigned int data[2] = {0u, 0u};
+        const glu::Buffer			shaderStorageBuffer		(m_context.getRenderContext());
+        gl.bindBuffer(GL_SHADER_STORAGE_BUFFER, *shaderStorageBuffer);
+        gl.bufferData(GL_SHADER_STORAGE_BUFFER, sizeof(unsigned int) * 2, data,
+                      GL_STATIC_DRAW);
+    // Bind shader storage buffer
+    gl.bindBufferBase(GL_SHADER_STORAGE_BUFFER, 1, *shaderStorageBuffer);
+
+    // Dispath compute
+    gl.dispatchCompute(1, 1, 1);
+
+    gl.finish();
+
+    
+        gl.bindBuffer(GL_SHADER_STORAGE_BUFFER, *shaderStorageBuffer);
+	void* ptr = gl.mapBufferRange(GL_SHADER_STORAGE_BUFFER, 0, sizeof(GLuint) * 2, GL_MAP_READ_BIT);
+
+	deMemcpy(&(data[0]), ptr, sizeof(GLuint) * 2);
+        tcu::print("--------------end: data[0] = %d, data[1] = %d\n", data[0], data[1]);
+
+	if (!gl.unmapBuffer(GL_SHADER_STORAGE_BUFFER))
+	{
+		tcu::print("Mapped buffer corrupted");
+	}
+
+
+	gl.bindBuffer(GL_SHADER_STORAGE_BUFFER, 0);
+	gl.useProgram(0);
+        tcu::print("\n=============================================\n");
+}
+
 TestCase::IterateResult AtomicCounterTest::iterate (void)
 {
+        testAtomicCounter();
+
 	const glw::Functions&		gl					= m_context.getRenderContext().getFunctions();
 	TestLog&					log					= m_testCtx.getLog();
 	const glu::Buffer			counterBuffer		(m_context.getRenderContext());
-- 
2.11.0