DataflowDebugDrawObject.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once 
 
#include "HitProxies.h"
#include "Dataflow/DataflowDebugDrawInterface.h"
#include "Dataflow/DataflowElement.h"
#include "GenericPlatform/ICursor.h"
 
#include "DataflowDebugDrawObject.generated.h"
 
#define UE_API DATAFLOWENGINE_API
 
class FPrimitiveDrawInterface;
 
struct FDataflowDebugDrawBaseObject : public IDataflowDebugDrawObject
{
    using Super = IDataflowDebugDrawObject;
    
    static FName StaticType() { return FName("FDataflowDebugDrawBaseObject"); }
    
    FDataflowDebugDrawBaseObject(IDataflowDebugDrawInterface::FDataflowElementsType& InDataflowElements) :
        IDataflowDebugDrawObject(), DataflowElements(InDataflowElements)
    {}
    
    virtual void PopulateDataflowElements() = 0;
 
    virtual void DrawDataflowElements(FPrimitiveDrawInterface* PDI) = 0;
 
    virtual FBox ComputeBoundingBox() const = 0;
 
    virtual bool IsA(FName InType) const override
    {
        return InType.ToString().Equals(StaticType().ToString()) 
            || Super::IsA(InType); 
    }
protected :
 
    IDataflowDebugDrawInterface::FDataflowElementsType& DataflowElements;
 
    int32 ElementsOffset = 0;
 
    int32 ElementsSize = 0;
};
 
template<typename ObjectType, typename... Args>
TRefCountPtr<ObjectType> MakeDebugDrawObject(Args&& ... args)
{
    TRefCountPtr<ObjectType> DataflowObject = MakeRefCount<ObjectType>(std::forward<Args>(args)...);
    DataflowObject->PopulateDataflowElements();
 
    return DataflowObject;
}
 
struct HDataflowElementHitProxy : public HHitProxy
{
    DECLARE_HIT_PROXY( UE_API )
 
    HDataflowElementHitProxy(int32 InElementIndex, FName InElementName)
        : HHitProxy(HPP_Foreground)
        , ElementIndex(InElementIndex)
        , ElementName(InElementName)
    {}
 
    //~ Begin HHitProxy interface
    virtual EMouseCursor::Type GetMouseCursor() override { return EMouseCursor::Crosshairs; }
    //~ End HHitProxy interface
 
    int32 ElementIndex;
 
    FName ElementName;
};
 
USTRUCT()
struct FDataflowProxyElement : public FDataflowBaseElement
{
    GENERATED_BODY()
 
    FDataflowProxyElement() {}
 
    FDataflowProxyElement(const FString& InElementName, FDataflowBaseElement* InParentElement, const FBox& InBoundingBox, const bool bInIsConstruction) :
        FDataflowBaseElement(InElementName, InParentElement, InBoundingBox, bInIsConstruction ), ElementProxy(nullptr) {}
 
    TRefCountPtr<HHitProxy> ElementProxy = nullptr;
 
    static FName StaticType() { return FName("FDatafloProxyElement"); }
 
    virtual bool IsA(FName InType) const override
    {
        return InType.ToString().Equals(StaticType().ToString())
            || Super::IsA(InType);
    }
};
 
#undef UE_API