module Game.LambdaHack.Client.AI
(
queryAI, pongAI
#ifdef EXPOSE_INTERNAL
, refreshTarget, pickAction
#endif
) where
import Control.Exception.Assert.Sugar
import qualified Data.EnumMap.Strict as EM
import qualified Data.Text as T
import Game.LambdaHack.Client.AI.HandleAbilityClient
import Game.LambdaHack.Client.AI.PickActorClient
import Game.LambdaHack.Client.AI.PickTargetClient
import Game.LambdaHack.Client.AI.Strategy
import Game.LambdaHack.Client.MonadClient
import Game.LambdaHack.Client.State
import Game.LambdaHack.Common.Actor
import Game.LambdaHack.Common.Faction
import Game.LambdaHack.Common.MonadStateRead
import Game.LambdaHack.Common.Msg
import Game.LambdaHack.Common.Random
import Game.LambdaHack.Common.Request
import Game.LambdaHack.Common.State
import Game.LambdaHack.Content.ModeKind
queryAI :: MonadClient m => ActorId -> m RequestAI
queryAI oldAid = do
side <- getsClient sside
fact <- getsState $ (EM.! side) . sfactionD
let mleader = gleader fact
wasLeader = fmap fst mleader == Just oldAid
(aidToMove, bToMove) <- pickActorToMove refreshTarget oldAid
RequestAnyAbility reqAny <-
if ftactic (gplayer fact) == TBlock && not wasLeader
then return $! RequestAnyAbility ReqWait
else pickAction (aidToMove, bToMove)
let req = ReqAITimed reqAny
mtgt2 <- getsClient $ fmap fst . EM.lookup aidToMove . stargetD
if wasLeader && mleader /= Just (aidToMove, mtgt2)
then return $! ReqAILeader aidToMove mtgt2 req
else return $! req
pongAI :: MonadClient m => m RequestAI
pongAI = return ReqAIPong
refreshTarget :: MonadClient m
=> ActorId -> (ActorId, Actor)
-> m (Maybe ((ActorId, Actor), (Target, PathEtc)))
refreshTarget oldLeader (aid, body) = do
side <- getsClient sside
assert (bfid body == side `blame` "AI tries to move an enemy actor"
`twith` (aid, body, side)) skip
assert (not (bproj body) `blame` "AI gets to manually move its projectiles"
`twith` (aid, body, side)) skip
stratTarget <- targetStrategy oldLeader aid
tgtMPath <-
if nullStrategy stratTarget then
return Nothing
else do
tmp <- rndToAction $ frequency $ bestVariant stratTarget
return $ Just tmp
oldTgt <- getsClient $ EM.lookup aid . stargetD
let _debug = T.unpack
$ "\nHandleAI symbol:" <+> tshow (bsymbol body)
<> ", aid:" <+> tshow aid
<> ", pos:" <+> tshow (bpos body)
<> "\nHandleAI oldTgt:" <+> tshow oldTgt
<> "\nHandleAI strTgt:" <+> tshow stratTarget
<> "\nHandleAI target:" <+> tshow tgtMPath
modifyClient $ \cli ->
cli {stargetD = EM.alter (const $ tgtMPath) aid (stargetD cli)}
return $! case tgtMPath of
Just (tgt, Just pathEtc) -> Just ((aid, body), (tgt, pathEtc))
_ -> Nothing
pickAction :: MonadClient m => (ActorId, Actor) -> m RequestAnyAbility
pickAction (aid, body) = do
side <- getsClient sside
assert (bfid body == side `blame` "AI tries to move enemy actor"
`twith` (aid, bfid body, side)) skip
assert (not (bproj body) `blame` "AI gets to manually move its projectiles"
`twith` (aid, bfid body, side)) skip
stratAction <- actionStrategy aid
rndToAction $ frequency $ bestVariant stratAction