1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "table/strings.h"

     6

#include "table/sprites.h"

     7

#include "functions.h"

     8

#include "window.h"

     9

#include "gui.h"

    10

#include "gfx.h"

    11

#include "player.h"

    12

#include "economy.h"

    13

#include "signs.h"

    14

#include "strings.h"

    15

#include "debug.h"

    16

#include "variables.h"

    17

    18

static uint _legend_excludebits;

    19

static uint _legend_cargobits;

    20

    21

/************************/

    22

/* GENERIC GRAPH DRAWER */

    23

/************************/

    24

    25

enum {GRAPH_NUM = 16};

    26

    27

typedef struct GraphDrawer {

    28

	uint sel; // bitmask of the players *excluded* (e.g. 11111111 means that no players are shown)

    29

	byte num_dataset;

    30

	byte num_on_x_axis;

    31

	byte month;

    32

	byte year;

    33

	bool include_neg;

    34

	byte num_vert_lines;

    35

	uint16 unk61A;

    36

	uint16 unk61C;

    37

	int left, top;

    38

	uint height;

    39

	StringID format_str_y_axis;

    40

	byte color_3, color_2, bg_line_color;

    41

	byte colors[GRAPH_NUM];

    42

	uint64 cost[GRAPH_NUM][24]; // last 2 years

    43

} GraphDrawer;

    44

    45

#define INVALID_VALUE 0x80000000

    46

    47

static void DrawGraph(const GraphDrawer *gw)

    48

