source/blender/modifiers/intern/MOD_functionpoints_cxx.cc


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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

#include "DNA_mesh_types.h"
#include "DNA_meshdata_types.h"
#include "DNA_modifier_types.h"

#include "BKE_mesh.h"
#include "BKE_modifier.h"

#include "BLI_math.h"

#include "FN_vtree_multi_function_network_generation.h"
#include "FN_multi_functions.h"
#include "FN_multi_function_common_context_ids.h"

#include "DEG_depsgraph.h"
#include "DEG_depsgraph_query.h"

using BKE::VirtualNodeTree;
using BKE::VNode;
using BLI::ArrayRef;
using BLI::float3;
using BLI::IndexRange;
using BLI::TemporaryVector;
using BLI::Vector;
using FN::MFContext;
using FN::MFInputSocket;
using FN::MFOutputSocket;
using FN::MFParamsBuilder;

extern "C" {
Mesh *MOD_functionpoints_do(FunctionPointsModifierData *fpmd,
                            const struct ModifierEvalContext *ctx);
}

Mesh *MOD_functionpoints_do(FunctionPointsModifierData *fpmd,
                            const struct ModifierEvalContext *ctx)
{
  if (fpmd->function_tree == nullptr) {
    return BKE_mesh_new_nomain(0, 0, 0, 0, 0);
  }

  bNodeTree *btree = (bNodeTree *)DEG_get_original_id((ID *)fpmd->function_tree);

  BKE::VirtualNodeTreeBuilder vtree_builder;
  vtree_builder.add_all_of_node_tree(btree);
  auto vtree = vtree_builder.build();

  BLI::ResourceCollector resources;
  auto function = FN::generate_vtree_multi_function(*vtree, resources);

  MFParamsBuilder params(*function, 1);
  params.add_readonly_single_input(&fpmd->control1);
  params.add_readonly_single_input(&fpmd->control2);

  FN::GenericVectorArray vector_array{FN::GET_TYPE<float3>(), 1};
  params.add_vector_output(vector_array);

  float current_frame = DEG_get_ctime(ctx->depsgraph);

  FN::MFContextBuilder context_builder;
  context_builder.add(FN::ContextIDs::current_frame, (void *)&current_frame);
  function->call(FN::MFMask({0}), params.build(), context_builder.build());

  ArrayRef<float3> output_points = vector_array[0].as_typed_ref<float3>();

  Mesh *mesh = BKE_mesh_new_nomain(output_points.size(), 0, 0, 0, 0);
  for (uint i = 0; i < output_points.size(); i++) {
    copy_v3_v3(mesh->mvert[i].co, output_points[i]);
  }

  return mesh;
}