blob: 8d254d8d1aa87fd6428d0634f1b4008684eea208 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
/* SPDX-License-Identifier: GPL-2.0-or-later */
#pragma once
#include "BKE_geometry_set.hh"
#include "NOD_socket_declarations.hh"
namespace blender::nodes::decl {
class GeometryBuilder;
class Geometry : public SocketDeclaration {
private:
blender::Vector<GeometryComponentType> supported_types_;
bool only_realized_data_ = false;
bool only_instances_ = false;
friend GeometryBuilder;
public:
using Builder = GeometryBuilder;
bNodeSocket &build(bNodeTree &ntree, bNode &node) const override;
bool matches(const bNodeSocket &socket) const override;
bool can_connect(const bNodeSocket &socket) const override;
Span<GeometryComponentType> supported_types() const;
bool only_realized_data() const;
bool only_instances() const;
};
class GeometryBuilder : public SocketDeclarationBuilder<Geometry> {
public:
GeometryBuilder &supported_type(GeometryComponentType supported_type);
GeometryBuilder &supported_type(blender::Vector<GeometryComponentType> supported_types);
GeometryBuilder &only_realized_data(bool value = true);
GeometryBuilder &only_instances(bool value = true);
};
} // namespace blender::nodes::decl
|