swr/rast: add additional jit utility functions
[mesa.git] / src / gallium / drivers / swr / swr_draw.cpp
1 /****************************************************************************
2 * Copyright (C) 2015 Intel Corporation. All Rights Reserved.
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice (including the next
12 * paragraph) shall be included in all copies or substantial portions of the
13 * Software.
14 *
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
18 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
21 * IN THE SOFTWARE.
22 ***************************************************************************/
23
24 #include "swr_screen.h"
25 #include "swr_context.h"
26 #include "swr_resource.h"
27 #include "swr_fence.h"
28 #include "swr_query.h"
29 #include "jit_api.h"
30
31 #include "util/u_draw.h"
32 #include "util/u_prim.h"
33
34 /*
35 * Draw vertex arrays, with optional indexing, optional instancing.
36 */
37 static void
38 swr_draw_vbo(struct pipe_context *pipe, const struct pipe_draw_info *info)
39 {
40 struct swr_context *ctx = swr_context(pipe);
41
42 if (!info->count_from_stream_output && !info->indirect &&
43 !info->primitive_restart &&
44 !u_trim_pipe_prim(info->mode, (unsigned*)&info->count))
45 return;
46
47 if (!swr_check_render_cond(pipe))
48 return;
49
50 if (info->indirect) {
51 util_draw_indirect(pipe, info);
52 return;
53 }
54
55 /* Update derived state, pass draw info to update function */
56 swr_update_derived(pipe, info);
57
58 swr_update_draw_context(ctx);
59
60 if (ctx->vs->pipe.stream_output.num_outputs) {
61 if (!ctx->vs->soFunc[info->mode]) {
62 STREAMOUT_COMPILE_STATE state = {0};
63 struct pipe_stream_output_info *so = &ctx->vs->pipe.stream_output;
64
65 state.numVertsPerPrim = u_vertices_per_prim(info->mode);
66
67 uint32_t offsets[MAX_SO_STREAMS] = {0};
68 uint32_t num = 0;
69
70 for (uint32_t i = 0; i < so->num_outputs; i++) {
71 assert(so->output[i].stream == 0); // @todo
72 uint32_t output_buffer = so->output[i].output_buffer;
73 if (so->output[i].dst_offset != offsets[output_buffer]) {
74 // hole - need to fill
75 state.stream.decl[num].bufferIndex = output_buffer;
76 state.stream.decl[num].hole = true;
77 state.stream.decl[num].componentMask =
78 (1 << (so->output[i].dst_offset - offsets[output_buffer]))
79 - 1;
80 num++;
81 offsets[output_buffer] = so->output[i].dst_offset;
82 }
83
84 state.stream.decl[num].bufferIndex = output_buffer;
85 state.stream.decl[num].attribSlot = so->output[i].register_index - 1;
86 state.stream.decl[num].componentMask =
87 ((1 << so->output[i].num_components) - 1)
88 << so->output[i].start_component;
89 state.stream.decl[num].hole = false;
90 num++;
91
92 offsets[output_buffer] += so->output[i].num_components;
93 }
94
95 state.stream.numDecls = num;
96
97 HANDLE hJitMgr = swr_screen(pipe->screen)->hJitMgr;
98 ctx->vs->soFunc[info->mode] = JitCompileStreamout(hJitMgr, state);
99 debug_printf("so shader %p\n", ctx->vs->soFunc[info->mode]);
100 assert(ctx->vs->soFunc[info->mode] && "Error: SoShader = NULL");
101 }
102
103 SwrSetSoFunc(ctx->swrContext, ctx->vs->soFunc[info->mode], 0);
104 }
105
106 struct swr_vertex_element_state *velems = ctx->velems;
107 velems->fsState.cutIndex = info->restart_index;
108 velems->fsState.bEnableCutIndex = info->primitive_restart;
109 velems->fsState.bPartialVertexBuffer = (info->min_index > 0);
110
111 swr_jit_fetch_key key;
112 swr_generate_fetch_key(key, velems);
113 auto search = velems->map.find(key);
114 if (search != velems->map.end()) {
115 velems->fsFunc = search->second;
116 } else {
117 HANDLE hJitMgr = swr_screen(ctx->pipe.screen)->hJitMgr;
118 velems->fsFunc = JitCompileFetch(hJitMgr, velems->fsState);
119
120 debug_printf("fetch shader %p\n", velems->fsFunc);
121 assert(velems->fsFunc && "Error: FetchShader = NULL");
122
123 velems->map.insert(std::make_pair(key, velems->fsFunc));
124 }
125
126 SwrSetFetchFunc(ctx->swrContext, velems->fsFunc);
127
128 /* Set up frontend state
129 * XXX setup provokingVertex & topologyProvokingVertex */
130 SWR_FRONTEND_STATE feState = {0};
131 if (ctx->rasterizer->flatshade_first) {
132 feState.provokingVertex = {1, 0, 0};
133 } else {
134 feState.provokingVertex = {2, 1, 2};
135 }
136
137 enum pipe_prim_type topology;
138 if (ctx->gs)
139 topology = (pipe_prim_type)ctx->gs->info.base.properties[TGSI_PROPERTY_GS_OUTPUT_PRIM];
140 else
141 topology = info->mode;
142
143 switch (topology) {
144 case PIPE_PRIM_TRIANGLE_FAN:
145 feState.topologyProvokingVertex = feState.provokingVertex.triFan;
146 break;
147 case PIPE_PRIM_TRIANGLE_STRIP:
148 case PIPE_PRIM_TRIANGLES:
149 feState.topologyProvokingVertex = feState.provokingVertex.triStripList;
150 break;
151 case PIPE_PRIM_QUAD_STRIP:
152 case PIPE_PRIM_QUADS:
153 if (ctx->rasterizer->flatshade_first)
154 feState.topologyProvokingVertex = 0;
155 else
156 feState.topologyProvokingVertex = 3;
157 break;
158 case PIPE_PRIM_LINES:
159 case PIPE_PRIM_LINE_LOOP:
160 case PIPE_PRIM_LINE_STRIP:
161 feState.topologyProvokingVertex = feState.provokingVertex.lineStripList;
162 break;
163 default:
164 feState.topologyProvokingVertex = 0;
165 }
166
167 feState.bEnableCutIndex = info->primitive_restart;
168 SwrSetFrontendState(ctx->swrContext, &feState);
169
170 if (info->indexed)
171 SwrDrawIndexedInstanced(ctx->swrContext,
172 swr_convert_prim_topology(info->mode),
173 info->count,
174 info->instance_count,
175 info->start,
176 info->index_bias,
177 info->start_instance);
178 else
179 SwrDrawInstanced(ctx->swrContext,
180 swr_convert_prim_topology(info->mode),
181 info->count,
182 info->instance_count,
183 info->start,
184 info->start_instance);
185 }
186
187
188 static void
189 swr_flush(struct pipe_context *pipe,
190 struct pipe_fence_handle **fence,
191 unsigned flags)
192 {
193 struct swr_context *ctx = swr_context(pipe);
194 struct swr_screen *screen = swr_screen(pipe->screen);
195 struct pipe_surface *cb = ctx->framebuffer.cbufs[0];
196
197 /* If the current renderTarget is the display surface, store tiles back to
198 * the surface, in preparation for present (swr_flush_frontbuffer).
199 * Other renderTargets get stored back when attachment changes or
200 * swr_surface_destroy */
201 if (cb && swr_resource(cb->texture)->display_target)
202 swr_store_dirty_resource(pipe, cb->texture, SWR_TILE_RESOLVED);
203
204 if (fence)
205 swr_fence_reference(pipe->screen, fence, screen->flush_fence);
206 }
207
208 void
209 swr_finish(struct pipe_context *pipe)
210 {
211 struct pipe_fence_handle *fence = nullptr;
212
213 swr_flush(pipe, &fence, 0);
214 swr_fence_finish(pipe->screen, NULL, fence, 0);
215 swr_fence_reference(pipe->screen, &fence, NULL);
216 }
217
218
219 /*
220 * Store SWR HotTiles back to renderTarget surface.
221 */
222 void
223 swr_store_render_target(struct pipe_context *pipe,
224 uint32_t attachment,
225 enum SWR_TILE_STATE post_tile_state)
226 {
227 struct swr_context *ctx = swr_context(pipe);
228 struct swr_draw_context *pDC = &ctx->swrDC;
229 struct SWR_SURFACE_STATE *renderTarget = &pDC->renderTargets[attachment];
230
231 /* Only proceed if there's a valid surface to store to */
232 if (renderTarget->pBaseAddress) {
233 swr_update_draw_context(ctx);
234 SWR_RECT full_rect =
235 {0, 0,
236 (int32_t)u_minify(renderTarget->width, renderTarget->lod),
237 (int32_t)u_minify(renderTarget->height, renderTarget->lod)};
238 SwrStoreTiles(ctx->swrContext,
239 1 << attachment,
240 post_tile_state,
241 full_rect);
242 }
243 }
244
245 void
246 swr_store_dirty_resource(struct pipe_context *pipe,
247 struct pipe_resource *resource,
248 enum SWR_TILE_STATE post_tile_state)
249 {
250 /* Only store resource if it has been written to */
251 if (swr_resource(resource)->status & SWR_RESOURCE_WRITE) {
252 struct swr_context *ctx = swr_context(pipe);
253 struct swr_screen *screen = swr_screen(pipe->screen);
254 struct swr_resource *spr = swr_resource(resource);
255
256 swr_draw_context *pDC = &ctx->swrDC;
257 SWR_SURFACE_STATE *renderTargets = pDC->renderTargets;
258 for (uint32_t i = 0; i < SWR_NUM_ATTACHMENTS; i++)
259 if (renderTargets[i].pBaseAddress == spr->swr.pBaseAddress ||
260 (spr->secondary.pBaseAddress &&
261 renderTargets[i].pBaseAddress == spr->secondary.pBaseAddress)) {
262 swr_store_render_target(pipe, i, post_tile_state);
263
264 /* Mesa thinks depth/stencil are fused, so we'll never get an
265 * explicit resource for stencil. So, if checking depth, then
266 * also check for stencil. */
267 if (spr->has_stencil && (i == SWR_ATTACHMENT_DEPTH)) {
268 swr_store_render_target(
269 pipe, SWR_ATTACHMENT_STENCIL, post_tile_state);
270 }
271
272 /* This fence signals StoreTiles completion */
273 swr_fence_submit(ctx, screen->flush_fence);
274
275 break;
276 }
277 }
278 }
279
280 void
281 swr_draw_init(struct pipe_context *pipe)
282 {
283 pipe->draw_vbo = swr_draw_vbo;
284 pipe->flush = swr_flush;
285 }