Andron666
343 lines
13 KiB
C++
343 lines
13 KiB
C++
/*
|
|
* Copyright (c) <2017> Side Effects Software Inc.
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*
|
|
*/
|
|
|
|
#include "SNewFilePathPicker.h"
|
|
|
|
#include "HoudiniApi.h"
|
|
#include "DesktopPlatformModule.h"
|
|
#include "Widgets/SBoxPanel.h"
|
|
#include "Framework/Application/SlateApplication.h"
|
|
#include "Widgets/Images/SImage.h"
|
|
#include "Widgets/Input/SEditableTextBox.h"
|
|
#include "Widgets/Input/SButton.h"
|
|
|
|
|
|
#define LOCTEXT_NAMESPACE "SNewFilePathPicker"
|
|
|
|
|
|
/* SNewFilePathPicker interface
|
|
*****************************************************************************/
|
|
|
|
void SNewFilePathPicker::Construct( const FArguments& InArgs )
|
|
{
|
|
BrowseDirectory = InArgs._BrowseDirectory;
|
|
BrowseTitle = InArgs._BrowseTitle;
|
|
FilePath = InArgs._FilePath;
|
|
FileTypeFilter = InArgs._FileTypeFilter;
|
|
OnPathPicked = InArgs._OnPathPicked;
|
|
IsNewFile = InArgs._IsNewFile;
|
|
|
|
ChildSlot
|
|
[
|
|
SNew(SHorizontalBox)
|
|
|
|
+ SHorizontalBox::Slot()
|
|
.FillWidth(1.0f)
|
|
.VAlign(VAlign_Center)
|
|
[
|
|
// file path text box
|
|
SAssignNew(TextBox, SEditableTextBox)
|
|
.Text(this, &SNewFilePathPicker::HandleTextBoxText)
|
|
.Font(InArgs._Font)
|
|
.SelectAllTextWhenFocused(true)
|
|
.ClearKeyboardFocusOnCommit(false)
|
|
.OnTextCommitted(this, &SNewFilePathPicker::HandleTextBoxTextCommitted)
|
|
.SelectAllTextOnCommit(false)
|
|
.IsReadOnly(InArgs._IsReadOnly)
|
|
]
|
|
|
|
+ SHorizontalBox::Slot()
|
|
.AutoWidth()
|
|
.Padding(4.0f, 0.0f, 0.0f, 0.0f)
|
|
.VAlign(VAlign_Center)
|
|
[
|
|
// browse button
|
|
SNew(SButton)
|
|
.ButtonStyle(InArgs._BrowseButtonStyle)
|
|
.ToolTipText(InArgs._BrowseButtonToolTip)
|
|
.OnClicked(this, &SNewFilePathPicker::HandleBrowseButtonClicked)
|
|
.ContentPadding(2.0f)
|
|
.ForegroundColor(FSlateColor::UseForeground())
|
|
.IsFocusable(false)
|
|
[
|
|
SNew(SImage)
|
|
.Image(InArgs._BrowseButtonImage)
|
|
.ColorAndOpacity(FSlateColor::UseForeground())
|
|
]
|
|
]
|
|
];
|
|
}
|
|
|
|
|
|
/* SNewFilePathPicker callbacks
|
|
*****************************************************************************/
|
|
#if PLATFORM_WINDOWS
|
|
|
|
#include "Windows/WindowsHWrapper.h"
|
|
#include "Windows/COMPointer.h"
|
|
//#include "Misc/Paths.h"
|
|
//#include "Misc/Guid.h"
|
|
#include "HAL/FileManager.h"
|
|
#include "Windows/AllowWindowsPlatformTypes.h"
|
|
#include <commdlg.h>
|
|
//#include <shellapi.h>
|
|
#include <shlobj.h>
|
|
//#include <Winver.h>
|
|
//#include <LM.h>
|
|
//#include <tlhelp32.h>
|
|
//#include <Psapi.h>
|
|
#include "Windows/HideWindowsPlatformTypes.h"
|
|
//#pragma comment( lib, "version.lib" )
|
|
|
|
bool FileDialogShared( bool bSave, const void* ParentWindowHandle, const FString& DialogTitle, const FString& DefaultPath, const FString& DefaultFile, const FString& FileTypes, uint32 Flags, TArray<FString>& OutFilenames, int32& OutFilterIndex )
|
|
{
|
|
FScopedSystemModalMode SystemModalScope;
|
|
|
|
bool bSuccess = false;
|
|
|
|
TComPtr<IFileDialog> FileDialog;
|
|
if ( SUCCEEDED( ::CoCreateInstance( bSave ? CLSID_FileSaveDialog : CLSID_FileOpenDialog, nullptr, CLSCTX_INPROC_SERVER, bSave ? IID_IFileSaveDialog : IID_IFileOpenDialog, IID_PPV_ARGS_Helper( &FileDialog ) ) ) )
|
|
{
|
|
if ( bSave )
|
|
{
|
|
// Set the default "filename"
|
|
if ( !DefaultFile.IsEmpty() )
|
|
{
|
|
FileDialog->SetFileName( *FPaths::GetCleanFilename( DefaultFile ) );
|
|
}
|
|
DWORD dwFlags = 0;
|
|
FileDialog->GetOptions( &dwFlags );
|
|
FileDialog->SetOptions( dwFlags & ~FOS_OVERWRITEPROMPT );
|
|
}
|
|
else
|
|
{
|
|
// Set this up as a multi-select picker
|
|
if ( Flags & EFileDialogFlags::Multiple )
|
|
{
|
|
DWORD dwFlags = 0;
|
|
FileDialog->GetOptions( &dwFlags );
|
|
FileDialog->SetOptions( dwFlags | FOS_ALLOWMULTISELECT );
|
|
}
|
|
}
|
|
|
|
// Set up common settings
|
|
FileDialog->SetTitle( *DialogTitle );
|
|
if ( !DefaultPath.IsEmpty() )
|
|
{
|
|
// SHCreateItemFromParsingName requires the given path be absolute and use \ rather than / as our normalized paths do
|
|
FString DefaultWindowsPath = FPaths::ConvertRelativePathToFull( DefaultPath );
|
|
DefaultWindowsPath.ReplaceInline( TEXT( "/" ), TEXT( "\\" ), ESearchCase::CaseSensitive );
|
|
|
|
TComPtr<IShellItem> DefaultPathItem;
|
|
if ( SUCCEEDED( ::SHCreateItemFromParsingName( *DefaultWindowsPath, nullptr, IID_PPV_ARGS( &DefaultPathItem ) ) ) )
|
|
{
|
|
FileDialog->SetFolder( DefaultPathItem );
|
|
}
|
|
}
|
|
|
|
// Set-up the file type filters
|
|
TArray<FString> UnformattedExtensions;
|
|
TArray<COMDLG_FILTERSPEC> FileDialogFilters;
|
|
{
|
|
// Split the given filter string (formatted as "Pair1String1|Pair1String2|Pair2String1|Pair2String2") into the Windows specific filter struct
|
|
FileTypes.ParseIntoArray( UnformattedExtensions, TEXT( "|" ), true );
|
|
|
|
if ( UnformattedExtensions.Num() % 2 == 0 )
|
|
{
|
|
FileDialogFilters.Reserve( UnformattedExtensions.Num() / 2 );
|
|
for ( int32 ExtensionIndex = 0; ExtensionIndex < UnformattedExtensions.Num();)
|
|
{
|
|
COMDLG_FILTERSPEC& NewFilterSpec = FileDialogFilters[FileDialogFilters.AddDefaulted()];
|
|
NewFilterSpec.pszName = *UnformattedExtensions[ExtensionIndex++];
|
|
NewFilterSpec.pszSpec = *UnformattedExtensions[ExtensionIndex++];
|
|
}
|
|
}
|
|
}
|
|
FileDialog->SetFileTypes( FileDialogFilters.Num(), FileDialogFilters.GetData() );
|
|
|
|
// Show the picker
|
|
if ( SUCCEEDED( FileDialog->Show( (HWND)ParentWindowHandle ) ) )
|
|
{
|
|
OutFilterIndex = 0;
|
|
if ( SUCCEEDED( FileDialog->GetFileTypeIndex( (UINT*)&OutFilterIndex ) ) )
|
|
{
|
|
OutFilterIndex -= 1; // GetFileTypeIndex returns a 1-based index
|
|
}
|
|
|
|
auto AddOutFilename = [&OutFilenames]( const FString& InFilename )
|
|
{
|
|
FString& OutFilename = OutFilenames[OutFilenames.Add( InFilename )];
|
|
OutFilename = IFileManager::Get().ConvertToRelativePath( *OutFilename );
|
|
FPaths::NormalizeFilename( OutFilename );
|
|
};
|
|
|
|
if ( bSave )
|
|
{
|
|
TComPtr<IShellItem> Result;
|
|
if ( SUCCEEDED( FileDialog->GetResult( &Result ) ) )
|
|
{
|
|
PWSTR pFilePath = nullptr;
|
|
if ( SUCCEEDED( Result->GetDisplayName( SIGDN_FILESYSPATH, &pFilePath ) ) )
|
|
{
|
|
bSuccess = true;
|
|
|
|
// Apply the selected extension if the given filename doesn't already have one
|
|
FString SaveFilePath = pFilePath;
|
|
if ( FileDialogFilters.IsValidIndex( OutFilterIndex ) )
|
|
{
|
|
// Build a "clean" version of the selected extension (without the wildcard)
|
|
FString CleanExtension = FileDialogFilters[OutFilterIndex].pszSpec;
|
|
if ( CleanExtension == TEXT( "*.*" ) )
|
|
{
|
|
CleanExtension.Reset();
|
|
}
|
|
else
|
|
{
|
|
const int32 WildCardIndex = CleanExtension.Find( TEXT( "*" ) );
|
|
if ( WildCardIndex != INDEX_NONE )
|
|
{
|
|
CleanExtension = CleanExtension.RightChop( WildCardIndex + 1 );
|
|
}
|
|
}
|
|
|
|
// We need to split these before testing the extension to avoid anything within the path being treated as a file extension
|
|
FString SaveFileName = FPaths::GetCleanFilename( SaveFilePath );
|
|
SaveFilePath = FPaths::GetPath( SaveFilePath );
|
|
|
|
// Apply the extension if the file name doesn't already have one
|
|
if ( FPaths::GetExtension( SaveFileName ).IsEmpty() && !CleanExtension.IsEmpty() )
|
|
{
|
|
SaveFileName = FPaths::SetExtension( SaveFileName, CleanExtension );
|
|
}
|
|
|
|
SaveFilePath /= SaveFileName;
|
|
}
|
|
AddOutFilename( SaveFilePath );
|
|
|
|
::CoTaskMemFree( pFilePath );
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
IFileOpenDialog* FileOpenDialog = static_cast<IFileOpenDialog*>( FileDialog.Get() );
|
|
|
|
TComPtr<IShellItemArray> Results;
|
|
if ( SUCCEEDED( FileOpenDialog->GetResults( &Results ) ) )
|
|
{
|
|
DWORD NumResults = 0;
|
|
Results->GetCount( &NumResults );
|
|
for ( DWORD ResultIndex = 0; ResultIndex < NumResults; ++ResultIndex )
|
|
{
|
|
TComPtr<IShellItem> Result;
|
|
if ( SUCCEEDED( Results->GetItemAt( ResultIndex, &Result ) ) )
|
|
{
|
|
PWSTR pFilePath = nullptr;
|
|
if ( SUCCEEDED( Result->GetDisplayName( SIGDN_FILESYSPATH, &pFilePath ) ) )
|
|
{
|
|
bSuccess = true;
|
|
AddOutFilename( pFilePath );
|
|
::CoTaskMemFree( pFilePath );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return bSuccess;
|
|
}
|
|
|
|
bool SaveFileDialog( const void* ParentWindowHandle, const FString& DialogTitle, const FString& DefaultPath, const FString& DefaultFile, const FString& FileTypes, uint32 Flags, TArray<FString>& OutFilenames )
|
|
{
|
|
int32 DummyFilterIndex = 0;
|
|
return FileDialogShared( true, ParentWindowHandle, DialogTitle, DefaultPath, DefaultFile, FileTypes, Flags, OutFilenames, DummyFilterIndex );
|
|
}
|
|
|
|
#else
|
|
|
|
bool SaveFileDialog( const void* ParentWindowHandle, const FString& DialogTitle, const FString& DefaultPath, const FString& DefaultFile, const FString& FileTypes, uint32 Flags, TArray<FString>& OutFilenames )
|
|
{
|
|
IDesktopPlatform* DesktopPlatform = FDesktopPlatformModule::Get();
|
|
return DesktopPlatform->SaveFileDialog( ParentWindowHandle, DialogTitle, DefaultPath, DefaultFile, FileTypes, Flags, OutFilenames );
|
|
}
|
|
#endif
|
|
|
|
FReply SNewFilePathPicker::HandleBrowseButtonClicked()
|
|
{
|
|
IDesktopPlatform* DesktopPlatform = FDesktopPlatformModule::Get();
|
|
|
|
if (DesktopPlatform == nullptr)
|
|
{
|
|
return FReply::Handled();
|
|
}
|
|
|
|
const FString DefaultPath = BrowseDirectory.IsSet()
|
|
? BrowseDirectory.Get()
|
|
: FPaths::GetPath(FilePath.Get());
|
|
|
|
// show the file browse dialog
|
|
TSharedPtr<SWindow> ParentWindow = FSlateApplication::Get().FindWidgetWindow(AsShared());
|
|
void* ParentWindowHandle = (ParentWindow.IsValid() && ParentWindow->GetNativeWindow().IsValid())
|
|
? ParentWindow->GetNativeWindow()->GetOSWindowHandle()
|
|
: nullptr;
|
|
|
|
TArray<FString> OutFiles;
|
|
|
|
// CG: Use SaveFileDialog instead of OpenFileDialog
|
|
if ( IsNewFile.Get() )
|
|
{
|
|
if ( SaveFileDialog( ParentWindowHandle, BrowseTitle.Get().ToString(), DefaultPath, TEXT( "" ), FileTypeFilter.Get(), EFileDialogFlags::None, OutFiles ) )
|
|
{
|
|
OnPathPicked.ExecuteIfBound( OutFiles[0] );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if ( DesktopPlatform->OpenFileDialog( ParentWindowHandle, BrowseTitle.Get().ToString(), DefaultPath, TEXT( "" ), FileTypeFilter.Get(), EFileDialogFlags::None, OutFiles ) )
|
|
{
|
|
OnPathPicked.ExecuteIfBound( OutFiles[0] );
|
|
}
|
|
}
|
|
|
|
return FReply::Handled();
|
|
}
|
|
|
|
|
|
FText SNewFilePathPicker::HandleTextBoxText() const
|
|
{
|
|
return FText::FromString(FilePath.Get());
|
|
}
|
|
|
|
|
|
void SNewFilePathPicker::HandleTextBoxTextCommitted( const FText& NewText, ETextCommit::Type CommitInfo )
|
|
{
|
|
OnPathPicked.ExecuteIfBound(NewText.ToString());
|
|
}
|
|
|
|
|
|
|
|
|
|
#undef LOCTEXT_NAMESPACE
|