UnrealEngineUWP/Engine/Source/Editor/LevelInstanceEditor/Private/LevelInstanceEditorMode.h

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

#include "CoreMinimal.h"
#include "Tools/UEdMode.h"
#include "Framework/Commands/UICommandList.h"
#include "LevelInstanceEditorMode.generated.h"

UCLASS()
class ULevelInstanceEditorMode : public UEdMode
{
	GENERATED_BODY()
public:
	static FEditorModeID EM_LevelInstanceEditorModeId;

	/** Constructor */
	ULevelInstanceEditorMode();

	/** Destructor */
	virtual ~ULevelInstanceEditorMode();

	// Begin UEdMode
	virtual void Enter() override;
	virtual void Exit() override;
	virtual void CreateToolkit() override;
	
	virtual bool IsCompatibleWith(FEditorModeID OtherModeID) const override;

	/** Check to see if an actor can be selected in this mode - no side effects */
	virtual bool IsSelectionDisallowed(AActor* InActor, bool bInSelection) const override;
	/** Only accept saving on current asset */
	virtual bool IsOperationSupportedForCurrentAsset(EAssetOperation InOperation) const { return InOperation == EAssetOperation::Save; }
	// End UEdMode

private:
	void OnPreBeginPIE(bool bSimulate);
	void UpdateEngineShowFlags();
	virtual void BindCommands() override;

	void ExitModeCommand();
	void ToggleContextRestrictionCommand();
	bool IsContextRestrictionCommandEnabled() const;

	bool bContextRestriction;
};