Cleanup: Rename variables, use shorter names

`src` and `dst` are perfectly clear, and avoid repeating unecessary
characters when writing the variables many times, allowing more space
for everything else.
This commit is contained in:
Hans Goudey 2021-06-21 23:02:00 -05:00
parent d086570c7a
commit dc3b7602ee
5 changed files with 54 additions and 55 deletions

View File

@ -196,7 +196,7 @@ class Spline {
* exceed the lifetime of the input data.
*/
virtual blender::fn::GVArrayPtr interpolate_to_evaluated(
const blender::fn::GVArray &source_data) const = 0;
const blender::fn::GVArray &src) const = 0;
blender::fn::GVArrayPtr interpolate_to_evaluated(blender::fn::GSpan data) const;
template<typename T>
blender::fn::GVArray_Typed<T> interpolate_to_evaluated(blender::Span<T> data) const
@ -332,8 +332,7 @@ class BezierSpline final : public Spline {
};
InterpolationData interpolation_data_from_index_factor(const float index_factor) const;
virtual blender::fn::GVArrayPtr interpolate_to_evaluated(
const blender::fn::GVArray &source_data) const override;
virtual blender::fn::GVArrayPtr interpolate_to_evaluated(const blender::fn::GVArray &src) const;
void evaluate_segment(const int index,
const int next_index,
@ -455,13 +454,12 @@ class NURBSpline final : public Spline {
blender::Span<blender::float3> evaluated_positions() const final;
blender::fn::GVArrayPtr interpolate_to_evaluated(
const blender::fn::GVArray &source_data) const final;
blender::fn::GVArrayPtr interpolate_to_evaluated(const blender::fn::GVArray &src) const final;
protected:
void correct_end_tangents() const final;
void calculate_knots() const;
void calculate_basis_cache() const;
blender::Span<BasisCache> calculate_basis_cache() const;
};
/**
@ -505,8 +503,7 @@ class PolySpline final : public Spline {
blender::Span<blender::float3> evaluated_positions() const final;
blender::fn::GVArrayPtr interpolate_to_evaluated(
const blender::fn::GVArray &source_data) const final;
blender::fn::GVArrayPtr interpolate_to_evaluated(const blender::fn::GVArray &src) const final;
protected:
void correct_end_tangents() const final;

View File

@ -25,6 +25,9 @@ using blender::float3;
using blender::IndexRange;
using blender::MutableSpan;
using blender::Span;
using blender::fn::GVArray;
using blender::fn::GVArray_For_ArrayContainer;
using blender::fn::GVArrayPtr;
SplinePtr BezierSpline::copy() const
{
@ -546,44 +549,44 @@ BezierSpline::InterpolationData BezierSpline::interpolation_data_from_index_fact
/* Use a spline argument to avoid adding this to the header. */
template<typename T>
static void interpolate_to_evaluated_impl(const BezierSpline &spline,
const blender::VArray<T> &source_data,
MutableSpan<T> result_data)
const blender::VArray<T> &src,
MutableSpan<T> dst)
{
BLI_assert(src.size() == spline.size());
BLI_assert(dst.size() == spline.evaluated_points_size());
Span<float> mappings = spline.evaluated_mappings();
for (const int i : result_data.index_range()) {
for (const int i : dst.index_range()) {
BezierSpline::InterpolationData interp = spline.interpolation_data_from_index_factor(
mappings[i]);
const T &value = source_data[interp.control_point_index];
const T &next_value = source_data[interp.next_control_point_index];
const T &value = src[interp.control_point_index];
const T &next_value = src[interp.next_control_point_index];
result_data[i] = blender::attribute_math::mix2(interp.factor, value, next_value);
dst[i] = blender::attribute_math::mix2(interp.factor, value, next_value);
}
}
blender::fn::GVArrayPtr BezierSpline::interpolate_to_evaluated(
const blender::fn::GVArray &source_data) const
GVArrayPtr BezierSpline::interpolate_to_evaluated(const GVArray &src) const
{
BLI_assert(source_data.size() == this->size());
BLI_assert(src.size() == this->size());
if (source_data.is_single()) {
return source_data.shallow_copy();
if (src.is_single()) {
return src.shallow_copy();
}
const int eval_size = this->evaluated_points_size();
if (eval_size == 1) {
return source_data.shallow_copy();
return src.shallow_copy();
}
blender::fn::GVArrayPtr new_varray;
blender::attribute_math::convert_to_static_type(source_data.type(), [&](auto dummy) {
GVArrayPtr new_varray;
blender::attribute_math::convert_to_static_type(src.type(), [&](auto dummy) {
using T = decltype(dummy);
if constexpr (!std::is_void_v<blender::attribute_math::DefaultMixer<T>>) {
Array<T> values(eval_size);
interpolate_to_evaluated_impl<T>(*this, source_data.typed<T>(), values);
new_varray = std::make_unique<blender::fn::GVArray_For_ArrayContainer<Array<T>>>(
std::move(values));
interpolate_to_evaluated_impl<T>(*this, src.typed<T>(), values);
new_varray = std::make_unique<GVArray_For_ArrayContainer<Array<T>>>(std::move(values));
}
});

View File

@ -26,7 +26,10 @@ using blender::float3;
using blender::IndexRange;
using blender::MutableSpan;
using blender::Span;
using blender::fn::GVArray;
using blender::fn::GVArray_For_ArrayContainer;
using blender::fn::GVArray_Typed;
using blender::fn::GVArrayPtr;
SplinePtr NURBSpline::copy() const
{
@ -326,15 +329,15 @@ static void calculate_basis_for_point(const float parameter,
basis_cache.start_index = start;
}
void NURBSpline::calculate_basis_cache() const
Span<NURBSpline::BasisCache> NURBSpline::calculate_basis_cache() const
{
if (!basis_cache_dirty_) {
return;
return basis_cache_;
}
std::lock_guard lock{basis_cache_mutex_};
if (!basis_cache_dirty_) {
return;
return basis_cache_;
}
const int points_len = this->size();
@ -371,50 +374,47 @@ void NURBSpline::calculate_basis_cache() const
}
basis_cache_dirty_ = false;
return basis_cache_;
}
template<typename T>
void interpolate_to_evaluated_impl(Span<NURBSpline::BasisCache> weights,
const blender::VArray<T> &source_data,
MutableSpan<T> result_data)
const blender::VArray<T> &src,
MutableSpan<T> dst)
{
const int points_len = source_data.size();
BLI_assert(result_data.size() == weights.size());
blender::attribute_math::DefaultMixer<T> mixer(result_data);
const int size = src.size();
BLI_assert(dst.size() == weights.size());
blender::attribute_math::DefaultMixer<T> mixer(dst);
for (const int i : result_data.index_range()) {
for (const int i : dst.index_range()) {
Span<float> point_weights = weights[i].weights;
const int start_index = weights[i].start_index;
for (const int j : point_weights.index_range()) {
const int point_index = (start_index + j) % points_len;
mixer.mix_in(i, source_data[point_index], point_weights[j]);
const int point_index = (start_index + j) % size;
mixer.mix_in(i, src[point_index], point_weights[j]);
}
}
mixer.finalize();
}
blender::fn::GVArrayPtr NURBSpline::interpolate_to_evaluated(
const blender::fn::GVArray &source_data) const
GVArrayPtr NURBSpline::interpolate_to_evaluated(const GVArray &src) const
{
BLI_assert(source_data.size() == this->size());
BLI_assert(src.size() == this->size());
if (source_data.is_single()) {
return source_data.shallow_copy();
if (src.is_single()) {
return src.shallow_copy();
}
this->calculate_basis_cache();
Span<BasisCache> weights(basis_cache_);
Span<BasisCache> basis_cache = this->calculate_basis_cache();
blender::fn::GVArrayPtr new_varray;
blender::attribute_math::convert_to_static_type(source_data.type(), [&](auto dummy) {
GVArrayPtr new_varray;
blender::attribute_math::convert_to_static_type(src.type(), [&](auto dummy) {
using T = decltype(dummy);
if constexpr (!std::is_void_v<blender::attribute_math::DefaultMixer<T>>) {
Array<T> values(this->evaluated_points_size());
interpolate_to_evaluated_impl<T>(weights, source_data.typed<T>(), values);
new_varray = std::make_unique<blender::fn::GVArray_For_ArrayContainer<Array<T>>>(
std::move(values));
interpolate_to_evaluated_impl<T>(basis_cache, src.typed<T>(), values);
new_varray = std::make_unique<GVArray_For_ArrayContainer<Array<T>>>(std::move(values));
}
});

View File

@ -22,6 +22,8 @@
using blender::float3;
using blender::MutableSpan;
using blender::Span;
using blender::fn::GVArray;
using blender::fn::GVArrayPtr;
SplinePtr PolySpline::copy() const
{
@ -115,10 +117,9 @@ Span<float3> PolySpline::evaluated_positions() const
* the original data. Therefore the lifetime of the returned virtual array must not be longer than
* the source data.
*/
blender::fn::GVArrayPtr PolySpline::interpolate_to_evaluated(
const blender::fn::GVArray &source_data) const
GVArrayPtr PolySpline::interpolate_to_evaluated(const GVArray &src) const
{
BLI_assert(source_data.size() == this->size());
BLI_assert(src.size() == this->size());
return source_data.shallow_copy();
return src.shallow_copy();
}

View File

@ -326,8 +326,6 @@ static void geo_node_curve_to_points_exec(GeoNodeExecParams params)
geometry_set = bke::geometry_set_realize_instances(geometry_set);
SCOPED_TIMER(__func__);
if (!geometry_set.has_curve()) {
params.set_output("Geometry", GeometrySet());
return;