88 lines
2.9 KiB
C++
88 lines
2.9 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
#pragma once
|
|
|
|
#include "Containers/Array.h"
|
|
#include "Containers/BitArray.h"
|
|
#include "Containers/Map.h"
|
|
#include "Containers/Set.h"
|
|
#include "Containers/SparseArray.h"
|
|
#include "Delegates/Delegate.h"
|
|
#include "Fbx/SSceneBaseMeshListView.h"
|
|
#include "Fbx/SSceneImportNodeTreeView.h"
|
|
#include "HAL/PlatformCrt.h"
|
|
#include "Input/Reply.h"
|
|
#include "Misc/Optional.h"
|
|
#include "Styling/SlateTypes.h"
|
|
#include "Templates/SharedPointer.h"
|
|
#include "Templates/TypeHash.h"
|
|
#include "Types/SlateEnums.h"
|
|
#include "Widgets/DeclarativeSyntaxSupport.h"
|
|
#include "Widgets/Views/STableViewBase.h"
|
|
#include "Widgets/Views/STreeView.h"
|
|
|
|
class FFbxAttributeInfo;
|
|
class FFbxSceneInfo;
|
|
class ITableRow;
|
|
class SWidget;
|
|
|
|
struct FTreeNodeValue
|
|
{
|
|
public:
|
|
FbxNodeInfoPtr CurrentNode;
|
|
FbxNodeInfoPtr OriginalNode;
|
|
};
|
|
|
|
class SFbxReimportSceneTreeView : public STreeView<FbxNodeInfoPtr>
|
|
{
|
|
public:
|
|
~SFbxReimportSceneTreeView();
|
|
SLATE_BEGIN_ARGS(SFbxReimportSceneTreeView)
|
|
: _SceneInfo(nullptr)
|
|
, _SceneInfoOriginal(nullptr)
|
|
, _NodeStatusMap(nullptr)
|
|
{}
|
|
SLATE_ARGUMENT(TSharedPtr<FFbxSceneInfo>, SceneInfo)
|
|
SLATE_ARGUMENT(TSharedPtr<FFbxSceneInfo>, SceneInfoOriginal)
|
|
SLATE_ARGUMENT(FbxSceneReimportStatusMapPtr, NodeStatusMap)
|
|
SLATE_END_ARGS()
|
|
|
|
/** Construct this widget */
|
|
void Construct(const FArguments& InArgs);
|
|
TSharedRef< ITableRow > OnGenerateRowFbxSceneTreeView(FbxNodeInfoPtr Item, const TSharedRef< STableViewBase >& OwnerTable);
|
|
void OnGetChildrenFbxSceneTreeView(FbxNodeInfoPtr InParent, TArray< FbxNodeInfoPtr >& OutChildren);
|
|
|
|
void OnToggleSelectAll(ECheckBoxState CheckType);
|
|
FReply OnExpandAll();
|
|
FReply OnCollapseAll();
|
|
|
|
static void FillNodeStatusMap(FbxSceneReimportStatusMapPtr NodeStatusMap, TSharedPtr<FFbxSceneInfo> SceneInfo, TSharedPtr<FFbxSceneInfo> SceneInfoOriginal);
|
|
|
|
protected:
|
|
TSharedPtr<FFbxSceneInfo> SceneInfo;
|
|
TSharedPtr<FFbxSceneInfo> SceneInfoOriginal;
|
|
FbxSceneReimportStatusMapPtr NodeStatusMap;
|
|
|
|
|
|
/** the elements we show in the tree view */
|
|
TArray<FbxNodeInfoPtr> FbxRootNodeArray;
|
|
|
|
/** Open a context menu for the current selection */
|
|
TSharedPtr<SWidget> OnOpenContextMenu();
|
|
void AddSelectionToImport();
|
|
void RemoveSelectionFromImport();
|
|
void SetSelectionImportState(bool MarkForImport);
|
|
void OnSelectionChanged(FbxNodeInfoPtr Item, ESelectInfo::Type SelectionType);
|
|
|
|
void GotoAsset(TSharedPtr<FFbxAttributeInfo> AssetAttribute);
|
|
void RecursiveSetImport(FbxNodeInfoPtr NodeInfoPtr, bool ImportStatus);
|
|
|
|
// Internal structure and function to create the tree view status data
|
|
TMap<FbxNodeInfoPtr, TSharedPtr<FTreeNodeValue>> NodeTreeData;
|
|
|
|
static void FillNodeStatusMapInternal(FbxSceneReimportStatusMapPtr NodeStatusMap
|
|
, TSharedPtr<FFbxSceneInfo> SceneInfo
|
|
, TSharedPtr<FFbxSceneInfo> SceneInfoOriginal
|
|
, TMap<FbxNodeInfoPtr, TSharedPtr<FTreeNodeValue>>* NodeTreeDataPtr
|
|
, TArray<FbxNodeInfoPtr>* FbxRootNodeArrayPtr);
|
|
};
|