diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/HWiid.hsc | 178 |
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 |