mirrors/monado
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/monado/monado.git synced 2025-01-16 11:55:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

comp: Implement new mesh shader

Browse source
This commit is contained in:
Jakob Bornecrantz 2019-10-23 22:00:22 +01:00 committed by Jakob Bornecrantz
parent 9d238d7313
commit 9b6625961a
5 changed files with 85 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/xrt/compositor/common/comp_vk.c
View file

@ -778,7 +778,9 @@ vk_get_device_functions(struct vk_bundle *vk)
vk->vkCmdBindDescriptorSets = GET_DEV_PROC(vk, vkCmdBindDescriptorSets);
vk->vkCmdBindPipeline = GET_DEV_PROC(vk, vkCmdBindPipeline);
vk->vkCmdBindVertexBuffers = GET_DEV_PROC(vk, vkCmdBindVertexBuffers);
vk->vkCmdBindIndexBuffer = GET_DEV_PROC(vk, vkCmdBindIndexBuffer);
vk->vkCmdDraw = GET_DEV_PROC(vk, vkCmdDraw);
vk->vkCmdDrawIndexed = GET_DEV_PROC(vk, vkCmdDrawIndexed);
vk->vkEndCommandBuffer = GET_DEV_PROC(vk, vkEndCommandBuffer);
vk->vkFreeCommandBuffers = GET_DEV_PROC(vk, vkFreeCommandBuffers);
vk->vkCreateRenderPass = GET_DEV_PROC(vk, vkCreateRenderPass);

2
src/xrt/compositor/common/comp_vk.h
View file

@ -132,7 +132,9 @@ struct vk_bundle
PFN_vkCmdBindDescriptorSets vkCmdBindDescriptorSets;
PFN_vkCmdBindPipeline vkCmdBindPipeline;
PFN_vkCmdBindVertexBuffers vkCmdBindVertexBuffers;
PFN_vkCmdBindIndexBuffer vkCmdBindIndexBuffer;
PFN_vkCmdDraw vkCmdDraw;
PFN_vkCmdDrawIndexed vkCmdDrawIndexed;
PFN_vkEndCommandBuffer vkEndCommandBuffer;
PFN_vkFreeCommandBuffers vkFreeCommandBuffers;

84
src/xrt/compositor/main/comp_distortion.c
View file

