input.cpp

Go to the documentation of this file.

#include "stdafx.h"
#include "include/input.h"
#include "include/debug.h"
#include "include/exceptions.h"

#pragma comment(lib, "dinput8.lib")
#pragma comment(lib, "dxguid.lib")

namespace yappy {
namespace input {

using error::checkDXResult;
using error::DIError;

DInput::DInput(HINSTANCE hInst, HWND hWnd, bool foreground, bool exclusive)
{
    m_key.fill(false);

    debug::writeLine(L"Initializing DirectInput...");

    HRESULT hr = S_OK;

    // Initialize DirectInput
    IDirectInput8 *ptmpDi;
    hr = ::DirectInput8Create(hInst, DIRECTINPUT_VERSION, IID_IDirectInput8,
        reinterpret_cast<void **>(&ptmpDi), NULL);
    checkDXResult<DIError>(hr, "DirectInput8Create() failed");
    m_pDi.reset(ptmpDi);

    // Key Device
    debug::writeLine(L"Get Keyboard...");
    IDirectInputDevice8 *ptmpKeyDevice;
    hr = m_pDi->CreateDevice(GUID_SysKeyboard, &ptmpKeyDevice, NULL);
    checkDXResult<DIError>(hr, "IDirectInput8::CreateDevice() failed");
    m_pKeyDevice.reset(ptmpKeyDevice);

    hr = m_pKeyDevice->SetDataFormat(&c_dfDIKeyboard);
    checkDXResult<DIError>(hr, "IDirectInputDevice8::SetDataFormat() failed");
    DWORD coopLevel = DISCL_NOWINKEY;
    coopLevel |= (foreground ? DISCL_FOREGROUND : DISCL_BACKGROUND);
    coopLevel |= (exclusive ? DISCL_EXCLUSIVE : DISCL_NONEXCLUSIVE);
    hr = m_pKeyDevice->SetCooperativeLevel(hWnd, coopLevel);
    checkDXResult<DIError>(hr, "IDirectInputDevice8::SetCooperativeLevel() failed");
    debug::writeLine(L"Get Keyboard OK");

    // GamePad Device
    updateControllers(hWnd, foreground, exclusive);

    debug::writeLine(L"Initializing DirectInput OK");
}

DInput::~DInput()
{
    // Unacquire (ignore errors)
    m_pKeyDevice->Unacquire();
    for (auto &dev : m_pPadDevs) {
        dev->Unacquire();
    }
    debug::writeLine(L"Finalize DirectInput");
}

// file local functions for callback
namespace {

BOOL CALLBACK enumDevicesCallback(LPCDIDEVICEINSTANCE lpddi, LPVOID pvRef)
{
    std::vector<DIDEVICEINSTANCE> *devs = reinterpret_cast<decltype(devs)>(pvRef);
    devs->emplace_back(*lpddi);
    return DIENUM_CONTINUE;
}

BOOL CALLBACK enumAxesCallback(LPCDIDEVICEOBJECTINSTANCE lpddoi, LPVOID pvRef)
{
    DIPROPRANGE range = { 0 };
    range.diph.dwSize = sizeof(range);
    range.diph.dwHeaderSize = sizeof(range.diph);
    range.diph.dwHow = DIPH_BYID;
    range.diph.dwObj = lpddoi->dwType;
    range.lMax = DInput::AXIS_RANGE;
    range.lMin = -DInput::AXIS_RANGE;
    IDirectInputDevice8 *pdev = reinterpret_cast<decltype(pdev)>(pvRef);
    pdev->SetProperty(DIPROP_RANGE, &range.diph);
    return DIENUM_CONTINUE;
}

}

void DInput::updateControllers(HWND hwnd, bool foreground, bool exclusive)
{
    debug::writeLine(L"Get controllers...");

    HRESULT hr = S_OK;

    m_padInstList.clear();
    m_pPadDevs.clear();

    hr = m_pDi->EnumDevices(DI8DEVCLASS_GAMECTRL, enumDevicesCallback,
        &m_padInstList, DIEDFL_ATTACHEDONLY);
    checkDXResult<DIError>(hr, "IDirectInput8::EnumDevices() failed");

    for (const auto &padInst : m_padInstList) {
        IDirectInputDevice8 *ptmpDevice;
        hr = m_pDi->CreateDevice(padInst.guidInstance, &ptmpDevice, nullptr);
        checkDXResult<DIError>(hr, "IDirectInput8::CreateDevice() failed");
        util::ComPtr<IDirectInputDevice8> pDevice(ptmpDevice);

        hr = pDevice->SetDataFormat(&c_dfDIJoystick);
        checkDXResult<DIError>(hr, "IDirectInputDevice8::SetDataFormat() failed");
        DWORD coopLevel = 0;
        coopLevel |= (foreground ? DISCL_FOREGROUND : DISCL_BACKGROUND);
        coopLevel |= (exclusive ? DISCL_EXCLUSIVE : DISCL_NONEXCLUSIVE);
        hr = pDevice->SetCooperativeLevel(hwnd, coopLevel);
        checkDXResult<DIError>(hr, "IDirectInputDevice8::SetCooperativeLevel() failed");
        hr = pDevice->EnumObjects(enumAxesCallback, pDevice.get(), DIDFT_AXIS);
        checkDXResult<DIError>(hr, "IDirectInputDevice8::EnumObjects() failed");

        m_pPadDevs.emplace_back(std::move(pDevice));
        DIJOYSTATE js = { 0 };
        m_pad.emplace_back(js);
    }

    {
        int i = 0;
        for (const auto &padInst : m_padInstList) {
            debug::writef(L"[Controller %d]", i);
            debug::writef(L"tszInstanceName = %s", padInst.tszInstanceName);
            debug::writef(L"tszProductName  = %s", padInst.tszProductName);
            i++;
        }
        debug::writef(L"%zu controller(s) found", m_pPadDevs.size());
    }
}

void DInput::processFrame()
{
    HRESULT hr = S_OK;

    // Keyboard
    m_key.fill(0);
    do {
        hr = m_pKeyDevice->Acquire();
    } while (hr == DIERR_INPUTLOST);
    if (hr != DIERR_OTHERAPPHASPRIO) {
        BYTE tmp[256];
        hr = m_pKeyDevice->GetDeviceState(sizeof(tmp), &tmp);
        if (SUCCEEDED(hr)) {
            for (int i = 0; i < 256; i++) {
                m_key[i] = ((tmp[i] & 0x80) != 0);
            }
        }
    }
    else {
        // DISCL_FOREGROUND and app is background now
        //debug::writeLine(L"DIERR_OTHERAPPHASPRIO");
    }

    // Pad
    // TODO: think twice about error handling
    // TODO: sudden disconnect?
    for (size_t i = 0; i < m_pPadDevs.size(); i++) {
        ZeroMemory(&m_pad[i], sizeof(m_pad[i]));
        memset(m_pad[i].rgdwPOV, -1, sizeof(m_pad[i].rgdwPOV));

        do {
            hr = m_pPadDevs[i]->Acquire();
        } while (hr == DIERR_INPUTLOST);
        if (hr == DIERR_OTHERAPPHASPRIO) {
            // ISCL_FOREGROUND and app is background now
            continue;
        }
        hr = m_pPadDevs[i]->Poll();
        hr = m_pPadDevs[i]->GetDeviceState(sizeof(m_pad[i]), &m_pad[i]);
        // if error, m_pad[i] will be cleared state
    }
}

DInput::KeyData DInput::getKeys() const
{
    return m_key;
}

int DInput::getPadCount() const
{
    ASSERT(m_pad.size() == m_pPadDevs.size());
    return static_cast<int>(m_pPadDevs.size());
}

void DInput::getPadState(DIJOYSTATE *state, int index) const
{
    *state = m_pad.at(index);
}


namespace {

struct DikConvEntry {
    const char *str;
    const wchar_t *wstr;
};

#define ENT(s) {s, L##s}
const DikConvEntry DikConvTable[DInput::KEY_NUM] = {
    // 0x00-0xff
    ENT("?"),       ENT("ESCAPE"),  ENT("1"),       ENT("2"),
    ENT("3"),       ENT("4"),       ENT("5"),       ENT("6"),
    ENT("7"),       ENT("8"),       ENT("9"),       ENT("0"),
    ENT("MINUS"),   ENT("EQUALS"),  ENT("BACK"),    ENT("TAB"),
    // 0x10-0x1f
    ENT("Q"),       ENT("W"),       ENT("E"),       ENT("R"),
    ENT("T"),       ENT("Y"),       ENT("U"),       ENT("I"),
    ENT("O"),       ENT("P"),       ENT("LBRACKET"),ENT("RBRACKET"),
    ENT("RETURN"),  ENT("LCONTROL"),ENT("A"),       ENT("S"),
    // 0x20-0x2f
    ENT("D"),       ENT("F"),       ENT("G"),       ENT("H"),
    ENT("J"),       ENT("K"),       ENT("L"),       ENT("SEMIC"),
    ENT("APOST"),   ENT("GRAVE"),   ENT("LSHIFT"),  ENT("BSLASH"),
    ENT("Z"),       ENT("X"),       ENT("C"),       ENT("V"),
    // 0x30-0x3f
    ENT("B"),       ENT("N"),       ENT("M"),       ENT("COMMA"),
    ENT("PERIOD"),  ENT("SLASH"),   ENT("RSHIFT"),  ENT("MULT"),
    ENT("LMENU"),   ENT("SPACE"),   ENT("CAPITAL"), ENT("F1"),
    ENT("F2"),      ENT("F3"),      ENT("F4"),      ENT("F5"),
    // 0x40-0x4f
    ENT("F6"),      ENT("F7"),      ENT("F8"),      ENT("F9"),
    ENT("F10"),     ENT("NUMLOCK"), ENT("SCROLL"),  ENT("NUMPAD7"),
    ENT("NUMPAD8"), ENT("NUMPAD9"), ENT("SUB"),     ENT("NUMPAD4"),
    ENT("NUMPAD5"), ENT("NUMPAD6"), ENT("ADD"),     ENT("NUMPAD1"),
    // 0x50-0x5f
    ENT("NUMPAD2"), ENT("NUMPAD3"), ENT("NUMPAD0"), ENT("DECIMAL"),
    ENT("?"),       ENT("?"),       ENT("OEM_102"), ENT("F11"),
    ENT("F12"),     ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    // 0x60-0x6f
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("F13"),     ENT("F14"),     ENT("F15"),     ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    // 0x70-0x7f
    ENT("KANA"),    ENT("?"),       ENT("?"),       ENT("ABNT_C1"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("CONVERT"), ENT("?"),       ENT("NOCONV"),
    ENT("?"),       ENT("YEN"),     ENT("ABNT_C2"), ENT("?"),
    // 0x80-0x8f
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("NUMPADEQ"),ENT("?"),       ENT("?"),
    // 0x90-0x9f
    ENT("PREVTR"),  ENT("AT"),      ENT("COLON"),   ENT("ULINE"),
    ENT("KANJI"),   ENT("STOP"),    ENT("AX"),      ENT("UNLABEL"),
    ENT("?"),       ENT("NEXTTR"),  ENT("?"),       ENT("?"),
    ENT("NUMPADET"),ENT("RCONTROL"),ENT("?"),       ENT("?"),
    // 0xa0-0xaf
    ENT("MUTE"),    ENT("CALC"),    ENT("PLYPAUSE"),ENT("?"),
    ENT("MSTOP"),   ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("VOLDOWN"), ENT("?"),
    // 0xb0-0xbf
    ENT("VOLUP"),   ENT("?"),       ENT("WEBHOME"), ENT("NUMPADCM"),
    ENT("?"),       ENT("DIVIDE"),  ENT("?"),       ENT("SYSRQ"),
    ENT("RMENU"),   ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),   ENT("?"),
    // 0xc0-0xcf
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("PAUSE"),   ENT("?"),       ENT("HOME"),
    ENT("UP"),      ENT("PGUP"),    ENT("?"),       ENT("LEFT"),
    ENT("?"),       ENT("RIGHT"),   ENT("?"),       ENT("END"),
    // 0xd0-0xdf
    ENT("DOWN"),    ENT("PGDN"),    ENT("INSERT"),  ENT("DELETE"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("LWIN"),
    ENT("RWIN"),    ENT("APPS"),    ENT("POWER"),   ENT("SLEEP"),
    // 0xe0-0xef
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("WAKE"),
    ENT("?"),       ENT("WEBSCH"),  ENT("WEBFAV"),  ENT("WEBREF"),
    ENT("WEBSTOP"), ENT("WEBFWD"),  ENT("WEBBACK"), ENT("MYCOM"),
    ENT("MAIL"),    ENT("MEDIASEL"),ENT("?"),       ENT("?"),
    // 0xf0-0xff
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
    ENT("?"),       ENT("?"),       ENT("?"),       ENT("?"),
};
#undef ENT

}   // namespace

const char *dikToString(int dik)
{
    if (dik < 0 || dik >= 256) {
        throw std::logic_error("invalid DIK");
    }
    return DikConvTable[dik].str;
}

const wchar_t *dikToWString(int dik)
{
    if (dik < 0 || dik >= 256) {
        throw std::logic_error("invalid DIK");
    }
    return DikConvTable[dik].wstr;
}

}   // namespace input
}   // namespace yappy