Start implementing yakus

This commit is contained in:
Sasha 2018-02-01 12:24:07 +01:00
parent 4659f308a2
commit 561df60b57

View file

@ -1,15 +1,57 @@
module KoiKoi where module KoiKoi where
import CCard import CCard
import Data.Bits (popCount, (.&.))
type Points = Int
data Player = Player { data Player = Player {
hand :: Pack hand :: Pack
, captured :: Pack , captured :: Pack
, scored :: [Yaku]
} }
data State = State { data State = State {
players :: (Player, Player) players :: (Player, Player)
, river :: Pack , river :: Pack
, deck :: [ Card ] , deck :: [ Card ]
, month :: Flower
, turn :: Bool , turn :: Bool
} }
data Yaku =
Goko
| Shiko
| AmeShiko
| Sanko
| InoShikaCho
| Tane
| Akatan
| Aotan
| Tan
| Kasu
| TsukimiZake
| HanamiZake
yakus :: [ Pack -> Maybe (Yaku, Points) ]
yakus = [
(\p -> lightsYaku (popCount p) (p `contains` RainMan)) . (lights .&.)
, is inoshikacho InoShikaCho
, moreThan 4 animals Tane
, is poetry Akatan
, is blue Aotan
, moreThan 4 ribbons Tan
, moreThan 9 plain Kasu
, is (set [SakeCup, FullMoon]) TsukimiZake
, is (set [SakeCup, CampCurtain]) HanamiZake
]
where
lightsYaku 5 _ = Just (Goko, 10)
lightsYaku 4 hasRainMan = if hasRainMan then Just (AmeShiko, 7) else Just (Shiko, 8)
lightsYaku n hasRainMan = if not hasRainMan && n > 2 then Just (Sanko, 8) else Nothing
moreThan points set yaku =
threshold yaku . (\x -> x - points) . popCount . (set .&.)
threshold yaku n
| n > 0 = Just (yaku, n)
| otherwise = Nothing
is set yaku pack = if set == pack then Just (yaku, 5) else Nothing