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_RGBA8888_UNORM
,
21 PIPE_FORMAT_BGRA8888_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 zNear
, float zFar
)
86 float half_width
= (float)width
/ 2.0f
;
87 float half_height
= (float)height
/ 2.0f
;
88 float half_depth
= ((float)zFar
- (float)zNear
) / 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
;
95 vp
.translate
[0] = half_width
+ x
;
96 vp
.translate
[1] = half_height
+ y
;
97 vp
.translate
[2] = half_depth
+ z
;
99 ctx
->set_viewport_states( ctx
, 0, 1, &vp
);
103 static void set_vertices( void )
105 struct pipe_vertex_element ve
[3];
106 struct pipe_vertex_buffer vbuf
[2];
107 struct pipe_index_buffer ibuf
;
110 memset(ve
, 0, sizeof ve
);
113 ve
[0].src_offset
= Offset(struct vertex
, position
);
114 ve
[0].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
115 ve
[0].vertex_buffer_index
= 0;
118 ve
[1].src_offset
= Offset(struct vertex
, color
);
119 ve
[1].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
120 ve
[1].vertex_buffer_index
= 0;
122 /* per-instance info */
123 ve
[2].src_offset
= 0;
124 ve
[2].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
125 ve
[2].vertex_buffer_index
= 1;
126 ve
[2].instance_divisor
= 1;
128 handle
= ctx
->create_vertex_elements_state(ctx
, 3, ve
);
129 ctx
->bind_vertex_elements_state(ctx
, handle
);
131 memset(&vbuf
, 0, sizeof vbuf
);
134 vbuf
[0].stride
= sizeof( struct vertex
);
135 vbuf
[0].buffer_offset
= 0;
136 vbuf
[0].buffer
= pipe_buffer_create_with_data(ctx
,
137 PIPE_BIND_VERTEX_BUFFER
,
143 vbuf
[1].stride
= sizeof( inst_data
[0] );
144 vbuf
[1].buffer_offset
= 0;
145 vbuf
[1].buffer
= pipe_buffer_create_with_data(ctx
,
146 PIPE_BIND_VERTEX_BUFFER
,
151 ctx
->set_vertex_buffers(ctx
, 0, 2, vbuf
);
154 ibuf
.buffer
= pipe_buffer_create_with_data(ctx
,
155 PIPE_BIND_INDEX_BUFFER
,
162 ctx
->set_index_buffer(ctx
, &ibuf
);
166 static void set_vertex_shader( void )
174 "DCL OUT[0], POSITION\n"
175 "DCL OUT[1], COLOR\n"
176 " 0: MOV OUT[1], IN[1]\n"
177 " 1: ADD OUT[0], IN[0], IN[2]\n" /* add instance pos to vertex pos */
180 handle
= graw_parse_vertex_shader(ctx
, text
);
181 ctx
->bind_vs_state(ctx
, handle
);
184 static void set_fragment_shader( void )
189 "DCL IN[0], COLOR, LINEAR\n"
190 "DCL OUT[0], COLOR\n"
191 " 0: MOV OUT[0], IN[0]\n"
194 handle
= graw_parse_fragment_shader(ctx
, text
);
195 ctx
->bind_fs_state(ctx
, handle
);
199 static void draw( void )
201 union pipe_color_union clear_color
= { {1,0,1,1} };
202 struct pipe_draw_info info
;
204 ctx
->clear(ctx
, PIPE_CLEAR_COLOR
, &clear_color
, 0, 0);
206 util_draw_init_info(&info
);
207 info
.indexed
= (draw_elements
!= 0);
208 info
.mode
= PIPE_PRIM_TRIANGLES
;
211 /* draw NUM_INST triangles */
212 info
.instance_count
= NUM_INST
;
214 ctx
->draw_vbo(ctx
, &info
);
216 ctx
->flush(ctx
, NULL
, 0);
218 graw_save_surface_to_file(ctx
, surf
, NULL
);
220 screen
->flush_frontbuffer(screen
, tex
, 0, 0, window
, NULL
);
224 static void init( void )
226 struct pipe_framebuffer_state fb
;
227 struct pipe_resource templat
;
228 struct pipe_surface surf_tmpl
;
231 /* It's hard to say whether window or screen should be created
232 * first. Different environments would prefer one or the other.
234 * Also, no easy way of querying supported formats if the screen
235 * cannot be created first.
237 for (i
= 0; formats
[i
] != PIPE_FORMAT_NONE
; i
++) {
238 screen
= graw_create_window_and_screen(0, 0, 300, 300,
241 if (window
&& screen
)
244 if (!screen
|| !window
) {
245 fprintf(stderr
, "Unable to create window\n");
249 ctx
= screen
->context_create(screen
, NULL
, 0);
253 templat
.target
= PIPE_TEXTURE_2D
;
254 templat
.format
= formats
[i
];
255 templat
.width0
= WIDTH
;
256 templat
.height0
= HEIGHT
;
258 templat
.array_size
= 1;
259 templat
.last_level
= 0;
260 templat
.nr_samples
= 1;
261 templat
.bind
= (PIPE_BIND_RENDER_TARGET
|
262 PIPE_BIND_DISPLAY_TARGET
);
264 tex
= screen
->resource_create(screen
,
269 surf_tmpl
.format
= templat
.format
;
270 surf_tmpl
.u
.tex
.level
= 0;
271 surf_tmpl
.u
.tex
.first_layer
= 0;
272 surf_tmpl
.u
.tex
.last_layer
= 0;
273 surf
= ctx
->create_surface(ctx
, tex
, &surf_tmpl
);
277 memset(&fb
, 0, sizeof fb
);
283 ctx
->set_framebuffer_state(ctx
, &fb
);
286 struct pipe_blend_state blend
;
288 memset(&blend
, 0, sizeof blend
);
289 blend
.rt
[0].colormask
= PIPE_MASK_RGBA
;
290 handle
= ctx
->create_blend_state(ctx
, &blend
);
291 ctx
->bind_blend_state(ctx
, handle
);
295 struct pipe_depth_stencil_alpha_state depthstencil
;
297 memset(&depthstencil
, 0, sizeof depthstencil
);
298 handle
= ctx
->create_depth_stencil_alpha_state(ctx
, &depthstencil
);
299 ctx
->bind_depth_stencil_alpha_state(ctx
, handle
);
303 struct pipe_rasterizer_state rasterizer
;
305 memset(&rasterizer
, 0, sizeof rasterizer
);
306 rasterizer
.cull_face
= PIPE_FACE_NONE
;
307 rasterizer
.half_pixel_center
= 1;
308 rasterizer
.bottom_edge_rule
= 1;
309 rasterizer
.depth_clip
= 1;
310 handle
= ctx
->create_rasterizer_state(ctx
, &rasterizer
);
311 ctx
->bind_rasterizer_state(ctx
, handle
);
314 set_viewport(0, 0, WIDTH
, HEIGHT
, 30, 1000);
317 set_fragment_shader();
321 static void options(int argc
, char *argv
[])
325 for (i
= 1; i
< argc
;) {
326 if (graw_parse_args(&i
, argc
, argv
)) {
329 if (strcmp(argv
[i
], "-e") == 0) {
338 printf("Using pipe_context::draw_elements_instanced()\n");
340 printf("Using pipe_context::draw_arrays_instanced()\n");
344 int main( int argc
, char *argv
[] )
350 graw_set_display_func( draw
);