{- |
Drawing of creatures. 
Takes into account damage etc. -}
module Dodge.Creature.Picture
    ( basicCrPict
    , circLine
    ) where
import Dodge.Data
import Dodge.Creature.Stance.Data
import Dodge.Creature.State.Data
--import Dodge.Creature.AlertLevel.Data
import Dodge.Picture.Layer
import Picture
import Geometry

import Control.Lens
import Data.List
import qualified Data.IntMap.Strict as IM
basicCrPict 
    :: Color    -- ^ Creature color
    -> Creature 
    -> Picture
basicCrPict col cr = pictures 
    [ onLayer CrLayer . piercingMod $ bluntScale naked 
--    , drawAwakeLevel cr
    , drawEquipment cr
    ]
  where 
    cdir = _crDir cr
    naked 
        | strikeMelee   = color white $ circleSolid $ _crRad cr
        | pdam > 200    = color red $ circleSolid $ _crRad cr
        | pdam > 99     = color white $ circleSolid $ _crRad cr
        | otherwise     = pictures [color col' $ circleSolid $ _crRad cr, circLine $ _crRad cr]
    pastDams = _crPastDamage $ _crState cr
    pdam = sum $ concatMap (map _dmAmount) pastDams
    col' = light . light . light $ light col
    bluntDam :: Maybe Point2
    bluntDam = find isBluntDam (concat pastDams) >>= (\dm -> (-.-) <$> dm ^? dmFrom <*> dm ^? dmTo)
    bluntScale = case fmap argV bluntDam of
        Just a -> rotate (a + cdir) . scale 0.8 1.2 . rotate (negate $ cdir + a)
        _   -> id
    isBluntDam Blunt{} = True
    isBluntDam _ = False
    piercingDam = find isPiercingDam (concat pastDams) >>= (\dm -> (-.-) <$> dm ^? dmFrom <*> dm ^? dmTo)
    isPiercingDam Piercing{} = True
    isPiercingDam _ = False
    piercingMod = case fmap argV piercingDam of
        Just a -> rotate (a + cdir) . scale 0.8 1.2 . rotate (negate $ cdir + a)
        _   -> id
    strikeMelee = case _crMeleeCooldown cr of
        Nothing -> False
        Just x  -> x > 5

--drawAwakeLevel
--    :: Creature
--    -> Picture
--drawAwakeLevel cr = case cr ^? crAttentionDir of
--    Just (AttentiveTo [0]) -> setPos . color red $ circleSolid 5
--    _        -> setPos . color blue $ circleSolid 5
--  where
--    setPos = translate 0 (_crRad cr)

drawEquipment 
    :: Creature 
    -> Picture
drawEquipment cr = pictures $ map f $ IM.toList (_crInv cr)
  where 
    f (i,it) = case it ^? itEquipPict of
        Just g | (cr ^? crStance . posture) == Just Aiming -> g cr i
        _      -> blank

circLine :: Float -> Picture
circLine x = line [(0,0),(x,0)]