// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#if WITH_EDITOR

#include "Framework/Application/SWindowTitleBar.h"
#include "PIEPreviewWindow.h"
#include "PIEPreviewWindowStyle.h"
#include "PIEPreviewWindowCoreStyle.h"

/**
* Implements a window PIE title bar widget.
*/
class PIEPREVIEWDEVICEPROFILESELECTOR_API SPIEPreviewWindowTitleBar
	: public SWindowTitleBar
{
private:
	/**
	* Creates widgets for this window's title bar area.
	*/
	virtual void MakeTitleBarContentWidgets(TSharedPtr< SWidget >& OutLeftContent, TSharedPtr< SWidget >& OutRightContent) override;

	const FSlateBrush* GetScreenRotationButtonImage() const;

	TSharedPtr<class SPIEPreviewWindow> GetOwnerWindow() const;

private:
	// Holds the screen rotation button.
	TSharedPtr<SButton> ScreenRotationButton;

	// Holds the clamp button.
	TSharedPtr<class SCheckBox> ClampWindowSizeCheckBox;
};

FORCEINLINE TSharedPtr<class SPIEPreviewWindow> SPIEPreviewWindowTitleBar::GetOwnerWindow() const
{
	TSharedPtr<SWindow> OwnerWindow = OwnerWindowPtr.Pin();
	TSharedPtr<SPIEPreviewWindow> PIEWindow = StaticCastSharedPtr<SPIEPreviewWindow>(OwnerWindow);

	return PIEWindow;
}

#endif