fix WARP being invisible, make it visible in heat display mode though

2 files changed, 5 insertions, 1 deletions

diff --git a/src/simulation/Simulation.cpp b/src/simulation/Simulation.cpp
index 21f8109..afb50c5 100644
--- a/src/simulation/Simulation.cpp
+++ b/src/simulation/Simulation.cpp
@@ -136,6 +136,10 @@ int Simulation::Load(int fullX, int fullY, GameSave * save)
 				}
 			}
 		}
+		if (parts[i].pavg[0] || parts[i].pavg[1])
+		{
+			parts[i].pavg[0] = parts[i].pavg[1] = 0;
+		}
 	}
 	parts_lastActiveIndex = NPART-1;
 	force_stacking_check = 1;
diff --git a/src/simulation/elements/WARP.cpp b/src/simulation/elements/WARP.cpp
index bb8f8e1..5dd19c5 100644
--- a/src/simulation/elements/WARP.cpp
+++ b/src/simulation/elements/WARP.cpp
@@ -43,7 +43,7 @@ Element_WARP::Element_WARP()
     HighTemperatureTransition = NT;
     
     Update = &Element_WARP::update;
-    
+    Graphics = &Element_WARP::graphics;
 }
 
 //#TPT-Directive ElementHeader Element_WARP static int update(UPDATE_FUNC_ARGS)