327 lines
14 KiB
C++
327 lines
14 KiB
C++
|
// Copyright Epic Games, Inc. All Rights Reserved.
|
||
|
|
|
||
|
|
#include "Fbx/SSceneImportStaticMeshListView.h"
|
||
|
|
#include "Widgets/SOverlay.h"
|
||
|
|
#include "Textures/SlateIcon.h"
|
||
|
|
#include "Framework/Commands/UIAction.h"
|
||
|
|
#include "Widgets/Images/SImage.h"
|
||
|
|
#include "Framework/MultiBox/MultiBoxBuilder.h"
|
||
|
|
#include "Widgets/Input/SCheckBox.h"
|
||
|
|
#include "EditorStyleSet.h"
|
||
|
|
#include "Styling/SlateIconFinder.h"
|
||
|
|
#include "SFbxSceneOptionWindow.h"
|
||
|
|
#include "FbxImporter.h"
|
||
|
|
|
||
|
|
#define LOCTEXT_NAMESPACE "SFbxSceneStaticMeshListView"
|
||
|
|
|
||
|
|
namespace FbxSceneImportStaticMesh
|
||
|
|
{
|
||
|
|
const FName SceneImportCheckBoxSelectionHeaderIdName(TEXT("CheckBoxSelectionHeaderId"));
|
||
|
|
const FName SceneImportClassIconHeaderIdName(TEXT("ClassIconHeaderId"));
|
||
|
|
const FName SceneImportAssetNameHeaderIdName(TEXT("AssetNameHeaderId"));
|
||
|
|
const FName SceneImportContentPathHeaderIdName(TEXT("ContentPathHeaderId"));
|
||
|
|
const FName SceneImportOptionsNameHeaderIdName(TEXT("OptionsNameHeaderId"));
|
||
|
|
} // namespace FbxSceneImportStaticMesh
|
||
|
|
|
||
|
|
/** The item used for visualizing the class in the tree. */
|
||
|
|
class SFbxMeshItemTableListViewRow: public SMultiColumnTableRow<FbxMeshInfoPtr>
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
SLATE_BEGIN_ARGS(SFbxMeshItemTableListViewRow)
|
||
|
|
: _FbxMeshInfo(nullptr), _GlobalImportSettings(nullptr)
|
||
|
|
{}
|
||
|
|
|
||
|
|
/** The item content. */
|
||
|
|
SLATE_ARGUMENT(FbxMeshInfoPtr, FbxMeshInfo)
|
||
|
|
SLATE_ARGUMENT(UnFbx::FBXImportOptions*, GlobalImportSettings)
|
||
|
|
SLATE_END_ARGS()
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Construct the widget
|
||
|
|
*
|
||
|
|
* @param InArgs A declaration from which to construct the widget
|
||
|
|
*/
|
||
|
|
void Construct(const FArguments& InArgs, const TSharedRef<STableViewBase>& InOwnerTableView)
|
||
|
|
{
|
||
|
|
FbxMeshInfo = InArgs._FbxMeshInfo;
|
||
|
|
GlobalImportSettings = InArgs._GlobalImportSettings;
|
||
|
|
|
||
|
|
// This is suppose to always be valid
|
||
|
|
check(FbxMeshInfo.IsValid());
|
||
|
|
check(GlobalImportSettings != nullptr);
|
||
|
|
|
||
|
|
SMultiColumnTableRow<FbxMeshInfoPtr>::Construct(
|
||
|
|
FSuperRowType::FArguments()
|
||
|
|
.Style(FEditorStyle::Get(), "DataTableEditor.CellListViewRow"),
|
||
|
|
InOwnerTableView);
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Overridden from SMultiColumnTableRow. Generates a widget for this column of the list view. */
|
||
|
|
virtual TSharedRef<SWidget> GenerateWidgetForColumn(const FName& ColumnName) override
|
||
|
|
{
|
||
|
|
if (ColumnName == FbxSceneImportStaticMesh::SceneImportCheckBoxSelectionHeaderIdName)
|
||
|
|
{
|
||
|
|
return SNew(SBox)
|
||
|
|
.HAlign(HAlign_Center)
|
||
|
|
[SNew(SCheckBox)
|
||
|
|
.OnCheckStateChanged(this, &SFbxMeshItemTableListViewRow::OnItemCheckChanged)
|
||
|
|
.IsChecked(this, &SFbxMeshItemTableListViewRow::IsItemChecked)];
|
||
|
|
}
|
||
|
|
else if (ColumnName == FbxSceneImportStaticMesh::SceneImportClassIconHeaderIdName)
|
||
|
|
{
|
||
|
|
UClass* IconClass = FbxMeshInfo->GetType();
|
||
|
|
const FSlateBrush* ClassIcon = FSlateIconFinder::FindIconBrushForClass(IconClass);
|
||
|
|
|
||
|
|
TSharedRef<SOverlay> IconContent = SNew(SOverlay) + SOverlay::Slot()
|
||
|
|
.HAlign(HAlign_Right)
|
||
|
|
.VAlign(VAlign_Center)
|
||
|
|
[SNew(SImage)
|
||
|
|
.Image(ClassIcon)];
|
||
|
|
|
||
|
|
// Add the component mobility icon
|
||
|
|
IconContent->AddSlot()
|
||
|
|
.HAlign(HAlign_Left)
|
||
|
|
[SNew(SImage)
|
||
|
|
.Image(this, &SFbxMeshItemTableListViewRow::GetBrushForOverrideIcon)];
|
||
|
|
return IconContent;
|
||
|
|
}
|
||
|
|
else if (ColumnName == FbxSceneImportStaticMesh::SceneImportAssetNameHeaderIdName)
|
||
|
|
{
|
||
|
|
return SNew(STextBlock)
|
||
|
|
.Text(FText::FromString(FbxMeshInfo->Name))
|
||
|
|
.ToolTipText(FText::FromString(FbxMeshInfo->Name));
|
||
|
|
}
|
||
|
|
else if (ColumnName == FbxSceneImportStaticMesh::SceneImportContentPathHeaderIdName)
|
||
|
|
{
|
||
|
|
return SNew(STextBlock)
|
||
|
|
.Text(this, &SFbxMeshItemTableListViewRow::GetAssetFullName)
|
||
|
|
.ColorAndOpacity(this, &SFbxMeshItemTableListViewRow::GetContentPathTextColor)
|
||
|
|
.ToolTipText(this, &SFbxMeshItemTableListViewRow::GetAssetFullName);
|
||
|
|
}
|
||
|
|
else if (ColumnName == FbxSceneImportStaticMesh::SceneImportOptionsNameHeaderIdName)
|
||
|
|
{
|
||
|
|
return SNew(STextBlock)
|
||
|
|
.Text(this, &SFbxMeshItemTableListViewRow::GetAssetOptionName)
|
||
|
|
.ToolTipText(this, &SFbxMeshItemTableListViewRow::GetAssetOptionName);
|
||
|
|
}
|
||
|
|
else if (ColumnName == FbxSceneBaseListViewColumn::PivotColumnId)
|
||
|
|
{
|
||
|
|
return SNew(STextBlock)
|
||
|
|
.Text(this, &SFbxMeshItemTableListViewRow::GetAssetPivotNodeName)
|
||
|
|
.ToolTipText(this, &SFbxMeshItemTableListViewRow::GetAssetPivotNodeName);
|
||
|
|
}
|
||
|
|
|
||
|
|
return SNullWidget::NullWidget;
|
||
|
|
}
|
||
|
|
|
||
|
|
const FSlateBrush* GetBrushForOverrideIcon() const
|
||
|
|
{
|
||
|
|
if (UFbxSceneImportFactory::DefaultOptionName.Compare(FbxMeshInfo->OptionName) != 0)
|
||
|
|
{
|
||
|
|
return FEditorStyle::GetBrush("FBXIcon.ImportOptionsOverride");
|
||
|
|
}
|
||
|
|
return FEditorStyle::GetBrush("FBXIcon.ImportOptionsDefault");
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
void OnItemCheckChanged(ECheckBoxState CheckType)
|
||
|
|
{
|
||
|
|
if (!FbxMeshInfo.IsValid())
|
||
|
|
return;
|
||
|
|
FbxMeshInfo->bImportAttribute = CheckType == ECheckBoxState::Checked;
|
||
|
|
}
|
||
|
|
|
||
|
|
ECheckBoxState IsItemChecked() const
|
||
|
|
{
|
||
|
|
return FbxMeshInfo->bImportAttribute ? ECheckBoxState::Checked : ECheckBoxState::Unchecked;
|
||
|
|
}
|
||
|
|
|
||
|
|
FSlateColor GetContentPathTextColor() const
|
||
|
|
{
|
||
|
|
return FbxMeshInfo->bOverridePath ? FLinearColor(0.75f, 0.75f, 0.0f, 1.0f) : FSlateColor::UseForeground();
|
||
|
|
}
|
||
|
|
|
||
|
|
FText GetAssetFullName() const
|
||
|
|
{
|
||
|
|
return FText::FromString(FbxMeshInfo->GetFullImportName());
|
||
|
|
}
|
||
|
|
|
||
|
|
FText GetAssetOptionName() const
|
||
|
|
{
|
||
|
|
return FText::FromString(FbxMeshInfo->OptionName);
|
||
|
|
}
|
||
|
|
|
||
|
|
FText GetAssetPivotNodeName() const
|
||
|
|
{
|
||
|
|
return GlobalImportSettings->bBakePivotInVertex ? FText::FromString(FbxMeshInfo->PivotNodeName) : FText::FromString(TEXT("-"));
|
||
|
|
}
|
||
|
|
|
||
|
|
/** The node info to build the tree view row from. */
|
||
|
|
FbxMeshInfoPtr FbxMeshInfo;
|
||
|
|
UnFbx::FBXImportOptions* GlobalImportSettings;
|
||
|
|
};
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////
|
||
|
|
// Static Mesh List
|
||
|
|
|
||
|
|
SFbxSceneStaticMeshListView::~SFbxSceneStaticMeshListView()
|
||
|
|
{
|
||
|
|
SceneInfo = nullptr;
|
||
|
|
GlobalImportSettings = nullptr;
|
||
|
|
SceneImportOptionsStaticMeshDisplay = nullptr;
|
||
|
|
CurrentMeshImportOptions = nullptr;
|
||
|
|
FbxMeshesArray.Empty();
|
||
|
|
SceneImportOptionsStaticMeshDisplay = nullptr;
|
||
|
|
|
||
|
|
OverrideNameOptions = nullptr;
|
||
|
|
OverrideNameOptionsMap = nullptr;
|
||
|
|
OptionComboBox = nullptr;
|
||
|
|
DefaultOptionNamePtr = nullptr;
|
||
|
|
}
|
||
|
|
|
||
|
|
void SFbxSceneStaticMeshListView::Construct(const SFbxSceneStaticMeshListView::FArguments& InArgs)
|
||
|
|
{
|
||
|
|
SceneInfo = InArgs._SceneInfo;
|
||
|
|
GlobalImportSettings = InArgs._GlobalImportSettings;
|
||
|
|
OverrideNameOptions = InArgs._OverrideNameOptions;
|
||
|
|
OverrideNameOptionsMap = InArgs._OverrideNameOptionsMap;
|
||
|
|
SceneImportOptionsStaticMeshDisplay = InArgs._SceneImportOptionsStaticMeshDisplay;
|
||
|
|
|
||
|
|
check(SceneInfo.IsValid());
|
||
|
|
check(GlobalImportSettings != nullptr);
|
||
|
|
check(OverrideNameOptions != nullptr);
|
||
|
|
check(OverrideNameOptionsMap != nullptr);
|
||
|
|
check(SceneImportOptionsStaticMeshDisplay != nullptr);
|
||
|
|
|
||
|
|
SFbxSceneOptionWindow::CopyStaticMeshOptionsToFbxOptions(GlobalImportSettings, SceneImportOptionsStaticMeshDisplay);
|
||
|
|
// Set the default options to the current global import settings
|
||
|
|
GlobalImportSettings->bTransformVertexToAbsolute = false;
|
||
|
|
GlobalImportSettings->StaticMeshLODGroup = NAME_None;
|
||
|
|
CurrentMeshImportOptions = GlobalImportSettings;
|
||
|
|
|
||
|
|
bool bNameExist = false;
|
||
|
|
for (TSharedPtr<FString> OverrideName: (*OverrideNameOptions))
|
||
|
|
{
|
||
|
|
if (OverrideName.Get()->Compare(UFbxSceneImportFactory::DefaultOptionName) == 0)
|
||
|
|
{
|
||
|
|
DefaultOptionNamePtr = OverrideName;
|
||
|
|
bNameExist = true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!bNameExist)
|
||
|
|
{
|
||
|
|
DefaultOptionNamePtr = MakeShareable(new FString(UFbxSceneImportFactory::DefaultOptionName));
|
||
|
|
OverrideNameOptions->Add(DefaultOptionNamePtr);
|
||
|
|
OverrideNameOptionsMap->Add(UFbxSceneImportFactory::DefaultOptionName, GlobalImportSettings);
|
||
|
|
}
|
||
|
|
|
||
|
|
for (auto MeshInfoIt = SceneInfo->MeshInfo.CreateIterator(); MeshInfoIt; ++MeshInfoIt)
|
||
|
|
{
|
||
|
|
FbxMeshInfoPtr MeshInfo = (*MeshInfoIt);
|
||
|
|
|
||
|
|
if (!MeshInfo->bIsSkelMesh && !MeshInfo->IsLod && !MeshInfo->IsCollision)
|
||
|
|
{
|
||
|
|
MeshInfo->OptionName = UFbxSceneImportFactory::DefaultOptionName;
|
||
|
|
FbxMeshesArray.Add(MeshInfo);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
SListView::Construct(
|
||
|
|
SListView::FArguments()
|
||
|
|
.ListItemsSource(&FbxMeshesArray)
|
||
|
|
.SelectionMode(ESelectionMode::Multi)
|
||
|
|
.OnGenerateRow(this, &SFbxSceneStaticMeshListView::OnGenerateRowFbxSceneListView)
|
||
|
|
.OnContextMenuOpening(this, &SFbxSceneStaticMeshListView::OnOpenContextMenu)
|
||
|
|
.OnSelectionChanged(this, &SFbxSSceneBaseMeshListView::OnSelectionChanged)
|
||
|
|
.HeaderRow(
|
||
|
|
SNew(SHeaderRow)
|
||
|
|
|
||
|
|
+ SHeaderRow::Column(FbxSceneImportStaticMesh::SceneImportCheckBoxSelectionHeaderIdName)
|
||
|
|
.FixedWidth(25)
|
||
|
|
.DefaultLabel(FText::GetEmpty())
|
||
|
|
[SNew(SCheckBox)
|
||
|
|
.HAlign(HAlign_Center)
|
||
|
|
.OnCheckStateChanged(this, &SFbxSceneStaticMeshListView::OnToggleSelectAll)]
|
||
|
|
|
||
|
|
+ SHeaderRow::Column(FbxSceneImportStaticMesh::SceneImportClassIconHeaderIdName)
|
||
|
|
.FixedWidth(20)
|
||
|
|
.DefaultLabel(FText::GetEmpty())
|
||
|
|
|
||
|
|
+ SHeaderRow::Column(FbxSceneImportStaticMesh::SceneImportAssetNameHeaderIdName)
|
||
|
|
.FillWidth(250)
|
||
|
|
.HAlignCell(EHorizontalAlignment::HAlign_Left)
|
||
|
|
.DefaultLabel(LOCTEXT("AssetNameHeaderName", "Asset Name"))
|
||
|
|
|
||
|
|
+ SHeaderRow::Column(FbxSceneImportStaticMesh::SceneImportOptionsNameHeaderIdName)
|
||
|
|
.FillWidth(200)
|
||
|
|
.HAlignCell(EHorizontalAlignment::HAlign_Left)
|
||
|
|
.DefaultLabel(LOCTEXT("OptionsNameHeaderName", "Options Name"))
|
||
|
|
|
||
|
|
/* + SHeaderRow::Column(SceneImportContentPathHeaderIdName)
|
||
|
|
.FillWidth(300)
|
||
|
|
.HAlignCell(EHorizontalAlignment::HAlign_Left)
|
||
|
|
.DefaultLabel(LOCTEXT("ContentPathHeaderName", "Content Path"))*/
|
||
|
|
));
|
||
|
|
}
|
||
|
|
|
||
|
|
TSharedRef<ITableRow> SFbxSceneStaticMeshListView::OnGenerateRowFbxSceneListView(FbxMeshInfoPtr Item, const TSharedRef<STableViewBase>& OwnerTable)
|
||
|
|
{
|
||
|
|
TSharedRef<SFbxMeshItemTableListViewRow> ReturnRow = SNew(SFbxMeshItemTableListViewRow, OwnerTable)
|
||
|
|
.FbxMeshInfo(Item)
|
||
|
|
.GlobalImportSettings(GlobalImportSettings);
|
||
|
|
return ReturnRow;
|
||
|
|
}
|
||
|
|
|
||
|
|
TSharedPtr<SWidget> SFbxSceneStaticMeshListView::OnOpenContextMenu()
|
||
|
|
{
|
||
|
|
TArray<FbxMeshInfoPtr> SelectedFbxMeshInfos;
|
||
|
|
int32 SelectCount = GetSelectedItems(SelectedFbxMeshInfos);
|
||
|
|
// Build up the menu for a selection
|
||
|
|
const bool bCloseAfterSelection = true;
|
||
|
|
FMenuBuilder MenuBuilder(bCloseAfterSelection, TSharedPtr<FUICommandList>());
|
||
|
|
|
||
|
|
const FSlateIcon PlusIcon(FEditorStyle::GetStyleSetName(), "Plus");
|
||
|
|
MenuBuilder.AddMenuEntry(LOCTEXT("CheckForImport", "Add Selection To Import"), FText(), PlusIcon, FUIAction(FExecuteAction::CreateSP(this, &SFbxSceneStaticMeshListView::AddSelectionToImport)));
|
||
|
|
const FSlateIcon MinusIcon(FEditorStyle::GetStyleSetName(), "PropertyWindow.Button_RemoveFromArray");
|
||
|
|
MenuBuilder.AddMenuEntry(LOCTEXT("UncheckForImport", "Remove Selection From Import"), FText(), MinusIcon, FUIAction(FExecuteAction::CreateSP(this, &SFbxSceneStaticMeshListView::RemoveSelectionFromImport)));
|
||
|
|
|
||
|
|
AddBakePivotMenu(MenuBuilder);
|
||
|
|
|
||
|
|
if (OverrideNameOptions->Num() > 0)
|
||
|
|
{
|
||
|
|
MenuBuilder.BeginSection("FbxScene_SM_OptionsSection", LOCTEXT("FbxScene_SM_Options", "Options:"));
|
||
|
|
{
|
||
|
|
for (auto OptionName: (*OverrideNameOptions))
|
||
|
|
{
|
||
|
|
MenuBuilder.AddMenuEntry(FText::FromString(*OptionName.Get()), FText(), FSlateIcon(), FUIAction(FExecuteAction::CreateSP(this, &SFbxSceneStaticMeshListView::AssignToOptions, *OptionName.Get())));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
MenuBuilder.EndSection();
|
||
|
|
}
|
||
|
|
|
||
|
|
return MenuBuilder.MakeWidget();
|
||
|
|
}
|
||
|
|
|
||
|
|
void SFbxSceneStaticMeshListView::OnFinishedChangingProperties(const FPropertyChangedEvent& PropertyChangedEvent)
|
||
|
|
{
|
||
|
|
SFbxSceneOptionWindow::CopyStaticMeshOptionsToFbxOptions(CurrentMeshImportOptions, SceneImportOptionsStaticMeshDisplay);
|
||
|
|
}
|
||
|
|
|
||
|
|
void SFbxSceneStaticMeshListView::OnChangedOverrideOptions(TSharedPtr<FString> ItemSelected, ESelectInfo::Type SelectInfo)
|
||
|
|
{
|
||
|
|
check(ItemSelected.IsValid());
|
||
|
|
if (ItemSelected.Get()->Compare(UFbxSceneImportFactory::DefaultOptionName) == 0)
|
||
|
|
{
|
||
|
|
CurrentMeshImportOptions = GlobalImportSettings;
|
||
|
|
}
|
||
|
|
else if (OverrideNameOptionsMap->Contains(*(ItemSelected.Get())))
|
||
|
|
{
|
||
|
|
CurrentMeshImportOptions = *OverrideNameOptionsMap->Find(*(ItemSelected.Get()));
|
||
|
|
}
|
||
|
|
SFbxSceneOptionWindow::CopyFbxOptionsToStaticMeshOptions(CurrentMeshImportOptions, SceneImportOptionsStaticMeshDisplay);
|
||
|
|
}
|
||
|
|
|
||
|
|
#undef LOCTEXT_NAMESPACE
|