Add NewGRFClass helper method for whether class ID is valid (for iteration)

This commit is contained in:
Jonathan G Rennison
2023-01-03 03:51:40 +00:00
parent 417416d19b
commit 6a7cad1b47
7 changed files with 19 additions and 8 deletions

View File

@@ -1202,7 +1202,7 @@ public:
this->station_classes.clear();
for (uint i = 0; i < StationClass::GetClassCount(); i++) {
for (uint i = 0; StationClass::IsClassIDValid((StationClassID)i); i++) {
StationClassID station_class_id = (StationClassID)i;
if (station_class_id == StationClassID::STAT_CLASS_WAYP) {
// Skip waypoints.
@@ -1236,7 +1236,7 @@ public:
/* Check if the previously selected station class is not available anymore as a
* result of starting a new game without the corresponding NewGRF. */
bool available = false;
for (uint i = 0; i < StationClass::GetClassCount(); ++i) {
for (uint i = 0; StationClass::IsClassIDValid((StationClassID)i); ++i) {
if ((StationClassID)i == _railstation.station_class) {
available = true;
break;