Files
UnrealEngineUWP/Engine/Plugins/Animation/ControlRig/Source/ControlRigDeveloper/Private/ControlRigSchema.cpp
helge mathee a5a053cabe Control Rig: Rename rigid body element to physics element
#rb sara.schvartzman
[FYI] danny.chapman
#jira UE-219409
#prefligt https://horde.devtools.epicgames.com/job/66963a31206ca881ae05f4ef
#rnx

[CL 34835213 by helge mathee in ue5-main branch]
2024-07-16 05:51:30 -04:00

63 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;
}
if (InStruct == FRigPhysicsSolverID::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);
}