added Is Input Connected method #96

MothCocoon · Mar 30, 2022 · 6f0c35b · 6f0c35b
1 parent 640e160
commit 6f0c35b
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/Source/Flow/Private/Nodes/FlowNode.cpp b/Source/Flow/Private/Nodes/FlowNode.cpp
@@ -239,6 +239,28 @@ TSet<UFlowNode*> UFlowNode::GetConnectedNodes() const
 	return Result;
 }
 
+bool UFlowNode::IsInputConnected(const FName& PinName) const
+{
+	if (GetFlowAsset())
+	{
+		for (const TPair<FGuid, UFlowNode*>& Pair : GetFlowAsset()->Nodes)
+		{
+			if (Pair.Value)
+			{
+				for (const TPair<FName, FConnectedPin>& Connection : Pair.Value->Connections)
+				{
+					if (Connection.Value.NodeGuid == NodeGuid && Connection.Value.PinName == PinName)
+					{
+						return true;
+					}
+				}
+			}
+		}
+	}
+
+	return false;
+}
+
 bool UFlowNode::IsOutputConnected(const FName& PinName) const
 {
 	return OutputPins.Contains(PinName) && Connections.Contains(PinName);

diff --git a/Source/Flow/Public/Nodes/FlowNode.h b/Source/Flow/Public/Nodes/FlowNode.h
@@ -173,6 +173,9 @@ class FLOW_API UFlowNode : public UObject, public IVisualLoggerDebugSnapshotInte
 	FConnectedPin GetConnection(const FName OutputName) const { return Connections.FindRef(OutputName); }
 	TSet<UFlowNode*> GetConnectedNodes() const;
 
+	UFUNCTION(BlueprintPure, Category= "FlowNode")
+	bool IsInputConnected(const FName& PinName) const;
+
 	UFUNCTION(BlueprintPure, Category= "FlowNode")
 	bool IsOutputConnected(const FName& PinName) const;