summaryrefslogtreecommitdiff
path: root/src/ai/api/ai_vehiclelist.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/ai/api/ai_vehiclelist.cpp')
-rw-r--r--src/ai/api/ai_vehiclelist.cpp37
1 files changed, 37 insertions, 0 deletions
diff --git a/src/ai/api/ai_vehiclelist.cpp b/src/ai/api/ai_vehiclelist.cpp
new file mode 100644
index 000000000..127ac77e9
--- /dev/null
+++ b/src/ai/api/ai_vehiclelist.cpp
@@ -0,0 +1,37 @@
+/* $Id$ */
+
+/** @file ai_vehiclelist.cpp Implementation of AIVehicleList and friends. */
+
+#include "ai_vehiclelist.hpp"
+#include "ai_station.hpp"
+#include "../../openttd.h"
+#include "../../company_func.h"
+#include "../../vehicle_base.h"
+
+AIVehicleList::AIVehicleList()
+{
+ Vehicle *v;
+ FOR_ALL_VEHICLES(v) {
+ if (v->owner == _current_company && v->IsPrimaryVehicle()) this->AddItem(v->index);
+ }
+}
+
+AIVehicleList_Station::AIVehicleList_Station(StationID station_id)
+{
+ if (!AIStation::IsValidStation(station_id)) return;
+
+ Vehicle *v;
+
+ FOR_ALL_VEHICLES(v) {
+ if (v->owner == _current_company && v->IsPrimaryVehicle()) {
+ const Order *order;
+
+ FOR_VEHICLE_ORDERS(v, order) {
+ if (order->IsType(OT_GOTO_STATION) && order->GetDestination() == station_id) {
+ this->AddItem(v->index);
+ break;
+ }
+ }
+ }
+ }
+}