AI refactor

2021-05-14 12:04:12 +02:00
parent 2dd3756682
commit 0b26761be5
14 changed files with 282 additions and 762 deletions
@@ -0,0 +1,75 @@
+module Dodge.Creature.LtAutoCrit
+    ( ltAutoCrit 
+    ) 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 Dodge.RandomHelp
+
+import Data.Maybe
+import qualified Data.IntMap.Strict as IM
+import Control.Lens
+import Control.Monad.State
+import System.Random
+
+
+ltAutoCrit :: Creature
+ltAutoCrit = defaultCreature
+    { _crPict = basicCrPict red
+    , _crUpdate = stateUpdate $ impulsiveAI $ composeInternalAIs
+        [ performActions
+        , watchUpdateStrat
+            [ (crHasTargetLOS, \w cr -> StrategyActions (ShootAt 0) 
+                [DoActionIf (not . crIsAiming) DrawWeapon
+                ,DoActionThen 
+                    (DoActionWhile crHasTargetLOS $ ArbitraryAction chooseMovement)
+                    (DoImpulses [ChangeStrategy WatchAndWait])
+                ]
+              )
+            , (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,ltAutoGun),(1,medkit 100)]
+    , _crInvSel = 0
+    , _crRad = 10
+    , _crHP = 500
+    }
+
+chooseMovement :: Creature -> World -> Action
+chooseMovement cr w 
+    | dist cpos p > 200 = DoImpulses [UseItem,TurnToward p 0.05 , MoveForward 3]
+    | dist cpos p < 80 = DoImpulses [UseItem,TurnToward p 0.05, MoveForward (-3) ]
+    | errorAngleVV 22 (p' -.- cpos) (unitVectorAtAngle (_crDir cr)) < 0.4
+          = DoImpulses [UseItem,TurnToward p' 0.01, Move (0,3)]
+    | otherwise
+          = DoImpulses [UseItem,TurnToward p' 0.05, Move (0,3)]
+  where
+    cpos = _crPos cr
+    tcr = _creatures w IM.! 0
+    p = _crPos tcr
+    v = vNormal $ p -.- cpos
+    p' = p +.+ 0.5 *.* (v -.- 20 *.* (normalizeV v))