1

/* $Id$ */

     2

     3

#include "../stdafx.h"

     4

#include "../openttd.h"

     5

#include "../gfx_func.h"

     6

#include "../variables.h"

     7

#include "../win32.h"

     8

#include "../blitter/factory.hpp"

     9

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

    10

#include "../core/math_func.hpp"

    11

#include "../core/random_func.hpp"

    12

#include "win32_v.h"

    13

#include <windows.h>

    14

#include <tchar.h>

    15

    16

static struct {

    17

	HWND main_wnd;

    18

	HBITMAP dib_sect;

    19

	void *buffer_bits;

    20

	HPALETTE gdi_palette;

    21

	int width;

    22

	int height;

    23

	int width_org;

    24

	int height_org;

    25

	bool fullscreen;

    26

	bool has_focus;

    27

	bool running;

    28

} _wnd;

    29

    30

bool _force_full_redraw;

    31

bool _window_maximize;

    32

uint _display_hz;

    33

uint _fullscreen_bpp;

    34

static uint16 _bck_resolution[2];

    35

#if !defined(UNICODE)

    36

uint _codepage;

    37

#endif

    38

    39

static void MakePalette()

    40

{

    41

	LOGPALETTE *pal;

    42

	uint i;

    43

    44

	pal = (LOGPALETTE*)alloca(sizeof(LOGPALETTE) + (256 - 1) * sizeof(PALETTEENTRY));

    45

    46

	pal->palVersion = 0x300;

    47

	pal->palNumEntries = 256;

    48

    49

	for (i = 0; i != 256; i++) {

    50

		pal->palPalEntry[i].peRed   = _cur_palette[i].r;

    51

		pal->palPalEntry[i].peGreen = _cur_palette[i].g;

    52

		pal->palPalEntry[i].peBlue  = _cur_palette[i].b;

    53

		pal->palPalEntry[i].peFlags = 0;

    54

    55

	}

    56

	_wnd.gdi_palette = CreatePalette(pal);

    57

	if (_wnd.gdi_palette == NULL) error("CreatePalette failed!\n");

    58

}

    59

    60

static void UpdatePalette(HDC dc, uint start, uint count)

    61

{

    62

	RGBQUAD rgb[256];

    63

	uint i;

    64

    65

	for (i = 0; i != count; i++) {

    66

		rgb[i].rgbRed   = _cur_palette[start + i].r;

    67

		rgb[i].rgbGreen = _cur_palette[start + i].g;

    68

		rgb[i].rgbBlue  = _cur_palette[start + i].b;

    69

		rgb[i].rgbReserved = 0;

    70

	}

    71

    72

	SetDIBColorTable(dc, start, count, rgb);

    73

}

    74

    75

struct VkMapping {

    76

	byte vk_from;

    77

	byte vk_count;

    78

	byte map_to;

    79

};

    80

    81

#define AS(x, z) {x, 0, z}

    82

#define AM(x, y, z, w) {x, y - x, z}

    83

    84

static const VkMapping _vk_mapping[] = {

    85

	/* Pageup stuff + up/down */

    86

	AM(VK_PRIOR,VK_DOWN, WKC_PAGEUP, WKC_DOWN),

    87

	/* Map letters & digits */

    88

	AM('A','Z','A','Z'),

    89

	AM('0','9','0','9'),

    90

    91

	AS(VK_ESCAPE,   WKC_ESC),

    92

	AS(VK_PAUSE,    WKC_PAUSE),

    93

	AS(VK_BACK,     WKC_BACKSPACE),

    94

	AM(VK_INSERT,   VK_DELETE, WKC_INSERT, WKC_DELETE),

    95

    96

	AS(VK_SPACE,    WKC_SPACE),

    97

	AS(VK_RETURN,   WKC_RETURN),

    98

	AS(VK_TAB,      WKC_TAB),

    99

   100

	/* Function keys */

   101

	AM(VK_F1, VK_F12, WKC_F1, WKC_F12),

   102

   103

	/* Numeric part */

   104

	AM(VK_NUMPAD0, VK_NUMPAD9, WKC_NUM_0, WKC_NUM_9),

   105

	AS(VK_DIVIDE,   WKC_NUM_DIV),

   106

	AS(VK_MULTIPLY, WKC_NUM_MUL),

   107

	AS(VK_SUBTRACT, WKC_NUM_MINUS),

   108

	AS(VK_ADD,      WKC_NUM_PLUS),

   109

	AS(VK_DECIMAL,  WKC_NUM_DECIMAL),

   110

   111

	/* Other non-letter keys */

   112

	AS(0xBF,  WKC_SLASH),

   113

	AS(0xBA,  WKC_SEMICOLON),

   114

	AS(0xBB,  WKC_EQUALS),

   115

	AS(0xDB,  WKC_L_BRACKET),

   116

	AS(0xDC,  WKC_BACKSLASH),

   117

	AS(0xDD,  WKC_R_BRACKET),

   118

   119

	AS(0xDE,  WKC_SINGLEQUOTE),

   120

	AS(0xBC,  WKC_COMMA),

   121

	AS(0xBD,  WKC_MINUS),

   122

	AS(0xBE,  WKC_PERIOD)

   123

};

   124

   125

