1

/* $Id$ */

     2

     3

/** @file ai_industry.cpp handles the query-related of the AIIndustry class */

     4

     5

#include "ai_industry.hpp"

     6

#include "ai_cargo.hpp"

     7

#include "ai_map.hpp"

     8

#include "../../industry.h"

     9

#include "../../strings_func.h"

    10

#include "table/strings.h"

    11

    12

/* static */ IndustryID AIIndustry::GetMaxIndustryID()

    13

{

    14

	return ::GetMaxIndustryIndex();

    15

}

    16

    17

/* static */ int32 AIIndustry::GetIndustryCount()

    18

{

    19

	return ::GetNumIndustries();

    20

}

    21

    22

/* static */ bool AIIndustry::IsValidIndustry(IndustryID industry_id)

    23

{

    24

	return ::IsValidIndustryID(industry_id);

    25

}

    26

    27

/* static */ char *AIIndustry::GetName(IndustryID industry_id)

    28

{

    29

	if (!IsValidIndustry(industry_id)) return NULL;

    30

	static const int len = 64;

    31

	char *industry_name = MallocT<char>(len);

    32

    33

	::SetDParam(0, industry_id);

    34

	::GetString(industry_name, STR_INDUSTRY, &industry_name[len - 1]);

    35

    36

	return industry_name;

    37

}

    38

    39

/* static */ int32 AIIndustry::GetProduction(IndustryID industry_id, CargoID cargo_id)

    40

{

    41

	if (!IsValidIndustry(industry_id)) return -1;

    42

	if (!AICargo::IsValidCargo(cargo_id)) return -1;

    43

    44

	const Industry *i = ::GetIndustry(industry_id);

    45

	const IndustrySpec *indsp = ::GetIndustrySpec(i->type);

    46

    47

	for (byte j = 0; j < lengthof(indsp->produced_cargo); j++)

    48

		if (indsp->produced_cargo[j] == cargo_id) return i->production_rate[j] * 8;

    49

    50

	return -1;

    51

}

    52

    53

/* static */ bool AIIndustry::IsCargoAccepted(IndustryID industry_id, CargoID cargo_id)

    54

{

    55

	if (!IsValidIndustry(industry_id)) return false;

    56

	if (!AICargo::IsValidCargo(cargo_id)) return false;

    57

    58

	const Industry *i = ::GetIndustry(industry_id);

    59

	const IndustrySpec *indsp = ::GetIndustrySpec(i->type);

    60

    61

	for (byte j = 0; j < lengthof(indsp->accepts_cargo); j++)

    62

		if (indsp->accepts_cargo[j] == cargo_id) return true;

    63

    64

	return false;

    65

}

    66

    67

/* static */ TileIndex AIIndustry::GetLocation(IndustryID industry_id)

    68

{

    69

	if (!IsValidIndustry(industry_id)) return INVALID_TILE;

    70

	return ::GetIndustry(industry_id)->xy;

    71

}

    72

    73

/* static */ uint16 AIIndustry::GetLastMonthProduction(IndustryID industry_id, CargoID cargo_id)

    74

{

    75

	if (!IsValidIndustry(industry_id)) return false;

    76

	if (!AICargo::IsValidCargo(cargo_id)) return false;

    77

    78

	const Industry *i = ::GetIndustry(industry_id);

    79

	const IndustrySpec *indsp = ::GetIndustrySpec(i->type);

    80

    81

	for (byte j = 0; j < lengthof(indsp->produced_cargo); j++)

    82

		if (indsp->produced_cargo[j] == cargo_id) return i->last_month_production[j];

    83

    84

	return false;

    85

}

    86

    87

/* static */ uint16 AIIndustry::GetLastMonthTransported(IndustryID industry_id, CargoID cargo_id)

    88

{

    89

	if (!IsValidIndustry(industry_id)) return false;

    90

	if (!AICargo::IsValidCargo(cargo_id)) return false;

    91

    92

	const Industry *i = ::GetIndustry(industry_id);

    93

	const IndustrySpec *indsp = ::GetIndustrySpec(i->type);

    94

    95

	for (byte j = 0; j < lengthof(indsp->produced_cargo); j++)

    96

		if (indsp->produced_cargo[j] == cargo_id) return i->last_month_transported[j];

    97

    98

	return false;

    99

}

   100

   101

/* static */ int32 AIIndustry::GetDistanceManhattanToTile(IndustryID industry_id, TileIndex tile)

   102

{

   103

	return AIMap::DistanceManhattan(tile, GetLocation(industry_id));

   104

}

   105

   106

/* static */ int32 AIIndustry::GetDistanceSquareToTile(IndustryID industry_id, TileIndex tile)

   107

{

   108

	return AIMap::DistanceSquare(tile, GetLocation(industry_id));

   109

}