Differential D14517 Diff 49977 source/blender/blenkernel/BKE_geometry_set.hh

Changeset View

Standalone View

source/blender/blenkernel/BKE_geometry_set.hh

Show All 16 Lines
#include "BLI_set.hh"		#include "BLI_set.hh"
#include "BLI_user_counter.hh"		#include "BLI_user_counter.hh"
#include "BLI_vector_set.hh"		#include "BLI_vector_set.hh"

#include "BKE_anonymous_attribute.hh"		#include "BKE_anonymous_attribute.hh"
#include "BKE_attribute_access.hh"		#include "BKE_attribute_access.hh"
#include "BKE_geometry_set.h"		#include "BKE_geometry_set.h"

#include "FN_field.hh"

struct Curves;		struct Curves;
struct Collection;		struct Collection;
struct Curve;		struct Curve;
struct CurveEval;		struct CurveEval;
struct Mesh;		struct Mesh;
struct Object;		struct Object;
struct PointCloud;		struct PointCloud;
struct Volume;		struct Volume;
▲ Show 20 Lines • Show All 979 Lines • ▼ Show 20 Lines	public:
*/		*/
Volume *get_for_write();		Volume *get_for_write();

bool owns_direct_data() const override;		bool owns_direct_data() const override;
void ensure_owns_direct_data() override;		void ensure_owns_direct_data() override;

static constexpr inline GeometryComponentType static_type = GEO_COMPONENT_TYPE_VOLUME;		static constexpr inline GeometryComponentType static_type = GEO_COMPONENT_TYPE_VOLUME;
};		};

namespace blender::bke {

class GeometryComponentFieldContext : public fn::FieldContext {
private:
const GeometryComponent &component_;
const AttributeDomain domain_;

public:
GeometryComponentFieldContext(const GeometryComponent &component, const AttributeDomain domain)
: component_(component), domain_(domain)
{
}

const GeometryComponent &geometry_component() const
{
return component_;
}

AttributeDomain domain() const
{
return domain_;
}
};

class GeometryFieldInput : public fn::FieldInput {
public:
using fn::FieldInput::FieldInput;

GVArray get_varray_for_context(const fn::FieldContext &context,
IndexMask mask,
ResourceScope &scope) const override;

virtual GVArray get_varray_for_context(const GeometryComponent &component,
AttributeDomain domain,
IndexMask mask) const = 0;
};

class AttributeFieldInput : public GeometryFieldInput {
private:
std::string name_;

public:
AttributeFieldInput(std::string name, const CPPType &type)
: GeometryFieldInput(type, name), name_(std::move(name))
{
category_ = Category::NamedAttribute;
}

template<typename T> static fn::Field<T> Create(std::string name)
{
const CPPType &type = CPPType::get<T>();
auto field_input = std::make_shared<AttributeFieldInput>(std::move(name), type);
return fn::Field<T>{field_input};
}

StringRefNull attribute_name() const
{
return name_;
}

GVArray get_varray_for_context(const GeometryComponent &component,
AttributeDomain domain,
IndexMask mask) const override;

std::string socket_inspection_name() const override;

uint64_t hash() const override;
bool is_equal_to(const fn::FieldNode &other) const override;
};

class IDAttributeFieldInput : public GeometryFieldInput {
public:
IDAttributeFieldInput() : GeometryFieldInput(CPPType::get<int>())
{
category_ = Category::Generated;
}

GVArray get_varray_for_context(const GeometryComponent &component,
AttributeDomain domain,
IndexMask mask) const override;

std::string socket_inspection_name() const override;

uint64_t hash() const override;
bool is_equal_to(const fn::FieldNode &other) const override;
};

VArray<float3> curve_normals_varray(const CurveComponent &component, const AttributeDomain domain);

VArray<float3> mesh_normals_varray(const MeshComponent &mesh_component,
const Mesh &mesh,
const IndexMask mask,
const AttributeDomain domain);

class NormalFieldInput : public GeometryFieldInput {
public:
NormalFieldInput() : GeometryFieldInput(CPPType::get<float3>())
{
category_ = Category::Generated;
}

GVArray get_varray_for_context(const GeometryComponent &component,
const AttributeDomain domain,
IndexMask mask) const override;

std::string socket_inspection_name() const override;

uint64_t hash() const override;
bool is_equal_to(const fn::FieldNode &other) const override;
};

class AnonymousAttributeFieldInput : public GeometryFieldInput {
private:
/**
* A strong reference is required to make sure that the referenced attribute is not removed
* automatically.
*/
StrongAnonymousAttributeID anonymous_id_;
std::string producer_name_;

public:
AnonymousAttributeFieldInput(StrongAnonymousAttributeID anonymous_id,
const CPPType &type,
std::string producer_name)
: GeometryFieldInput(type, anonymous_id.debug_name()),
anonymous_id_(std::move(anonymous_id)),
producer_name_(producer_name)
{
category_ = Category::AnonymousAttribute;
}

template<typename T>
static fn::Field<T> Create(StrongAnonymousAttributeID anonymous_id, std::string producer_name)
{
const CPPType &type = CPPType::get<T>();
auto field_input = std::make_shared<AnonymousAttributeFieldInput>(
std::move(anonymous_id), type, std::move(producer_name));
return fn::Field<T>{field_input};
}

GVArray get_varray_for_context(const GeometryComponent &component,
AttributeDomain domain,
IndexMask mask) const override;

std::string socket_inspection_name() const override;

uint64_t hash() const override;
bool is_equal_to(const fn::FieldNode &other) const override;
};

} // namespace blender::bke