summaryrefslogtreecommitdiff
path: root/src/newgrf_airport.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/newgrf_airport.cpp')
-rw-r--r--src/newgrf_airport.cpp147
1 files changed, 145 insertions, 2 deletions
diff --git a/src/newgrf_airport.cpp b/src/newgrf_airport.cpp
index 20c0ba0e7..bec9f6146 100644
--- a/src/newgrf_airport.cpp
+++ b/src/newgrf_airport.cpp
@@ -14,9 +14,14 @@
#include "newgrf_airport.h"
#include "date_func.h"
#include "settings_type.h"
+#include "core/alloc_type.hpp"
+#include "newgrf.h"
+#include "table/strings.h"
-AirportSpec AirportSpec::dummy = {NULL, NULL, 0, 0, 0, 0, 0, MIN_YEAR, MIN_YEAR, ATP_TTDP_LARGE};
-AirportSpec AirportSpec::oilrig = {NULL, NULL, 0, 1, 1, 0, 4, MIN_YEAR, MIN_YEAR, ATP_TTDP_OILRIG};
+static AirportClass _airport_classes[APC_MAX];
+
+AirportSpec AirportSpec::dummy = {NULL, NULL, 0, 0, 0, 0, 0, MIN_YEAR, MIN_YEAR, ATP_TTDP_LARGE, APC_BEGIN};
+AirportSpec AirportSpec::oilrig = {NULL, NULL, 0, 1, 1, 0, 4, MIN_YEAR, MIN_YEAR, ATP_TTDP_OILRIG, APC_BEGIN};
AirportSpec AirportSpec::specs[NUM_AIRPORTS];
@@ -62,3 +67,141 @@ void AirportSpec::ResetAirports()
memset(&AirportSpec::specs, 0, sizeof(AirportSpec::specs));
memcpy(&AirportSpec::specs, &_origin_airport_specs, sizeof(AirportSpec) * NUM_AIRPORTS);
}
+
+/**
+ * Allocate an airport class for the given class id
+ * @param cls A 32 bit value identifying the class
+ * @return Index into _airport_classes of allocated class
+ */
+AirportClassID AllocateAirportClass(uint32 cls)
+{
+ for (AirportClassID i = APC_BEGIN; i < APC_MAX; i++) {
+ if (_airport_classes[i].id == cls) {
+ /* ClassID is already allocated, so reuse it. */
+ return i;
+ } else if (_airport_classes[i].id == 0) {
+ /* This class is empty, so allocate it to the ClassID. */
+ _airport_classes[i].id = cls;
+ return i;
+ }
+ }
+
+ grfmsg(2, "AllocateAirportClass: already allocated %d classes, using small airports class", APC_MAX);
+ return APC_SMALL;
+}
+
+/**
+ * Set the name of an airport class
+ * @param id The id of the class to change the name from
+ * @param name The new name for the class
+ */
+void SetAirportClassName(AirportClassID id, StringID name)
+{
+ assert(id < APC_MAX);
+ _airport_classes[id].name = name;
+}
+
+/**
+ * Retrieve the name of an airport class
+ * @param id The id of the airport class to get the name from
+ * @return The name of the given class
+ */
+StringID GetAirportClassName(AirportClassID id)
+{
+ assert(id < APC_MAX);
+ return _airport_classes[id].name;
+}
+
+/**
+ * Get the number of airport classes in use
+ * @return The number of airport classes
+ */
+uint GetNumAirportClasses()
+{
+ uint i;
+ for (i = APC_BEGIN; i < APC_MAX && _airport_classes[i].id != 0; i++) {}
+ return i;
+}
+
+/**
+ * Return the number of airports for the given airport class.
+ * @param id Index of the airport class.
+ * @return Number of airports in the class.
+ */
+uint GetNumAirportsInClass(AirportClassID id)
+{
+ assert(id < APC_MAX);
+ return _airport_classes[id].airports;
+}
+
+/**
+ * Tie an airport spec to its airport class.
+ * @param statspec The airport spec.
+ */
+static void BindAirportSpecToClass(AirportSpec *as)
+{
+ assert(as->aclass < APC_MAX);
+ AirportClass *airport_class = &_airport_classes[as->aclass];
+
+ int i = airport_class->airports++;
+ airport_class->spec = ReallocT(airport_class->spec, airport_class->airports);
+
+ airport_class->spec[i] = as;
+}
+
+/**
+ * Tie all airportspecs to their class.
+ */
+void BindAirportSpecs()
+{
+ for (int i = 0; i < NUM_AIRPORTS; i++) {
+ AirportSpec *as = AirportSpec::GetWithoutOverride(i);
+ BindAirportSpecToClass(as);
+ }
+}
+
+
+/**
+ * Retrieve an airport spec from a class.
+ * @param aclass Index of the airport class.
+ * @param airport The airport index with the class.
+ * @return The station spec.
+ */
+const AirportSpec *GetAirportSpecFromClass(AirportClassID aclass, uint airport)
+{
+ assert(aclass < APC_MAX);
+ assert(airport < _airport_classes[aclass].airports);
+
+ return _airport_classes[aclass].spec[airport];
+}
+
+/**
+ * Reset airport classes to their default state.
+ * This includes initialising the defaults classes with an empty
+ * entry, for standard airports.
+ */
+void ResetAirportClasses()
+{
+ for (AirportClassID i = APC_BEGIN; i < APC_MAX; i++) {
+ _airport_classes[i].id = 0;
+ _airport_classes[i].name = STR_EMPTY;
+ _airport_classes[i].airports = 0;
+
+ free(_airport_classes[i].spec);
+ _airport_classes[i].spec = NULL;
+ }
+
+ /* Set up initial data */
+ AirportClassID id = AllocateAirportClass('SMAL');
+ SetAirportClassName(id, STR_AIRPORT_CLASS_SMALL);
+
+ id = AllocateAirportClass('LARG');
+ SetAirportClassName(id, STR_AIRPORT_CLASS_LARGE);
+
+ id = AllocateAirportClass('HUB_');
+ SetAirportClassName(id, STR_AIRPORT_CLASS_HUB);
+
+ id = AllocateAirportClass('HELI');
+ SetAirportClassName(id, STR_AIRPORT_CLASS_HELIPORTS);
+}
+