使用WM_KEYDOWN、WM_KEYUP结合map处理按键时的延迟优化咨询

现有实现的核心问题 #

• 容器性能瓶颈：使用std::map存储按键状态，每次查找、插入操作均为O(log n)复杂度，多键同时按下时频繁的容器操作会引发延迟。
• 消息处理顺序不合理：主循环中先调用DispatchMessage再处理按键消息，窗口过程的额外逻辑可能阻塞输入状态更新。
• 重复消息未过滤：Windows按住按键时会发送重复的WM_KEYDOWN消息，现有代码未做处理，易导致状态冗余更新。
• 动态内存开销：std::map的动态插入会产生内存分配开销，长期运行后可能积累大量无用的按键状态条目。

优化方案 #

1. 替换为高效容器 #

虚拟键码（VK_*）范围为0-255，直接使用固定大小的std::array存储状态，访问复杂度为O(1)，彻底消除map的查找/插入开销。

2. 调整消息处理顺序 #

在分发消息到窗口过程前优先处理按键消息，确保输入状态及时更新。

3. 过滤重复按键消息 #

通过WM_KEYDOWN消息的lParam第30位判断是否为重复消息，避免重复设置按键状态。

4. 完善状态重置逻辑 #

处理窗口失去焦点事件，重置所有按键状态，避免后台残留的按键状态导致逻辑错误。

优化后的代码示例 #

基础定义 #

#include <array>
#include <algorithm>

enum class KeyStatus { NONE_GET, Down, Hold, Up };
using Key = int; // 与Windows虚拟键码对应

// 初始化所有按键状态为NONE_GET
std::array<KeyStatus, 256> StatusKey{};

按键事件处理 #

void UpdateKeyEvent(const MSG& msg) noexcept
{
    switch (msg.message)
    {
    case WM_KEYDOWN:
    case WM_SYSKEYDOWN: // 处理系统按键（如Alt组合键）
        // 检查是否为重复按键消息
        bool isRepeat = (msg.lParam & 0x40000000) != 0;
        if (!isRepeat)
        {
            Key key = static_cast<Key>(msg.wParam);
            if (key >= 0 && key < 256)
            {
                auto& status = StatusKey[key];
                if (status == KeyStatus::NONE_GET || status == KeyStatus::Up)
                {
                    status = KeyStatus::Down;
                }
            }
        }
        break;

    case WM_KEYUP:
    case WM_SYSKEYUP:
        Key key = static_cast<Key>(msg.wParam);
        if (key >= 0 && key < 256)
        {
            StatusKey[key] = KeyStatus::Up;
        }
        break;

    case WM_KILLFOCUS:
        // 窗口失去焦点时重置所有按键状态
        std::fill(StatusKey.begin(), StatusKey.end(), KeyStatus::NONE_GET);
        break;
    }
}

状态更新与查询 #

void UpdateStatusKey() noexcept
{
    for (auto& status : StatusKey)
    {
        if (status == KeyStatus::Down)
        {
            status = KeyStatus::Hold;
        }
    }
}

float GetKeyRaw(const Key& key) noexcept
{
    if (key >= 0 && key < 256)
    {
        auto status = StatusKey[key];
        if (status != KeyStatus::Up && status != KeyStatus::NONE_GET)
        {
            return 1.0f;
        }
    }
    return 0.0f;
}

优化后的主循环 #

MSG Msg;
while (true)
{
    // 批量处理所有待处理消息
    while (PeekMessage(&Msg, nullptr, 0, 0, PM_REMOVE))
    {
        if (Msg.message == WM_QUIT)
            goto ExitMainLoop;

        UpdateKeyEvent(Msg); // 优先处理输入
        TranslateMessage(&Msg);
        DispatchMessage(&Msg);
    }

    // 帧更新逻辑
    UpdateStatusKey();
    // 执行游戏/应用核心逻辑...
}
ExitMainLoop:

额外优化建议 #

• 使用MsgWaitForMultipleObjects替代PeekMessage，在无消息时进入等待状态，降低CPU占用同时保证输入响应及时性。
• 若需支持更多特殊按键（如鼠标按键、多媒体键），可扩展数组范围或结合std::unordered_map处理超出基础虚拟键码的按键。

内容的提问来源于stack exchange，提问作者Hải Duy