2 * Test draw instancing.
8 #include "state_tracker/graw.h"
9 #include "pipe/p_screen.h"
10 #include "pipe/p_context.h"
11 #include "pipe/p_state.h"
12 #include "pipe/p_defines.h"
14 #include "util/u_memory.h" /* Offset() */
15 #include "util/u_draw_quad.h"
16 #include "util/u_inlines.h"
19 enum pipe_format formats
[] = {
20 PIPE_FORMAT_R8G8B8A8_UNORM
,
21 PIPE_FORMAT_B8G8R8A8_UNORM
,
25 static const int WIDTH
= 300;
26 static const int HEIGHT
= 300;
28 static struct pipe_screen
*screen
= NULL
;
29 static struct pipe_context
*ctx
= NULL
;
30 static struct pipe_surface
*surf
= NULL
;
31 static struct pipe_resource
*tex
= NULL
;
32 static void *window
= NULL
;
40 static int draw_elements
= 0;
45 * Each vertex has three attributes: position, color and translation.
46 * The translation attribute is a per-instance attribute. See
47 * "instance_divisor" below.
49 static struct vertex vertices
[4] =
52 { 0.0f
, -0.3f
, 0.0f
, 1.0f
}, /* pos */
53 { 1.0f
, 0.0f
, 0.0f
, 1.0f
} /* color */
56 { -0.2f
, 0.3f
, 0.0f
, 1.0f
},
57 { 0.0f
, 1.0f
, 0.0f
, 1.0f
}
60 { 0.2f
, 0.3f
, 0.0f
, 1.0f
},
61 { 0.0f
, 0.0f
, 1.0f
, 1.0f
}
68 static float inst_data
[NUM_INST
][4] =
70 { -0.50f
, 0.4f
, 0.0f
, 0.0f
},
71 { -0.25f
, 0.1f
, 0.0f
, 0.0f
},
72 { 0.00f
, 0.2f
, 0.0f
, 0.0f
},
73 { 0.25f
, 0.1f
, 0.0f
, 0.0f
},
74 { 0.50f
, 0.3f
, 0.0f
, 0.0f
}
78 static ushort indices
[3] = { 0, 2, 1 };
81 static void set_viewport( float x
, float y
,
82 float width
, float height
,
83 float near
, float far
)
86 float half_width
= (float)width
/ 2.0f
;
87 float half_height
= (float)height
/ 2.0f
;
88 float half_depth
= ((float)far
- (float)near
) / 2.0f
;
89 struct pipe_viewport_state vp
;
91 vp
.scale
[0] = half_width
;
92 vp
.scale
[1] = half_height
;
93 vp
.scale
[2] = half_depth
;
96 vp
.translate
[0] = half_width
+ x
;
97 vp
.translate
[1] = half_height
+ y
;
98 vp
.translate
[2] = half_depth
+ z
;
99 vp
.translate
[3] = 0.0f
;
101 ctx
->set_viewport_state( ctx
, &vp
);
105 static void set_vertices( void )
107 struct pipe_vertex_element ve
[3];
108 struct pipe_vertex_buffer vbuf
[2];
109 struct pipe_index_buffer ibuf
;
112 memset(ve
, 0, sizeof ve
);
115 ve
[0].src_offset
= Offset(struct vertex
, position
);
116 ve
[0].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
117 ve
[0].vertex_buffer_index
= 0;
120 ve
[1].src_offset
= Offset(struct vertex
, color
);
121 ve
[1].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
122 ve
[1].vertex_buffer_index
= 0;
124 /* per-instance info */
125 ve
[2].src_offset
= 0;
126 ve
[2].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
127 ve
[2].vertex_buffer_index
= 1;
128 ve
[2].instance_divisor
= 1;
130 handle
= ctx
->create_vertex_elements_state(ctx
, 3, ve
);
131 ctx
->bind_vertex_elements_state(ctx
, handle
);
133 memset(&vbuf
, 0, sizeof vbuf
);
136 vbuf
[0].stride
= sizeof( struct vertex
);
137 vbuf
[0].buffer_offset
= 0;
138 vbuf
[0].buffer
= pipe_buffer_create_with_data(ctx
,
139 PIPE_BIND_VERTEX_BUFFER
,
145 vbuf
[1].stride
= sizeof( inst_data
[0] );
146 vbuf
[1].buffer_offset
= 0;
147 vbuf
[1].buffer
= pipe_buffer_create_with_data(ctx
,
148 PIPE_BIND_VERTEX_BUFFER
,
153 ctx
->set_vertex_buffers(ctx
, 0, 2, vbuf
);
156 ibuf
.buffer
= pipe_buffer_create_with_data(ctx
,
157 PIPE_BIND_INDEX_BUFFER
,
164 ctx
->set_index_buffer(ctx
, &ibuf
);
168 static void set_vertex_shader( void )
176 "DCL OUT[0], POSITION\n"
177 "DCL OUT[1], COLOR\n"
178 " 0: MOV OUT[1], IN[1]\n"
179 " 1: ADD OUT[0], IN[0], IN[2]\n" /* add instance pos to vertex pos */
182 handle
= graw_parse_vertex_shader(ctx
, text
);
183 ctx
->bind_vs_state(ctx
, handle
);
186 static void set_fragment_shader( void )
191 "DCL IN[0], COLOR, LINEAR\n"
192 "DCL OUT[0], COLOR\n"
193 " 0: MOV OUT[0], IN[0]\n"
196 handle
= graw_parse_fragment_shader(ctx
, text
);
197 ctx
->bind_fs_state(ctx
, handle
);
201 static void draw( void )
203 union pipe_color_union clear_color
= { {1,0,1,1} };
204 struct pipe_draw_info info
;
206 ctx
->clear(ctx
, PIPE_CLEAR_COLOR
, &clear_color
, 0, 0);
208 util_draw_init_info(&info
);
209 info
.indexed
= (draw_elements
!= 0);
210 info
.mode
= PIPE_PRIM_TRIANGLES
;
213 /* draw NUM_INST triangles */
214 info
.instance_count
= NUM_INST
;
216 ctx
->draw_vbo(ctx
, &info
);
218 ctx
->flush(ctx
, NULL
, 0);
220 graw_save_surface_to_file(ctx
, surf
, NULL
);
222 screen
->flush_frontbuffer(screen
, tex
, 0, 0, window
);
226 static void init( void )
228 struct pipe_framebuffer_state fb
;
229 struct pipe_resource templat
;
230 struct pipe_surface surf_tmpl
;
233 /* It's hard to say whether window or screen should be created
234 * first. Different environments would prefer one or the other.
236 * Also, no easy way of querying supported formats if the screen
237 * cannot be created first.
239 for (i
= 0; formats
[i
] != PIPE_FORMAT_NONE
; i
++) {
240 screen
= graw_create_window_and_screen(0, 0, 300, 300,
243 if (window
&& screen
)
246 if (!screen
|| !window
) {
247 fprintf(stderr
, "Unable to create window\n");
251 ctx
= screen
->context_create(screen
, NULL
);
255 templat
.target
= PIPE_TEXTURE_2D
;
256 templat
.format
= formats
[i
];
257 templat
.width0
= WIDTH
;
258 templat
.height0
= HEIGHT
;
260 templat
.array_size
= 1;
261 templat
.last_level
= 0;
262 templat
.nr_samples
= 1;
263 templat
.bind
= (PIPE_BIND_RENDER_TARGET
|
264 PIPE_BIND_DISPLAY_TARGET
);
266 tex
= screen
->resource_create(screen
,
271 surf_tmpl
.format
= templat
.format
;
272 surf_tmpl
.u
.tex
.level
= 0;
273 surf_tmpl
.u
.tex
.first_layer
= 0;
274 surf_tmpl
.u
.tex
.last_layer
= 0;
275 surf
= ctx
->create_surface(ctx
, tex
, &surf_tmpl
);
279 memset(&fb
, 0, sizeof fb
);
285 ctx
->set_framebuffer_state(ctx
, &fb
);
288 struct pipe_blend_state blend
;
290 memset(&blend
, 0, sizeof blend
);
291 blend
.rt
[0].colormask
= PIPE_MASK_RGBA
;
292 handle
= ctx
->create_blend_state(ctx
, &blend
);
293 ctx
->bind_blend_state(ctx
, handle
);
297 struct pipe_depth_stencil_alpha_state depthstencil
;
299 memset(&depthstencil
, 0, sizeof depthstencil
);
300 handle
= ctx
->create_depth_stencil_alpha_state(ctx
, &depthstencil
);
301 ctx
->bind_depth_stencil_alpha_state(ctx
, handle
);
305 struct pipe_rasterizer_state rasterizer
;
307 memset(&rasterizer
, 0, sizeof rasterizer
);
308 rasterizer
.cull_face
= PIPE_FACE_NONE
;
309 rasterizer
.half_pixel_center
= 1;
310 rasterizer
.bottom_edge_rule
= 1;
311 rasterizer
.depth_clip
= 1;
312 handle
= ctx
->create_rasterizer_state(ctx
, &rasterizer
);
313 ctx
->bind_rasterizer_state(ctx
, handle
);
316 set_viewport(0, 0, WIDTH
, HEIGHT
, 30, 1000);
319 set_fragment_shader();
323 static void options(int argc
, char *argv
[])
327 for (i
= 1; i
< argc
;) {
328 if (graw_parse_args(&i
, argc
, argv
)) {
331 if (strcmp(argv
[i
], "-e") == 0) {
340 printf("Using pipe_context::draw_elements_instanced()\n");
342 printf("Using pipe_context::draw_arrays_instanced()\n");
346 int main( int argc
, char *argv
[] )
352 graw_set_display_func( draw
);