src/ai/api/ai_cargo.cpp
author rubidium
Sun, 03 Feb 2008 20:17:54 +0000
branchnoai
changeset 9724 b39bc69bb2f2
parent 9723 eee46cb39750
child 9736 183b38e0a480
permissions -rw-r--r--
(svn r12051) [NoAI] -Sync: with trunk (r11795:12050).
/* $Id$ */

/** @file ai_cargo.cpp handles the query-related of the AICargo class */

#include "ai_cargo.hpp"
#include "../../cargotype.h"
#include "../../economy_func.h"
#include "../../core/alloc_func.hpp"
#include "../../core/bitmath_func.hpp"

/* static */ bool AICargo::IsValidCargo(CargoID cargo_type)
{
	return (cargo_type < NUM_CARGO && GetCargo(cargo_type)->IsValid());
}

char *AICargo::GetCargoLabel(CargoID cargo_type)
{
	if (!this->IsValidCargo(cargo_type)) return NULL;
	const CargoSpec *cargo = GetCargo(cargo_type);

	/* cargo->label is a uint32 packing a 4 character non-terminated string,
	 * like "PASS", "COAL", "OIL_". New ones can be defined by NewGRFs */
	char *cargo_label = MallocT<char>(sizeof(cargo->label) + 1);
	for (uint i = 0; i < sizeof(cargo->label); i++) {
		cargo_label[i] = GB(cargo->label, (sizeof(cargo->label) - i - 1) * 8, 8);
	}
	cargo_label[sizeof(cargo->label)] = '\0';
	return cargo_label;
}

bool AICargo::IsFreight(CargoID cargo_type)
{
	if (!this->IsValidCargo(cargo_type)) return false;
	const CargoSpec *cargo = GetCargo(cargo_type);
	return cargo->is_freight;
}

int32 AICargo::GetCargoIncome(uint32 distance, uint32 days_in_transit, CargoID cargo_type)
{
	if (!this->IsValidCargo(cargo_type)) return 0;
	return GetTransportedGoodsIncome(1, distance, days_in_transit, cargo_type);
}