source/core/program.h

   1 #ifndef MSP_GL_PROGRAM_H_
   2 #define MSP_GL_PROGRAM_H_
   3
   4 #include <map>
   5 #include <string>
   6 #include <vector>
   7 #include <msp/datafile/objectloader.h>
   8 #include "module.h"
   9 #include "program_backend.h"
  10 #include "reflectdata.h"
  11
  12 namespace Msp {
  13 namespace GL {
  14
  15 /**
  16 A shader program consisting of one or more stages.
  17
  18 Programs are created from Modules.  Specialization values can be applied to
  19 customize behaviour of the module.
  20 */
  21 class Program: public ProgramBackend
  22 {
  23         friend ProgramBackend;
  24
  25 public:
  26         class Loader: public DataFile::CollectionObjectLoader<Program>
  27         {
  28         public:
  29                 Loader(Program &, Collection &);
  30
  31         private:
  32                 void module(const std::string &);
  33         };
  34
  35 private:
  36         class SpecializationLoader: public DataFile::Loader
  37         {
  38         private:
  39                 std::map<std::string, int> &spec_values;
  40
  41                 static ActionMap shared_actions;
  42
  43         public:
  44                 SpecializationLoader(std::map<std::string, int> &);
  45
  46         private:
  47                 virtual void init_actions();
  48
  49                 void specialize_bool(const std::string &, bool);
  50                 void specialize_int(const std::string &, int);
  51         };
  52
  53         ReflectData reflect_data;
  54         SpirVModule *specialized_spirv = 0;
  55
  56 public:
  57         /// Constructs an empty Program with no shader stages attached.
  58         Program() = default;
  59
  60         /// Constructs a Program from a Module, with specialization constants.
  61         Program(const Module &, const std::map<std::string, int> & = std::map<std::string, int>());
  62
  63         Program(Program &&);
  64         ~Program();
  65
  66         void add_stages(const Module &, const std::map<std::string, int> & = std::map<std::string, int>());
  67 private:
  68         void collect_uniforms(const SpirVModule &);
  69         void collect_block_uniforms(const SpirVModule::Structure &, const std::string &, unsigned, std::vector<std::string> &);
  70         void collect_attributes(const SpirVModule &);
  71         void collect_builtins(const SpirVModule &);
  72         void collect_builtins(const SpirVModule::Structure &);
  73
  74 public:
  75         using ProgramBackend::is_compute;
  76
  77         ReflectData::LayoutHash get_uniform_layout_hash() const { return reflect_data.layout_hash; }
  78         unsigned get_n_descriptor_sets() const { return reflect_data.n_descriptor_sets; }
  79         unsigned get_push_constants_size() const { return reflect_data.push_constants_size; }
  80         const std::vector<ReflectData::UniformBlockInfo> &get_uniform_blocks() const { return reflect_data.uniform_blocks; }
  81         const ReflectData::UniformBlockInfo &get_uniform_block_info(const std::string &) const;
  82         const std::vector<ReflectData::UniformInfo> &get_uniforms() const { return reflect_data.uniforms; }
  83         const ReflectData::UniformInfo &get_uniform_info(const std::string &) const;
  84         const ReflectData::UniformInfo &get_uniform_info(Tag) const;
  85         int get_uniform_location(const std::string &) const;
  86         int get_uniform_location(Tag) const;
  87         int get_uniform_binding(Tag) const;
  88         bool uses_uniform_block_binding(int b) const { return uses_binding(b|ReflectData::UNIFORM_BLOCK_BINDING); }
  89         bool uses_texture_binding(unsigned b) const { return uses_binding(b|ReflectData::TEXTURE_BINDING); }
  90 private:
  91         bool uses_binding(int) const;
  92 public:
  93         const std::vector<ReflectData::AttributeInfo> &get_attributes() const { return reflect_data.attributes; }
  94         const ReflectData::AttributeInfo &get_attribute_info(const std::string &) const;
  95         int get_attribute_location(const std::string &) const;
  96         unsigned get_n_clip_distances() const { return reflect_data.n_clip_distances; }
  97         const LinAl::Vector<unsigned, 3> &get_compute_workgroup_size() const { return reflect_data.compute_wg_size; }
  98
  99         using ProgramBackend::set_debug_name;
 100 };
 101
 102 } // namespace GL
 103 } // namespace Msp
 104
 105 #endif