diff --git a/src/newgrf_class.h b/src/newgrf_class.h index 4bac546d35..a0073d1dd4 100644 --- a/src/newgrf_class.h +++ b/src/newgrf_class.h @@ -12,15 +12,16 @@ #include "strings_type.h" +#include + /** * Struct containing information relating to NewGRF classes for stations and airports. */ template struct NewGRFClass { private: - uint count; ///< Number of specs in this class. - uint ui_count; ///< Number of specs in this class potentially available to the user. - Tspec **spec; ///< Array of specifications. + uint ui_count; ///< Number of specs in this class potentially available to the user. + std::vector spec; ///< List of specifications. /** * The actual classes. @@ -41,7 +42,7 @@ public: void Insert(Tspec *spec); /** Get the number of allocated specs within the class. */ - uint GetSpecCount() const { return this->count; } + uint GetSpecCount() const { return static_cast(this->spec.size()); } /** Get the number of potentially user-available specs within the class. */ uint GetUISpecCount() const { return this->ui_count; } int GetUIFromIndex(int index) const; diff --git a/src/newgrf_class_func.h b/src/newgrf_class_func.h index f97ba58b95..b6ab8b455c 100644 --- a/src/newgrf_class_func.h +++ b/src/newgrf_class_func.h @@ -28,11 +28,9 @@ DEFINE_NEWGRF_CLASS_METHOD(void)::ResetClass() { this->global_id = 0; this->name = STR_EMPTY; - this->count = 0; this->ui_count = 0; - free(this->spec); - this->spec = nullptr; + this->spec.clear(); } /** Reset the classes, i.e. clear everything. */ @@ -75,12 +73,9 @@ DEFINE_NEWGRF_CLASS_METHOD(Tid)::Allocate(uint32 global_id) */ DEFINE_NEWGRF_CLASS_METHOD(void)::Insert(Tspec *spec) { - uint i = this->count++; - this->spec = ReallocT(this->spec, this->count); + this->spec.push_back(spec); - this->spec[i] = spec; - - if (this->IsUIAvailable(i)) this->ui_count++; + if (this->IsUIAvailable(static_cast(this->spec.size() - 1))) this->ui_count++; } /** @@ -197,7 +192,8 @@ DEFINE_NEWGRF_CLASS_METHOD(const Tspec *)::GetByGrf(uint32 grfid, byte local_id, uint j; for (Tid i = (Tid)0; i < Tmax; i++) { - for (j = 0; j < classes[i].count; j++) { + uint count = static_cast(classes[i].spec.size()); + for (j = 0; j < count; j++) { const Tspec *spec = classes[i].spec[j]; if (spec == nullptr) continue; if (spec->grf_prop.grffile->grfid == grfid && spec->grf_prop.local_id == local_id) {