// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include "Containers/Array.h"
#include "Input/Reply.h"
#include "SSceneOutliner.h"
#include "Templates/SharedPointer.h"

struct FSceneOutlinerInitializationOptions;

class SDataLayerOutliner : public SSceneOutliner
{
public:
	void Construct(const FArguments& InArgs, const FSceneOutlinerInitializationOptions& InitOptions)
	{
		SSceneOutliner::Construct(InArgs, InitOptions);
	}

	SDataLayerOutliner() {}
	virtual ~SDataLayerOutliner() {}

	void CustomAddToToolbar(TSharedPtr<class SHorizontalBox> Toolbar) override;
private:
	bool CanAddSelectedActorsToSelectedDataLayers() const;
	bool CanRemoveSelectedActorsFromSelectedDataLayers() const;
	FReply OnAddSelectedActorsToSelectedDataLayersClicked();
	FReply OnRemoveSelectedActorsFromSelectedDataLayersClicked();
	TArray<class UDataLayerInstance*> GetSelectedDataLayers() const;
};