static uint MapWindowsKey(uint sym)

   126

{

   127

	const VkMapping *map;

   128

	uint key = 0;

   129

   130

	for (map = _vk_mapping; map != endof(_vk_mapping); ++map) {

   131

		if ((uint)(sym - map->vk_from) <= map->vk_count) {

   132

			key = sym - map->vk_from + map->map_to;

   133

			break;

   134

		}

   135

	}

   136

   137

	if (GetAsyncKeyState(VK_SHIFT)   < 0) key |= WKC_SHIFT;

   138

	if (GetAsyncKeyState(VK_CONTROL) < 0) key |= WKC_CTRL;

   139

	if (GetAsyncKeyState(VK_MENU)    < 0) key |= WKC_ALT;

   140

	return key;

   141

}

   142

   143

static bool AllocateDibSection(int w, int h);

   144

   145

static void ClientSizeChanged(int w, int h)

   146

{

   147

	// allocate new dib section of the new size

   148

	if (AllocateDibSection(w, h)) {

   149

		// mark all palette colors dirty

   150

		_pal_first_dirty = 0;

   151

		_pal_count_dirty = 256;

   152

		GameSizeChanged();

   153

   154

		// redraw screen

   155

		if (_wnd.running) {

   156

			_screen.dst_ptr = _wnd.buffer_bits;

   157

			UpdateWindows();

   158

		}

   159

	}

   160

}

   161

   162

#ifdef _DEBUG

   163

// Keep this function here..

   164

// It allows you to redraw the screen from within the MSVC debugger

   165

int RedrawScreenDebug()

   166

{

   167

	HDC dc,dc2;

   168

	static int _fooctr;

   169

	HBITMAP old_bmp;

   170

	HPALETTE old_palette;

   171

   172

	_screen.dst_ptr = _wnd.buffer_bits;

   173

	UpdateWindows();

   174

   175

	dc = GetDC(_wnd.main_wnd);

   176

	dc2 = CreateCompatibleDC(dc);

   177

   178

	old_bmp = (HBITMAP)SelectObject(dc2, _wnd.dib_sect);

   179

	old_palette = SelectPalette(dc, _wnd.gdi_palette, FALSE);

   180

	BitBlt(dc, 0, 0, _wnd.width, _wnd.height, dc2, 0, 0, SRCCOPY);

   181

	SelectPalette(dc, old_palette, TRUE);

   182

	SelectObject(dc2, old_bmp);

   183

	DeleteDC(dc2);

   184

	ReleaseDC(_wnd.main_wnd, dc);

   185

   186

	return _fooctr++;

   187

}

   188

#endif

   189

   190

/* Windows 95 will not have a WM_MOUSELEAVE message, so define it if needed */

   191

#if !defined(WM_MOUSELEAVE)

   192

#define WM_MOUSELEAVE 0x02A3

   193

#endif

   194

#define TID_POLLMOUSE 1

   195

#define MOUSE_POLL_DELAY 75

   196

   197

static void CALLBACK TrackMouseTimerProc(HWND hwnd, UINT msg, UINT event, DWORD time)

   198

