You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
3.7 KiB
97 lines
3.7 KiB
// Copyright 2019 yangxiangyun
|
|
// All Rights Reserved
|
|
|
|
#include "NGAGraphPinConnectionFactory.h"
|
|
|
|
#include "ConnectionDrawingPolicy/NGAMaterialGraphConnectionDrawingPolicy.h"
|
|
#include "ConnectionDrawingPolicy/NGABlueprintConnectionDrawingPolicy.h"
|
|
#include "ConnectionDrawingPolicy/NGASoundCueConnectionDrawingPolicy.h"
|
|
#include "ConnectionDrawingPolicy/NGAAnimGraphConnectionDrawingPolicy.h"
|
|
|
|
#include "MaterialGraph/MaterialGraphSchema.h"
|
|
#include "EdGraphSchema_K2.h"
|
|
#include "AnimationStateMachineSchema.h"
|
|
|
|
#ifdef NGA_WITH_ENGINE_CPP
|
|
#include "StateMachineConnectionDrawingPolicy.cpp"
|
|
#include "AnimationStateNodes/SGraphNodeAnimTransition.cpp"
|
|
#else
|
|
#include "EngineCppFiles/StateMachineConnectionDrawingPolicy.cpp"
|
|
#include "EngineCppFiles/SGraphNodeAnimTransition.cpp"
|
|
#endif
|
|
|
|
#include "AnimationGraphSchema.h"
|
|
#include "Version.h"
|
|
|
|
|
|
|
|
//before 419 engine will use last factory in factory array when creation connection draw policy
|
|
//this will override previous factory even if the last one can not actual create correct factory
|
|
//we need to make sure its always the last factory,and it need to include all function of any other factory
|
|
FConnectionDrawingPolicy* FNGAGraphPinConnectionFactory::CreateConnectionPolicy(const class UEdGraphSchema* Schema, int32 InBackLayerID, int32 InFrontLayerID, float ZoomFactor, const class FSlateRect& InClippingRect, class FSlateWindowElementList& InDrawElements, class UEdGraph* InGraphObj) const
|
|
{
|
|
TSharedPtr<FNGAGraphPinConnectionFactoryPayLoadData> payLoadData;
|
|
if (PayLoadData->HoveredGraphPanel.IsValid() && PayLoadData->HoveredGraphPanel.Pin()->GetGraphObj() == InGraphObj)
|
|
{
|
|
payLoadData = PayLoadData;
|
|
}
|
|
|
|
if (Schema->IsA(UMaterialGraphSchema::StaticClass()))
|
|
{
|
|
return new FNGAMaterialGraphConnectionDrawingPolicy(InBackLayerID, InFrontLayerID, ZoomFactor, InClippingRect, InDrawElements, InGraphObj, payLoadData);
|
|
}
|
|
else if (Schema->IsA(UAnimationGraphSchema::StaticClass()))
|
|
{
|
|
return new FNGAAnimGraphConnectionDrawingPolicy(InBackLayerID, InFrontLayerID, ZoomFactor, InClippingRect, InDrawElements, InGraphObj, payLoadData);
|
|
}
|
|
else if (Schema->IsA(UEdGraphSchema_K2::StaticClass()))
|
|
{
|
|
return new FNGABPGraphConnectionDrawingPolicy(InBackLayerID, InFrontLayerID, ZoomFactor, InClippingRect, InDrawElements, InGraphObj, payLoadData);
|
|
}
|
|
#if ENGINE_MAJOR_VERSION == 4 && ENGINE_MINOR_VERSION < 19
|
|
else if (Schema->IsA(USoundCueGraphSchema::StaticClass()))
|
|
{
|
|
return new FNGASoundCueGraphConnectionDrawingPolicy(InBackLayerID, InFrontLayerID, ZoomFactor, InClippingRect, InDrawElements, InGraphObj, payLoadData);
|
|
}
|
|
else if (Schema->IsA(UAnimationStateMachineSchema::StaticClass()))
|
|
{
|
|
return new FStateMachineConnectionDrawingPolicy(InBackLayerID, InFrontLayerID, ZoomFactor, InClippingRect, InDrawElements, InGraphObj);
|
|
}
|
|
#endif
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
|
|
void FNGAGraphPinConnectionFactory::SetLazyConnectPayloadData(TWeakPtr<SGraphNode> InHoveredNode, TArray<UEdGraphPin*> InDraggingPins)
|
|
{
|
|
PayLoadData->HoveredNode = InHoveredNode;
|
|
PayLoadData->DraggingPins = InDraggingPins;
|
|
}
|
|
|
|
|
|
void FNGAGraphPinConnectionFactory::SetAutoConnectPayloadData(TArray<TWeakPtr<SGraphPin>> InAutoConnectStartPins, TArray<TWeakPtr<SGraphPin>> InAutoConnectEndPins)
|
|
{
|
|
PayLoadData->AutoConnectStartPins = InAutoConnectStartPins;
|
|
PayLoadData->AutoConnectEndPins = InAutoConnectEndPins;
|
|
}
|
|
|
|
|
|
void FNGAGraphPinConnectionFactory::ResetLazyConnectPayloadData()
|
|
{
|
|
PayLoadData->HoveredNode.Reset();
|
|
PayLoadData->DraggingPins.Empty();
|
|
}
|
|
|
|
|
|
void FNGAGraphPinConnectionFactory::ResetAutoConnectPayloadData()
|
|
{
|
|
PayLoadData->AutoConnectStartPins.Empty();
|
|
PayLoadData->AutoConnectEndPins.Empty();
|
|
}
|
|
|
|
|
|
void FNGAGraphPinConnectionFactory::ResetInsertNodePayloadData()
|
|
{
|
|
PayLoadData->InsertNodePinInfos.Empty();
|
|
}
|