tests/spec/arb_shader_storage_buffer_object/max-ssbo-size.c

   1 /*
   2  * Copyright © 2015 Intel Corporation
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  21  * DEALINGS IN THE SOFTWARE.
  22  */
  23
  24 /** @file max-ssbo-size.c
  25  *
  26  * Tests linking and drawing with shader storage buffer objects of size
  27  * MAX_SHADER_STORAGE_BLOCK_SIZE.
  28  *
  29  * Based on ARB_uniform_buffer_object's maxuniformblocksize.c
  30  */
  31
  32 #include "piglit-util-gl.h"
  33
  34 PIGLIT_GL_TEST_CONFIG_BEGIN
  35
  36         config.supports_gl_compat_version = 32;
  37         config.supports_gl_core_version = 32;
  38         config.window_visual = PIGLIT_GL_VISUAL_RGBA | PIGLIT_GL_VISUAL_DOUBLE;
  39         config.khr_no_error_support = PIGLIT_NO_ERRORS;
  40
  41 PIGLIT_GL_TEST_CONFIG_END
  42
  43 static enum {
  44         VS,
  45         VS_EXCEED,
  46         FS,
  47         FS_EXCEED,
  48 } mode;
  49
  50 static void
  51 usage(const char *name)
  52 {
  53         fprintf(stderr, "usage: %s <vs | vsexceed | fs | fsexceed>\n",
  54                 name);
  55         piglit_report_result(PIGLIT_FAIL);
  56 }
  57
  58 enum piglit_result
  59 piglit_display(void)
  60 {
  61         const char *vs_ssbo_template =
  62                 "#version 130\n"
  63                 "#extension GL_ARB_shader_storage_buffer_object : enable\n"
  64                 "#extension GL_ARB_uniform_buffer_object : enable\n"
  65                 "\n"
  66                 "varying vec4 vary;"
  67                 "in vec4 piglit_vertex;\n"
  68                 "\n"
  69                 "layout(std140) buffer ssbo {\n"
  70                 "       vec4 v[%d];\n"
  71                 "};\n"
  72                 "uniform int i;\n"
  73                 "\n"
  74                 "void main() {\n"
  75                 "       gl_Position = piglit_vertex;\n"
  76                 "       vary = v[i];\n"
  77                 "}\n";
  78
  79         const char *fs_template =
  80                 "#version 130\n"
  81                 "#extension GL_ARB_shader_storage_buffer_object : enable\n"
  82                 "\n"
  83                 "varying vec4 vary;"
  84                 "\n"
  85                 "void main() {\n"
  86                 "       gl_FragColor = vary;\n"
  87                 "}\n";
  88
  89         const char *vs_template =
  90                 "#version 130\n"
  91                 "#extension GL_ARB_shader_storage_buffer_object : enable\n"
  92                 "in vec4 piglit_vertex;\n"
  93                 "\n"
  94                 "void main() {\n"
  95                 "       gl_Position = piglit_vertex;\n"
  96                 "}\n";
  97
  98         const char *fs_ssbo_template =
  99                 "#version 130\n"
 100                 "#extension GL_ARB_shader_storage_buffer_object : enable\n"
 101                 "#extension GL_ARB_uniform_buffer_object : enable\n"
 102                 "\n"
 103                 "layout(std140) buffer ssbo {\n"
 104                 "       vec4 v[%d];\n"
 105                 "};\n"
 106                 "uniform int i;\n"
 107                 "\n"
 108                 "void main() {\n"
 109                 "       gl_FragColor = v[i];\n"
 110                 "}\n";
 111
 112         char *vs_source, *fs_source;
 113         GLint max_size, vec4s, i_location;
 114         GLuint vs, fs, prog, bo;
 115         GLenum target;
 116         float *data;
 117         size_t size;
 118         bool pass = true;
 119         bool link_should_fail;
 120         GLint num_vertex_ssbo;
 121         const float green[4] = { 0, 1, 0, 0 };
 122         int test_index;
 123
 124         piglit_require_extension("GL_ARB_shader_storage_buffer_object");
 125         piglit_require_extension("GL_ARB_uniform_buffer_object");
 126
 127         glGetIntegerv(GL_MAX_SHADER_STORAGE_BLOCK_SIZE, &max_size);
 128         printf("Max shader storage block size: %d\n", max_size);
 129         vec4s = max_size / 4 / 4;
 130
 131         glGetIntegerv(GL_MAX_VERTEX_SHADER_STORAGE_BLOCKS, &num_vertex_ssbo);
 132
 133         switch (mode) {
 134         case VS:
 135                 target = GL_VERTEX_SHADER;
 136                 link_should_fail = false;
 137                 test_index = vec4s - 1;
 138                 if (num_vertex_ssbo == 0)
 139                         piglit_report_result(PIGLIT_SKIP);
 140                 break;
 141         case VS_EXCEED:
 142                 target = GL_VERTEX_SHADER;
 143                 link_should_fail = true;
 144                 vec4s++;
 145                 test_index = vec4s - 2;
 146                 if (num_vertex_ssbo == 0)
 147                         piglit_report_result(PIGLIT_SKIP);
 148                 break;
 149         case FS:
 150                 target = GL_FRAGMENT_SHADER;
 151                 link_should_fail = false;
 152                 test_index = vec4s - 1;
 153                 break;
 154         case FS_EXCEED:
 155                 target = GL_FRAGMENT_SHADER;
 156                 link_should_fail = true;
 157                 vec4s++;
 158                 test_index = vec4s - 2;
 159                 break;
 160         default:
 161                 assert(false);
 162                 target = GL_NONE;
 163                 link_should_fail = false;
 164         }
 165
 166         switch (target) {
 167         case GL_VERTEX_SHADER:
 168                 (void)!asprintf(&vs_source, vs_ssbo_template, vec4s);
 169                 (void)!asprintf(&fs_source, "%s", fs_template);
 170                 printf("Testing VS with shader storage block vec4 v[%d]\n", vec4s);
 171                 break;
 172         case GL_FRAGMENT_SHADER:
 173                 (void)!asprintf(&vs_source, "%s", vs_template);
 174                 (void)!asprintf(&fs_source, fs_ssbo_template, vec4s);
 175                 printf("Testing FS with shader storage block vec4 v[%d]\n", vec4s);
 176                 break;
 177         default:
 178                 piglit_report_result(PIGLIT_FAIL);
 179         }
 180
 181         vs = piglit_compile_shader_text(GL_VERTEX_SHADER, vs_source);
 182         fs = piglit_compile_shader_text(GL_FRAGMENT_SHADER, fs_source);
 183
 184         prog = glCreateProgram();
 185         glAttachShader(prog, vs);
 186         glAttachShader(prog, fs);
 187         glLinkProgram(prog);
 188
 189         if (link_should_fail) {
 190                 if (!piglit_link_check_status_quiet(prog)) {
 191                         printf("Failed to link with shader storage block vec4 "
 192                                "v[%d]\n", vec4s);
 193                         piglit_report_result(PIGLIT_PASS);
 194                 }
 195         } else {
 196                 if (!piglit_link_check_status_quiet(prog)) {
 197                         fprintf(stderr,
 198                                 "Failed to link with shader storage block vec4 "
 199                                 "v[%d]\n", vec4s);
 200                         return PIGLIT_FAIL;
 201                 }
 202         }
 203
 204         size = vec4s * 4 * sizeof(float);
 205         glGenBuffers(1, &bo);
 206         glBindBuffer(GL_SHADER_STORAGE_BUFFER, bo);
 207         glBufferData(GL_SHADER_STORAGE_BUFFER, size, NULL, GL_DYNAMIC_DRAW);
 208         data = glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_WRITE);
 209         memset(data, 0, size);
 210
 211         /* The whole shader storage buffer will be zeros, except for the
 212          * entry at v[test_index] which will be green.
 213          */
 214         data[test_index * 4 + 0] = green[0];
 215         data[test_index * 4 + 1] = green[1];
 216         data[test_index * 4 + 2] = green[2];
 217         data[test_index * 4 + 3] = green[3];
 218         glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
 219
 220         glUseProgram(prog);
 221         i_location = glGetUniformLocation(prog, "i");
 222         glUniform1i(i_location, test_index);
 223
 224         glShaderStorageBlockBinding(prog, 0, 0);
 225         glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 0, bo);
 226         piglit_draw_rect(-1, -1, 2, 2);
 227
 228         pass = piglit_probe_rect_rgba(0, 0, piglit_width, piglit_height, green);
 229
 230         glDeleteProgram(prog);
 231
 232         piglit_present_results();
 233
 234         return pass ? PIGLIT_PASS : PIGLIT_FAIL;
 235 }
 236
 237
 238 void
 239 piglit_init(int argc, char **argv)
 240 {
 241         if (argc < 2)
 242                 usage(argv[0]);
 243
 244         if (strcmp(argv[1], "vs") == 0)
 245                 mode = VS;
 246         else if (strcmp(argv[1], "vsexceed") == 0)
 247                 mode = VS_EXCEED;
 248         else if (strcmp(argv[1], "fs") == 0)
 249                 mode = FS;
 250         else if (strcmp(argv[1], "fsexceed") == 0)
 251                 mode = FS_EXCEED;
 252         else
 253                 usage(argv[0]);
 254 }