1 /* Display a cleared blue window. This demo has no dependencies on
2 * any utility code, just the graw interface and gallium.
5 #include "state_tracker/graw.h"
6 #include "pipe/p_screen.h"
7 #include "pipe/p_context.h"
8 #include "pipe/p_shader_tokens.h"
9 #include "pipe/p_state.h"
10 #include "pipe/p_defines.h"
11 #include <stdio.h> /* for fread(), etc */
13 #include "util/u_inlines.h"
14 #include "util/u_memory.h" /* Offset() */
15 #include "util/u_draw_quad.h"
16 #include "util/u_box.h"
18 static const char *filename
= NULL
;
19 unsigned show_fps
= 0;
20 unsigned draw_strip
= 0;
23 static void usage(char *name
)
25 fprintf(stderr
, "usage: %s [ options ] shader_filename\n", name
);
27 fprintf(stderr
, "\n" );
28 fprintf(stderr
, "options:\n");
29 fprintf(stderr
, " -fps show frames per second\n");
30 fprintf(stderr
, " -strip renders a triangle strip\n");
35 enum pipe_format formats
[] = {
36 PIPE_FORMAT_R8G8B8A8_UNORM
,
37 PIPE_FORMAT_B8G8R8A8_UNORM
,
41 static const int WIDTH
= 250;
42 static const int HEIGHT
= 250;
44 static struct pipe_screen
*screen
= NULL
;
45 static struct pipe_context
*ctx
= NULL
;
46 static struct pipe_resource
*rttex
= NULL
;
47 static struct pipe_resource
*constbuf1
= NULL
;
48 static struct pipe_resource
*constbuf2
= NULL
;
49 static struct pipe_surface
*surf
= NULL
;
50 static struct pipe_sampler_view
*sv
= NULL
;
51 static void *sampler
= NULL
;
52 static void *window
= NULL
;
53 static struct pipe_resource
*samptex
= NULL
;
62 /* Vertex data matches progs/fp/fp-tri.c, but flipped in Y dimension
63 * so that the final images are the same.
65 static struct vertex vertices
[] =
67 { { 0.9, 0.9, 0.0, 1.0 },
73 { { 0.9, -0.9, 0.0, 1.0 },
79 { {-0.9, 0.0, 0.0, 1.0 },
86 static struct vertex vertices_strip
[] =
88 { { 0.9, 0.9, 0.0, 1.0 },
94 { { 0.9, -0.9, 0.0, 1.0 },
100 { {-0.9, 0.9, 0.0, 1.0 },
106 { {-0.9, -0.9, 0.0, 1.0 },
113 static float constants1
[] =
131 static float constants2
[] =
149 static void init_fs_constbuf( void )
151 struct pipe_resource templat
;
154 templat
.target
= PIPE_BUFFER
;
155 templat
.format
= PIPE_FORMAT_R8_UNORM
;
156 templat
.width0
= sizeof(constants1
);
159 templat
.array_size
= 1;
160 templat
.last_level
= 0;
161 templat
.nr_samples
= 1;
162 templat
.bind
= PIPE_BIND_CONSTANT_BUFFER
;
164 constbuf1
= screen
->resource_create(screen
, &templat
);
165 if (constbuf1
== NULL
)
167 constbuf2
= screen
->resource_create(screen
, &templat
);
168 if (constbuf2
== NULL
)
172 u_box_2d(0,0,sizeof(constants1
),1, &box
);
174 ctx
->transfer_inline_write(ctx
,
184 pipe_set_constant_buffer(ctx
,
185 PIPE_SHADER_GEOMETRY
, 0,
189 u_box_2d(0,0,sizeof(constants2
),1, &box
);
191 ctx
->transfer_inline_write(ctx
,
201 pipe_set_constant_buffer(ctx
,
202 PIPE_SHADER_GEOMETRY
, 1,
208 static void set_viewport( float x
, float y
,
209 float width
, float height
,
210 float near
, float far
)
213 float half_width
= (float)width
/ 2.0f
;
214 float half_height
= (float)height
/ 2.0f
;
215 float half_depth
= ((float)far
- (float)near
) / 2.0f
;
216 struct pipe_viewport_state vp
;
218 vp
.scale
[0] = half_width
;
219 vp
.scale
[1] = half_height
;
220 vp
.scale
[2] = half_depth
;
223 vp
.translate
[0] = half_width
+ x
;
224 vp
.translate
[1] = half_height
+ y
;
225 vp
.translate
[2] = half_depth
+ z
;
226 vp
.translate
[3] = 0.0f
;
228 ctx
->set_viewport_state( ctx
, &vp
);
231 static void set_vertices( void )
233 struct pipe_vertex_element ve
[4];
234 struct pipe_vertex_buffer vbuf
;
237 memset(ve
, 0, sizeof ve
);
239 ve
[0].src_offset
= Offset(struct vertex
, position
);
240 ve
[0].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
241 ve
[1].src_offset
= Offset(struct vertex
, color
);
242 ve
[1].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
243 ve
[2].src_offset
= Offset(struct vertex
, texcoord
);
244 ve
[2].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
245 ve
[3].src_offset
= Offset(struct vertex
, generic
);
246 ve
[3].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
248 handle
= ctx
->create_vertex_elements_state(ctx
, 4, ve
);
249 ctx
->bind_vertex_elements_state(ctx
, handle
);
251 vbuf
.stride
= sizeof( struct vertex
);
252 vbuf
.buffer_offset
= 0;
254 vbuf
.buffer
= screen
->user_buffer_create(screen
,
256 sizeof(vertices_strip
),
257 PIPE_BIND_VERTEX_BUFFER
);
259 vbuf
.buffer
= screen
->user_buffer_create(screen
,
262 PIPE_BIND_VERTEX_BUFFER
);
265 ctx
->set_vertex_buffers(ctx
, 1, &vbuf
);
268 static void set_vertex_shader( void )
277 "DCL OUT[0], POSITION\n"
278 "DCL OUT[1], COLOR[0]\n"
279 "DCL OUT[2], GENERIC[0]\n"
280 "DCL OUT[3], GENERIC[1]\n"
281 " MOV OUT[0], IN[0]\n"
282 " MOV OUT[1], IN[1]\n"
283 " MOV OUT[2], IN[2]\n"
284 " MOV OUT[3], IN[3]\n"
287 handle
= graw_parse_vertex_shader(ctx
, text
);
288 ctx
->bind_vs_state(ctx
, handle
);
291 static void set_fragment_shader( void )
296 "DCL IN[0], COLOR, LINEAR\n"
297 "DCL OUT[0], COLOR\n"
298 " 0: MOV OUT[0], IN[0]\n"
301 handle
= graw_parse_fragment_shader(ctx
, text
);
302 ctx
->bind_fs_state(ctx
, handle
);
306 static void set_geometry_shader( void )
313 if ((f
= fopen(filename
, "r")) == NULL
) {
314 fprintf(stderr
, "Couldn't open %s\n", filename
);
318 sz
= fread(buf
, 1, sizeof(buf
), f
);
320 printf("file too long\n");
323 printf("%.*s\n", sz
, buf
);
326 handle
= graw_parse_geometry_shader(ctx
, buf
);
327 ctx
->bind_gs_state(ctx
, handle
);
332 static void draw( void )
334 union pipe_color_union clear_color
= { {.1,.3,.5,0} };
336 ctx
->clear(ctx
, PIPE_CLEAR_COLOR
, &clear_color
, 0, 0);
338 util_draw_arrays(ctx
, PIPE_PRIM_TRIANGLE_STRIP
, 0, 4);
340 util_draw_arrays(ctx
, PIPE_PRIM_TRIANGLES
, 0, 3);
342 ctx
->flush(ctx
, NULL
);
344 graw_save_surface_to_file(ctx
, surf
, NULL
);
346 screen
->flush_frontbuffer(screen
, rttex
, 0, 0, window
);
351 static void init_tex( void )
353 struct pipe_sampler_view sv_template
;
354 struct pipe_sampler_state sampler_desc
;
355 struct pipe_resource templat
;
357 ubyte tex2d
[SIZE
][SIZE
][4];
361 for (s
= 0; s
< SIZE
; s
++) {
362 for (t
= 0; t
< SIZE
; t
++) {
365 tex2d
[t
][s
][0] = (x
) ? 0 : 63;
366 tex2d
[t
][s
][1] = (x
) ? 0 : 128;
368 tex2d
[t
][s
][3] = 0xff;
371 int x
= ((s
^ t
) >> 2) & 1;
372 tex2d
[t
][s
][0] = s
*255/(SIZE
-1);
373 tex2d
[t
][s
][1] = t
*255/(SIZE
-1);
374 tex2d
[t
][s
][2] = (x
) ? 0 : 128;
375 tex2d
[t
][s
][3] = 0xff;
381 tex2d
[0][0][1] = 255;
382 tex2d
[0][0][2] = 255;
387 tex2d
[0][1][2] = 255;
388 tex2d
[0][1][3] = 255;
390 tex2d
[1][0][0] = 255;
391 tex2d
[1][0][1] = 255;
393 tex2d
[1][0][3] = 255;
395 tex2d
[1][1][0] = 255;
398 tex2d
[1][1][3] = 255;
401 templat
.target
= PIPE_TEXTURE_2D
;
402 templat
.format
= PIPE_FORMAT_B8G8R8A8_UNORM
;
403 templat
.width0
= SIZE
;
404 templat
.height0
= SIZE
;
406 templat
.array_size
= 1;
407 templat
.last_level
= 0;
408 templat
.nr_samples
= 1;
409 templat
.bind
= PIPE_BIND_SAMPLER_VIEW
;
412 samptex
= screen
->resource_create(screen
,
417 u_box_2d(0,0,SIZE
,SIZE
, &box
);
419 ctx
->transfer_inline_write(ctx
,
428 /* Possibly read back & compare against original data:
432 struct pipe_transfer
*t
;
434 t
= pipe_get_transfer(ctx
, samptex
,
435 0, 0, /* level, layer */
437 0, 0, SIZE
, SIZE
); /* x, y, width, height */
439 ptr
= ctx
->transfer_map(ctx
, t
);
441 if (memcmp(ptr
, tex2d
, sizeof tex2d
) != 0) {
446 ctx
->transfer_unmap(ctx
, t
);
448 ctx
->transfer_destroy(ctx
, t
);
451 memset(&sv_template
, 0, sizeof sv_template
);
452 sv_template
.format
= samptex
->format
;
453 sv_template
.texture
= samptex
;
454 sv_template
.swizzle_r
= 0;
455 sv_template
.swizzle_g
= 1;
456 sv_template
.swizzle_b
= 2;
457 sv_template
.swizzle_a
= 3;
458 sv
= ctx
->create_sampler_view(ctx
, samptex
, &sv_template
);
462 ctx
->set_fragment_sampler_views(ctx
, 1, &sv
);
465 memset(&sampler_desc
, 0, sizeof sampler_desc
);
466 sampler_desc
.wrap_s
= PIPE_TEX_WRAP_REPEAT
;
467 sampler_desc
.wrap_t
= PIPE_TEX_WRAP_REPEAT
;
468 sampler_desc
.wrap_r
= PIPE_TEX_WRAP_REPEAT
;
469 sampler_desc
.min_img_filter
= PIPE_TEX_FILTER_NEAREST
;
470 sampler_desc
.min_mip_filter
= PIPE_TEX_MIPFILTER_NONE
;
471 sampler_desc
.mag_img_filter
= PIPE_TEX_FILTER_NEAREST
;
472 sampler_desc
.compare_mode
= PIPE_TEX_COMPARE_NONE
;
473 sampler_desc
.compare_func
= 0;
474 sampler_desc
.normalized_coords
= 1;
475 sampler_desc
.max_anisotropy
= 0;
477 sampler
= ctx
->create_sampler_state(ctx
, &sampler_desc
);
481 ctx
->bind_fragment_sampler_states(ctx
, 1, &sampler
);
485 static void init( void )
487 struct pipe_framebuffer_state fb
;
488 struct pipe_resource templat
;
489 struct pipe_surface surf_tmpl
;
492 /* It's hard to say whether window or screen should be created
493 * first. Different environments would prefer one or the other.
495 * Also, no easy way of querying supported formats if the screen
496 * cannot be created first.
498 for (i
= 0; formats
[i
] != PIPE_FORMAT_NONE
; i
++) {
499 screen
= graw_create_window_and_screen(0, 0, 300, 300,
502 if (window
&& screen
)
505 if (!screen
|| !window
) {
506 fprintf(stderr
, "Unable to create window\n");
510 ctx
= screen
->context_create(screen
, NULL
);
514 templat
.target
= PIPE_TEXTURE_2D
;
515 templat
.format
= formats
[i
];
516 templat
.width0
= WIDTH
;
517 templat
.height0
= HEIGHT
;
519 templat
.array_size
= 1;
520 templat
.last_level
= 0;
521 templat
.nr_samples
= 1;
522 templat
.bind
= (PIPE_BIND_RENDER_TARGET
|
523 PIPE_BIND_DISPLAY_TARGET
);
525 rttex
= screen
->resource_create(screen
,
530 surf_tmpl
.format
= templat
.format
;
531 surf_tmpl
.usage
= PIPE_BIND_RENDER_TARGET
;
532 surf_tmpl
.u
.tex
.level
= 0;
533 surf_tmpl
.u
.tex
.first_layer
= 0;
534 surf_tmpl
.u
.tex
.last_layer
= 0;
535 surf
= ctx
->create_surface(ctx
, rttex
, &surf_tmpl
);
539 memset(&fb
, 0, sizeof fb
);
545 ctx
->set_framebuffer_state(ctx
, &fb
);
548 struct pipe_blend_state blend
;
550 memset(&blend
, 0, sizeof blend
);
551 blend
.rt
[0].colormask
= PIPE_MASK_RGBA
;
552 handle
= ctx
->create_blend_state(ctx
, &blend
);
553 ctx
->bind_blend_state(ctx
, handle
);
557 struct pipe_depth_stencil_alpha_state depthstencil
;
559 memset(&depthstencil
, 0, sizeof depthstencil
);
560 handle
= ctx
->create_depth_stencil_alpha_state(ctx
, &depthstencil
);
561 ctx
->bind_depth_stencil_alpha_state(ctx
, handle
);
565 struct pipe_rasterizer_state rasterizer
;
567 memset(&rasterizer
, 0, sizeof rasterizer
);
568 rasterizer
.cull_face
= PIPE_FACE_NONE
;
569 rasterizer
.gl_rasterization_rules
= 1;
570 rasterizer
.depth_clip
= 1;
571 handle
= ctx
->create_rasterizer_state(ctx
, &rasterizer
);
572 ctx
->bind_rasterizer_state(ctx
, handle
);
575 set_viewport(0, 0, WIDTH
, HEIGHT
, 30, 1000);
582 set_fragment_shader();
583 set_geometry_shader();
586 static void args(int argc
, char *argv
[])
590 for (i
= 1; i
< argc
;) {
591 if (graw_parse_args(&i
, argc
, argv
)) {
594 if (strcmp(argv
[i
], "-fps") == 0) {
598 else if (strcmp(argv
[i
], "-strip") == 0) {
602 else if (i
== argc
- 1) {
618 int main( int argc
, char *argv
[] )
623 graw_set_display_func( draw
);