Fix predicating instructions before return in loops.

src/Shader/Shader.cpp

@@ -1090,14 +1090,14 @@ namespace sw
 		return opcode == OPCODE_BREAK || opcode == OPCODE_BREAKC || opcode == OPCODE_BREAKP;
 	}
 
-	bool Shader::Instruction::isLoopOrSwitch() const
+	bool Shader::Instruction::isLoop() const
 	{
-		return opcode == OPCODE_LOOP || opcode == OPCODE_REP || opcode == OPCODE_WHILE || opcode == OPCODE_SWITCH;
+		return opcode == OPCODE_LOOP || opcode == OPCODE_REP || opcode == OPCODE_WHILE;
 	}
 
-	bool Shader::Instruction::isEndLoopOrSwitch() const
+	bool Shader::Instruction::isEndLoop() const
 	{
-		return opcode == OPCODE_ENDLOOP || opcode == OPCODE_ENDREP || opcode == OPCODE_ENDWHILE || opcode == OPCODE_ENDSWITCH;;
+		return opcode == OPCODE_ENDLOOP || opcode == OPCODE_ENDREP || opcode == OPCODE_ENDWHILE;
 	}
 
 	bool Shader::Instruction::isPredicated() const
@@ -1657,6 +1657,7 @@ namespace sw
 		int breakDepth = 0;
 		int continueDepth = 0;
 		bool leaveReturn = false;
+		unsigned int functionBegin = 0;
 
 		for(unsigned int i = 0; i < instruction.size(); i++)
 		{
@@ -1688,11 +1689,11 @@ namespace sw
 
 			if(breakDepth > 0)
 			{
-				if(instruction[i]->isLoopOrSwitch())   // Nested loop or switch, don't make the end of it disable the break execution mask
+				if(instruction[i]->isLoop() || instruction[i]->opcode == OPCODE_SWITCH)   // Nested loop or switch, don't make the end of it disable the break execution mask
 				{
 					breakDepth++;
 				}
-				else if(instruction[i]->isEndLoopOrSwitch())
+				else if(instruction[i]->isEndLoop() || instruction[i]->opcode == OPCODE_ENDSWITCH)
 				{
 					breakDepth--;
 				}
@@ -1713,11 +1714,11 @@ namespace sw
 
 			if(continueDepth > 0)
 			{
-				if(instruction[i]->isLoopOrSwitch())   // Nested loop or switch, don't make the end of it disable the break execution mask
+				if(instruction[i]->isLoop() || instruction[i]->opcode == OPCODE_SWITCH)   // Nested loop or switch, don't make the end of it disable the break execution mask
 				{
 					continueDepth++;
 				}
-				else if(instruction[i]->isEndLoopOrSwitch())
+				else if(instruction[i]->isEndLoop() || instruction[i]->opcode == OPCODE_ENDSWITCH)
 				{
 					continueDepth--;
 				}
@@ -1734,11 +1735,29 @@ namespace sw
 			if(instruction[i]->opcode == OPCODE_LEAVE)
 			{
 				leaveReturn = true;
+
+				// Mark loop body instructions prior to the return statement
+				for(unsigned int l = functionBegin; l < i; l++)
+				{
+					if(instruction[l]->isLoop())
+					{
+						for(unsigned int r = l + 1; r < i; r++)
+						{
+							instruction[r]->analysisLeave = true;
+						}
+
+						break;
+					}
+				}
 			}
 			else if(instruction[i]->opcode == OPCODE_RET)   // End of the function
 			{
 				leaveReturn = false;
 			}
+			else if(instruction[i]->opcode == OPCODE_LABEL)
+			{
+				functionBegin = i;
+			}
 
 			if(leaveReturn)
 			{

src/Shader/Shader.hpp

@@ -495,8 +495,8 @@ namespace sw
 			bool isBranch() const;
 			bool isCall() const;
 			bool isBreak() const;
-			bool isLoopOrSwitch() const;
-			bool isEndLoopOrSwitch() const;
+			bool isLoop() const;
+			bool isEndLoop() const;
 
 			bool isPredicated() const;