1

/* $Id$ */

     2

     3

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

     4

     5

#include "ai_cargo.hpp"

     6

#include "../../cargotype.h"

     7

#include "../../player.h" // For economy.h

     8

#include "../../economy.h"

     9

    10

/* static */ bool AICargo::IsValidCargo(CargoID cargo_type)

    11

{

    12

	return (cargo_type < NUM_CARGO && GetCargo(cargo_type)->IsValid());

    13

}

    14

    15

char *AICargo::GetCargoLabel(CargoID cargo_type)

    16

{

    17

	if (!this->IsValidCargo(cargo_type)) return NULL;

    18

	const CargoSpec *cargo = GetCargo(cargo_type);

    19

    20

	/* cargo->label is a uint32 packing a 4 character non-terminated string,

    21

	 * like "PASS", "COAL", "OIL_". New ones can be defined by NewGRFs */

    22

	char *cargo_label = MallocT<char>(sizeof(cargo->label) + 1);

    23

	for (uint i = 0; i < sizeof(cargo->label); i++) {

    24

		cargo_label[i] = GB(cargo->label, (sizeof(cargo->label) - i - 1) * 8, 8);

    25

	}

    26

	cargo_label[sizeof(cargo->label)] = '\0';

    27

	return cargo_label;

    28

}

    29

    30

bool AICargo::IsFreight(CargoID cargo_type)

    31

{

    32

	if (!this->IsValidCargo(cargo_type)) return false;

    33

	const CargoSpec *cargo = GetCargo(cargo_type);

    34

	return cargo->is_freight;

    35

}

    36

    37

int32 AICargo::GetCargoIncome(uint32 distance, uint32 days_in_transit, CargoID cargo_type)

    38

{

    39

	if (!this->IsValidCargo(cargo_type)) return 0;

    40

	return GetTransportedGoodsIncome(1, distance, days_in_transit, cargo_type);

    41

}