dlls/joy.cpl/main.c

   1 /*
   2  * Joystick testing control panel applet
   3  *
   4  * Copyright 2012 Lucas Fialho Zawacki
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  *
  20  */
  21
  22 #define NONAMELESSUNION
  23 #define COBJMACROS
  24 #define CONST_VTABLE
  25
  26 #include <stdarg.h>
  27 #include <windef.h>
  28 #include <winbase.h>
  29 #include <winuser.h>
  30 #include <commctrl.h>
  31 #include <cpl.h>
  32 #include "ole2.h"
  33
  34 #include "wine/debug.h"
  35 #include "joy.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(joycpl);
  38
  39 DECLSPEC_HIDDEN HMODULE hcpl;
  40
  41 /*********************************************************************
  42  *  DllMain
  43  */
  44 BOOL WINAPI DllMain(HINSTANCE hdll, DWORD reason, LPVOID reserved)
  45 {
  46     TRACE("(%p, %d, %p)\n", hdll, reason, reserved);
  47
  48     switch (reason)
  49     {
  50         case DLL_WINE_PREATTACH:
  51             return FALSE;  /* prefer native version */
  52
  53         case DLL_PROCESS_ATTACH:
  54             DisableThreadLibraryCalls(hdll);
  55             hcpl = hdll;
  56     }
  57     return TRUE;
  58 }
  59
  60 /***********************************************************************
  61  *  enum_callback [internal]
  62  *   Enumerates, creates and sets the common data format for all the joystick devices.
  63  *   First time it checks if space for the joysticks was already reserved
  64  *   and if not, just counts how many there are.
  65  */
  66 static BOOL CALLBACK enum_callback(const DIDEVICEINSTANCEW *instance, void *context)
  67 {
  68     struct JoystickData *data = context;
  69     struct Joystick *joystick;
  70     DIDEVCAPS caps;
  71
  72     if (data->joysticks == NULL)
  73     {
  74         data->num_joysticks += 1;
  75         return DIENUM_CONTINUE;
  76     }
  77
  78     joystick = &data->joysticks[data->cur_joystick];
  79     data->cur_joystick += 1;
  80
  81     IDirectInput8_CreateDevice(data->di, &instance->guidInstance, &joystick->device, NULL);
  82     IDirectInputDevice8_SetDataFormat(joystick->device, &c_dfDIJoystick);
  83
  84     joystick->instance = *instance;
  85
  86     caps.dwSize = sizeof(caps);
  87     IDirectInputDevice8_GetCapabilities(joystick->device, &caps);
  88
  89     joystick->num_buttons = caps.dwButtons;
  90     joystick->num_axes = caps.dwAxes;
  91
  92     return DIENUM_CONTINUE;
  93 }
  94
  95 /***********************************************************************
  96  *  initialize_joysticks [internal]
  97  */
  98 static void initialize_joysticks(struct JoystickData *data)
  99 {
 100     data->num_joysticks = 0;
 101     data->cur_joystick = 0;
 102     IDirectInput8_EnumDevices(data->di, DI8DEVCLASS_GAMECTRL, enum_callback, data, DIEDFL_ATTACHEDONLY);
 103     data->joysticks = HeapAlloc(GetProcessHeap(), 0, sizeof(struct Joystick) * data->num_joysticks);
 104
 105     /* Get all the joysticks */
 106     IDirectInput8_EnumDevices(data->di, DI8DEVCLASS_GAMECTRL, enum_callback, data, DIEDFL_ATTACHEDONLY);
 107 }
 108
 109 /***********************************************************************
 110  *  destroy_joysticks [internal]
 111  */
 112 static void destroy_joysticks(struct JoystickData *data)
 113 {
 114     int i;
 115
 116     for (i = 0; i < data->num_joysticks; i++)
 117     {
 118         IDirectInputDevice8_Unacquire(data->joysticks[i].device);
 119         IDirectInputDevice8_Release(data->joysticks[i].device);
 120     }
 121
 122     HeapFree(GetProcessHeap(), 0, data->joysticks);
 123 }
 124
 125 /*********************************************************************
 126  * list_dlgproc [internal]
 127  *
 128  */
 129 INT_PTR CALLBACK list_dlgproc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
 130 {
 131     TRACE("(%p, 0x%08x/%d, 0x%lx)\n", hwnd, msg, msg, lparam);
 132     switch (msg)
 133     {
 134         case WM_INITDIALOG:
 135         {
 136             int i;
 137             struct JoystickData *data = (struct JoystickData*) ((PROPSHEETPAGEW*)lparam)->lParam;
 138
 139             /* Set dialog information */
 140             for (i = 0; i < data->num_joysticks; i++)
 141             {
 142                 struct Joystick *joy = &data->joysticks[i];
 143                 SendDlgItemMessageW(hwnd, IDC_JOYSTICKLIST, LB_ADDSTRING, 0, (LPARAM) joy->instance.tszInstanceName);
 144             }
 145
 146             return TRUE;
 147         }
 148
 149         case WM_COMMAND:
 150
 151             switch (LOWORD(wparam))
 152             {
 153                 case IDC_BUTTONDISABLE:
 154                     FIXME("Disable selected joystick from being enumerated\n");
 155                     break;
 156
 157                 case IDC_BUTTONENABLE:
 158                     FIXME("Re-Enable selected joystick\n");
 159                     break;
 160             }
 161
 162             return TRUE;
 163
 164         case WM_NOTIFY:
 165             return TRUE;
 166
 167         default:
 168             break;
 169     }
 170     return FALSE;
 171 }
 172
 173 /*********************************************************************
 174  * Joystick testing functions
 175  *
 176  */
 177 static void poll_input(const struct Joystick *joy, DIJOYSTATE *state)
 178 {
 179     HRESULT  hr;
 180
 181     hr = IDirectInputDevice8_Poll(joy->device);
 182
 183     /* If it failed, try to acquire the joystick */
 184     if (FAILED(hr))
 185     {
 186         hr = IDirectInputDevice8_Acquire(joy->device);
 187         while (hr == DIERR_INPUTLOST) hr = IDirectInputDevice8_Acquire(joy->device);
 188     }
 189
 190     if (hr == DIERR_OTHERAPPHASPRIO) return;
 191
 192     IDirectInputDevice8_GetDeviceState(joy->device, sizeof(DIJOYSTATE), state);
 193 }
 194
 195 static DWORD WINAPI input_thread(void *param)
 196 {
 197     DIJOYSTATE state;
 198     struct JoystickData *data = param;
 199
 200     ZeroMemory(&state, sizeof(state));
 201
 202     while (!data->stop)
 203     {
 204         int i;
 205         poll_input(&data->joysticks[data->chosen_joystick], &state);
 206
 207         /* Indicate pressed buttons */
 208         for (i = 0; i < data->joysticks[data->chosen_joystick].num_buttons; i++)
 209             if (state.rgbButtons[i])
 210                 SendMessageW(data->buttons[i], BM_SETSTATE, TRUE, 0);
 211
 212         Sleep(TEST_POLL_TIME);
 213
 214         /* Reset button state */
 215         for (i = 0; i < data->joysticks[data->chosen_joystick].num_buttons; i++)
 216             SendMessageW(data->buttons[i], BM_SETSTATE, FALSE, 0);
 217     }
 218
 219     return 0;
 220 }
 221
 222 static void test_handle_joychange(HWND hwnd, struct JoystickData *data)
 223 {
 224     int i;
 225
 226     if (data->num_joysticks == 0) return;
 227
 228     data->chosen_joystick = SendDlgItemMessageW(hwnd, IDC_TESTSELECTCOMBO, CB_GETCURSEL, 0, 0);
 229
 230     /* Enable only  buttons present in the device */
 231     for (i = 0; i < TEST_MAX_BUTTONS; i++)
 232         ShowWindow(data->buttons[i], i <= data->joysticks[data->chosen_joystick].num_buttons);
 233 }
 234
 235 /*********************************************************************
 236  * button_number_to_wchar [internal]
 237  *  Transforms an integer in the interval [0,99] into a 2 character WCHAR string
 238  */
 239 static void button_number_to_wchar(int n, WCHAR str[3])
 240 {
 241     str[1] = n % 10 + '0';
 242     n /= 10;
 243     str[0] = n % 10 + '0';
 244     str[2] = '\0';
 245 }
 246
 247 static void draw_joystick_buttons(HWND hwnd, struct JoystickData* data)
 248 {
 249     int i;
 250     int row = 0, col = 0;
 251     WCHAR button_label[3];
 252     HINSTANCE hinst = (HINSTANCE) GetWindowLongPtrW(hwnd, GWLP_HINSTANCE);
 253     static WCHAR button_class[] = {'B','u','t','t','o','n','\0'};
 254
 255     for (i = 0; i < TEST_MAX_BUTTONS; i++)
 256     {
 257         if ((i % TEST_BUTTON_COL_MAX) == 0 && i != 0)
 258         {
 259             row += 1;
 260             col = 0;
 261         }
 262
 263         button_number_to_wchar(i + 1, button_label);
 264
 265         data->buttons[i] = CreateWindowW(button_class, button_label, WS_CHILD,
 266             TEST_BUTTON_X + TEST_NEXT_BUTTON_X*col, TEST_BUTTON_Y + TEST_NEXT_BUTTON_Y*row,
 267             TEST_BUTTON_SIZE_X, TEST_BUTTON_SIZE_Y,
 268             hwnd, NULL, NULL, hinst);
 269
 270         col += 1;
 271     }
 272 }
 273
 274 /*********************************************************************
 275  * test_dlgproc [internal]
 276  *
 277  */
 278 static INT_PTR CALLBACK test_dlgproc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
 279 {
 280     static HANDLE thread;
 281     static struct JoystickData *data;
 282     TRACE("(%p, 0x%08x/%d, 0x%lx)\n", hwnd, msg, msg, lparam);
 283
 284     switch (msg)
 285     {
 286         case WM_INITDIALOG:
 287         {
 288             int i;
 289
 290             data = (struct JoystickData*) ((PROPSHEETPAGEW*)lparam)->lParam;
 291
 292             /* Add enumerated joysticks to the combobox */
 293             for (i = 0; i < data->num_joysticks; i++)
 294             {
 295                 struct Joystick *joy = &data->joysticks[i];
 296                 SendDlgItemMessageW(hwnd, IDC_TESTSELECTCOMBO, CB_ADDSTRING, 0, (LPARAM) joy->instance.tszInstanceName);
 297             }
 298
 299             draw_joystick_buttons(hwnd, data);
 300
 301             return TRUE;
 302         }
 303
 304         case WM_COMMAND:
 305             switch(wparam)
 306             {
 307                 case MAKEWPARAM(IDC_TESTSELECTCOMBO, CBN_SELCHANGE):
 308                     test_handle_joychange(hwnd, data);
 309                 break;
 310             }
 311             return TRUE;
 312
 313         case WM_NOTIFY:
 314             switch(((LPNMHDR)lparam)->code)
 315             {
 316                 case PSN_SETACTIVE:
 317                 {
 318                     DWORD tid;
 319
 320                     /* Initialize input thread */
 321                     if (data->num_joysticks > 0)
 322                     {
 323                         data->stop = FALSE;
 324
 325                         /* Set the first joystick as default */
 326                         SendDlgItemMessageW(hwnd, IDC_TESTSELECTCOMBO, CB_SETCURSEL, 0, 0);
 327                         test_handle_joychange(hwnd, data);
 328
 329                         thread = CreateThread(NULL, 0, input_thread, (void*) data, 0, &tid);
 330                     }
 331                 }
 332                 break;
 333
 334                 case PSN_RESET:
 335                     /* Stop input thread */
 336                     data->stop = TRUE;
 337                     CloseHandle(thread);
 338                 break;
 339             }
 340             return TRUE;
 341     }
 342     return FALSE;
 343 }
 344
 345 /******************************************************************************
 346  * propsheet_callback [internal]
 347  */
 348 static int CALLBACK propsheet_callback(HWND hwnd, UINT msg, LPARAM lparam)
 349 {
 350     TRACE("(%p, 0x%08x/%d, 0x%lx)\n", hwnd, msg, msg, lparam);
 351     switch (msg)
 352     {
 353         case PSCB_INITIALIZED:
 354             break;
 355     }
 356     return 0;
 357 }
 358
 359 /******************************************************************************
 360  * display_cpl_sheets [internal]
 361  *
 362  * Build and display the dialog with all control panel propertysheets
 363  *
 364  */
 365 static void display_cpl_sheets(HWND parent, struct JoystickData *data)
 366 {
 367     INITCOMMONCONTROLSEX icex;
 368     PROPSHEETPAGEW psp[NUM_PROPERTY_PAGES];
 369     PROPSHEETHEADERW psh;
 370     DWORD id = 0;
 371
 372     OleInitialize(NULL);
 373     /* Initialize common controls */
 374     icex.dwSize = sizeof(INITCOMMONCONTROLSEX);
 375     icex.dwICC = ICC_LISTVIEW_CLASSES | ICC_BAR_CLASSES;
 376     InitCommonControlsEx(&icex);
 377
 378     ZeroMemory(&psh, sizeof(psh));
 379     ZeroMemory(psp, sizeof(psp));
 380
 381     /* Fill out all PROPSHEETPAGE */
 382     psp[id].dwSize = sizeof (PROPSHEETPAGEW);
 383     psp[id].hInstance = hcpl;
 384     psp[id].u.pszTemplate = MAKEINTRESOURCEW(IDD_LIST);
 385     psp[id].pfnDlgProc = list_dlgproc;
 386     psp[id].lParam = (INT_PTR) data;
 387     id++;
 388
 389     psp[id].dwSize = sizeof (PROPSHEETPAGEW);
 390     psp[id].hInstance = hcpl;
 391     psp[id].u.pszTemplate = MAKEINTRESOURCEW(IDD_TEST);
 392     psp[id].pfnDlgProc = test_dlgproc;
 393     psp[id].lParam = (INT_PTR) data;
 394     id++;
 395
 396     psp[id].dwSize = sizeof (PROPSHEETPAGEW);
 397     psp[id].hInstance = hcpl;
 398     psp[id].u.pszTemplate = MAKEINTRESOURCEW(IDD_FORCEFEEDBACK);
 399     psp[id].pfnDlgProc = NULL;
 400     psp[id].lParam = (INT_PTR) data;
 401     id++;
 402
 403     /* Fill out the PROPSHEETHEADER */
 404     psh.dwSize = sizeof (PROPSHEETHEADERW);
 405     psh.dwFlags = PSH_PROPSHEETPAGE | PSH_USEICONID | PSH_USECALLBACK;
 406     psh.hwndParent = parent;
 407     psh.hInstance = hcpl;
 408     psh.pszCaption = MAKEINTRESOURCEW(IDS_CPL_NAME);
 409     psh.nPages = id;
 410     psh.u3.ppsp = psp;
 411     psh.pfnCallback = propsheet_callback;
 412
 413     /* display the dialog */
 414     PropertySheetW(&psh);
 415
 416     OleUninitialize();
 417 }
 418
 419 /*********************************************************************
 420  * CPlApplet (joy.cpl.@)
 421  *
 422  * Control Panel entry point
 423  *
 424  * PARAMS
 425  *  hWnd    [I] Handle for the Control Panel Window
 426  *  command [I] CPL_* Command
 427  *  lParam1 [I] first extra Parameter
 428  *  lParam2 [I] second extra Parameter
 429  *
 430  * RETURNS
 431  *  Depends on the command
 432  *
 433  */
 434 LONG CALLBACK CPlApplet(HWND hwnd, UINT command, LPARAM lParam1, LPARAM lParam2)
 435 {
 436     static struct JoystickData data;
 437     TRACE("(%p, %u, 0x%lx, 0x%lx)\n", hwnd, command, lParam1, lParam2);
 438
 439     switch (command)
 440     {
 441         case CPL_INIT:
 442         {
 443             HRESULT hr;
 444
 445             /* Initialize dinput */
 446             hr = DirectInput8Create(GetModuleHandleW(NULL), DIRECTINPUT_VERSION, &IID_IDirectInput8W, (void**)&data.di, NULL);
 447
 448             if (FAILED(hr))
 449             {
 450                 ERR("Failed to initialize DirectInput: 0x%08x\n", hr);
 451                 return FALSE;
 452             }
 453
 454             /* Then get all the connected joysticks */
 455             initialize_joysticks(&data);
 456
 457             return TRUE;
 458         }
 459         case CPL_GETCOUNT:
 460             return 1;
 461
 462         case CPL_INQUIRE:
 463         {
 464             CPLINFO *appletInfo = (CPLINFO *) lParam2;
 465
 466             appletInfo->idName = IDS_CPL_NAME;
 467             appletInfo->idInfo = IDS_CPL_INFO;
 468             appletInfo->lData = 0;
 469             return TRUE;
 470         }
 471
 472         case CPL_DBLCLK:
 473             display_cpl_sheets(hwnd, &data);
 474             break;
 475
 476         case CPL_STOP:
 477             destroy_joysticks(&data);
 478
 479             /* And destroy dinput too */
 480             IDirectInput8_Release(data.di);
 481             break;
 482     }
 483
 484     return FALSE;
 485 }