Update

2025-04-17 23:59:17 +08:00
commit 88536f22da
57 changed files with 8094 additions and 0 deletions
--- a/Source/FLESHEditor/Public/DismembermentGraph/DismembermentGraphSchema.h
+++ b/Source/FLESHEditor/Public/DismembermentGraph/DismembermentGraphSchema.h
@@ -0,0 +1,168 @@
+#pragma once
+
+#include "CoreMinimal.h"
+#include "EdGraph/EdGraphSchema.h"
+#include "DismembermentGraphSchema.generated.h"
+
+class UDismembermentGraphNode;
+
+/**
+ * Connection response
+ */
+USTRUCT()
+struct FDismembermentGraphConnectionResponse
+{
+    GENERATED_BODY()
+
+    // Response type
+    UPROPERTY()
+    int32 Response;
+
+    // Response text
+    UPROPERTY()
+    FText Message;
+
+    // Pin names that would be broken
+    UPROPERTY()
+    TArray<FText> BreakingPins;
+
+    // Constructor
+    FDismembermentGraphConnectionResponse()
+        : Response(0)
+    {
+    }
+};
+
+/**
+ * Connection response types
+ */
+struct FDismembermentGraphConnectionResponse_K2
+{
+    enum Type
+    {
+        // No error
+        OK = 0,
+        // Generic error
+        ERROR_INCOMPATIBLE = 1,
+        // Disallowed pin connection
+        ERROR_DISALLOWED = 2,
+        // Self-connection not allowed
+        ERROR_SELF_CONNECTION = 3,
+        // Cycle not allowed
+        ERROR_CYCLE = 4
+    };
+};
+
+/**
+ * Pin type
+ */
+USTRUCT()
+struct FDismembermentGraphPinType
+{
+    GENERATED_BODY()
+
+    // Pin category
+    UPROPERTY()
+    FName PinCategory;
+
+    // Constructor
+    FDismembermentGraphPinType()
+    {
+    }
+
+    // Constructor with category
+    FDismembermentGraphPinType(const FName& InPinCategory)
+        : PinCategory(InPinCategory)
+    {
+    }
+
+    // Equality operator
+    bool operator==(const FDismembermentGraphPinType& Other) const
+    {
+        return PinCategory == Other.PinCategory;
+    }
+
+    // Inequality operator
+    bool operator!=(const FDismembermentGraphPinType& Other) const
+    {
+        return !(*this == Other);
+    }
+};
+
+/**
+ * Dismemberment graph schema
+ */
+UCLASS()
+class FLESHEDITOR_API UDismembermentGraphSchema : public UEdGraphSchema
+{
+    GENERATED_BODY()
+
+public:
+    // Pin categories
+    static const FName PC_Exec;
+    static const FName PC_Bone;
+    static const FName PC_Cut;
+    static const FName PC_Blood;
+    static const FName PC_Physics;
+    static const FName PC_Organ;
+    static const FName PC_Wound;
+
+    // UEdGraphSchema interface
+    virtual void GetGraphContextActions(FGraphContextMenuBuilder& ContextMenuBuilder) const override;
+    virtual void GetContextMenuActions(UToolMenu* Menu, UGraphNodeContextMenuContext* Context) const override;
+    virtual const FPinConnectionResponse CanCreateConnection(const UEdGraphPin* A, const UEdGraphPin* B) const override;
+    virtual bool TryCreateConnection(UEdGraphPin* A, UEdGraphPin* B) const override;
+    virtual bool ShouldHidePinDefaultValue(UEdGraphPin* Pin) const override;
+    virtual FLinearColor GetPinTypeColor(const FEdGraphPinType& PinType) const override;
+    virtual void BreakNodeLinks(UEdGraphNode& TargetNode) const override;
+    virtual void BreakPinLinks(UEdGraphPin& TargetPin, bool bSendsNodeNotification) const override;
+    virtual void BreakSinglePinLink(UEdGraphPin* SourcePin, UEdGraphPin* TargetPin) const override;
+    virtual void DroppedAssetsOnGraph(const TArray<FAssetData>& Assets, const FVector2D& GraphPosition, UEdGraph* Graph) const override;
+    virtual void DroppedAssetsOnNode(const TArray<FAssetData>& Assets, const FVector2D& GraphPosition, UEdGraphNode* Node) const override;
+    virtual void DroppedAssetsOnPin(const TArray<FAssetData>& Assets, const FVector2D& GraphPosition, UEdGraphPin* Pin) const override;
+    virtual void GetAssetsNodeHoverMessage(const TArray<FAssetData>& Assets, const UEdGraphNode* HoverNode, FString& OutTooltipText, bool& OutOkIcon) const override;
+    virtual void GetAssetsPinHoverMessage(const TArray<FAssetData>& Assets, const UEdGraphPin* HoverPin, FString& OutTooltipText, bool& OutOkIcon) const override;
+    // End of UEdGraphSchema interface
+
+    /**
+     * Check if two pins can be connected
+     * @param PinA - First pin
+     * @param PinB - Second pin
+     * @param OutResponse - Connection response
+     * @return True if the pins can be connected
+     */
+    bool CanConnectPins(const UEdGraphPin* PinA, const UEdGraphPin* PinB, FDismembermentGraphConnectionResponse& OutResponse) const;
+
+    /**
+     * Check if connecting two pins would create a cycle
+     * @param PinA - First pin
+     * @param PinB - Second pin
+     * @return True if connecting the pins would create a cycle
+     */
+    bool WouldCreateCycle(const UEdGraphPin* PinA, const UEdGraphPin* PinB) const;
+
+    /**
+     * Get the pin type from a pin
+     * @param Pin - The pin to get the type from
+     * @return The pin type
+     */
+    static FDismembermentGraphPinType GetPinType(const UEdGraphPin* Pin);
+
+    /**
+     * Get the pin type color
+     * @param PinType - The pin type
+     * @return The pin color
+     */
+    static FLinearColor GetPinTypeColor(const FDismembermentGraphPinType& PinType);
+
+    /**
+     * Create a new node
+     * @param NodeClass - Class of the node to create
+     * @param ParentGraph - Graph to create the node in
+     * @param NodePosX - X position of the node
+     * @param NodePosY - Y position of the node
+     * @param bSelectNewNode - Whether to select the new node
+     * @return The created node
+     */
+    static UDismembermentGraphNode* CreateNode(TSubclassOf<UDismembermentGraphNode> NodeClass, UEdGraph* ParentGraph, float NodePosX, float NodePosY, bool bSelectNewNode = true);
+};