{

   199

	RECT rc;

   200

	POINT pt;

   201

   202

	/* Get the rectangle of our window and translate it to screen coordinates.

   203

	 * Compare this with the current screen coordinates of the mouse and if it

   204

	 * falls outside of the area or our window we have left the window. */

   205

	GetClientRect(hwnd, &rc);

   206

	MapWindowPoints(hwnd, HWND_DESKTOP, (LPPOINT)(LPRECT)&rc, 2);

   207

	GetCursorPos(&pt);

   208

   209

	if (!PtInRect(&rc, pt) || (WindowFromPoint(pt) != hwnd)) {

   210

		KillTimer(hwnd, event);

   211

		PostMessage(hwnd, WM_MOUSELEAVE, 0, 0L);

   212

	}

   213

}

   214

   215

static bool MakeWindow(bool full_screen)

   216

{

   217

	_fullscreen = full_screen;

   218

   219

	// recreate window?

   220

	if ((full_screen || _wnd.fullscreen) && _wnd.main_wnd) {

   221

		DestroyWindow(_wnd.main_wnd);

   222

		_wnd.main_wnd = 0;

   223

	}

   224

   225

#if defined(WINCE)

   226

	/* WinCE is always fullscreen */

   227

#else

   228

	if (full_screen) {

   229

		DEVMODE settings;

   230

   231

		/* Make sure we are always at least the screen-depth of the blitter */

   232

		if (_fullscreen_bpp < BlitterFactoryBase::GetCurrentBlitter()->GetScreenDepth()) _fullscreen_bpp = BlitterFactoryBase::GetCurrentBlitter()->GetScreenDepth();

   233

   234

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

   235

		settings.dmSize = sizeof(settings);

   236

		settings.dmFields =

   237

			(_fullscreen_bpp != 0 ? DM_BITSPERPEL : 0) |

   238

			DM_PELSWIDTH |

   239

			DM_PELSHEIGHT |

   240

			(_display_hz != 0 ? DM_DISPLAYFREQUENCY : 0);

   241

		settings.dmBitsPerPel = _fullscreen_bpp;

   242

		settings.dmPelsWidth  = _wnd.width_org;

   243

		settings.dmPelsHeight = _wnd.height_org;

   244

		settings.dmDisplayFrequency = _display_hz;

   245

   246

		if (ChangeDisplaySettings(&settings, CDS_FULLSCREEN) != DISP_CHANGE_SUCCESSFUL) {

   247

			MakeWindow(false);  // don't care about the result

   248

			return false;  // the request failed

   249

		}

   250

	} else if (_wnd.fullscreen) {

   251

		// restore display?

   252

		ChangeDisplaySettings(NULL, 0);

   253

	}

   254

#endif

   255

   256

	{

   257

		RECT r;

   258

		DWORD style, showstyle;

   259

		int x, y, w, h;

   260

   261

		showstyle = SW_SHOWNORMAL;

   262

		_wnd.fullscreen = full_screen;

   263

		if (_wnd.fullscreen) {

   264

			style = WS_POPUP;

   265

			SetRect(&r, 0, 0, _wnd.width_org, _wnd.height_org);

   266

		} else {

   267

			style = WS_OVERLAPPEDWINDOW;

   268

			/* On window creation, check if we were in maximize mode before */

   269

			if (_window_maximize) showstyle = SW_SHOWMAXIMIZED;

   270

			SetRect(&r, 0, 0, _wnd.width, _wnd.height);

   271

		}

   272

   273

#if !defined(WINCE)

   274

		AdjustWindowRect(&r, style, FALSE);

   275

#endif

   276

		w = r.right - r.left;

   277

		h = r.bottom - r.top;

   278

		x = (GetSystemMetrics(SM_CXSCREEN) - w) / 2;

   279

		y = (GetSystemMetrics(SM_CYSCREEN) - h) / 2;

   280

   281

		if (_wnd.main_wnd) {

   282

			ShowWindow(_wnd.main_wnd, SW_SHOWNORMAL); // remove maximize-flag

   283

			SetWindowPos(_wnd.main_wnd, 0, x, y, w, h, SWP_NOACTIVATE | SWP_NOOWNERZORDER | SWP_NOZORDER);

   284

		} else {

   285

			extern const char _openttd_revision[];

   286

			TCHAR Windowtitle[50];

   287

   288

			_sntprintf(Windowtitle, sizeof(Windowtitle), _T("OpenTTD %s"), MB_TO_WIDE(_openttd_revision));

   289

   290

			_wnd.main_wnd = CreateWindow(_T("OTTD"), Windowtitle, style, x, y, w, h, 0, 0, GetModuleHandle(NULL), 0);

   291

			if (_wnd.main_wnd == NULL) error("CreateWindow failed");

   292

			ShowWindow(_wnd.main_wnd, showstyle);

   293

		}

   294

	}

   295

	GameSizeChanged(); // invalidate all windows, force redraw

   296

	return true; // the request succedded

   297

}

   298

   299

