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_func.h"

     7

#include "../../network/network.h"

     8

#include "../../player_func.h"

     9

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

    10

#include "../../vehicle_func.h"

    11

#include "../ai.h"

    12

#include "../ai_threads.h"

    13

    14

void AIObject::SetDoCommandDelay(uint ticks)

    15

{

    16

	assert(ticks > 0);

    17

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

    18

}

    19

    20

uint AIObject::GetDoCommandDelay()

    21

{

    22

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

    23

}

    24

    25

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

    26

{

    27

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

    28

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

    29

}

    30

    31

AIModeProc *AIObject::GetDoCommandMode()

    32

{

    33

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

    34

}

    35

    36

AIObject *AIObject::GetDoCommandModeInstance()

    37

{

    38

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

    39

}

    40

    41

void AIObject::SetDoCommandCosts(Money value)

    42

{

    43

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

    44

}

    45

    46

void AIObject::IncreaseDoCommandCosts(Money value)

    47

{

    48

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

    49

}

    50

    51

Money AIObject::GetDoCommandCosts()

    52

{

    53

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

    54

}

    55

    56

void AIObject::SetLastCommandRes(bool res)

    57

{

    58

	AIObject::GetDoCommandStruct(_current_player)->last_command_res = res;

    59

}

    60

    61

bool AIObject::GetLastCommandRes()

    62

{

    63

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

    64

}

    65

    66

void AIObject::SetNewVehicleID(VehicleID vehicle)

    67

{

    68

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

    69

}

    70

    71

VehicleID AIObject::GetNewVehicleID()

    72

{

    73

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

    74

}

    75

    76

void AIObject::SetNewSignID(SignID sign)

    77

{

    78

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

    79

}

    80

    81

SignID AIObject::GetNewSignID()

    82

{

    83

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

    84

}

    85

    86

void *&AIObject::GetEventPointer()

    87

{

    88

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

    89

}

    90

    91

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

    92

{

    93

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

    94

	static AIObject::AIDoCommandStruct command_struct[MAX_PLAYERS];

    95

	static bool initialized = false;

    96

    97

	/* Make sure all memory is NULL when we start */

    98

	if (!initialized) {

    99

		initialized = true;

   100

		memset(&command_struct, 0, sizeof(command_struct));

   101

	}

   102

   103

	return &command_struct[player];

   104

}

   105

   106

void AIObject::ResetInternalPlayerData()

   107

{

   108

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

   109

	command_struct->mode = NULL;

   110

	command_struct->delay = 1;

   111

	command_struct->costs = CommandCost();

   112

	if (command_struct->event_data != NULL) AIEventController::FreeEventPointer();

   113

	command_struct->event_data = NULL;

   114

}

   115

   116

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

   117

{

   118

	uint32 flags = 0;

   119

	PlayerID old_lp;

   120

	CommandCost res;

   121

	const char *tmp_cmdtext;

   122

   123

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

   124

	if (water_protection) flags |= DC_NO_WATER;

   125

   126

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

   127

	tmp_cmdtext = _cmd_text;

   128

   129

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

   130

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

   131

	/* The command failed, so return */

   132

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

   133

   134

	/* Restore _cmd_text */

   135

	_cmd_text = tmp_cmdtext;

   136

   137

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

   138

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

   139

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

   140

		return true;

   141

	}

   142

   143

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

   144

	 * over the network */

   145

	if (flags & DC_NO_WATER) procc |= CMD_NO_WATER;

   146

   147

#ifdef ENABLE_NETWORK

   148

	/* Send the command */

   149

	if (_networking) {

   150

		/* NetworkSend_Command needs _local_player to be set correctly, so

   151

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

   152

		old_lp = _local_player;

   153

		_local_player = _current_player;

   154

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

   155

		_local_player = old_lp;

   156

   157

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

   158

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

   159

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

   160

		if (!AIObject::GetLastCommandRes()) res = CMD_ERROR;

   161

	} else {

   162

#else

   163

	{

   164

#endif

   165

		/* For SinglePlayer we execute the command immediatly */

   166

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

   167

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

   168

		AIObject::SetNewVehicleID(_new_vehicle_id);

   169

		AIObject::SetNewSignID(_new_sign_id);

   170

   171

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

   172

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

   173

	}

   174

   175

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

   176

   177

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

   178

	return true;

   179

}