st/mesa: keep serialized NIR instead of nir_shader in st_program
[mesa.git] / src / mesa / state_tracker / st_program.h
index d8c7b08d4e7a14e849812ed3a804f5a0db95f632..cb085d18bd7e5a6a394462963eb5525f66dfbd2c 100644 (file)
 #define ST_PROGRAM_H
 
 #include "main/mtypes.h"
+#include "main/atifragshader.h"
 #include "program/program.h"
 #include "pipe/p_state.h"
+#include "tgsi/tgsi_from_mesa.h"
 #include "st_context.h"
+#include "st_texture.h"
 #include "st_glsl_to_tgsi.h"
 
-
 #ifdef __cplusplus
 extern "C" {
 #endif
 
+#define ST_DOUBLE_ATTRIB_PLACEHOLDER 0xff
+
+struct st_external_sampler_key
+{
+   GLuint lower_nv12;             /**< bitmask of 2 plane YUV samplers */
+   GLuint lower_iyuv;             /**< bitmask of 3 plane YUV samplers */
+   GLuint lower_xy_uxvx;          /**< bitmask of 2 plane YUV samplers */
+   GLuint lower_yx_xuxv;          /**< bitmask of 2 plane YUV samplers */
+   GLuint lower_ayuv;
+   GLuint lower_xyuv;
+};
+
+static inline struct st_external_sampler_key
+st_get_external_sampler_key(struct st_context *st, struct gl_program *prog)
+{
+   unsigned mask = prog->ExternalSamplersUsed;
+   struct st_external_sampler_key key;
+
+   memset(&key, 0, sizeof(key));
+
+   while (unlikely(mask)) {
+      unsigned unit = u_bit_scan(&mask);
+      struct st_texture_object *stObj =
+            st_get_texture_object(st->ctx, prog, unit);
+
+      switch (st_get_view_format(stObj)) {
+      case PIPE_FORMAT_NV12:
+      case PIPE_FORMAT_P016:
+         key.lower_nv12 |= (1 << unit);
+         break;
+      case PIPE_FORMAT_IYUV:
+         key.lower_iyuv |= (1 << unit);
+         break;
+      case PIPE_FORMAT_YUYV:
+         key.lower_yx_xuxv |= (1 << unit);
+         break;
+      case PIPE_FORMAT_UYVY:
+         key.lower_xy_uxvx |= (1 << unit);
+         break;
+      case PIPE_FORMAT_AYUV:
+         key.lower_ayuv |= (1 << unit);
+         break;
+      case PIPE_FORMAT_XYUV:
+         key.lower_xyuv |= (1 << unit);
+         break;
+      default:
+         printf("mesa: st_get_external_sampler_key: unhandled pipe format %u\n",
+               st_get_view_format(stObj));
+         break;
+      }
+   }
+
+   return key;
+}
 
 /** Fragment program variant key */
 struct st_fp_variant_key
@@ -58,14 +114,29 @@ struct st_fp_variant_key
    GLuint drawpixels:1;           /**< glDrawPixels variant */
    GLuint scaleAndBias:1;         /**< glDrawPixels w/ scale and/or bias? */
    GLuint pixelMaps:1;            /**< glDrawPixels w/ pixel lookup map? */
-   GLuint drawpixels_z:1;         /**< glDrawPixels(GL_DEPTH) */
-   GLuint drawpixels_stencil:1;   /**< glDrawPixels(GL_STENCIL) */
 
    /** for ARB_color_buffer_float */
    GLuint clamp_color:1;
 
    /** for ARB_sample_shading */
    GLuint persample_shading:1;
+
+   /** needed for ATI_fragment_shader */
+   GLuint fog:2;
+
+   /** for ARB_depth_clamp */
+   GLuint lower_depth_clamp:1;
+
+   /** for OpenGL 1.0 on modern hardware */
+   GLuint lower_two_sided_color:1;
+
+   GLuint lower_flatshade:1;
+   enum compare_func lower_alpha_func:3;
+
+   /** needed for ATI_fragment_shader */
+   char texture_targets[MAX_NUM_FRAGMENT_REGISTERS_ATI];
+
+   struct st_external_sampler_key external;
 };
 
 
@@ -77,41 +148,39 @@ struct st_fp_variant
    /** Parameters which generated this version of fragment program */
    struct st_fp_variant_key key;
 
-   struct pipe_shader_state tgsi;
-
    /** Driver's compiled shader */
    void *driver_shader;
 
    /** For glBitmap variants */
-   struct gl_program_parameter_list *parameters;
    uint bitmap_sampler;
 
+   /** For glDrawPixels variants */
+   unsigned drawpix_sampler;
+   unsigned pixelmap_sampler;
+
    /** next in linked list */
    struct st_fp_variant *next;
 };
 
 
-/**
- * Derived from Mesa gl_fragment_program:
- */
-struct st_fragment_program
+/** Shader key shared by other shaders */
+struct st_common_variant_key
 {
-   struct gl_fragment_program Base;
-   struct glsl_to_tgsi_visitor* glsl_to_tgsi;
-
-   struct st_fp_variant *variants;
-};
+   struct st_context *st;          /**< variants are per-context */
+   bool passthrough_edgeflags;
 
