1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "debug.h"

     6

#include "functions.h"

     7

#include "gui.h"

     8

#include "spritecache.h"

     9

#include "strings.h"

    10

#include "table/sprites.h"

    11

#include "table/strings.h"

    12

#include "map.h"

    13

#include "viewport.h"

    14

#include "window.h"

    15

#include "vehicle.h"

    16

#include "station.h"

    17

#include "gfx.h"

    18

#include "town.h"

    19

#include "signs.h"

    20

#include "waypoint.h"

    21

#include "variables.h"

    22

#include "train.h"

    23

    24

#define VIEWPORT_DRAW_MEM (65536 * 2)

    25

    26

static bool _added_tile_sprite;

    27

static bool _offset_ground_sprites;

    28

    29

typedef struct StringSpriteToDraw {

    30

	uint16 string;

    31

	uint16 color;

    32

	struct StringSpriteToDraw *next;

    33

	int32 x;

    34

	int32 y;

    35

	uint32 params[3];

    36

	uint16 width;

    37

} StringSpriteToDraw;

    38

    39

typedef struct TileSpriteToDraw {

    40

	uint32 image;

    41

	struct TileSpriteToDraw *next;

    42

	int32 x;

    43

	int32 y;

    44

	byte z;

    45

} TileSpriteToDraw;

    46

    47

typedef struct ChildScreenSpriteToDraw {

    48

	uint32 image;

    49

	int32 x;

    50

	int32 y;

    51

	struct ChildScreenSpriteToDraw *next;

    52

} ChildScreenSpriteToDraw;

    53

    54

typedef struct ParentSpriteToDraw {

    55

	uint32 image;

    56

	int32 left;

    57

	int32 top;

    58

	int32 right;

    59

	int32 bottom;

    60

	int32 tile_x;

    61

	int32 tile_y;

    62

	int32 tile_right;

    63

	int32 tile_bottom;

    64

	ChildScreenSpriteToDraw *child;

    65

	byte unk16;

    66

	byte tile_z;

    67

	byte tile_z_bottom;

    68

} ParentSpriteToDraw;

    69

    70

// Quick hack to know how much memory to reserve when allocating from the spritelist

    71

// to prevent a buffer overflow.

    72

#define LARGEST_SPRITELIST_STRUCT ParentSpriteToDraw

    73

    74

typedef struct ViewportDrawer {

    75

	DrawPixelInfo dpi;

    76

    77

	byte *spritelist_mem;

    78

	const byte *eof_spritelist_mem;

    79

    80

	StringSpriteToDraw **last_string, *first_string;

    81

	TileSpriteToDraw **last_tile, *first_tile;

    82

    83

	ChildScreenSpriteToDraw **last_child;

    84

    85

	ParentSpriteToDraw **parent_list;

    86

	ParentSpriteToDraw * const *eof_parent_list;

    87

    88

	byte combine_sprites;

    89

    90

	int offs_x, offs_y; // used when drawing ground sprites relative

    91

} ViewportDrawer;

    92

    93

static ViewportDrawer *_cur_vd;

    94

    95

TileHighlightData _thd;

    96

static TileInfo *_cur_ti;

    97

    98

    99

static Point MapXYZToViewport(const ViewPort *vp, uint x, uint y, uint z)

   100

{

   101

	Point p = RemapCoords(x, y, z);

   102

	p.x -= vp->virtual_width / 2;

   103

	p.y -= vp->virtual_height / 2;

   104

	return p;

   105

}

   106

   107

void AssignWindowViewport(Window *w, int x, int y,

   108

	int width, int height, uint32 follow_flags, byte zoom)

   109

