From 98bbe024957d24488e6d9bf14ce1a6cc80296ab0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maximilian=20Ke=C3=9Fler?= <git@maximilian-kessler.de>
Date: Sat, 8 Jul 2023 09:47:10 +0200
Subject: [PATCH] improve on game: add todo and provide list of dark suits

---
 hanabi/hanab_game.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/hanabi/hanab_game.py b/hanabi/hanab_game.py
index 9ea8bba..08d00f8 100644
--- a/hanabi/hanab_game.py
+++ b/hanabi/hanab_game.py
@@ -105,6 +105,10 @@ class Action:
 
 
 class HanabiInstance:
+    # TODO Max: Deal with the following variants:
+    # - Critical fours (need to calculate dark suits differently)
+    # - Reversed (need to store information somehow and pass this to the hanabi game class)
+    # - Up or Down (in the long run we also want this, but seems a bit tedious, not needed now)
     def __init__(
             self,
             deck: List[DeckCard],
@@ -170,6 +174,10 @@ class HanabiInstance:
     def clue_increment(self):
         return 0.5 if self.clue_starved else 1
 
+    @property
+    def dark_suits(self):
+        return list(range(self.num_suits - self.num_dark_suits, self.num_suits))
+
 
 class GameState:
     def __init__(self, instance: HanabiInstance):