static LRESULT CALLBACK WndProcGdi(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)

   300

{

   301

	static uint32 keycode = 0;

   302

	static bool console = false;

   303

   304

	switch (msg) {

   305

		case WM_CREATE:

   306

			SetTimer(hwnd, TID_POLLMOUSE, MOUSE_POLL_DELAY, (TIMERPROC)TrackMouseTimerProc);

   307

			break;

   308

   309

		case WM_PAINT: {

   310

			PAINTSTRUCT ps;

   311

			HDC dc,dc2;

   312

			HBITMAP old_bmp;

   313

			HPALETTE old_palette;

   314

   315

			BeginPaint(hwnd, &ps);

   316

			dc = ps.hdc;

   317

			dc2 = CreateCompatibleDC(dc);

   318

			old_bmp = (HBITMAP)SelectObject(dc2, _wnd.dib_sect);

   319

			old_palette = SelectPalette(dc, _wnd.gdi_palette, FALSE);

   320

   321

			if (_pal_count_dirty != 0) {

   322

				Blitter *blitter = BlitterFactoryBase::GetCurrentBlitter();

   323

   324

				switch (blitter->UsePaletteAnimation()) {

   325

					case Blitter::PALETTE_ANIMATION_VIDEO_BACKEND:

   326

						UpdatePalette(dc2, _pal_first_dirty, _pal_count_dirty);

   327

						break;

   328

   329

					case Blitter::PALETTE_ANIMATION_BLITTER:

   330

						blitter->PaletteAnimate(_pal_first_dirty, _pal_count_dirty);

   331

						break;

   332

   333

					case Blitter::PALETTE_ANIMATION_NONE:

   334

						break;

   335

   336

					default:

   337

						NOT_REACHED();

   338

				}

   339

				_pal_count_dirty = 0;

   340

			}

   341

   342

			BitBlt(dc, 0, 0, _wnd.width, _wnd.height, dc2, 0, 0, SRCCOPY);

   343

			SelectPalette(dc, old_palette, TRUE);

   344

			SelectObject(dc2, old_bmp);

   345

			DeleteDC(dc2);

   346

			EndPaint(hwnd, &ps);

   347

			return 0;

   348

		}

   349

   350

		case WM_PALETTECHANGED:

   351

			if ((HWND)wParam == hwnd) return 0;

   352

			/* FALLTHROUGH */

   353

   354

		case WM_QUERYNEWPALETTE: {

   355

			HDC hDC = GetWindowDC(hwnd);

   356

			HPALETTE hOldPalette = SelectPalette(hDC, _wnd.gdi_palette, FALSE);

   357

			UINT nChanged = RealizePalette(hDC);

   358

   359

			SelectPalette(hDC, hOldPalette, TRUE);

   360

			ReleaseDC(hwnd, hDC);

   361

			if (nChanged) InvalidateRect(hwnd, NULL, FALSE);

   362

			return 0;

   363

		}

   364

   365

		case WM_CLOSE:

   366

			HandleExitGameRequest();

   367

			return 0;

   368

   369

		case WM_DESTROY:

   370

			if (_window_maximize) {

   371

				_cur_resolution[0] = _bck_resolution[0];

   372

				_cur_resolution[1] = _bck_resolution[1];

   373

			}

   374

			return 0;

   375

   376

		case WM_LBUTTONDOWN:

   377

			SetCapture(hwnd);

   378

			_left_button_down = true;

   379

			HandleMouseEvents();

   380

			return 0;

   381

   382

		case WM_LBUTTONUP:

   383

			ReleaseCapture();

   384

			_left_button_down = false;

   385

			_left_button_clicked = false;

   386

			HandleMouseEvents();

   387

			return 0;

   388

   389

		case WM_RBUTTONDOWN:

   390

			SetCapture(hwnd);

   391

			_right_button_down = true;

   392

			_right_button_clicked = true;

   393

			HandleMouseEvents();

   394

			return 0;

   395

   396

		case WM_RBUTTONUP:

   397

			ReleaseCapture();

   398

			_right_button_down = false;

   399

			HandleMouseEvents();

   400

			return 0;

   401

   402

		case WM_MOUSELEAVE:

   403

			UndrawMouseCursor();

   404

			_cursor.in_window = false;

   405

   406

			if (!_left_button_down && !_right_button_down) MyShowCursor(true);

   407

			HandleMouseEvents();

   408

			return 0;

   409

   410

		case WM_MOUSEMOVE: {

   411

			int x = (int16)LOWORD(lParam);

   412

			int y = (int16)HIWORD(lParam);

   413

			POINT pt;

   414

   415

			/* If the mouse was not in the window and it has moved it means it has

   416

			 * come into the window, so start drawing the mouse. Also start

   417

			 * tracking the mouse for exiting the window */

   418

			if (!_cursor.in_window) {

   419

				_cursor.in_window = true;

   420

				SetTimer(hwnd, TID_POLLMOUSE, MOUSE_POLL_DELAY, (TIMERPROC)TrackMouseTimerProc);

   421

   422

				DrawMouseCursor();

   423

			}

   424

   425

			if (_cursor.fix_at) {

   426

				int dx = x - _cursor.pos.x;

   427

				int dy = y - _cursor.pos.y;

   428

				if (dx != 0 || dy != 0) {

   429

					_cursor.delta.x += dx;

   430

					_cursor.delta.y += dy;

   431

   432

					pt.x = _cursor.pos.x;

   433

					pt.y = _cursor.pos.y;

   434

   435

					ClientToScreen(hwnd, &pt);

   436

					SetCursorPos(pt.x, pt.y);

   437

				}

   438

			} else {

   439

				_cursor.delta.x += x - _cursor.pos.x;

   440

				_cursor.delta.y += y - _cursor.pos.y;

   441

				_cursor.pos.x = x;

   442

				_cursor.pos.y = y;

   443

				_cursor.dirty = true;

   444

			}

   445

			MyShowCursor(false);

   446

			HandleMouseEvents();

   447

			return 0;

   448

		}

   449

   450

#if !defined(UNICODE)

   451

		case WM_INPUTLANGCHANGE: {

   452

			TCHAR locale[6];

   453

			LCID lcid = GB(lParam, 0, 16);

   454

   455

			int len = GetLocaleInfo(lcid, LOCALE_IDEFAULTANSICODEPAGE, locale, lengthof(locale));

   456

			if (len != 0) _codepage = _ttoi(locale);

   457

			return 1;

   458

		}

   459

#endif /* UNICODE */

   460

   461

		case WM_DEADCHAR:

   462

			console = GB(lParam, 16, 8) == 41;

   463

			return 0;

   464

   465

		case WM_CHAR: {

   466

			/* Silently drop all non-text messages as those were handled by WM_KEYDOWN */

   467

			if (wParam < VK_SPACE) return 0;

   468

			uint scancode = GB(lParam, 16, 8);

   469

			uint charcode = wParam;

   470

   471

			/* If the console key is a dead-key, we need to press it twice to get a WM_CHAR message.

   472

			 * But we then get two WM_CHAR messages, so ignore the first one */

   473

			if (console && scancode == 41) {

   474

				console = false;

   475

				return 0;

   476

			}

   477

   478

#if !defined(UNICODE)

   479

			wchar_t w;

   480

			int len = MultiByteToWideChar(_codepage, 0, (char*)&charcode, 1, &w, 1);

   481

			charcode = len == 1 ? w : 0;

   482

#endif /* UNICODE */

   483

   484

			/* No matter the keyboard layout, we will map the '~' to the console */

   485

			scancode = scancode == 41 ? (int)WKC_BACKQUOTE : keycode;

   486

			HandleKeypress(GB(charcode, 0, 16) | (scancode << 16));

   487

			return 0;

   488

		}

   489

   490

		case WM_KEYDOWN: {

   491

			keycode = MapWindowsKey(wParam);

   492