Keep list of all vehicles in Level class

Newly created vehicles are now to be added to the Level, which
automatically updates and renders them. This infrastructure will
be needed for depth sorting and eventually level saving.

1 files changed, 5 insertions, 5 deletions

diff --git a/src/main.cpp b/src/main.cpp
index 0dd2dea..9281b2c 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -15,11 +15,15 @@ int main(int argc, char** argv)
 
 	Level level(32, 32, tiles);
 	Bitmap bitmap(Width, Height);
+
 	Train engine(level);
 	Train wagon(level);
 
 	engine.addVehicle(&wagon);
 
+	level.addVehicle(engine);
+	level.addVehicle(wagon);
+
 	int xOffs = 0, yOffs = 0;
 	int xDrag, yDrag;
 	Point2D hoveredTile = { 0, 0 };
@@ -65,12 +69,8 @@ int main(int argc, char** argv)
 		window.update();
 		bitmap.clear(0xff224466);
 
+		level.update();
 		level.draw(bitmap, xOffs, yOffs);
-		engine.update(level);
-		engine.draw(bitmap, xOffs, yOffs);
-
-		wagon.update(level);
-		wagon.draw(bitmap, xOffs, yOffs);
 
 		int xx = (hoveredTile.x - hoveredTile.y) * (TileSize / 2) - xOffs;
 		int yy = (hoveredTile.x + hoveredTile.y) * (TileSize / 4) - yOffs;