From 567a99c3bfad5d73f127876e66257f658edd804d Mon Sep 17 00:00:00 2001
From: Ted John <ted@brambles.org>
Date: Sat, 8 Oct 2016 23:28:35 +0100
Subject: [PATCH] Improve / fix tunnel generation

---
 test/testpaint/generate.cpp | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/test/testpaint/generate.cpp b/test/testpaint/generate.cpp
index 8008e04ade..73cfe91d9a 100644
--- a/test/testpaint/generate.cpp
+++ b/test/testpaint/generate.cpp
@@ -605,9 +605,12 @@ private:
             }
         }
 
-        if (tunnelOffset[0] == tunnelOffset[1] &&
-            tunnelOffset[0] == tunnelOffset[2] &&
-            tunnelOffset[0] == tunnelOffset[3])
+        if (tunnelType[0] == 0xFF)
+        {
+            return;
+        }
+
+        if (AllMatch(tunnelOffset, 4) && AllMatch(tunnelType, 4))
         {
             GenerateTunnelCall(tabs, tunnelOffset[0], tunnelType[0]);
         }
@@ -625,7 +628,7 @@ private:
     {
         if (offset == 0)
         {
-            WriteLine(tabs, "paint_util_push_tunnel_rotated(direction, height, TUNNEL_%d);", offset);
+            WriteLine(tabs, "paint_util_push_tunnel_rotated(direction, height, TUNNEL_%d);", type);
         }
         else if (offset < 0)
         {