1/* Test texture swizzles */
2
3#include <stdio.h>
4
5#include "graw_util.h"
6
7
8static struct graw_info info;
9
10static struct pipe_resource *texture = NULL;
11static struct pipe_sampler_view *sv = NULL;
12static void *sampler = NULL;
13
14static const int WIDTH = 300;
15static const int HEIGHT = 300;
16
17struct vertex {
18   float position[4];
19   float color[4];
20};
21
22static struct vertex vertices[] =
23{
24   { { 0.9, -0.9, 0.0, 1.0 },
25     { 1, 0, 0, 1 } },
26
27   { { 0.9,  0.9, 0.0, 1.0 },
28     { 1, 1, 0, 1 } },
29
30   { {-0.9,  0.9, 0.0, 1.0 },
31     { 0, 1, 0, 1 } },
32
33   { {-0.9,  -0.9, 0.0, 1.0 },
34     { 0, 0, 0, 1 } },
35};
36
37
38static void set_vertices(void)
39{
40   struct pipe_vertex_element ve[2];
41   struct pipe_vertex_buffer vbuf;
42   void *handle;
43
44   memset(ve, 0, sizeof ve);
45
46   ve[0].src_offset = Offset(struct vertex, position);
47   ve[0].src_format = PIPE_FORMAT_R32G32B32A32_FLOAT;
48   ve[1].src_offset = Offset(struct vertex, color);
49   ve[1].src_format = PIPE_FORMAT_R32G32B32A32_FLOAT;
50
51   handle = info.ctx->create_vertex_elements_state(info.ctx, 2, ve);
52   info.ctx->bind_vertex_elements_state(info.ctx, handle);
53
54   memset(&vbuf, 0, sizeof vbuf);
55
56   vbuf.stride = sizeof(struct vertex);
57   vbuf.buffer_offset = 0;
58   vbuf.buffer = pipe_buffer_create_with_data(info.ctx,
59                                              PIPE_BIND_VERTEX_BUFFER,
60                                              PIPE_USAGE_DEFAULT,
61                                              sizeof(vertices),
62                                              vertices);
63
64   info.ctx->set_vertex_buffers(info.ctx, 0, 1, &vbuf);
65}
66
67static void set_vertex_shader(void)
68{
69   void *handle;
70   const char *text =
71      "VERT\n"
72      "DCL IN[0]\n"
73      "DCL IN[1]\n"
74      "DCL OUT[0], POSITION\n"
75      "DCL OUT[1], GENERIC[0]\n"
76      "  0: MOV OUT[1], IN[1]\n"
77      "  1: MOV OUT[0], IN[0]\n"
78      "  2: END\n";
79
80   handle = graw_parse_vertex_shader(info.ctx, text);
81   info.ctx->bind_vs_state(info.ctx, handle);
82}
83
84static void set_fragment_shader(void)
85{
86   void *handle;
87   const char *text =
88      "FRAG\n"
89      "DCL IN[0], GENERIC[0], PERSPECTIVE\n"
90      "DCL OUT[0], COLOR\n"
91      "DCL SAMP[0]\n"
92      "DCL SVIEW[0], 2D, FLOAT\n"
93      "  0: TXP OUT[0], IN[0], SAMP[0], 2D\n"
94      "  2: END\n";
95
96   handle = graw_parse_fragment_shader(info.ctx, text);
97   info.ctx->bind_fs_state(info.ctx, handle);
98}
99
100
101static void draw(void)
102{
103   union pipe_color_union clear_color;
104
105   clear_color.f[0] = 0.5;
106   clear_color.f[1] = 0.5;
107   clear_color.f[2] = 0.5;
108   clear_color.f[3] = 1.0;
109
110   info.ctx->clear(info.ctx, PIPE_CLEAR_COLOR, &clear_color, 0, 0);
111   util_draw_arrays(info.ctx, PIPE_PRIM_QUADS, 0, 4);
112   info.ctx->flush(info.ctx, NULL, 0);
113
114   graw_util_flush_front(&info);
115}
116
117
118
119static void
120init_tex(const unsigned swizzle[4])
121{
122#define SIZE 256
123   struct pipe_sampler_view sv_template;
124   ubyte tex2d[SIZE][SIZE][4];
125   int s, t;
126
127   for (s = 0; s < SIZE; s++) {
128      for (t = 0; t < SIZE; t++) {
129         tex2d[t][s][0] = 0;  /*B*/
130         tex2d[t][s][1] = t;  /*G*/
131         tex2d[t][s][2] = s;  /*R*/
132         tex2d[t][s][3] = 1;  /*A*/
133      }
134   }
135
136   texture = graw_util_create_tex2d(&info, SIZE, SIZE,
137                                    PIPE_FORMAT_B8G8R8A8_UNORM, tex2d);
138
139   memset(&sv_template, 0, sizeof sv_template);
140   sv_template.format = texture->format;
141   sv_template.texture = texture;
142   sv_template.swizzle_r = swizzle[0];
143   sv_template.swizzle_g = swizzle[1];
144   sv_template.swizzle_b = swizzle[2];
145   sv_template.swizzle_a = swizzle[3];
146   sv = info.ctx->create_sampler_view(info.ctx, texture, &sv_template);
147   if (sv == NULL)
148      exit(5);
149
150   info.ctx->set_sampler_views(info.ctx, PIPE_SHADER_FRAGMENT, 0, 1, &sv);
151
152   sampler = graw_util_create_simple_sampler(&info,
153                                             PIPE_TEX_WRAP_REPEAT,
154                                             PIPE_TEX_FILTER_NEAREST);
155
156   info.ctx->bind_sampler_states(info.ctx, PIPE_SHADER_FRAGMENT,
157                                 0, 1, &sampler);
158#undef SIZE
159}
160
161
162static void
163init(const unsigned swizzle[4])
164{
165   if (!graw_util_create_window(&info, WIDTH, HEIGHT, 1, FALSE))
166      exit(1);
167
168   graw_util_default_state(&info, FALSE);
169
170   graw_util_viewport(&info, 0, 0, WIDTH, HEIGHT, 30, 10000);
171
172   init_tex(swizzle);
173
174   set_vertices();
175   set_vertex_shader();
176   set_fragment_shader();
177}
178
179
180static unsigned
181char_to_swizzle(char c)
182{
183   switch (c) {
184   case 'r':
185      return PIPE_SWIZZLE_X;
186   case 'g':
187      return PIPE_SWIZZLE_Y;
188   case 'b':
189      return PIPE_SWIZZLE_Z;
190   case 'a':
191      return PIPE_SWIZZLE_W;
192   case '0':
193      return PIPE_SWIZZLE_0;
194   case '1':
195      return PIPE_SWIZZLE_1;
196   default:
197      return PIPE_SWIZZLE_X;
198   }
199}
200
201
202int main(int argc, char *argv[])
203{
204   const char swizzle_names[] = "rgba01";
205   uint swizzle[4];
206   int i;
207
208   swizzle[0] = PIPE_SWIZZLE_X;
209   swizzle[1] = PIPE_SWIZZLE_Y;
210   swizzle[2] = PIPE_SWIZZLE_Z;
211   swizzle[3] = PIPE_SWIZZLE_W;
212
213   for (i = 1; i < argc; i++) {
214      swizzle[i-1] = char_to_swizzle(argv[i][0]);
215   }
216
217   printf("Example:\n");
218   printf("  tex-swizzle r 0 g 1\n");
219   printf("Current swizzle = ");
220   for (i = 0; i < 4; i++) {
221      printf("%c", swizzle_names[swizzle[i]]);
222   }
223   printf("\n");
224
225   init(swizzle);
226
227   graw_set_display_func(draw);
228   graw_main_loop();
229   return 0;
230}
231