1

/* $Id$ */

     2

     3

/** @file ai_gui.cpp */

     4

     5

#include "../stdafx.h"

     6

#include "../openttd.h"

     7

#include "../gui.h"

     8

#include "../window_gui.h"

     9

#include "../player_func.h"

    10

#include "../player_base.h"

    11

#include "../player_gui.h"

    12

#include "../economy_func.h"

    13

#include "../variables.h"

    14

#include "../cargotype.h"

    15

#include "../strings_func.h"

    16

#include "../core/alloc_func.hpp"

    17

#include "../window_func.h"

    18

#include "../date_func.h"

    19

#include "../gfx_func.h"

    20

#include "../debug.h"

    21

#include "../command_func.h"

    22

    23

#include "ai.h"

    24

#include "api/ai_types.hpp"

    25

#include "api/ai_controller.hpp"

    26

#include "api/ai_object.hpp"

    27

#include "api/ai_log.hpp"

    28

#include "ai_info.hpp"

    29

    30

#include "table/strings.h"

    31

#include "../table/sprites.h"

    32

    33

extern AIInfo *AI_GetPlayerInfo(PlayerID player);

    34

    35

struct AIDebugWindow : public Window {

    36

	enum AIDebugWindowWidgets {

    37

		AID_WIDGET_CLOSEBOX = 0,

    38

		AID_WIDGET_CAPTION,

    39

		AID_WIDGET_VIEW,

    40

		AID_WIDGET_NAME_TEXT,

    41

		AID_WIDGET_RELOAD_TOGGLE,

    42

		AID_WIDGET_LOG_PANEL,

    43

		AID_WIDGET_SCROLLBAR,

    44

		AID_WIDGET_UNUSED_1,

    45

		AID_WIDGET_UNUSED_2,

    46

		AID_WIDGET_UNUSED_3,

    47

		AID_WIDGET_UNUSED_4,

    48

		AID_WIDGET_UNUSED_5,

    49

		AID_WIDGET_UNUSED_6,

    50

		AID_WIDGET_UNUSED_7,

    51

    52

		AID_WIDGET_COMPANY_BUTTON_START,

    53

		AID_WIDGET_COMPANY_BUTTON_END = AID_WIDGET_COMPANY_BUTTON_START + 7,

    54

	};

    55

    56

	static PlayerID ai_debug_player;

    57

	int redraw_timer;

    58

    59

	AIDebugWindow(const WindowDesc *desc, WindowNumber number) : Window(desc, number)

    60

	{

    61

		/* Disable the players who are not active or not an AI */

    62

		for (PlayerID i = PLAYER_FIRST; i < MAX_PLAYERS; i++) {

    63

			this->SetWidgetDisabledState(i + AID_WIDGET_COMPANY_BUTTON_START, !GetPlayer(i)->is_active || !GetPlayer(i)->is_ai);

    64

		}

    65

		this->DisableWidget(AID_WIDGET_RELOAD_TOGGLE);

    66

    67

		this->vscroll.cap = 14;

    68

		this->vscroll.pos = 0;

    69

		this->resize.step_height = 12;

    70

    71

		if (ai_debug_player != INVALID_PLAYER) this->LowerWidget(ai_debug_player + AID_WIDGET_COMPANY_BUTTON_START);

    72

    73

		this->FindWindowPlacementAndResize(desc);

    74

	}

    75

    76

	virtual void OnPaint()

    77

