summaryrefslogtreecommitdiff
path: root/dealconvert/formats/rzd.py
blob: 55dade231636c9b10cd1fa1461036c960260694d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import sys

from . import DealFormat
from .. import dto

class RZDFormat(DealFormat):
    number_warning = 'WARNING: .rzd file format assumes consequent deal numbers from 1'

    @property
    def suffix(self):
        return '.rzd'

    def parse_content(self, content):
        dealset = []
        header = None
        number = 1
        while True:
            data = content.read(13)
            if len(data) < 13:
                if len(data) != 0:
                    print 'WARNING: .rzd data truncated: %s' % (data)
                break
            if header is None:
                header = data
                continue
            deal = dto.Deal()
            deal.number = number
            deal.dealer = deal.get_dealer(number)
            deal.vulnerable = deal.get_vulnerability(number)
            for card, byte in enumerate(data):
                byte = ord(byte)
                for suit in range(3, -1, -1):
                    deal.hands[byte%4][suit].append(self.cards[card])
                    byte /= 4
            dealset.append(deal)
            number += 1
        return dealset

    def output_content(self, out_file, dealset):
        board_count = len(dealset)
        out_file.write(chr(board_count%256))
        out_file.write(chr(board_count/256))
        out_file.write(' '*11)
        values = [None] * 52
        for deal in dealset:
            for i, hand in enumerate(deal.hands):
                for suit, cards in enumerate(hand):
                    for card in cards:
                        try:
                            idx = self.cards.index(card)
                        except ValueError:
                            print 'ERROR: invalid card: %s' % (card)
                            sys.exit()
                        values[idx*4+suit] = i
            for i in range(0, 13):
                byte = 0
                for j in range(0, 4):
                    byte *= 4
                    byte += values[4*i+j]
                out_file.write(chr(byte))