Skip to content
Snippets Groups Projects
Select Git revision
  • f6c638d9afa0d1b840efa546d7ddb4c00cd8437a
  • master default protected
2 results

controller.test.cpp

Blame
    • DisplayClusterExtensionsEditor.cpp 1.48 KiB 
    #include "DisplayClusterExtensionsEditor.h"

#include "ComponentVisualizers.h"
#include "GrabbingBehaviorOnLineVisualizer.h"
#include "GrabbingBehaviorPlaneVisualizer.h"

#include "GrabbingBehaviorOnPlaneComponent.h"
#include "GrabbingBehaviorOnLineComponent.h"

#include "UnrealEdGlobals.h"
#include "Editor/UnrealEdEngine.h"

IMPLEMENT_GAME_MODULE(FDisplayClusterExtensionsEditorModule, DisplayClusterExtensionsEditor);

#define LOCTEXT_NAMESPACE "DisplayClusterExtensionsEdito"

void FDisplayClusterExtensionsEditorModule::StartupModule()
{
	if (GUnrealEd != NULL)
	{
		TSharedPtr<FComponentVisualizer> LineVisualizer = MakeShareable(new FGrabbingBehaviorOnLineVisualizer());

		if (LineVisualizer.IsValid())
		{
			GUnrealEd->RegisterComponentVisualizer(UGrabbingBehaviorOnLineComponent::StaticClass()->GetFName(), LineVisualizer);
			LineVisualizer->OnRegister();
		}

		TSharedPtr<FComponentVisualizer> PlaneVisualizer = MakeShareable(new FGrabbingBehaviorPlaneVisualizer());
		
		if (PlaneVisualizer.IsValid())
		{
			GUnrealEd->RegisterComponentVisualizer(UGrabbingBehaviorOnPlaneComponent::StaticClass()->GetFName(), PlaneVisualizer);
			PlaneVisualizer->OnRegister();
		}
	}
}

void FDisplayClusterExtensionsEditorModule::ShutdownModule()
{
	if (GUnrealEd != NULL)
	{
		GUnrealEd->UnregisterComponentVisualizer(UGrabbingBehaviorOnLineComponent::StaticClass()->GetFName());
		GUnrealEd->UnregisterComponentVisualizer(UGrabbingBehaviorOnPlaneComponent::StaticClass()->GetFName());
	}
}

#undef LOCTEXT_NAMESPACE