1/*
2 * Copyright 2019 Valve Corporation
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 * on the rights to use, copy, modify, merge, publish, distribute, sub
8 * license, and/or sell copies of the Software, and to permit persons to whom
9 * the 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 NON-INFRINGEMENT. IN NO EVENT SHALL
18 * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM,
19 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
20 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
21 * USE OR OTHER DEALINGS IN THE SOFTWARE.
22 */
23
24#ifndef AC_SHADER_ARGS_H
25#define AC_SHADER_ARGS_H
26
27#include <stdbool.h>
28#include <stdint.h>
29
30/* Maximum dwords of inline push constants when the indirect path is still used */
31#define AC_MAX_INLINE_PUSH_CONSTS_WITH_INDIRECT 8
32/* Maximum dwords of inline push constants when the indirect path is not used */
33#define AC_MAX_INLINE_PUSH_CONSTS 32
34
35enum ac_arg_regfile
36{
37   AC_ARG_SGPR,
38   AC_ARG_VGPR,
39};
40
41enum ac_arg_type
42{
43   AC_ARG_FLOAT,
44   AC_ARG_INT,
45   AC_ARG_CONST_PTR,       /* Pointer to i8 array */
46   AC_ARG_CONST_FLOAT_PTR, /* Pointer to f32 array */
47   AC_ARG_CONST_PTR_PTR,   /* Pointer to pointer to i8 array */
48   AC_ARG_CONST_DESC_PTR,  /* Pointer to v4i32 array */
49   AC_ARG_CONST_IMAGE_PTR, /* Pointer to v8i32 array */
50};
51
52struct ac_arg {
53   uint16_t arg_index;
54   bool used;
55};
56
57#define AC_MAX_ARGS 384 /* including all VS->TCS IO */
58
59struct ac_shader_args {
60   /* Info on how to declare arguments */
61   struct {
62      enum ac_arg_type type;
63      enum ac_arg_regfile file;
64      uint8_t offset;
65      uint8_t size;
66      bool skip;
67   } args[AC_MAX_ARGS];
68
69   uint16_t arg_count;
70   uint16_t num_sgprs_used;
71   uint16_t num_vgprs_used;
72
73   uint16_t return_count;
74   uint16_t num_sgprs_returned;
75   uint16_t num_vgprs_returned;
76
77   /* VS */
78   struct ac_arg base_vertex;
79   struct ac_arg start_instance;
80   struct ac_arg draw_id;
81   struct ac_arg vertex_buffers;
82   struct ac_arg vertex_id;
83   struct ac_arg vs_rel_patch_id;
84   struct ac_arg vs_prim_id;
85   struct ac_arg instance_id;
86
87   /* Merged shaders */
88   struct ac_arg tess_offchip_offset;
89   struct ac_arg merged_wave_info;
90   /* On gfx10:
91    *  - bits 0..11: ordered_wave_id
92    *  - bits 12..20: number of vertices in group
93    *  - bits 22..30: number of primitives in group
94    */
95   struct ac_arg gs_tg_info;
96   struct ac_arg scratch_offset;
97
98   /* TCS */
99   struct ac_arg tcs_factor_offset;
100   struct ac_arg tcs_wave_id; /* gfx11+ */
101   struct ac_arg tcs_patch_id;
102   struct ac_arg tcs_rel_ids;
103
104   /* TES */
105   struct ac_arg tes_u;
106   struct ac_arg tes_v;
107   struct ac_arg tes_rel_patch_id;
108   struct ac_arg tes_patch_id;
109
110   /* GS */
111   struct ac_arg es2gs_offset;      /* separate legacy ES */
112   struct ac_arg gs2vs_offset;      /* legacy GS */
113   struct ac_arg gs_wave_id;        /* legacy GS */
114   struct ac_arg gs_attr_offset;    /* gfx11+: attribute ring offset in 512B increments */
115   struct ac_arg gs_vtx_offset[6];  /* GFX6-8: [0-5], GFX9+: [0-2] packed */
116   struct ac_arg gs_prim_id;
117   struct ac_arg gs_invocation_id;
118
119   /* Streamout */
120   struct ac_arg streamout_config;
121   struct ac_arg streamout_write_index;
122   struct ac_arg streamout_offset[4];
123
124   /* PS */
125   struct ac_arg frag_pos[4];
126   struct ac_arg front_face;
127   struct ac_arg ancillary;
128   struct ac_arg sample_coverage;
129   struct ac_arg prim_mask;
130   struct ac_arg persp_sample;
131   struct ac_arg persp_center;
132   struct ac_arg persp_centroid;
133   struct ac_arg pull_model;
134   struct ac_arg linear_sample;
135   struct ac_arg linear_center;
136   struct ac_arg linear_centroid;
137
138   /* CS */
139   struct ac_arg local_invocation_ids;
140   struct ac_arg num_work_groups;
141   struct ac_arg workgroup_ids[3];
142   struct ac_arg tg_size;
143
144   /* Mesh and task shaders */
145   struct ac_arg task_ring_entry; /* Pointer into the draw and payload rings. */
146
147   /* Vulkan only */
148   struct ac_arg push_constants;
149   struct ac_arg inline_push_consts[AC_MAX_INLINE_PUSH_CONSTS];
150   uint64_t inline_push_const_mask;
151   struct ac_arg view_index;
152   struct ac_arg sbt_descriptors;
153   struct ac_arg ray_launch_size_addr;
154   struct ac_arg force_vrs_rates;
155};
156
157void ac_add_arg(struct ac_shader_args *info, enum ac_arg_regfile regfile, unsigned registers,
158                enum ac_arg_type type, struct ac_arg *arg);
159void ac_add_return(struct ac_shader_args *info, enum ac_arg_regfile regfile);
160
161#endif
162