Code/CryEngine/CrySchematyc2/Deprecated/DocGraphNodes/DocGraphBranchNode.cpp

   1 // Copyright 2001-2019 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "Deprecated/DocGraphNodes/DocGraphBranchNode.h"
   5
   6 #include <CrySchematyc2/ICompiler.h>
   7 #include <CrySchematyc2/LibUtils.h>
   8 #include <CrySchematyc2/Deprecated/DocUtils.h>
   9 #include <CrySchematyc2/Deprecated/IGlobalFunction.h>
  10
  11 #include "Deprecated/DocGraphNodes/DocGraphNodeBase.h"
  12 #include "Script/ScriptVariableDeclaration.h"
  13
  14 namespace Schematyc2
  15 {
  16         //////////////////////////////////////////////////////////////////////////
  17         CDocGraphBranchNode::CDocGraphBranchNode(IScriptFile& file, IDocGraph& graph, const SGUID& guid, const SGUID& contextGUID, const SGUID& refGUID, Vec2 pos)
  18                 : CDocGraphNodeBase(file, graph, guid, "Branch", EScriptGraphNodeType::Branch, contextGUID, refGUID, pos)
  19                 , m_bValue(false)
  20         {
  21                 Refresh(SScriptRefreshParams(EScriptRefreshReason::Other));
  22         }
  23
  24         //////////////////////////////////////////////////////////////////////////
  25         IAnyConstPtr CDocGraphBranchNode::GetCustomOutputDefault() const
  26         {
  27                 return IAnyConstPtr();
  28         }
  29
  30         //////////////////////////////////////////////////////////////////////////
  31         size_t CDocGraphBranchNode::AddCustomOutput(const IAny& value)
  32         {
  33                 return INVALID_INDEX;
  34         }
  35
  36         //////////////////////////////////////////////////////////////////////////
  37         void CDocGraphBranchNode::EnumerateOptionalOutputs(const ScriptGraphNodeOptionalOutputEnumerator& enumerator) {}
  38
  39         //////////////////////////////////////////////////////////////////////////
  40         size_t CDocGraphBranchNode::AddOptionalOutput(const char* szName, EScriptGraphPortFlags flags, const CAggregateTypeId& typeId)
  41         {
  42                 return INVALID_INDEX;
  43         }
  44
  45         //////////////////////////////////////////////////////////////////////////
  46         void CDocGraphBranchNode::RemoveOutput(size_t outputIdx) {}
  47
  48         //////////////////////////////////////////////////////////////////////////
  49         void CDocGraphBranchNode::Refresh(const SScriptRefreshParams& params)
  50         {
  51                 CRY_PROFILE_FUNCTION(PROFILE_LOADING_ONLY);
  52                 CDocGraphNodeBase::Refresh(params);
  53                 CDocGraphNodeBase::AddInput("In", EScriptGraphPortFlags::MultiLink | EScriptGraphPortFlags::Execute);
  54                 CDocGraphNodeBase::AddInput("Value", EScriptGraphPortFlags::None, GetAggregateTypeId<bool>());
  55                 CDocGraphNodeBase::AddOutput("True", EScriptGraphPortFlags::Execute);
  56                 CDocGraphNodeBase::AddOutput("False", EScriptGraphPortFlags::Execute);
  57         }
  58
  59         //////////////////////////////////////////////////////////////////////////
  60         void CDocGraphBranchNode::Serialize(Serialization::IArchive& archive)
  61         {
  62                 CRY_PROFILE_FUNCTION(PROFILE_LOADING_ONLY);
  63                 CDocGraphNodeBase::Serialize(archive);
  64                 archive(m_bValue, "value", "Value");
  65         }
  66
  67         //////////////////////////////////////////////////////////////////////////
  68         void CDocGraphBranchNode::PreCompileSequence(IDocGraphSequencePreCompiler& preCompiler, size_t outputIdx) const {}
  69
  70         //////////////////////////////////////////////////////////////////////////
  71         void CDocGraphBranchNode::LinkSequence(IDocGraphSequenceLinker& linker, size_t outputIdx, const LibFunctionId& functionId) const {}
  72
  73         //////////////////////////////////////////////////////////////////////////
  74         void CDocGraphBranchNode::Compile(IDocGraphNodeCompiler& compiler, EDocGraphSequenceStep sequenceStep, size_t portIdx) const
  75         {
  76                 switch(sequenceStep)
  77                 {
  78                 case EDocGraphSequenceStep::BeginInput:
  79                         {
  80                                 CompileInputs(compiler);
  81                                 break;
  82                         }
  83                 case EDocGraphSequenceStep::BeginOutput:
  84                         {
  85                                 switch(portIdx)
  86                                 {
  87                                 case EOutput::True:
  88                                         {
  89                                                 CompileTrue(compiler);
  90                                                 break;
  91                                         }
  92                                 case EOutput::False:
  93                                         {
  94                                                 CompileFalse(compiler);
  95                                                 break;
  96                                         }
  97                                 }
  98                                 break;
  99                         }
 100                 case EDocGraphSequenceStep::EndInput:
 101                         {
 102                                 CompileEnd(compiler);
 103                                 break;
 104                         }
 105                 }
 106         }
 107
 108         //////////////////////////////////////////////////////////////////////////
 109         void CDocGraphBranchNode::CompileInputs(IDocGraphNodeCompiler& compiler) const
 110         {
 111                 compiler.CreateStackFrame(*this, EStackFrame::Value);
 112                 const size_t stackPos = compiler.FindInputOnStack(*this, EInput::Value);
 113                 if(stackPos != INVALID_INDEX)
 114                 {
 115                         if(stackPos != (compiler.GetStackSize() - 1))
 116                         {
 117                                 compiler.Copy(stackPos, INVALID_INDEX, MakeAny(m_bValue), CDocGraphNodeBase::GetGUID(), GetInputName(EInput::Value));
 118                         }
 119                         else
 120                         {
 121                                 compiler.SetDebugInput(CDocGraphNodeBase::GetGUID(), GetInputName(EInput::Value));
 122                         }
 123                 }
 124                 else
 125                 {
 126                         compiler.Push(MakeAny(m_bValue), CDocGraphNodeBase::GetGUID(), GetInputName(EInput::Value));
 127                 }
 128         }
 129
 130         //////////////////////////////////////////////////////////////////////////
 131         void CDocGraphBranchNode::CompileTrue(IDocGraphNodeCompiler& compiler) const
 132         {
 133                 compiler.BranchIfZero(*this, EMarker::BranchFalse);
 134                 compiler.CollapseStackFrame(*this, EStackFrame::Value);
 135                 compiler.CreateStackFrame(*this, EStackFrame::Value);
 136         }
 137
 138         //////////////////////////////////////////////////////////////////////////
 139         void CDocGraphBranchNode::CompileFalse(IDocGraphNodeCompiler& compiler) const
 140         {
 141                 compiler.Branch(*this, EMarker::BranchEnd);
 142                 compiler.CreateMarker(*this, EMarker::BranchFalse);
 143                 compiler.CollapseStackFrame(*this, EStackFrame::Value);
 144                 compiler.CreateStackFrame(*this, EStackFrame::Value);
 145         }
 146
 147         //////////////////////////////////////////////////////////////////////////
 148         void CDocGraphBranchNode::CompileEnd(IDocGraphNodeCompiler& compiler) const
 149         {
 150                 compiler.CreateMarker(*this, EMarker::BranchEnd);
 151                 compiler.CollapseStackFrame(*this, EStackFrame::Value);
 152         }
 153 }