Include ambient heat in undo

2 files changed, 6 insertions, 3 deletions

diff --git a/src/air.c b/src/air.c
index 0fe404d..c65e1a0 100644
--- a/src/air.c
+++ b/src/air.c
@@ -18,9 +18,10 @@ float vx[YRES/CELL][XRES/CELL], ovx[YRES/CELL][XRES/CELL];
 float vy[YRES/CELL][XRES/CELL], ovy[YRES/CELL][XRES/CELL];
 float pv[YRES/CELL][XRES/CELL], opv[YRES/CELL][XRES/CELL];
 
-float cb_vx[YRES/CELL][XRES/CELL], cb_ovx[YRES/CELL][XRES/CELL];
-float cb_vy[YRES/CELL][XRES/CELL], cb_ovy[YRES/CELL][XRES/CELL];
-float cb_pv[YRES/CELL][XRES/CELL], cb_opv[YRES/CELL][XRES/CELL];
+float cb_vx[YRES/CELL][XRES/CELL];
+float cb_vy[YRES/CELL][XRES/CELL];
+float cb_pv[YRES/CELL][XRES/CELL];
+float cb_hv[YRES/CELL][XRES/CELL];
 
 float fvx[YRES/CELL][XRES/CELL], fvy[YRES/CELL][XRES/CELL];
 
diff --git a/src/main.c b/src/main.c
index f198319..b3d3f49 100644
--- a/src/main.c
+++ b/src/main.c
@@ -2404,6 +2404,7 @@ int main(int argc, char *argv[])
 						vx[cby][cbx] = cb_vx[cby][cbx];
 						vy[cby][cbx] = cb_vy[cby][cbx];
 						pv[cby][cbx] = cb_pv[cby][cbx];
+						hv[cby][cbx] = cb_hv[cby][cbx];
 						bmap[cby][cbx] = cb_bmap[cby][cbx];
 						emap[cby][cbx] = cb_emap[cby][cbx];
 					}
@@ -3126,6 +3127,7 @@ int main(int argc, char *argv[])
 								cb_vx[cby][cbx] = vx[cby][cbx];
 								cb_vy[cby][cbx] = vy[cby][cbx];
 								cb_pv[cby][cbx] = pv[cby][cbx];
+								cb_hv[cby][cbx] = hv[cby][cbx];
 								cb_bmap[cby][cbx] = bmap[cby][cbx];
 								cb_emap[cby][cbx] = emap[cby][cbx];
 							}