Index: src/gpu/vk/GrVkProgramDataManager.cpp |
diff --git a/src/gpu/vk/GrVkProgramDataManager.cpp b/src/gpu/vk/GrVkProgramDataManager.cpp |
new file mode 100644 |
index 0000000000000000000000000000000000000000..0ca03ad0b13cc90cf064ef038472b24775348d09 |
--- /dev/null |
+++ b/src/gpu/vk/GrVkProgramDataManager.cpp |
@@ -0,0 +1,315 @@ |
+/* |
+* Copyright 2016 Google Inc. |
+* |
+* Use of this source code is governed by a BSD-style license that can be |
+* found in the LICENSE file. |
+*/ |
+ |
+#include "GrVkProgramDataManager.h" |
+ |
+#include "GrVkGpu.h" |
+#include "GrVkUniformBuffer.h" |
+ |
+GrVkProgramDataManager::GrVkProgramDataManager(const UniformInfoArray& uniforms, |
+ uint32_t vertexUniformSize, |
+ uint32_t fragmentUniformSize) |
+ : fVertexUniformSize(vertexUniformSize) |
+ , fFragmentUniformSize(fragmentUniformSize) { |
+ fVertexUniformData.reset(vertexUniformSize); |
+ fFragmentUniformData.reset(fragmentUniformSize); |
+ int count = uniforms.count(); |
+ fUniforms.push_back_n(count); |
+ // We must add uniforms in same order is the UniformInfoArray so that UniformHandles already |
+ // owned by other objects will still match up here. |
+ for (int i = 0; i < count; i++) { |
+ Uniform& uniform = fUniforms[i]; |
+ const GrVkUniformHandler::UniformInfo uniformInfo = uniforms[i]; |
+ SkASSERT(GrGLSLShaderVar::kNonArray == uniformInfo.fVariable.getArrayCount() || |
+ uniformInfo.fVariable.getArrayCount() > 0); |
+ SkDEBUGCODE( |
+ uniform.fArrayCount = uniformInfo.fVariable.getArrayCount(); |
+ uniform.fType = uniformInfo.fVariable.getType(); |
+ ); |
+ uniform.fBinding = uniformInfo.fBinding; |
+ uniform.fOffset = uniformInfo.fUBOffset; |
+ SkDEBUGCODE( |
+ uniform.fSetNumber = uniformInfo.fSetNumber; |
+ ); |
+ } |
+} |
+ |
+void GrVkProgramDataManager::set1f(UniformHandle u, float v0) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kFloat_GrSLType); |
+ SkASSERT(GrGLSLShaderVar::kNonArray == uni.fArrayCount); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ SkASSERT(sizeof(float) == 4); |
+ memcpy(buffer, &v0, sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::set1fv(UniformHandle u, |
+ int arrayCount, |
+ const float v[]) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kFloat_GrSLType); |
+ SkASSERT(arrayCount > 0); |
+ SkASSERT(arrayCount <= uni.fArrayCount || |
+ (1 == arrayCount && GrGLSLShaderVar::kNonArray == uni.fArrayCount)); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ SkASSERT(sizeof(float) == 4); |
+ memcpy(buffer, v, arrayCount * sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::set2f(UniformHandle u, float v0, float v1) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kVec2f_GrSLType); |
+ SkASSERT(GrGLSLShaderVar::kNonArray == uni.fArrayCount); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ SkASSERT(sizeof(float) == 4); |
+ float v[2] = { v0, v1 }; |
+ memcpy(buffer, v, 2 * sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::set2fv(UniformHandle u, |
+ int arrayCount, |
+ const float v[]) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kVec2f_GrSLType); |
+ SkASSERT(arrayCount > 0); |
+ SkASSERT(arrayCount <= uni.fArrayCount || |
+ (1 == arrayCount && GrGLSLShaderVar::kNonArray == uni.fArrayCount)); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ SkASSERT(sizeof(float) == 4); |
+ memcpy(buffer, v, arrayCount * 2* sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::set3f(UniformHandle u, float v0, float v1, float v2) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kVec3f_GrSLType); |
+ SkASSERT(GrGLSLShaderVar::kNonArray == uni.fArrayCount); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ SkASSERT(sizeof(float) == 4); |
+ float v[3] = { v0, v1, v2 }; |
+ memcpy(buffer, v, 3 * sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::set3fv(UniformHandle u, |
+ int arrayCount, |
+ const float v[]) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kVec3f_GrSLType); |
+ SkASSERT(arrayCount > 0); |
+ SkASSERT(arrayCount <= uni.fArrayCount || |
+ (1 == arrayCount && GrGLSLShaderVar::kNonArray == uni.fArrayCount)); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ SkASSERT(sizeof(float) == 4); |
+ memcpy(buffer, v, arrayCount * 3 * sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::set4f(UniformHandle u, float v0, float v1, float v2, float v3) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kVec4f_GrSLType); |
+ SkASSERT(GrGLSLShaderVar::kNonArray == uni.fArrayCount); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ SkASSERT(sizeof(float) == 4); |
+ float v[4] = { v0, v1, v2, v3 }; |
+ memcpy(buffer, v, 4 * sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::set4fv(UniformHandle u, |
+ int arrayCount, |
+ const float v[]) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kVec4f_GrSLType); |
+ SkASSERT(arrayCount > 0); |
+ SkASSERT(arrayCount <= uni.fArrayCount || |
+ (1 == arrayCount && GrGLSLShaderVar::kNonArray == uni.fArrayCount)); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ SkASSERT(sizeof(float) == 4); |
+ memcpy(buffer, v, arrayCount * 4 * sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::setMatrix3f(UniformHandle u, const float matrix[]) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kMat33f_GrSLType); |
+ SkASSERT(GrGLSLShaderVar::kNonArray == uni.fArrayCount); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ |
+ SkASSERT(sizeof(float) == 4); |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ memcpy(buffer, &matrix[0], 3 * sizeof(float)); |
+ buffer = static_cast<char*>(buffer) + 4*sizeof(float); |
+ memcpy(buffer, &matrix[3], 3 * sizeof(float)); |
+ buffer = static_cast<char*>(buffer) + 4 * sizeof(float); |
+ memcpy(buffer, &matrix[6], 3 * sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::setMatrix3fv(UniformHandle u, |
+ int arrayCount, |
+ const float matrices[]) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kMat33f_GrSLType); |
+ SkASSERT(arrayCount > 0); |
+ SkASSERT(arrayCount <= uni.fArrayCount || |
+ (1 == arrayCount && GrGLSLShaderVar::kNonArray == uni.fArrayCount)); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ SkASSERT(sizeof(float) == 4); |
+ buffer = static_cast<char*>(buffer)+uni.fOffset; |
+ for (int i = 0; i < arrayCount; ++i) { |
+ const float* matrix = &matrices[9 * i]; |
+ memcpy(buffer, &matrix[0], 3 * sizeof(float)); |
+ buffer = static_cast<char*>(buffer)+4 * sizeof(float); |
+ memcpy(buffer, &matrix[3], 3 * sizeof(float)); |
+ buffer = static_cast<char*>(buffer)+4 * sizeof(float); |
+ memcpy(buffer, &matrix[6], 3 * sizeof(float)); |
+ buffer = static_cast<char*>(buffer)+4 * sizeof(float); |
+ } |
+} |
+ |
+ |
+void GrVkProgramDataManager::setMatrix4f(UniformHandle u, const float matrix[]) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kMat44f_GrSLType); |
+ SkASSERT(GrGLSLShaderVar::kNonArray == uni.fArrayCount); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ SkASSERT(sizeof(float) == 4); |
+ memcpy(buffer, matrix, 16 * sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::setMatrix4fv(UniformHandle u, |
+ int arrayCount, |
+ const float matrices[]) const { |
+ const Uniform& uni = fUniforms[u.toIndex()]; |
+ SkASSERT(uni.fType == kMat44f_GrSLType); |
+ SkASSERT(arrayCount > 0); |
+ SkASSERT(arrayCount <= uni.fArrayCount || |
+ (1 == arrayCount && GrGLSLShaderVar::kNonArray == uni.fArrayCount)); |
+ SkASSERT(GrVkUniformHandler::kUniformBufferDescSet == uni.fSetNumber); |
+ |
+ void* buffer; |
+ if (GrVkUniformHandler::kVertexBinding == uni.fBinding) { |
+ buffer = fVertexUniformData.get(); |
+ } else { |
+ SkASSERT(GrVkUniformHandler::kFragBinding == uni.fBinding); |
+ buffer = fFragmentUniformData.get(); |
+ } |
+ buffer = static_cast<char*>(buffer) + uni.fOffset; |
+ SkASSERT(sizeof(float) == 4); |
+ memcpy(buffer, matrices, arrayCount * 16 * sizeof(float)); |
+} |
+ |
+void GrVkProgramDataManager::uploadUniformBuffers(const GrVkGpu* gpu, |
+ GrVkUniformBuffer* vertexBuffer, |
+ GrVkUniformBuffer* fragmentBuffer) const { |
+ if (vertexBuffer) { |
+ vertexBuffer->addMemoryBarrier(gpu, |
+ VK_ACCESS_UNIFORM_READ_BIT, |
+ VK_ACCESS_HOST_WRITE_BIT, |
+ VK_PIPELINE_STAGE_VERTEX_SHADER_BIT, |
+ VK_PIPELINE_STAGE_HOST_BIT, |
+ false); |
+ SkAssertResult(vertexBuffer->updateData(gpu, fVertexUniformData.get(), fVertexUniformSize)); |
+ } |
+ |
+ if (fragmentBuffer) { |
+ fragmentBuffer->addMemoryBarrier(gpu, |
+ VK_ACCESS_UNIFORM_READ_BIT, |
+ VK_ACCESS_HOST_WRITE_BIT, |
+ VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT, |
+ VK_PIPELINE_STAGE_HOST_BIT, |
+ false); |
+ SkAssertResult(fragmentBuffer->updateData(gpu, fFragmentUniformData.get(), |
+ fFragmentUniformSize)); |
+ } |
+} |