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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
|
{-# LANGUAGE ForeignFunctionInterface, DeriveDataTypeable, GeneralizedNewtypeDeriving #-}
module HWiid ( BDAddr(..)
, Wiimote
, WiimoteIRSource(..)
, WiimoteState(..)
, WiimoteExtState(..)
, WiimoteMesg(..)
, nullWiimote
, bdAddrAny
, hwiidFlagMesgInterface
, hwiidFlagNonblock
, hwiidReportStatus
, hwiidReportButtons
, hwiidReportAcc
, hwiidReportIR
, hwiidReportNunchuk
, hwiidReportExt
, hwiidLed1
, hwiidLed2
, hwiidLed3
, hwiidLed4
, hwiidButton2
, hwiidButton1
, hwiidButtonB
, hwiidButtonA
, hwiidButtonMinus
, hwiidButtonHome
, hwiidButtonLeft
, hwiidButtonRight
, hwiidButtonDown
, hwiidButtonUp
, hwiidButtonPlus
, hwiidNunchukButtonZ
, hwiidNunchukButtonC
, hwiidIRMaxX
, hwiidIRMaxY
, hwiidMesgTypeStatus
, hwiidMesgTypeButton
, hwiidExtNone
, hwiidExtNunchuk
, hwiidOpen
, hwiidOpenTimeout
, hwiidClose
, hwiidGetState
, hwiidSetReportMode
, hwiidSetLed
, hwiidGetMesg
) where
import Data.Bits
import Data.Int
import Data.Typeable
import Data.Word
import Foreign.C.Types
import Foreign.Marshal.Alloc (alloca, free)
import Foreign.Marshal.Array (peekArray)
import Foreign.Marshal.Utils (with)
import Foreign.Ptr
import Foreign.Storable
import System.Posix.Clock (TimeSpec)
#include <cwiid.h>
hwiidFlagMesgInterface :: CInt
hwiidFlagMesgInterface = (#const CWIID_FLAG_MESG_IFC)
hwiidFlagNonblock :: CInt
hwiidFlagNonblock = (#const CWIID_FLAG_NONBLOCK)
hwiidReportStatus :: Word8
hwiidReportStatus = (#const CWIID_RPT_STATUS)
hwiidReportButtons :: Word8
hwiidReportButtons = (#const CWIID_RPT_BTN)
hwiidReportAcc :: Word8
hwiidReportAcc = (#const CWIID_RPT_ACC)
hwiidReportIR :: Word8
hwiidReportIR = (#const CWIID_RPT_IR)
hwiidReportNunchuk :: Word8
hwiidReportNunchuk = (#const CWIID_RPT_NUNCHUK)
hwiidReportExt :: Word8
hwiidReportExt = hwiidReportNunchuk
hwiidLed1 :: Word8
hwiidLed1 = (#const CWIID_LED1_ON)
hwiidLed2 :: Word8
hwiidLed2 = (#const CWIID_LED2_ON)
hwiidLed3 :: Word8
hwiidLed3 = (#const CWIID_LED3_ON)
hwiidLed4 :: Word8
hwiidLed4 = (#const CWIID_LED4_ON)
hwiidButton2 :: Word16
hwiidButton2 = (#const CWIID_BTN_2)
hwiidButton1 :: Word16
hwiidButton1 = (#const CWIID_BTN_1)
hwiidButtonB :: Word16
hwiidButtonB = (#const CWIID_BTN_B)
hwiidButtonA :: Word16
hwiidButtonA = (#const CWIID_BTN_A)
hwiidButtonMinus :: Word16
hwiidButtonMinus = (#const CWIID_BTN_MINUS)
hwiidButtonHome :: Word16
hwiidButtonHome = (#const CWIID_BTN_HOME)
hwiidButtonLeft :: Word16
hwiidButtonLeft = (#const CWIID_BTN_LEFT)
hwiidButtonRight :: Word16
hwiidButtonRight = (#const CWIID_BTN_RIGHT)
hwiidButtonDown :: Word16
hwiidButtonDown = (#const CWIID_BTN_DOWN)
hwiidButtonUp :: Word16
hwiidButtonUp = (#const CWIID_BTN_UP)
hwiidButtonPlus :: Word16
hwiidButtonPlus = (#const CWIID_BTN_PLUS)
hwiidNunchukButtonZ :: Word16
hwiidNunchukButtonZ = (#const CWIID_NUNCHUK_BTN_Z)
hwiidNunchukButtonC :: Word16
hwiidNunchukButtonC = (#const CWIID_NUNCHUK_BTN_C)
hwiidIRMaxX :: CInt
hwiidIRMaxX = (#const CWIID_IR_X_MAX)
hwiidIRMaxY :: CInt
hwiidIRMaxY = (#const CWIID_IR_Y_MAX)
hwiidMesgTypeStatus :: (#type enum cwiid_mesg_type)
hwiidMesgTypeStatus = (#const CWIID_MESG_STATUS)
hwiidMesgTypeButton :: (#type enum cwiid_mesg_type)
hwiidMesgTypeButton = (#const CWIID_MESG_BTN)
hwiidExtNone :: (#type enum cwiid_ext_type)
hwiidExtNone = (#const CWIID_EXT_NONE)
hwiidExtNunchuk :: (#type enum cwiid_ext_type)
hwiidExtNunchuk = (#const CWIID_EXT_NUNCHUK)
data BDAddr = BDAddr (Word8, Word8, Word8, Word8, Word8, Word8)
deriving (Eq, Ord, Show)
instance Storable BDAddr where
sizeOf _ = (#size bdaddr_t)
alignment _ = alignment (undefined :: CInt)
peek addr = do
b0 <- peekByteOff addr 0
b1 <- peekByteOff addr 1
b2 <- peekByteOff addr 2
b3 <- peekByteOff addr 3
b4 <- peekByteOff addr 4
b5 <- peekByteOff addr 5
return $ BDAddr (b0, b1, b2, b3, b4, b5)
poke addr (BDAddr (b0, b1, b2, b3, b4, b5)) = do
pokeByteOff addr 0 b0
pokeByteOff addr 1 b1
pokeByteOff addr 2 b2
pokeByteOff addr 3 b3
pokeByteOff addr 4 b4
pokeByteOff addr 5 b5
bdAddrAny :: BDAddr
bdAddrAny = BDAddr (0, 0, 0, 0, 0, 0)
data WiimoteIRSource = WiimoteIRSource
{ irValid :: CChar
, irPosX :: Word16
, irPosY :: Word16
, irSize :: Int8
} deriving (Eq, Show)
instance Storable WiimoteIRSource where
sizeOf _ = (#size struct cwiid_ir_src)
alignment _ = alignment (undefined :: CInt)
peek src = do
valid <- (#peek struct cwiid_ir_src, valid) src
posx <- (#peek struct cwiid_ir_src, pos[0]) src
posy <- (#peek struct cwiid_ir_src, pos[1]) src
size <- (#peek struct cwiid_ir_src, size) src
return $ WiimoteIRSource valid posx posy size
poke _ _ = error "Can't write WiimoteIRSource"
data WiimoteState = WiimoteState
{ stateLed :: Word8
, stateRumble :: Word8
, stateBattery :: Word8
, stateButtons :: Word16
, stateAccX :: Word8
, stateAccY :: Word8
, stateAccZ :: Word8
, stateIRSources :: [WiimoteIRSource]
, stateExt :: WiimoteExtState
} deriving (Eq, Show)
instance Storable WiimoteState where
sizeOf _ = (#size struct cwiid_state)
alignment _ = alignment (undefined :: CInt)
peek state = do
led <- (#peek struct cwiid_state, led) state
rumble <- (#peek struct cwiid_state, rumble) state
battery <- (#peek struct cwiid_state, battery) state
buttons <- (#peek struct cwiid_state, buttons) state
accX <- (#peek struct cwiid_state, acc[0]) state
accY <- (#peek struct cwiid_state, acc[1]) state
accZ <- (#peek struct cwiid_state, acc[2]) state
irSources <- peekArray (#const CWIID_IR_SRC_COUNT) $ (#ptr struct cwiid_state, ir_src) state
exttype <- (#peek struct cwiid_state, ext_type) state
extstate <- peekExtState exttype $ (#ptr struct cwiid_state, ext) state
return $ WiimoteState led rumble battery buttons accX accY accZ (filter (\src -> (irValid src) /= 0) irSources) extstate
poke _ _ = fail "Can't write WiimoteState"
data WiimoteMesg = WiimoteStatusMesg
{ mesgType :: (#type enum cwiid_mesg_type)
, mesgBattery :: Word8
}
| WiimoteButtonMesg
{ mesgType :: (#type enum cwiid_mesg_type)
, mesgButtons :: Word16
}
| WiimoteMesgOther
{ mesgType :: (#type enum cwiid_mesg_type)
}
deriving (Eq, Show)
instance Storable WiimoteMesg where
sizeOf _ = (#size union cwiid_mesg)
alignment _ = alignment (undefined :: CInt)
peek mesg = do
mesgtype <- (#peek union cwiid_mesg, type) mesg
case () of
_ | mesgtype == hwiidMesgTypeStatus -> do
battery <- (#peek struct cwiid_status_mesg, battery) mesg
return $ WiimoteStatusMesg mesgtype battery
| mesgtype == hwiidMesgTypeButton -> do
buttons <- (#peek struct cwiid_btn_mesg, buttons) mesg
return $ WiimoteButtonMesg mesgtype buttons
| otherwise -> return $ WiimoteMesgOther mesgtype
poke _ _ = fail "Can't write WiimoteMesg"
data WiimoteExtState = WiimoteNunchukState
{ extType :: (#type enum cwiid_ext_type)
, extNunchukStickX :: Word8
, extNunchukStickY :: Word8
, extNunchukButtons :: Word8
}
| WiimoteOtherState
{ extType :: (#type enum cwiid_ext_type)
}
deriving (Eq, Show)
peekExtState :: (#type enum cwiid_ext_type) -> Ptr WiimoteExtState -> IO WiimoteExtState
peekExtState exttype state
| exttype == hwiidExtNunchuk = do
posx <- (#peek struct nunchuk_state, stick[0]) state
posy <- (#peek struct nunchuk_state, stick[1]) state
buttons <- (#peek struct nunchuk_state, buttons) state
return $ WiimoteNunchukState exttype posx posy buttons
| otherwise = return $ WiimoteOtherState exttype
newtype Wiimote = Wiimote (Ptr Wiimote) deriving (Eq, Ord, Show, Storable)
nullWiimote :: Wiimote
nullWiimote = Wiimote nullPtr
foreign import ccall unsafe "cwiid.h cwiid_open_timeout"
cwiid_open_timeout :: Ptr BDAddr -> CInt -> CInt -> IO Wiimote
hwiidOpenTimeout :: BDAddr -> CInt -> CInt -> IO Wiimote
hwiidOpenTimeout addr flags timeout = with addr $ \addrptr -> cwiid_open_timeout addrptr flags timeout
defaultTimeout :: CInt
defaultTimeout = 5
hwiidOpen :: BDAddr -> CInt -> IO Wiimote
hwiidOpen addr flags = hwiidOpenTimeout addr flags defaultTimeout
foreign import ccall unsafe "cwiid.h cwiid_close"
hwiidClose :: Wiimote -> IO CInt
foreign import ccall unsafe "cwiid.h cwiid_get_state"
cwiid_get_state :: Wiimote -> Ptr WiimoteState -> IO CInt
hwiidGetState :: Wiimote -> IO WiimoteState
hwiidGetState wiimote = alloca $ \state -> do
cwiid_get_state wiimote state
peek state
foreign import ccall unsafe "cwiid.h cwiid_set_rpt_mode"
hwiidSetReportMode :: Wiimote -> Word8 -> IO CInt
foreign import ccall unsafe "cwiid.h cwiid_set_led"
hwiidSetLed :: Wiimote -> Word8 -> IO CInt
foreign import ccall unsafe "cwiid.h cwiid_get_mesg"
cwiid_get_mesg :: Wiimote -> Ptr CInt -> Ptr (Ptr WiimoteMesg) -> Ptr TimeSpec -> IO CInt
hwiidGetMesg :: Wiimote -> IO [WiimoteMesg]
hwiidGetMesg wiimote = alloca $ \countptr -> alloca $ \arrayptr -> alloca $ \timestamp -> do
ret <- cwiid_get_mesg wiimote countptr arrayptr timestamp
count <- peek countptr
array <- peek arrayptr
if (ret == 0) then do
list <- peekArray (fromIntegral count) array
free array
return list
else return []
|