// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once

#include "CoreMinimal.h"
#include "Input/Reply.h"
#include "Widgets/DeclarativeSyntaxSupport.h"
#include "Widgets/Views/STableViewBase.h"
#include "Widgets/Views/STableRow.h"
#include "Widgets/Views/STreeView.h"
#include "Fbx/SSceneImportNodeTreeView.h"
#include "Fbx/SSceneBaseMeshListView.h"

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);
};