Fix: Link drag searcher for Compare node trying to use color type in less than operation for color socket #104617
|
@ -93,7 +93,7 @@ static void node_init(bNodeTree * /*tree*/, bNode *node)
|
||||||
|
|
||||||
class SocketSearchOp {
|
class SocketSearchOp {
|
||||||
public:
|
public:
|
||||||
std::string socket_name;
|
const StringRef socket_name;
|
||||||
eNodeSocketDatatype data_type;
|
eNodeSocketDatatype data_type;
|
||||||
NodeCompareOperation operation;
|
NodeCompareOperation operation;
|
||||||
NodeCompareMode mode = NODE_COMPARE_MODE_ELEMENT;
|
NodeCompareMode mode = NODE_COMPARE_MODE_ELEMENT;
|
||||||
|
@ -107,45 +107,85 @@ class SocketSearchOp {
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
static std::optional<eNodeSocketDatatype> fixed_compare_type_for_operation_if_possible(
|
||||||
|
const eNodeSocketDatatype type, const NodeCompareOperation operation)
|
||||||
|
{
|
||||||
|
switch (type) {
|
||||||
|
case SOCK_BOOLEAN: {
|
||||||
|
if (ELEM(operation, NODE_COMPARE_COLOR_BRIGHTER, NODE_COMPARE_COLOR_DARKER)) {
|
||||||
|
return SOCK_RGBA;
|
||||||
|
}
|
||||||
|
return SOCK_INT;
|
||||||
|
}
|
||||||
|
case SOCK_INT:
|
||||||
|
case SOCK_FLOAT:
|
||||||
|
case SOCK_VECTOR: {
|
||||||
|
if (ELEM(operation, NODE_COMPARE_COLOR_BRIGHTER, NODE_COMPARE_COLOR_DARKER)) {
|
||||||
|
return SOCK_RGBA;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case SOCK_RGBA: {
|
||||||
|
if (!ELEM(operation,
|
||||||
|
NODE_COMPARE_COLOR_BRIGHTER,
|
||||||
|
NODE_COMPARE_COLOR_DARKER,
|
||||||
|
NODE_COMPARE_EQUAL,
|
||||||
|
NODE_COMPARE_NOT_EQUAL)) {
|
||||||
|
return SOCK_VECTOR;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case SOCK_STRING: {
|
||||||
|
if (!ELEM(operation, NODE_COMPARE_EQUAL, NODE_COMPARE_NOT_EQUAL)) {
|
||||||
|
return std::nullopt;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
default: {
|
||||||
|
BLI_assert_unreachable();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return type;
|
||||||
|
}
|
||||||
|
|
||||||
static void node_gather_link_searches(GatherLinkSearchOpParams ¶ms)
|
static void node_gather_link_searches(GatherLinkSearchOpParams ¶ms)
|
||||||
{
|
{
|
||||||
const eNodeSocketDatatype type = static_cast<eNodeSocketDatatype>(params.other_socket().type);
|
const eNodeSocketDatatype type = eNodeSocketDatatype(params.other_socket().type);
|
||||||
if (!ELEM(type, SOCK_BOOLEAN, SOCK_FLOAT, SOCK_RGBA, SOCK_VECTOR, SOCK_INT, SOCK_STRING)) {
|
if (!ELEM(type, SOCK_INT, SOCK_BOOLEAN, SOCK_FLOAT, SOCK_VECTOR, SOCK_RGBA, SOCK_STRING)) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
const eNodeSocketDatatype mode_type = (type == SOCK_BOOLEAN) ? SOCK_INT : type;
|
const StringRef socket_name = params.in_out() == SOCK_IN ? "A" : "Result";
|
||||||
const bool string_type = (type == SOCK_STRING);
|
|
||||||
|
|
||||||
const std::string socket_name = params.in_out() == SOCK_IN ? "A" : "Result";
|
|
||||||
|
|
||||||
for (const EnumPropertyItem *item = rna_enum_node_compare_operation_items;
|
for (const EnumPropertyItem *item = rna_enum_node_compare_operation_items;
|
||||||
item->identifier != nullptr;
|
item->identifier != nullptr;
|
||||||
item++) {
|
item++) {
|
||||||
if (item->name != nullptr && item->identifier[0] != '\0') {
|
if (item->name == nullptr || item->identifier[0] == '\0') {
|
||||||
if (!string_type &&
|
continue;
|
||||||
ELEM(item->value, NODE_COMPARE_COLOR_BRIGHTER, NODE_COMPARE_COLOR_DARKER)) {
|
|
||||||
params.add_item(IFACE_(item->name),
|
|
||||||
SocketSearchOp{socket_name,
|
|
||||||
SOCK_RGBA,
|
|
||||||
static_cast<NodeCompareOperation>(item->value)});
|
|
||||||
}
|
|
||||||
else if ((!string_type) ||
|
|
||||||
(string_type && ELEM(item->value, NODE_COMPARE_EQUAL, NODE_COMPARE_NOT_EQUAL))) {
|
|
||||||
params.add_item(IFACE_(item->name),
|
|
||||||
SocketSearchOp{socket_name,
|
|
||||||
mode_type,
|
|
||||||
static_cast<NodeCompareOperation>(item->value)});
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const NodeCompareOperation operation = NodeCompareOperation(item->value);
|
||||||
|
const std::optional<eNodeSocketDatatype> fixed_type =
|
||||||
|
fixed_compare_type_for_operation_if_possible(type, operation);
|
||||||
|
|
||||||
|
if (!fixed_type.has_value()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
params.add_item(IFACE_(item->name), SocketSearchOp{socket_name, *fixed_type, operation});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (params.in_out() != SOCK_IN) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (type == SOCK_STRING) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
/* Add Angle socket. */
|
/* Add Angle socket. */
|
||||||
if (!string_type && params.in_out() == SOCK_IN) {
|
params.add_item(
|
||||||
params.add_item(
|
IFACE_("Angle"),
|
||||||
IFACE_("Angle"),
|
SocketSearchOp{
|
||||||
SocketSearchOp{
|
"Angle", SOCK_VECTOR, NODE_COMPARE_GREATER_THAN, NODE_COMPARE_MODE_DIRECTION});
|
||||||
"Angle", SOCK_VECTOR, NODE_COMPARE_GREATER_THAN, NODE_COMPARE_MODE_DIRECTION});
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static void node_label(const bNodeTree * /*tree*/, const bNode *node, char *label, int maxlen)
|
static void node_label(const bNodeTree * /*tree*/, const bNode *node, char *label, int maxlen)
|
||||||
|
|
Loading…
Reference in New Issue