	{

    78

		/* Check if the currently selected player is still active. */

    79

		if (ai_debug_player == INVALID_PLAYER || !GetPlayer(ai_debug_player)->is_active) {

    80

			if (ai_debug_player != INVALID_PLAYER) {

    81

				/* Raise and disable the widget for the previous selection. */

    82

				this->RaiseWidget(ai_debug_player + AID_WIDGET_COMPANY_BUTTON_START);

    83

				this->DisableWidget(ai_debug_player + AID_WIDGET_COMPANY_BUTTON_START);

    84

    85

				ai_debug_player = INVALID_PLAYER;

    86

			}

    87

    88

			for (PlayerID i = PLAYER_FIRST; i < MAX_PLAYERS; i++) {

    89

				if (GetPlayer(i)->is_active && GetPlayer(i)->is_ai) {

    90

					/* Lower the widget corresponding to this player. */

    91

					this->LowerWidget(i + AID_WIDGET_COMPANY_BUTTON_START);

    92

    93

					ai_debug_player = i;

    94

					break;

    95

				}

    96

			}

    97

		}

    98

    99

		/* Update "Reload AI" button */

   100

		this->SetWidgetDisabledState(AID_WIDGET_RELOAD_TOGGLE, ai_debug_player == INVALID_PLAYER);

   101

   102

		/* Draw standard stuff */

   103

		this->DrawWidgets();

   104

   105

		/* If there are no active players, don't display anything else. */

   106

		if (ai_debug_player == INVALID_PLAYER) return;

   107

   108

		/* Paint the player icons */

   109

		for (PlayerID i = PLAYER_FIRST; i < MAX_PLAYERS; i++) {

   110

			if (!GetPlayer(i)->is_active || !GetPlayer(i)->is_ai) {

   111

				/* Check if we have the player as an active player */

   112

				if (!this->IsWidgetDisabled(i + AID_WIDGET_COMPANY_BUTTON_START)) {

   113

					/* Bah, player gone :( */

   114

					this->DisableWidget(i + AID_WIDGET_COMPANY_BUTTON_START);

   115

   116

					/* We need a repaint */

   117

					this->SetDirty();

   118

				}

   119

				continue;

   120

			}

   121

   122

			/* Check if we have the player marked as inactive */

   123

			if (this->IsWidgetDisabled(i + AID_WIDGET_COMPANY_BUTTON_START)) {

   124

				/* New AI! Yippie :p */

   125

				this->EnableWidget(i + AID_WIDGET_COMPANY_BUTTON_START);

   126

   127

				/* We need a repaint */

   128

				this->SetDirty();

   129

			}

   130

   131

			byte x = (i == ai_debug_player) ? 1 : 0;

   132

			DrawPlayerIcon(i, i * 37 + 13 + x, 16 + x);

   133

		}

   134

   135

		/* Draw the AI name */

   136

		AIInfo *info = AI_GetPlayerInfo(ai_debug_player);

   137

		assert(info != NULL);

   138

		DoDrawString(info->GetName(), 7, 34, TC_BLACK);

   139

   140

		PlayerID old_cp = _current_player;

   141

		_current_player = ai_debug_player;

   142

		AILog::LogData *log = (AILog::LogData *)AIObject::GetLogPointer();

   143

		_current_player = old_cp;

   144

   145

		SetVScrollCount(this, (log == NULL) ? 0 : log->used);

   146

		this->InvalidateWidget(AID_WIDGET_SCROLLBAR);

   147

		if (log == NULL) return;

   148

   149

		int y = 6;

   150

		for (int i = this->vscroll.pos; i < (this->vscroll.cap + this->vscroll.pos); i++) {

   151

			uint pos = (log->count + log->pos - i) % log->count;

   152

			if (log->lines[pos] == NULL) break;

   153

   154

			uint colour;

   155

			switch (log->type[pos]) {

   156

				case AILog::LOG_SQ_INFO:  colour = TC_BLACK;  break;

   157

				case AILog::LOG_SQ_ERROR: colour = TC_RED;    break;

   158

				case AILog::LOG_INFO:     colour = TC_BLACK;  break;

   159

				case AILog::LOG_WARNING:  colour = TC_YELLOW; break;

   160

				case AILog::LOG_ERROR:    colour = TC_RED;    break;

   161

				default:                  colour = TC_BLACK;  break;

   162

			}

   163

   164

			DoDrawStringTruncated(log->lines[pos], 7, this->widget[AID_WIDGET_LOG_PANEL].top + y, colour, this->widget[AID_WIDGET_LOG_PANEL].right - this->widget[AID_WIDGET_LOG_PANEL].left - 14);

   165

			y += 12;

   166

		}

   167

	}

   168

   169

	virtual void OnClick(Point pt, int widget)

   170

	{

   171

		/* Check which button is clicked */

   172

		if (IsInsideMM(widget, AID_WIDGET_COMPANY_BUTTON_START, AID_WIDGET_COMPANY_BUTTON_END + 1)) {

   173

			/* Is it no on disable? */

   174

			if (!this->IsWidgetDisabled(widget)) {

   175

				this->RaiseWidget(ai_debug_player + AID_WIDGET_COMPANY_BUTTON_START);

   176

				ai_debug_player = (PlayerID)(widget - AID_WIDGET_COMPANY_BUTTON_START);

   177

				this->LowerWidget(ai_debug_player + AID_WIDGET_COMPANY_BUTTON_START);

   178

				this->SetDirty();

   179

			}

   180

		}

   181

		if (widget == AID_WIDGET_RELOAD_TOGGLE && !this->IsWidgetDisabled(widget)) {

   182

			/* Set the current AI as forced next AI */

   183

			AIInfo *info = AI_GetPlayerInfo(ai_debug_player);

   185

   186

			/* First kill the company of the AI, then start a new one. This should start the current AI again */

   187

			DoCommandP(0, 2, ai_debug_player, NULL, CMD_PLAYER_CTRL);

   188

			DoCommandP(0, 1, 0, NULL, CMD_PLAYER_CTRL);

   189

		}

   190

	}

   191

   192

	virtual void OnTimeout()

   193

	{

   194

		this->RaiseWidget(AID_WIDGET_RELOAD_TOGGLE);

   195

		this->SetDirty();

   196

	}

   197

   198

	virtual void OnInvalidateData(int data = 0)

   199

	{

   200

		if (data == -1 || ai_debug_player == data) this->SetDirty();

   201

	}

   202

   203

	virtual void OnResize(Point new_size, Point delta)

   204

	{

   205

		this->vscroll.cap += delta.y / (int)this->resize.step_height;

   206

	}