{

    49

    50

	int i,j,k;

    51

	uint x,y,old_x,old_y;

    52

	int color;

    53

	int right, bottom;

    54

	int num_x, num_dataset;

    55

	const uint64 *row_ptr, *col_ptr;

    56

	int64 mx;

    57

	int adj_height;

    58

	uint64 y_scaling, tmp;

    59

	int64 value;

    60

	int64 cur_val;

    61

	uint sel;

    62

    63

	/* the colors and cost array of GraphDrawer must accomodate

    64

	 * both values for cargo and players. So if any are higher, quit */

    65

	assert(GRAPH_NUM >= NUM_CARGO && GRAPH_NUM >= MAX_PLAYERS);

    66

    67

	color = _color_list[gw->bg_line_color].window_color_1b;

    68

    69

	/* draw the vertical lines */

    70

	i = gw->num_vert_lines; assert(i > 0);

    71

	x = gw->left + 66;

    72

	bottom = gw->top + gw->height - 1;

    73

	do {

    74

		GfxFillRect(x, gw->top, x, bottom, color);

    75

		x += 22;

    76

	} while (--i);

    77

    78

	/* draw the horizontal lines */

    79

	i = 9;

    80

	x = gw->left + 44;

    81

	y = gw->height + gw->top;

    82

	right = gw->left + 44 + gw->num_vert_lines*22-1;

    83

    84

	do {

    85

		GfxFillRect(x, y, right, y, color);

    86

		y -= gw->height >> 3;

    87

	} while (--i);

    88

    89

	/* draw vertical edge line */

    90

	GfxFillRect(x, gw->top, x, bottom, gw->color_2);

    91

    92

	adj_height = gw->height;

    93

	if (gw->include_neg) adj_height >>= 1;

    94

    95

	/* draw horiz edge line */

    96

	y = adj_height + gw->top;

    97

	GfxFillRect(x, y, right, y, gw->color_2);

    98

    99

	/* find the max element */

   100

	if (gw->num_on_x_axis == 0)

   101

		return;

   102

   103

	num_dataset = gw->num_dataset;

   104

	assert(num_dataset > 0);

   105

   106

	row_ptr = gw->cost[0];

   107

	mx = 0;

   108

		/* bit selection for the showing of various players, base max element

   109

		 * on to-be shown player-information. This way the graph can scale */

   110

	sel = gw->sel;

   111

	do {

   112

		if (!(sel&1)) {

   113

			num_x = gw->num_on_x_axis;

   114

			assert(num_x > 0);

   115

			col_ptr = row_ptr;

   116

			do {

   117

				if (*col_ptr != INVALID_VALUE) {

   118

					mx = max64(mx, myabs64(*col_ptr));

   119

				}

   120

			} while (col_ptr++, --num_x);

   121

		}

   122

	} while (sel>>=1, row_ptr+=24, --num_dataset);

   123

   124

	/* setup scaling */

   125

	y_scaling = INVALID_VALUE;

   126

	value = adj_height * 2;

   127

   128

	if (mx > value) {

   129

		mx = (mx + 7) & ~7;

   130

		y_scaling = (((uint64) (value>>1) << 32) / mx);

   131

		value = mx;

   132

	}

   133

   134

	/* draw text strings on the y axis */

   135

	tmp = value;

   136

	if (gw->include_neg) tmp >>= 1;

   137

	x = gw->left + 45;

   138

	y = gw->top - 3;

   139

	i = 9;

   140

	do {

   141

		SetDParam(0, gw->format_str_y_axis);

   142

		SetDParam64(1, (int64)tmp);

   143

		tmp -= (value >> 3);

   144

		DrawStringRightAligned(x, y, STR_0170, gw->color_3);

   145

		y += gw->height >> 3;

   146

	} while (--i);

   147

   148

	/* draw strings on the x axis */

   149

	if (gw->month != 0xFF) {

   150

		x = gw->left + 44;

   151

		y = gw->top + gw->height + 1;

   152

		j = gw->month;

   153

		k = gw->year + MAX_YEAR_BEGIN_REAL;

   154

		i = gw->num_on_x_axis;assert(i>0);

   155

		do {

   156

			SetDParam(2, k);

   157

			SetDParam(0, j + STR_0162_JAN);

   158

			SetDParam(1, j + STR_0162_JAN + 2);

   159

			DrawString(x, y, j == 0 ? STR_016F : STR_016E, gw->color_3);

   160

   161

			j += 3;

   162

			if (j >= 12) {

   163

				j = 0;

   164

				k++;

   165

			}

   166

			x += 22;

   167

		} while (--i);

   168

	} else {

   169

		x = gw->left + 52;

   170

		y = gw->top + gw->height + 1;

   171

		j = gw->unk61A;

   172

		i = gw->num_on_x_axis;assert(i>0);

   173

		do {

   174

			SetDParam(0, j);

   175

			DrawString(x, y, STR_01CB, gw->color_3);

   176

			j += gw->unk61C;

   177

			x += 22;

   178

		} while (--i);

   179

	}

   180

   181

	/* draw lines and dots */

   182

	i = 0;

   183

	row_ptr = gw->cost[0];

   184

	sel = gw->sel; // show only selected lines. GraphDrawer qw->sel set in Graph-Legend (_legend_excludebits)

   185

	do {

   186

		if (!(sel & 1)) {

   187

			x = gw->left + 55;

   188

			j = gw->num_on_x_axis;assert(j>0);

   189

			col_ptr = row_ptr;

   190

			color = gw->colors[i];

   191

			old_y = old_x = INVALID_VALUE;

   192

			do {

   193

				cur_val = *col_ptr++;

   194

				if (cur_val != INVALID_VALUE) {

   195

					y = adj_height - BIGMULSS64(cur_val, y_scaling >> 1, 31) + gw->top;

   196

   197

					GfxFillRect(x-1, y-1, x+1, y+1, color);

   198

					if (old_x != INVALID_VALUE)

   199

						GfxDrawLine(old_x, old_y, x, y, color);

   200

   201

					old_x = x;

   202

					old_y = y;

   203

				} else {

   204

					old_x = INVALID_VALUE;

   205

				}

   206

			} while (x+=22,--j);

   207

		}

   208

	} while (sel>>=1,row_ptr+=24, ++i < gw->num_dataset);

   209

}

   210

   211

/****************/

   212

/* GRAPH LEGEND */

   213

/****************/

   214

   215

void DrawPlayerIcon(int p, int x, int y)

   216

{

   217

	DrawSprite(SPRITE_PALETTE(PLAYER_SPRITE_COLOR(p) + 0x2EB), x, y);

   218

}

   219

   220

static void GraphLegendWndProc(Window *w, WindowEvent *e)

   221

