Files
sara schvartzman 223638bc56 Control Rig: Trigger connector event when filtering connector candidates
#jira UE-202227
#rb Helge.Mathee
#rnx

[CL 30288978 by sara schvartzman in ue5-main branch]
2023-12-13 06:24:34 -05:00

59 lines
1.6 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "ControlRigSchema.h"
#include "ControlRigBlueprint.h"
#include "RigVMModel/RigVMController.h"
#include "Rigs/RigHierarchyPose.h"
#include "Curves/CurveFloat.h"
#include "Units/RigUnitContext.h"
#include "Units/Modules/RigUnit_ConnectorExecution.h"
#include "Units/Modules/RigUnit_ConnectionCandidates.h"
UControlRigSchema::UControlRigSchema(const FObjectInitializer& ObjectInitializer)
: Super(ObjectInitializer)
{
SetExecuteContextStruct(FControlRigExecuteContext::StaticStruct());
}
bool UControlRigSchema::ShouldUnfoldStruct(URigVMController* InController, const UStruct* InStruct) const
{
RIGVMSCHEMA_DEFAULT_FUNCTION_BODY
if(!Super::ShouldUnfoldStruct(InController, InStruct))
{
return false;
}
if (InStruct == TBaseStructure<FQuat>::Get())
{
return false;
}
if (InStruct == FRuntimeFloatCurve::StaticStruct())
{
return false;
}
if (InStruct == FRigPose::StaticStruct())
{
return false;
}
return true;
}
bool UControlRigSchema::SupportsUnitFunction(URigVMController* InController, const FRigVMFunction* InUnitFunction) const
{
if (InUnitFunction->Struct == FRigUnit_ConnectorExecution::StaticStruct() ||
InUnitFunction->Struct == FRigUnit_GetCandidates::StaticStruct() ||
InUnitFunction->Struct == FRigUnit_DiscardMatches::StaticStruct())
{
if (UControlRigBlueprint* Blueprint = Cast<UControlRigBlueprint>(InController->GetOuter()))
{
if (Blueprint->IsControlRigModule())
{
return true;
}
return false;
}
}
return Super::SupportsUnitFunction(InController, InUnitFunction);
}