diff options
Diffstat (limited to 'ws2015/ffp/blaetter')
-rw-r--r-- | ws2015/ffp/blaetter/07/FFP_U07_Monaden2.hs | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/ws2015/ffp/blaetter/07/FFP_U07_Monaden2.hs b/ws2015/ffp/blaetter/07/FFP_U07_Monaden2.hs index 18eacfa..8e92dee 100644 --- a/ws2015/ffp/blaetter/07/FFP_U07_Monaden2.hs +++ b/ws2015/ffp/blaetter/07/FFP_U07_Monaden2.hs | |||
@@ -278,18 +278,19 @@ type MM = Map ([Monster], [Monster]) Double -- MM stands for MonsterManual | |||
278 | 278 | ||
279 | memoWinchance :: [Monster] -> [Monster] -> Double | 279 | memoWinchance :: [Monster] -> [Monster] -> Double |
280 | memoWinchance atts defs = runST (newSTRef Map.empty >>= memoWinchance' atts defs) | 280 | memoWinchance atts defs = runST (newSTRef Map.empty >>= memoWinchance' atts defs) |
281 | where | ||
282 | memoWinchance' [] _ _ = return 0 -- We do this here so we don't clutter our manual with entries for empty lists | ||
283 | memoWinchance' _ [] _ = return 1 | ||
284 | memoWinchance' atts defs mmRef = recall >>= fromMaybeM (winchance >>= memoize) | ||
285 | 281 | ||
282 | memoWinchance' :: forall s. [Monster] -> [Monster] -> STRef s MM -> ST s Double | ||
283 | memoWinchance' [] _ _ = return 0 -- We do this here so we don't clutter our manual with entries for empty lists | ||
284 | memoWinchance' _ [] _ = return 1 | ||
285 | memoWinchance' atts defs mmRef = recall >>= fromMaybeM (winchance >>= memoize) | ||
286 | where | ||
286 | recall = Map.lookup (atts, defs) <$> readSTRef mmRef | 287 | recall = Map.lookup (atts, defs) <$> readSTRef mmRef |
287 | memoize x = modifySTRef mmRef (Map.insert (atts, defs) x) >> return x | 288 | memoize x = modifySTRef mmRef (Map.insert (atts, defs) x) >> return x |
288 | winchance = winchance' atts defs mmRef | 289 | winchance = winchance' atts defs mmRef |
289 | |||
290 | fromMaybeM x Nothing = x | 290 | fromMaybeM x Nothing = x |
291 | fromMaybeM _ (Just x) = return x | 291 | fromMaybeM _ (Just x) = return x |
292 | 292 | ||
293 | winchance' :: forall s. [Monster] -> [Monster] -> STRef s MM -> ST s Double | ||
293 | winchance' allatts@(a:atts) alldefs@(d:defs) mmRef | 294 | winchance' allatts@(a:atts) alldefs@(d:defs) mmRef |
294 | | hitpoints a <= 0 = memoWinchance' atts alldefs mmRef | 295 | | hitpoints a <= 0 = memoWinchance' atts alldefs mmRef |
295 | | hitpoints d <= 0 = memoWinchance' allatts defs mmRef | 296 | | hitpoints d <= 0 = memoWinchance' allatts defs mmRef |