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

    12

#include <windows.h>

    13

#include <tchar.h>

    14

    15

static struct {

    16

	HWND main_wnd;

    17

	HBITMAP dib_sect;

    18

	void *buffer_bits;

    19

	HPALETTE gdi_palette;

    20

	int width;

    21

	int height;

    22

	int width_org;

    23

	int height_org;

    24

	bool fullscreen;

    25

	bool has_focus;

    26

	bool running;

    27

} _wnd;

    28

    29

bool _force_full_redraw;

    30

bool _window_maximize;

    31

uint _display_hz;

    32

uint _fullscreen_bpp;

    33

static uint16 _bck_resolution[2];

    34

#if !defined(UNICODE)

    35

uint _codepage;

    36

#endif

    37

    38

static void MakePalette()

    39

{

    40

	LOGPALETTE *pal;

    41

	uint i;

    42

    43

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

    44

    45

	pal->palVersion = 0x300;

    46

	pal->palNumEntries = 256;

    47

    48

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

    49

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

    50

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

    51

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

    52

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

    53

    54

	}

    55

	_wnd.gdi_palette = CreatePalette(pal);

    56

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

    57

}

    58

    59

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

    60

{

    61

	RGBQUAD rgb[256];

    62

	uint i;

    63

    64

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

    65

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

    66

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

    67

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

    68

		rgb[i].rgbReserved = 0;

    69

	}

    70

    71

	SetDIBColorTable(dc, start, count, rgb);

    72

}

    73

    74

struct VkMapping {

    75

	byte vk_from;

    76

	byte vk_count;

    77

	byte map_to;

    78

};

    79

    80

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

    81

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

    82

    83

static const VkMapping _vk_mapping[] = {

    84

	/* Pageup stuff + up/down */

    85

	AM(VK_PRIOR,VK_DOWN, WKC_PAGEUP, WKC_DOWN),

    86

	/* Map letters & digits */

    87

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

    88

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

    89

    90

	AS(VK_ESCAPE,   WKC_ESC),

    91

	AS(VK_PAUSE,    WKC_PAUSE),

    92

	AS(VK_BACK,     WKC_BACKSPACE),

    93

	AM(VK_INSERT,   VK_DELETE, WKC_INSERT, WKC_DELETE),

    94

    95

	AS(VK_SPACE,    WKC_SPACE),

    96

	AS(VK_RETURN,   WKC_RETURN),

    97

	AS(VK_TAB,      WKC_TAB),

    98

    99

	/* Function keys */

   100

	AM(VK_F1, VK_F12, WKC_F1, WKC_F12),

   101

   102

	/* Numeric part */

   103

	AM(VK_NUMPAD0, VK_NUMPAD9, WKC_NUM_0, WKC_NUM_9),

   104

	AS(VK_DIVIDE,   WKC_NUM_DIV),

   105

	AS(VK_MULTIPLY, WKC_NUM_MUL),

   106

	AS(VK_SUBTRACT, WKC_NUM_MINUS),

   107

	AS(VK_ADD,      WKC_NUM_PLUS),

   108

	AS(VK_DECIMAL,  WKC_NUM_DECIMAL),

   109

   110

	/* Other non-letter keys */

   111

	AS(0xBF,  WKC_SLASH),

   112

	AS(0xBA,  WKC_SEMICOLON),

   113

	AS(0xBB,  WKC_EQUALS),

   114

	AS(0xDB,  WKC_L_BRACKET),

   115

	AS(0xDC,  WKC_BACKSLASH),

   116

	AS(0xDD,  WKC_R_BRACKET),

   117

   118

	AS(0xDE,  WKC_SINGLEQUOTE),

   119

	AS(0xBC,  WKC_COMMA),

   120

	AS(0xBD,  WKC_MINUS),

   121

	AS(0xBE,  WKC_PERIOD)

   122

};

   123

   124

static uint MapWindowsKey(uint sym)

   125

{

   126

	const VkMapping *map;

   127

	uint key = 0;

   128

   129

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

   130

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

   131

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

   132

			break;

   133

		}

   134

	}

   135

   136

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

   137

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

   138

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

   139

	return key;

   140

}

   141

   142

static bool AllocateDibSection(int w, int h);

   143

   144

static void ClientSizeChanged(int w, int h)

   145

{

   146

	// allocate new dib section of the new size

   147

	if (AllocateDibSection(w, h)) {

   148

		// mark all palette colors dirty

   149

		_pal_first_dirty = 0;

   150

		_pal_count_dirty = 256;

   151

		GameSizeChanged();

   152

   153

		// redraw screen

   154

		if (_wnd.running) {

   155

			_screen.dst_ptr = _wnd.buffer_bits;

   156

			UpdateWindows();

   157

		}

   158

	}

   159

}

   160

   161

#ifdef _DEBUG

   162

// Keep this function here..

   163

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

   164

int RedrawScreenDebug()

   165

