SoShaderBuilder.h

Go to the documentation of this file.

/*************************************************************************************
**
** Copyright 2011, MeVis Medical Solutions AG
**
** The user may use this file in accordance with the license agreement provided with
** the Software or, alternatively, in accordance with the terms contained in a
** written agreement between the user and MeVis Medical Solutions AG.
**
** For further information use the contact form at https://www.mevislab.de/contact
**
**************************************************************************************/
 
#ifndef SO_SHADER_BUILDER_H
#define SO_SHADER_BUILDER_H
 
#include "SoShaderSystem.h"
 
#include <string>
#include <map>
#include <sstream>
#include <set>
 
typedef std::vector<std::string> StringVector;
 
class INVENTOR_SHADER_API SoShaderRepository
{
public:
  struct SourceInfo {
 
    bool isValid() const { return code.size()>0; }
 
    std::string  version;
    std::string  code;
    StringVector includes;
  };
 
  const SourceInfo& lookup(const std::string name) const;
 
  void insert(const std::string name, const SourceInfo& info);
 
  void addSource(const char* data);
 
private:
  typedef std::map<std::string, SourceInfo> SourceInfoMap;
 
  SourceInfoMap _repository;
  SourceInfo    _emptyInfo;
};
 
class SoGLCompleteShader;
 
class INVENTOR_SHADER_API SoShaderBuilder
{
public:
  enum ShaderType {
    FRAGMENT_SHADER = 1,
    VERTEX_SHADER = 2,
    GEOMETRY_SHADER = 4
  };
 
  SoShaderBuilder();
 
  void addSource(const char* data);
 
  void addRepository(SoShaderRepository* repository);
 
  std::string createShader(const std::string name, const std::vector<std::string>& directives = std::vector<std::string>(), SoShaderRepository* repository = nullptr);
 
  SoGLCompleteShader* createCompleteShader(const std::string name, const std::vector<std::string>& directives = std::vector<std::string>(),
    SoShaderRepository* repository = nullptr, int directivesFlag = FRAGMENT_SHADER | GEOMETRY_SHADER | VERTEX_SHADER);
 
private:
 
  struct ShaderString {
 
    std::ostringstream directives;
    std::ostringstream header;
    std::ostringstream body;
 
    void addDirectived(const std::vector<std::string>& directives);
 
    std::string toString();
 
    std::set<std::string> usedIncludes;
  };
 
  const SoShaderRepository::SourceInfo& lookup(const std::string name) const;
 
  void addCode(const SoShaderRepository::SourceInfo& code, ShaderString& result);
 
  std::vector<SoShaderRepository*> _repositories;
  SoShaderRepository _defaultRepository;
 
  SoShaderRepository::SourceInfo _emptySource;
};
 
#endif