Tweak weapons

src/Dodge/Item/Attachment.hs

@@ -1,4 +1,9 @@
 module Dodge.Item.Attachment
+    ( charFiringStratI
+    , scrollCharMode
+    , decreaseFuse
+    , increaseFuse
+    )
     where
 import Dodge.Data
 import Dodge.Item.Attachment.Data
@@ -6,19 +11,9 @@ import Dodge.Default
 
 import Control.Lens hiding ((|>),(<|))
 import Data.Sequence
-import Data.Maybe (fromJust)
+import Data.Maybe
 import qualified Data.IntMap.Strict as IM
 
-incCharMode' :: ItAttachment -> ItAttachment
-incCharMode' = itCharMode %~ cycleL
-  where
-    cycleL (x :<| xs) = xs |> x
-    cycleL xs = xs
-decCharMode' :: ItAttachment -> ItAttachment
-decCharMode' = itCharMode %~ cycleR
-  where
-    cycleR (xs :|> x) = x <| xs
-    cycleR xs = xs
 
 scrollCharMode
     :: Float    -- ^ Amount scrolled
@@ -34,34 +29,32 @@ incCharMode
     :: Int      -- ^ Inventory selection
     -> World 
     -> World
-incCharMode i = creatures . ix 0 . crInv . ix i . itAttachment %~ incCharMode'
+incCharMode i = creatures . ix 0 . crInv . ix i . itAttachment . itCharMode %~ cycleL
+  where
+    cycleL (x :<| xs) = xs |> x
+    cycleL xs = xs
+
 decCharMode 
     :: Int      -- ^ Inventory selection
     -> World 
     -> World
-decCharMode i = creatures . ix 0 . crInv . ix i . itAttachment %~ decCharMode'
+decCharMode i = creatures . ix 0 . crInv . ix i . itAttachment . itCharMode %~ cycleR
-
-charFiringStrat 
-    :: [(Char, Creature -> World -> World)] -- ^ Different firing effects for different characters
-    -> Creature      -- ^ Creature id
-    -> World 
-    -> World
-charFiringStrat strats cr w = case w ^? creatures . ix cid . crInv 
-        . ix (_crInvSel $ _creatures w IM.! cid) . itAttachment . itCharMode of
-    Just (c :<| _) -> fromJust (Prelude.lookup c strats) cr w
-    _ -> w
   where
-    cid = _crID cr
+    cycleR (xs :|> x) = x <| xs
-charFiringStratI
+    cycleR xs = xs
-    :: [(Char, (Item -> Creature -> World -> World) -> Item -> Creature -> World -> World)] -- ^ Different firing effects for different characters
+
-    -> (Item -> Creature -> World -> World)
+type ChainEffect = 
+    (Item -> Creature -> World -> World)
     -> Item
     -> Creature
     -> World 
     -> World
+charFiringStratI
+    :: [(Char, ChainEffect)] -- ^ Different firing effects for different characters
+    -> ChainEffect
 charFiringStratI strats eff item cr w = case w ^? creatures . ix cid . crInv 
         . ix (_crInvSel $ _creatures w IM.! cid) . itAttachment . itCharMode of
-    Just (c :<| _) -> fromJust (Prelude.lookup c strats) eff item cr w
+    Just (c :<| _) -> fromMaybe id (Prelude.lookup c strats) eff item cr w
     _ -> w
   where
     cid = _crID cr
@@ -84,7 +77,8 @@ increaseFuse fuse itid w = w
 decreaseFuse 
     :: Int  -- ^ Old fuse time
     -> Int  -- ^ Inventory item reference (I believe)
-    -> World -> World
+    -> World 
+    -> World
 decreaseFuse fuse itid w = w 
     & creatures . ix 0 . crInv . ix itid %~ 
         ( itScrollUp .~ decreaseFuse newTime )

src/Dodge/Item/Weapon.hs

@@ -67,7 +67,7 @@ pistol = defaultGun
         , useTimeCheckI
         , withSoundI 0 
         , useAmmo 1
---        , withRandomDirI 0.1
+        , randSpreadDir
         , withMuzFlareI
         ]
     , _itLeftClickUse = Nothing

src/Dodge/Item/Weapon/AutoGun.hs

@@ -32,10 +32,13 @@ autoGun = defaultAutoGun
     , _itUse   = useAmmoParams
     , _itUseModifiers =
         [ ammoCheckI
+        , charFiringStratI
+            [('S', hammerCheckI)
+            ]
         , useTimeCheckI
         , charFiringStratI
-            [('M', torqueBefore 0.05) 
+            [('M', torqueBefore 0.08) 
-            ,('S', hammerCheckI)
+            ,('S', torqueBefore 0.05) 
             ]
         , withSoundI autoGunSound
         , useAmmo 1

src/Dodge/Item/Weapon/TriggerType.hs

@@ -31,6 +31,7 @@ module Dodge.Item.Weapon.TriggerType
     , shootL
     , useTimeCheckI
     , ammoCheckI
+    , randSpreadDir
     )
     where
 import Dodge.Data
@@ -282,6 +283,14 @@ withMuzFlareI f it cr w = tempLightForAt 3 pos2 -- . muzzleFlashAt pos2
   where 
     --pos = _crPos cr +.+ _crRad cr *.* unitVectorAtAngle (_crDir cr)
     pos2 = _crPos cr +.+ (2 * _crRad cr) *.* unitVectorAtAngle (_crDir cr)
+{- | Applies the effect to a randomly rotated creature,
+ - rotation amount given by wpSpread  -}
+randSpreadDir :: ChainEffect
+randSpreadDir f it cr w = f it (cr & crDir +~ a) $ set randGen g w
+  where 
+    acc = _wpSpread it
+    (a, g) = randomR (-acc,acc) $ _randGen w
+
 {- | Applies the effect to a randomly rotated creature.  -}
 withRandomDirI
     :: Float    -- ^ Max possible rotation

src/Dodge/Render/Picture.hs

@@ -9,7 +9,7 @@ import Dodge.Picture
 import Dodge.Picture.Layer
 import Dodge.Render.HUD
 import Dodge.Render.MenuScreen
-import Dodge.Graph
+--import Dodge.Graph
 import Geometry
 import Picture
 import Polyhedra.Data
@@ -59,7 +59,8 @@ customMouseCursor w =
     $ pictures [ line [V2 (-5) 0,V2 5 0] , line [V2 0 (-5),V2 0 5] ]
 
 testPic :: World -> Picture
-testPic w = color green . pictures . map (flip thickLine 5 . tflat2) . graphToEdges $ _pathGraph w
+testPic _ = blank
+--testPic w = color green . pictures . map (flip thickLine 5 . tflat2) . graphToEdges $ _pathGraph w
 
 crDraw :: World -> Creature -> Picture
 crDraw w c = _crPict c c w