diff options
Diffstat (limited to 'src/ai/api')
-rw-r--r-- | src/ai/api/ai_engine.cpp | 8 | ||||
-rw-r--r-- | src/ai/api/ai_event_types.cpp | 9 |
2 files changed, 7 insertions, 10 deletions
diff --git a/src/ai/api/ai_engine.cpp b/src/ai/api/ai_engine.cpp index 3948a406a..669c44ec5 100644 --- a/src/ai/api/ai_engine.cpp +++ b/src/ai/api/ai_engine.cpp @@ -6,9 +6,10 @@ #include "ai_cargo.hpp" #include "../../company_func.h" #include "../../strings_func.h" -#include "../../aircraft.h" #include "../../vehicle_func.h" #include "../../settings_type.h" +#include "../../rail.h" +#include "../../engine_base.h" #include "../../articulated_vehicles.h" #include "table/strings.h" @@ -77,11 +78,8 @@ } break; case VEH_SHIP: - return e->u.ship.capacity; - break; - case VEH_AIRCRAFT: - return AircraftDefaultCargoCapacity(e->GetDefaultCargoType(), &e->u.air); + return e->GetDisplayDefaultCapacity(); break; default: NOT_REACHED(); diff --git a/src/ai/api/ai_event_types.cpp b/src/ai/api/ai_event_types.cpp index 469d53776..f02cfc957 100644 --- a/src/ai/api/ai_event_types.cpp +++ b/src/ai/api/ai_event_types.cpp @@ -3,9 +3,11 @@ /** @file ai_event_types.cpp Implementation of all EventTypes. */ #include "ai_event_types.hpp" +#include "../../command_type.h" #include "../../strings_func.h" #include "../../settings_type.h" -#include "../../aircraft.h" +#include "../../rail.h" +#include "../../engine_base.h" #include "../../articulated_vehicles.h" #include "table/strings.h" @@ -41,11 +43,8 @@ int32 AIEventEnginePreview::GetCapacity() } break; case VEH_SHIP: - return e->u.ship.capacity; - break; - case VEH_AIRCRAFT: - return AircraftDefaultCargoCapacity(e->GetDefaultCargoType(), &e->u.air); + return e->GetDisplayDefaultCapacity(); break; default: NOT_REACHED(); |