mitkVtkShaderRepository.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef _MITKVTKSHADERREPOSITORY_H_
#define _MITKVTKSHADERREPOSITORY_H_

#include "mitkIShaderRepository.h"
#include <vtkShaderProgram2.h>
#include <vtkXMLShader.h>

class vtkXMLDataElement;
class vtkXMLMaterial;
class vtkProperty;

namespace mitk
{
  class VtkShaderRepository : public IShaderRepository
  {
  protected:
    class Shader : public IShaderRepository::Shader
    {
    public:
      mitkClassMacro(Shader, IShaderRepository::Shader) itkFactorylessNewMacro(Self)

        class Uniform : public itk::Object
      {
      public:
        mitkClassMacroItkParent(Uniform, itk::Object) itkFactorylessNewMacro(Self)

          enum Type {
            glsl_none,
            glsl_float,
            glsl_vec2,
            glsl_vec3,
            glsl_vec4,
            glsl_int,
            glsl_ivec2,
            glsl_ivec3,
            glsl_ivec4
          };

        Uniform();

        ~Uniform();

        Type type;
        std::string name;

        int defaultInt[4];
        float defaultFloat[4];

        void LoadFromXML(vtkXMLDataElement *e);
      };

      std::list<Uniform::Pointer> uniforms;

      Shader();

      ~Shader();

      void SetVertexShaderCode(const std::string &code);
      std::string GetVertexShaderCode() const;

      void SetFragmentShaderCode(const std::string &code);
      std::string GetFragmentShaderCode() const;

      void SetGeometryShaderCode(const std::string &code);
      std::string GetGeometryShaderCode() const;

      std::list<Uniform::Pointer> GetUniforms() const;

    private:
      friend class VtkShaderRepository;

      std::string m_VertexShaderCode;
      std::string m_FragmentShaderCode;
      std::string m_GeometryShaderCode;

      void LoadXmlShader(std::istream &stream);
    };

    void LoadShaders();

    Shader::Pointer GetShaderImpl(const std::string &name) const;

  private:
    std::list<Shader::Pointer> shaders;

    static int shaderId;
    static const bool debug;

  public:
    VtkShaderRepository();

    ~VtkShaderRepository();

    ShaderProgram::Pointer CreateShaderProgram() override;

    std::list<IShaderRepository::Shader::Pointer> GetShaders() const override;

    IShaderRepository::Shader::Pointer GetShader(const std::string &name) const override;

    IShaderRepository::Shader::Pointer GetShader(int id) const override;

    void AddDefaultProperties(mitk::DataNode *node, mitk::BaseRenderer *renderer, bool overwrite) const override;

    int LoadShader(std::istream &stream, const std::string &name) override;

    bool UnloadShader(int id) override;

    void UpdateShaderProgram(mitk::IShaderRepository::ShaderProgram *shaderProgram,
                             DataNode *node,
                             BaseRenderer *renderer) const override;
  };

} // end of namespace mitk
#endif