1 /* 2 * Copyright 2019 Google Inc. 3 * 4 * Use of this source code is governed by a BSD-style license that can be 5 * found in the LICENSE file. 6 */ 7 8 #ifndef GrDawnProgramDataManager_DEFINED 9 #define GrDawnProgramDataManager_DEFINED 10 11 #include "src/gpu/dawn/GrDawnRingBuffer.h" 12 #include "src/gpu/dawn/GrDawnUniformHandler.h" 13 #include "src/gpu/glsl/GrGLSLProgramDataManager.h" 14 #include "dawn/webgpu_cpp.h" 15 16 #include "src/core/SkAutoMalloc.h" 17 18 class GrDawnGpu; 19 class GrDawnUniformBuffer; 20 21 class GrDawnProgramDataManager : public GrGLSLProgramDataManager { 22 public: 23 typedef GrDawnUniformHandler::UniformInfoArray UniformInfoArray; 24 25 GrDawnProgramDataManager(const UniformInfoArray&, uint32_t uniformBufferSize); 26 27 void set1i(UniformHandle, int32_t) const override; 28 void set1iv(UniformHandle, int arrayCount, const int32_t v[]) const override; 29 void set2i(UniformHandle, int32_t, int32_t) const override; 30 void set2iv(UniformHandle, int arrayCount, const int32_t v[]) const override; 31 void set3i(UniformHandle, int32_t, int32_t, int32_t) const override; 32 void set3iv(UniformHandle, int arrayCount, const int32_t v[]) const override; 33 void set4i(UniformHandle, int32_t, int32_t, int32_t, int32_t) const override; 34 void set4iv(UniformHandle, int arrayCount, const int32_t v[]) const override; 35 36 void set1f(UniformHandle, float v0) const override; 37 void set1fv(UniformHandle, int arrayCount, const float v[]) const override; 38 void set2f(UniformHandle, float, float) const override; 39 void set2fv(UniformHandle, int arrayCount, const float v[]) const override; 40 void set3f(UniformHandle, float, float, float) const override; 41 void set3fv(UniformHandle, int arrayCount, const float v[]) const override; 42 void set4f(UniformHandle, float, float, float, float) const override; 43 void set4fv(UniformHandle, int arrayCount, const float v[]) const override; 44 // matrices are column-major, the first two upload a single matrix, the latter two upload 45 // arrayCount matrices into a uniform array. 46 void setMatrix2f(UniformHandle, const float matrix[]) const override; 47 void setMatrix3f(UniformHandle, const float matrix[]) const override; 48 void setMatrix4f(UniformHandle, const float matrix[]) const override; 49 void setMatrix2fv(UniformHandle, int arrayCount, const float matrices[]) const override; 50 void setMatrix3fv(UniformHandle, int arrayCount, const float matrices[]) const override; 51 void setMatrix4fv(UniformHandle, int arrayCount, const float matrices[]) const override; 52 53 // for nvpr only setPathFragmentInputTransform(VaryingHandle u,int components,const SkMatrix & matrix)54 void setPathFragmentInputTransform(VaryingHandle u, int components, 55 const SkMatrix& matrix) const override { 56 SK_ABORT("Only supported in NVPR, which is not in Dawn"); 57 } 58 59 void uploadUniformBuffers(GrDawnGpu* gpu, GrDawnRingBuffer::Slice uniformBuffer) const; 60 uniformBufferSize()61 uint32_t uniformBufferSize() const { return fUniformBufferSize; } 62 private: 63 struct Uniform { 64 uint32_t fOffset; 65 SkDEBUGCODE( 66 GrSLType fType; 67 int fArrayCount; 68 ); 69 }; 70 71 template<int N> inline void setMatrices(UniformHandle, int arrayCount, 72 const float matrices[]) const; 73 74 void* getBufferPtrAndMarkDirty(const Uniform& uni) const; 75 76 uint32_t fUniformBufferSize; 77 78 SkTArray<Uniform, true> fUniforms; 79 80 mutable SkAutoMalloc fUniformData; 81 mutable bool fUniformsDirty; 82 }; 83 84 #endif 85