#91 Visualize in graph that Flow Node class is marked as "Deprecated"

MothCocoon · Feb 13, 2022 · d836e91 · d836e91
1 parent 18b064e
commit d836e91
Show file tree

Hide file tree

Showing 5 changed files with 38 additions and 2 deletions.
diff --git a/Source/Flow/Private/Nodes/FlowNode.cpp b/Source/Flow/Private/Nodes/FlowNode.cpp
@@ -25,14 +25,16 @@ UFlowNode::UFlowNode(const FObjectInitializer& ObjectInitializer)
 	: Super(ObjectInitializer)
 #if WITH_EDITOR
 	, GraphNode(nullptr)
+	, bCanDelete(true)
+	, bCanDuplicate(true)
+	, bNodeDeprecated(false)
 #endif
 	, bPreloaded(false)
 	, ActivationState(EFlowNodeState::NeverActivated)
 {
 #if WITH_EDITOR
 	Category = TEXT("Uncategorized");
 	NodeStyle = EFlowNodeStyle::Default;
-	bCanDelete = bCanDuplicate = true;
 #endif
 
 	InputPins = {DefaultInputPin};

diff --git a/Source/Flow/Public/Nodes/FlowNode.h b/Source/Flow/Public/Nodes/FlowNode.h
@@ -28,6 +28,7 @@ class FLOW_API UFlowNode : public UObject, public IVisualLoggerDebugSnapshotInte
 	friend class SFlowGraphNode;
 	friend class UFlowAsset;
 	friend class UFlowGraphNode;
+	friend class UFlowGraphSchema;
 
 //////////////////////////////////////////////////////////////////////////
 // Node
@@ -47,6 +48,13 @@ class FLOW_API UFlowNode : public UObject, public IVisualLoggerDebugSnapshotInte
 	uint8 bCanDelete : 1;
 	uint8 bCanDuplicate : 1;
 
+	UPROPERTY(EditDefaultsOnly, Category = "FlowNode")
+	bool bNodeDeprecated;
+
+	// If this node is deprecated, it might be replaced by another node
+	UPROPERTY(EditDefaultsOnly, Category = "FlowNode")
+	TSubclassOf<UFlowNode> ReplacedBy;
+
 public:
 	FFlowNodeEvent OnReconstructionRequested;
 #endif

diff --git a/Source/FlowEditor/Private/Graph/FlowGraphSchema.cpp b/Source/FlowEditor/Private/Graph/FlowGraphSchema.cpp
@@ -306,7 +306,17 @@ void UFlowGraphSchema::GetCommentAction(FGraphActionMenuBuilder& ActionMenuBuild
 
 bool UFlowGraphSchema::IsFlowNodePlaceable(const UClass* Class)
 {
-	return !Class->HasAnyClassFlags(CLASS_Abstract) && !Class->HasAnyClassFlags(CLASS_NotPlaceable) && !Class->HasAnyClassFlags(CLASS_Deprecated);
+	if (Class->HasAnyClassFlags(CLASS_Abstract) || Class->HasAnyClassFlags(CLASS_NotPlaceable) || Class->HasAnyClassFlags(CLASS_Deprecated))
+	{
+		return false;
+	}
+
+	if (const UFlowNode* DefaultObject = Class->GetDefaultObject<UFlowNode>())
+	{
+		return !DefaultObject->bNodeDeprecated;
+	}
+
+	return true; 
 }
 
 void UFlowGraphSchema::OnBlueprintPreCompile(UBlueprint* Blueprint)

diff --git a/Source/FlowEditor/Private/Graph/Widgets/SFlowGraphNode.cpp b/Source/FlowEditor/Private/Graph/Widgets/SFlowGraphNode.cpp
@@ -351,6 +351,21 @@ void SFlowGraphNode::UpdateGraphNode()
 	CreateAdvancedViewArrow(InnerVerticalBox);
 }
 
+void SFlowGraphNode::UpdateErrorInfo()
+{
+	if (const UFlowNode* FlowNode = FlowGraphNode->GetFlowNode())
+	{
+		if (FlowNode->GetClass()->HasAnyClassFlags(CLASS_Deprecated) || FlowNode->bNodeDeprecated)
+		{
+			ErrorMsg = FlowNode->ReplacedBy ? FString::Printf(TEXT(" REPLACED BY: %s "), *FlowNode->ReplacedBy->GetName()) : FString(TEXT(" DEPRECATED! "));
+			ErrorColor = FEditorStyle::GetColor("ErrorReporting.WarningBackgroundColor");
+			return;
+		}
+	}
+
+	SGraphNode::UpdateErrorInfo();
+}
+
 TSharedRef<SWidget> SFlowGraphNode::CreateNodeContentArea()
 {
 	return SNew(SBorder)

diff --git a/Source/FlowEditor/Public/Graph/Widgets/SFlowGraphNode.h b/Source/FlowEditor/Public/Graph/Widgets/SFlowGraphNode.h
@@ -35,6 +35,7 @@ class SFlowGraphNode : public SGraphNode
 
 	// SGraphNode
 	virtual void UpdateGraphNode() override;
+	virtual void UpdateErrorInfo() override;
 	virtual TSharedRef<SWidget> CreateNodeContentArea() override;
 	virtual const FSlateBrush* GetNodeBodyBrush() const override;