card.py

#!/usr/bin/python
#-*- coding: utf-8 -*-
from suit import *
from bcolor import *


class Card:
    """
    Card for the Hanabi game.
    """

    def __init__(self, suit=Suit.unknown, value=0):
        self.value = value
        self.suit = suit

    def getSuit(self):
        return self.suit

    @classmethod
    def getSuitFromInt(cardID):
        return Suit(int((cardID - 1) / 5) + 1)

    def getValue(self):
        return self.value

    @classmethod
    def getValueFromInt(cls, cardID):
        return ((cardID - 1) % 5) + 1

    def setSuit(self, suit):
        self.suit = suit

    def setValue(self, value):
        self.value = value

    def toString(self):
        return Suit.toColor(self.suit) + " " + Suit.toString(self.suit) + " " + str(self.value) + Bcolor.END

    def toBinary(self):
        return pad([int(i) for i in str(bin(Suit.toInt(self.getSuit())))[2:]], 3) + pad([int(i) for i in str(bin(self.getValue()))[2:]], 3)

    def __str__(self):
        return self.toString()

    def __int__(self):
        """ overload of int() for easy value comparisons
            1 <= Suit.toInt(self.suit) <= 6
            0 <= (Suit.toInt(self.suit) - 1) <= 5
            0 <= (Suit.toInt(self.suit) - 1) * 5 <= 25
            1 <= (Suit.toInt(self.suit) - 1) * 5 + self.value <= 30
            this stretches our nColors*nValues = 6*5 = 30 different types
            of cards over 30 different values, which allows us to order them
        """
        return (Suit.toInt(self.suit) - 1) * 5 + self.value

    def __lt__(self, other):
        """ Overload of the < operator for sorting
        """
        return int(self) < int(other)

    def __le__(self, other):
        """ Overload of the <= operator
        """
        return int(self) <= int(other)

    def __gt__(self, other):
        """ Overload of the > operator
        """
        return int(self) > int(other)

    def __ge__(self, other):
        """ Overload of the >= operator
        """
        return int(self) >= int(other)

    def __eq__(self, other):
        """ Overload of the == operator
        """
        return int(self) == int(other)

    def __ne__(self, other):
        """ Overload of the != operator
        """
        return int(self) != int(other)


def pad(x, size):
    while len(x) < size:
        x = [0] + x
    return x