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 complete shader program.  Programs can be assembled of individual Shaders or
  17 generated with a set of standard features.
  18 */
  19 class Program: public ProgramBackend
  20 {
  21         friend ProgramBackend;
  22
  23 public:
  24         class Loader: public DataFile::CollectionObjectLoader<Program>
  25         {
  26         public:
  27                 Loader(Program &, Collection &);
  28
  29         private:
  30                 void module(const std::string &);
  31         };
  32
  33 private:
  34         class SpecializationLoader: public DataFile::Loader
  35         {
  36         private:
  37                 std::map<std::string, int> &spec_values;
  38
  39                 static ActionMap shared_actions;
  40
  41         public:
  42                 SpecializationLoader(std::map<std::string, int> &);
  43
  44         private:
  45                 virtual void init_actions();
  46
  47                 void specialize_bool(const std::string &, bool);
  48                 void specialize_int(const std::string &, int);
  49         };
  50
  51         ReflectData reflect_data;
  52
  53 public:
  54         /// Constructs an empty Program with no shader stages attached.
  55         Program() = default;
  56
  57         /// Constructs a Program from a Module, with specialization constants.
  58         Program(const Module &, const std::map<std::string, int> & = std::map<std::string, int>());
  59
  60         void add_stages(const Module &, const std::map<std::string, int> & = std::map<std::string, int>());
  61 private:
  62         void collect_uniforms(const SpirVModule &, const std::map<unsigned, int> &);
  63         void collect_block_uniforms(const SpirVModule::Structure &, const std::string &, unsigned, const std::map<unsigned, int> &, std::vector<std::string> &);
  64         void collect_attributes(const SpirVModule &);
  65
  66 public:
  67         ReflectData::LayoutHash get_uniform_layout_hash() const { return reflect_data.layout_hash; }
  68         const std::vector<ReflectData::UniformBlockInfo> &get_uniform_blocks() const { return reflect_data.uniform_blocks; }
  69         const ReflectData::UniformBlockInfo &get_uniform_block_info(const std::string &) const;
  70         const std::vector<ReflectData::UniformInfo> &get_uniforms() const { return reflect_data.uniforms; }
  71         const ReflectData::UniformInfo &get_uniform_info(const std::string &) const;
  72         const ReflectData::UniformInfo &get_uniform_info(Tag) const;
  73         int get_uniform_location(const std::string &) const;
  74         int get_uniform_location(Tag) const;
  75         int get_uniform_binding(Tag) const;
  76         const std::vector<ReflectData::AttributeInfo> &get_attributes() const { return reflect_data.attributes; }
  77         const ReflectData::AttributeInfo &get_attribute_info(const std::string &) const;
  78         int get_attribute_location(const std::string &) const;
  79
  80         using ProgramBackend::set_debug_name;
  81 };
  82
  83 } // namespace GL
  84 } // namespace Msp
  85
  86 #endif