{

   110

	ViewPort *vp;

   111

	Point pt;

   112

	uint32 bit;

   113

   114

	for (vp = _viewports, bit = 1; ; vp++, bit <<= 1) {

   115

		assert(vp != endof(_viewports));

   116

		if (vp->width == 0) break;

   117

	}

   118

	_active_viewports |= bit;

   119

   120

	vp->left = x + w->left;

   121

	vp->top = y + w->top;

   122

	vp->width = width;

   123

	vp->height = height;

   124

   125

	vp->zoom = zoom;

   126

   127

	vp->virtual_width = width << zoom;

   128

	vp->virtual_height = height << zoom;

   129

   130

	if (follow_flags & 0x80000000) {

   131

		const Vehicle *veh;

   132

   133

		WP(w, vp_d).follow_vehicle = (VehicleID)(follow_flags & 0xFFFF);

   134

		veh = GetVehicle(WP(w, vp_d).follow_vehicle);

   135

		pt = MapXYZToViewport(vp, veh->x_pos, veh->y_pos, veh->z_pos);

   136

	} else {

   137

		uint x = TileX(follow_flags) * TILE_SIZE;

   138

		uint y = TileY(follow_flags) * TILE_SIZE;

   139

   140

		WP(w, vp_d).follow_vehicle = INVALID_VEHICLE;

   141

		pt = MapXYZToViewport(vp, x, y, GetSlopeZ(x, y));

   142

	}

   143

   144

	WP(w, vp_d).scrollpos_x = pt.x;

   145

	WP(w, vp_d).scrollpos_y = pt.y;

   146

	w->viewport = vp;

   147

	vp->virtual_left = 0;//pt.x;

   148

	vp->virtual_top = 0;//pt.y;

   149

}

   150

   151

static Point _vp_move_offs;

   152

   153

static void DoSetViewportPosition(Window *w, int left, int top, int width, int height)

   154

{

   155

	for (; w < _last_window; w++) {

   156

		if (left + width > w->left &&

   157

				w->left + w->width > left &&

   158

				top + height > w->top &&

   159

				w->top + w->height > top) {

   160

   161

			if (left < w->left) {

   162

				DoSetViewportPosition(w, left, top, w->left - left, height);

   163

				DoSetViewportPosition(w, left + (w->left - left), top, width - (w->left - left), height);

   164

				return;

   165

			}

   166

   167

			if (left + width > w->left + w->width) {

   168

				DoSetViewportPosition(w, left, top, (w->left + w->width - left), height);

   169

				DoSetViewportPosition(w, left + (w->left + w->width - left), top, width - (w->left + w->width - left) , height);

   170

				return;

   171

			}

   172

   173

			if (top < w->top) {

   174

				DoSetViewportPosition(w, left, top, width, (w->top - top));

   175

				DoSetViewportPosition(w, left, top + (w->top - top), width, height - (w->top - top));

   176

				return;

   177

			}

   178

   179

			if (top + height > w->top + w->height) {

   180

				DoSetViewportPosition(w, left, top, width, (w->top + w->height - top));

   181

				DoSetViewportPosition(w, left, top + (w->top + w->height - top), width , height - (w->top + w->height - top));

   182

				return;

   183

			}

   184

   185

			return;

   186

		}

   187

	}

   188

   189

	{

   190

		int xo = _vp_move_offs.x;

   191

		int yo = _vp_move_offs.y;

   192

   193

		if (abs(xo) >= width || abs(yo) >= height) {

   194

			/* fully_outside */

   195

			RedrawScreenRect(left, top, left + width, top + height);

   196

			return;

   197

		}

   198

   199

		GfxScroll(left, top, width, height, xo, yo);

   200

   201

		if (xo > 0) {

   202

			RedrawScreenRect(left, top, xo + left, top + height);

   203

			left += xo;

   204

			width -= xo;

   205

		} else if (xo < 0) {

   206

			RedrawScreenRect(left+width+xo, top, left+width, top + height);

   207

			width += xo;

   208

		}

   209

   210

		if (yo > 0) {

   211

			RedrawScreenRect(left, top, width+left, top + yo);

   212

		} else if (yo < 0) {

   213

			RedrawScreenRect(left, top + height + yo, width+left, top + height);

   214

		}

   215

	}

   216

}

   217

   218

