DataflowContextObject.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "UObject/Object.h"
#include "Dataflow/DataflowEdNode.h"
#include "Dataflow/DataflowNodeParameters.h"
#include "DataflowContextObject.generated.h"
 
#define UE_API DATAFLOWENGINE_API
 
struct FManagedArrayCollection;
class UDataflow;
namespace UE::Dataflow
{
    class IDataflowConstructionViewMode;
    template<class Base> class TEngineContext;
    typedef TEngineContext<FContextSingle> FEngineContext;
}
 
UCLASS(MinimalAPI)
class UDataflowContextObject : public UObject
{
    GENERATED_BODY()
public:
 
    void SetSelectedNode(TObjectPtr<UDataflowEdNode> InSelectedNode) { SelectedNode = InSelectedNode; }
    TObjectPtr<UDataflowEdNode> GetSelectedNode() const { return SelectedNode; }
 
    template<typename NodeType>
    NodeType* GetSelectedNodeOfType() const
    {
        if (SelectedNode && SelectedNode->GetDataflowNode())
        {
            return SelectedNode->GetDataflowNode()->AsType<NodeType>();
        }
        return nullptr;
    }
 
    void SetRenderCollection(const TSharedPtr<FManagedArrayCollection>& InCollection) { RenderCollection = InCollection; }
    TSharedPtr<const FManagedArrayCollection> GetRenderCollection() const { return RenderCollection; }
 
    void SetConstructionViewMode(const UE::Dataflow::IDataflowConstructionViewMode* InMode) { ConstructionViewMode = InMode; }
    const UE::Dataflow::IDataflowConstructionViewMode* GetConstructionViewMode() const { return ConstructionViewMode; }
    
    virtual void SetDataflowContext(const TSharedPtr<UE::Dataflow::FEngineContext>& InDataflowContext) { DataflowContext = InDataflowContext; }
    const TSharedPtr<UE::Dataflow::FEngineContext>& GetDataflowContext() const { return DataflowContext; }
    TSharedPtr<UE::Dataflow::FEngineContext>& GetDataflowContext() { return DataflowContext; }
 
    virtual void SetDataflowAsset(const TObjectPtr<UDataflow>& InAsset) { DataflowGraph = InAsset; }
    TObjectPtr<UDataflow> GetDataflowAsset() const { return DataflowGraph;  }
 
    void SetSelectedCollection(TSharedPtr<const FManagedArrayCollection> InSelectedCollection, bool bInUsingInputCollection) 
    { 
        SelectedCollection = InSelectedCollection; 
        bUsingInputCollection = bInUsingInputCollection;
    }
    const TSharedPtr<const FManagedArrayCollection> GetSelectedCollection() const { return SelectedCollection; }
    bool IsUsingInputCollection() const { return bUsingInputCollection; }
 
 
    //~ UObject interface
    static UE_API void AddReferencedObjects(UObject* InThis, FReferenceCollector& Collector);
 
protected:
 
    TSharedPtr<FManagedArrayCollection> RenderCollection = nullptr;
 
    UPROPERTY(Transient, SkipSerialization)
    TObjectPtr<UDataflowEdNode> SelectedNode = nullptr;
 
    const UE::Dataflow::IDataflowConstructionViewMode* ConstructionViewMode = nullptr;
 
    TSharedPtr<UE::Dataflow::FEngineContext> DataflowContext = nullptr;
 
    UPROPERTY(Transient, SkipSerialization)
    TObjectPtr<UDataflow> DataflowGraph = nullptr;
 
    TSharedPtr<const FManagedArrayCollection> SelectedCollection = nullptr;
 
    bool bUsingInputCollection = false;
 
};
 
#undef UE_API