Ability/Ability/Plugins/EzAbility/Source/EzAbilityEditor/Public/SEzAbleTreeViewWidget.h

// Fill out your copyright notice in the Description page of Project Settings.

#pragma once

#include "CoreMinimal.h"
#include "Widgets/SCompoundWidget.h"
#include "Widgets/Layout/SScrollBox.h"
#include "EzAbilityState.h"

/**
 * 
 */
class EZABILITYEDITOR_API SEzAbleTreeViewWidget : public SCompoundWidget
{
public:
	SLATE_BEGIN_ARGS(SEzAbleTreeViewWidget)
	{}
	SLATE_END_ARGS()

	/** Constructs this widget with InArgs */
	void Construct(const FArguments& InArgs);
	FReply HandleAddStateButton();

	// Treeview handlers
	TSharedRef<ITableRow> HandleGenerateRow(TWeakObjectPtr<UEzAbilityState> InState, const TSharedRef<STableViewBase>& InOwnerTableView);

public:

	TSharedPtr<STreeView<TWeakObjectPtr<UEzAbilityState>>> TreeView;
	TSharedPtr<SScrollBox> ViewBox;
};