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 "../../industry.h"

     8

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

     9

#include "table/strings.h"

    10

    11

IndustryID AIIndustry::GetMaxIndustryID()

    12

{

    13

	return ::GetMaxIndustryIndex();

    14

}

    15

    16

int32 AIIndustry::GetIndustryCount()

    17

{

    18

	return ::GetNumIndustries();

    19

}

    20

    21

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

    22

{

    23

	return ::IsValidIndustryID(industry_id);

    24

}

    25

    26

char *AIIndustry::GetName(IndustryID industry_id)

    27

{

    28

	if (!this->IsValidIndustry(industry_id)) return NULL;

    29

	static const int len = 64;

    30

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

    31

    32

	SetDParam(0, industry_id);

    33

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

    34

    35

	return industry_name;

    36

}

    37

    38

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

    39

{

    40

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

    41

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

    42

    43

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

    44

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

    45

    46

	if (indsp->produced_cargo[0] == cargo_id) return i->production_rate[0];

    47

	if (indsp->produced_cargo[1] == cargo_id) return i->production_rate[1];

    48

	return -1;

    49

}

    50

    51

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

    52

{

    53

	if (!AIIndustry::IsValidIndustry(industry_id)) return false;

    54

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

    55

    56

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

    57

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

    58

    59

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

    60

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

    61

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

    62

    63

	return false;

    64

}

    65

    66

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

    67

{

    68

	if (!AIIndustry::IsValidIndustry(industry_id)) return INVALID_TILE;

    69

	return ::GetIndustry(industry_id)->xy;

    70

}