new project

1 files changed, 140 insertions, 0 deletions

diff --git a/deck.py b/deck.py
new file mode 100644
index 0000000..21039f1
--- /dev/null
+++ b/deck.py
@@ -0,0 +1,140 @@
+#Copyright (c) 2009-11 Walter Bender
+
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the
+# Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+# Boston, MA 02111-1307, USA.
+
+import gtk
+from random import randrange
+
+from card import Card
+from genpieces import generate_tile_1_line, generate_tile_2_lines, \
+    generate_board
+
+
+class Deck:
+    ''' Class for defining deck of card '''
+
+    def __init__(self, sprites, scale):
+        ''' Create the deck of cards. '''
+        self.cards = []
+        for a in range(16):
+            self.cards.append(Card(sprites, generate_tile_1_line(
+                        -1, 0, 0, 0, scale)))
+            self.cards[-1].set_connections([0, 0, 0, 1])
+        for a in range(4):
+            self.cards.append(Card(sprites, generate_tile_1_line(
+                        -1, 0, 1, 0, scale)))
+            self.cards[-1].set_connections([0, 1, 0, 1])
+        for a in range(12):
+            self.cards.append(Card(sprites, generate_tile_2_lines(
+                        -1, 0, 1, 0, 0, 0, 0, 1, scale)))
+            self.cards[-1].set_connections([0, 1, 1, 1])
+        for a in range(16):
+            self.cards.append(Card(sprites, generate_tile_2_lines(
+                        -1, 0, 0, 0, 0, -1, 0, 0, scale)))
+            self.cards[-1].set_connections([1, 0, 0, 1])
+        for a in range(4):
+            self.cards.append(Card(sprites, generate_tile_2_lines(
+                        -1, 0, 1, 0, 0, -1, 0, 1, scale)))
+            self.cards[-1].set_connections([1, 1, 1, 1])
+        for a in range(8):
+            self.cards.append(Card(sprites, generate_tile_2_lines(
+                        -1, 0, 0, 1, 0, -1, 1, 0, scale)))
+            self.cards[-1].set_connections([1, 1, 1, 1])
+        for a in range(4):
+            self.cards.append(Card(sprites, generate_tile_2_lines(
+                        -1, 0, 0, 0, 0, -1, 1, 0, scale)))
+            self.cards[-1].set_connections([1, 1, 0, 1])
+        # Remember the current position in the deck.
+        self.index = 0
+
+        # And a playing surface
+        self.board = Card(sprites, generate_board(scale), card_type='board')
+        self.board.spr.set_layer(1)
+
+    def shuffle(self):
+        ''' Shuffle the deck (Knuth algorithm). '''
+        decksize = self.count()
+        # Hide all the cards.
+        for c in self.cards:
+            c.hide_card()
+        # Randomize the card order.
+        for n in range(decksize):
+            i = randrange(decksize - n)
+            self.swap_cards(n, decksize - 1 - i)            
+        # Reset the index to the beginning of the deck after a shuffle,
+        self.index = 0
+        self.hide()
+        return
+
+    def restore(self, saved_deck_indices):
+        ''' Restore the deck upon resume. '''
+        decksize = len(saved_deck_indices)
+        # If we have a short deck, then we need to abort.
+        if self.count() < decksize:
+            return False
+        _deck = []
+        for i in saved_deck_indices:
+             _deck.append(self.index_to_card(i))
+        for i in range(decksize):
+             self.cards[i] = _deck[i]
+        return True
+
+    def swap_cards(self,i,j):
+        ''' Swap the position of two cards in the deck. '''
+        tmp = self.cards[j]
+        self.cards[j] = self.cards[i]
+        self.cards[i] = tmp
+        return
+
+    def spr_to_card(self, spr):
+        ''' Given a sprite, find the corresponding card in the deck. '''
+        if spr == self.board.spr:
+            return self.board
+        for c in self.cards:
+            if c.spr == spr:
+                return c
+        return None
+
+    def index_to_card(self, i):
+        ''' Given a card index, find the corresponding card in the deck. '''
+        for c in self.cards:
+            if c.index == i:
+                return c
+        return None
+
+    def deal_next_card(self):
+        ''' Return the next card from the deck. '''
+        if self.empty():
+            return None
+        next_card = self.cards[self.index]
+        self.index += 1
+        return next_card
+ 
+    def empty(self):
+        ''' Is the deck empty? '''
+        if self.cards_remaining() > 0:
+            return False
+        else:
+            return True
+
+    def cards_remaining(self):
+        ''' Return how many cards are remaining in the deck. '''
+        return(self.count()-self.index)
+
+    def hide(self):
+        ''' Hide the deck. '''
+        for c in self.cards:
+            if c is not None:
+                c.hide_card()
+
+    def count(self):
+        ''' Return the length of the deck. '''
+        return len(self.cards)