From ed5ced71c852387aec030acc19ad096e55bcbd82 Mon Sep 17 00:00:00 2001
From: rubidium <rubidium@openttd.org>
Date: Tue, 21 Jul 2009 11:20:34 +0000
Subject: (svn r16897) -Codechange: use the 'generic' station spec to station
 allocation for waypoints too

---
 src/waypoint_cmd.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'src/waypoint_cmd.cpp')

diff --git a/src/waypoint_cmd.cpp b/src/waypoint_cmd.cpp
index 2f10c7dea..882ba6097 100644
--- a/src/waypoint_cmd.cpp
+++ b/src/waypoint_cmd.cpp
@@ -190,7 +190,7 @@ CommandCost CmdBuildTrainWaypoint(TileIndex tile, DoCommandFlag flags, uint32 p1
 		SetDepotWaypointReservation(tile, reserved);
 		MarkTileDirtyByTile(tile);
 
-		wp->AssignStationSpec(p1);
+		AllocateSpecToStation(GetCustomStationSpec(STAT_CLASS_WAYP, p1), wp, true);
 
 		wp->delete_ctr = 0;
 		wp->build_date = _date;
@@ -246,6 +246,8 @@ CommandCost RemoveTrainWaypoint(TileIndex tile, DoCommandFlag flags, bool justre
 		}
 		YapfNotifyTrackLayoutChange(tile, track);
 		if (v != NULL) TryPathReserve(v, true);
+
+		DeallocateSpecFromStation(wp, wp->num_specs > 0 ? 1 : 0);
 	}
 
 	return CommandCost(EXPENSES_CONSTRUCTION, _price.remove_train_depot);
-- 
cgit v1.2.3-54-g00ecf