tests/spec/arb_compute_shader/common.c

   1 /*
   2  * Copyright (c) 2014 - 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 DEALINGS
  21  * IN THE SOFTWARE.
  22  */
  23
  24 /** @file common.c
  25  *
  26  * Utility code for building a compute shader program.
  27  */
  28
  29 #include "common.h"
  30
  31 char *
  32 concat(char *hunk0, ...)
  33 {
  34         char *s = hunk0;
  35         char *hunk;
  36         va_list ap;
  37
  38         va_start(ap, hunk0);
  39
  40         while ((hunk = va_arg(ap, char *))) {
  41                 char *t = s;
  42                 (void)!asprintf(&s, "%s\n%s", t, hunk);
  43                 free(t);
  44                 free(hunk);
  45         }
  46
  47         va_end(ap);
  48         return s;
  49 }
  50
  51 GLuint
  52 generate_cs_prog(unsigned x, unsigned y, unsigned z, char *ext,
  53                  char *src)
  54 {
  55         char *source = NULL;
  56
  57         if (ext == NULL)
  58                 ext = hunk("");
  59
  60         (void)!asprintf(&source,
  61                  "#version 330\n"
  62                  "#extension GL_ARB_compute_shader : enable\n"
  63                  "%s\n"
  64                  "layout(local_size_x = %d, local_size_y = %d, local_size_z = %d) in;\n"
  65                  "\n"
  66                  "%s\n",
  67                  ext,
  68                  x, y, z,
  69                  src
  70                  );
  71         free(ext);
  72         free(src);
  73
  74         GLuint prog = glCreateProgram();
  75
  76         GLuint shader =
  77                 piglit_compile_shader_text_nothrow(GL_COMPUTE_SHADER, source);
  78
  79         free(source);
  80
  81         if (!shader) {
  82                 glDeleteProgram(prog);
  83                 return 0;
  84         }
  85
  86         glAttachShader(prog, shader);
  87
  88         glLinkProgram(prog);
  89
  90         glDeleteShader(shader);
  91
  92         if (!piglit_link_check_status(prog)) {
  93                 glDeleteProgram(prog);
  94                 return 0;
  95         }
  96
  97         return prog;
  98 }