+   /** for ARB_color_buffer_float */
+   bool clamp_color;
 
+   /** both for ARB_depth_clamp */
+   bool lower_depth_clamp;
+   bool clip_negative_one_to_one;
 
-/** Vertex program variant key */
-struct st_vp_variant_key
-{
-   struct st_context *st;          /**< variants are per-context */
-   boolean passthrough_edgeflags;
+   /** lower glPointSize to gl_PointSize */
+   boolean lower_point_size;
 
-   /** for ARB_color_buffer_float */
-   boolean clamp_color;
+   /* for user-defined clip-planes */
+   uint8_t lower_ucp;
 };
 
 
@@ -124,13 +193,13 @@ struct st_vp_variant
    /* Parameters which generated this translated version of a vertex
     * shader:
     */
-   struct st_vp_variant_key key;
+   struct st_common_variant_key key;
 
    /**
-    * TGSI tokens (to later generate a 'draw' module shader for
-    * selection/feedback/rasterpos)
+    * The shader variant saved for the draw module to later emulate
+    * selection/feedback/rasterpos.
     */
-   struct pipe_shader_state tgsi;
+   const struct tgsi_token *tokens;
 
    /** Driver's compiled shader */
    void *driver_shader;
@@ -143,114 +212,76 @@ struct st_vp_variant
 
    /** similar to that in st_vertex_program, but with edgeflags info too */
    GLuint num_inputs;
-};
-
-
-/**
- * Derived from Mesa gl_fragment_program:
- */
-struct st_vertex_program
-{
-   struct gl_vertex_program Base;  /**< The Mesa vertex program */
-   struct glsl_to_tgsi_visitor* glsl_to_tgsi;
-
-   /** maps a Mesa VERT_ATTRIB_x to a packed TGSI input index */
-   GLuint input_to_index[VERT_ATTRIB_MAX];
-   /** maps a TGSI input index back to a Mesa VERT_ATTRIB_x */
-   GLuint index_to_input[PIPE_MAX_SHADER_INPUTS];
-   GLuint num_inputs;
-
-   /** Maps VARYING_SLOT_x to slot */
-   GLuint result_to_output[VARYING_SLOT_MAX];
-   ubyte output_semantic_name[VARYING_SLOT_MAX];
-   ubyte output_semantic_index[VARYING_SLOT_MAX];
-   GLuint num_outputs;
-
-   /** List of translated variants of this vertex program.
-    */
-   struct st_vp_variant *variants;
-};
 
-
-
-/** Geometry program variant key */
-struct st_gp_variant_key
-{
-   struct st_context *st;          /**< variants are per-context */
-   /* no other fields yet */
+   /** Bitfield of VERT_BIT_* bits of mesa vertex processing inputs */
+   GLbitfield vert_attrib_mask;
 };
 
 
 /**
  * Geometry program variant.
  */
-struct st_gp_variant
+struct st_common_variant
 {
-   /* Parameters which generated this translated version of a vertex */
-   struct st_gp_variant_key key;
+   /* Parameters which generated this variant. */
+   struct st_common_variant_key key;
 
    void *driver_shader;
 
-   struct st_gp_variant *next;
+   struct st_common_variant *next;
 };
 
 
 /**
- * Derived from Mesa gl_geometry_program:
+ * Derived from Mesa gl_program:
  */
-struct st_geometry_program
+struct st_program
 {
-   struct gl_geometry_program Base;  /**< The Mesa geometry program */
+   struct gl_program Base;
+   struct pipe_shader_state state;
    struct glsl_to_tgsi_visitor* glsl_to_tgsi;
+   struct ati_fragment_shader *ati_fs;
+   uint64_t affected_states; /**< ST_NEW_* flags to mark dirty when binding */
 
-   struct st_gp_variant *variants;
-};
+   void *nir_binary;
+   unsigned nir_size;
 
+  /* used when bypassing glsl_to_tgsi: */
+   struct gl_shader_program *shader_program;
 
+   union {
+      struct st_common_variant *variants;
+      struct st_vp_variant *vp_variants;
+      struct st_fp_variant *fp_variants;
+   };
+};
 
-static INLINE struct st_fragment_program *
-st_fragment_program( struct gl_fragment_program *fp )
+
+struct st_vertex_program
 {
-   return (struct st_fragment_program *)fp;
-}
+   struct st_program Base;
 
+   /** maps a TGSI input index back to a Mesa VERT_ATTRIB_x */
+   ubyte index_to_input[PIPE_MAX_ATTRIBS];
+   ubyte num_inputs;
+   /** Reverse mapping of the above */
+   ubyte input_to_index[VERT_ATTRIB_MAX];
 
-static INLINE struct st_vertex_program *
-st_vertex_program( struct gl_vertex_program *vp )
-{
-   return (struct st_vertex_program *)vp;
-}
+   /** Maps VARYING_SLOT_x to slot */
+   ubyte result_to_output[VARYING_SLOT_MAX];
+};
 
