summaryrefslogtreecommitdiff
path: root/src/bcdd/ParScore.py
blob: e23a83910bb55fd4c2a75b501ab9c5ca1371cd86 (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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
import re

from .BCalcWrapper import BCalcWrapper as bcw
from .ParContract import ParContract
from .Exceptions import FieldNotFoundException


class ParScore(object):
    _pbn_contract_pattern = re.compile(r'(\d)([CDHSN])(X?)\s+([NESW])')
    _pbn_score_pattern = re.compile(r'(NS|EW)\s+(-?\d})')
    _jfr_contract_pattern = re.compile(r'^(\d)([CDHSN])(D?)([NESW])(-?\d+)$')

    def __init__(self, board):
        self._board = board

    def get_pbn_par_contract(self):
        contract_field = self._board.get_optimum_result()
        if 'Pass' == contract_field:
            return ParContract()
        contract_match = self._pbn_contract_pattern.match(contract_field)
        if not contract_match:
            raise ParScoreInvalidException(
                'Invalid format for OptimumResult field: ' + contract_field)
        score_field = self._board.get_optimum_score()
        score_match = self._pbn_score_pattern.match(score_field)
        if not score_match:
            raise ParScoreInvalidException(
                'Invalid format for OptimumScore field: ' + scoreField)
        score = int(score_match.group(2))
        if 'EW' == score_match.group(1):
            score = -score
        contract = ParContract(
            int(contract_match.group(1)),
            contract_match.group(2)[0],
            contract_match.group(4)[0],
            'X' == contract_match.group(3),
            score)
        return contract.validate()

    def get_jfr_par_contract(self):
        par_string = self._board.get_minimax()
        par_match = self._jfr_contract_pattern.match(par_string)
        if not par_match:
            raise ParScoreInvalidException(
                'Invalid format for Minimax field: ' + par_string)
        if '0' == par_match.group(4):
            return ParContract() # pass-out
        contract = ParContract(
            int(par_match.group(1)),
            par_match.group(2)[0],
            par_match.group(4)[0],
            'D' == par_match.group(3),
            int(par_match.group(5)))
        return contract.validate()

    def _determine_vulnerability(self, vulnerability, declarer):
        vulnerability = vulnerability.upper()
        return vulnerability in ['ALL', 'BOTH'] \
            or (vulnerability not in ['LOVE', 'NONE'] \
                and declarer in vulnerability)

    def _get_highest_makeable_contract(self, dd_table,
                                       for_ns=True, for_ew=True):
        contract = ParContract()
        tricks = 0
        for i in range(3, -1, -1):
            if ((i % 2 == 0) and for_ns) \
               or ((i % 2 == 1) and for_ew):
                for j in range(0, 5):
                    level = dd_table[i][j] - 6
                    denomination = bcw.DENOMINATIONS.index(
                        contract.denomination) \
                        if contract.denomination in bcw.DENOMINATIONS \
                           else -1
                    if (level > contract.level) \
                       or ((level == contract.level) \
                           and (j > denomination)):
                        contract.level = level
                        contract.denomination = bcw.DENOMINATIONS[j]
                        contract.declarer = bcw.PLAYERS[i]
                        tricks = dd_table[i][j]
        vulnerability = self._board.get_vulnerable().upper()
        vulnerable = self._determine_vulnerability(
            vulnerability, contract.declarer)
        contract.score = contract.calculate_score(tricks, vulnerable)
        return contract

    def get_dd_table_par_contract(self, dd_table):
        dealer = self._board.get_dealer()
        vulnerability = self._board.get_vulnerable().upper()
        ns_highest = self._get_highest_makeable_contract(
            dd_table, True, False);
        ew_highest = self._get_highest_makeable_contract(
            dd_table, False, True)
        if ns_highest == ew_highest:
            return ns_highest.validate() \
                if dealer in ['N', 'S'] else ew_highest.validate()
        highest = max(ns_highest, ew_highest)
        other_side_highest = min(ew_highest, ns_highest)
        ns_playing = highest.declarer in ['N', 'S']
        defense_vulnerability = self._determine_vulnerability(
            vulnerability, 'E' if ns_playing else 'N')
        highest_defense = highest.get_defense(dd_table, defense_vulnerability)
        if highest_defense is not None:
            # Highest contract has profitable defense
            return highest_defense.validate()
        denomination_index = bcw.DENOMINATIONS.index(highest.denomination) \
                             if highest.denomination in bcw.DENOMINATIONS \
                                else -1
        declarer_index = bcw.PLAYERS.index(highest.declarer) \
                         if highest.declarer in bcw.PLAYERS else -1
        player_indexes = [declarer_index, (declarer_index + 2) % 4]
        vulnerable = self._determine_vulnerability(
            vulnerability, highest.declarer)
        score_squared = highest.score * highest.score
        possible_optimums = []
        for i in range(0, 5):
            for player in player_indexes:
                level = highest.level
                if i > denomination_index:
                    level -= 1
                while level > 0:
                    contract = ParContract(
                        level,
                        bcw.DENOMINATIONS[i],
                        bcw.PLAYERS[player],
                        False, 0)
                    contract.score = contract.calculate_score(
                        dd_table[player][i], vulnerable)
                    if other_side_highest > contract:
                        # Contract is lower than other side's contract
                        break
                    if (highest.score * contract.score) > 0:
                        # Contract makes
                        if abs(contract.score) >= abs(highest.score):
                            # Contract is profitable
                            defense = contract.get_defense(
                                dd_table, defense_vulnerability)
                            if defense is not None \
                               and (contract.score * contract.score) \
                               > (contract.score * defense.score):
                                # Contract has defense
                                possible_optimums.append(defense)
                                # So lower contracts will too.
                                break
                            else:
                                # Contract does not have defense
                                possible_optimums.append(contract)
                        else:
                            # Contract is not profitable
                            break
                    level -= 1
        for contract in possible_optimums:
            if abs(contract.score) > abs(highest.score):
                # Contract is more profitable
                highest = contract
            else:
                if contract.score == highest.score:
                    if highest > contract:
                        # Equally profitable, but lower
                        highest = contract
        return highest.validate()

    def get_par_contract(self, dd_table):
        try:
            return self.get_jfr_par_contract()
        except FieldNotFoundException:
            try:
                return self.get_pbn_par_contract()
            except FieldNotFoundException:
                return self.get_dd_table_par_contract(dd_table)