111 lines
3.1 KiB
C++
111 lines
3.1 KiB
C++
#pragma once
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "WingServer.h"
|
|
#include "WingBasics.h"
|
|
#include "WingFetcher.h"
|
|
#include "WingProperty.h"
|
|
#include "WingUtils.h"
|
|
#include "EdGraph/EdGraph.h"
|
|
#include "EdGraph/EdGraphPin.h"
|
|
#include "GraphPin_Disconnect.generated.h"
|
|
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// ---------------------------------------------------------------------------
|
|
// ---------------------------------------------------------------------------
|
|
|
|
USTRUCT()
|
|
struct FDisconnectPinEntry
|
|
{
|
|
GENERATED_BODY()
|
|
|
|
UPROPERTY()
|
|
FString Pin;
|
|
|
|
UPROPERTY(EditAnywhere, meta=(Optional))
|
|
FString TargetPin;
|
|
};
|
|
|
|
|
|
UCLASS()
|
|
class UWing_GraphPin_Disconnect : public UWingHandler
|
|
{
|
|
GENERATED_BODY()
|
|
|
|
public:
|
|
UPROPERTY(EditAnywhere, meta=(Description="Target graph"))
|
|
FString Graph;
|
|
|
|
UPROPERTY(EditAnywhere, meta=(Description="Array of {pin, targetPin?} objects. If targetPin is omitted, all connections on the pin are broken."))
|
|
FWingJsonArray Disconnections;
|
|
|
|
virtual void Register() override
|
|
{
|
|
UWingServer::AddHandler(this,
|
|
TEXT("Disconnect pins in a graph (Blueprint or Material). "
|
|
"Can disconnect a specific link or all links on a pin."));
|
|
}
|
|
virtual void Handle() override
|
|
{
|
|
WingFetcher F(WingOut::Stdout);
|
|
UEdGraph* G = F.Walk(Graph).Cast<UEdGraph>();
|
|
if (!G) return;
|
|
|
|
int32 SuccessCount = 0;
|
|
int32 TotalDisconnected = 0;
|
|
|
|
FDisconnectPinEntry Entry;
|
|
TArray<FWingProperty> EntryProps = FWingProperty::GetAll(&Entry);
|
|
for (const TSharedPtr<FJsonValue>& DiscVal : Disconnections.Array)
|
|
{
|
|
if (!FWingProperty::PopulateFromJson(EntryProps, *DiscVal, false, WingOut::Stdout)) continue;
|
|
|
|
WingFetcher FP(G, WingOut::Stdout);
|
|
UWingGraphPinRef* PinRef = FP.Walk(Entry.Pin).Cast<UWingGraphPinRef>();
|
|
if (!PinRef) continue;
|
|
UEdGraphPin* Pin = WingUtils::CheckGetPin(PinRef->Node, PinRef->PinName, WingOut::Stdout);
|
|
if (!Pin) continue;
|
|
|
|
int32 DisconnectedCount = 0;
|
|
|
|
if (!Entry.TargetPin.IsEmpty())
|
|
{
|
|
WingFetcher FT(G, WingOut::Stdout);
|
|
UWingGraphPinRef* TargetRef = FT.Walk(Entry.TargetPin).Cast<UWingGraphPinRef>();
|
|
if (!TargetRef) continue;
|
|
UEdGraphPin* Target = WingUtils::CheckGetPin(TargetRef->Node, TargetRef->PinName, WingOut::Stdout);
|
|
if (!Target) continue;
|
|
|
|
if (!Pin->LinkedTo.Contains(Target))
|
|
{
|
|
WingOut::Stdout.Printf(TEXT("Error: %s.%s is not connected to %s.%s\n"),
|
|
*WingUtils::FormatName(Pin->GetOwningNode()), *WingUtils::FormatName(Pin),
|
|
*WingUtils::FormatName(Target->GetOwningNode()), *WingUtils::FormatName(Target));
|
|
continue;
|
|
}
|
|
|
|
Pin->BreakLinkTo(Target);
|
|
DisconnectedCount = 1;
|
|
}
|
|
else
|
|
{
|
|
DisconnectedCount = Pin->LinkedTo.Num();
|
|
if (DisconnectedCount > 0)
|
|
{
|
|
Pin->BreakAllPinLinks(true);
|
|
}
|
|
}
|
|
|
|
WingOut::Stdout.Printf(TEXT("Disconnected %d link(s) from %s.%s\n"),
|
|
DisconnectedCount,
|
|
*WingUtils::FormatName(Pin->GetOwningNode()), *WingUtils::FormatName(Pin));
|
|
SuccessCount++;
|
|
TotalDisconnected += DisconnectedCount;
|
|
}
|
|
|
|
WingOut::Stdout.Printf(TEXT("Done: %d/%d succeeded, %d links broken.\n"),
|
|
SuccessCount, Disconnections.Array.Num(), TotalDisconnected);
|
|
}
|
|
};
|