static void SetViewportPosition(Window *w, int x, int y)

   219

{

   220

	ViewPort *vp = w->viewport;

   221

	int old_left = vp->virtual_left;

   222

	int old_top = vp->virtual_top;

   223

	int i;

   224

	int left, top, width, height;

   225

   226

	vp->virtual_left = x;

   227

	vp->virtual_top = y;

   228

   229

	old_left >>= vp->zoom;

   230

	old_top >>= vp->zoom;

   231

	x >>= vp->zoom;

   232

	y >>= vp->zoom;

   233

   234

	old_left -= x;

   235

	old_top -= y;

   236

   237

	if (old_top == 0 && old_left == 0) return;

   238

   239

	_vp_move_offs.x = old_left;

   240

	_vp_move_offs.y = old_top;

   241

   242

	left = vp->left;

   243

	top = vp->top;

   244

	width = vp->width;

   245

	height = vp->height;

   246

   247

	if (left < 0) {

   248

		width += left;

   249

		left = 0;

   250

	}

   251

   252

	i = left + width - _screen.width;

   253

	if (i >= 0) width -= i;

   254

   255

	if (width > 0) {

   256

		if (top < 0) {

   257

			height += top;

   258

			top = 0;

   259

		}

   260

   261

		i = top + height - _screen.height;

   262

		if (i >= 0) height -= i;

   263

   264

		if (height > 0) DoSetViewportPosition(w + 1, left, top, width, height);

   265

	}

   266

}

   267

   268

   269

ViewPort *IsPtInWindowViewport(const Window *w, int x, int y)

   270

{

   271

	ViewPort *vp = w->viewport;

   272

   273

	if (vp != NULL &&

   274

	    IS_INT_INSIDE(x, vp->left, vp->left + vp->width) &&

   275

			IS_INT_INSIDE(y, vp->top, vp->top + vp->height))

   276

		return vp;

   277

   278

	return NULL;

   279

}

   280

   281

static Point TranslateXYToTileCoord(const ViewPort *vp, int x, int y)

   282

{

   283

	int z;

   284

	Point pt;

   285

	int a,b;

   286

   287

	if ( (uint)(x -= vp->left) >= (uint)vp->width ||

   288

				(uint)(y -= vp->top) >= (uint)vp->height) {

   289

				Point pt = {-1, -1};

   290

				return pt;

   291

	}

   292

   293

	x = ((x << vp->zoom) + vp->virtual_left) >> 2;

   294

	y = ((y << vp->zoom) + vp->virtual_top) >> 1;

   295

   296

#if !defined(NEW_ROTATION)

   297

	a = y-x;

   298

	b = y+x;

   299

#else

   300

	a = x+y;

   301

	b = x-y;

   302

#endif

   303

	z = GetSlopeZ(a, b) >> 1;

   304

	z = GetSlopeZ(a+z, b+z) >> 1;

   305

	z = GetSlopeZ(a+z, b+z) >> 1;

   306

	z = GetSlopeZ(a+z, b+z) >> 1;

   307

	z = GetSlopeZ(a+z, b+z) >> 1;

   308

   309

	pt.x = a+z;

   310

	pt.y = b+z;

   311

   312

	if ((uint)pt.x >= MapMaxX() * TILE_SIZE || (uint)pt.y >= MapMaxY() * TILE_SIZE) {

   313

		pt.x = pt.y = -1;

   314

	}

   315

   316

	return pt;

   317

}

   318

   319

/* When used for zooming, check area below current coordinates (x,y)

   320

 * and return the tile of the zoomed out/in position (zoom_x, zoom_y)

   321

 * when you just want the tile, make x = zoom_x and y = zoom_y */

   322

static Point GetTileFromScreenXY(int x, int y, int zoom_x, int zoom_y)

   323

