108 lines
4.2 KiB
C++
108 lines
4.2 KiB
C++
// Copyright 2017 ~ 2022 Critical Failure Studio Ltd. All rights reserved.
|
|
|
|
#pragma once
|
|
#include "CoreMinimal.h"
|
|
#include "Compilers/PaperZDAnimBPCompilerContext.h"
|
|
|
|
class FCompilerResultsLog;
|
|
class FPaperZDAnimDataLinkRecord;
|
|
class UPaperZDAnimGraphNode_Base;
|
|
|
|
/**
|
|
* Compile time accessor to the running "Compiler Context".
|
|
*/
|
|
class FPaperZDAnimBPCompilerAccess
|
|
{
|
|
/* Only grant access during compilation (private constructor, only the compiler can generate this accessor). */
|
|
friend class FPaperZDAnimBPCompilerContext;
|
|
FPaperZDAnimBPCompilerAccess(FPaperZDAnimBPCompilerContext* InCompiler)
|
|
: CompilerContext(InCompiler)
|
|
{
|
|
check(CompilerContext);
|
|
}
|
|
|
|
/* Compiler context we're wrapping. */
|
|
FPaperZDAnimBPCompilerContext* CompilerContext;
|
|
|
|
public:
|
|
/* Request a given compiler handle. */
|
|
template <typename T>
|
|
T* GetHandle()
|
|
{
|
|
TUniquePtr<IPaperZDAnimBPCompilerHandle>* HandlePtr = CompilerContext->CompileHandles.Find(T::GetClassId());
|
|
if (HandlePtr)
|
|
{
|
|
IPaperZDAnimBPCompilerHandle* HandleBase = (*HandlePtr).Get();
|
|
return static_cast<T*>(HandleBase);
|
|
}
|
|
else
|
|
{
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
/* Spawns an intermediate node associated with the source node (for error purposes) */
|
|
template <typename NodeType>
|
|
NodeType* SpawnIntermediateNode(UEdGraphNode* SourceNode, UEdGraph* ParentGraph = nullptr)
|
|
{
|
|
return CompilerContext->SpawnIntermediateNode<NodeType>(SourceNode, ParentGraph);
|
|
}
|
|
|
|
/** Spawn an intermediate function graph for this compilation using the specified desired name (which may be modified to make it unique */
|
|
UEdGraph* SpawnIntermediateFunctionGraph(const FString& InDesiredFunctionName);
|
|
|
|
/* Obtains the list of compile time generated function graphs. */
|
|
TArray<UEdGraph*>& GetGeneratedFunctionGraphs() const { return CompilerContext->GeneratedFunctionGraphs; }
|
|
|
|
/* Find the generated FProperty that represents the given GraphNode. */
|
|
FProperty* GetAllocatedPropertyFromNode(UPaperZDAnimGraphNode_Base* GraphNode) const;
|
|
|
|
/* Gets the message log associated to the compiler. */
|
|
FCompilerResultsLog& GetMessageLog() const;
|
|
|
|
/* Adds an AnimDataLinkRecord. */
|
|
void AddAnimDataLinkRecord(const FPaperZDAnimDataLinkRecord& Record);
|
|
|
|
/* Obtains the consolidated event graph. */
|
|
UEdGraph* GetConsolidatedEventGraph() const;
|
|
|
|
/* Obtains the AnimBlueprint being compiled. */
|
|
const UPaperZDAnimBP* GetAnimBP() const;
|
|
|
|
/* Processes all the supplied animation nodes. */
|
|
void ProcessAnimationNodes(TArray<UPaperZDAnimGraphNode_Base*>& AnimNodeList);
|
|
|
|
/* Prunes any nodes that aren't reachable via an AnimData link. */
|
|
void PruneIsolatedAnimationNodes(const TArray<UPaperZDAnimGraphNode_Base*>& RootSet, TArray<UPaperZDAnimGraphNode_Base*>& GraphNodes);
|
|
|
|
/* Performs standard validation on the graph (outputs point to inputs, no more than one connection to each input, types match on both ends, etc...). */
|
|
bool ValidateGraphIsWellFormed(UEdGraph* InGraph) const;
|
|
|
|
/* Returns the allocation index of the specified node, processing it if it was pending. */
|
|
int32 GetAllocationIndexOfNode(UPaperZDAnimGraphNode_Base* VisualAnimNode) const;
|
|
|
|
/* Gets all anim graph nodes that are piped into the provided node (traverses input pins). */
|
|
void GetLinkedAnimNodes(UPaperZDAnimGraphNode_Base* InGraphNode, TArray<UPaperZDAnimGraphNode_Base*>& LinkedAnimNodes) const;
|
|
|
|
/* Obtains the map that relates the source nodes and the processed nodes. */
|
|
TMap<UPaperZDAnimGraphNode_Base*, UPaperZDAnimGraphNode_Base*>& GetSourceNodeToProcessedNodeMap() const;
|
|
|
|
/* Delegate called when starting compiling a class. */
|
|
FOnCompileClassSignature& OnStartCompilingClass();
|
|
|
|
/* Delegate called when finishing compiling a class. */
|
|
FOnCompileClassSignature& OnFinishCompilingClass();
|
|
|
|
/* Delegate called when PostExpansionStep is called. */
|
|
FOnPostExpansionStepSignature& OnPostExpansionStep();
|
|
|
|
/* Delegate called when copying the default data to the CDO. */
|
|
FOnCopyTermDefaultsSignature& OnCopyTermDefaultsToDefaultObject();
|
|
|
|
/* Delegate called just before the animation nodes are processed. */
|
|
FOnProcessAnimationNodesSignature& OnPreProcessAnimationNodes();
|
|
|
|
/* Delegate called after the animation nodes have been processed. */
|
|
FOnProcessAnimationNodesSignature& OnPostProcessAnimationNodes();
|
|
};
|