{

   222

	const Player* p;

   223

   224

	switch(e->event) {

   225

	case WE_PAINT:

   226

		FOR_ALL_PLAYERS(p) {

   227

			if (!p->is_active) SETBIT(_legend_excludebits, p->index);

   228

		}

   229

		w->click_state = (~_legend_excludebits) << 3;

   230

		DrawWindowWidgets(w);

   231

   232

		FOR_ALL_PLAYERS(p) {

   233

			if (!p->is_active) continue;

   234

   235

			DrawPlayerIcon(p->index, 4, 18+p->index*12);

   236

   237

			SetDParam(0, p->name_1);

   238

			SetDParam(1, p->name_2);

   239

			SetDParam(2, GetPlayerNameString(p->index, 3));

   240

			DrawString(21,17+p->index*12,STR_7021,HASBIT(_legend_excludebits, p->index) ? 0x10 : 0xC);

   241

		}

   242

		break;

   243

   244

	case WE_CLICK:

   245

		if (IS_INT_INSIDE(e->click.widget, 3, 11)) {

   246

			_legend_excludebits ^= (1 << (e->click.widget - 3));

   247

			SetWindowDirty(w);

   248

			InvalidateWindow(WC_INCOME_GRAPH, 0);

   249

			InvalidateWindow(WC_OPERATING_PROFIT, 0);

   250

			InvalidateWindow(WC_DELIVERED_CARGO, 0);

   251

			InvalidateWindow(WC_PERFORMANCE_HISTORY, 0);

   252

			InvalidateWindow(WC_COMPANY_VALUE, 0);

   253

		}

   254

		break;

   255

	}

   256

}

   257

   258

