139 lines
3.7 KiB
C++
139 lines
3.7 KiB
C++
#pragma once
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "MCPServer.h"
|
|
#include "MCPHandler.h"
|
|
#include "MCPFetcher.h"
|
|
#include "MCPJson.h"
|
|
#include "MCPUtils.h"
|
|
#include "Animation/AnimSequence.h"
|
|
#include "Animation/BlendSpace.h"
|
|
#include "AnimBlueprint_SetBlendSpaceSamples.generated.h"
|
|
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// ---------------------------------------------------------------------------
|
|
// ---------------------------------------------------------------------------
|
|
|
|
USTRUCT()
|
|
struct FBlendSpaceSampleEntry
|
|
{
|
|
GENERATED_BODY()
|
|
|
|
UPROPERTY()
|
|
FString AnimationAsset;
|
|
|
|
UPROPERTY()
|
|
float X = 0.0f;
|
|
|
|
UPROPERTY()
|
|
float Y = 0.0f;
|
|
};
|
|
|
|
UCLASS()
|
|
class UMCP_AnimBlueprint_SetBlendSpaceSamples : public UObject, public IMCPHandler
|
|
{
|
|
GENERATED_BODY()
|
|
|
|
public:
|
|
UPROPERTY(meta=(Description="Blend Space package path"))
|
|
FString BlendSpace;
|
|
|
|
UPROPERTY(meta=(Optional, Description="Display name for the X axis"))
|
|
FString AxisXName;
|
|
|
|
UPROPERTY(meta=(Optional, Description="Display name for the Y axis"))
|
|
FString AxisYName;
|
|
|
|
UPROPERTY(meta=(Optional, Description="Minimum value for X axis"))
|
|
float AxisXMin = 0.0f;
|
|
|
|
UPROPERTY(meta=(Optional, Description="Maximum value for X axis"))
|
|
float AxisXMax = 0.0f;
|
|
|
|
UPROPERTY(meta=(Optional, Description="Minimum value for Y axis"))
|
|
float AxisYMin = 0.0f;
|
|
|
|
UPROPERTY(meta=(Optional, Description="Maximum value for Y axis"))
|
|
float AxisYMax = 0.0f;
|
|
|
|
UPROPERTY(meta=(Optional, Description="Array of sample points, each with animationAsset, x, y"))
|
|
FMCPJsonArray Samples;
|
|
|
|
virtual FString GetDescription() const override
|
|
{
|
|
return TEXT("Set axis parameters and animation sample points on a Blend Space. "
|
|
"Replaces all existing samples.");
|
|
}
|
|
|
|
virtual void Handle() override
|
|
{
|
|
// Load the blend space
|
|
MCPFetcher F;
|
|
UBlendSpace* BS = F.Asset(BlendSpace).Cast<UBlendSpace>();
|
|
if (!BS) return;
|
|
|
|
// Set axis parameters
|
|
const FBlendParameter& ParamX = BS->GetBlendParameter(0);
|
|
const FBlendParameter& ParamY = BS->GetBlendParameter(1);
|
|
|
|
// We need to modify BlendParameters directly — use const_cast since there's no setter API
|
|
FBlendParameter& MutableParamX = const_cast<FBlendParameter&>(ParamX);
|
|
FBlendParameter& MutableParamY = const_cast<FBlendParameter&>(ParamY);
|
|
|
|
if (!AxisXName.IsEmpty()) MutableParamX.DisplayName = AxisXName;
|
|
if (AxisXMin != 0.0f) MutableParamX.Min = AxisXMin;
|
|
if (AxisXMax != 0.0f) MutableParamX.Max = AxisXMax;
|
|
|
|
if (!AxisYName.IsEmpty()) MutableParamY.DisplayName = AxisYName;
|
|
if (AxisYMin != 0.0f) MutableParamY.Min = AxisYMin;
|
|
if (AxisYMax != 0.0f) MutableParamY.Max = AxisYMax;
|
|
|
|
// Clear existing samples (delete from end to start)
|
|
int32 NumExisting = BS->GetNumberOfBlendSamples();
|
|
for (int32 i = NumExisting - 1; i >= 0; --i)
|
|
{
|
|
BS->DeleteSample(i);
|
|
}
|
|
|
|
// Add new samples
|
|
int32 SamplesSet = 0;
|
|
|
|
for (const TSharedPtr<FJsonValue>& SampleVal : Samples.Array)
|
|
{
|
|
FBlendSpaceSampleEntry Entry;
|
|
if (!MCPJson::PopulateFromJson(FBlendSpaceSampleEntry::StaticStruct(), &Entry, SampleVal)) return;
|
|
|
|
UAnimSequence* AnimSeq = nullptr;
|
|
if (!Entry.AnimationAsset.IsEmpty())
|
|
{
|
|
MCPFetcher F2;
|
|
AnimSeq = F2.Asset(Entry.AnimationAsset).Cast<UAnimSequence>();
|
|
if (!AnimSeq) return;
|
|
}
|
|
|
|
FVector SampleValue(Entry.X, Entry.Y, 0.0f);
|
|
if (AnimSeq)
|
|
{
|
|
BS->AddSample(AnimSeq, SampleValue);
|
|
}
|
|
else
|
|
{
|
|
BS->AddSample(SampleValue);
|
|
}
|
|
SamplesSet++;
|
|
}
|
|
|
|
BS->ValidateSampleData();
|
|
|
|
// Save
|
|
bool bSaved = MCPUtils::SaveGenericPackage(BS);
|
|
|
|
UMCPServer::Printf(TEXT("Set %d samples on %s\n"), SamplesSet, *MCPUtils::FormatName(BS));
|
|
if (!bSaved)
|
|
{
|
|
UMCPServer::Print(TEXT("WARNING: package save failed\n"));
|
|
}
|
|
}
|
|
};
|