编写完顶点渲染程序，先保存一下
This commit is contained in:
+274
-13
@@ -1,37 +1,298 @@
|
||||
#include "FaceApp.h"
|
||||
#include "common/helpers.h"
|
||||
#include "hardcode_data.h"
|
||||
#include <sstream>
|
||||
|
||||
FaceApp* FaceApp::faceIns = nullptr;
|
||||
|
||||
FaceApp::FaceApp(/* args */)
|
||||
{
|
||||
faceIns = this;
|
||||
}
|
||||
|
||||
FaceApp::~FaceApp()
|
||||
{
|
||||
}
|
||||
|
||||
bool FaceApp::LoadOBJ(const std::string& filename,
|
||||
std::vector<TextureLoadingVertexStructure>& vertices,
|
||||
std::vector<uint32_t>& indices) {
|
||||
|
||||
// 临时存储从OBJ文件读取的原始数据
|
||||
std::vector<float> temp_positions;
|
||||
std::vector<float> temp_texcoords;
|
||||
std::vector<float> temp_normals;
|
||||
|
||||
// 用于处理顶点索引
|
||||
std::vector<int> vertexIndices, uvIndices, normalIndices;
|
||||
|
||||
std::vector<char> data = readFile(filename);
|
||||
|
||||
// 将 vector<char> 转换为以 null 结尾的字符串（安全做法）
|
||||
std::string content(data.begin(), data.end());
|
||||
std::istringstream iss(content); // 用字符串创建字符串流
|
||||
|
||||
std::string line;
|
||||
while (std::getline(iss, line)) {
|
||||
// 跳过空行和注释行
|
||||
if (line.empty() || line[0] == '#') {
|
||||
continue;
|
||||
}
|
||||
|
||||
std::istringstream iss(line);
|
||||
std::string type;
|
||||
iss >> type;
|
||||
|
||||
if (type == "v") { // 顶点位置
|
||||
float x, y, z;
|
||||
iss >> x >> y >> z;
|
||||
temp_positions.push_back(x);
|
||||
temp_positions.push_back(y);
|
||||
temp_positions.push_back(z);
|
||||
}
|
||||
else if (type == "vt") { // 纹理坐标
|
||||
float u, v;
|
||||
iss >> u >> v;
|
||||
temp_texcoords.push_back(u);
|
||||
temp_texcoords.push_back(1 - v);
|
||||
}
|
||||
else if (type == "vn") { // 法线
|
||||
float nx, ny, nz;
|
||||
iss >> nx >> ny >> nz;
|
||||
temp_normals.push_back(nx);
|
||||
temp_normals.push_back(ny);
|
||||
temp_normals.push_back(nz);
|
||||
}
|
||||
else if (type == "f") { // 面（三角形）
|
||||
std::string vertex1, vertex2, vertex3;
|
||||
iss >> vertex1 >> vertex2 >> vertex3;
|
||||
|
||||
// 处理每个顶点的索引
|
||||
for (const std::string& vertex : { vertex1, vertex2, vertex3 }) {
|
||||
std::istringstream viss(vertex);
|
||||
std::string v, vt, vn;
|
||||
|
||||
// 解析顶点索引格式：v/vt/vn 或 v//vn 或 v
|
||||
std::getline(viss, v, '/');
|
||||
std::getline(viss, vt, '/');
|
||||
std::getline(viss, vn, '/');
|
||||
|
||||
int posIndex = std::stoi(v) - 1; // OBJ索引从1开始
|
||||
int texIndex = -1, normIndex = -1;
|
||||
|
||||
if (!vt.empty()) texIndex = std::stoi(vt) - 1;
|
||||
if (!vn.empty()) normIndex = std::stoi(vn) - 1;
|
||||
|
||||
vertexIndices.push_back(posIndex);
|
||||
uvIndices.push_back(texIndex);
|
||||
normalIndices.push_back(normIndex);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// 创建顶点数据
|
||||
vertices.clear();
|
||||
indices.clear();
|
||||
|
||||
// 用于去重的哈希映射
|
||||
std::map<std::string, uint32_t> vertexMap;
|
||||
|
||||
for (size_t i = 0; i < vertexIndices.size(); i++) {
|
||||
int posIndex = vertexIndices[i];
|
||||
int texIndex = uvIndices[i];
|
||||
int normIndex = normalIndices[i];
|
||||
|
||||
// 创建唯一标识符
|
||||
std::string vertexKey = std::to_string(posIndex) + "/" +
|
||||
std::to_string(texIndex) + "/" +
|
||||
std::to_string(normIndex);
|
||||
|
||||
// 检查是否已经存在相同的顶点
|
||||
if (vertexMap.find(vertexKey) != vertexMap.end()) {
|
||||
// 使用现有顶点的索引
|
||||
indices.push_back(vertexMap[vertexKey]);
|
||||
}
|
||||
else {
|
||||
// 创建新顶点
|
||||
TextureLoadingVertexStructure vertex;
|
||||
|
||||
// 设置位置
|
||||
if (posIndex >= 0 && posIndex * 3 + 2 < temp_positions.size()) {
|
||||
vertex.pos[0] = temp_positions[posIndex * 3];
|
||||
vertex.pos[1] = temp_positions[posIndex * 3 + 1];
|
||||
vertex.pos[2] = temp_positions[posIndex * 3 + 2];
|
||||
}
|
||||
else {
|
||||
vertex.pos[0] = vertex.pos[1] = vertex.pos[2] = 0.0f;
|
||||
}
|
||||
|
||||
// 设置纹理坐标
|
||||
if (texIndex >= 0 && texIndex * 2 + 1 < temp_texcoords.size()) {
|
||||
vertex.uv[0] = temp_texcoords[texIndex * 2];
|
||||
vertex.uv[1] = temp_texcoords[texIndex * 2 + 1];
|
||||
}
|
||||
else {
|
||||
vertex.uv[0] = vertex.uv[1] = 0.0f;
|
||||
}
|
||||
|
||||
// 设置法线
|
||||
if (normIndex >= 0 && normIndex * 3 + 2 < temp_normals.size()) {
|
||||
vertex.normal[0] = temp_normals[normIndex * 3];
|
||||
vertex.normal[1] = temp_normals[normIndex * 3 + 1];
|
||||
vertex.normal[2] = temp_normals[normIndex * 3 + 2];
|
||||
}
|
||||
else {
|
||||
vertex.normal[0] = vertex.normal[1] = 0.0f;
|
||||
vertex.normal[2] = 1.0f; // 默认法线
|
||||
}
|
||||
|
||||
// 添加新顶点并记录索引
|
||||
uint32_t newIndex = static_cast<uint32_t>(vertices.size());
|
||||
vertices.push_back(vertex);
|
||||
indices.push_back(newIndex);
|
||||
obj_vertices_map[newIndex] = posIndex;
|
||||
vertexMap[vertexKey] = newIndex;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void FaceApp::render(VkCommandBuffer commandBuffer)
|
||||
{
|
||||
Application::render(commandBuffer);
|
||||
|
||||
}
|
||||
|
||||
void FaceApp::initVulkan()
|
||||
{
|
||||
Application::initVulkan();
|
||||
|
||||
VmaAllocatorCreateInfo allocatorInfo = {};
|
||||
allocatorInfo.physicalDevice = physicalDevice;
|
||||
allocatorInfo.device = device;
|
||||
allocatorInfo.instance = instance;
|
||||
vmaCreateAllocator(&allocatorInfo, &allocator);
|
||||
createVertexBuffer();
|
||||
LoadOBJ("face_picture_3dmax.obj", obj_vertices, obj_indices);
|
||||
uploadVertexData();
|
||||
}
|
||||
|
||||
|
||||
void FaceApp::generate_quad()
|
||||
void FaceApp::createVertexBuffer()
|
||||
{
|
||||
//std::vector<TextureLoadingVertexStructure>& vertices = obj_vertices;
|
||||
//std::vector<uint32_t> indices = obj_indices;
|
||||
VkDeviceSize vertexBufferSize = sizeof(TextureLoadingVertexStructure) * obj_vertices.capacity();
|
||||
VkDeviceSize indexBufferSize = sizeof(uint32_t) * obj_indices.capacity();
|
||||
|
||||
//index_count = static_cast<uint32_t>(indices.size());
|
||||
// 创建顶点缓冲区（设备本地，用于渲染）
|
||||
VkBufferCreateInfo bufferInfo = { VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO };
|
||||
bufferInfo.size = vertexBufferSize;
|
||||
bufferInfo.usage = VK_BUFFER_USAGE_VERTEX_BUFFER_BIT | VK_BUFFER_USAGE_TRANSFER_DST_BIT;
|
||||
bufferInfo.sharingMode = VK_SHARING_MODE_EXCLUSIVE;
|
||||
|
||||
//auto vertex_buffer_size = vkb::to_u32(vertices.size() * sizeof(TextureLoadingVertexStructure));
|
||||
//auto index_buffer_size = vkb::to_u32(indices.size() * sizeof(uint32_t));
|
||||
VmaAllocationCreateInfo allocInfo = {};
|
||||
allocInfo.usage = VMA_MEMORY_USAGE_GPU_ONLY;
|
||||
|
||||
//vkb::core::DeviceC& cDevice = reinterpret_cast<vkb::core::DeviceC&>(*v_device);
|
||||
vmaCreateBuffer(allocator, &bufferInfo, &allocInfo, &m_vertexBuffer, &m_vertexBufferAllocation, nullptr);
|
||||
|
||||
//auto vertex_buffer = std::make_unique<vkb::core::BufferC>(cDevice, vertex_buffer_size,VK_BUFFER_USAGE_TRANSFER_DST_BIT | VK_BUFFER_USAGE_VERTEX_BUFFER_BIT,VMA_MEMORY_USAGE_CPU_TO_GPU);
|
||||
//vertex_buffer->update(vertices.data(), vertex_buffer_size);
|
||||
// 创建暂存缓冲区（CPU可见，用于上传数据）
|
||||
bufferInfo.usage = VK_BUFFER_USAGE_TRANSFER_SRC_BIT;
|
||||
allocInfo.usage = VMA_MEMORY_USAGE_CPU_TO_GPU;
|
||||
|
||||
//index_buffer = std::make_unique<vkb::core::BufferC>(cDevice,index_buffer_size,VK_BUFFER_USAGE_TRANSFER_DST_BIT | VK_BUFFER_USAGE_INDEX_BUFFER_BIT,VMA_MEMORY_USAGE_CPU_TO_GPU);
|
||||
vmaCreateBuffer(allocator, &bufferInfo, &allocInfo, &m_stagingBuffer, &m_stagingBufferAllocation, nullptr);
|
||||
|
||||
//index_buffer->update(indices.data(), index_buffer_size);
|
||||
// 创建索引缓冲区
|
||||
bufferInfo.size = indexBufferSize;
|
||||
bufferInfo.usage = VK_BUFFER_USAGE_INDEX_BUFFER_BIT | VK_BUFFER_USAGE_TRANSFER_DST_BIT;
|
||||
allocInfo.usage = VMA_MEMORY_USAGE_GPU_ONLY;
|
||||
|
||||
vmaCreateBuffer(allocator, &bufferInfo, &allocInfo, &m_indexBuffer, &m_indexBufferAllocation, nullptr);
|
||||
}
|
||||
|
||||
void ReceiveFacePoint(float* pos, int pointCount, int width, int height)
|
||||
{
|
||||
FaceApp* self = FaceApp::Get();
|
||||
if (self != nullptr)
|
||||
{
|
||||
FaceApp::Get()->update_face_vertex_buffer(pos, pointCount);
|
||||
}
|
||||
}
|
||||
|
||||
void FaceApp::update_face_vertex_buffer(float* pos, int pointCount)
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(mtx_point);
|
||||
last_update_time = getCurrentTimeMillis();
|
||||
|
||||
|
||||
for (int i = 0; i < obj_vertices.size(); ++i)
|
||||
{
|
||||
|
||||
int face_index = obj_vertices_map[HardCodeData::Get().indexMap[i]];
|
||||
float x = pos[face_index * 3 + 0];
|
||||
float y = pos[face_index * 3 + 1];
|
||||
float z = pos[face_index * 3 + 2];
|
||||
|
||||
obj_vertices[i].pos[0] = x;
|
||||
obj_vertices[i].pos[1] = y;
|
||||
obj_vertices[i].pos[2] = z;
|
||||
}
|
||||
|
||||
uploadVertexData();
|
||||
}
|
||||
|
||||
void FaceApp::uploadVertexData() {
|
||||
// 上传顶点数据
|
||||
void* data;
|
||||
vmaMapMemory(allocator, m_stagingBufferAllocation, &data);
|
||||
memcpy(data, obj_vertices.data(), sizeof(TextureLoadingVertexStructure) * obj_vertices.size());
|
||||
vmaUnmapMemory(allocator, m_stagingBufferAllocation);
|
||||
|
||||
// 复制到设备内存
|
||||
copyBuffer(m_stagingBuffer, m_vertexBuffer, sizeof(TextureLoadingVertexStructure) * obj_vertices.size());
|
||||
|
||||
// 上传索引数据（如果需要暂存缓冲区，可以创建另一个）
|
||||
vmaMapMemory(allocator, m_stagingBufferAllocation, &data);
|
||||
memcpy(data, obj_indices.data(), sizeof(uint32_t) * obj_indices.size());
|
||||
vmaUnmapMemory(allocator, m_stagingBufferAllocation);
|
||||
|
||||
copyBuffer(m_stagingBuffer, m_indexBuffer, sizeof(uint32_t) * obj_indices.size());
|
||||
}
|
||||
|
||||
void FaceApp::copyBuffer(VkBuffer srcBuffer, VkBuffer dstBuffer, VkDeviceSize size)
|
||||
{
|
||||
VkCommandBuffer commandBuffer = beginSingleTimeCommands();
|
||||
|
||||
VkBufferCopy copyRegion = {};
|
||||
copyRegion.size = size;
|
||||
vkCmdCopyBuffer(commandBuffer, srcBuffer, dstBuffer, 1, ©Region);
|
||||
|
||||
endSingleTimeCommands(commandBuffer);
|
||||
}
|
||||
|
||||
VkCommandBuffer FaceApp::beginSingleTimeCommands() {
|
||||
VkCommandBufferAllocateInfo allocInfo = { VK_STRUCTURE_TYPE_COMMAND_BUFFER_ALLOCATE_INFO };
|
||||
allocInfo.level = VK_COMMAND_BUFFER_LEVEL_PRIMARY;
|
||||
allocInfo.commandPool = commandPool;
|
||||
allocInfo.commandBufferCount = 1;
|
||||
|
||||
VkCommandBuffer commandBuffer;
|
||||
vkAllocateCommandBuffers(device, &allocInfo, &commandBuffer);
|
||||
|
||||
VkCommandBufferBeginInfo beginInfo = { VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO };
|
||||
beginInfo.flags = VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT;
|
||||
|
||||
vkBeginCommandBuffer(commandBuffer, &beginInfo);
|
||||
return commandBuffer;
|
||||
}
|
||||
|
||||
void FaceApp::endSingleTimeCommands(VkCommandBuffer commandBuffer) {
|
||||
vkEndCommandBuffer(commandBuffer);
|
||||
|
||||
VkSubmitInfo submitInfo = { VK_STRUCTURE_TYPE_SUBMIT_INFO };
|
||||
submitInfo.commandBufferCount = 1;
|
||||
submitInfo.pCommandBuffers = &commandBuffer;
|
||||
|
||||
vkQueueSubmit(graphicsQueue, 1, &submitInfo, VK_NULL_HANDLE);
|
||||
vkQueueWaitIdle(graphicsQueue);
|
||||
|
||||
vkFreeCommandBuffers(device, commandPool, 1, &commandBuffer);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user