   207

};

   208

   209

PlayerID AIDebugWindow::ai_debug_player = INVALID_PLAYER;

   210

   211

static const Widget _ai_debug_widgets[] = {

   212

{   WWT_CLOSEBOX,   RESIZE_NONE,    14,     0,    10,     0,    13, STR_00C5,                   STR_018B_CLOSE_WINDOW},                 // AID_WIDGET_CLOSEBOX

   213

{    WWT_CAPTION,  RESIZE_RIGHT,    14,    11,   298,     0,    13, STR_AI_DEBUG,               STR_018C_WINDOW_TITLE_DRAG_THIS},       // AID_WIDGET_CAPTION

   214

{      WWT_PANEL,  RESIZE_RIGHT,    14,     0,   298,    14,    27, 0x0,                        STR_NULL},                              // AID_WIDGET_VIEW

   215

   216

{      WWT_PANEL,  RESIZE_RIGHT,    14,     0,   149,    28,    47, 0x0,                        STR_AI_DEBUG_NAME_TIP},                 // AID_WIDGET_NAME_TEXT

   217

{ WWT_PUSHTXTBTN,     RESIZE_LR,    14,   150,   298,    28,    47, STR_AI_DEBUG_RELOAD,        STR_AI_DEBUG_RELOAD_TIP},               // AID_WIDGET_RELOAD_TOGGLE

   218

{      WWT_PANEL,     RESIZE_RB,    14,     0,   286,    48,   227, 0x0,                        STR_NULL},                              // AID_WIDGET_LOG_PANEL

   219

{  WWT_SCROLLBAR,    RESIZE_LRB,    14,   287,   298,    48,   215, STR_NULL,                   STR_0190_SCROLL_BAR_SCROLLS_LIST},      // AID_WIDGET_SCROLLBAR

   220

/* As this is WIP, leave the next few so we can work a bit with the GUI */

   221

{      WWT_EMPTY,   RESIZE_NONE,    14,     0,   298,    88,   107, 0x0,                        STR_NULL},                              // AID_WIDGET_UNUSED_1

   222

{      WWT_EMPTY,   RESIZE_NONE,    14,     0,   298,   108,   127, 0x0,                        STR_NULL},                              // AID_WIDGET_UNUSED_2

   223

{      WWT_EMPTY,   RESIZE_NONE,    14,     0,   298,   128,   147, 0x0,                        STR_NULL},                              // AID_WIDGET_UNUSED_3

   224

{      WWT_EMPTY,   RESIZE_NONE,    14,     0,   298,   148,   167, 0x0,                        STR_NULL},                              // AID_WIDGET_UNUSED_4

   225

{      WWT_EMPTY,   RESIZE_NONE,    14,     0,   298,   168,   187, 0x0,                        STR_NULL},                              // AID_WIDGET_UNUSED_5

   226

{      WWT_EMPTY,   RESIZE_NONE,    14,     0,   298,   188,   207, 0x0,                        STR_NULL},                              // AID_WIDGET_UNUSED_6

   227

{      WWT_EMPTY,   RESIZE_NONE,    14,     0,   298,   208,   227, 0x0,                        STR_NULL},                              // AID_WIDGET_UNUSED_7

   228

   229

{      WWT_PANEL,   RESIZE_NONE,    14,     2,    38,    14,    26, 0x0,                        STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, // AID_WIDGET_COMPANY_BUTTON_START

   230

{      WWT_PANEL,   RESIZE_NONE,    14,    39,    75,    14,    26, 0x0,                        STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   231

{      WWT_PANEL,   RESIZE_NONE,    14,    76,   112,    14,    26, 0x0,                        STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   232

{      WWT_PANEL,   RESIZE_NONE,    14,   113,   149,    14,    26, 0x0,                        STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   233

{      WWT_PANEL,   RESIZE_NONE,    14,   150,   186,    14,    26, 0x0,                        STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   234

{      WWT_PANEL,   RESIZE_NONE,    14,   187,   223,    14,    26, 0x0,                        STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   235

{      WWT_PANEL,   RESIZE_NONE,    14,   224,   260,    14,    26, 0x0,                        STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   236

{      WWT_PANEL,   RESIZE_NONE,    14,   261,   297,    14,    26, 0x0,                        STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, // AID_WIDGET_COMPANY_BUTTON_END

   237

{  WWT_RESIZEBOX,   RESIZE_LRTB,    14,   287,   298,   216,   227, STR_NULL,                   STR_RESIZE_BUTTON},

   238

{   WIDGETS_END},

   239

};

   240

   241

static const WindowDesc _ai_debug_desc = {

   242

	WDP_AUTO, WDP_AUTO, 299, 228, 299, 228,

   243

	WC_AI_DEBUG, WC_NONE,

   244

	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_RESIZABLE,

   245

	_ai_debug_widgets

   246

};

   247

   248

void ShowAIDebugWindow()

   249

{

   250

	AllocateWindowDescFront<AIDebugWindow>(&_ai_debug_desc, 0);

   251

}