MaterialExpressionNeuralPostProcessNode.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "UObject/ObjectMacros.h"
#include "Materials/MaterialExpressionCustomOutput.h"
#include "MaterialExpressionNeuralPostProcessNode.generated.h"
 
struct FPropertyChangedEvent;
enum EMaterialSamplerType : int;
 
UENUM()
enum ENeuralIndexType : int
{
    NIT_TextureIndex UMETA(DisplayName = "Texture Index"),
 
    NIT_BufferIndex UMETA(DisplayName = "Buffer Index"),
 
};
 
UCLASS(MinimalAPI, collapsecategories, hidecategories=Object)
class UMaterialExpressionNeuralNetworkInput : public UMaterialExpressionCustomOutput 
{
    GENERATED_UCLASS_BODY()
    
    
    UPROPERTY(meta = (RequiredInput = "false"))
    FExpressionInput Coordinates;
 
    UPROPERTY()
    FExpressionInput Input0;
 
    UPROPERTY(meta = (RequiredInput = "false"))
    FExpressionInput Mask;
 
    UPROPERTY(EditAnywhere, CateGory = "Default", meta = (DisplayName = "Index Type"))
    TEnumAsByte<ENeuralIndexType> NeuralIndexType;
    
#if WITH_EDITOR
 
    virtual int32 Compile(class FMaterialCompiler* Compiler, int32 OutputIndex) override;
    virtual void GetCaption(TArray<FString>& OutCaptions) const override;
    virtual EMaterialValueType GetInputValueType(int32 InputIndex) override;
 
#endif // WITH_EDITOR
 
    //~ Begin UMaterialExpressionCustomOutput Interface
    virtual int32 GetNumOutputs() const override;
    virtual FString GetFunctionName() const override;
    //~ End UMaterialExpressionCustomOutput Interface
};
 
 
// The input node that holds the output from the neural network to the post process material 
UCLASS(MinimalAPI, collapsecategories, hidecategories = Object)
class UMaterialExpressionNeuralNetworkOutput : public UMaterialExpression
{
    GENERATED_UCLASS_BODY()
 
    
    UPROPERTY(meta = (RequiredInput = "false"))
    FExpressionInput Coordinates;
 
    UPROPERTY(EditAnywhere, Category = "Default", meta = (DisplayName = "Index Type"))
    TEnumAsByte<ENeuralIndexType> NeuralIndexType;
 
#if WITH_EDITOR
    virtual int32 Compile(class FMaterialCompiler* Compiler, int32 OutputIndex) override;
    virtual void GetCaption(TArray<FString>& OutCaptions) const override;
    virtual EMaterialValueType GetInputValueType(int32 InputIndex) override;
 
#endif // WITH_EDITOR
 
};