CommonNodes.h

// Copyright (C) Entropy Software LLC - All Rights Reserved
 
#pragma once
 
#include "ActivationTypes.h"
#include "Context.h"
#include "GraphNode.h"
 
namespace SmartEngine
{
 
#pragma pack(push, 4)
struct ConcatNodeCInfo : GraphNodeCInfo
{
    IGraphNode* const* inputs = nullptr;
 
    int inputCount = 0;
};
 
struct ExtractNodeCInfo : GraphNodeCInfo
{
    IGraphNode* input = nullptr;
 
    int columnStartIndex = 0;
 
    int columnCount = -1;
};
 
enum class ChoiceType : byte
{
    Random,
 
    Max,
 
    Count
};
SMARTENGINE_ENUM_STRING(ChoiceType, "Random", "Max");
 
struct ChoiceNodeCInfo : GraphNodeCInfo
{
    IGraphNode* input = nullptr;
 
    ChoiceType type = ChoiceType::Random;
 
    int axis = 1;
};
 
struct ActivationNodeCInfo : GraphNodeCInfo
{
    IGraphNode* input = nullptr;
 
    ActivationType type = ActivationType::None;
};
 
struct NormalNodeCInfo : GraphNodeCInfo
{
    IGraphNode* mean = nullptr;
 
    IGraphNode* variance = nullptr;
};
 
struct MinimumNodeCInfo : GraphNodeCInfo
{
    IGraphNode* A = nullptr;
 
    IGraphNode* B = nullptr;
};
 
struct MaximumNodeCInfo : GraphNodeCInfo
{
    IGraphNode* A = nullptr;
 
    IGraphNode* B = nullptr;
};
 
struct StopGradientNodeCInfo : GraphNodeCInfo
{
    IGraphNode* input = nullptr;
};
 
struct MultiplexerNodeCInfo : GraphNodeCInfo
{
    IGraphNode* selector = nullptr;
 
    IGraphNode* const* inputs = nullptr;
 
    int inputCount = 0;
 
    ChoiceType selectionMethod = ChoiceType::Max;
};
#pragma pack(pop)
 
SMARTENGINE_EXPORT ObjectPtr<IGraphNode> CreateConcatNode(const ConcatNodeCInfo& cinfo);
 
SMARTENGINE_EXPORT ObjectPtr<IGraphNode> CreateExtractNode(const ExtractNodeCInfo& cinfo);
 
SMARTENGINE_EXPORT ObjectPtr<IGraphNode> CreateChoiceNode(const ChoiceNodeCInfo& cinfo);
 
SMARTENGINE_EXPORT ObjectPtr<IGraphNode> CreateActivationNode(const ActivationNodeCInfo& cinfo);
 
SMARTENGINE_EXPORT ObjectPtr<IGraphNode> CreateNormalNode(const NormalNodeCInfo& cinfo);
 
SMARTENGINE_EXPORT ObjectPtr<IGraphNode> CreateMinimumNode(const MinimumNodeCInfo& cinfo);
 
SMARTENGINE_EXPORT ObjectPtr<IGraphNode> CreateMaximumNode(const MaximumNodeCInfo& cinfo);
 
SMARTENGINE_EXPORT ObjectPtr<IGraphNode> CreateStopGradientNode(const StopGradientNodeCInfo& cinfo);
 
SMARTENGINE_EXPORT ObjectPtr<IGraphNode> CreateMultiplexerNode(const MultiplexerNodeCInfo& cinfo);
 
extern "C"
{
    SMARTENGINE_EXPORT ObjPtr ConcatNode_CreateInstance(const ConcatNodeCInfo& cinfo);
    SMARTENGINE_EXPORT ObjPtr ExtractNode_CreateInstance(const ExtractNodeCInfo& cinfo);
    SMARTENGINE_EXPORT ObjPtr ChoiceNode_CreateInstance(const ChoiceNodeCInfo& cinfo);
    SMARTENGINE_EXPORT ObjPtr ActivationNode_CreateInstance(const ActivationNodeCInfo& cinfo);
    SMARTENGINE_EXPORT ObjPtr NormalNode_CreateInstance(const NormalNodeCInfo& cinfo);
    SMARTENGINE_EXPORT ObjPtr MinimumNode_CreateInstance(const MinimumNodeCInfo& cinfo);
    SMARTENGINE_EXPORT ObjPtr MaximumNode_CreateInstance(const MaximumNodeCInfo& cinfo);
    SMARTENGINE_EXPORT ObjPtr StopGradientNode_CreateInstance(const StopGradientNodeCInfo& cinfo);
    SMARTENGINE_EXPORT ObjPtr MultiplexerNode_CreateInstance(const MultiplexerNodeCInfo& cinfo);
}
 
} // namespace SmartEngine