{

   166

	HDC dc,dc2;

   167

	static int _fooctr;

   168

	HBITMAP old_bmp;

   169

	HPALETTE old_palette;

   170

   171

	_screen.dst_ptr = _wnd.buffer_bits;

   172

	UpdateWindows();

   173

   174

	dc = GetDC(_wnd.main_wnd);

   175

	dc2 = CreateCompatibleDC(dc);

   176

   177

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

   178

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

   179

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

   180

	SelectPalette(dc, old_palette, TRUE);

   181

	SelectObject(dc2, old_bmp);

   182

	DeleteDC(dc2);

   183

	ReleaseDC(_wnd.main_wnd, dc);

   184

   185

	return _fooctr++;

   186

}

   187

#endif

   188

   189

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

   190

#if !defined(WM_MOUSELEAVE)

   191

#define WM_MOUSELEAVE 0x02A3

   192

#endif

   193

#define TID_POLLMOUSE 1

   194

#define MOUSE_POLL_DELAY 75

   195

   196

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

   197

{

   198

	RECT rc;

   199

	POINT pt;

   200

   201

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

   202

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

   203

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

   204

	GetClientRect(hwnd, &rc);

   205

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

   206

	GetCursorPos(&pt);

   207

   208

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

   209

		KillTimer(hwnd, event);

   210

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

   211

	}

   212

}

   213

   214

static bool MakeWindow(bool full_screen)

   215

{

   216

	_fullscreen = full_screen;

   217

   218

	// recreate window?

   219

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

   220

		DestroyWindow(_wnd.main_wnd);

   221

		_wnd.main_wnd = 0;

   222

	}

   223

   224

#if defined(WINCE)

   225

	/* WinCE is always fullscreen */

   226

#else

   227

	if (full_screen) {

   228

		DEVMODE settings;

   229

   230

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

   231

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

   232

   233

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

   234

		settings.dmSize = sizeof(settings);

   235

		settings.dmFields =

   236

			(_fullscreen_bpp != 0 ? DM_BITSPERPEL : 0) |

   237

			DM_PELSWIDTH |

   238

			DM_PELSHEIGHT |

   239

			(_display_hz != 0 ? DM_DISPLAYFREQUENCY : 0);

   240

		settings.dmBitsPerPel = _fullscreen_bpp;

   241

		settings.dmPelsWidth  = _wnd.width_org;

   242

		settings.dmPelsHeight = _wnd.height_org;

   243

		settings.dmDisplayFrequency = _display_hz;

   244

   245

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

   246

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

   247

			return false;  // the request failed

   248

		}

   249

	} else if (_wnd.fullscreen) {

   250

		// restore display?

   251

		ChangeDisplaySettings(NULL, 0);

   252

	}

   253

#endif

   254

   255

	{

   256

		RECT r;

   257

		DWORD style, showstyle;

   258

		int x, y, w, h;

   259

   260

		showstyle = SW_SHOWNORMAL;

   261

		_wnd.fullscreen = full_screen;

   262

		if (_wnd.fullscreen) {

   263

			style = WS_POPUP;

   264

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

   265

		} else {

   266

			style = WS_OVERLAPPEDWINDOW;

   267

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

   268

			if (_window_maximize) showstyle = SW_SHOWMAXIMIZED;

   269

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

   270

		}

   271

   272

#if !defined(WINCE)

   273

		AdjustWindowRect(&r, style, FALSE);

   274

#endif

   275

		w = r.right - r.left;

   276

		h = r.bottom - r.top;

   277

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

   278

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

   279

   280

		if (_wnd.main_wnd) {

   281

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

   282

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

   283

		} else {

   284

			extern const char _openttd_revision[];

   285

			TCHAR Windowtitle[50];

   286

   287

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

   288

   289

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

   290

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

   291

			ShowWindow(_wnd.main_wnd, showstyle);

   292

		}

   293

	}

   294

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

   295

	return true; // the request succedded

   296

}

   297

   298

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

   299

