1 | #pragma once |
---|---|
2 | |
3 | #include <iostream> |
4 | #include <fstream> |
5 | #include <stdexcept> |
6 | #include <algorithm> |
7 | #include <chrono> |
8 | #include <vector> |
9 | #include <cstring> |
10 | #include <cstdlib> |
11 | #include <cstdint> |
12 | #include <array> |
13 | #include <optional> |
14 | #include <set> |
15 | #include "taichi/ui/utils/utils.h" |
16 | #include "taichi/ui/backends/vulkan/vertex.h" |
17 | |
18 | #include "taichi/ui/backends/vulkan/app_context.h" |
19 | #include "taichi/ui/backends/vulkan/swap_chain.h" |
20 | #include "taichi/ui/backends/vulkan/renderable.h" |
21 | #include "taichi/program/field_info.h" |
22 | #include "taichi/ui/backends/vulkan/scene.h" |
23 | |
24 | namespace taichi::ui { |
25 | |
26 | namespace vulkan { |
27 | |
28 | class Mesh final : public Renderable { |
29 | public: |
30 | Mesh(AppContext *app_context, VertexAttributes vbo_attrs); |
31 | |
32 | void update_data(const MeshInfo &info, const Scene &scene); |
33 | |
34 | void record_this_frame_commands( |
35 | taichi::lang::CommandList *command_list) override; |
36 | |
37 | private: |
38 | struct UniformBufferObject { |
39 | Scene::SceneUniformBuffer scene; |
40 | alignas(16) glm::vec3 color; |
41 | int use_per_vertex_color; |
42 | int two_sided; |
43 | float has_attribute; |
44 | }; |
45 | |
46 | int num_instances_{0}; |
47 | int start_instance_{0}; |
48 | |
49 | size_t mesh_ssbo_size_{0}; |
50 | DeviceAllocationUnique mesh_storage_buffer_{nullptr}; |
51 | DeviceAllocationUnique mesh_staging_storage_buffer_{nullptr}; |
52 | |
53 | void resize_mesh_storage_buffers(size_t ssbo_size); |
54 | }; |
55 | |
56 | } // namespace vulkan |
57 | |
58 | } // namespace taichi::ui |
59 |