Add pistolCrit file, tweak shell spin params

2021-05-13 19:21:52 +02:00
parent 2402e1e868
commit eed7865192
7 changed files with 147 additions and 45 deletions
@@ -0,0 +1,72 @@
+module Dodge.Creature.PistolCrit
+    ( pistolCrit 
+    ) where
+import Dodge.Data
+import Dodge.Default
+import Dodge.Creature.Picture
+import Dodge.Creature.Test
+import Dodge.Creature.ActionRat
+import Dodge.Creature.ChooseTarget
+import Dodge.Creature.SetTarget
+import Dodge.Creature.Rationality
+import Dodge.Creature.AlertLevel
+import Dodge.Creature.State
+import Dodge.Creature.State.Data
+import Dodge.Item.Weapon
+import Dodge.Item.Consumable
+import Geometry
+import Picture
+
+import Data.Maybe
+import qualified Data.IntMap.Strict as IM
+import Control.Lens
+
+pistolCrit :: Creature
+pistolCrit = defaultCreature
+    { _crPict = basicCrPict red
+    , _crUpdate = stateUpdate $ impulsiveAI $ composeInternalAIs
+        [ performActions
+        , watchUpdateStrat
+            [ (crHasTargetLOS, \w cr -> StrategyActions (ShootAt 0) [chooseMovement cr w])
+            , (crAwayFromPost, goToPostStrat)
+            ]
+        , basicPerceptionUpdate [0]
+        , doStrategyActions
+        , reloadOverride
+        , targetYouWhenCognizant
+        , overrideInternal (onBoth (&&) crHasTarget (crStratConMatches (GetTo (0,0))))
+            $ \ _ -> crActionPlan . crStrategy .~ WatchAndWait
+        ]
+    , _crActionPlan = ActionPlan
+        { _crImpulse    = []
+        , _crAction     = []
+        , _crStrategy   = StrategyActions WatchAndWait [StartSentinelPost]
+        , _crGoal       = []
+        }
+    , _crInv = IM.fromList [(0,pistol),(1,medkit 100)]
+    , _crInvSel = 0
+    , _crRad = 10
+    , _crHP = 500
+    }
+
+chooseMovement :: Creature -> World -> Action
+chooseMovement cr w = undefined
+
+retreatAction :: Creature -> Creature -> Action
+retreatAction tcr cr = 
+    [TurnToward retreatOffset 0.2]
+    `DoImpulsesAlongside`
+    NoAction
+  where 
+    cpos = _crPos cr
+    tpos = _crPos tcr
+    retreatOffset = 
+        let a | dist cpos tpos < 50 = 0
+              | isLeftOfA (_crDir cr) (argV $ tpos -.- cpos)
+                    = -0.7
+              | otherwise = 0.7
+        in  fromMaybe tpos $ intersectLineLine' cpos 
+                                                     (cpos +.+ rotateV a (tpos -.- cpos))
+                                                     tpos
+                                                     (tpos +.+ vNormal (cpos -.- tpos))
+