DIEX4.CPP
上传用户:bangxh
上传日期:2007-01-31
资源大小:42235k
文件大小:16k
源码类别:

Windows编程

开发平台:

Visual C++

  1. /**************************************************************************
  2.     DIEX4.CPP - DirectInput simple sample 4
  3.     Demonstrates an application which retrieves buffered keyboard data
  4.     in non-exclusive mode via a game loop.
  5.  **************************************************************************/
  6. /**************************************************************************
  7.     (C) Copyright 1997 Microsoft Corp.  All rights reserved.
  8.     You have a royalty-free right to use, modify, reproduce and
  9.     distribute the Sample Files (and/or any modified version) in
  10.     any way you find useful, provided that you agree that
  11.     Microsoft has no warranty obligations or liability for any
  12.     Sample Application Files which are modified.
  13.  **************************************************************************/
  14. #include <windows.h>
  15. #include <dinput.h>
  16. #include "diex4.h"
  17. /****************************************************************************
  18.  *
  19.  *      Global Parameters
  20.  *
  21.  ****************************************************************************/
  22. #define DINPUT_BUFFERSIZE       16      /* Number of buffer elements */
  23. /****************************************************************************
  24.  *
  25.  *      Global variables
  26.  *
  27.  ****************************************************************************/
  28. char c_szClassName[] = "DIEX4";
  29. HINSTANCE       g_hinst;                /* My instance handle */
  30. BOOL            g_fPaused = TRUE;       /* Should I be paused? */
  31. /****************************************************************************
  32.  *
  33.  *      DirectInput globals
  34.  *
  35.  ****************************************************************************/
  36. LPDIRECTINPUT           g_pdi;
  37. LPDIRECTINPUTDEVICE     g_pKeyboard;
  38. char                    g_szText[1024]; /* What we display in client area */
  39. /****************************************************************************
  40.  *
  41.  *      Complain
  42.  *
  43.  *      Whine and moan.
  44.  *
  45.  ****************************************************************************/
  46. void
  47. Complain(
  48.     HWND hwndOwner,
  49.     HRESULT hr,
  50.     LPCSTR pszMessage
  51. )
  52. {
  53.     MessageBox(hwndOwner, pszMessage, "DirectInput Sample", MB_OK);
  54. }
  55. /****************************************************************************
  56.  *
  57.  *      DIInit
  58.  *
  59.  *      Initialize the DirectInput variables.
  60.  *
  61.  *      This entails the following four functions:
  62.  *
  63.  *          DirectInputCreate
  64.  *          IDirectInput::CreateDevice
  65.  *          IDirectInputDevice::SetDataFormat
  66.  *          IDirectInputDevice::SetCooperativeLevel
  67.  *
  68.  *      Reading buffered data requires another function:
  69.  *
  70.  *          IDirectInputDevice::SetProperty(DIPROP_BUFFERSIZE)
  71.  *
  72.  ****************************************************************************/
  73. BOOL
  74. DIInit(
  75.     HWND hwnd
  76. )
  77. {
  78.     HRESULT hr;
  79.     /*
  80.      *  Register with the DirectInput subsystem and get a pointer
  81.      *  to a IDirectInput interface we can use.
  82.      *
  83.      *  Parameters:
  84.      *
  85.      *      g_hinst
  86.      *
  87.      *          Instance handle to our application or DLL.
  88.      *
  89.      *      DIRECTINPUT_VERSION
  90.      *
  91.      *          The version of DirectInput we were designed for.
  92.      *          We take the value from the <dinput.h> header file.
  93.      *
  94.      *      &g_pdi
  95.      *
  96.      *          Receives pointer to the IDirectInput interface
  97.      *          that was created.
  98.      *
  99.      *      NULL
  100.      *
  101.      *          We do not use OLE aggregation, so this parameter
  102.      *          must be NULL.
  103.      *
  104.      */
  105.     hr = DirectInputCreate(g_hinst, DIRECTINPUT_VERSION, &g_pdi, NULL);
  106.     if (FAILED(hr)) {
  107.         Complain(hwnd, hr, "DirectInputCreate");
  108.         return FALSE;
  109.     }
  110.     /*
  111.      *  Obtain an interface to the system keyboard device.
  112.      *
  113.      *  Parameters:
  114.      *
  115.      *      GUID_SysKeyboard
  116.      *
  117.      *          The instance GUID for the device we wish to access.
  118.      *          GUID_SysKeyboard is a predefined instance GUID that
  119.      *          always refers to the system keyboard device.
  120.      *
  121.      *      &g_pKeyboard
  122.      *
  123.      *          Receives pointer to the IDirectInputDevice interface
  124.      *          that was created.
  125.      *
  126.      *      NULL
  127.      *
  128.      *          We do not use OLE aggregation, so this parameter
  129.      *          must be NULL.
  130.      *
  131.      */
  132.     hr = g_pdi->CreateDevice(GUID_SysKeyboard, &g_pKeyboard, NULL);
  133.     if (FAILED(hr)) {
  134.         Complain(hwnd, hr, "CreateDevice");
  135.         return FALSE;
  136.     }
  137.     /*
  138.      *  Set the data format to "keyboard format".
  139.      *
  140.      *  A data format specifies which controls on a device we
  141.      *  are interested in, and how they should be reported.
  142.      *
  143.      *  This tells DirectInput that we are interested in all keys
  144.      *  on the device, and they should be reported as DirectInput
  145.      *  DIK_* codes.
  146.      *
  147.      *  Parameters:
  148.      *
  149.      *      c_dfDIKeyboard
  150.      *
  151.      *          Predefined data format which describes
  152.      *          an array of 256 bytes, one per scancode.
  153.      */
  154.     hr = g_pKeyboard->SetDataFormat(&c_dfDIKeyboard);
  155.     if (FAILED(hr)) {
  156.         Complain(hwnd, hr, "SetDataFormat");
  157.         return FALSE;
  158.     }
  159.     /*
  160.      *  Set the cooperativity level to let DirectInput know how
  161.      *  this device should interact with the system and with other
  162.      *  DirectInput applications.
  163.      *
  164.      *  Parameters:
  165.      *
  166.      *      DISCL_NONEXCLUSIVE
  167.      *
  168.      *          Retrieve keyboard data when acquired, not interfering
  169.      *          with any other applications which are reading keyboard
  170.      *          data.
  171.      *
  172.      *      DISCL_FOREGROUND
  173.      *
  174.      *          If the user switches away from our application,
  175.      *          automatically release the keyboard back to the system.
  176.      *
  177.      */
  178.     hr = g_pKeyboard->SetCooperativeLevel(hwnd,
  179.                                        DISCL_NONEXCLUSIVE | DISCL_FOREGROUND);
  180.     if (FAILED(hr)) {
  181.         Complain(hwnd, hr, "SetCooperativeLevel");
  182.         return FALSE;
  183.     }
  184.     /*
  185.      *  IMPORTANT STEP IF YOU WANT TO USE BUFFERED DEVICE DATA!
  186.      *
  187.      *  DirectInput uses unbuffered I/O (buffer size = 0) by default.
  188.      *  If you want to read buffered data, you need to set a nonzero
  189.      *  buffer size.
  190.      *
  191.      *  Set the buffer size to DINPUT_BUFFERSIZE (defined above) elements.
  192.      *
  193.      *  The buffer size is a DWORD property associated with the device.
  194.      */
  195.     DIPROPDWORD dipdw =
  196.         {
  197.             {
  198.                 sizeof(DIPROPDWORD),        // diph.dwSize
  199.                 sizeof(DIPROPHEADER),       // diph.dwHeaderSize
  200.                 0,                          // diph.dwObj
  201.                 DIPH_DEVICE,                // diph.dwHow
  202.             },
  203.             DINPUT_BUFFERSIZE,              // dwData
  204.         };
  205.     hr = g_pKeyboard->SetProperty(DIPROP_BUFFERSIZE, &dipdw.diph);
  206.     if (FAILED(hr)) {
  207.         Complain(hwnd, hr, "Set buffer size");
  208.         return FALSE;
  209.     }
  210.     return TRUE;
  211. }
  212. /****************************************************************************
  213.  *
  214.  *      DITerm
  215.  *
  216.  *      Terminate our usage of DirectInput.
  217.  *
  218.  ****************************************************************************/
  219. void
  220. DITerm(void)
  221. {
  222.     /*
  223.      *  Destroy any lingering IDirectInputDevice object.
  224.      */
  225.     if (g_pKeyboard) {
  226.         /*
  227.          *  Cleanliness is next to godliness.  Unacquire the device
  228.          *  one last time just in case we got really confused and tried
  229.          *  to exit while the device is still acquired.
  230.          */
  231.         g_pKeyboard->Unacquire();
  232.         g_pKeyboard->Release();
  233.         g_pKeyboard = NULL;
  234.     }
  235.     /*
  236.      *  Destroy any lingering IDirectInput object.
  237.      */
  238.     if (g_pdi) {
  239.         g_pdi->Release();
  240.         g_pdi = NULL;
  241.     }
  242. }
  243. /****************************************************************************
  244.  *
  245.  *      Ex_OnPaint
  246.  *
  247.  *      Display the current keyboard state.
  248.  *
  249.  ****************************************************************************/
  250. LRESULT
  251. Ex_OnPaint(
  252.     HWND hwnd
  253. )
  254. {
  255.     PAINTSTRUCT ps;
  256.     HDC hdc = BeginPaint(hwnd, &ps);
  257.     if (hdc) {
  258.         ExtTextOut(hdc, 0, 0, ETO_OPAQUE, &ps.rcPaint, g_szText,
  259.                    lstrlen(g_szText), NULL);
  260.         EndPaint(hwnd, &ps);
  261.     }
  262.     return 0;
  263. }
  264. /****************************************************************************
  265.  *
  266.  *      Ex_OneFrame
  267.  *
  268.  *      The game plays here.
  269.  *
  270.  *      Our "game" consists entirely of reading keyboard data
  271.  *      and displaying it.
  272.  *
  273.  ****************************************************************************/
  274. void
  275. Ex_OneFrame(HWND hwnd)
  276. {
  277.     if (g_pKeyboard) {
  278.         DIDEVICEOBJECTDATA rgod[DINPUT_BUFFERSIZE]; /* Receives buffered data */
  279.         DWORD cod;
  280.         HRESULT hr;
  281.     again:;
  282.         cod = DINPUT_BUFFERSIZE;
  283.         hr = g_pKeyboard->GetDeviceData(sizeof(DIDEVICEOBJECTDATA),
  284.                                         rgod, &cod, 0);
  285.         if (hr != DI_OK) {
  286.             /*
  287.              *  We got an error or we got DI_BUFFEROVERFLOW.
  288.              *
  289.              *  Either way, it means that continuous contact with the
  290.              *  device has been lost, either due to an external
  291.              *  interruption, or because the buffer overflowed
  292.              *  and some events were lost.
  293.              *
  294.              *  Consequently, if a button was pressed at the time
  295.              *  the buffer overflowed or the connection was broken,
  296.              *  the corresponding "up" message might have been lost.
  297.              *
  298.              *  But since our simple sample doesn't actually have
  299.              *  any state associated with button up or down events,
  300.              *  there is no state to reset.  (In a real game, ignoring
  301.              *  the buffer overflow would result in the game thinking
  302.              *  a key was held down when in fact it isn't; it's just
  303.              *  that the "up" event got lost because the buffer
  304.              *  overflowed.)
  305.              *
  306.              *  If we want to be cleverer, we could do a
  307.              *  GetDeviceState() and compare the current state
  308.              *  against the state we think the device is in,
  309.              *  and process all the states that are currently
  310.              *  different from our private state.
  311.              *
  312.              */
  313.             /* << insert recovery code here if you need any >> */
  314.             if (hr == DIERR_INPUTLOST) {
  315.                 hr = g_pKeyboard->Acquire();
  316.                 if (SUCCEEDED(hr)) {
  317.                     goto again;
  318.                 }
  319.             }
  320.         }
  321.         /*
  322.          *  In order for it to be worth our while to parse the
  323.          *  buffer elements, the GetDeviceData must have succeeded,
  324.          *  and we must have received some data at all.
  325.          */
  326.         if (SUCCEEDED(hr) && cod > 0) {
  327.             char szBuf[1024];
  328.             DWORD iod;
  329.             /*
  330.              *  Study each of the buffer elements and process them.
  331.              *
  332.              *  Since we really don't do anything, our "processing"
  333.              *  consists merely of squirting the name into our
  334.              *  local buffer.
  335.              */
  336.             for (iod = 0; iod < cod; iod++) {
  337.                 wsprintf(szBuf, "%02x was %s",
  338.                          rgod[iod].dwOfs,
  339.                          (rgod[iod].dwData & 0x80) ? "pressed" : "released");
  340.             }
  341.             /*
  342.              *  Trigger a repaint only if the status string changed.
  343.              *  This avoids flicker.
  344.              */
  345.             if (lstrcmp(g_szText, szBuf)) {
  346.                 lstrcpy(g_szText, szBuf);
  347.                 InvalidateRect(hwnd, NULL, TRUE);
  348.             }
  349.         }
  350.     }
  351.     /*
  352.      *  Sleep for a few milliseconds to simulate a 30fps frame rate.
  353.      */
  354.     Sleep(1000 / 30);
  355. }
  356. /****************************************************************************
  357.  *
  358.  *      Ex_SyncAcquire
  359.  *
  360.  *      Acquire or unacquire the keyboard, depending on the the g_fPaused
  361.  *      flag.  This synchronizes the device with our internal view of
  362.  *      the world.
  363.  *
  364.  ****************************************************************************/
  365. void
  366. Ex_SyncAcquire(HWND hwnd)
  367. {
  368.     if (g_fPaused) {
  369.         if (g_pKeyboard) g_pKeyboard->Unacquire();
  370.     } else {
  371.         if (g_pKeyboard) g_pKeyboard->Acquire();
  372.     }
  373. }
  374. /****************************************************************************
  375.  *
  376.  *      Ex_WndProc
  377.  *
  378.  *      Window procedure for simple sample.
  379.  *
  380.  ****************************************************************************/
  381. LRESULT CALLBACK
  382. Ex_WndProc(
  383.     HWND hwnd,
  384.     UINT msg,
  385.     WPARAM wParam,
  386.     LPARAM lParam
  387. )
  388. {
  389.     switch (msg) {
  390.     case WM_PAINT:      return Ex_OnPaint(hwnd);
  391.     /*
  392.      *  WM_ACTIVATE
  393.      *
  394.      *      Windows sends this message when the window becomes
  395.      *      the active window or stops being the active window.
  396.      *
  397.      *      wParam = WA_INACTIVE if window is no longer active
  398.      *
  399.      *      wParam = WA_ACTIVE or WA_CLICKACTIVE if window is now active
  400.      *
  401.      *      If we are losing activation, then pause.
  402.      *
  403.      *      If we are gaining activation, then unpause.
  404.      *
  405.      *      After deciding whether we are paused or unpaused,
  406.      *      tell DirectInput that we don't (paused) or do (unpaused)
  407.      *      want non-exclusive access to the keyboard.
  408.      *
  409.      */
  410.     case WM_ACTIVATE:
  411.         g_fPaused = wParam == WA_INACTIVE;
  412.         Ex_SyncAcquire(hwnd);
  413.         break;
  414.     case WM_DESTROY:
  415.         PostQuitMessage(0);
  416.         break;
  417.     }
  418.     return DefWindowProc(hwnd, msg, wParam, lParam);
  419. }
  420. /****************************************************************************
  421.  *
  422.  *      AppInit
  423.  *
  424.  *      Set up everything the application needs to get started.
  425.  *
  426.  ****************************************************************************/
  427. HWND
  428. AppInit(
  429.     HINSTANCE hinst,
  430.     int nCmdShow
  431. )
  432. {
  433.     /*
  434.      *  Save instance handle for future reference.
  435.      */
  436.     g_hinst = hinst;
  437.     /*
  438.      *  Set up the window class.
  439.      */
  440.     WNDCLASS wc;
  441.     wc.hCursor        = LoadCursor(0, IDC_ARROW);
  442.     wc.hIcon          = LoadIcon(NULL, MAKEINTRESOURCE(IDI_APPLICATION));
  443.     wc.lpszMenuName   = NULL;
  444.     wc.lpszClassName  = c_szClassName;
  445.     wc.hbrBackground  = 0;
  446.     wc.hInstance      = hinst;
  447.     wc.style          = 0;
  448.     wc.lpfnWndProc    = Ex_WndProc;
  449.     wc.cbClsExtra     = 0;
  450.     wc.cbWndExtra     = 0;
  451.     if (!RegisterClass(&wc)) {
  452.         return NULL;
  453.     }
  454.     HWND hwnd = CreateWindow(
  455.                     c_szClassName,                  // Class name
  456.                     "DIEX4 - Alt+F4 to exit",       // Caption
  457.                     WS_OVERLAPPEDWINDOW,            // Style
  458.                     CW_USEDEFAULT, CW_USEDEFAULT,   // Position
  459.                     CW_USEDEFAULT, CW_USEDEFAULT,   // Size
  460.                     NULL,                           // No parent
  461.                     NULL,                           // No menu
  462.                     g_hinst,                        // inst handle
  463.                     0                               // no params
  464.                     );
  465.     if (!DIInit(hwnd)) {
  466.         DestroyWindow(hwnd);
  467.         return NULL;
  468.     }
  469.     ShowWindow(hwnd, nCmdShow);
  470.     return hwnd;
  471. }
  472. /****************************************************************************
  473.  *
  474.  *      WinMain
  475.  *
  476.  *      Application entry point.
  477.  *
  478.  ****************************************************************************/
  479. int PASCAL
  480. WinMain(HINSTANCE hinst, HINSTANCE hinstPrev, LPSTR szCmdLine, int nCmdShow)
  481. {
  482.     MSG msg;
  483.     msg.wParam = 0;         /* In case something goes horribly wrong */
  484.     HWND hwnd = AppInit(hinst, nCmdShow);
  485.     if (hwnd) {
  486.         /*
  487.          *  Standard game loop.
  488.          */
  489.         for (;;) {
  490.             if (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {
  491.                 /* If it's a quit message, we're outta here */
  492.                 if (msg.message == WM_QUIT) {
  493.                     break;
  494.                 } else {
  495.                     TranslateMessage(&msg);
  496.                     DispatchMessage(&msg);
  497.                 }
  498.             } else if (g_fPaused) {
  499.                 WaitMessage();
  500.             } else {
  501.                 Ex_OneFrame(hwnd);
  502.             }
  503.         }
  504.     }
  505.     DITerm();
  506.     return msg.wParam;
  507. }