UnrealEngineUWP/Engine/Source/Developer/Datasmith/DatasmithExporterUI/Private/DirectLinkUI.cpp

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

#include "DirectLinkUI.h"

#include "DatasmithDirectLink.h"
#include "DatasmithExporterManager.h"
#include "DirectLinkEndpoint.h"
#include "Widgets/SDirectLinkStreamManager.h"

#include "Containers/Queue.h"
#include "CoreGlobals.h"
#include "Delegates/Delegate.h"
#include "Framework/Application/SWindowTitleBar.h"
#include "Framework/Application/SlateApplication.h"
#include "Hal/Platform.h"
#include "Misc/ConfigCacheIni.h"
#include "Misc/ScopeLock.h"
#include "Styling/CoreStyle.h"
#include "Templates/SharedPointer.h"
#include "Widgets/Input/SButton.h"
#include "Widgets/Layout/SBorder.h"
#include "Widgets/SBoxPanel.h"
#include "Widgets/SWidget.h"
#include "Widgets/SWindow.h"

#define LOCTEXT_NAMESPACE "DirectLinkUI"

namespace DirectLinkUIUtils
{
	FString GetConfigPath()
	{
		return FPaths::Combine( FPaths::GeneratedConfigDir(), TEXT("DirectLinkExporter") ).Append( TEXT(".ini") );;
	}

	FString GetConfigCacheDirectorySectionAndValue()
	{
		return TEXT("DLCacheFolder");
	}

	EHorizontalAlignment GetWindowTitleAlignement()
	{
		EWindowTitleAlignment::Type TitleAlignment = FSlateApplicationBase::Get().GetPlatformApplication()->GetWindowTitleAlignment();
		EHorizontalAlignment TitleContentAlignment;

		if ( TitleAlignment == EWindowTitleAlignment::Left )
		{
			TitleContentAlignment = HAlign_Left;
		}
		else if ( TitleAlignment == EWindowTitleAlignment::Center )
		{
			TitleContentAlignment = HAlign_Center;
		}
		else
		{
			TitleContentAlignment = HAlign_Right;
		}
		return TitleContentAlignment;
	}
}

FDirectLinkUI::FDirectLinkUI()
{
	FString ConfigPath = DirectLinkUIUtils::GetConfigPath();
	FString DirectLinkCacheSectionAndValue = DirectLinkUIUtils::GetConfigCacheDirectorySectionAndValue();

	FScopeLock Lock( &CriticalSectionCacheDirectory );
	if ( !GConfig->GetString( *DirectLinkCacheSectionAndValue, *DirectLinkCacheSectionAndValue, DirectLinkCacheDirectory, ConfigPath ) )
	{
		DirectLinkCacheDirectory = FPaths::Combine( FPlatformProcess::UserTempDir(), TEXT("DLExporter") );
	}
}

void FDirectLinkUI::OpenDirectLinkStreamWindow()
{
	FSimpleDelegate RunOnUIThread;

	TSharedRef<DirectLink::FEndpoint, ESPMode::ThreadSafe> Enpoint = FDatasmithDirectLink::GetEnpoint();

	RunOnUIThread.BindLambda( [this, Enpoint]()
	{
		if ( TSharedPtr<SWindow> OpennWindow = DirectLinkWindow.Pin() )
		{
			// Cancel the minimize
			OpennWindow->BringToFront();

			OpennWindow->HACK_ForceToFront();
			OpennWindow->FlashWindow();
		}
		else
		{
			// This window setup might be an issue on mac where users often expect os borders on their window
			TSharedRef<SWindow> Window = SNew( SWindow )
				.CreateTitleBar( false )
				.ClientSize( FVector2D( 640, 480 ) )
				.AutoCenter( EAutoCenter::PrimaryWorkArea )
				.SizingRule( ESizingRule::UserSized )
				.FocusWhenFirstShown( true )
				.Title( LOCTEXT("DirectlinkStreamManagerWindowTitle", "Datasmith Direct Link Connection Status") );

			TSharedRef<SWindowTitleBar> WindowTitleBar = SNew( SWindowTitleBar, Window, nullptr, DirectLinkUIUtils::GetWindowTitleAlignement() )
				.Visibility( EVisibility::Visible )
				.ShowAppIcon( false );

			Window->SetTitleBar( WindowTitleBar );

			Window->SetContent(
					SNew(SVerticalBox)
					+ SVerticalBox::Slot()
					.AutoHeight()
					[
						WindowTitleBar
					]
					+ SVerticalBox::Slot()
					.FillHeight( 1.f )
					[
						SNew( SBorder )
						.BorderImage( FCoreStyle::Get().GetBrush( "ToolPanel.GroupBorder" ) )
						[
							SNew( SDirectLinkStreamManager, Enpoint )
							// There can only be one window that change this value (no locks required here)
							.DefaultCacheDirectory( DirectLinkCacheDirectory )
							// The slate application should always be close before this module.
							.OnCacheDirectoryChanged_Raw( this, &FDirectLinkUI::OnCacheDirectoryChanged )
						]
					]
				);

			DirectLinkWindow = Window;
			FSlateApplication::Get().AddWindow( Window, true );
			Window->HACK_ForceToFront();
		}
	});

#if IS_PROGRAM
	FDatasmithExporterManager::PushCommandIntoGameThread(MoveTemp(RunOnUIThread));
#else
	RunOnUIThread.Execute();
#endif
}

const TCHAR* FDirectLinkUI::GetDirectLinkCacheDirectory()
{
	// This function should always be called from one thread only
	static const uint32 FirstCallThreadId = FPlatformTLS::GetCurrentThreadId();
	check(FirstCallThreadId == FPlatformTLS::GetCurrentThreadId());

	{
		FScopeLock ReadLock( &CriticalSectionCacheDirectory );
		LastReturnedCacheDirectory = DirectLinkCacheDirectory;
	}

	return *LastReturnedCacheDirectory;
}

void FDirectLinkUI::OnCacheDirectoryChanged(const FString& InNewCacheDirectory)
{
	{
		FScopeLock ReadLock( &CriticalSectionCacheDirectory );
		DirectLinkCacheDirectory = InNewCacheDirectory;
	}

	FString ConfigPath = DirectLinkUIUtils::GetConfigPath();
	FString DirectLinkCacheSectionAndValue = DirectLinkUIUtils::GetConfigCacheDirectorySectionAndValue();

	// Save to config file
	GConfig->SetString( *DirectLinkCacheSectionAndValue, *DirectLinkCacheSectionAndValue, *InNewCacheDirectory, ConfigPath );
}

#undef LOCTEXT_NAMESPACE