CGNICO/FLESH
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
Jeffreytsai1004
2025-04-18 18:17:02 +08:00
parent 80fdce96b2
commit e15eafe5d4
18 changed files with 259 additions and 493 deletions
Download Patch File Download Diff File Expand all files Collapse all files

122
Source/FLESHEditor/Private/DismembermentGraph/DismembermentGraphEditorModule.cpp
View File

@@ -11,20 +11,51 @@
#include "Modules/ModuleManager.h"
#include "GraphEditorActions.h"
#include "AssetTypeActions_Base.h"
#include "DismembermentGraph/DismembermentGraphSchema.h"
#include "DismembermentGraph/SDismembermentGraphNode.h"
#include "DismembermentGraph/DismembermentGraphEditorFactory.h"
#include "DismembermentGraph/DismembermentGraphPalette.h"
#include "DismembermentGraph/DismembermentGraphNodeFactory.h"
#include "Styling/AppStyle.h"
#include "GraphEditorSettings.h"
#include "EdGraph/EdGraphSchema.h"
#define LOCTEXT_NAMESPACE "DismembermentGraphEditor"
/**
* Dismemberment graph schema actions factory
* Dismemberment graph asset type actions
*/
class FDismembermentGraphSchemaActionMenuBuilder : public FGraphSchemaActionMenuBuilder
class FDismembermentGraphAssetTypeActions : public FAssetTypeActions_Base
{
public:
// Constructor
FDismembermentGraphSchemaActionMenuBuilder(const UEdGraphSchema* InSchema)
: FGraphSchemaActionMenuBuilder(InSchema)
FDismembermentGraphAssetTypeActions(EAssetTypeCategories::Type InAssetCategory)
: AssetCategory(InAssetCategory)
{
}
// FAssetTypeActions_Base interface
virtual FText GetName() const override { return LOCTEXT("DismembermentGraphAssetName", "Dismemberment Graph"); }
virtual FColor GetTypeColor() const override { return FColor(255, 64, 64); }
virtual UClass* GetSupportedClass() const override { return UDismembermentGraphAsset::StaticClass(); }
virtual uint32 GetCategories() override { return AssetCategory; }
virtual bool HasActions(const TArray<UObject*>& InObjects) const override { return false; }
virtual void OpenAssetEditor(const TArray<UObject*>& InObjects, TSharedPtr<IToolkitHost> EditWithinLevelEditor) override
{
for (UObject* Object : InObjects)
{
if (UDismembermentGraphAsset* GraphAsset = Cast<UDismembermentGraphAsset>(Object))
{
TSharedRef<FDismembermentGraphEditor> NewEditor = MakeShareable(new FDismembermentGraphEditor());
NewEditor->InitDismembermentGraphEditor(EToolkitMode::Standalone, EditWithinLevelEditor, GraphAsset);
}
}
}
// End of FAssetTypeActions_Base interface
private:
// Asset category
EAssetTypeCategories::Type AssetCategory;
};
/**
@@ -43,7 +74,7 @@ public:
DismembermentGraphAssetCategoryBit = AssetTools.RegisterAdvancedAssetCategory(FName(TEXT("Dismemberment")), LOCTEXT("DismembermentGraphAssetCategory", "Dismemberment"));
// Register asset type actions
TSharedPtr<IAssetTypeActions> Action = MakeShareable(new FDismembermentGraphAssetTypeActions(DismembermentGraphAssetCategoryBit));
TSharedRef<IAssetTypeActions> Action = MakeShareable(new FDismembermentGraphAssetTypeActions(DismembermentGraphAssetCategoryBit));
AssetTools.RegisterAssetTypeActions(Action);
RegisteredAssetTypeActions.Add(Action);
@@ -100,84 +131,7 @@ private:
}
};
/**
* Dismemberment graph asset type actions
*/
class FDismembermentGraphAssetTypeActions : public FAssetTypeActions_Base
{
public:
// Constructor
FDismembermentGraphAssetTypeActions(EAssetTypeCategories::Type InAssetCategory)
: AssetCategory(InAssetCategory)
{
}
// FAssetTypeActions_Base interface
virtual FText GetName() const override { return LOCTEXT("DismembermentGraphAssetName", "Dismemberment Graph"); }
virtual FColor GetTypeColor() const override { return FColor(255, 64, 64); }
virtual UClass* GetSupportedClass() const override { return UDismembermentGraphAsset::StaticClass(); }
virtual uint32 GetCategories() override { return AssetCategory; }
virtual bool HasActions(const TArray<UObject*>& InObjects) const override { return false; }
virtual void OpenAssetEditor(const TArray<UObject*>& InObjects, TSharedPtr<IToolkitHost> EditWithinLevelEditor) override
{
for (UObject* Object : InObjects)
{
if (UDismembermentGraphAsset* GraphAsset = Cast<UDismembermentGraphAsset>(Object))
{
TSharedRef<FDismembermentGraphEditor> NewEditor = MakeShareable(new FDismembermentGraphEditor());
NewEditor->InitDismembermentGraphEditor(EToolkitMode::Standalone, EditWithinLevelEditor, GraphAsset);
}
}
}
// End of FAssetTypeActions_Base interface
private:
// Asset category
EAssetTypeCategories::Type AssetCategory;
};
/**
* Dismemberment graph node factory
*/
class FDismembermentGraphNodeFactory : public FGraphPanelNodeFactory
{
public:
// Constructor
FDismembermentGraphNodeFactory()
: NodeClass(nullptr)
{
}
FDismembermentGraphNodeFactory(UClass* InNodeClass, const FText& InDisplayName, const FText& InTooltip)
: NodeClass(InNodeClass)
, DisplayName(InDisplayName)
, Tooltip(InTooltip)
{
}
// FGraphPanelNodeFactory interface
virtual TSharedPtr<SGraphNode> CreateNode(UEdGraphNode* Node) const override
{
if (NodeClass && Node->IsA(NodeClass))
{
return SNew(SDismembermentGraphNode, Node);
}
return nullptr;
}
// End of FGraphPanelNodeFactory interface
private:
// Node class
UClass* NodeClass;
// Display name
FText DisplayName;
// Tooltip
FText Tooltip;
};
IMPLEMENT_MODULE(FDismembermentGraphEditorModule, FLESHEditor)
// Comment out this line to avoid module redefinition
// IMPLEMENT_MODULE(FDismembermentGraphEditorModule, FLESHEditor)
#undef LOCTEXT_NAMESPACE