ShaderCache.hpp

#pragma once
 
#include <array>
#include <Gfx/Graph/RenderState.hpp>
 
#include <score/gfx/OpenGL.hpp>
#include <score/tools/Debug.hpp>
#include <score/tools/std/StringHash.hpp>
 
#include <ossia/detail/hash_map.hpp>
 
#if __has_include(<QtShaderTools/rhi/qshaderbaker.h>)
#include <QtShaderTools/rhi/qshaderbaker.h>
#else
#include <QtShaderTools/private/qshaderbaker_p.h>
#endif
 
namespace score::gfx
{
 
SCORE_PLUGIN_GFX_EXPORT
QByteArray hlslIntrinsicCollision(const QByteArray& src) noexcept;
 
struct SCORE_PLUGIN_GFX_EXPORT ShaderCache
{
public:
  static const std::pair<QShader, QString>& get(
      const RenderState& v, const QByteArray& shader, QShader::Stage stage,
      int multiViewCount = 0);
  static const std::pair<QShader, QString>& get(
      GraphicsApi api, const QShaderVersion& v, const QByteArray& shader,
      QShader::Stage stage, int multiViewCount = 0);
 
private:
  ShaderCache();
 
  struct Baker
  {
    explicit Baker(GraphicsApi api, const QShaderVersion& v, int multiViewCount);
 
    GraphicsApi api;
    QShaderVersion version;
    int multiViewCount{};
    QShaderBaker baker;
 
    // Keyed by SOURCE, per STAGE. The stage has to be part of the key: the same
    // source text baked as a vertex shader and as a fragment shader would
    // otherwise share one entry, handing back whichever stage was baked first
    // -- with the wrong stage's SPIR-V inside. The baker itself is already per
    // (api, version, multiViewCount).
    //
    // An array rather than a composite key: QShader::Stage is a small dense
    // enum, so this costs one indirection and, unlike appending the stage to
    // the QByteArray, does not copy the whole shader source on every lookup.
    static constexpr int stageCount = 6; // Vertex..Compute, QShader::Stage
    std::array<ossia::hash_map<QByteArray, std::pair<QShader, QString>>, stageCount>
        shaders;
 
    auto& forStage(QShader::Stage s) noexcept
    {
      const int i = int(s);
      SCORE_ASSERT(i >= 0 && i < stageCount);
      return shaders[i];
    }
  };
 
  std::vector<std::unique_ptr<Baker>> m_bakers;
};
}