1

/* $Id$ */

     2

     3

/** @file ai_company.cpp handles the functions of the AICompany class */

     4

     5

#include "ai_company.hpp"

     7

#include "../../player.h"

    10

#include "../../variables.h"

    12

#include "table/strings.h"

    13

    14

AICompany::CompanyIndex AICompany::ResolveCompanyIndex(AICompany::CompanyIndex company)

    15

{

    16

	if (company == MY_COMPANY) return (CompanyIndex)((byte)_current_player);

    17

    18

	return (IsValidPlayer((PlayerID)company) && GetPlayer((PlayerID)company)->is_active) ? company : INVALID_COMPANY;

    19

}

    20

    21

bool AICompany::SetCompanyName(const char *name)

    22

{

    23

	if (name == NULL) return false;

    24

    25

	_cmd_text = name;

    26

	return this->DoCommand(0, 0, 0, CMD_CHANGE_COMPANY_NAME);

    27

}

    28

    29

char *AICompany::GetCompanyName(AICompany::CompanyIndex company)

    30

{

    31

	company = this->ResolveCompanyIndex(company);

    32

	if (company == INVALID_COMPANY) return NULL;

    33

    34

	static const int len = 64;

    35

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

    36

    37

	SetDParam(0, GetPlayer((PlayerID)company)->index);

    38

	GetString(company_name, STR_COMPANY_NAME, &company_name[len - 1]);

    39

	return company_name;

    40

}

    41

    42

bool AICompany::SetPresidentName(const char *name)

    43

{

    44

	if (name == NULL) return false;

    45

    46

	_cmd_text = name;

    47

	return this->DoCommand(0, 0, 0, CMD_CHANGE_PRESIDENT_NAME);

    48

}

    49

    50

char *AICompany::GetPresidentName(AICompany::CompanyIndex company)

    51

{

    52

	company = this->ResolveCompanyIndex(company);

    53

    54

	static const int len = 64;

    55

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

    56

	if (company != INVALID_COMPANY) {

    57

		SetDParam(0, GetPlayer((PlayerID)company)->index);

    58

		GetString(president_name, STR_PLAYER_NAME, &president_name[len - 1]);

    59

	} else {

    60

		*president_name = '\0';

    61

	}

    62

    63

	return president_name;

    64

}

    65

    66

int32 AICompany::GetCompanyValue(AICompany::CompanyIndex company)

    67

{

    68

	company = this->ResolveCompanyIndex(company);

    69

	if (company == INVALID_COMPANY) return 0;

    70

    71

	return GetPlayer((PlayerID)company)->cur_economy.company_value;

    72

}

    73

    74

int32 AICompany::GetBankBalance(AICompany::CompanyIndex company)

    75

{

    76

	company = this->ResolveCompanyIndex(company);

    77

	if (company == INVALID_COMPANY) return 0;

    78

    79

	return GetPlayer((PlayerID)company)->player_money;

    80

}

    81

    82

int32 AICompany::GetLoanAmount()

    83

{

    84

	return GetPlayer(_current_player)->current_loan;

    85

}

    86

    87

int32 AICompany::GetMaxLoanAmount()

    88

{

    89

	return _economy.max_loan;

    90

}

    91

    92

int32 AICompany::GetLoanInterval()

    93

{

    94

	return LOAN_INTERVAL;

    95

}

    96

    97

bool AICompany::SetLoanAmount(int32 loan)

    98

{

    99

	if (loan < 0 ||

   100

			(loan % this->GetLoanInterval()) != 0 ||

   101

			loan > this->GetMaxLoanAmount() ||

   102

			(loan - this->GetLoanAmount() + this->GetBankBalance(MY_COMPANY)) < 0) {

   103

		return false;

   104

	}

   105

   106

	if (loan == this->GetLoanAmount()) return true;

   107

   108

	return this->DoCommand(0,

   109

			abs(loan - this->GetLoanAmount()), 2,

   110

			(loan > this->GetLoanAmount()) ? CMD_INCREASE_LOAN : CMD_DECREASE_LOAN);

   111

}

   112

   113

bool AICompany::SetMinimumLoanAmount(int32 loan)

   114

{

   115

	if (loan < 0) return false;

   116

   117

	int32 over_interval = loan % this->GetLoanInterval();

   118

	if (over_interval != 0) loan += this->GetLoanInterval() - over_interval;

   119

   120

	if (loan > this->GetMaxLoanAmount()) return false;

   121

   122

	this->SetLoanAmount(loan);

   123

   124

	return this->GetLoanAmount() == loan;

   125

}