1

/* $Id$ */

     2

     3

/** @file ai_object.cpp handles the commands-related functions of the AIObject class */

     4

     5

#include "ai_object.hpp"

     6

#include "../../command.h"

     7

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

     8

#include "table/strings.h" // for signs.h

     9

#include "../../signs.h" // for _new_sign_id

    10

#include "../ai.h"

    11

#include "../ai_threads.h"

    12

    13

void AIObject::SetDoCommandDelay(uint ticks)

    14

{

    15

	assert(ticks > 0);

    16

	AIObject::GetDoCommandStruct(_current_player)->delay = ticks;

    17

}

    18

    19

uint AIObject::GetDoCommandDelay()

    20

{

    21

	return AIObject::GetDoCommandStruct(_current_player)->delay;

    22

}

    23

    24

void AIObject::SetDoCommandMode(AIModeProc *proc, AIObject *instance)

    25

{

    26

	AIObject::GetDoCommandStruct(_current_player)->mode = proc;

    27

	AIObject::GetDoCommandStruct(_current_player)->mode_instance = instance;

    28

}

    29

    30

AIModeProc *AIObject::GetDoCommandMode()

    31

{

    32

	return AIObject::GetDoCommandStruct(_current_player)->mode;

    33

}

    34

    35

AIObject *AIObject::GetDoCommandModeInstance()

    36

{

    37

	return AIObject::GetDoCommandStruct(_current_player)->mode_instance;

    38

}

    39

    40

void AIObject::SetDoCommandCosts(Money value)

    41

{

    42

	AIObject::GetDoCommandStruct(_current_player)->costs = CommandCost(value);

    43

}

    44

    45

void AIObject::IncreaseDoCommandCosts(Money value)

    46

{

    47

	AIObject::GetDoCommandStruct(_current_player)->costs.AddCost(value);

    48

}

    49

    50

Money AIObject::GetDoCommandCosts()

    51

{

    52

	return AIObject::GetDoCommandStruct(_current_player)->costs.GetCost();

    53

}

    54

    55

void AIObject::SetNewVehicleID(VehicleID vehicle)

    56

{

    57

	AIObject::GetDoCommandStruct(_current_player)->new_vehicle_id = vehicle;

    58

}

    59

    60

VehicleID AIObject::GetNewVehicleID()

    61

{

    62

	return AIObject::GetDoCommandStruct(_current_player)->new_vehicle_id;

    63

}

    64

    65

void AIObject::SetNewSignID(SignID sign)

    66

{

    67

	AIObject::GetDoCommandStruct(_current_player)->new_sign_id = sign;

    68

}

    69

    70

SignID AIObject::GetNewSignID()

    71

{

    72

	return AIObject::GetDoCommandStruct(_current_player)->new_sign_id;

    73

}

    74

    75

AIObject::AIDoCommandStruct *AIObject::GetDoCommandStruct(PlayerID player)

    76

{

    77

	/* Storage for data on per-AI level */

    78

	static AIObject::AIDoCommandStruct command_struct[MAX_PLAYERS];

    79

    80

	return &command_struct[player];

    81

}

    82

    83

void AIObject::ResetInternalPlayerData()

    84

{

    85

	AIObject::AIDoCommandStruct *command_struct = GetDoCommandStruct(_current_player);

    86

	command_struct->mode = NULL;

    87

	command_struct->delay = 1;

    88

	command_struct->costs = CommandCost();

    89

}

    90

    91

bool AIObject::DoCommand(TileIndex tile, uint32 p1, uint32 p2, uint procc)

    92

{

    93

	uint32 flags = 0;

    94

	PlayerID old_lp;

    95

	CommandCost res;

    96

	const char* tmp_cmdtext;

    97

    98

	flags |= DC_NO_WATER;

    99

	if (procc != CMD_LANDSCAPE_CLEAR) flags |= DC_AUTO;

   100

   101

	/* The test already resets _cmd_text, so backup the pointer */

   102

	tmp_cmdtext = _cmd_text;

   103

   104

	/* First, do a test-run to see if we can do this */

   105

	res = ::DoCommand(tile, p1, p2, flags, procc);

   106

	/* The command failed, so return */

   107

	if (::CmdFailed(res)) return false;

   108

   109

	/* Restore _cmd_text */

   110

	_cmd_text = tmp_cmdtext;

   111

   112

	/* Check what the callback wants us to do */

   113

	if (AIObject::GetDoCommandMode() != NULL && !AIObject::GetDoCommandMode()(tile, p1, p2, procc, res)) {

   114

		AIObject::IncreaseDoCommandCosts(res.GetCost());

   115

		return true;

   116

	}

   117

   118

	/* If we did a DC_EXEC, and the command did not return an error, execute it

   119

	 * over the network */

   120

	if (flags & DC_AUTO)     procc |= CMD_AUTO;

   121

	if (flags & DC_NO_WATER) procc |= CMD_NO_WATER;

   122

   123

#ifdef ENABLE_NETWORK

   124

	/* Send the command */

   125

	if (_networking) {

   126

		/* NetworkSend_Command needs _local_player to be set correctly, so

   127

		 * adjust it, and put it back right after the function */

   128

		old_lp = _local_player;

   129

		_local_player = _current_player;

   130

		::NetworkSend_Command(tile, p1, p2, procc, CcAI);

   131

		_local_player = old_lp;

   132

   133

		/* Suspend the AI till the command is really executed */

   134

		AI_SuspendPlayer(_current_player, -(int)AIObject::GetDoCommandDelay());

   135

		/* Check if the callback still agrees with us, else return error */

   136

		if (!AI_GetCallbackResult(_current_player)) res = CMD_ERROR;

   137

	} else {

   138

#else

   139

	{

   140

#endif

   141

		/* For SinglePlayer we execute the command immediatly */

   142

		::DoCommandP(tile, p1, p2, NULL, procc);

   143

		/* Store some values inside the AIObject static memory */

   144

		AIObject::SetNewVehicleID(_new_vehicle_id);

   145

		AIObject::SetNewSignID(_new_sign_id);

   146

   147

		/* Suspend the AI player for 1 tick, so it simulates MultiPlayer */

   148

		AI_SuspendPlayer(_current_player, AIObject::GetDoCommandDelay());

   149

	}

   150

   151

	if (::CmdFailed(res)) return false;

   152

   153

	AIObject::IncreaseDoCommandCosts(res.GetCost());

   154

	return true;

   155

}