tweaking #154: changes to add/remove Custom Inputs

- wrap it with WITH_EDITOR - replace check with with simple if-statement, we don't editor to crash on accidentally provided invalid data
MothCocoon · May 11, 2023 · 8b6c610 · 8b6c610
1 parent ba9b57b
commit 8b6c610
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/Source/Flow/Private/FlowAsset.cpp b/Source/Flow/Private/FlowAsset.cpp
@@ -188,19 +188,23 @@ void UFlowAsset::HarvestNodeConnections()
 		}
 	}
 }
-#endif // WITH_EDITOR
 
 void UFlowAsset::AddCustomInput(const FName& InName)
 {
-	check(!CustomInputs.Contains(InName));
-	CustomInputs.Add(InName);
+	if (!CustomInputs.Contains(InName))
+	{
+		CustomInputs.Add(InName);
+	}
 }
 
 void UFlowAsset::RemoveCustomInput(const FName& InName)
 {
-	check(CustomInputs.Contains(InName));
-	CustomInputs.Remove(InName);
+	if (CustomInputs.Contains(InName))
+	{
+		CustomInputs.Remove(InName);
+	}
 }
+#endif // WITH_EDITOR
 
 UFlowNode* UFlowAsset::GetDefaultEntryNode() const
 {

diff --git a/Source/Flow/Public/FlowAsset.h b/Source/Flow/Public/FlowAsset.h
@@ -187,8 +187,10 @@ class FLOW_API UFlowAsset : public UObject
 	const TArray<FName>& GetCustomOutputs() const { return CustomOutputs; }
 
 protected:
+#if WITH_EDITOR
 	void AddCustomInput(const FName& InName);
 	void RemoveCustomInput(const FName& InName);
+#endif
 
 //////////////////////////////////////////////////////////////////////////
 // Instances of the template asset