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
24namespace taichi::ui {
25
26namespace vulkan {
27
28class 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