TransformSources.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "BaseGizmos/GizmoInterfaces.h"
#include "BaseGizmos/TransformProxy.h"
#include "CoreMinimal.h"
#include "Delegates/Delegate.h"
#include "Math/MathFwd.h"
#include "Math/Transform.h"
#include "Math/Vector.h"
#include "Misc/AssertionMacros.h"
#include "Templates/Casts.h"
#include "Templates/Function.h"
#include "UObject/Object.h"
#include "UObject/ObjectMacros.h"
#include "UObject/ObjectPtr.h"
#include "UObject/Package.h"
#include "UObject/ScriptInterface.h"
#include "UObject/UObjectGlobals.h"
 
#include "TransformSources.generated.h"
 
class USceneComponent;
class UTransformProxy;
 
 
 
UCLASS(MinimalAPI)
class UGizmoBaseTransformSource : public UObject, public IGizmoTransformSource
{
    GENERATED_BODY()
public:
    virtual FTransform GetTransform() const
    {
        return FTransform::Identity;
    }
 
    virtual void SetTransform(const FTransform& NewTransform)
    {
        check(false);   // not implemented
    }
 
public:
    DECLARE_MULTICAST_DELEGATE_OneParam(FOnGizmoTransformSourceChanged, IGizmoTransformSource*);
    FOnGizmoTransformSourceChanged OnTransformChanged;
};
 
 
 
 
UCLASS(MinimalAPI)
class UGizmoComponentWorldTransformSource : public UGizmoBaseTransformSource
{
    GENERATED_BODY()
public:
 
    INTERACTIVETOOLSFRAMEWORK_API virtual FTransform GetTransform() const override;
 
    INTERACTIVETOOLSFRAMEWORK_API virtual void SetTransform(const FTransform& NewTransform) override;
 
    UPROPERTY()
    TObjectPtr<USceneComponent> Component;
 
    UPROPERTY()
    bool bModifyComponentOnTransform = true;
 
 
public:
    static UGizmoComponentWorldTransformSource* Construct(
        USceneComponent* Component,
        UObject* Outer = (UObject*)GetTransientPackage())
    {
        UGizmoComponentWorldTransformSource* NewSource = NewObject<UGizmoComponentWorldTransformSource>(Outer);
        NewSource->Component = Component;
        return NewSource;
    }
};
 
 
struct FSeparateScaleProvider
{
    TFunction<FVector(void)> GetScale = []() { return FVector::OneVector; };
    TFunction<void(FVector)> SetScale = [](FVector) {};
};
 
 
UCLASS(MinimalAPI)
class UGizmoScaledTransformSource : public UGizmoBaseTransformSource
{
    GENERATED_BODY()
public:
 
    INTERACTIVETOOLSFRAMEWORK_API virtual FTransform GetTransform() const override;
 
    INTERACTIVETOOLSFRAMEWORK_API virtual void SetTransform(const FTransform& NewTransform) override;
 
    UPROPERTY()
    TScriptInterface<IGizmoTransformSource> ChildTransformSource;
 
    FSeparateScaleProvider ScaleProvider;
 
    INTERACTIVETOOLSFRAMEWORK_API FTransform GetScaledTransform() const;
 
public:
    static UGizmoScaledTransformSource* Construct(
        IGizmoTransformSource* ChildSource,
        FSeparateScaleProvider ScaleProviderIn,
        UObject* Outer = (UObject*)GetTransientPackage())
    {
        UGizmoScaledTransformSource* NewSource = NewObject<UGizmoScaledTransformSource>(Outer);
        NewSource->ChildTransformSource = Cast<UObject>(ChildSource);
        NewSource->ScaleProvider = ScaleProviderIn;
        return NewSource;
    }
};
 
 
 
 
 
 
UCLASS(MinimalAPI)
class UGizmoTransformProxyTransformSource : public UGizmoBaseTransformSource
{
    GENERATED_BODY()
public:
 
    INTERACTIVETOOLSFRAMEWORK_API virtual FTransform GetTransform() const override;
 
    INTERACTIVETOOLSFRAMEWORK_API virtual void SetTransform(const FTransform& NewTransform) override;
 
    UPROPERTY()
    TObjectPtr<UTransformProxy> Proxy;
 
    bool bOverrideSetPivotMode = false;
 
    DECLARE_MULTICAST_DELEGATE_OneParam(FOnGizmoTransformSourcePivotChanged, IGizmoTransformSource*);
    FOnGizmoTransformSourcePivotChanged OnPivotChanged;
 
public:
    static UGizmoTransformProxyTransformSource* Construct(
        UTransformProxy* Proxy,
        UObject* Outer = (UObject*)GetTransientPackage())
    {
        UGizmoTransformProxyTransformSource* NewSource = NewObject<UGizmoTransformProxyTransformSource>(Outer);
        NewSource->Proxy = Proxy;
        return NewSource;
    }
};
 
 
UCLASS(MinimalAPI)
class UGizmoScaledAndUnscaledTransformSources : public UGizmoBaseTransformSource
{
    GENERATED_BODY()
public:
 
    INTERACTIVETOOLSFRAMEWORK_API virtual FTransform GetTransform() const override;
 
    INTERACTIVETOOLSFRAMEWORK_API virtual void SetTransform(const FTransform& NewTransform) override;
 
    UPROPERTY()
    TScriptInterface<IGizmoTransformSource> ScaledTransformSource;
 
    UPROPERTY()
    TScriptInterface<IGizmoTransformSource> UnscaledTransformSource;
 
public:
    static UGizmoScaledAndUnscaledTransformSources* Construct(
        IGizmoTransformSource* ScaledSource,
        USceneComponent* GizmoComponentIn,
        UObject* Outer = (UObject*)GetTransientPackage())
    {
        UGizmoScaledAndUnscaledTransformSources* NewSource = NewObject<UGizmoScaledAndUnscaledTransformSources>(Outer);
        NewSource->ScaledTransformSource = Cast<UObject>(ScaledSource);
 
        UGizmoComponentWorldTransformSource* WrappedGizmo = UGizmoComponentWorldTransformSource::Construct(GizmoComponentIn, Outer);
        WrappedGizmo->bModifyComponentOnTransform = false;
 
        NewSource->UnscaledTransformSource = WrappedGizmo;
        return NewSource;
    }
 
    static UGizmoScaledAndUnscaledTransformSources* Construct(
        IGizmoTransformSource* ScaledSource,
        IGizmoTransformSource* UnscaledSource,
        UObject* Outer = (UObject*)GetTransientPackage())
    {
        UGizmoScaledAndUnscaledTransformSources* NewSource = NewObject<UGizmoScaledAndUnscaledTransformSources>(Outer);
        NewSource->ScaledTransformSource = Cast<UObject>(ScaledSource);
        NewSource->UnscaledTransformSource = Cast<UObject>(UnscaledSource);
        return NewSource;
    }
};