shaderprogram.h

Go to the documentation of this file.

#ifndef ION_GFX_SHADERPROGRAM_H_
#define ION_GFX_SHADERPROGRAM_H_

#include "ion/base/referent.h"
#include "ion/gfx/resourceholder.h"
#include "ion/gfx/shader.h"

namespace ion {
namespace gfx {

class ShaderInputRegistry;
typedef base::ReferentPtr<ShaderInputRegistry>::Type ShaderInputRegistryPtr;

class ShaderProgram;
typedef base::ReferentPtr<ShaderProgram>::Type ShaderProgramPtr;
typedef base::WeakReferentPtr<ShaderProgram> ShaderProgramWeakPtr;

class ION_API ShaderProgram : public ShaderBase {
 public:
  enum Changes {
    kVertexShaderChanged = kNumBaseChanges,
    kFragmentShaderChanged,
    kNumChanges
  };

  explicit ShaderProgram(const ShaderInputRegistryPtr& registry);

  const ShaderInputRegistryPtr& GetRegistry() const { return registry_; }

  void SetVertexShader(const ShaderPtr& shader) {
    if (Shader* old_shader = vertex_shader_.Get().Get())
      old_shader->RemoveReceiver(this);
    vertex_shader_.Set(shader);
    if (shader.Get())
      shader->AddReceiver(this);
  }
  const ShaderPtr& GetVertexShader() const {
    return vertex_shader_.Get();
  }

  void SetFragmentShader(const ShaderPtr& shader) {
    if (Shader* old_shader = fragment_shader_.Get().Get())
      old_shader->RemoveReceiver(this);
    fragment_shader_.Set(shader);
    if (shader.Get())
      shader->AddReceiver(this);
  }
  const ShaderPtr& GetFragmentShader() const {
    return fragment_shader_.Get();
  }

  void SetConcurrent(bool value);
  bool IsConcurrent() const { return concurrent_; }

  static const ShaderProgramPtr BuildFromStrings(
    const std::string& id_string,
    const ShaderInputRegistryPtr& registry_ptr,
    const std::string& vertex_shader_string,
    const std::string& fragment_shader_string,
    const base::AllocatorPtr& allocator);

 protected:
  ~ShaderProgram() override;

 private:
  void OnNotify(const base::Notifier* notifier) override;

  Field<ShaderPtr> vertex_shader_;
  Field<ShaderPtr> fragment_shader_;
  ShaderInputRegistryPtr registry_;
  bool concurrent_;
  bool concurrent_set_;
};

}  // namespace gfx
}  // namespace ion

#endif  // ION_GFX_SHADERPROGRAM_H_