80 lines
2.4 KiB
C++
80 lines
2.4 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "Modules/ModuleInterface.h"
|
|
#include "EdMode.h"
|
|
#include "EditorModeTools.h"
|
|
|
|
class FEditorViewportClient;
|
|
class FEdModeTexture;
|
|
class FScopedTransaction;
|
|
class FViewport;
|
|
|
|
/**
|
|
* Texture mode module
|
|
*/
|
|
class FTextureAlignModeModule : public IModuleInterface
|
|
{
|
|
private:
|
|
TSharedPtr<class FEdModeTexture> EdModeTexture;
|
|
public:
|
|
/**
|
|
* Called right after the module's DLL has been loaded and the module object has been created
|
|
*/
|
|
virtual void StartupModule() override;
|
|
|
|
/**
|
|
* Called before the module is unloaded, right before the module object is destroyed.
|
|
*/
|
|
virtual void ShutdownModule() override;};
|
|
|
|
/**
|
|
* Allows texture alignment on BSP surfaces via the widget.
|
|
*/
|
|
class FEdModeTexture : public FEdMode
|
|
{
|
|
public:
|
|
FEdModeTexture();
|
|
virtual ~FEdModeTexture();
|
|
|
|
/** Stores the coordinate system that was active when the mode was entered so it can restore it later. */
|
|
ECoordSystem SaveCoordSystem;
|
|
|
|
virtual void Enter() override;
|
|
virtual void Exit() override;
|
|
virtual FVector GetWidgetLocation() const override;
|
|
virtual bool ShouldDrawWidget() const override;
|
|
virtual bool GetCustomDrawingCoordinateSystem( FMatrix& InMatrix, void* InData ) override;
|
|
virtual bool GetCustomInputCoordinateSystem( FMatrix& InMatrix, void* InData ) override;
|
|
virtual EAxisList::Type GetWidgetAxisToDraw( UE::Widget::EWidgetMode InWidgetMode ) const override;
|
|
virtual bool StartTracking(FEditorViewportClient* InViewportClient, FViewport* InViewport) override;
|
|
virtual bool EndTracking(FEditorViewportClient* InViewportClient, FViewport* InViewport) override;
|
|
virtual bool AllowWidgetMove() override { return false; }
|
|
virtual bool IsCompatibleWith(FEditorModeID OtherModeID) const override;
|
|
|
|
protected:
|
|
/** The current transaction. */
|
|
FScopedTransaction* ScopedTransaction;
|
|
/* The world that brush that we started tracking with belongs to. Cleared when tracking ends. */
|
|
UWorld* TrackingWorld;
|
|
};
|
|
|
|
/**
|
|
* FModeTool_Texture
|
|
*/
|
|
class FModeTool_Texture : public FModeTool
|
|
{
|
|
public:
|
|
FModeTool_Texture();
|
|
|
|
virtual bool InputDelta(FEditorViewportClient* InViewportClient,FViewport* InViewport,FVector& InDrag,FRotator& InRot,FVector& InScale);
|
|
|
|
// override these to allow this tool to keep track of the users dragging during a single drag event
|
|
virtual bool StartModify() { PreviousInputDrag = FVector::ZeroVector; return true; }
|
|
virtual bool EndModify() { return true; }
|
|
|
|
private:
|
|
FVector PreviousInputDrag;
|
|
};
|