static const Widget _graph_legend_widgets[] = {

   259

{    WWT_TEXTBTN,   RESIZE_NONE,    14,     0,    10,     0,    13, STR_00C5, STR_018B_CLOSE_WINDOW},

   260

{    WWT_CAPTION,   RESIZE_NONE,    14,    11,   249,     0,    13, STR_704E_KEY_TO_COMPANY_GRAPHS, STR_018C_WINDOW_TITLE_DRAG_THIS},

   261

{     WWT_IMGBTN,   RESIZE_NONE,    14,     0,   249,    14,   113, 0x0,STR_NULL},

   262

{     WWT_IMGBTN,   RESIZE_NONE,    14,     2,   247,    16,    27, 0x0,STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   263

{     WWT_IMGBTN,   RESIZE_NONE,    14,     2,   247,    28,    39, 0x0,STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   264

{     WWT_IMGBTN,   RESIZE_NONE,    14,     2,   247,    40,    51, 0x0,STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   265

{     WWT_IMGBTN,   RESIZE_NONE,    14,     2,   247,    52,    63, 0x0,STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   266

{     WWT_IMGBTN,   RESIZE_NONE,    14,     2,   247,    64,    75, 0x0,STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   267

{     WWT_IMGBTN,   RESIZE_NONE,    14,     2,   247,    76,    87, 0x0,STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   268

{     WWT_IMGBTN,   RESIZE_NONE,    14,     2,   247,    88,    99, 0x0,STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   269

{     WWT_IMGBTN,   RESIZE_NONE,    14,     2,   247,   100,   111, 0x0,STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY},

   270

{   WIDGETS_END},

   271

};

   272

   273

static const WindowDesc _graph_legend_desc = {

   274

	-1, -1, 250, 114,

   275

	WC_GRAPH_LEGEND,0,

   276

	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET,

   277

	_graph_legend_widgets,

   278

	GraphLegendWndProc

   279

};

   280

   281

static void ShowGraphLegend(void)

   282

{

   283

	AllocateWindowDescFront(&_graph_legend_desc, 0);

   284

}

   285

   286

/********************/

   287

/* OPERATING PROFIT */

   288

/********************/

   289

   290

static void SetupGraphDrawerForPlayers(GraphDrawer *gd)

   291

{

   292

	const Player* p;

   293

	uint excludebits = _legend_excludebits;

   294

	int nums;

   295

	int mo,yr;

   296

   297

	// Exclude the players which aren't valid

   298

	FOR_ALL_PLAYERS(p) {

   299

		if (!p->is_active) SETBIT(excludebits,p->index);

   300

	}

   301

	gd->sel = excludebits;

   302

	gd->num_vert_lines = 24;

   303

   304

	nums = 0;

   305

	FOR_ALL_PLAYERS(p) {

   306

		if (p->is_active) nums = max(nums,p->num_valid_stat_ent);

   307

	}

   308

	gd->num_on_x_axis = min(nums,24);

   309

   310

	mo = (_cur_month/3-nums)*3;

   311

	yr = _cur_year;

   312

	while (mo < 0) {

   313

		yr--;

   314

		mo += 12;

   315

	}

   316

   317

	gd->year = yr;

   318

	gd->month = mo;

   319

}

   320

   321

static void OperatingProfitWndProc(Window *w, WindowEvent *e)

   322

{

   323

	switch(e->event) {

   324

	case WE_PAINT: {

   325

		GraphDrawer gd;

   326

		const Player* p;

   327

		int i,j;

   328

		int numd;

   329

   330

		DrawWindowWidgets(w);

   331

   332

		gd.left = 2;

   333

		gd.top = 18;

   334

		gd.height = 136;

   335

		gd.include_neg = true;

   336

		gd.format_str_y_axis = STR_CURRCOMPACT;

   337

		gd.color_3 = 0x10;

   338

		gd.color_2 = 0xD7;

   339

		gd.bg_line_color = 0xE;

   340

   341

		SetupGraphDrawerForPlayers(&gd);

   342

   343

		numd = 0;

   344

		FOR_ALL_PLAYERS(p) {

   345

			if (p->is_active) {

   346

				gd.colors[numd] = _color_list[p->player_color].window_color_bgb;

   347

				for(j=gd.num_on_x_axis,i=0; --j >= 0;) {

   348

					gd.cost[numd][i] = (j >= p->num_valid_stat_ent) ? INVALID_VALUE : (uint64)(p->old_economy[j].income + p->old_economy[j].expenses);

   349

					i++;

   350

				}

   351

			}

   352

			numd++;

   353

		}

   354

   355

		gd.num_dataset = numd;

   356

   357

		DrawGraph(&gd);

   358

	}	break;

   359

	case WE_CLICK:

   360

		if (e->click.widget == 2) /* Clicked on Legend */

   361

			ShowGraphLegend();

   362

		break;

   363

	}

   364

}

   365

   366

static const Widget _operating_profit_widgets[] = {

   367

{    WWT_TEXTBTN,   RESIZE_NONE,    14,     0,    10,     0,    13, STR_00C5,												STR_018B_CLOSE_WINDOW},

   368

{    WWT_CAPTION,   RESIZE_NONE,    14,    11,   525,     0,    13, STR_7025_OPERATING_PROFIT_GRAPH, STR_018C_WINDOW_TITLE_DRAG_THIS},

   369

{ WWT_PUSHTXTBTN,   RESIZE_NONE,    14,   526,   575,     0,    13, STR_704C_KEY,										STR_704D_SHOW_KEY_TO_GRAPHS},

   370

{     WWT_IMGBTN,   RESIZE_NONE,    14,     0,   575,    14,   173, 0x0,															STR_NULL},

   371

{   WIDGETS_END},

   372

};

   373

   374

static const WindowDesc _operating_profit_desc = {

   375

	-1, -1, 576, 174,

   376

	WC_OPERATING_PROFIT,0,

   377

	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS,

   378

	_operating_profit_widgets,

   379

	OperatingProfitWndProc

   380

};

   381

   382

   383

void ShowOperatingProfitGraph(void)

   384

{

   385

	if (AllocateWindowDescFront(&_operating_profit_desc, 0)) {

   386

		InvalidateWindow(WC_GRAPH_LEGEND, 0);

   387

	}

   388

}

   389

   390

   391

/****************/

   392

/* INCOME GRAPH */

   393

/****************/

   394

   395

static void IncomeGraphWndProc(Window *w, WindowEvent *e)

   396

{

   397

	switch(e->event) {

   398

	case WE_PAINT: {

   399

		GraphDrawer gd;

   400

		const Player* p;

   401

		int i,j;

   402

		int numd;

   403

   404

		DrawWindowWidgets(w);

   405

   406

		gd.left = 2;

   407

		gd.top = 18;

   408

		gd.height = 104;

   409

		gd.include_neg = false;

   410

		gd.format_str_y_axis = STR_CURRCOMPACT;

   411

		gd.color_3 = 0x10;

   412

		gd.color_2 = 0xD7;

   413

		gd.bg_line_color = 0xE;

   414

		SetupGraphDrawerForPlayers(&gd);

   415

   416

		numd = 0;

   417

		FOR_ALL_PLAYERS(p) {

   418

			if (p->is_active) {

   419

				gd.colors[numd] = _color_list[p->player_color].window_color_bgb;

   420

				for(j=gd.num_on_x_axis,i=0; --j >= 0;) {

   421

					gd.cost[numd][i] = (j >= p->num_valid_stat_ent) ? INVALID_VALUE : (uint64)p->old_economy[j].income;

   422

					i++;

   423

				}

   424

			}

   425

			numd++;

   426

		}

   427

   428

		gd.num_dataset = numd;

   429

   430

		DrawGraph(&gd);

   431

		break;

   432

	}

   433

   434

	case WE_CLICK:

   435

		if (e->click.widget == 2)

   436

			ShowGraphLegend();

   437

		break;

   438

	}

   439

}

   440

   441

static const Widget _income_graph_widgets[] = {

   442

{    WWT_TEXTBTN,   RESIZE_NONE,    14,     0,    10,     0,    13, STR_00C5,							STR_018B_CLOSE_WINDOW},

   443

{    WWT_CAPTION,   RESIZE_NONE,    14,    11,   525,     0,    13, STR_7022_INCOME_GRAPH, STR_018C_WINDOW_TITLE_DRAG_THIS},

   444

{ WWT_PUSHTXTBTN,   RESIZE_NONE,    14,   526,   575,     0,    13, STR_704C_KEY,					STR_704D_SHOW_KEY_TO_GRAPHS},

   445

{     WWT_IMGBTN,   RESIZE_NONE,    14,     0,   575,    14,   141, 0x0,										STR_NULL},

   446

{   WIDGETS_END},

   447

};

   448

   449

static const WindowDesc _income_graph_desc = {

   450

	-1, -1, 576, 142,

   451

	WC_INCOME_GRAPH,0,

   452

	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS,

   453

	_income_graph_widgets,

   454

	IncomeGraphWndProc

   455

};

   456

   457

void ShowIncomeGraph(void)

   458

{

   459

	if (AllocateWindowDescFront(&_income_graph_desc, 0)) {

   460

		InvalidateWindow(WC_GRAPH_LEGEND, 0);

   461

	}

   462

}

   463

   464

/*******************/

   465

/* DELIVERED CARGO */

   466

/*******************/

   467

   468

static void DeliveredCargoGraphWndProc(Window *w, WindowEvent *e)

   469

{

   470

	switch(e->event) {

   471

	case WE_PAINT: {

   472

		GraphDrawer gd;

   473

		const Player* p;

   474

		int i,j;

   475

		int numd;

   476

   477

		DrawWindowWidgets(w);

   478

   479

		gd.left = 2;

   480

		gd.top = 18;

   481

		gd.height = 104;

   482

		gd.include_neg = false;

   483

		gd.format_str_y_axis = STR_7024;

   484

		gd.color_3 = 0x10;

   485

		gd.color_2 = 0xD7;

   486

		gd.bg_line_color = 0xE;

   487

		SetupGraphDrawerForPlayers(&gd);

   488

   489

		numd = 0;

   490

		FOR_ALL_PLAYERS(p) {

   491

			if (p->is_active) {

   492

				gd.colors[numd] = _color_list[p->player_color].window_color_bgb;

   493

				for(j=gd.num_on_x_axis,i=0; --j >= 0;) {

   494

					gd.cost[numd][i] = (j >= p->num_valid_stat_ent) ? INVALID_VALUE : (uint64)p->old_economy[j].delivered_cargo;

   495

					i++;

   496

				}

   497

			}

   498

			numd++;

   499

		}

   500

   501

		gd.num_dataset = numd;

   502

   503

		DrawGraph(&gd);

   504

		break;

   505

	}

   506

   507

	case WE_CLICK:

   508

		if (e->click.widget == 2)

   509

			ShowGraphLegend();

   510

		break;