GeoEditorEdMode.cpp

// Copyright 1998-2018 Epic Games, Inc. All Rights Reserved.

#include "GeoEditorEdMode.h"
#include "GeoEditorEdModeToolkit.h"
#include "GeoEditorCommands.h"
#include "Toolkits/ToolkitManager.h"
#include "Engine/Selection.h"
#include "EditorModeManager.h"

const FEditorModeID FGeoEditorEdMode::EM_GeoEditorEdModeId = TEXT("EM_GeoEditorEdMode");

FGeoEditorEdMode::FGeoEditorEdMode()
{
	GEditor->AddOnScreenDebugMessage(-1, 1.f, FColor::White, "HAI");
	UICommandList = MakeShareable(new FUICommandList);
	
}

FGeoEditorEdMode::~FGeoEditorEdMode()
{

}

void FGeoEditorEdMode::Enter()
{
	FEdMode::Enter();

	if (!Toolkit.IsValid() && UsesToolkits())
	{
		Toolkit = MakeShareable(new FGeoEditorEdModeToolkit);
		Toolkit->Init(Owner->GetToolkitHost());
	}
}

void FGeoEditorEdMode::Exit()
{
	if (Toolkit.IsValid())
	{
		FToolkitManager::Get().CloseToolkit(Toolkit.ToSharedRef());
		Toolkit.Reset();
	}

	// Call base Exit method to ensure proper cleanup
	FEdMode::Exit();
}

bool FGeoEditorEdMode::UsesToolkits() const
{
	return true;
}


bool FGeoEditorEdMode::InputKey(FEditorViewportClient* InViewportClient, FViewport* InViewport, FKey InKey, EInputEvent InEvent)
{
	bRightMouseDown = InKey == EKeys::RightMouseButton;

	bCtrlPressed = InViewportClient->IsCtrlPressed();
	bAltPressed = InViewportClient->IsAltPressed();
	bShiftPressed = InViewportClient->IsShiftPressed();

	bool bHandled = false;

	if (!bHandled && InEvent != EInputEvent::IE_Released)
	{
		if (UICommandList->ProcessCommandBindings(InKey, FSlateApplication::Get().GetModifierKeys(), false))
		{
			bHandled = true;
		}
	}
	if (!bHandled)
	{
		const bool bKeyPressed = InEvent == EInputEvent::IE_Pressed;
		const bool bKeyReleased = InEvent == EInputEvent::IE_Released;

		if (InKey == EKeys::U && bCtrlPressed)
		{
			GEditor->AddOnScreenDebugMessage(-1, 1.f, FColor::White, "HAI");
			USelection* SelectedActors = GEditor->GetSelectedActors();
			TArray<AActor*> Actors;
			for (FSelectionIterator Iter(*SelectedActors); Iter; ++Iter)
			{
				AActor* SelectedActor = Cast<AActor>(*Iter);
				if (SelectedActor)
				{
					SelectedActor->Destroy();
				}
			}
		}
	}
	return bHandled;
}

void FGeoEditorEdMode::BindCommands()
{
	const FGeoEditorCommands& Commands = FGeoEditorCommands::Get();

	UICommandList->MapAction(Commands.AlignBrushVerticesToGrid, FExecuteAction::CreateRaw(this, &FGeoEditorEdMode::AlignBrushVerticesToGrid));

}

void FGeoEditorEdMode::AlignBrushVerticesToGrid()
{
	GEditor->AddOnScreenDebugMessage(-1, 1.f, FColor::White, "HAI");
	USelection* SelectedActors = GEditor->GetSelectedActors();
	TArray<AActor*> Actors;
	for (FSelectionIterator Iter(*SelectedActors); Iter; ++Iter)
	{
		AActor* SelectedActor = Cast<AActor>(*Iter);
		if (SelectedActor)
		{
			SelectedActor->Destroy();
		}
	}
}