UnrealEngineUWP/Engine/Source/Editor/DetailCustomizations/Private/ComponentsTree.h

// Copyright 1998-2014 Epic Games, Inc. All Rights Reserved.


#pragma once

typedef TSharedPtr<class FComponentTreeNode> FComponentTreeNodePtrType;
typedef STreeView<FComponentTreeNodePtrType> SComponentTreeType;

//////////////////////////////////////////////////////////////////////////

class FComponentTreeNode : public TSharedFromThis<FComponentTreeNode>
{
public:
	FComponentTreeNode(UActorComponent* InComponent);

	TWeakObjectPtr<UActorComponent>	Component;

	TArray<FComponentTreeNodePtrType> ChildNodes;
};

//////////////////////////////////////////////////////////////////////////

class SComponentRowWidget : public SCompoundWidget
{
public:
	SLATE_BEGIN_ARGS( SComponentRowWidget ){}
	SLATE_END_ARGS()

	void Construct( const FArguments& InArgs, FComponentTreeNodePtrType InNodePtr );

protected:
	/** Pointer to node we represent */
	TWeakPtr<FComponentTreeNode> NodePtr;
};

//////////////////////////////////////////////////////////////////////////

class SComponentsTree : public SCompoundWidget
{

public:

	SLATE_BEGIN_ARGS( SComponentsTree ){}

	SLATE_END_ARGS()

	void Construct( const FArguments& InArgs, AActor* InActor );

	void UpdateTree();

	// Tree delegates
	TSharedRef<ITableRow> MakeTableRowWidget( FComponentTreeNodePtrType InNodePtr, const TSharedRef<STableViewBase>& OwnerTable );
	void OnGetChildrenForTree( FComponentTreeNodePtrType InNodePtr, TArray<FComponentTreeNodePtrType>& OutChildren );
	void OnTreeSelectionChanged(FComponentTreeNodePtrType InSelectedNodePtr, ESelectInfo::Type SelectInfo);

	void OnSelectedCompClass(TSubclassOf<UActorComponent> CompClass);

private:

	TWeakObjectPtr<AActor>				Actor;

	TArray<FComponentTreeNodePtrType>	RootNodes;

	TSharedPtr<SComponentTreeType>		TreeWidget;

	TSharedPtr<class IDetailsView>		PropertyView;
};