From 7b77f010bb12c1fab44df0c667fd12fbeaa51350 Mon Sep 17 00:00:00 2001
From: Lynix <lynix680@gmail.com>
Date: Fri, 6 Jan 2017 01:06:49 +0100
Subject: [PATCH] Renderer/ShaderAst: Fix Mat4x4 handling

---
 include/Nazara/Renderer/ShaderAst.inl | 50 ++++++++++++++++++++++++---
 1 file changed, 46 insertions(+), 4 deletions(-)

diff --git a/include/Nazara/Renderer/ShaderAst.inl b/include/Nazara/Renderer/ShaderAst.inl
index 9bde01926..398956187 100644
--- a/include/Nazara/Renderer/ShaderAst.inl
+++ b/include/Nazara/Renderer/ShaderAst.inl
@@ -21,9 +21,11 @@ namespace Nz
 					return 3;
 
 				case ExpressionType::Float4:
-				case ExpressionType::Mat4x4:
 					return 4;
 
+				case ExpressionType::Mat4x4:
+					return 16;
+
 				default:
 					return 1;
 			}
@@ -85,9 +87,49 @@ namespace Nz
 		left(std::move(Left)),
 		right(std::move(Right))
 		{
-			//TODO: AstParseError
-			if (left->GetExpressionType() != right->GetExpressionType())
-				throw std::runtime_error("Left expression type must match right expression type");
+			ExpressionType leftType = left->GetExpressionType();
+			ExpressionType rightType = right->GetExpressionType();
+
+			if (leftType != rightType)
+			{
+				switch (op)
+				{
+					case BinaryType::Add:
+					case BinaryType::Divide:
+					case BinaryType::Equality:
+					case BinaryType::Substract:
+					{
+						//TODO: AstParseError
+						throw std::runtime_error("Left expression type must match right expression type");
+					}
+
+					case BinaryType::Multiply:
+					{
+						switch (leftType)
+						{
+							case ExpressionType::Mat4x4:
+							{
+								switch (rightType)
+								{
+									case ExpressionType::Float4:
+									case ExpressionType::Mat4x4:
+										break;
+
+									//TODO: AstParseError
+									default:
+										throw std::runtime_error("Left expression type is not compatible with right expression type");
+								}
+
+								break;
+							}
+
+							default:
+								//TODO: AstParseError
+								throw std::runtime_error("Left expression type must match right expression type");
+						}
+					}
+				}
+			}
 		}
 
 		inline Branch::Branch(ExpressionPtr condition, StatementPtr trueStatement, StatementPtr falseStatement)