76 lines
2.0 KiB
C++
76 lines
2.0 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "Containers/Map.h"
|
|
#include "UObject/NameTypes.h"
|
|
#include "Templates/SharedPointer.h"
|
|
#include "Framework/Commands/UICommandInfo.h"
|
|
#include "Framework/MultiBox/MultiBoxBuilder.h"
|
|
#include "Framework/Commands/Commands.h"
|
|
|
|
class FEditorViewportClient;
|
|
class UToolMenu;
|
|
struct FToolMenuSection;
|
|
|
|
class UNREALED_API FRayTracingDebugVisualizationMenuCommands : public TCommands<FRayTracingDebugVisualizationMenuCommands>
|
|
{
|
|
public:
|
|
enum class FVisualizationType : uint8
|
|
{
|
|
Overview,
|
|
Standard,
|
|
Performance,
|
|
Timing,
|
|
Other,
|
|
};
|
|
|
|
struct FVisualizationRecord
|
|
{
|
|
FName Name;
|
|
TSharedPtr<FUICommandInfo> Command;
|
|
FVisualizationType Type;
|
|
|
|
FVisualizationRecord()
|
|
: Name()
|
|
, Command()
|
|
, Type(FVisualizationType::Overview)
|
|
{
|
|
}
|
|
};
|
|
|
|
typedef TMultiMap<FName, FVisualizationRecord> TVisualizationModeCommandMap;
|
|
typedef TVisualizationModeCommandMap::TConstIterator TCommandConstIterator;
|
|
|
|
FRayTracingDebugVisualizationMenuCommands();
|
|
|
|
TCommandConstIterator CreateCommandConstIterator() const;
|
|
|
|
UE_DEPRECATED(5.6, "Please use the version taking UToolMenu* as argument")
|
|
static void BuildVisualisationSubMenu(FMenuBuilder& Menu);
|
|
|
|
static void BuildVisualisationSubMenu(UToolMenu* InMenu);
|
|
|
|
virtual void RegisterCommands() override;
|
|
|
|
void BindCommands(FUICommandList& CommandList, const TSharedPtr<FEditorViewportClient>& Client) const;
|
|
|
|
inline bool IsPopulated() const
|
|
{
|
|
return CommandMap.Num() > 0;
|
|
}
|
|
|
|
private:
|
|
void BuildCommandMap();
|
|
|
|
/** Remove this function when removing deprecated BuildVisualisationSubMenu(FMenuBuilder& Menu) */
|
|
bool AddCommandTypeToMenu(FMenuBuilder& Menu, const FVisualizationType Type) const;
|
|
|
|
bool AddCommandTypeToSection(FToolMenuSection& InSection, const FVisualizationType Type) const;
|
|
|
|
static void ChangeRayTracingDebugVisualizationMode(TWeakPtr<FEditorViewportClient> WeakClient, FName InName);
|
|
static bool IsRayTracingDebugVisualizationModeSelected(TWeakPtr<FEditorViewportClient> WeakClient, FName InName);
|
|
|
|
TVisualizationModeCommandMap CommandMap;
|
|
};
|