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];
109 memset(ve
, 0, sizeof ve
);
112 ve
[0].src_offset
= Offset(struct vertex
, position
);
113 ve
[0].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
114 ve
[0].vertex_buffer_index
= 0;
117 ve
[1].src_offset
= Offset(struct vertex
, color
);
118 ve
[1].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
119 ve
[1].vertex_buffer_index
= 0;
121 /* per-instance info */
122 ve
[2].src_offset
= 0;
123 ve
[2].src_format
= PIPE_FORMAT_R32G32B32A32_FLOAT
;
124 ve
[2].vertex_buffer_index
= 1;
125 ve
[2].instance_divisor
= 1;
127 handle
= ctx
->create_vertex_elements_state(ctx
, 3, ve
);
128 ctx
->bind_vertex_elements_state(ctx
, handle
);
130 memset(&vbuf
, 0, sizeof vbuf
);
133 vbuf
[0].stride
= sizeof( struct vertex
);
134 vbuf
[0].buffer_offset
= 0;
135 vbuf
[0].buffer
.resource
= pipe_buffer_create_with_data(ctx
,
136 PIPE_BIND_VERTEX_BUFFER
,
142 vbuf
[1].stride
= sizeof( inst_data
[0] );
143 vbuf
[1].buffer_offset
= 0;
144 vbuf
[1].buffer
.resource
= pipe_buffer_create_with_data(ctx
,
145 PIPE_BIND_VERTEX_BUFFER
,
150 ctx
->set_vertex_buffers(ctx
, 0, 2, vbuf
);
153 static void set_vertex_shader( void )
161 "DCL OUT[0], POSITION\n"
162 "DCL OUT[1], COLOR\n"
163 " 0: MOV OUT[1], IN[1]\n"
164 " 1: ADD OUT[0], IN[0], IN[2]\n" /* add instance pos to vertex pos */
167 handle
= graw_parse_vertex_shader(ctx
, text
);
168 ctx
->bind_vs_state(ctx
, handle
);
171 static void set_fragment_shader( void )
176 "DCL IN[0], COLOR, LINEAR\n"
177 "DCL OUT[0], COLOR\n"
178 " 0: MOV OUT[0], IN[0]\n"
181 handle
= graw_parse_fragment_shader(ctx
, text
);
182 ctx
->bind_fs_state(ctx
, handle
);
186 static void draw( void )
188 union pipe_color_union clear_color
= { {1,0,1,1} };
189 struct pipe_draw_info info
;
191 ctx
->clear(ctx
, PIPE_CLEAR_COLOR
, &clear_color
, 0, 0);
194 util_draw_init_info(&info
);
195 info
.index_size
= draw_elements
? 2 : 0;
196 info
.mode
= PIPE_PRIM_TRIANGLES
;
199 /* draw NUM_INST triangles */
200 info
.instance_count
= NUM_INST
;
203 if (info
.index_size
) {
204 info
.index
.resource
=
205 pipe_buffer_create_with_data(ctx
,
206 PIPE_BIND_INDEX_BUFFER
,
212 ctx
->draw_vbo(ctx
, &info
);
214 pipe_resource_reference(&info
.index
.resource
, NULL
);
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 memset(&templat
, 0, sizeof(templat
));
254 templat
.target
= PIPE_TEXTURE_2D
;
255 templat
.format
= formats
[i
];
256 templat
.width0
= WIDTH
;
257 templat
.height0
= HEIGHT
;
259 templat
.array_size
= 1;
260 templat
.last_level
= 0;
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_near
= 1;
310 rasterizer
.depth_clip_far
= 1;
311 handle
= ctx
->create_rasterizer_state(ctx
, &rasterizer
);
312 ctx
->bind_rasterizer_state(ctx
, handle
);
315 set_viewport(0, 0, WIDTH
, HEIGHT
, 30, 1000);
318 set_fragment_shader();
322 static void options(int argc
, char *argv
[])
326 for (i
= 1; i
< argc
;) {
327 if (graw_parse_args(&i
, argc
, argv
)) {
330 if (strcmp(argv
[i
], "-e") == 0) {
339 printf("Using pipe_context::draw_elements_instanced()\n");
341 printf("Using pipe_context::draw_arrays_instanced()\n");
345 int main( int argc
, char *argv
[] )
351 graw_set_display_func( draw
);