{

   300

	static uint32 keycode = 0;

   301

	static bool console = false;

   302

   303

	switch (msg) {

   304

		case WM_CREATE:

   305

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

   306

			break;

   307

   308

		case WM_PAINT: {

   309

			PAINTSTRUCT ps;

   310

			HDC dc,dc2;

   311

			HBITMAP old_bmp;

   312

			HPALETTE old_palette;

   313

   314

			BeginPaint(hwnd, &ps);

   315

			dc = ps.hdc;

   316

			dc2 = CreateCompatibleDC(dc);

   317

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

   318

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

   319

   320

			if (_pal_count_dirty != 0) {

   321

				Blitter *blitter = BlitterFactoryBase::GetCurrentBlitter();

   322

   323

				switch (blitter->UsePaletteAnimation()) {

   324

					case Blitter::PALETTE_ANIMATION_VIDEO_BACKEND:

   325

						UpdatePalette(dc2, _pal_first_dirty, _pal_count_dirty);

   326

						break;

   327

   328

					case Blitter::PALETTE_ANIMATION_BLITTER:

   329

						blitter->PaletteAnimate(_pal_first_dirty, _pal_count_dirty);

   330

						break;

   331

   332

					case Blitter::PALETTE_ANIMATION_NONE:

   333

						break;

   334

   335

					default:

   336

						NOT_REACHED();

   337

				}

   338

				_pal_count_dirty = 0;

   339

			}

   340

   341

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

   342

			SelectPalette(dc, old_palette, TRUE);

   343

			SelectObject(dc2, old_bmp);

   344

			DeleteDC(dc2);

   345

			EndPaint(hwnd, &ps);

   346

			return 0;

   347

		}

   348

   349

		case WM_PALETTECHANGED:

   350

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

   351

			/* FALLTHROUGH */

   352

   353

		case WM_QUERYNEWPALETTE: {

   354

			HDC hDC = GetWindowDC(hwnd);

   355

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

   356

			UINT nChanged = RealizePalette(hDC);

   357

   358

			SelectPalette(hDC, hOldPalette, TRUE);

   359

			ReleaseDC(hwnd, hDC);

   360

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

   361

			return 0;

   362

		}

   363

   364

		case WM_CLOSE:

   365

			HandleExitGameRequest();

   366

			return 0;

   367

   368

		case WM_DESTROY:

   369

			if (_window_maximize) {

   370

				_cur_resolution[0] = _bck_resolution[0];

   371

				_cur_resolution[1] = _bck_resolution[1];

   372

			}

   373

			return 0;

   374

   375

		case WM_LBUTTONDOWN:

   376

			SetCapture(hwnd);

   377

			_left_button_down = true;

   378

			HandleMouseEvents();

   379

			return 0;

   380

   381

		case WM_LBUTTONUP:

   382

			ReleaseCapture();

   383

			_left_button_down = false;

   384

			_left_button_clicked = false;

   385

			HandleMouseEvents();

   386

			return 0;

   387

   388

		case WM_RBUTTONDOWN:

   389

			SetCapture(hwnd);

   390

			_right_button_down = true;

   391

			_right_button_clicked = true;

   392

			HandleMouseEvents();

   393

			return 0;

   394

   395

		case WM_RBUTTONUP:

   396

			ReleaseCapture();

   397

			_right_button_down = false;

   398

			HandleMouseEvents();

   399

			return 0;

   400

   401

		case WM_MOUSELEAVE:

   402

			UndrawMouseCursor();

   403

			_cursor.in_window = false;

   404

   405

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

   406

			HandleMouseEvents();

   407

			return 0;

   408

   409

		case WM_MOUSEMOVE: {

   410

			int x = (int16)LOWORD(lParam);

   411

			int y = (int16)HIWORD(lParam);

   412

			POINT pt;

   413

   414

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

   415

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

   416

			 * tracking the mouse for exiting the window */

   417

			if (!_cursor.in_window) {

   418

				_cursor.in_window = true;

   419

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

   420

   421

				DrawMouseCursor();

   422

			}

   423

   424

			if (_cursor.fix_at) {

   425

				int dx = x - _cursor.pos.x;

   426

				int dy = y - _cursor.pos.y;

   427

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

   428

					_cursor.delta.x += dx;

   429

					_cursor.delta.y += dy;

   430

   431

					pt.x = _cursor.pos.x;

   432

					pt.y = _cursor.pos.y;

   433

   434

					ClientToScreen(hwnd, &pt);

   435

					SetCursorPos(pt.x, pt.y);

   436

				}

   437

			} else {

   438

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

   439

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

   440

				_cursor.pos.x = x;

   441

				_cursor.pos.y = y;

   442

				_cursor.dirty = true;

   443

			}

   444

			MyShowCursor(false);

   445

			HandleMouseEvents();

   446

			return 0;

   447

		}

   448

   449

#if !defined(UNICODE)

   450

		case WM_INPUTLANGCHANGE: {

   451

			TCHAR locale[6];

   452

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

   453

   454

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

   455

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

   456

			return 1;

   457

		}

   458

#endif /* UNICODE */

   459

   460

		case WM_DEADCHAR:

   461

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

   462

			return 0;

   463

   464

		case WM_CHAR: {

   465

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

   466

			if (wParam < VK_SPACE) return 0;

   467

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

   468

			uint charcode = wParam;

   469

   470

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

   471

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

   472

			if (console && scancode == 41) {

   473

				console = false;

   474

				return 0;

   475

			}

   476

   477

#if !defined(UNICODE)

   478

			wchar_t w;

   479

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

   480

			charcode = len == 1 ? w : 0;

   481

#endif /* UNICODE */

   482

   483

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

   484

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

   485

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

   486

			return 0;

   487

		}

   488

   489

		case WM_KEYDOWN: {

   490

			keycode = MapWindowsKey(wParam);

   491

   492

			/* Silently drop all text messages as those will be handled by WM_CHAR

   493

			 * WM_KEYDOWN only handles CTRL+ commands and special keys like VK_LEFT, etc. */

   494

			if (keycode == 0 || (keycode > WKC_PAUSE && GB(keycode, 13, 4) == 0)) return 0;

   495

   496

			/* Keys handled in WM_CHAR */

   497

			if ((uint)(GB(keycode, 0, 12) - WKC_NUM_DIV) <= WKC_MINUS - WKC_NUM_DIV) return 0;