summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/HWiid.hsc178
1 files changed, 176 insertions, 2 deletions
diff --git a/src/HWiid.hsc b/src/HWiid.hsc
index 6f4679d..3699a05 100644
--- a/src/HWiid.hsc
+++ b/src/HWiid.hsc
@@ -1,9 +1,183 @@
-{-# LANGUAGE ForeignFunctionInterface, DeriveDataTypeable #-}
+{-# LANGUAGE ForeignFunctionInterface, DeriveDataTypeable, GeneralizedNewtypeDeriving #-}
-module HWiid (
+module HWiid ( BDAddr(..)
+ , Wiimote
+ , WiimoteState(..)
+ , nullWiimote
+ , bdAddrAny
+ , hwiidLed1
+ , hwiidLed2
+ , hwiidLed3
+ , hwiidLed4
+ , hwiidReportStatus
+ , hwiidReportButtons
+ , hwiidOpen
+ , hwiidOpenTimeout
+ , hwiidClose
+ , hwiidGetState
+ , hwiidSetReportMode
+ , hwiidSetLed
+ , hwiidButton2
+ , hwiidButton1
+ , hwiidButtonB
+ , hwiidButtonA
+ , hwiidButtonMinus
+ , hwiidButtonHome
+ , hwiidButtonLeft
+ , hwiidButtonRight
+ , hwiidButtonDown
+ , hwiidButtonUp
+ , hwiidButtonPlus
+ , hwiidNunchukButtonZ
+ , hwiidNunchukButtonC
) where
+import Data.Word
+import Foreign.C.Types
+import Foreign.Marshal.Alloc (alloca)
+import Foreign.Marshal.Utils (with)
+import Foreign.Ptr
+import Foreign.Storable
+
+
#include <cwiid.h>
+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 WiimoteState = WiimoteState
+ { stateButtons :: Word16
+ } deriving (Eq, Show)
+
+instance Storable WiimoteState where
+ sizeOf _ = (#size struct cwiid_state)
+ alignment _ = alignment (undefined :: CInt)
+
+ peek state = do
+ buttons <- (#peek struct cwiid_state, buttons) state
+ return $ WiimoteState (buttons)
+
+ poke state (WiimoteState (buttons)) = do
+ (#poke struct cwiid_state, buttons) state buttons
+
+
+newtype Wiimote = Wiimote (Ptr Wiimote) deriving (Eq, Ord, Show, Storable)
+
+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)
+
+hwiidReportStatus :: Word8
+hwiidReportStatus = (#const CWIID_RPT_STATUS)
+
+hwiidReportButtons :: Word8
+hwiidReportButtons = (#const CWIID_RPT_BTN)
+
+
+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)
+
+
+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