Code/CryEngine/CrySchematyc2/Script/GraphNodes/ScriptGraphBranchNode.cpp

   1 // Copyright 2001-2019 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "Script/GraphNodes/ScriptGraphBranchNode.h"
   5
   6 #include <CrySchematyc2/IDomainContext.h>
   7 #include <CrySchematyc2/Script/IScriptGraphNodeCompiler.h>
   8 #include <CrySchematyc2/Script/IScriptGraphNodeCreationMenu.h>
   9
  10 #include "Script/GraphNodes/ScriptGraphNodeFactory.h"
  11
  12 namespace Schematyc2
  13 {
  14         const SGUID CScriptGraphBranchNode::s_typeGUID = "4F2A4D42-ED2A-40A3-9555-DBB1210A36D8";
  15
  16         CScriptGraphBranchNode::CScriptGraphBranchNode()
  17                 : m_bValue(false)
  18         {
  19         }
  20
  21         CScriptGraphBranchNode::CScriptGraphBranchNode(const SGUID& guid)
  22                 : CScriptGraphNodeBase(guid)
  23                 , m_bValue(false)
  24         {}
  25
  26         CScriptGraphBranchNode::CScriptGraphBranchNode(const SGUID& guid, const Vec2& pos)
  27                 : CScriptGraphNodeBase(guid, pos)
  28                 , m_bValue(false)
  29         {}
  30
  31         SGUID CScriptGraphBranchNode::GetTypeGUID() const
  32         {
  33                 return s_typeGUID;
  34         }
  35
  36         EScriptGraphColor CScriptGraphBranchNode::GetColor() const
  37         {
  38                 return EScriptGraphColor::Green;
  39         }
  40
  41         void CScriptGraphBranchNode::Refresh(const SScriptRefreshParams& params)
  42         {
  43                 CRY_PROFILE_FUNCTION(PROFILE_LOADING_ONLY);
  44
  45                 CScriptGraphNodeBase::Refresh(params);
  46
  47                 CScriptGraphNodeBase::SetName("Branch");
  48                 CScriptGraphNodeBase::AddInput(EScriptGraphPortFlags::MultiLink | EScriptGraphPortFlags::Execute, "In");
  49                 CScriptGraphNodeBase::AddInput(EScriptGraphPortFlags::Data, "Value", GetAggregateTypeId<bool>());
  50
  51                 CScriptGraphNodeBase::AddOutput(EScriptGraphPortFlags::Execute, "True");
  52                 CScriptGraphNodeBase::AddOutput(EScriptGraphPortFlags::Execute, "False");
  53         }
  54
  55         void CScriptGraphBranchNode::Serialize(Serialization::IArchive& archive)
  56         {
  57                 CRY_PROFILE_FUNCTION(PROFILE_LOADING_ONLY);
  58
  59                 CScriptGraphNodeBase::Serialize(archive);
  60
  61                 archive(m_bValue, "value", "Value");
  62         }
  63
  64         void CScriptGraphBranchNode::RemapGUIDs(IGUIDRemapper& guidRemapper)
  65         {
  66                 CScriptGraphNodeBase::RemapGUIDs(guidRemapper);
  67         }
  68
  69         void CScriptGraphBranchNode::Compile_New(IScriptGraphNodeCompiler& compiler) const
  70         {
  71                 compiler.BindCallback(&Execute);
  72
  73                 compiler.BindInput(EInputIdx::Value, m_bValue);
  74         }
  75
  76         void CScriptGraphBranchNode::RegisterCreator(CScriptGraphNodeFactory& factory)
  77         {
  78                 class CCreator : public IScriptGraphNodeCreator
  79                 {
  80                         class CNodeCreationMenuCommand : public IScriptGraphNodeCreationMenuCommand
  81                         {
  82                         public:
  83
  84                                 // IScriptGraphNodeCreationMenuCommand
  85                                 IScriptGraphNodePtr Execute(const Vec2& pos)
  86                                 {
  87                                         return std::make_shared<CScriptGraphBranchNode>(gEnv->pSchematyc2->CreateGUID(), pos);
  88                                 }
  89                                 // ~IScriptGraphNodeCreationMenuCommand
  90                         };
  91
  92                 public:
  93
  94                         // IScriptGraphNodeCreator
  95                         virtual SGUID GetTypeGUID() const override
  96                         {
  97                                 return CScriptGraphBranchNode::s_typeGUID;
  98                         }
  99
 100                         virtual IScriptGraphNodePtr CreateNode() override
 101                         {
 102                                 return std::make_shared<CScriptGraphBranchNode>(gEnv->pSchematyc2->CreateGUID());
 103                         }
 104
 105                         virtual void PopulateNodeCreationMenu(IScriptGraphNodeCreationMenu& nodeCreationMenu, const IDomainContext& domainContext, const IScriptGraphExtension& graph) override
 106                         {
 107                                 nodeCreationMenu.AddOption("Branch", "Branch on input Bool value", "", std::make_shared<CNodeCreationMenuCommand>());
 108                         }
 109                         // ~IScriptGraphNodeCreator
 110                 };
 111
 112                 factory.RegisterCreator(std::make_shared<CCreator>());
 113         }
 114
 115         SRuntimeResult CScriptGraphBranchNode::Execute(IObject* pObject, const SRuntimeActivationParams& activationParams, CRuntimeNodeData& data)
 116         {
 117                 const bool* pValue = data.GetInput<bool>(EInputIdx::Value);
 118                 SCHEMATYC2_SYSTEM_ASSERT_FATAL(pValue != nullptr);
 119
 120                 const uint32 output = *pValue ? EOutputIdx::True : EOutputIdx::False;
 121                 return SRuntimeResult(ERuntimeStatus::Continue, output);
 122         }
 123 }