summaryrefslogtreecommitdiffhomepage
path: root/opentrack/win32-joystick.hpp
blob: f7629c3d90d9801fa3dee312a523fd582110ceab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
#pragma once

#ifdef _WIN32

#include <cstring>
#include <memory>
#include <vector>
#include <functional>
#include <algorithm>
#include <unordered_map>
#ifndef DIRECTINPUT_VERSION
#   define DIRECTINPUT_VERSION 0x800
#endif
#include <dinput.h>
#include <windows.h>
#include "opentrack-compat/timer.hpp"
#include <QString>
#include <QDebug>
#include <QMutex>
#include <QMutexLocker>
#include <QMainWindow>

namespace std {
template<>
struct hash<QString>
{
    std::size_t operator()(const QString& value) const
    {
        return qHash(value);
    }
};
}

#ifdef BUILD_api
#   include "opentrack-compat/export.hpp"
#else
#   include "opentrack-compat/import.hpp"
#endif

struct OPENTRACK_EXPORT win32_joy_ctx
{
    using fn = std::function<void(const QString& guid, int btn, bool held)>;
    
    enum { joy_axis_size = 65535 };
    
    struct joy_info
    {
        QString name, guid;
    };

    void poll(fn f);
    bool poll_axis(const QString& guid, int axes[8]);
    std::vector<joy_info> get_joy_info();
    
    win32_joy_ctx(const win32_joy_ctx&) = delete;
    win32_joy_ctx& operator=(const win32_joy_ctx&) = delete;
    
    win32_joy_ctx();
    ~win32_joy_ctx();
    
private:
    enum { joylist_refresh_ms = 100 };
    
    QMutex mtx;
    Timer timer_joylist;
    QMainWindow fake_main_window;
    LPDIRECTINPUT8 di;
    
    static QString guid_to_string(const GUID guid);
    void release();
    void refresh(bool first);
    
    struct joy;
    static std::unordered_map<QString, std::shared_ptr<joy>>& joys();
    
    struct joy
    {
        LPDIRECTINPUTDEVICE8 joy_handle;
        QString guid, name;
        bool pressed[128];
        Timer first_timer;

        joy(LPDIRECTINPUTDEVICE8 handle, const QString& guid, const QString& name)
            : joy_handle(handle), guid(guid), name(name)
        {
            qDebug() << "got joy" << guid;
            for (int i = 0; i < 128; i++)
                pressed[i] = false;
        }

        ~joy()
        {
            qDebug() << "nix joy" << guid;
            release();
        }

        void release();
        bool poll(fn f);
    };

    class enum_state
    {
        std::unordered_map<QString, std::shared_ptr<joy>> joys;
        QMainWindow& fake_main_window;
        LPDIRECTINPUT8 di;
        
        std::vector<QString> all;
        static BOOL CALLBACK EnumJoysticksCallback(const DIDEVICEINSTANCE* pdidInstance, VOID* pContext);
        static BOOL CALLBACK EnumObjectsCallback(const DIDEVICEOBJECTINSTANCE* pdidoi, VOID* ctx);
    public:
        enum_state(std::unordered_map<QString, std::shared_ptr<joy>>& joys, QMutex &mtx, QMainWindow& fake_main_window, LPDIRECTINPUT8 di);
    };
};

#endif