Shader: Add function parameters and return handling

include/Nazara/Shader/SpirvBlock.inl

@@ -7,16 +7,70 @@
 
 namespace Nz
 {
-	inline SpirvBlock::SpirvBlock(SpirvWriter& writer)
+	inline SpirvBlock::SpirvBlock(SpirvWriter& writer) :
+	m_isTerminated(false)
 	{
 		m_labelId = writer.AllocateResultId();
 		Append(SpirvOp::OpLabel, m_labelId);
 	}
 
+	inline std::size_t SpirvBlock::Append(SpirvOp opcode, const OpSize& wordCount)
+	{
+		HandleSpirvOp(opcode);
+
+		return SpirvSectionBase::Append(opcode, wordCount);
+	}
+
+	template<typename ...Args>
+	std::size_t SpirvBlock::Append(SpirvOp opcode, Args&&... args)
+	{
+		HandleSpirvOp(opcode);
+
+		return SpirvSectionBase::Append(opcode, std::forward<Args>(args)...);
+	}
+
+	template<typename F>
+	std::size_t SpirvBlock::AppendVariadic(SpirvOp opcode, F&& callback)
+	{
+		HandleSpirvOp(opcode);
+
+		return SpirvSectionBase::AppendVariadic(opcode, std::forward<F>(callback));
+	}
+
 	inline UInt32 SpirvBlock::GetLabelId() const
 	{
 		return m_labelId;
 	}
+
+	inline bool SpirvBlock::IsTerminated() const
+	{
+		return m_isTerminated;
+	}
+
+	inline bool SpirvBlock::IsTerminationInstruction(SpirvOp op)
+	{
+		switch (op)
+		{
+			case SpirvOp::OpBranch:
+			case SpirvOp::OpBranchConditional:
+			case SpirvOp::OpKill:
+			case SpirvOp::OpReturn:
+			case SpirvOp::OpReturnValue:
+			case SpirvOp::OpSwitch:
+			case SpirvOp::OpUnreachable:
+				return true;
+
+			default:
+				return false;
+		}
+	}
+
+	inline void SpirvBlock::HandleSpirvOp(SpirvOp op)
+	{
+		assert(!m_isTerminated);
+		if (IsTerminationInstruction(op))
+			m_isTerminated = true;
+	}
 }
 
 #include <Nazara/Shader/DebugOff.hpp>