Project02/ProjectFish/Source/ProjectFishEditor/Private/ProjectFishEditor.cpp

#include "ProjectFishEditor.h"

#include "AssetTypeActions_Base.h"
#include "DataTableRowSelectorCustomization.h"
#include "AssetActions/BagShapeAssetTypeAction.h"

#define LOCTEXT_NAMESPACE "FProjectFishEditorModule"

void FProjectFishEditorModule::StartupModule()
{
    FPropertyEditorModule& PropModule = FModuleManager::LoadModuleChecked<FPropertyEditorModule>("PropertyEditor");
    PropModule.RegisterCustomPropertyTypeLayout("SkillDataConfig"
        , FOnGetPropertyTypeCustomizationInstance::CreateStatic(&FDataTableRowSelectorCustomization::MakeInstance));

    //注册资源类型编辑器
    RegisterAssetTypeActions();
}

void FProjectFishEditorModule::ShutdownModule()
{
    if ( FPropertyEditorModule* PropModule = FModuleManager::GetModulePtr<FPropertyEditorModule>("PropertyEditor"))
    {
        PropModule->UnregisterCustomPropertyTypeLayout("SkillDataConfig");
    }
    UnregisterAssetTypeActions();
}

void FProjectFishEditorModule::RegisterAssetTypeActions()
{
    FAssetToolsModule& AssetToolsModule = FModuleManager::LoadModuleChecked<FAssetToolsModule>("AssetTools");
    IAssetTools& AssetTools = AssetToolsModule.Get();

    //BagShape
    TSharedRef<FAssetTypeActions_Base> BagShapeAction = MakeShareable(new FBagShapeAssetTypeAction);
    AssetTools.RegisterAssetTypeActions(BagShapeAction);
    CreatedAssetTypeActions.Add(BagShapeAction);
}

void FProjectFishEditorModule::UnregisterAssetTypeActions()
{
    if (FModuleManager::Get().IsModuleLoaded("AssetTools"))
    {
        FAssetToolsModule& AssetToolsModule = FModuleManager::LoadModuleChecked<FAssetToolsModule>("AssetTools");
        IAssetTools& AssetTools = AssetToolsModule.Get();
        for (auto action : CreatedAssetTypeActions)
        {
            AssetTools.UnregisterAssetTypeActions(action.ToSharedRef());
        }
    }
    CreatedAssetTypeActions.Empty();
}

#undef LOCTEXT_NAMESPACE
    
IMPLEMENT_MODULE(FProjectFishEditorModule, ProjectFishEditor)