// MyWindow.h
#pragma once

#include "CoreMinimal.h"
#include "DevUIWindow.h"
#include "MyWindow.generated.h"

UCLASS()
class UMyWindow : public UDevUIWindow
{
	GENERATED_BODY()
public:

	/**
	* Draws the window if enabled
	* @return True to indicate this window should remain open
	*/
	bool Draw(UWorld* World) override;

	/** Returns FString that will be used for the title */
	FString GetWindowName() override;

	/** Returns the window help text */
	FString GetWindowHelp() override;

	/** Returns ImGui windows flags */
	ImGuiWindowFlags GetWindowFlags() const override;
};

// MyWindow.cpp
#include "MyWindow.h"

bool UMyWindow::Draw(UWorld* World)
{
	DrawViewModeButtons(World);
	return true;
}

FString UMyWindow::GetWindowName()
{
	return TEXT("My Window");
}

FString UMyWindow::GetWindowHelp()
{
	return TEXT("My window help text!");
}

ImGuiWindowFlags UMyWindow::GetWindowFlags() const
{
	return ImGuiWindowFlags_None;
}

void UMyWindow::DrawViewModeButtons(UWorld* World)
{
	ImGui::BeginGroup();
	ImGui::Text(TCHAR_TO_ANSI(TEXT("View modes")));

	if (ImGui::Button(TCHAR_TO_ANSI(TEXT("Lit"))))
	{
		World->GetGameViewport()->ViewModeIndex = EViewModeIndex::VMI_Lit;
	}
	ImGui::SameLine();

	if (ImGui::Button(TCHAR_TO_ANSI(TEXT("Unlit"))))
	{
		World->GetGameViewport()->ViewModeIndex = EViewModeIndex::VMI_Unlit;
	}
	ImGui::SameLine();

	if (ImGui::Button(TCHAR_TO_ANSI(TEXT("Wireframe"))))
	{
		World->GetGameViewport()->ViewModeIndex = EViewModeIndex::VMI_Wireframe;
	}

	ImGui::EndGroup();
}