1 files changed, 77 insertions, 0 deletions
diff --git a/src/kosmickrisp/compiler/msl_private.h b/src/kosmickrisp/compiler/msl_private.h
new file mode 100644
index 00000000000..9ccd2bd7922
--- /dev/null
+++ b/src/kosmickrisp/compiler/msl_private.h
@@ -0,0 +1,77 @@
+/*
+ * Copyright 2025 LunarG, Inc.
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: MIT
+ */
+
+#pragma once
+
+#include "util/string_buffer.h"
+#include "nir.h"
+
+struct io_slot_info {
+   nir_alu_type type;
+   uint32_t interpolation;
+   unsigned num_components;
+   bool centroid;
+   bool sample;
+};
+
+struct nir_to_msl_ctx {
+   FILE *output;
+   struct hash_table *types;
+   nir_shader *shader;
+   struct _mesa_string_buffer *text;
+   unsigned short indentlevel;
+   struct io_slot_info inputs_info[NUM_TOTAL_VARYING_SLOTS];
+   struct io_slot_info outputs_info[NUM_TOTAL_VARYING_SLOTS];
+};
+
+#define P_IND(ctx, ...)                                                        \
+   do {                                                                        \
+      for (unsigned i = 0; i < (ctx)->indentlevel; i++)                        \
+         _mesa_string_buffer_append((ctx)->text, "    ");                      \
+      _mesa_string_buffer_printf((ctx)->text, __VA_ARGS__);                    \
+   } while (0);
+
+#define P(ctx, ...) _mesa_string_buffer_printf((ctx)->text, __VA_ARGS__);
+
+#define P_INDENT(ctx)                                                          \
+   do {                                                                        \
+      for (unsigned i = 0; i < (ctx)->indentlevel; i++)                        \
+         _mesa_string_buffer_append((ctx)->text, "    ");                      \
+   } while (0)
+
+/* Perform type inference. The returned value is a
+ * map from nir_def* to base type.*/
+
+struct hash_table *msl_infer_types(nir_shader *shader);
+
+const char *msl_type_for_def(struct hash_table *types, nir_def *def);
+
+const char *msl_uint_type(uint8_t bit_size, uint8_t num_components);
+
+const char *msl_type_for_src(struct hash_table *types, nir_src *src);
+
+const char *msl_bitcast_for_src(struct hash_table *types, nir_src *src);
+
+void msl_src_as_const(struct nir_to_msl_ctx *ctx, nir_src *src);
+
+void msl_emit_io_blocks(struct nir_to_msl_ctx *ctx, nir_shader *shader);
+
+void msl_emit_output_var(struct nir_to_msl_ctx *ctx, nir_shader *shader);
+
+void msl_gather_io_info(struct nir_to_msl_ctx *ctx,
+                        struct io_slot_info *info_array_input,
+                        struct io_slot_info *info_array_output);
+
+const char *msl_input_name(struct nir_to_msl_ctx *ctx, unsigned location);
+
+const char *msl_output_name(struct nir_to_msl_ctx *ctx, unsigned location);
+
+bool msl_src_is_float(struct nir_to_msl_ctx *ctx, nir_src *src);
+bool msl_def_is_sampler(struct nir_to_msl_ctx *ctx, nir_def *def);
+
+void msl_nir_lower_subgroups(nir_shader *nir);
+
+bool msl_nir_lower_algebraic_late(nir_shader *shader);