119 lines
3.8 KiB
C++
119 lines
3.8 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "PointLightGizmoFactory.h"
|
|
#include "EditorModeManager.h"
|
|
#include "BaseGizmos/CombinedTransformGizmo.h"
|
|
#include "InteractiveGizmoManager.h"
|
|
#include "SubTransformProxy.h"
|
|
#include "Engine/PointLight.h"
|
|
#include "PointLightGizmo.h"
|
|
#include "LightGizmosModule.h"
|
|
#include "Selection.h"
|
|
|
|
#include UE_INLINE_GENERATED_CPP_BY_NAME(PointLightGizmoFactory)
|
|
|
|
bool UPointLightGizmoFactory::CanBuildGizmoForSelection(FEditorModeTools* ModeTools) const
|
|
{
|
|
/** Since there can only be one factory active at a time in the current iteration
|
|
* of the tools framework, this function only returns true if /all/ items in the
|
|
* current selection are point lights and then creates point light gizmos for them
|
|
* If not, the selection falls through to the next factory
|
|
*/
|
|
TArray<AActor*> SelectedActors;
|
|
ModeTools->GetSelectedActors()->GetSelectedObjects<AActor>(SelectedActors);
|
|
|
|
for (const AActor* Actor : SelectedActors)
|
|
{
|
|
if (!Cast<APointLight>(Actor))
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// Only return true if all selected actors are Point Light
|
|
return true;
|
|
}
|
|
|
|
TArray<UInteractiveGizmo*> UPointLightGizmoFactory::BuildGizmoForSelection(FEditorModeTools* ModeTools, UInteractiveGizmoManager* GizmoManager) const
|
|
{
|
|
TArray<UInteractiveGizmo*> Gizmos;
|
|
TArray<AActor*> SelectedActors;
|
|
ModeTools->GetSelectedActors()->GetSelectedObjects<AActor>(SelectedActors);
|
|
|
|
// Create a point light gizmo for each actor
|
|
for (AActor* Actor : SelectedActors)
|
|
{
|
|
UPointLightGizmo* LightGizmo = Cast<UPointLightGizmo>(GizmoManager->CreateGizmo(FLightGizmosModule::PointLightGizmoType));
|
|
|
|
LightGizmo->SetSelectedObject(Cast<APointLight>(Actor));
|
|
LightGizmo->CreateLightGizmo();
|
|
|
|
Gizmos.Add(LightGizmo);
|
|
}
|
|
|
|
|
|
// Create one transform gizmo for the whole selection
|
|
ETransformGizmoSubElements Elements = ETransformGizmoSubElements::None;
|
|
bool bUseContextCoordinateSystem = true;
|
|
UE::Widget::EWidgetMode WidgetMode = ModeTools->GetWidgetMode();
|
|
|
|
switch (WidgetMode)
|
|
{
|
|
case UE::Widget::EWidgetMode::WM_Translate:
|
|
Elements = ETransformGizmoSubElements::TranslateAllAxes | ETransformGizmoSubElements::TranslateAllPlanes;
|
|
break;
|
|
case UE::Widget::EWidgetMode::WM_Rotate:
|
|
Elements = ETransformGizmoSubElements::None; // Rotation doesn't make sense for point lights
|
|
break;
|
|
case UE::Widget::EWidgetMode::WM_Scale:
|
|
Elements = ETransformGizmoSubElements::None; // Attenuation can be scaled using custom gizmo anyways, so scaling doesn't make sense
|
|
bUseContextCoordinateSystem = false;
|
|
break;
|
|
case UE::Widget::EWidgetMode::WM_2D:
|
|
Elements = ETransformGizmoSubElements::RotateAxisY | ETransformGizmoSubElements::TranslatePlaneXZ;
|
|
break;
|
|
default:
|
|
Elements = ETransformGizmoSubElements::FullTranslateRotateScale;
|
|
break;
|
|
}
|
|
|
|
UCombinedTransformGizmo* TransformGizmo = GizmoManager->CreateCustomTransformGizmo(Elements);
|
|
TransformGizmo->bUseContextCoordinateSystem = bUseContextCoordinateSystem;
|
|
|
|
USubTransformProxy* TransformProxy = NewObject<USubTransformProxy>();
|
|
|
|
// Get the SubTransformProxy from each PointLightGizmo and connect it to the main Transform Proxy
|
|
for (UInteractiveGizmo* Gizmo : Gizmos)
|
|
{
|
|
UPointLightGizmo* LightGizmo = Cast<UPointLightGizmo>(Gizmo);
|
|
|
|
if (LightGizmo)
|
|
{
|
|
TransformProxy->AddSubTransformProxy(LightGizmo->GetTransformProxy());
|
|
}
|
|
|
|
}
|
|
|
|
TransformGizmo->SetActiveTarget(TransformProxy);
|
|
TransformGizmo->SetVisibility(SelectedActors.Num() > 0);
|
|
|
|
Gizmos.Add(TransformGizmo);
|
|
|
|
return Gizmos;
|
|
}
|
|
|
|
void UPointLightGizmoFactory::ConfigureGridSnapping(bool bGridEnabled, bool bRotGridEnabled, const TArray<UInteractiveGizmo*>& Gizmos) const
|
|
{
|
|
for (UInteractiveGizmo *Gizmo : Gizmos)
|
|
{
|
|
UCombinedTransformGizmo* TransformGizmo = Cast<UCombinedTransformGizmo>(Gizmo);
|
|
|
|
if (TransformGizmo)
|
|
{
|
|
TransformGizmo->bSnapToWorldGrid = bGridEnabled;
|
|
TransformGizmo->bSnapToWorldRotGrid = bRotGridEnabled;
|
|
}
|
|
}
|
|
}
|
|
|