{

   324

	Window *w;

   325

	ViewPort *vp;

   326

	Point pt;

   327

   328

	if ( (w = FindWindowFromPt(x, y)) != NULL &&

   329

			 (vp = IsPtInWindowViewport(w, x, y)) != NULL)

   330

				return TranslateXYToTileCoord(vp, zoom_x, zoom_y);

   331

   332

	pt.y = pt.x = -1;

   333

	return pt;

   334

}

   335

   336

Point GetTileBelowCursor(void)

   337

{

   338

	return GetTileFromScreenXY(_cursor.pos.x, _cursor.pos.y, _cursor.pos.x, _cursor.pos.y);

   339

}

   340

   341

   342

Point GetTileZoomCenterWindow(bool in, Window * w)

   343

{

   344

	int x, y;

   345

	ViewPort * vp;

   346

   347

	vp = w->viewport;

   348

   349

	if (in) {

   350

		x = ((_cursor.pos.x - vp->left) >> 1) + (vp->width >> 2);

   351

		y = ((_cursor.pos.y - vp->top) >> 1) + (vp->height >> 2);

   352

	} else {

   353

		x = vp->width - (_cursor.pos.x - vp->left);

   354

		y = vp->height - (_cursor.pos.y - vp->top);

   355

	}

   356

	/* Get the tile below the cursor and center on the zoomed-out center */

   357

	return GetTileFromScreenXY(_cursor.pos.x, _cursor.pos.y, x + vp->left, y + vp->top);

   358

}

   359

   360

void DrawGroundSpriteAt(uint32 image, int32 x, int32 y, byte z)

   361

{

   362

	ViewportDrawer *vd = _cur_vd;

   363

	TileSpriteToDraw *ts;

   364

   365

	assert((image & SPRITE_MASK) < MAX_SPRITES);

   366

   367

	if (vd->spritelist_mem >= vd->eof_spritelist_mem) {

   368

		DEBUG(misc, 0) ("Out of sprite mem");

   369

		return;

   370

	}

   371

	ts = (TileSpriteToDraw*)vd->spritelist_mem;

   372

   373

	vd->spritelist_mem += sizeof(TileSpriteToDraw);

   374

   375

	ts->image = image;

   376

	ts->next = NULL;

   377

	ts->x = x;

   378

	ts->y = y;

   379

	ts->z = z;

   380

	*vd->last_tile = ts;

   381

	vd->last_tile = &ts->next;

   382

}

   383

   384

void DrawGroundSprite(uint32 image)

   385

{

   386

	if (_offset_ground_sprites) {

   387

		// offset ground sprite because of foundation?

   388

		AddChildSpriteScreen(image, _cur_vd->offs_x, _cur_vd->offs_y);

   389

	} else {

   390

		_added_tile_sprite = true;

   391

		DrawGroundSpriteAt(image, _cur_ti->x, _cur_ti->y, _cur_ti->z);

   392

	}

   393

}

   394

   395

   396

void OffsetGroundSprite(int x, int y)

   397

{

   398

	_cur_vd->offs_x = x;

   399

	_cur_vd->offs_y = y;

   400

	_offset_ground_sprites = true;

   401

}

   402

   403

static void AddCombinedSprite(uint32 image, int x, int y, byte z)

   404

{

   405

	const ViewportDrawer *vd = _cur_vd;

   406

	Point pt = RemapCoords(x, y, z);

   407

	const Sprite* spr = GetSprite(image & SPRITE_MASK);

   408

   409

	if (pt.x + spr->x_offs >= vd->dpi.left + vd->dpi.width ||

   410

			pt.x + spr->x_offs + spr->width <= vd->dpi.left ||

   411

			pt.y + spr->y_offs >= vd->dpi.top + vd->dpi.height ||

   412

			pt.y + spr->y_offs + spr->height <= vd->dpi.top)

   413

		return;

   414

   415

	AddChildSpriteScreen(image, pt.x - vd->parent_list[-1]->left, pt.y - vd->parent_list[-1]->top);

   416

}

   417

   418

   419

