src/gallium/tests/trivial/quad-tex.c

   1 /**************************************************************************
   2  *
   3  * Copyright © 2010 Jakob Bornecrantz
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the "Software"),
   7  * to deal in the Software without restriction, including without limitation
   8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   9  * and/or sell copies of the Software, and to permit persons to whom the
  10  * Software is furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice (including the next
  13  * paragraph) shall be included in all copies or substantial portions of the
  14  * Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  21  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  22  * DEALINGS IN THE SOFTWARE.
  23  *
  24  **************************************************************************/
  25
  26
  27 #define USE_TRACE 0
  28 #define WIDTH 300
  29 #define HEIGHT 300
  30 #define NEAR 0
  31 #define FAR 1
  32 #define FLIP 0
  33
  34 /* pipe_*_state structs */
  35 #include "pipe/p_state.h"
  36 /* pipe_context */
  37 #include "pipe/p_context.h"
  38 /* pipe_screen */
  39 #include "pipe/p_screen.h"
  40 /* PIPE_* */
  41 #include "pipe/p_defines.h"
  42 /* TGSI_SEMANTIC_{POSITION|GENERIC} */
  43 #include "pipe/p_shader_tokens.h"
  44 /* pipe_buffer_* helpers */
  45 #include "util/u_inlines.h"
  46
  47 /* constant state object helper */
  48 #include "cso_cache/cso_context.h"
  49
  50 #include "util/macros.h"
  51 /* u_sampler_view_default_template */
  52 #include "util/u_sampler.h"
  53 /* debug_dump_surface_bmp */
  54 #include "util/u_debug_image.h"
  55 /* util_draw_vertex_buffer helper */
  56 #include "util/u_draw_quad.h"
  57 /* FREE & CALLOC_STRUCT */
  58 #include "util/u_memory.h"
  59 /* util_make_[fragment|vertex]_passthrough_shader */
  60 #include "util/u_simple_shaders.h"
  61 /* to get a hardware pipe driver */
  62 #include "pipe-loader/pipe_loader.h"
  63
  64 struct program
  65 {
  66         struct pipe_loader_device *dev;
  67         struct pipe_screen *screen;
  68         struct pipe_context *pipe;
  69         struct cso_context *cso;
  70
  71         struct pipe_blend_state blend;
  72         struct pipe_depth_stencil_alpha_state depthstencil;
  73         struct pipe_rasterizer_state rasterizer;
  74         struct pipe_sampler_state sampler;
  75         struct pipe_viewport_state viewport;
  76         struct pipe_framebuffer_state framebuffer;
  77         struct cso_velems_state velem;
  78
  79         void *vs;
  80         void *fs;
  81
  82         union pipe_color_union clear_color;
  83
  84         struct pipe_resource *vbuf;
  85         struct pipe_resource *target;
  86         struct pipe_resource *tex;
  87         struct pipe_sampler_view *view;
  88 };
  89
  90 static void init_prog(struct program *p)
  91 {
  92         struct pipe_surface surf_tmpl;
  93         ASSERTED int ret;
  94
  95         /* find a hardware device */
  96         ret = pipe_loader_probe(&p->dev, 1);
  97         assert(ret);
  98
  99         /* init a pipe screen */
 100         p->screen = pipe_loader_create_screen(p->dev);
 101         assert(p->screen);
 102
 103         /* create the pipe driver context and cso context */
 104         p->pipe = p->screen->context_create(p->screen, NULL, 0);
 105         p->cso = cso_create_context(p->pipe, 0);
 106
 107         /* set clear color */
 108         p->clear_color.f[0] = 0.3;
 109         p->clear_color.f[1] = 0.1;
 110         p->clear_color.f[2] = 0.3;
 111         p->clear_color.f[3] = 1.0;
 112
 113         /* vertex buffer */
 114         {
 115                 float vertices[4][2][4] = {
 116                         {
 117                                 { 0.9f, 0.9f, 0.0f, 1.0f },
 118                                 { 1.0f, 1.0f, 0.0f, 1.0f }
 119                         },
 120                         {
 121                                 { -0.9f, 0.9f, 0.0f, 1.0f },
 122                                 {  0.0f, 1.0f, 0.0f, 1.0f }
 123                         },
 124                         {
 125                                 { -0.9f, -0.9f, 0.0f, 1.0f },
 126                                 {  0.0f,  0.0f, 1.0f, 1.0f }
 127                         },
 128                         {
 129                                 { 0.9f, -0.9f, 0.0f, 1.0f },
 130                                 { 1.0f,  0.0f, 1.0f, 1.0f }
 131                         }
 132                 };
 133
 134                 p->vbuf = pipe_buffer_create(p->screen, PIPE_BIND_VERTEX_BUFFER,
 135                                              PIPE_USAGE_DEFAULT, sizeof(vertices));
 136                 pipe_buffer_write(p->pipe, p->vbuf, 0, sizeof(vertices), vertices);
 137         }
 138
 139         /* render target texture */
 140         {
 141                 struct pipe_resource tmplt;
 142                 memset(&tmplt, 0, sizeof(tmplt));
 143                 tmplt.target = PIPE_TEXTURE_2D;
 144                 tmplt.format = PIPE_FORMAT_B8G8R8A8_UNORM; /* All drivers support this */
 145                 tmplt.width0 = WIDTH;
 146                 tmplt.height0 = HEIGHT;
 147                 tmplt.depth0 = 1;
 148                 tmplt.array_size = 1;
 149                 tmplt.last_level = 0;
 150                 tmplt.bind = PIPE_BIND_RENDER_TARGET;
 151
 152                 p->target = p->screen->resource_create(p->screen, &tmplt);
 153         }
 154
 155         /* sampler texture */
 156         {
 157                 uint32_t *ptr;
 158                 struct pipe_transfer *t;
 159                 struct pipe_resource t_tmplt;
 160                 struct pipe_sampler_view v_tmplt;
 161                 struct pipe_box box;
 162
 163                 memset(&t_tmplt, 0, sizeof(t_tmplt));
 164                 t_tmplt.target = PIPE_TEXTURE_2D;
 165                 t_tmplt.format = PIPE_FORMAT_B8G8R8A8_UNORM; /* All drivers support this */
 166                 t_tmplt.width0 = 2;
 167                 t_tmplt.height0 = 2;
 168                 t_tmplt.depth0 = 1;
 169                 t_tmplt.array_size = 1;
 170                 t_tmplt.last_level = 0;
 171                 t_tmplt.bind = PIPE_BIND_RENDER_TARGET;
 172
 173                 p->tex = p->screen->resource_create(p->screen, &t_tmplt);
 174
 175                 memset(&box, 0, sizeof(box));
 176                 box.width = 2;
 177                 box.height = 2;
 178                 box.depth = 1;
 179
 180                 ptr = p->pipe->transfer_map(p->pipe, p->tex, 0, PIPE_TRANSFER_WRITE, &box, &t);
 181                 ptr[0] = 0xffff0000;
 182                 ptr[1] = 0xff0000ff;
 183                 ptr[2] = 0xff00ff00;
 184                 ptr[3] = 0xffffff00;
 185                 p->pipe->transfer_unmap(p->pipe, t);
 186
 187                 u_sampler_view_default_template(&v_tmplt, p->tex, p->tex->format);
 188
 189                 p->view = p->pipe->create_sampler_view(p->pipe, p->tex, &v_tmplt);
 190         }
 191
 192         /* disabled blending/masking */
 193         memset(&p->blend, 0, sizeof(p->blend));
 194         p->blend.rt[0].colormask = PIPE_MASK_RGBA;
 195
 196         /* no-op depth/stencil/alpha */
 197         memset(&p->depthstencil, 0, sizeof(p->depthstencil));
 198
 199         /* rasterizer */
 200         memset(&p->rasterizer, 0, sizeof(p->rasterizer));
 201         p->rasterizer.cull_face = PIPE_FACE_NONE;
 202         p->rasterizer.half_pixel_center = 1;
 203         p->rasterizer.bottom_edge_rule = 1;
 204         p->rasterizer.depth_clip_near = 1;
 205         p->rasterizer.depth_clip_far = 1;
 206
 207         /* sampler */
 208         memset(&p->sampler, 0, sizeof(p->sampler));
 209         p->sampler.wrap_s = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 210         p->sampler.wrap_t = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 211         p->sampler.wrap_r = PIPE_TEX_WRAP_CLAMP_TO_EDGE;
 212         p->sampler.min_mip_filter = PIPE_TEX_MIPFILTER_NONE;
 213         p->sampler.min_img_filter = PIPE_TEX_MIPFILTER_LINEAR;
 214         p->sampler.mag_img_filter = PIPE_TEX_MIPFILTER_LINEAR;
 215         p->sampler.normalized_coords = 1;
 216
 217         surf_tmpl.format = PIPE_FORMAT_B8G8R8A8_UNORM; /* All drivers support this */
 218         surf_tmpl.u.tex.level = 0;
 219         surf_tmpl.u.tex.first_layer = 0;
 220         surf_tmpl.u.tex.last_layer = 0;
 221         /* drawing destination */
 222         memset(&p->framebuffer, 0, sizeof(p->framebuffer));
 223         p->framebuffer.width = WIDTH;
 224         p->framebuffer.height = HEIGHT;
 225         p->framebuffer.nr_cbufs = 1;
 226         p->framebuffer.cbufs[0] = p->pipe->create_surface(p->pipe, p->target, &surf_tmpl);
 227
 228         /* viewport, depth isn't really needed */
 229         {
 230                 float x = 0;
 231                 float y = 0;
 232                 float z = NEAR;
 233                 float half_width = (float)WIDTH / 2.0f;
 234                 float half_height = (float)HEIGHT / 2.0f;
 235                 float half_depth = ((float)FAR - (float)NEAR) / 2.0f;
 236                 float scale, bias;
 237
 238                 if (FLIP) {
 239                         scale = -1.0f;
 240                         bias = (float)HEIGHT;
 241                 } else {
 242                         scale = 1.0f;
 243                         bias = 0.0f;
 244                 }
 245
 246                 p->viewport.scale[0] = half_width;
 247                 p->viewport.scale[1] = half_height * scale;
 248                 p->viewport.scale[2] = half_depth;
 249
 250                 p->viewport.translate[0] = half_width + x;
 251                 p->viewport.translate[1] = (half_height + y) * scale + bias;
 252                 p->viewport.translate[2] = half_depth + z;
 253         }
 254
 255         /* vertex elements state */
 256         memset(&p->velem, 0, sizeof(p->velem));
 257         p->velem.count = 2;
 258
 259         p->velem.velems[0].src_offset = 0 * 4 * sizeof(float); /* offset 0, first element */
 260         p->velem.velems[0].instance_divisor = 0;
 261         p->velem.velems[0].vertex_buffer_index = 0;
 262         p->velem.velems[0].src_format = PIPE_FORMAT_R32G32B32A32_FLOAT;
 263
 264         p->velem.velems[1].src_offset = 1 * 4 * sizeof(float); /* offset 16, second element */
 265         p->velem.velems[1].instance_divisor = 0;
 266         p->velem.velems[1].vertex_buffer_index = 0;
 267         p->velem.velems[1].src_format = PIPE_FORMAT_R32G32B32A32_FLOAT;
 268
 269         /* vertex shader */
 270         {
 271                 const enum tgsi_semantic semantic_names[] =
 272                    { TGSI_SEMANTIC_POSITION, TGSI_SEMANTIC_GENERIC };
 273                 const uint semantic_indexes[] = { 0, 0 };
 274                 p->vs = util_make_vertex_passthrough_shader(p->pipe, 2, semantic_names, semantic_indexes, FALSE);
 275         }
 276
 277         /* fragment shader */
 278         p->fs = util_make_fragment_tex_shader(p->pipe, TGSI_TEXTURE_2D,
 279                                               TGSI_INTERPOLATE_LINEAR,
 280                                               TGSI_RETURN_TYPE_FLOAT,
 281                                               TGSI_RETURN_TYPE_FLOAT, false,
 282                                               false);
 283 }
 284
 285 static void close_prog(struct program *p)
 286 {
 287         cso_destroy_context(p->cso);
 288
 289         p->pipe->delete_vs_state(p->pipe, p->vs);
 290         p->pipe->delete_fs_state(p->pipe, p->fs);
 291
 292         pipe_surface_reference(&p->framebuffer.cbufs[0], NULL);
 293         pipe_sampler_view_reference(&p->view, NULL);
 294         pipe_resource_reference(&p->target, NULL);
 295         pipe_resource_reference(&p->tex, NULL);
 296         pipe_resource_reference(&p->vbuf, NULL);
 297
 298         p->pipe->destroy(p->pipe);
 299         p->screen->destroy(p->screen);
 300         pipe_loader_release(&p->dev, 1);
 301
 302         FREE(p);
 303 }
 304
 305 static void draw(struct program *p)
 306 {
 307         const struct pipe_sampler_state *samplers[] = {&p->sampler};
 308
 309         /* set the render target */
 310         cso_set_framebuffer(p->cso, &p->framebuffer);
 311
 312         /* clear the render target */
 313         p->pipe->clear(p->pipe, PIPE_CLEAR_COLOR, NULL, &p->clear_color, 0, 0);
 314
 315         /* set misc state we care about */
 316         cso_set_blend(p->cso, &p->blend);
 317         cso_set_depth_stencil_alpha(p->cso, &p->depthstencil);
 318         cso_set_rasterizer(p->cso, &p->rasterizer);
 319         cso_set_viewport(p->cso, &p->viewport);
 320
 321         /* sampler */
 322         cso_set_samplers(p->cso, PIPE_SHADER_FRAGMENT, 1, samplers);
 323
 324         /* texture sampler view */
 325         cso_set_sampler_views(p->cso, PIPE_SHADER_FRAGMENT, 1, &p->view);
 326
 327         /* shaders */
 328         cso_set_fragment_shader_handle(p->cso, p->fs);
 329         cso_set_vertex_shader_handle(p->cso, p->vs);
 330
 331         /* vertex element data */
 332         cso_set_vertex_elements(p->cso, &p->velem);
 333
 334         util_draw_vertex_buffer(p->pipe, p->cso,
 335                                 p->vbuf, 0, 0,
 336                                 PIPE_PRIM_QUADS,
 337                                 4,  /* verts */
 338                                 2); /* attribs/vert */
 339
 340         p->pipe->flush(p->pipe, NULL, 0);
 341
 342         debug_dump_surface_bmp(p->pipe, "result.bmp", p->framebuffer.cbufs[0]);
 343 }
 344
 345 int main(int argc, char** argv)
 346 {
 347         struct program *p = CALLOC_STRUCT(program);
 348
 349         init_prog(p);
 350         draw(p);
 351         close_prog(p);
 352
 353         return 0;
 354 }