Files
UnrealEngine/Engine/Source/Editor/LevelInstanceEditor/Private/LevelInstanceEditorModule.h
2025-05-18 13:04:45 +08:00

134 lines
4.5 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include "CoreMinimal.h"
#include "LevelInstance/ILevelInstanceEditorModule.h"
#include "UObject/GCObject.h"
#include "PropertyEditorArchetypePolicy.h"
#include "PropertyEditorEditConstPolicy.h"
#include "Tools/Modes.h"
class AActor;
class ULevel;
class IInputBehaviorSource;
enum class EMapChangeType : uint8;
class ILevelEditor;
class ISceneOutliner;
class ISceneOutlinerColumn;
/**
* The module holding all of the UI related pieces for LevelInstance management
*/
class FLevelInstanceEditorModule : public ILevelInstanceEditorModule, public FGCObject
{
public:
virtual ~FLevelInstanceEditorModule(){}
/**
* Called right after the module DLL has been loaded and the module object has been created
*/
virtual void StartupModule();
/**
* Called before the module is unloaded, right before the module object is destroyed.
*/
virtual void ShutdownModule();
virtual void BroadcastTryExitEditorMode() override;
DECLARE_DERIVED_EVENT(FLevelInstanceEditorModule, ILevelInstanceEditorModule::FExitEditorModeEvent, FExitEditorModeEvent);
virtual FExitEditorModeEvent& OnExitEditorMode() override { return ExitEditorModeEvent; }
DECLARE_DERIVED_EVENT(FLevelInstanceEditorModule, ILevelInstanceEditorModule::FTryExitEditorModeEvent, FTryExitEditorModeEvent);
virtual FTryExitEditorModeEvent& OnTryExitEditorMode() override { return TryExitEditorModeEvent; }
virtual bool IsEditInPlaceStreamingEnabled() const override;
virtual bool IsSubSelectionEnabled() const override;
virtual void AddReferencedObjects(FReferenceCollector& Collector) override;
virtual FString GetReferencerName() const override { return TEXT("LevelInstanceEditorModule"); }
private:
virtual void UpdateEditorMode(bool bActivated) override;
void OnEditorModeIDChanged(const FEditorModeID& InModeID, bool bIsEnteringMode);
void OnLevelActorDeleted(AActor* Actor);
void CanMoveActorToLevel(const AActor* ActorToMove, const ULevel* DestLevel, bool& bOutCanMove);
void ExtendContextMenu();
void OnLevelEditorCreated(TSharedPtr<ILevelEditor> InLevelEditor);
void RegisterToFirstLevelEditor();
FName GetBrowseToLevelInstanceAsset(const UObject* Object) const;
void RegisterLevelInstanceColumn();
void UnregisterLevelInstanceColumn();
TSharedRef<ISceneOutlinerColumn> CreateLevelInstanceColumn(ISceneOutliner& SceneOutliner) const;
class FPropertyEditorPolicy : public PropertyEditorPolicy::IEditConstPolicy,
public PropertyEditorPolicy::IArchetypePolicy
{
public:
FPropertyEditorPolicy(ILevelInstanceEditorModule::IPropertyOverridePolicy* InPropertyOverridePolicy)
: PropertyOverridePolicy(InPropertyOverridePolicy)
{
check(PropertyOverridePolicy);
PropertyEditorPolicy::RegisterEditConstPolicy(this);
PropertyEditorPolicy::RegisterArchetypePolicy(this);
}
virtual ~FPropertyEditorPolicy()
{
PropertyEditorPolicy::UnregisterEditConstPolicy(this);
PropertyEditorPolicy::UnregisterArchetypePolicy(this);
}
virtual UObject* GetArchetypeForObject(const UObject* Object) const override
{
return PropertyOverridePolicy->GetArchetypeForObject(Object);
}
virtual bool CanEditProperty(const FEditPropertyChain& PropertyChain, const UObject* Object) const override
{
return PropertyOverridePolicy->CanEditProperty(PropertyChain, Object);
}
virtual bool CanEditProperty(const FProperty* Property, const UObject* Object) const override
{
return PropertyOverridePolicy->CanEditProperty(Property, Object);
}
ILevelInstanceEditorModule::IPropertyOverridePolicy* PropertyOverridePolicy = nullptr;
};
TUniquePtr<FPropertyEditorPolicy> PropertyEditorPolicy;
virtual bool IsPropertyEditConst(const FEditPropertyChain& PropertyChain, UObject* Object) override
{
return PropertyEditorPolicy::IsPropertyEditConst(PropertyChain, Object);
}
virtual bool IsPropertyEditConst(const FProperty* Property, UObject* Object) override
{
return PropertyEditorPolicy::IsPropertyEditConst(Property, Object);
}
virtual UObject* GetArchetype(const UObject* Object) override
{
return PropertyEditorPolicy::GetArchetype(Object);
}
virtual void SetPropertyOverridePolicy(ILevelInstanceEditorModule::IPropertyOverridePolicy* InPropertyOverridePolicy) override
{
PropertyEditorPolicy.Reset();
if (InPropertyOverridePolicy)
{
PropertyEditorPolicy = MakeUnique<FPropertyEditorPolicy>(InPropertyOverridePolicy);
}
}
FExitEditorModeEvent ExitEditorModeEvent;
FTryExitEditorModeEvent TryExitEditorModeEvent;
TScriptInterface<IInputBehaviorSource> DefaultBehaviorSource;
};