void AddSortableSpriteToDraw(uint32 image, int x, int y, int w, int h, byte dz, byte z)

   420

{

   421

	ViewportDrawer *vd = _cur_vd;

   422

	ParentSpriteToDraw *ps;

   423

	const Sprite *spr;

   424

	Point pt;

   425

   426

	assert((image & SPRITE_MASK) < MAX_SPRITES);

   427

   428

	if (vd->combine_sprites == 2) {

   429

		AddCombinedSprite(image, x, y, z);

   430

		return;

   431

	}

   432

   433

	vd->last_child = NULL;

   434

   435

	if (vd->spritelist_mem >= vd->eof_spritelist_mem) {

   436

		DEBUG(misc, 0) ("Out of sprite mem");

   437

		return;

   438

	}

   439

	ps = (ParentSpriteToDraw*)vd->spritelist_mem;

   440

   441

	if (vd->parent_list >= vd->eof_parent_list) {

   442

		// This can happen rarely, mostly when you zoom out completely

   443

		//  and have a lot of stuff that moves (and is added to the

   444

		//  sort-list, this function). To solve it, increase

   445

		//  parent_list somewhere below to a higher number.

   446

		// This can not really hurt you, it just gives some black

   447

		//  spots on the screen ;)

   448

		DEBUG(misc, 0) ("Out of sprite mem (parent_list)");

   449

		return;

   450

	}

   451

   452

	vd->spritelist_mem += sizeof(ParentSpriteToDraw);

   453

   454

	ps->image = image;

   455

	ps->tile_x = x;

   456

	ps->tile_right = x + w - 1;

   457

   458

	ps->tile_y = y;

   459

	ps->tile_bottom = y + h - 1;

   460

   461

	ps->tile_z = z;

   462

	ps->tile_z_bottom = z + dz - 1;

   463

   464

	pt = RemapCoords(x, y, z);

   465

   466

	spr = GetSprite(image & SPRITE_MASK);

   467

	if ((ps->left   = (pt.x += spr->x_offs)) >= vd->dpi.left + vd->dpi.width ||

   468

			(ps->right  = (pt.x +  spr->width )) <= vd->dpi.left ||

   469

			(ps->top    = (pt.y += spr->y_offs)) >= vd->dpi.top + vd->dpi.height ||

   470

			(ps->bottom = (pt.y +  spr->height)) <= vd->dpi.top) {

   471

		return;

   472

	}

   473

   474

	ps->unk16 = 0;

   475

	ps->child = NULL;

   476

	vd->last_child = &ps->child;

   477

   478

	*vd->parent_list++ = ps;

   479

   480

	if (vd->combine_sprites == 1) vd->combine_sprites = 2;

   481

}

   482

   483

void StartSpriteCombine(void)

   484

{

   485

	_cur_vd->combine_sprites = 1;

   486

}

   487

   488

void EndSpriteCombine(void)

   489

{

   490

	_cur_vd->combine_sprites = 0;

   491

}

   492

   493

void AddChildSpriteScreen(uint32 image, int x, int y)

   494

{

   495

	ViewportDrawer *vd = _cur_vd;

   496

	ChildScreenSpriteToDraw *cs;

   497

   498

	assert((image & SPRITE_MASK) < MAX_SPRITES);

   499

   500

	if (vd->spritelist_mem >= vd->eof_spritelist_mem) {

   501

		DEBUG(misc, 0) ("Out of sprite mem");

   502

		return;

   503

	}

   504

	cs = (ChildScreenSpriteToDraw*)vd->spritelist_mem;

   505

   506

	if (vd->last_child == NULL) return;

   507

   508

	vd->spritelist_mem += sizeof(ChildScreenSpriteToDraw);

   509