NodeRenderer.hpp

#pragma once
#include <Gfx/Graph/Node.hpp>
 
namespace score::gfx
{
 
class SCORE_PLUGIN_GFX_EXPORT NodeRenderer
{
public:
  explicit NodeRenderer(const Node& node)
      : node{node}
  {
  }
  virtual ~NodeRenderer();
 
  virtual TextureRenderTarget renderTargetForInput(const Port& input) = 0;
  virtual QRhiBuffer* bufferForInput(const Port& input);
  virtual QRhiBuffer* bufferForOutput(const Port& output);
 
  virtual void
  inputAboutToFinish(RenderList& renderer, const Port& p, QRhiResourceUpdateBatch*&);
 
  virtual void init(RenderList& renderer, QRhiResourceUpdateBatch& res) = 0;
  virtual void update(RenderList& renderer, QRhiResourceUpdateBatch& res, Edge* edge)
      = 0;
 
  virtual void runInitialPasses(
      RenderList&, QRhiCommandBuffer& commands, QRhiResourceUpdateBatch*& res,
      Edge& edge);
 
  virtual void runRenderPass(RenderList&, QRhiCommandBuffer& commands, Edge& edge);
 
  virtual void release(RenderList&) = 0;
 
  void checkForChanges()
  {
    materialChanged = node.hasMaterialChanged(materialChangedIndex);
    renderTargetSpecsChanged
        = node.hasRenderTargetChanged(renderTargetSpecsChangedIndex);
  }
 
  void process(int32_t port, const ossia::geometry_spec& v)
  {
    if(this->geometry != v)
    {
      this->geometry = v;
      geometryChanged = true;
    }
    else
    {
      if(this->geometry.meshes)
      {
        for(auto& mesh : this->geometry.meshes->meshes)
        {
          for(auto& buf : mesh.buffers)
          {
            if(buf.dirty)
            {
              geometryChanged = true;
              break;
            }
          }
          if(geometryChanged)
            break;
        }
      }
    }
  }
 
  const Node& node;
 
  ossia::geometry_spec geometry;
 
  int32_t id{-1};
  bool materialChanged{false};
  bool geometryChanged{false};
  bool renderTargetSpecsChanged{false};
 
private:
  int64_t materialChangedIndex{-1};
  int64_t renderTargetSpecsChangedIndex{-1};
};
 
using PassMap = ossia::small_vector<std::pair<Edge*, Pipeline>, 2>;
SCORE_PLUGIN_GFX_EXPORT
void defaultPassesInit(
    PassMap& passes, const std::vector<Edge*>& edges, RenderList& renderer,
    const Mesh& mesh, const QShader& v, const QShader& f, QRhiBuffer* processUBO,
    QRhiBuffer* matUBO, std::span<const Sampler> samplers,
    std::span<QRhiShaderResourceBinding> additionalBindings = {});
 
SCORE_PLUGIN_GFX_EXPORT
void defaultRenderPass(
    RenderList& renderer, const Mesh& mesh, const MeshBuffers& bufs,
    QRhiCommandBuffer& cb, Edge& edge, PassMap& passes);
 
SCORE_PLUGIN_GFX_EXPORT
void quadRenderPass(
    RenderList& renderer, const MeshBuffers& bufs, QRhiCommandBuffer& cb, Edge& edge,
    PassMap& passes);
 
class SCORE_PLUGIN_GFX_EXPORT GenericNodeRenderer : public score::gfx::NodeRenderer
{
public:
  GenericNodeRenderer(const NodeModel& node) noexcept
      : NodeRenderer{node}
  {
  }
 
  TextureRenderTarget renderTargetForInput(const Port& p) override;
  virtual ~GenericNodeRenderer() { }
 
  ossia::small_vector<Sampler, 8> m_samplers;
 
  QShader m_vertexS;
  QShader m_fragmentS;
 
  // Pipeline
  PassMap m_p;
 
  MeshBuffers m_meshbufs;
 
  QRhiBuffer* m_processUBO{};
 
  DefaultShaderMaterial m_material;
 
  const score::gfx::Mesh* m_mesh{};
 
  // Render loop
  void
  defaultMeshInit(RenderList& renderer, const Mesh& mesh, QRhiResourceUpdateBatch& res);
  void processUBOInit(RenderList& renderer);
  void defaultPassesInit(RenderList& renderer, const Mesh& mesh);
  void defaultPassesInit(
      RenderList& renderer, const Mesh& mesh, const QShader& v, const QShader& f,
      std::span<QRhiShaderResourceBinding> additionalBindings = {});
 
  void init(RenderList& renderer, QRhiResourceUpdateBatch& res) override;
 
  void defaultUBOUpdate(RenderList& renderer, QRhiResourceUpdateBatch& res);
  void defaultMeshUpdate(RenderList& renderer, QRhiResourceUpdateBatch& res);
  void update(RenderList& renderer, QRhiResourceUpdateBatch& res, Edge* edge) override;
 
  void defaultRelease(RenderList&);
  void release(RenderList&) override;
 
  void defaultRenderPass(
      RenderList&, const Mesh& mesh, QRhiCommandBuffer& commands, Edge& edge);
 
  void defaultRenderPass(
      RenderList&, const Mesh& mesh, QRhiCommandBuffer& commands, Edge& edge,
      PassMap& passes);
 
  void runRenderPass(RenderList&, QRhiCommandBuffer& commands, Edge& edge) override;
};
 
}