341 lines
13 KiB
C++
341 lines
13 KiB
C++
/*
|
|
* Vulkan Example - Animated gears using multiple uniform buffers
|
|
*
|
|
* See readme.md for details
|
|
*
|
|
* Copyright (C) 2016 by Sascha Willems - www.saschawillems.de
|
|
*
|
|
* This code is licensed under the MIT license (MIT) (http://opensource.org/licenses/MIT)
|
|
*/
|
|
|
|
#include "vulkangear.h"
|
|
|
|
int32_t VulkanGear::newVertex(std::vector<Vertex> *vBuffer, float x, float y, float z, const glm::vec3& normal)
|
|
{
|
|
Vertex v(glm::vec3(x, y, z), normal, color);
|
|
vBuffer->push_back(v);
|
|
return vBuffer->size() - 1;
|
|
}
|
|
|
|
void VulkanGear::newFace(std::vector<uint32_t> *iBuffer, int a, int b, int c)
|
|
{
|
|
iBuffer->push_back(a);
|
|
iBuffer->push_back(b);
|
|
iBuffer->push_back(c);
|
|
}
|
|
|
|
VulkanGear::~VulkanGear()
|
|
{
|
|
// Clean up vulkan resources
|
|
vkDestroyBuffer(vulkanDevice->logicalDevice, uniformData.buffer, nullptr);
|
|
vkFreeMemory(vulkanDevice->logicalDevice, uniformData.memory, nullptr);
|
|
|
|
vertexBuffer.destroy();
|
|
indexBuffer.destroy();
|
|
}
|
|
|
|
void VulkanGear::generate(GearInfo *gearinfo, VkQueue queue)
|
|
{
|
|
this->color = gearinfo->color;
|
|
this->pos = gearinfo->pos;
|
|
this->rotOffset = gearinfo->rotOffset;
|
|
this->rotSpeed = gearinfo->rotSpeed;
|
|
|
|
std::vector<Vertex> vBuffer;
|
|
std::vector<uint32_t> iBuffer;
|
|
|
|
int i, j;
|
|
float r0, r1, r2;
|
|
float ta, da;
|
|
float u1, v1, u2, v2, len;
|
|
float cos_ta, cos_ta_1da, cos_ta_2da, cos_ta_3da, cos_ta_4da;
|
|
float sin_ta, sin_ta_1da, sin_ta_2da, sin_ta_3da, sin_ta_4da;
|
|
int32_t ix0, ix1, ix2, ix3, ix4, ix5;
|
|
|
|
r0 = gearinfo->innerRadius;
|
|
r1 = gearinfo->outerRadius - gearinfo->toothDepth / 2.0;
|
|
r2 = gearinfo->outerRadius + gearinfo->toothDepth / 2.0;
|
|
da = 2.0 * M_PI / gearinfo->numTeeth / 4.0;
|
|
|
|
glm::vec3 normal;
|
|
|
|
for (i = 0; i < gearinfo->numTeeth; i++)
|
|
{
|
|
ta = i * 2.0 * M_PI / gearinfo->numTeeth;
|
|
|
|
cos_ta = cos(ta);
|
|
cos_ta_1da = cos(ta + da);
|
|
cos_ta_2da = cos(ta + 2 * da);
|
|
cos_ta_3da = cos(ta + 3 * da);
|
|
cos_ta_4da = cos(ta + 4 * da);
|
|
sin_ta = sin(ta);
|
|
sin_ta_1da = sin(ta + da);
|
|
sin_ta_2da = sin(ta + 2 * da);
|
|
sin_ta_3da = sin(ta + 3 * da);
|
|
sin_ta_4da = sin(ta + 4 * da);
|
|
|
|
u1 = r2 * cos_ta_1da - r1 * cos_ta;
|
|
v1 = r2 * sin_ta_1da - r1 * sin_ta;
|
|
len = sqrt(u1 * u1 + v1 * v1);
|
|
u1 /= len;
|
|
v1 /= len;
|
|
u2 = r1 * cos_ta_3da - r2 * cos_ta_2da;
|
|
v2 = r1 * sin_ta_3da - r2 * sin_ta_2da;
|
|
|
|
// front face
|
|
normal = glm::vec3(0.0, 0.0, 1.0);
|
|
ix0 = newVertex(&vBuffer, r0 * cos_ta, r0 * sin_ta, gearinfo->width * 0.5, normal);
|
|
ix1 = newVertex(&vBuffer, r1 * cos_ta, r1 * sin_ta, gearinfo->width * 0.5, normal);
|
|
ix2 = newVertex(&vBuffer, r0 * cos_ta, r0 * sin_ta, gearinfo->width * 0.5, normal);
|
|
ix3 = newVertex(&vBuffer, r1 * cos_ta_3da, r1 * sin_ta_3da, gearinfo->width * 0.5, normal);
|
|
ix4 = newVertex(&vBuffer, r0 * cos_ta_4da, r0 * sin_ta_4da, gearinfo->width * 0.5, normal);
|
|
ix5 = newVertex(&vBuffer, r1 * cos_ta_4da, r1 * sin_ta_4da, gearinfo->width * 0.5, normal);
|
|
newFace(&iBuffer, ix0, ix1, ix2);
|
|
newFace(&iBuffer, ix1, ix3, ix2);
|
|
newFace(&iBuffer, ix2, ix3, ix4);
|
|
newFace(&iBuffer, ix3, ix5, ix4);
|
|
|
|
// front sides of teeth
|
|
normal = glm::vec3(0.0, 0.0, 1.0);
|
|
ix0 = newVertex(&vBuffer, r1 * cos_ta, r1 * sin_ta, gearinfo->width * 0.5, normal);
|
|
ix1 = newVertex(&vBuffer, r2 * cos_ta_1da, r2 * sin_ta_1da, gearinfo->width * 0.5, normal);
|
|
ix2 = newVertex(&vBuffer, r1 * cos_ta_3da, r1 * sin_ta_3da, gearinfo->width * 0.5, normal);
|
|
ix3 = newVertex(&vBuffer, r2 * cos_ta_2da, r2 * sin_ta_2da, gearinfo->width * 0.5, normal);
|
|
newFace(&iBuffer, ix0, ix1, ix2);
|
|
newFace(&iBuffer, ix1, ix3, ix2);
|
|
|
|
// back face
|
|
normal = glm::vec3(0.0, 0.0, -1.0);
|
|
ix0 = newVertex(&vBuffer, r1 * cos_ta, r1 * sin_ta, -gearinfo->width * 0.5, normal);
|
|
ix1 = newVertex(&vBuffer, r0 * cos_ta, r0 * sin_ta, -gearinfo->width * 0.5, normal);
|
|
ix2 = newVertex(&vBuffer, r1 * cos_ta_3da, r1 * sin_ta_3da, -gearinfo->width * 0.5, normal);
|
|
ix3 = newVertex(&vBuffer, r0 * cos_ta, r0 * sin_ta, -gearinfo->width * 0.5, normal);
|
|
ix4 = newVertex(&vBuffer, r1 * cos_ta_4da, r1 * sin_ta_4da, -gearinfo->width * 0.5, normal);
|
|
ix5 = newVertex(&vBuffer, r0 * cos_ta_4da, r0 * sin_ta_4da, -gearinfo->width * 0.5, normal);
|
|
newFace(&iBuffer, ix0, ix1, ix2);
|
|
newFace(&iBuffer, ix1, ix3, ix2);
|
|
newFace(&iBuffer, ix2, ix3, ix4);
|
|
newFace(&iBuffer, ix3, ix5, ix4);
|
|
|
|
// back sides of teeth
|
|
normal = glm::vec3(0.0, 0.0, -1.0);
|
|
ix0 = newVertex(&vBuffer, r1 * cos_ta_3da, r1 * sin_ta_3da, -gearinfo->width * 0.5, normal);
|
|
ix1 = newVertex(&vBuffer, r2 * cos_ta_2da, r2 * sin_ta_2da, -gearinfo->width * 0.5, normal);
|
|
ix2 = newVertex(&vBuffer, r1 * cos_ta, r1 * sin_ta, -gearinfo->width * 0.5, normal);
|
|
ix3 = newVertex(&vBuffer, r2 * cos_ta_1da, r2 * sin_ta_1da, -gearinfo->width * 0.5, normal);
|
|
newFace(&iBuffer, ix0, ix1, ix2);
|
|
newFace(&iBuffer, ix1, ix3, ix2);
|
|
|
|
// draw outward faces of teeth
|
|
normal = glm::vec3(v1, -u1, 0.0);
|
|
ix0 = newVertex(&vBuffer, r1 * cos_ta, r1 * sin_ta, gearinfo->width * 0.5, normal);
|
|
ix1 = newVertex(&vBuffer, r1 * cos_ta, r1 * sin_ta, -gearinfo->width * 0.5, normal);
|
|
ix2 = newVertex(&vBuffer, r2 * cos_ta_1da, r2 * sin_ta_1da, gearinfo->width * 0.5, normal);
|
|
ix3 = newVertex(&vBuffer, r2 * cos_ta_1da, r2 * sin_ta_1da, -gearinfo->width * 0.5, normal);
|
|
newFace(&iBuffer, ix0, ix1, ix2);
|
|
newFace(&iBuffer, ix1, ix3, ix2);
|
|
|
|
normal = glm::vec3(cos_ta, sin_ta, 0.0);
|
|
ix0 = newVertex(&vBuffer, r2 * cos_ta_1da, r2 * sin_ta_1da, gearinfo->width * 0.5, normal);
|
|
ix1 = newVertex(&vBuffer, r2 * cos_ta_1da, r2 * sin_ta_1da, -gearinfo->width * 0.5, normal);
|
|
ix2 = newVertex(&vBuffer, r2 * cos_ta_2da, r2 * sin_ta_2da, gearinfo->width * 0.5, normal);
|
|
ix3 = newVertex(&vBuffer, r2 * cos_ta_2da, r2 * sin_ta_2da, -gearinfo->width * 0.5, normal);
|
|
newFace(&iBuffer, ix0, ix1, ix2);
|
|
newFace(&iBuffer, ix1, ix3, ix2);
|
|
|
|
normal = glm::vec3(v2, -u2, 0.0);
|
|
ix0 = newVertex(&vBuffer, r2 * cos_ta_2da, r2 * sin_ta_2da, gearinfo->width * 0.5, normal);
|
|
ix1 = newVertex(&vBuffer, r2 * cos_ta_2da, r2 * sin_ta_2da, -gearinfo->width * 0.5, normal);
|
|
ix2 = newVertex(&vBuffer, r1 * cos_ta_3da, r1 * sin_ta_3da, gearinfo->width * 0.5, normal);
|
|
ix3 = newVertex(&vBuffer, r1 * cos_ta_3da, r1 * sin_ta_3da, -gearinfo->width * 0.5, normal);
|
|
newFace(&iBuffer, ix0, ix1, ix2);
|
|
newFace(&iBuffer, ix1, ix3, ix2);
|
|
|
|
normal = glm::vec3(cos_ta, sin_ta, 0.0);
|
|
ix0 = newVertex(&vBuffer, r1 * cos_ta_3da, r1 * sin_ta_3da, gearinfo->width * 0.5, normal);
|
|
ix1 = newVertex(&vBuffer, r1 * cos_ta_3da, r1 * sin_ta_3da, -gearinfo->width * 0.5, normal);
|
|
ix2 = newVertex(&vBuffer, r1 * cos_ta_4da, r1 * sin_ta_4da, gearinfo->width * 0.5, normal);
|
|
ix3 = newVertex(&vBuffer, r1 * cos_ta_4da, r1 * sin_ta_4da, -gearinfo->width * 0.5, normal);
|
|
newFace(&iBuffer, ix0, ix1, ix2);
|
|
newFace(&iBuffer, ix1, ix3, ix2);
|
|
|
|
// draw inside radius cylinder
|
|
ix0 = newVertex(&vBuffer, r0 * cos_ta, r0 * sin_ta, -gearinfo->width * 0.5, glm::vec3(-cos_ta, -sin_ta, 0.0));
|
|
ix1 = newVertex(&vBuffer, r0 * cos_ta, r0 * sin_ta, gearinfo->width * 0.5, glm::vec3(-cos_ta, -sin_ta, 0.0));
|
|
ix2 = newVertex(&vBuffer, r0 * cos_ta_4da, r0 * sin_ta_4da, -gearinfo->width * 0.5, glm::vec3(-cos_ta_4da, -sin_ta_4da, 0.0));
|
|
ix3 = newVertex(&vBuffer, r0 * cos_ta_4da, r0 * sin_ta_4da, gearinfo->width * 0.5, glm::vec3(-cos_ta_4da, -sin_ta_4da, 0.0));
|
|
newFace(&iBuffer, ix0, ix1, ix2);
|
|
newFace(&iBuffer, ix1, ix3, ix2);
|
|
}
|
|
|
|
int vertexBufferSize = vBuffer.size() * sizeof(Vertex);
|
|
int indexBufferSize = iBuffer.size() * sizeof(uint32_t);
|
|
|
|
bool useStaging = true;
|
|
|
|
if (useStaging)
|
|
{
|
|
vk::Buffer vertexStaging, indexStaging;
|
|
|
|
// Create staging buffers
|
|
// Vertex data
|
|
vulkanDevice->createBuffer(
|
|
VK_BUFFER_USAGE_TRANSFER_SRC_BIT,
|
|
VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT,
|
|
&vertexStaging,
|
|
vertexBufferSize,
|
|
vBuffer.data());
|
|
// Index data
|
|
vulkanDevice->createBuffer(
|
|
VK_BUFFER_USAGE_TRANSFER_SRC_BIT,
|
|
VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT,
|
|
&indexStaging,
|
|
indexBufferSize,
|
|
iBuffer.data());
|
|
|
|
// Create device local buffers
|
|
// Vertex buffer
|
|
vulkanDevice->createBuffer(
|
|
VK_BUFFER_USAGE_VERTEX_BUFFER_BIT | VK_BUFFER_USAGE_TRANSFER_DST_BIT,
|
|
VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT,
|
|
&vertexBuffer,
|
|
vertexBufferSize);
|
|
// Index buffer
|
|
vulkanDevice->createBuffer(
|
|
VK_BUFFER_USAGE_INDEX_BUFFER_BIT | VK_BUFFER_USAGE_TRANSFER_DST_BIT,
|
|
VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT,
|
|
&indexBuffer,
|
|
indexBufferSize);
|
|
|
|
// Copy from staging buffers
|
|
VkCommandBuffer copyCmd = vulkanDevice->createCommandBuffer(VK_COMMAND_BUFFER_LEVEL_PRIMARY, true);
|
|
|
|
VkBufferCopy copyRegion = {};
|
|
|
|
copyRegion.size = vertexBufferSize;
|
|
vkCmdCopyBuffer(
|
|
copyCmd,
|
|
vertexStaging.buffer,
|
|
vertexBuffer.buffer,
|
|
1,
|
|
©Region);
|
|
|
|
copyRegion.size = indexBufferSize;
|
|
vkCmdCopyBuffer(
|
|
copyCmd,
|
|
indexStaging.buffer,
|
|
indexBuffer.buffer,
|
|
1,
|
|
©Region);
|
|
|
|
vulkanDevice->flushCommandBuffer(copyCmd, queue, true);
|
|
|
|
vkDestroyBuffer(vulkanDevice->logicalDevice, vertexStaging.buffer, nullptr);
|
|
vkFreeMemory(vulkanDevice->logicalDevice, vertexStaging.memory, nullptr);
|
|
vkDestroyBuffer(vulkanDevice->logicalDevice, indexStaging.buffer, nullptr);
|
|
vkFreeMemory(vulkanDevice->logicalDevice, indexStaging.memory, nullptr);
|
|
}
|
|
else
|
|
{
|
|
// Vertex buffer
|
|
vulkanDevice->createBuffer(
|
|
VK_BUFFER_USAGE_VERTEX_BUFFER_BIT,
|
|
VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT,
|
|
&vertexBuffer,
|
|
vertexBufferSize,
|
|
vBuffer.data());
|
|
// Index buffer
|
|
vulkanDevice->createBuffer(
|
|
VK_BUFFER_USAGE_INDEX_BUFFER_BIT,
|
|
VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT,
|
|
&indexBuffer,
|
|
indexBufferSize,
|
|
iBuffer.data());
|
|
}
|
|
|
|
indexCount = iBuffer.size();
|
|
|
|
prepareUniformBuffer();
|
|
}
|
|
|
|
void VulkanGear::draw(VkCommandBuffer cmdbuffer, VkPipelineLayout pipelineLayout)
|
|
{
|
|
VkDeviceSize offsets[1] = { 0 };
|
|
vkCmdBindDescriptorSets(cmdbuffer, VK_PIPELINE_BIND_POINT_GRAPHICS, pipelineLayout, 0, 1, &descriptorSet, 0, NULL);
|
|
vkCmdBindVertexBuffers(cmdbuffer, 0, 1, &vertexBuffer.buffer, offsets);
|
|
vkCmdBindIndexBuffer(cmdbuffer, indexBuffer.buffer, 0, VK_INDEX_TYPE_UINT32);
|
|
vkCmdDrawIndexed(cmdbuffer, indexCount, 1, 0, 0, 1);
|
|
}
|
|
|
|
void VulkanGear::updateUniformBuffer(glm::mat4 perspective, glm::vec3 rotation, float zoom, float timer)
|
|
{
|
|
ubo.projection = perspective;
|
|
|
|
ubo.view = glm::lookAt(
|
|
glm::vec3(0, 0, -zoom),
|
|
glm::vec3(-1.0, -1.5, 0),
|
|
glm::vec3(0, 1, 0)
|
|
);
|
|
ubo.view = glm::rotate(ubo.view, glm::radians(rotation.x), glm::vec3(1.0f, 0.0f, 0.0f));
|
|
ubo.view = glm::rotate(ubo.view, glm::radians(rotation.y), glm::vec3(0.0f, 1.0f, 0.0f));
|
|
|
|
ubo.model = glm::mat4();
|
|
ubo.model = glm::translate(ubo.model, pos);
|
|
rotation.z = (rotSpeed * timer) + rotOffset;
|
|
ubo.model = glm::rotate(ubo.model, glm::radians(rotation.z), glm::vec3(0.0f, 0.0f, 1.0f));
|
|
|
|
ubo.normal = glm::inverseTranspose(ubo.view * ubo.model);
|
|
|
|
ubo.lightPos = glm::vec3(0.0f, 0.0f, 2.5f);
|
|
ubo.lightPos.x = sin(glm::radians(timer)) * 8.0f;
|
|
ubo.lightPos.z = cos(glm::radians(timer)) * 8.0f;
|
|
|
|
uint8_t *pData;
|
|
VK_CHECK_RESULT(vkMapMemory(vulkanDevice->logicalDevice, uniformData.memory, 0, sizeof(ubo), 0, (void **)&pData));
|
|
memcpy(pData, &ubo, sizeof(ubo));
|
|
vkUnmapMemory(vulkanDevice->logicalDevice, uniformData.memory);
|
|
}
|
|
|
|
void VulkanGear::setupDescriptorSet(VkDescriptorPool pool, VkDescriptorSetLayout descriptorSetLayout)
|
|
{
|
|
VkDescriptorSetAllocateInfo allocInfo =
|
|
vkTools::initializers::descriptorSetAllocateInfo(
|
|
pool,
|
|
&descriptorSetLayout,
|
|
1);
|
|
|
|
VK_CHECK_RESULT(vkAllocateDescriptorSets(vulkanDevice->logicalDevice, &allocInfo, &descriptorSet));
|
|
|
|
// Binding 0 : Vertex shader uniform buffer
|
|
VkWriteDescriptorSet writeDescriptorSet =
|
|
vkTools::initializers::writeDescriptorSet(
|
|
descriptorSet,
|
|
VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER,
|
|
0,
|
|
&uniformData.descriptor);
|
|
|
|
vkUpdateDescriptorSets(vulkanDevice->logicalDevice, 1, &writeDescriptorSet, 0, NULL);
|
|
}
|
|
|
|
void VulkanGear::prepareUniformBuffer()
|
|
{
|
|
// Vertex shader uniform buffer block
|
|
VkMemoryAllocateInfo allocInfo = vkTools::initializers::memoryAllocateInfo();
|
|
VkMemoryRequirements memReqs;
|
|
|
|
VkBufferCreateInfo bufferInfo = vkTools::initializers::bufferCreateInfo(
|
|
VK_BUFFER_USAGE_UNIFORM_BUFFER_BIT,
|
|
sizeof(ubo));
|
|
|
|
VK_CHECK_RESULT(vkCreateBuffer(vulkanDevice->logicalDevice, &bufferInfo, nullptr, &uniformData.buffer));
|
|
vkGetBufferMemoryRequirements(vulkanDevice->logicalDevice, uniformData.buffer, &memReqs);
|
|
allocInfo.allocationSize = memReqs.size;
|
|
allocInfo.memoryTypeIndex = vulkanDevice->getMemoryType(memReqs.memoryTypeBits, VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT | VK_MEMORY_PROPERTY_HOST_COHERENT_BIT);
|
|
VK_CHECK_RESULT(vkAllocateMemory(vulkanDevice->logicalDevice, &allocInfo, nullptr, &uniformData.memory));
|
|
VK_CHECK_RESULT(vkBindBufferMemory(vulkanDevice->logicalDevice, uniformData.buffer, uniformData.memory, 0));
|
|
|
|
uniformData.descriptor.buffer = uniformData.buffer;
|
|
uniformData.descriptor.offset = 0;
|
|
uniformData.descriptor.range = sizeof(ubo);
|
|
uniformData.allocSize = allocInfo.allocationSize;
|
|
}
|