-static INLINE struct st_geometry_program *
-st_geometry_program( struct gl_geometry_program *gp )
-{
-   return (struct st_geometry_program *)gp;
-}
 
-static INLINE void
-st_reference_vertprog(struct st_context *st,
-                      struct st_vertex_program **ptr,
-                      struct st_vertex_program *prog)
+static inline struct st_program *
+st_program( struct gl_program *cp )
 {
-   _mesa_reference_program(st->ctx,
-                           (struct gl_program **) ptr,
-                           (struct gl_program *) prog);
+   return (struct st_program *)cp;
 }
 
-static INLINE void
-st_reference_geomprog(struct st_context *st,
-                      struct st_geometry_program **ptr,
-                      struct st_geometry_program *prog)
-{
-   _mesa_reference_program(st->ctx,
-                           (struct gl_program **) ptr,
-                           (struct gl_program *) prog);
-}
-
-static INLINE void
-st_reference_fragprog(struct st_context *st,
-                      struct st_fragment_program **ptr,
-                      struct st_fragment_program *prog)
+static inline void
+st_reference_prog(struct st_context *st,
+                  struct st_program **ptr,
+                  struct st_program *prog)
 {
    _mesa_reference_program(st->ctx,
                            (struct gl_program **) ptr,
@@ -260,75 +291,69 @@ st_reference_fragprog(struct st_context *st,
 /**
  * This defines mapping from Mesa VARYING_SLOTs to TGSI GENERIC slots.
  */
-static INLINE unsigned
+static inline unsigned
 st_get_generic_varying_index(struct st_context *st, GLuint attr)
 {
-   if (attr >= VARYING_SLOT_VAR0) {
-      if (st->needs_texcoord_semantic)
-         return attr - VARYING_SLOT_VAR0;
-      else
-         return 9 + (attr - VARYING_SLOT_VAR0);
-   }
-   if (attr == VARYING_SLOT_PNTC) {
-      assert(!st->needs_texcoord_semantic);
-      return 8;
-   }
-   if (attr >= VARYING_SLOT_TEX0 && attr <= VARYING_SLOT_TEX7) {
-      assert(!st->needs_texcoord_semantic);
-      return attr - VARYING_SLOT_TEX0;
-   }
-
-   assert(0);
-   return 0;
+   return tgsi_get_generic_gl_varying_index((gl_varying_slot)attr,
+                                            st->needs_texcoord_semantic);
 }
 
+extern void
+st_set_prog_affected_state_flags(struct gl_program *prog);
 
 extern struct st_vp_variant *
 st_get_vp_variant(struct st_context *st,
-                  struct st_vertex_program *stvp,
-                  const struct st_vp_variant_key *key);
+                  struct st_program *stvp,
+                  const struct st_common_variant_key *key);
 
 
 extern struct st_fp_variant *
 st_get_fp_variant(struct st_context *st,
-                  struct st_fragment_program *stfp,
+                  struct st_program *stfp,
                   const struct st_fp_variant_key *key);
 
-
-extern struct st_gp_variant *
-st_get_gp_variant(struct st_context *st,
-                  struct st_geometry_program *stgp,
-                  const struct st_gp_variant_key *key);
-
-
-extern void
-st_prepare_vertex_program(struct gl_context *ctx,
-                          struct st_vertex_program *stvp);
-
-extern GLboolean
-st_prepare_fragment_program(struct gl_context *ctx,
-                            struct st_fragment_program *stfp);
-
+extern struct st_common_variant *
+st_get_common_variant(struct st_context *st,
+                      struct st_program *p,
+                      const struct st_common_variant_key *key);
 
 extern void
 st_release_vp_variants( struct st_context *st,
-                        struct st_vertex_program *stvp );
+                        struct st_program *stvp );
 
 extern void
 st_release_fp_variants( struct st_context *st,
-                        struct st_fragment_program *stfp );
+                        struct st_program *stfp );
 
 extern void
-st_release_gp_variants(struct st_context *st,
-                       struct st_geometry_program *stgp);
+st_release_common_variants(struct st_context *st, struct st_program *p);
 
 extern void
 st_destroy_program_variants(struct st_context *st);
 
+extern void
+st_finalize_nir_before_variants(struct nir_shader *nir);
+
+extern void
+st_prepare_vertex_program(struct st_program *stvp);
 
 extern void
-st_print_current_vertex_program(void);
+st_translate_stream_output_info(struct gl_program *prog);
+
+extern bool
+st_translate_vertex_program(struct st_context *st,
+                            struct st_program *stvp);
+
+extern bool
+st_translate_fragment_program(struct st_context *st,
+                              struct st_program *stfp);
 
+extern bool
+st_translate_common_program(struct st_context *st,
+                            struct st_program *stp);
+
+extern void
+st_finalize_program(struct st_context *st, struct gl_program *prog);
 
 #ifdef __cplusplus
 }