Merge branch 'template_train_replacement' into template_train_replacement-sx

src/saveload/afterload.cpp

@@ -2991,6 +2991,9 @@ bool AfterLoadGame()
 	ResetSignalHandlers();
 
 	AfterLoadLinkGraphs();
+
+	AfterLoadTemplateVehiclesUpdateImage();
+
 	return true;
 }
 
@@ -3028,4 +3031,5 @@ void ReloadNewGRFData()
 	/* redraw the whole screen */
 	MarkWholeScreenDirty();
 	CheckTrainsLengths();
+	AfterLoadTemplateVehiclesUpdateImage();
 }

src/saveload/saveload_internal.h

@@ -48,6 +48,8 @@ void ConnectMultiheadedTrains();
 Engine *GetTempDataEngine(EngineID index);
 void CopyTempEngineData();
 
+void AfterLoadTemplateVehiclesUpdateImage();
+
 extern int32 _saved_scrollpos_x;
 extern int32 _saved_scrollpos_y;
 extern ZoomLevelByte _saved_scrollpos_zoom;

src/saveload/tbtr_template_veh_sl.cpp

@@ -1,6 +1,8 @@
 #include "../stdafx.h"
 
 #include "../tbtr_template_vehicle.h"
+#include "../tbtr_template_vehicle_func.h"
+#include "../train.h"
 
 #include "saveload.h"
 
@@ -94,6 +96,37 @@ void AfterLoadTemplateVehicles()
 	}
 }
 
+void AfterLoadTemplateVehiclesUpdateImage()
+{
+	TemplateVehicle *tv;
+
+	FOR_ALL_TEMPLATES(tv) {
+		if (tv->Prev() == NULL) {
+			StringID err;
+			Train* t = VirtualTrainFromTemplateVehicle(tv, err);
+			if (t != NULL) {
+				int tv_len = 0;
+				for (TemplateVehicle *u = tv; u != NULL; u = u->Next()) {
+					tv_len++;
+				}
+				int t_len = 0;
+				for (Train *u = t; u != NULL; u = u->Next()) {
+					t_len++;
+				}
+				if (t_len == tv_len) {
+					Train *v = t;
+					for (TemplateVehicle *u = tv; u != NULL; u = u->Next(), v = v->Next()) {
+						u->spritenum = v->spritenum;
+						u->cur_image = v->GetImage(DIR_W, EIT_PURCHASE);
+						u->image_width = v->GetDisplayImageWidth();
+					}
+				}
+				delete t;
+			}
+		}
+	}
+}
+
 extern const ChunkHandler _template_vehicle_chunk_handlers[] = {
 	{'TMPL', Save_TMPLS, Load_TMPLS, Ptrs_TMPLS, NULL, CH_ARRAY | CH_LAST},
 };