diff --git a/src/core/pool_func.hpp b/src/core/pool_func.hpp index 3ac28f4729..1687b4fafb 100644 --- a/src/core/pool_func.hpp +++ b/src/core/pool_func.hpp @@ -108,6 +108,7 @@ DEFINE_POOL_METHOD(void)::FreeItem(size_t index) this->data[index] = NULL; this->first_free = min(this->first_free, index); this->items--; + if (!this->cleaning) Titem::PostDestructor(index); } DEFINE_POOL_METHOD(void)::CleanPool() diff --git a/src/core/pool_type.hpp b/src/core/pool_type.hpp index 2d32df724f..2f35b4a88c 100644 --- a/src/core/pool_type.hpp +++ b/src/core/pool_type.hpp @@ -221,6 +221,15 @@ struct Pool { { return Tpool->items; } + + /** + * Dummy function called after destructor of each member. + * If you want to use it, override it in PoolItem's subclass. + * @param index index of deleted item + * @note when this function is called, PoolItem::Get(index) == NULL. + * @note it's called only when !CleaningPool() + */ + static FORCEINLINE void PostDestructor(size_t index) { } }; private: diff --git a/src/station.cpp b/src/station.cpp index 4db8c7c0f3..32166493f3 100644 --- a/src/station.cpp +++ b/src/station.cpp @@ -94,8 +94,6 @@ Station::~Station() /* Remove all news items */ DeleteStationNews(this->index); - InvalidateWindowData(WC_SELECT_STATION, 0, 0); - for (CargoID c = 0; c < NUM_CARGO; c++) { goods[c].cargo.Truncate(0); } @@ -108,6 +106,16 @@ Station::~Station() } +/** + * Invalidating of the JoinStation window has to be done + * after removing item from the pool. + * @param index index of deleted item + */ +void Station::PostDestructor(size_t index) +{ + InvalidateWindowData(WC_SELECT_STATION, 0, 0); +} + /** * Get the primary road stop (the first road stop) that the given vehicle can load/unload. * @param v the vehicle to get the first road stop for diff --git a/src/station_base.h b/src/station_base.h index b91fb499e6..deb01bfc66 100644 --- a/src/station_base.h +++ b/src/station_base.h @@ -186,6 +186,8 @@ public: { return Station::Get(GetStationIndex(tile)); } + + static void PostDestructor(size_t index); }; #define FOR_ALL_STATIONS_FROM(var, start) FOR_ALL_ITEMS_FROM(Station, station_index, var, start)