142 lines
4.2 KiB
C++
142 lines
4.2 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "ControlRigPythonLogDetails.h"
|
|
#include "Widgets/SWidget.h"
|
|
#include "DetailLayoutBuilder.h"
|
|
#include "DetailCategoryBuilder.h"
|
|
#include "DetailWidgetRow.h"
|
|
#include "IDetailChildrenBuilder.h"
|
|
#include "Widgets/Input/SButton.h"
|
|
#include "Widgets/Text/STextBlock.h"
|
|
#include "HAL/PlatformApplicationMisc.h"
|
|
#include "IPropertyUtilities.h"
|
|
#include "IPythonScriptPlugin.h"
|
|
#include "RigVMPythonUtils.h"
|
|
|
|
#define LOCTEXT_NAMESPACE "ControlRigCompilerDetails"
|
|
|
|
void FControlRigPythonLogDetails::CustomizeHeader(TSharedRef<IPropertyHandle> InStructPropertyHandle, FDetailWidgetRow& HeaderRow, IPropertyTypeCustomizationUtils& StructCustomizationUtils)
|
|
{
|
|
HeaderRow
|
|
.NameContent()
|
|
[
|
|
InStructPropertyHandle->CreatePropertyNameWidget()
|
|
]
|
|
.ValueContent()
|
|
[
|
|
InStructPropertyHandle->CreatePropertyValueWidget()
|
|
];
|
|
|
|
TArray<UObject*> Objects;
|
|
InStructPropertyHandle->GetOuterObjects(Objects);
|
|
ensure(Objects.Num() == 1); // This is in here to ensure we are only showing the modifier details in the blueprint editor
|
|
|
|
for (UObject* Object : Objects)
|
|
{
|
|
if (Object->IsA<UControlRigBlueprint>())
|
|
{
|
|
BlueprintBeingCustomized = Cast<UControlRigBlueprint>(Object);
|
|
}
|
|
}
|
|
}
|
|
|
|
void FControlRigPythonLogDetails::CustomizeChildren(TSharedRef<IPropertyHandle> InStructPropertyHandle, IDetailChildrenBuilder& StructBuilder, IPropertyTypeCustomizationUtils& StructCustomizationUtils)
|
|
{
|
|
if (InStructPropertyHandle->IsValidHandle())
|
|
{
|
|
uint32 NumChildren = 0;
|
|
InStructPropertyHandle->GetNumChildren(NumChildren);
|
|
|
|
for (uint32 ChildIndex = 0; ChildIndex < NumChildren; ChildIndex++)
|
|
{
|
|
StructBuilder.AddProperty(InStructPropertyHandle->GetChildHandle(ChildIndex).ToSharedRef());
|
|
}
|
|
|
|
StructBuilder.AddCustomRow(LOCTEXT("Commands", "Python Commands"))
|
|
.NameContent()
|
|
[
|
|
SNew(STextBlock)
|
|
.Text(FText::FromString(TEXT("Commands")))
|
|
.Font(IDetailLayoutBuilder::GetDetailFont())
|
|
]
|
|
.ValueContent()
|
|
[
|
|
SNew(SVerticalBox)
|
|
+ SVerticalBox::Slot()
|
|
[
|
|
SNew(SButton)
|
|
.OnClicked(this, &FControlRigPythonLogDetails::OnCopyPythonScriptClicked)
|
|
.ContentPadding(FMargin(2))
|
|
.Content()
|
|
[
|
|
SNew(STextBlock)
|
|
.Justification(ETextJustify::Center)
|
|
.Text(LOCTEXT("CopyPythonScript", "Copy Python Script"))
|
|
]
|
|
]
|
|
+ SVerticalBox::Slot()
|
|
[
|
|
SNew(SButton)
|
|
.OnClicked(this, &FControlRigPythonLogDetails::OnRunPythonContextClicked)
|
|
.ContentPadding(FMargin(2))
|
|
.Content()
|
|
[
|
|
SNew(STextBlock)
|
|
.Justification(ETextJustify::Center)
|
|
.Text(LOCTEXT("RunPythonContext", "Run Python Context"))
|
|
]
|
|
]
|
|
];
|
|
}
|
|
}
|
|
|
|
FReply FControlRigPythonLogDetails::OnCopyPythonScriptClicked()
|
|
{
|
|
if (BlueprintBeingCustomized)
|
|
{
|
|
FString NewName = BlueprintBeingCustomized->GetPathName();
|
|
int32 DotIndex = NewName.Find(TEXT("."), ESearchCase::IgnoreCase, ESearchDir::FromEnd);
|
|
if (DotIndex != INDEX_NONE)
|
|
{
|
|
NewName = NewName.Left(DotIndex);
|
|
}
|
|
|
|
TArray<FString> Commands = BlueprintBeingCustomized->GeneratePythonCommands(NewName);
|
|
FString FullScript = FString::Join(Commands, TEXT("\n"));
|
|
FPlatformApplicationMisc::ClipboardCopy(*FullScript);
|
|
}
|
|
return FReply::Handled();
|
|
}
|
|
|
|
FReply FControlRigPythonLogDetails::OnRunPythonContextClicked()
|
|
{
|
|
if (BlueprintBeingCustomized)
|
|
{
|
|
FString BlueprintName = BlueprintBeingCustomized->GetPathName();
|
|
int32 DotIndex = BlueprintName.Find(TEXT("."), ESearchCase::IgnoreCase, ESearchDir::FromEnd);
|
|
if (DotIndex != INDEX_NONE)
|
|
{
|
|
BlueprintName = BlueprintName.Left(DotIndex);
|
|
}
|
|
|
|
TArray<FString> PyCommands = {
|
|
TEXT("import unreal"),
|
|
FString::Printf(TEXT("blueprint = unreal.load_object(name = '%s', outer = None)"), *BlueprintName),
|
|
TEXT("library = blueprint.get_local_function_library()"),
|
|
TEXT("library_controller = blueprint.get_controller(library)"),
|
|
TEXT("hierarchy = blueprint.hierarchy"),
|
|
TEXT("hierarchy_controller = hierarchy.get_controller()")};
|
|
|
|
for (FString& Command : PyCommands)
|
|
{
|
|
RigVMPythonUtils::Print(BlueprintBeingCustomized->GetFName().ToString(), Command);
|
|
|
|
// Run the Python commands
|
|
IPythonScriptPlugin::Get()->ExecPythonCommand(*Command);
|
|
}
|
|
}
|
|
return FReply::Handled();
|
|
}
|
|
|
|
#undef LOCTEXT_NAMESPACE
|