@ -191,12 +191,24 @@ comp_distortion_init(struct comp_distortion *d,
d->distortion_model = distortion_model;
//! Add support for 1 channels as well.
assert(parts->distortion.mesh.data == NULL ||
assert(parts->distortion.mesh.vertices == NULL ||
parts->distortion.mesh.num_uv_channels == 3);
assert(parts->distortion.mesh.indices == NULL ||
parts->distortion.mesh.total_num_indices != 0);
assert(parts->distortion.mesh.indices == NULL ||
parts->distortion.mesh.num_indices[0] != 0);
assert(parts->distortion.mesh.indices == NULL ||
parts->distortion.mesh.num_indices[1] != 0);
d->vbo_mesh.data = parts->distortion.mesh.data;
d->vbo_mesh.stride = parts->distortion.mesh.stride;
d->vbo_mesh.num = parts->distortion.mesh.num_vertex;
d->mesh.vertices = parts->distortion.mesh.vertices;
d->mesh.stride = parts->distortion.mesh.stride;
d->mesh.num_vertices = parts->distortion.mesh.num_vertices;
d->mesh.indices = parts->distortion.mesh.indices;
d->mesh.total_num_indices = parts->distortion.mesh.total_num_indices;
d->mesh.num_indices[0] = parts->distortion.mesh.num_indices[0];
d->mesh.num_indices[1] = parts->distortion.mesh.num_indices[1];
d->mesh.offset_indices[0] = parts->distortion.mesh.offset_indices[0];
d->mesh.offset_indices[1] = parts->distortion.mesh.offset_indices[1];
d->ubo_vp_data[0].flip_y = flip_y;
d->ubo_vp_data[1].flip_y = flip_y;
@ -220,6 +232,7 @@ comp_distortion_destroy(struct comp_distortion *d)
_buffer_destroy(vk, &d->ubo_handle);
_buffer_destroy(vk, &d->vbo_handle);
_buffer_destroy(vk, &d->index_handle);
_buffer_destroy(vk, &d->ubo_viewport_handles[0]);
_buffer_destroy(vk, &d->ubo_viewport_handles[1]);
@ -235,12 +248,22 @@ comp_distortion_init_pipeline(struct comp_distortion *d,
struct vk_bundle *vk = d->vk;
VkResult ret;
VkPolygonMode polygonMode = VK_POLYGON_MODE_FILL;
if (d->quirk_draw_lines) {
polygonMode = VK_POLYGON_MODE_LINE;
}
VkPrimitiveTopology topology = VK_PRIMITIVE_TOPOLOGY_TRIANGLE_LIST;
if (d->mesh.total_num_indices > 0) {
topology = VK_PRIMITIVE_TOPOLOGY_TRIANGLE_STRIP;
}
VkPipelineInputAssemblyStateCreateInfo input_assembly_state = {
.sType =
VK_STRUCTURE_TYPE_PIPELINE_INPUT_ASSEMBLY_STATE_CREATE_INFO,
.pNext = NULL,
.flags = 0,
.topology = VK_PRIMITIVE_TOPOLOGY_TRIANGLE_LIST,
.topology = topology,
.primitiveRestartEnable = VK_FALSE,
};
@ -250,8 +273,8 @@ comp_distortion_init_pipeline(struct comp_distortion *d,
.flags = 0,
.depthClampEnable = VK_FALSE,
.rasterizerDiscardEnable = VK_FALSE,
.polygonMode = VK_POLYGON_MODE_FILL,
.cullMode = VK_CULL_MODE_NONE, // Hack for now, should only be back.
.polygonMode = polygonMode,
.cullMode = VK_CULL_MODE_BACK_BIT,
.frontFace = VK_FRONT_FACE_CLOCKWISE,
.depthBiasEnable = VK_FALSE,
.depthBiasConstantFactor = 0.f,
@ -398,7 +421,7 @@ comp_distortion_init_pipeline(struct comp_distortion *d,
vertex_input_binding_description.binding = 0;
vertex_input_binding_description.inputRate = VK_VERTEX_INPUT_RATE_VERTEX;
vertex_input_binding_description.stride = d->vbo_mesh.stride;
vertex_input_binding_description.stride = d->mesh.stride;
vertex_input_state.vertexAttributeDescriptionCount = 2;
vertex_input_state.pVertexAttributeDescriptions = vertex_input_attribute_descriptions;
@ -688,7 +711,16 @@ comp_distortion_draw_mesh(struct comp_distortion *d,
vk->vkCmdBindVertexBuffers(command_buffer, 0, 1,
&(d->vbo_handle.buffer), offsets);
vk->vkCmdDraw(command_buffer, d->vbo_mesh.num, 1, 0, 0);
if (d->mesh.total_num_indices > 0) {
vk->vkCmdBindIndexBuffer(command_buffer, d->index_handle.buffer,
0, VK_INDEX_TYPE_UINT32);
vk->vkCmdDrawIndexed(command_buffer, d->mesh.num_indices[eye],
1, d->mesh.offset_indices[eye], 0, 0);
} else {
vk->vkCmdDraw(command_buffer, d->mesh.num_vertices, 1, 0, 0);
}
}
// Update fragment shader hmd warp uniform block
@ -733,12 +765,12 @@ comp_distortion_update_uniform_buffer_warp(struct comp_distortion *d,
d->ubo_pano.aberr[1] = c->xdev->hmd->distortion.pano.aberration_k[1];
d->ubo_pano.aberr[2] = c->xdev->hmd->distortion.pano.aberration_k[2];
d->ubo_pano.aberr[3] = c->xdev->hmd->distortion.pano.aberration_k[3];
d->ubo_pano.lens_center[0][0] = c->xdev->hmd->views[0].lens_center.x_meters;
d->ubo_pano.lens_center[0][1] = c->xdev->hmd->views[0].lens_center.y_meters;
d->ubo_pano.lens_center[1][0] = c->xdev->hmd->views[1].lens_center.x_meters;
d->ubo_pano.lens_center[1][1] = c->xdev->hmd->views[1].lens_center.y_meters;
d->ubo_pano.viewport_scale[0] = c->xdev->hmd->views[0].display.w_meters;
d->ubo_pano.viewport_scale[1] = c->xdev->hmd->views[0].display.h_meters;
d->ubo_pano.lens_center[0][0] = 0.5;//c->xdev->hmd->views[0].lens_center.x_meters;
d->ubo_pano.lens_center[0][1] = 0.5;//c->xdev->hmd->views[0].lens_center.y_meters;
d->ubo_pano.lens_center[1][0] = 0.5;//c->xdev->hmd->views[1].lens_center.x_meters;
d->ubo_pano.lens_center[1][1] = 0.5;//c->xdev->hmd->views[1].lens_center.y_meters;
d->ubo_pano.viewport_scale[0] = 1.0;//c->xdev->hmd->views[0].display.w_meters;
d->ubo_pano.viewport_scale[1] = 1.0;//c->xdev->hmd->views[0].display.h_meters;
d->ubo_pano.warp_scale = c->xdev->hmd->distortion.pano.warp_scale;
memcpy(d->ubo_handle.mapped, &d->ubo_pano, sizeof(d->ubo_pano));
@ -863,18 +895,21 @@ comp_distortion_init_buffers(struct comp_distortion *d,
VkMemoryPropertyFlags memory_property_flags = 0;
VkBufferUsageFlags ubo_usage_flags = 0;
VkBufferUsageFlags vbo_usage_flags = 0;
VkBufferUsageFlags index_usage_flags = 0;
VkResult ret;
// Using the same flags for all ubos and vbos uniform buffers.
ubo_usage_flags |= VK_BUFFER_USAGE_UNIFORM_BUFFER_BIT;
vbo_usage_flags |= VK_BUFFER_USAGE_VERTEX_BUFFER_BIT;
index_usage_flags |= VK_BUFFER_USAGE_INDEX_BUFFER_BIT;
memory_property_flags |= VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT;
memory_property_flags |= VK_MEMORY_PROPERTY_HOST_COHERENT_BIT;
// Distortion ubo and vbo sizes.
VkDeviceSize ubo_size = 0;
VkDeviceSize vbo_size = 0;
VkDeviceSize index_size = 0;
switch (d->distortion_model) {
case XRT_DISTORTION_MODEL_PANOTOOLS:
@ -882,7 +917,8 @@ comp_distortion_init_buffers(struct comp_distortion *d,
break;
case XRT_DISTORTION_MODEL_MESHUV:
ubo_size = sizeof(d->ubo_pano);
vbo_size = d->vbo_mesh.stride * d->vbo_mesh.num;
vbo_size = d->mesh.stride * d->mesh.num_vertices;
index_size = sizeof(int) * d->mesh.total_num_indices;
break;
case XRT_DISTORTION_MODEL_VIVE:
// Vive data
@ -935,7 +971,7 @@ comp_distortion_init_buffers(struct comp_distortion *d,
}
ret = _create_buffer(vk, vbo_usage_flags, memory_property_flags,
&d->vbo_handle, vbo_size, d->vbo_mesh.data);
&d->vbo_handle, vbo_size, d->mesh.vertices);
if (ret != VK_SUCCESS) {
VK_DEBUG(vk, "Failed to create mesh vbo buffer!");
}
@ -943,4 +979,18 @@ comp_distortion_init_buffers(struct comp_distortion *d,
if (ret != VK_SUCCESS) {
VK_DEBUG(vk, "Failed to map mesh vbo buffer!");
}
if (index_size == 0) {
return;
}
ret = _create_buffer(vk, index_usage_flags, memory_property_flags,
&d->index_handle, index_size, d->mesh.indices);
if (ret != VK_SUCCESS) {
VK_DEBUG(vk, "Failed to create mesh index buffer!");
}
ret = _buffer_map(vk, &d->index_handle, index_size, 0);
if (ret != VK_SUCCESS) {
VK_DEBUG(vk, "Failed to map mesh vbo buffer!");
}
}

14
src/xrt/compositor/main/comp_distortion.h
View file

@ -56,6 +56,7 @@ struct comp_distortion
struct comp_uniform_buffer ubo_handle;
struct comp_uniform_buffer vbo_handle;
struct comp_uniform_buffer index_handle;
struct comp_uniform_buffer ubo_viewport_handles[2];
enum xrt_distortion_model distortion_model;
@ -78,13 +79,16 @@ struct comp_distortion
float grow_for_undistort;
} ubo_vive;
// vec2 for pos, vec2 for uv
struct
{
void *data;
float *vertices;
int *indices;
size_t stride;
size_t num;
} vbo_mesh;
size_t num_vertices;
size_t num_indices[2];
size_t offset_indices[2];
size_t total_num_indices;
} mesh;
struct
{
@ -98,6 +102,8 @@ struct comp_distortion
VkDescriptorSetLayout descriptor_set_layout;
VkDescriptorSet descriptor_sets[2];
bool quirk_draw_lines;
};

19
src/xrt/compositor/shaders/mesh.vert
View file

@ -31,20 +31,10 @@ void main()
ubo_vp.rot.zw,
};
vec2 pos = 2.0 * (in_pos_ruv.xy - vec2(0.5, 0.5));
out_ruv = in_pos_ruv.zw;
out_guv = in_guv_buv.xy;
out_buv = in_guv_buv.zw;
// A hack for now.
if (ubo_vp.viewport_id == 1) {
pos.x = -pos.x;
out_ruv.x = 1.0 - out_ruv.x;
out_guv.x = 1.0 - out_guv.x;
out_buv.x = 1.0 - out_buv.x;
}
vec2 pos = rot * in_pos_ruv.xy;
out_ruv = in_pos_ruv.zw;
out_guv = in_guv_buv.xy;
out_buv = in_guv_buv.zw;
if (ubo_vp.flip_y) {
out_ruv.y = 1.0 - out_ruv.y;
@ -52,6 +42,5 @@ void main()
out_buv.y = 1.0 - out_buv.y;
}
pos = rot * pos;
gl_Position = vec4(pos, 0.0f, 1.0f);
}