1 /*
2  * Copyright 2010 Christoph Bumiller
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 shall be included in
12  * all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
17  * THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
18  * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF
19  * OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20  * SOFTWARE.
21  */
22 
23 #include "pipe/p_defines.h"
24 
25 #include "nvc0_context.h"
26 
27 #include "nv50/codegen/nv50_ir_driver.h"
28 
29 /* If only they told use the actual semantic instead of just GENERIC ... */
30 static void
nvc0_mesa_varying_hack(struct nv50_ir_varying * var)31 nvc0_mesa_varying_hack(struct nv50_ir_varying *var)
32 {
33    unsigned c;
34 
35    if (var->sn != TGSI_SEMANTIC_GENERIC)
36       return;
37 
38    if (var->si <= 7) /* gl_TexCoord */
39       for (c = 0; c < 4; ++c)
40          var->slot[c] = (0x300 + var->si * 0x10 + c * 0x4) / 4;
41    else
42    if (var->si == 9) /* gl_PointCoord */
43       for (c = 0; c < 4; ++c)
44          var->slot[c] = (0x2e0 + c * 0x4) / 4;
45    else
46       for (c = 0; c < 4; ++c) /* move down user varyings (first has index 8) */
47          var->slot[c] -= 0x80 / 4;
48 }
49 
50 static uint32_t
nvc0_shader_input_address(unsigned sn,unsigned si,unsigned ubase)51 nvc0_shader_input_address(unsigned sn, unsigned si, unsigned ubase)
52 {
53    switch (sn) {
54    case NV50_SEMANTIC_TESSFACTOR:   return 0x000 + si * 0x4;
55    case TGSI_SEMANTIC_PRIMID:       return 0x060;
56    case TGSI_SEMANTIC_PSIZE:        return 0x06c;
57    case TGSI_SEMANTIC_POSITION:     return 0x070;
58    case TGSI_SEMANTIC_GENERIC:      return ubase + si * 0x10;
59    case TGSI_SEMANTIC_FOG:          return 0x270;
60    case TGSI_SEMANTIC_COLOR:        return 0x280 + si * 0x10;
61    case TGSI_SEMANTIC_BCOLOR:       return 0x2a0 + si * 0x10;
62    case NV50_SEMANTIC_CLIPDISTANCE: return 0x2c0 + si * 0x4;
63    case TGSI_SEMANTIC_CLIPDIST:     return 0x2c0 + si * 0x10;
64    case TGSI_SEMANTIC_CLIPVERTEX:   return 0x260;
65    case NV50_SEMANTIC_POINTCOORD:   return 0x2e0;
66    case NV50_SEMANTIC_TESSCOORD:    return 0x2f0;
67    case TGSI_SEMANTIC_INSTANCEID:   return 0x2f8;
68    case TGSI_SEMANTIC_VERTEXID:     return 0x2fc;
69    case NV50_SEMANTIC_TEXCOORD:     return 0x300 + si * 0x10;
70    case TGSI_SEMANTIC_FACE:         return 0x3fc;
71    case NV50_SEMANTIC_INVOCATIONID: return ~0;
72    default:
73       assert(!"invalid TGSI input semantic");
74       return ~0;
75    }
76 }
77 
78 static uint32_t
nvc0_shader_output_address(unsigned sn,unsigned si,unsigned ubase)79 nvc0_shader_output_address(unsigned sn, unsigned si, unsigned ubase)
80 {
81    switch (sn) {
82    case NV50_SEMANTIC_TESSFACTOR:    return 0x000 + si * 0x4;
83    case TGSI_SEMANTIC_PRIMID:        return 0x060;
84    case NV50_SEMANTIC_LAYER:         return 0x064;
85    case NV50_SEMANTIC_VIEWPORTINDEX: return 0x068;
86    case TGSI_SEMANTIC_PSIZE:         return 0x06c;
87    case TGSI_SEMANTIC_POSITION:      return 0x070;
88    case TGSI_SEMANTIC_GENERIC:       return ubase + si * 0x10;
89    case TGSI_SEMANTIC_FOG:           return 0x270;
90    case TGSI_SEMANTIC_COLOR:         return 0x280 + si * 0x10;
91    case TGSI_SEMANTIC_BCOLOR:        return 0x2a0 + si * 0x10;
92    case NV50_SEMANTIC_CLIPDISTANCE:  return 0x2c0 + si * 0x4;
93    case TGSI_SEMANTIC_CLIPDIST:      return 0x2c0 + si * 0x10;
94    case TGSI_SEMANTIC_CLIPVERTEX:    return 0x260;
95    case NV50_SEMANTIC_TEXCOORD:      return 0x300 + si * 0x10;
96    case TGSI_SEMANTIC_EDGEFLAG:      return ~0;
97    default:
98       assert(!"invalid TGSI output semantic");
99       return ~0;
100    }
101 }
102 
103 static int
nvc0_vp_assign_input_slots(struct nv50_ir_prog_info * info)104 nvc0_vp_assign_input_slots(struct nv50_ir_prog_info *info)
105 {
106    unsigned i, c, n;
107 
108    for (n = 0, i = 0; i < info->numInputs; ++i) {
109       switch (info->in[i].sn) {
110       case TGSI_SEMANTIC_INSTANCEID: /* for SM4 only, in TGSI they're SVs */
111       case TGSI_SEMANTIC_VERTEXID:
112          info->in[i].mask = 0x1;
113          info->in[i].slot[0] =
114             nvc0_shader_input_address(info->in[i].sn, 0, 0) / 4;
115          continue;
116       default:
117          break;
118       }
119       for (c = 0; c < 4; ++c)
120          info->in[i].slot[c] = (0x80 + n * 0x10 + c * 0x4) / 4;
121       ++n;
122    }
123 
124    return 0;
125 }
126 
127 static int
nvc0_sp_assign_input_slots(struct nv50_ir_prog_info * info)128 nvc0_sp_assign_input_slots(struct nv50_ir_prog_info *info)
129 {
130    unsigned ubase = MAX2(0x80, 0x20 + info->numPatchConstants * 0x10);
131    unsigned offset;
132    unsigned i, c;
133 
134    for (i = 0; i < info->numInputs; ++i) {
135       offset = nvc0_shader_input_address(info->in[i].sn,
136                                          info->in[i].si, ubase);
137       if (info->in[i].patch && offset >= 0x20)
138          offset = 0x20 + info->in[i].si * 0x10;
139 
140       if (info->in[i].sn == NV50_SEMANTIC_TESSCOORD)
141          info->in[i].mask &= 3;
142 
143       for (c = 0; c < 4; ++c)
144          info->in[i].slot[c] = (offset + c * 0x4) / 4;
145 
146       nvc0_mesa_varying_hack(&info->in[i]);
147    }
148 
149    return 0;
150 }
151 
152 static int
nvc0_fp_assign_output_slots(struct nv50_ir_prog_info * info)153 nvc0_fp_assign_output_slots(struct nv50_ir_prog_info *info)
154 {
155    unsigned count = info->prop.fp.numColourResults * 4;
156    unsigned i, c;
157 
158    for (i = 0; i < info->numOutputs; ++i)
159       if (info->out[i].sn == TGSI_SEMANTIC_COLOR)
160          for (c = 0; c < 4; ++c)
161             info->out[i].slot[c] = info->out[i].si * 4 + c;
162 
163    if (info->io.sampleMask < PIPE_MAX_SHADER_OUTPUTS)
164       info->out[info->io.sampleMask].slot[0] = count++;
165    else
166    if (info->target >= 0xe0)
167       count++; /* on Kepler, depth is always last colour reg + 2 */
168 
169    if (info->io.fragDepth < PIPE_MAX_SHADER_OUTPUTS)
170       info->out[info->io.fragDepth].slot[2] = count;
171 
172    return 0;
173 }
174 
175 static int
nvc0_sp_assign_output_slots(struct nv50_ir_prog_info * info)176 nvc0_sp_assign_output_slots(struct nv50_ir_prog_info *info)
177 {
178    unsigned ubase = MAX2(0x80, 0x20 + info->numPatchConstants * 0x10);
179    unsigned offset;
180    unsigned i, c;
181 
182    for (i = 0; i < info->numOutputs; ++i) {
183       offset = nvc0_shader_output_address(info->out[i].sn,
184                                           info->out[i].si, ubase);
185       if (info->out[i].patch && offset >= 0x20)
186          offset = 0x20 + info->out[i].si * 0x10;
187 
188       for (c = 0; c < 4; ++c)
189          info->out[i].slot[c] = (offset + c * 0x4) / 4;
190 
191       nvc0_mesa_varying_hack(&info->out[i]);
192    }
193 
194    return 0;
195 }
196 
197 static int
nvc0_program_assign_varying_slots(struct nv50_ir_prog_info * info)198 nvc0_program_assign_varying_slots(struct nv50_ir_prog_info *info)
199 {
200    int ret;
201 
202    if (info->type == PIPE_SHADER_VERTEX)
203       ret = nvc0_vp_assign_input_slots(info);
204    else
205       ret = nvc0_sp_assign_input_slots(info);
206    if (ret)
207       return ret;
208 
209    if (info->type == PIPE_SHADER_FRAGMENT)
210       ret = nvc0_fp_assign_output_slots(info);
211    else
212       ret = nvc0_sp_assign_output_slots(info);
213    return ret;
214 }
215 
216 static INLINE void
nvc0_vtgp_hdr_update_oread(struct nvc0_program * vp,uint8_t slot)217 nvc0_vtgp_hdr_update_oread(struct nvc0_program *vp, uint8_t slot)
218 {
219    uint8_t min = (vp->hdr[4] >> 12) & 0xff;
220    uint8_t max = (vp->hdr[4] >> 24);
221 
222    min = MIN2(min, slot);
223    max = MAX2(max, slot);
224 
225    vp->hdr[4] = (max << 24) | (min << 12);
226 }
227 
228 /* Common part of header generation for VP, TCP, TEP and GP. */
229 static int
nvc0_vtgp_gen_header(struct nvc0_program * vp,struct nv50_ir_prog_info * info)230 nvc0_vtgp_gen_header(struct nvc0_program *vp, struct nv50_ir_prog_info *info)
231 {
232    unsigned i, c, a;
233 
234    for (i = 0; i < info->numInputs; ++i) {
235       if (info->in[i].patch)
236          continue;
237       for (c = 0; c < 4; ++c) {
238          a = info->in[i].slot[c];
239          if (info->in[i].mask & (1 << c)) {
240             if (info->in[i].sn != NV50_SEMANTIC_TESSCOORD)
241                vp->hdr[5 + a / 32] |= 1 << (a % 32);
242             else
243                nvc0_vtgp_hdr_update_oread(vp, info->in[i].slot[c]);
244          }
245       }
246    }
247 
248    for (i = 0; i < info->numOutputs; ++i) {
249       if (info->out[i].patch)
250          continue;
251       for (c = 0; c < 4; ++c) {
252          if (!(info->out[i].mask & (1 << c)))
253             continue;
254          assert(info->out[i].slot[c] >= 0x40 / 4);
255          a = info->out[i].slot[c] - 0x40 / 4;
256          vp->hdr[13 + a / 32] |= 1 << (a % 32);
257          if (info->out[i].oread)
258             nvc0_vtgp_hdr_update_oread(vp, info->out[i].slot[c]);
259       }
260    }
261 
262    for (i = 0; i < info->numSysVals; ++i) {
263       switch (info->sv[i].sn) {
264       case TGSI_SEMANTIC_PRIMID:
265          vp->hdr[5] |= 1 << 24;
266          break;
267       case TGSI_SEMANTIC_INSTANCEID:
268          vp->hdr[10] |= 1 << 30;
269          break;
270       case TGSI_SEMANTIC_VERTEXID:
271          vp->hdr[10] |= 1 << 31;
272          break;
273       default:
274          break;
275       }
276    }
277 
278    vp->vp.clip_enable = info->io.clipDistanceMask;
279    for (i = 0; i < 8; ++i)
280       if (info->io.cullDistanceMask & (1 << i))
281          vp->vp.clip_mode |= 1 << (i * 4);
282 
283    if (info->io.genUserClip < 0)
284       vp->vp.num_ucps = PIPE_MAX_CLIP_PLANES + 1; /* prevent rebuilding */
285 
286    return 0;
287 }
288 
289 static int
nvc0_vp_gen_header(struct nvc0_program * vp,struct nv50_ir_prog_info * info)290 nvc0_vp_gen_header(struct nvc0_program *vp, struct nv50_ir_prog_info *info)
291 {
292    vp->hdr[0] = 0x20061 | (1 << 10);
293    vp->hdr[4] = 0xff000;
294 
295    vp->hdr[18] = info->io.clipDistanceMask;
296 
297    return nvc0_vtgp_gen_header(vp, info);
298 }
299 
300 #if defined(PIPE_SHADER_HULL) || defined(PIPE_SHADER_DOMAIN)
301 static void
nvc0_tp_get_tess_mode(struct nvc0_program * tp,struct nv50_ir_prog_info * info)302 nvc0_tp_get_tess_mode(struct nvc0_program *tp, struct nv50_ir_prog_info *info)
303 {
304    if (info->prop.tp.outputPrim == PIPE_PRIM_MAX) {
305       tp->tp.tess_mode = ~0;
306       return;
307    }
308    switch (info->prop.tp.domain) {
309    case PIPE_PRIM_LINES:
310       tp->tp.tess_mode = NVC0_3D_TESS_MODE_PRIM_ISOLINES;
311       break;
312    case PIPE_PRIM_TRIANGLES:
313       tp->tp.tess_mode = NVC0_3D_TESS_MODE_PRIM_TRIANGLES;
314       if (info->prop.tp.winding > 0)
315          tp->tp.tess_mode |= NVC0_3D_TESS_MODE_CW;
316       break;
317    case PIPE_PRIM_QUADS:
318       tp->tp.tess_mode = NVC0_3D_TESS_MODE_PRIM_QUADS;
319       break;
320    default:
321       tp->tp.tess_mode = ~0;
322       return;
323    }
324    if (info->prop.tp.outputPrim != PIPE_PRIM_POINTS)
325       tp->tp.tess_mode |= NVC0_3D_TESS_MODE_CONNECTED;
326 
327    switch (info->prop.tp.partitioning) {
328    case PIPE_TESS_PART_INTEGER:
329    case PIPE_TESS_PART_POW2:
330       tp->tp.tess_mode |= NVC0_3D_TESS_MODE_SPACING_EQUAL;
331       break;
332    case PIPE_TESS_PART_FRACT_ODD:
333       tp->tp.tess_mode |= NVC0_3D_TESS_MODE_SPACING_FRACTIONAL_ODD;
334       break;
335    case PIPE_TESS_PART_FRACT_EVEN:
336       tp->tp.tess_mode |= NVC0_3D_TESS_MODE_SPACING_FRACTIONAL_EVEN;
337       break;
338    default:
339       assert(!"invalid tessellator partitioning");
340       break;
341    }
342 }
343 #endif
344 
345 #ifdef PIPE_SHADER_HULL
346 static int
nvc0_tcp_gen_header(struct nvc0_program * tcp,struct nv50_ir_prog_info * info)347 nvc0_tcp_gen_header(struct nvc0_program *tcp, struct nv50_ir_prog_info *info)
348 {
349    unsigned opcs = 6; /* output patch constants (at least the TessFactors) */
350 
351    tcp->tp.input_patch_size = info->prop.tp.inputPatchSize;
352 
353    if (info->numPatchConstants)
354       opcs = 8 + info->numPatchConstants * 4;
355 
356    tcp->hdr[0] = 0x20061 | (2 << 10);
357 
358    tcp->hdr[1] = opcs << 24;
359    tcp->hdr[2] = info->prop.tp.outputPatchSize << 24;
360 
361    tcp->hdr[4] = 0xff000; /* initial min/max parallel output read address */
362 
363    nvc0_vtgp_gen_header(tcp, info);
364 
365    nvc0_tp_get_tess_mode(tcp, info);
366 
367    return 0;
368 }
369 #endif
370 
371 #ifdef PIPE_SHADER_DOMAIN
372 static int
nvc0_tep_gen_header(struct nvc0_program * tep,struct nv50_ir_prog_info * info)373 nvc0_tep_gen_header(struct nvc0_program *tep, struct nv50_ir_prog_info *info)
374 {
375    tep->tp.input_patch_size = ~0;
376 
377    tep->hdr[0] = 0x20061 | (3 << 10);
378    tep->hdr[4] = 0xff000;
379 
380    nvc0_vtgp_gen_header(tep, info);
381 
382    nvc0_tp_get_tess_mode(tep, info);
383 
384    tep->hdr[18] |= 0x3 << 12; /* ? */
385 
386    return 0;
387 }
388 #endif
389 
390 static int
nvc0_gp_gen_header(struct nvc0_program * gp,struct nv50_ir_prog_info * info)391 nvc0_gp_gen_header(struct nvc0_program *gp, struct nv50_ir_prog_info *info)
392 {
393    gp->hdr[0] = 0x20061 | (4 << 10);
394 
395    gp->hdr[2] = MIN2(info->prop.gp.instanceCount, 32) << 24;
396 
397    switch (info->prop.gp.outputPrim) {
398    case PIPE_PRIM_POINTS:
399       gp->hdr[3] = 0x01000000;
400       gp->hdr[0] |= 0xf0000000;
401       break;
402    case PIPE_PRIM_LINE_STRIP:
403       gp->hdr[3] = 0x06000000;
404       gp->hdr[0] |= 0x10000000;
405       break;
406    case PIPE_PRIM_TRIANGLE_STRIP:
407       gp->hdr[3] = 0x07000000;
408       gp->hdr[0] |= 0x10000000;
409       break;
410    default:
411       assert(0);
412       break;
413    }
414 
415    gp->hdr[4] = info->prop.gp.maxVertices & 0x1ff;
416 
417    return nvc0_vtgp_gen_header(gp, info);
418 }
419 
420 #define NVC0_INTERP_FLAT          (1 << 0)
421 #define NVC0_INTERP_PERSPECTIVE   (2 << 0)
422 #define NVC0_INTERP_LINEAR        (3 << 0)
423 #define NVC0_INTERP_CENTROID      (1 << 2)
424 
425 static uint8_t
nvc0_hdr_interp_mode(const struct nv50_ir_varying * var)426 nvc0_hdr_interp_mode(const struct nv50_ir_varying *var)
427 {
428    if (var->linear)
429       return NVC0_INTERP_LINEAR;
430    if (var->flat)
431       return NVC0_INTERP_FLAT;
432    return NVC0_INTERP_PERSPECTIVE;
433 }
434 
435 static int
nvc0_fp_gen_header(struct nvc0_program * fp,struct nv50_ir_prog_info * info)436 nvc0_fp_gen_header(struct nvc0_program *fp, struct nv50_ir_prog_info *info)
437 {
438    unsigned i, c, a, m;
439 
440    /* just 00062 on Kepler */
441    fp->hdr[0] = 0x20062 | (5 << 10);
442    fp->hdr[5] = 0x80000000; /* getting a trap if FRAG_COORD_UMASK.w = 0 */
443 
444    if (info->prop.fp.usesDiscard)
445       fp->hdr[0] |= 0x8000;
446    if (info->prop.fp.numColourResults > 1)
447       fp->hdr[0] |= 0x4000;
448    if (info->io.sampleMask < PIPE_MAX_SHADER_OUTPUTS)
449       fp->hdr[19] |= 0x1;
450    if (info->prop.fp.writesDepth) {
451       fp->hdr[19] |= 0x2;
452       fp->flags[0] = 0x11; /* deactivate ZCULL */
453    }
454 
455    for (i = 0; i < info->numInputs; ++i) {
456       m = nvc0_hdr_interp_mode(&info->in[i]);
457       for (c = 0; c < 4; ++c) {
458          if (!(info->in[i].mask & (1 << c)))
459             continue;
460          a = info->in[i].slot[c];
461          if (info->in[i].slot[0] >= (0x060 / 4) &&
462              info->in[i].slot[0] <= (0x07c / 4)) {
463             fp->hdr[5] |= 1 << (24 + (a - 0x060 / 4));
464          } else
465          if (info->in[i].slot[0] >= (0x2c0 / 4) &&
466              info->in[i].slot[0] <= (0x2fc / 4)) {
467             fp->hdr[14] |= (1 << (a - 0x280 / 4)) & 0x03ff0000;
468          } else {
469             if (info->in[i].slot[c] < (0x040 / 4) ||
470                 info->in[i].slot[c] > (0x380 / 4))
471                continue;
472             a *= 2;
473             if (info->in[i].slot[0] >= (0x300 / 4))
474                a -= 32;
475             fp->hdr[4 + a / 32] |= m << (a % 32);
476          }
477       }
478    }
479 
480    for (i = 0; i < info->numOutputs; ++i) {
481       if (info->out[i].sn == TGSI_SEMANTIC_COLOR)
482          fp->hdr[18] |= info->out[i].mask << info->out[i].slot[0];
483    }
484 
485    fp->fp.early_z = info->prop.fp.earlyFragTests;
486 
487    return 0;
488 }
489 
490 static struct nvc0_transform_feedback_state *
nvc0_program_create_tfb_state(const struct nv50_ir_prog_info * info,const struct pipe_stream_output_info * pso)491 nvc0_program_create_tfb_state(const struct nv50_ir_prog_info *info,
492                               const struct pipe_stream_output_info *pso)
493 {
494    struct nvc0_transform_feedback_state *tfb;
495    unsigned b, i, c;
496 
497    tfb = MALLOC_STRUCT(nvc0_transform_feedback_state);
498    if (!tfb)
499       return NULL;
500    for (b = 0; b < 4; ++b) {
501       tfb->stride[b] = pso->stride[b] * 4;
502       tfb->varying_count[b] = 0;
503    }
504    memset(tfb->varying_index, 0xff, sizeof(tfb->varying_index)); /* = skip */
505 
506    for (i = 0; i < pso->num_outputs; ++i) {
507       unsigned s = pso->output[i].start_component;
508       unsigned p = pso->output[i].dst_offset;
509       b = pso->output[i].output_buffer;
510 
511       for (c = 0; c < pso->output[i].num_components; ++c)
512          tfb->varying_index[b][p++] =
513             info->out[pso->output[i].register_index].slot[s + c];
514 
515       tfb->varying_count[b] = MAX2(tfb->varying_count[b], p);
516    }
517    for (b = 0; b < 4; ++b) // zero unused indices (looks nicer)
518       for (c = tfb->varying_count[b]; c & 3; ++c)
519          tfb->varying_index[b][c] = 0;
520 
521    return tfb;
522 }
523 
524 #ifdef DEBUG
525 static void
nvc0_program_dump(struct nvc0_program * prog)526 nvc0_program_dump(struct nvc0_program *prog)
527 {
528    unsigned pos;
529 
530    for (pos = 0; pos < sizeof(prog->hdr) / sizeof(prog->hdr[0]); ++pos)
531       debug_printf("HDR[%02lx] = 0x%08x\n",
532                    pos * sizeof(prog->hdr[0]), prog->hdr[pos]);
533 
534    debug_printf("shader binary code (0x%x bytes):", prog->code_size);
535    for (pos = 0; pos < prog->code_size / 4; ++pos) {
536       if ((pos % 8) == 0)
537          debug_printf("\n");
538       debug_printf("%08x ", prog->code[pos]);
539    }
540    debug_printf("\n");
541 }
542 #endif
543 
544 boolean
nvc0_program_translate(struct nvc0_program * prog,uint16_t chipset)545 nvc0_program_translate(struct nvc0_program *prog, uint16_t chipset)
546 {
547    struct nv50_ir_prog_info *info;
548    int ret;
549 
550    info = CALLOC_STRUCT(nv50_ir_prog_info);
551    if (!info)
552       return FALSE;
553 
554    info->type = prog->type;
555    info->target = chipset;
556    info->bin.sourceRep = NV50_PROGRAM_IR_TGSI;
557    info->bin.source = (void *)prog->pipe.tokens;
558 
559    info->io.genUserClip = prog->vp.num_ucps;
560    info->io.ucpBase = 256;
561    info->io.ucpBinding = 15;
562 
563    info->assignSlots = nvc0_program_assign_varying_slots;
564 
565 #ifdef DEBUG
566    info->optLevel = debug_get_num_option("NV50_PROG_OPTIMIZE", 3);
567    info->dbgFlags = debug_get_num_option("NV50_PROG_DEBUG", 0);
568 #else
569    info->optLevel = 3;
570 #endif
571 
572    ret = nv50_ir_generate_code(info);
573    if (ret) {
574       NOUVEAU_ERR("shader translation failed: %i\n", ret);
575       goto out;
576    }
577    if (info->bin.syms) /* we don't need them yet */
578       FREE(info->bin.syms);
579 
580    prog->code = info->bin.code;
581    prog->code_size = info->bin.codeSize;
582    prog->immd_data = info->immd.buf;
583    prog->immd_size = info->immd.bufSize;
584    prog->relocs = info->bin.relocData;
585    prog->max_gpr = MAX2(4, (info->bin.maxGPR + 1));
586 
587    prog->vp.need_vertex_id = info->io.vertexId < PIPE_MAX_SHADER_INPUTS;
588 
589    if (info->io.edgeFlagOut < PIPE_MAX_ATTRIBS)
590       info->out[info->io.edgeFlagOut].mask = 0; /* for headergen */
591    prog->vp.edgeflag = info->io.edgeFlagIn;
592 
593    switch (prog->type) {
594    case PIPE_SHADER_VERTEX:
595       ret = nvc0_vp_gen_header(prog, info);
596       break;
597 #ifdef PIPE_SHADER_HULL
598    case PIPE_SHADER_HULL:
599       ret = nvc0_tcp_gen_header(prog, info);
600       break;
601 #endif
602 #ifdef PIPE_SHADER_DOMAIN
603    case PIPE_SHADER_DOMAIN:
604       ret = nvc0_tep_gen_header(prog, info);
605       break;
606 #endif
607    case PIPE_SHADER_GEOMETRY:
608       ret = nvc0_gp_gen_header(prog, info);
609       break;
610    case PIPE_SHADER_FRAGMENT:
611       ret = nvc0_fp_gen_header(prog, info);
612       break;
613    default:
614       ret = -1;
615       NOUVEAU_ERR("unknown program type: %u\n", prog->type);
616       break;
617    }
618    if (ret)
619       goto out;
620 
621    if (info->bin.tlsSpace) {
622       assert(info->bin.tlsSpace < (1 << 24));
623       prog->hdr[0] |= 1 << 26;
624       prog->hdr[1] |= info->bin.tlsSpace; /* l[] size */
625       prog->need_tls = TRUE;
626    }
627    /* TODO: factor 2 only needed where joinat/precont is used,
628     *       and we only have to count non-uniform branches
629     */
630    /*
631    if ((info->maxCFDepth * 2) > 16) {
632       prog->hdr[2] |= (((info->maxCFDepth * 2) + 47) / 48) * 0x200;
633       prog->need_tls = TRUE;
634    }
635    */
636    if (info->io.globalAccess)
637       prog->hdr[0] |= 1 << 16;
638 
639    if (prog->pipe.stream_output.num_outputs)
640       prog->tfb = nvc0_program_create_tfb_state(info,
641                                                 &prog->pipe.stream_output);
642 
643 out:
644    FREE(info);
645    return !ret;
646 }
647 
648 boolean
nvc0_program_upload_code(struct nvc0_context * nvc0,struct nvc0_program * prog)649 nvc0_program_upload_code(struct nvc0_context *nvc0, struct nvc0_program *prog)
650 {
651    struct nvc0_screen *screen = nvc0->screen;
652    int ret;
653    uint32_t size = prog->code_size + NVC0_SHADER_HEADER_SIZE;
654    uint32_t lib_pos = screen->lib_code->start;
655    uint32_t code_pos;
656 
657    /* c[] bindings need to be aligned to 0x100, but we could use relocations
658     * to save space. */
659    if (prog->immd_size) {
660       prog->immd_base = size;
661       size = align(size, 0x40);
662       size += prog->immd_size + 0xc0; /* add 0xc0 for align 0x40 -> 0x100 */
663    }
664    /* On Fermi, SP_START_ID must be aligned to 0x40.
665     * On Kepler, the first instruction must be aligned to 0x80 because
666     * latency information is expected only at certain positions.
667     */
668    if (screen->base.class_3d >= NVE4_3D_CLASS)
669       size = size + 0x70;
670    size = align(size, 0x40);
671 
672    ret = nouveau_heap_alloc(screen->text_heap, size, prog, &prog->mem);
673    if (ret) {
674       NOUVEAU_ERR("out of code space\n");
675       return FALSE;
676    }
677    prog->code_base = prog->mem->start;
678    prog->immd_base = align(prog->mem->start + prog->immd_base, 0x100);
679    assert((prog->immd_size == 0) || (prog->immd_base + prog->immd_size <=
680                                      prog->mem->start + prog->mem->size));
681 
682    if (screen->base.class_3d >= NVE4_3D_CLASS) {
683       switch (prog->mem->start & 0xff) {
684       case 0x40: prog->code_base += 0x70; break;
685       case 0x80: prog->code_base += 0x30; break;
686       case 0xc0: prog->code_base += 0x70; break;
687       default:
688          prog->code_base += 0x30;
689          assert((prog->mem->start & 0xff) == 0x00);
690          break;
691       }
692    }
693    code_pos = prog->code_base + NVC0_SHADER_HEADER_SIZE;
694 
695    if (prog->relocs)
696       nv50_ir_relocate_code(prog->relocs, prog->code, code_pos, lib_pos, 0);
697 
698 #ifdef DEBUG
699    if (debug_get_bool_option("NV50_PROG_DEBUG", FALSE))
700       nvc0_program_dump(prog);
701 #endif
702 
703    nvc0->base.push_data(&nvc0->base, screen->text, prog->code_base,
704                         NOUVEAU_BO_VRAM, NVC0_SHADER_HEADER_SIZE, prog->hdr);
705    nvc0->base.push_data(&nvc0->base, screen->text,
706                         prog->code_base + NVC0_SHADER_HEADER_SIZE,
707                         NOUVEAU_BO_VRAM, prog->code_size, prog->code);
708    if (prog->immd_size)
709       nvc0->base.push_data(&nvc0->base,
710                            screen->text, prog->immd_base, NOUVEAU_BO_VRAM,
711                            prog->immd_size, prog->immd_data);
712 
713    BEGIN_NVC0(nvc0->base.pushbuf, NVC0_3D(MEM_BARRIER), 1);
714    PUSH_DATA (nvc0->base.pushbuf, 0x1011);
715 
716    return TRUE;
717 }
718 
719 /* Upload code for builtin functions like integer division emulation. */
720 void
nvc0_program_library_upload(struct nvc0_context * nvc0)721 nvc0_program_library_upload(struct nvc0_context *nvc0)
722 {
723    struct nvc0_screen *screen = nvc0->screen;
724    int ret;
725    uint32_t size;
726    const uint32_t *code;
727 
728    if (screen->lib_code)
729       return;
730 
731    nv50_ir_get_target_library(screen->base.device->chipset, &code, &size);
732    if (!size)
733       return;
734 
735    ret = nouveau_heap_alloc(screen->text_heap, align(size, 0x100), NULL,
736                             &screen->lib_code);
737    if (ret)
738       return;
739 
740    nvc0->base.push_data(&nvc0->base,
741                         screen->text, screen->lib_code->start, NOUVEAU_BO_VRAM,
742                         size, code);
743    /* no need for a memory barrier, will be emitted with first program */
744 }
745 
746 void
nvc0_program_destroy(struct nvc0_context * nvc0,struct nvc0_program * prog)747 nvc0_program_destroy(struct nvc0_context *nvc0, struct nvc0_program *prog)
748 {
749    const struct pipe_shader_state pipe = prog->pipe;
750    const ubyte type = prog->type;
751 
752    if (prog->mem)
753       nouveau_heap_free(&prog->mem);
754 
755    if (prog->code)
756       FREE(prog->code);
757    if (prog->immd_data)
758       FREE(prog->immd_data);
759    if (prog->relocs)
760       FREE(prog->relocs);
761    if (prog->tfb) {
762       if (nvc0->state.tfb == prog->tfb)
763          nvc0->state.tfb = NULL;
764       FREE(prog->tfb);
765    }
766 
767    memset(prog, 0, sizeof(*prog));
768 
769    prog->pipe = pipe;
770    prog->type = type;
771 }
772