1 /*
2  * Copyright 2018 Collabora Ltd.
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 ZINK_COMPILER_H
25 #define ZINK_COMPILER_H
26 
27 #include "pipe/p_defines.h"
28 #include "pipe/p_state.h"
29 
30 #include "compiler/nir/nir.h"
31 #include "compiler/shader_info.h"
32 
33 #include <vulkan/vulkan.h>
34 
35 struct pipe_screen;
36 struct zink_context;
37 struct zink_screen;
38 struct zink_gfx_program;
39 
40 struct nir_shader_compiler_options;
41 struct nir_shader;
42 
43 struct set;
44 
45 struct tgsi_token;
46 struct zink_so_info {
47    struct pipe_stream_output_info so_info;
48    unsigned *so_info_slots;
49 };
50 
51 
52 const void *
53 zink_get_compiler_options(struct pipe_screen *screen,
54                           enum pipe_shader_ir ir,
55                           enum pipe_shader_type shader);
56 
57 struct nir_shader *
58 zink_tgsi_to_nir(struct pipe_screen *screen, const struct tgsi_token *tokens);
59 
60 struct zink_shader {
61    struct nir_shader *nir;
62 
63    struct zink_so_info streamout;
64 
65    struct {
66       int index;
67       int binding;
68       VkDescriptorType type;
69    } bindings[PIPE_MAX_CONSTANT_BUFFERS + PIPE_MAX_SHADER_SAMPLER_VIEWS];
70    size_t num_bindings;
71    struct set *programs;
72 
73    bool has_geometry_shader; // vertex shaders need to know if a geometry shader exists
74 };
75 
76 VkShaderModule
77 zink_shader_compile(struct zink_screen *screen, struct zink_shader *zs,
78                     unsigned char *shader_slot_map, unsigned char *shader_slots_reserved);
79 
80 struct zink_shader *
81 zink_shader_create(struct zink_screen *screen, struct nir_shader *nir,
82                  const struct pipe_stream_output_info *so_info);
83 
84 void
85 zink_shader_free(struct zink_context *ctx, struct zink_shader *shader);
86 
87 #endif
88