59 lines
1.5 KiB
C++
59 lines
1.5 KiB
C++
#include "MCPNotifier.h"
|
|
#include "EdGraph/EdGraphNode.h"
|
|
#include "EdGraph/EdGraph.h"
|
|
#include "Engine/Blueprint.h"
|
|
#include "Materials/Material.h"
|
|
#include "Kismet2/BlueprintEditorUtils.h"
|
|
#include "MaterialEditingLibrary.h"
|
|
|
|
void MCPNotifier::AddTouchedObject(UObject* Obj)
|
|
{
|
|
if (!Obj) return;
|
|
bool bAlreadyInSet = false;
|
|
TouchedSet.Add(Obj, &bAlreadyInSet);
|
|
if (bAlreadyInSet) return;
|
|
TouchedArray.Add(Obj);
|
|
Obj->PreEditChange(nullptr);
|
|
}
|
|
|
|
void MCPNotifier::SendNotifications()
|
|
{
|
|
TSet<UEdGraphNode*> Nodes;
|
|
TSet<UEdGraph*> Graphs;
|
|
TSet<UMaterial*> Materials;
|
|
TSet<UBlueprint*> Blueprints;
|
|
for (int32 i = TouchedArray.Num() - 1; i >= 0; --i)
|
|
{
|
|
UObject* Obj = TouchedArray[i];
|
|
Obj->PostEditChange();
|
|
Obj->MarkPackageDirty();
|
|
|
|
if (UEdGraphNode* Node = ::Cast<UEdGraphNode>(Obj))
|
|
Nodes.Add(Node);
|
|
|
|
if (UEdGraph* Graph = ::Cast<UEdGraph>(Obj))
|
|
Graphs.Add(Graph);
|
|
|
|
if (UBlueprint* BP = ::Cast<UBlueprint>(Obj))
|
|
Blueprints.Add(BP);
|
|
|
|
if (UMaterialInterface* MatIface = ::Cast<UMaterialInterface>(Obj))
|
|
if (UMaterial* BaseMat = MatIface->GetMaterial())
|
|
Materials.Add(BaseMat);
|
|
}
|
|
for (UEdGraphNode* Node : Nodes)
|
|
Node->ReconstructNode();
|
|
for (UEdGraph* Graph : Graphs)
|
|
Graph->NotifyGraphChanged();
|
|
for (UMaterial *Material : Materials)
|
|
UMaterialEditingLibrary::RebuildMaterialInstanceEditors(Material);
|
|
for (UBlueprint *Blueprint : Blueprints)
|
|
FBlueprintEditorUtils::MarkBlueprintAsStructurallyModified(Blueprint);
|
|
|
|
if (GEditor)
|
|
GEditor->RedrawAllViewports();
|
|
|
|
TouchedSet.Empty();
|
|
TouchedArray.Empty();
|
|
}
|