i965g: add constant buffer setter
[mesa.git] / src / gallium / drivers / i965 / brw_pipe_shader.c
1 /*
2 Copyright (C) Intel Corp. 2006. All Rights Reserved.
3 Intel funded Tungsten Graphics (http://www.tungstengraphics.com) to
4 develop this 3D driver.
5
6 Permission is hereby granted, free of charge, to any person obtaining
7 a copy of this software and associated documentation files (the
8 "Software"), to deal in the Software without restriction, including
9 without limitation the rights to use, copy, modify, merge, publish,
10 distribute, sublicense, and/or sell copies of the Software, and to
11 permit persons to whom the Software is furnished to do so, subject to
12 the following conditions:
13
14 The above copyright notice and this permission notice (including the
15 next paragraph) shall be included in all copies or substantial
16 portions of the Software.
17
18 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
19 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
21 IN NO EVENT SHALL THE COPYRIGHT OWNER(S) AND/OR ITS SUPPLIERS BE
22 LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
23 OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
24 WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25
26 **********************************************************************/
27 /*
28 * Authors:
29 * Keith Whitwell <keith@tungstengraphics.com>
30 */
31
32 #include "util/u_memory.h"
33
34 #include "tgsi/tgsi_parse.h"
35 #include "tgsi/tgsi_scan.h"
36
37 #include "brw_context.h"
38 #include "brw_util.h"
39 #include "brw_wm.h"
40
41
42 /**
43 * Determine if the given shader uses complex features such as flow
44 * conditionals, loops, subroutines.
45 */
46 GLboolean brw_wm_has_flow_control(const struct brw_fragment_shader *fp)
47 {
48 return (fp->info.opcode_count[TGSI_OPCODE_ARL] > 0 ||
49 fp->info.opcode_count[TGSI_OPCODE_IF] > 0 ||
50 fp->info.opcode_count[TGSI_OPCODE_ENDIF] > 0 || /* redundant - IF */
51 fp->info.opcode_count[TGSI_OPCODE_CAL] > 0 ||
52 fp->info.opcode_count[TGSI_OPCODE_BRK] > 0 || /* redundant - BGNLOOP */
53 fp->info.opcode_count[TGSI_OPCODE_RET] > 0 || /* redundant - CAL */
54 fp->info.opcode_count[TGSI_OPCODE_BGNLOOP] > 0);
55 }
56
57
58
59 static void brw_bind_fs_state( struct pipe_context *pipe, void *prog )
60 {
61 struct brw_context *brw = brw_context(pipe);
62
63 brw->curr.fragment_shader = (struct brw_fragment_shader *)prog;
64 brw->state.dirty.mesa |= PIPE_NEW_FRAGMENT_SHADER;
65 }
66
67 static void brw_bind_vs_state( struct pipe_context *pipe, void *prog )
68 {
69 struct brw_context *brw = brw_context(pipe);
70
71 brw->curr.vertex_shader = (struct brw_vertex_shader *)prog;
72 brw->state.dirty.mesa |= PIPE_NEW_VERTEX_SHADER;
73 }
74
75
76
77 static void *brw_create_fs_state( struct pipe_context *pipe,
78 const struct pipe_shader_state *shader )
79 {
80 struct brw_context *brw = brw_context(pipe);
81 struct brw_fragment_shader *fs;
82 int i;
83
84 fs = CALLOC_STRUCT(brw_fragment_shader);
85 if (fs == NULL)
86 return NULL;
87
88 /* Duplicate tokens, scan shader
89 */
90 fs->id = brw->program_id++;
91 fs->has_flow_control = brw_wm_has_flow_control(fs);
92
93 fs->tokens = tgsi_dup_tokens(shader->tokens);
94 if (fs->tokens == NULL)
95 goto fail;
96
97 tgsi_scan_shader(fs->tokens, &fs->info);
98
99 for (i = 0; i < fs->info.num_inputs; i++)
100 if (fs->info.input_semantic_name[i] == TGSI_SEMANTIC_POSITION)
101 fs->uses_depth = 1;
102
103 if (fs->info.uses_kill)
104 fs->iz_lookup |= IZ_PS_KILL_ALPHATEST_BIT;
105
106 if (fs->info.writes_z)
107 fs->iz_lookup |= IZ_PS_COMPUTES_DEPTH_BIT;
108
109 return (void *)fs;
110
111 fail:
112 FREE(fs);
113 return NULL;
114 }
115
116
117 static void *brw_create_vs_state( struct pipe_context *pipe,
118 const struct pipe_shader_state *shader )
119 {
120 struct brw_context *brw = brw_context(pipe);
121
122 struct brw_vertex_shader *vs = CALLOC_STRUCT(brw_vertex_shader);
123 if (vs == NULL)
124 return NULL;
125
126 /* Duplicate tokens, scan shader
127 */
128 vs->id = brw->program_id++;
129 //vs->has_flow_control = brw_wm_has_flow_control(vs);
130
131 /* Tell the draw module about this shader:
132 */
133
134 /* Done:
135 */
136 return (void *)vs;
137 }
138
139
140 static void brw_delete_fs_state( struct pipe_context *pipe, void *prog )
141 {
142 struct brw_context *brw = brw_context(pipe);
143 struct brw_fragment_shader *fs = (struct brw_fragment_shader *)prog;
144
145 brw->sws->bo_unreference(fs->const_buffer);
146 FREE( (void *)fs->tokens );
147 FREE( fs );
148 }
149
150
151 static void brw_delete_vs_state( struct pipe_context *pipe, void *prog )
152 {
153 struct brw_fragment_shader *vs = (struct brw_fragment_shader *)prog;
154
155 /* Delete draw shader
156 */
157 FREE( (void *)vs->tokens );
158 FREE( vs );
159 }
160
161
162 static void brw_set_constant_buffer(struct pipe_context *pipe,
163 uint shader, uint index,
164 const struct pipe_constant_buffer *buf)
165 {
166 struct brw_context *brw = brw_context(pipe);
167
168 assert(index == 0);
169
170 if (shader == PIPE_SHADER_FRAGMENT) {
171 pipe_buffer_reference( &brw->curr.fragment_constants,
172 buf->buffer );
173
174 brw->state.dirty.mesa |= PIPE_NEW_FRAGMENT_CONSTANTS;
175 }
176 else {
177 pipe_buffer_reference( &brw->curr.vertex_constants,
178 buf->buffer );
179
180 brw->state.dirty.mesa |= PIPE_NEW_VERTEX_CONSTANTS;
181 }
182 }
183
184
185 void brw_pipe_shader_init( struct brw_context *brw )
186 {
187 brw->base.set_constant_buffer = brw_set_constant_buffer;
188
189 brw->base.create_vs_state = brw_create_vs_state;
190 brw->base.bind_vs_state = brw_bind_vs_state;
191 brw->base.delete_vs_state = brw_delete_vs_state;
192
193 brw->base.create_fs_state = brw_create_fs_state;
194 brw->base.bind_fs_state = brw_bind_fs_state;
195 brw->base.delete_fs_state = brw_delete_fs_state;
196 }
197
198 void brw_pipe_shader_cleanup( struct brw_context *brw )
199 {
200 pipe_buffer_reference( &brw->curr.fragment_constants, NULL );
201 pipe_buffer_reference( &brw->curr.vertex_constants, NULL );
202 }