diff options
Diffstat (limited to 'src/WiimotePlayer.hs')
| -rw-r--r-- | src/WiimotePlayer.hs | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/src/WiimotePlayer.hs b/src/WiimotePlayer.hs new file mode 100644 index 0000000..2487d17 --- /dev/null +++ b/src/WiimotePlayer.hs @@ -0,0 +1,44 @@ +{-# LANGUAGE DeriveDataTypeable #-} + +module WiimotePlayer ( WiimotePlayer(..) + , newWiimotePlayer + ) where + +import Control.Monad +import Data.Bits +import Data.Ratio ((%)) +import Data.Typeable +import HWiid + +import Player + + +data WiimotePlayer = WiimotePlayer Wiimote + deriving (Typeable, Show) + +instance Player WiimotePlayer where + playerUpdate (WiimotePlayer wiimote) tank = do + buttons <- hwiidGetState wiimote >>= return . stateButtons + messages <- hwiidGetMesg wiimote + + let shoot = any (\m -> (mesgType m == hwiidMesgTypeButton) && (test (mesgButtons m) hwiidButtonB)) $ messages + x = (if (test buttons hwiidButtonLeft) then (-1) else 0) + (if (test buttons hwiidButtonRight) then 1 else 0) + y = (if (test buttons hwiidButtonDown) then (-1) else 0) + (if (test buttons hwiidButtonUp) then 1 else 0) + --ax = aimx - (fromRational . toRational . tankX $ tank) + --ay = aimy - (fromRational . toRational . tankY $ tank) + move = (x /= 0 || y /= 0) + + angle = if move then Just $ fromRational $ round ((atan2 y x)*1000000*180/pi)%1000000 else Nothing + --aangle = if (ax /= 0 || ay /= 0) then Just $ fromRational $ round ((atan2 ay ax)*1000000*180/pi)%1000000 else Nothing + return (WiimotePlayer wiimote, angle, move, Nothing, shoot) + + +newWiimotePlayer :: IO WiimotePlayer +newWiimotePlayer = do + wiimote <- hwiidOpen bdAddrAny (hwiidFlagMesgInterface .|. hwiidFlagNonblock) + when (wiimote == nullWiimote) $ fail "Wiimote error" + hwiidSetReportMode wiimote hwiidReportButtons + return $ WiimotePlayer wiimote + +test :: (Bits a) => a -> a -> Bool +test field bits = (field .&. bits) == bits
\ No newline at end of file |