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/common/canvas_base.h"
23
24namespace taichi::ui {
25
26namespace vulkan {
27
28class Lines final : public Renderable {
29 public:
30 Lines(AppContext *app_context, VertexAttributes vbo_attrs);
31
32 void update_data(const LinesInfo &info);
33
34 void create_graphics_pipeline() final;
35
36 void record_prepass_this_frame_commands(
37 taichi::lang::CommandList *command_list) override;
38
39 void record_this_frame_commands(
40 taichi::lang::CommandList *command_list) override;
41
42 private:
43 struct UniformBufferObject {
44 alignas(16) glm::vec3 color;
45 float line_width;
46 int per_vertex_color_offset;
47 int vertex_stride;
48 int start_vertex;
49 int start_index;
50 int num_vertices;
51 int is_indexed;
52 float aspect_ratio;
53 };
54
55 uint64_t lines_count_{0};
56
57 taichi::lang::Pipeline *quad_expand_pipeline_{nullptr};
58
59 std::unique_ptr<taichi::lang::DeviceAllocationGuard> vbo_translated_{nullptr};
60 std::unique_ptr<taichi::lang::DeviceAllocationGuard> ibo_translated_{nullptr};
61};
62
63} // namespace vulkan
64
65} // namespace taichi::ui
66