src/ai/api/ai_cargolist.cpp
branchnoai
changeset 11164 67338721eb26
parent 10339 ce6cd68d9eb8
--- a/src/ai/api/ai_cargolist.cpp	Wed Jul 09 13:32:13 2008 +0000
+++ b/src/ai/api/ai_cargolist.cpp	Thu Jul 17 23:45:52 2008 +0000
@@ -3,8 +3,10 @@
 /** @file ai_cargolist.cpp Implementation of AICargoList and friends. */
 
 #include "ai_cargolist.hpp"
+#include "ai_industry.hpp"
 #include "../../openttd.h"
 #include "../../cargotype.h"
+#include "../../industry.h"
 
 AICargoList::AICargoList()
 {
@@ -16,3 +18,28 @@
 	}
 }
 
+AICargoList_IndustryAccepting::AICargoList_IndustryAccepting(IndustryID industry_id)
+{
+	if (!AIIndustry::IsValidIndustry(industry_id)) return;
+
+	Industry *ind = ::GetIndustry(industry_id);
+	for (uint i = 0; i < lengthof(ind->accepts_cargo); i++) {
+		CargoID cargo_id = ind->accepts_cargo[i];
+		if (cargo_id != CT_INVALID) {
+			this->AddItem(cargo_id);
+		}
+	}
+}
+
+AICargoList_IndustryProducing::AICargoList_IndustryProducing(IndustryID industry_id)
+{
+	if (!AIIndustry::IsValidIndustry(industry_id)) return;
+
+	Industry *ind = ::GetIndustry(industry_id);
+	for (uint i = 0; i < lengthof(ind->produced_cargo); i++) {
+		CargoID cargo_id = ind->produced_cargo[i];
+		if (cargo_id != CT_INVALID) {
+			this->AddItem(cargo_id);
+		}
+	}
+}