[Bf-blender-cvs] [09009ce062e] temp-geometry-nodes-fields-prototype: automatically change type in attribute node when selecting attribute

[Jacques Lucke]

Commit: 09009ce062e21214d32893ff2858f7bd66887fed
Author: Jacques Lucke
Date:   Fri Jul 30 13:56:40 2021 +0200
Branches: temp-geometry-nodes-fields-prototype
https://developer.blender.org/rB09009ce062e21214d32893ff2858f7bd66887fed

automatically change type in attribute node when selecting attribute

===================================================================

M	source/blender/editors/space_node/node_geometry_attribute_search.cc

===================================================================

diff --git a/source/blender/editors/space_node/node_geometry_attribute_search.cc b/source/blender/editors/space_node/node_geometry_attribute_search.cc
index be6c7a05981..afbc0430be8 100644
--- a/source/blender/editors/space_node/node_geometry_attribute_search.cc
+++ b/source/blender/editors/space_node/node_geometry_attribute_search.cc
@@ -176,6 +176,35 @@ static void attribute_search_exec_fn(bContext *C, void *data_v, void *item_v)
   bNodeSocketValueString *value = static_cast<bNodeSocketValueString *>(socket.default_value);
   BLI_strncpy(value->value, item->name.c_str(), MAX_NAME);
 
+  if (data->node->type == GEO_NODE_ATTRIBUTE) {
+    NodeGeometryAttribute *storage = (NodeGeometryAttribute *)data->node->storage;
+    switch (item->data_type) {
+      case CD_PROP_FLOAT: {
+        storage->output_type = SOCK_FLOAT;
+        break;
+      }
+      case CD_PROP_INT32: {
+        storage->output_type = SOCK_INT;
+        break;
+      }
+      case CD_PROP_FLOAT2:
+      case CD_PROP_FLOAT3: {
+        storage->output_type = SOCK_VECTOR;
+        break;
+      }
+      case CD_PROP_BOOL: {
+        storage->output_type = SOCK_BOOLEAN;
+        break;
+      }
+      case CD_PROP_COLOR: {
+        storage->output_type = SOCK_RGBA;
+        break;
+      }
+    }
+    snode_update(CTX_wm_space_node(C), (bNode *)data->node);
+    ntreeUpdateTree(CTX_data_main(C), (bNodeTree *)data->tree);
+  }
+
   ED_undo_push(C, "Assign Attribute Name");
 }