Dudo/dudo.py at master · machocam/Dudo · GitHub

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
from __future__ import division
from operator import itemgetter
import math


all_cards_ranked = {
    "aces" : 5,
    "kings" : 4,
    "queens" : 3,
    "jacks" : 2,
    "10s" : 1,
    "9s" : 0
}

my_game = {
        "aces" : 1,
        "kings" : 2,
        "queens" : 2,
        "jacks" : 0,
        "10s" : 0,
        "9s" : 1
        }


def calculate_chances_call (num_players, fijos, call,):
    chances = 0.0
    num_dice = (num_players - 1) * 5
    if fijos:
        wanted_num_card = call[1] - my_game[call[0]]
    else:
        wanted_num_card = call[1] - (my_game[call[0]] + my_game["aces"])


    if wanted_num_card <= 0:
        chances += 1
    else:
        if call [0] == "aces" or fijos:   #If we are fijos then getting a card is same as ace
            for num in list(range(num_dice + 1))[call[1] - my_game[call[0]]:]:
                chances += combi(num_dice, num) * (1/6)**num * (5/6)**(num_dice - num) #This is the binomial formula of probability
        else :
            for num in list(range(num_dice + 1))[wanted_num_card:]:
                chances += combi(num_dice, num) * (1/3)**num * (2/3)**(num_dice - num) #This is the binomial formula of probability
    return chances

def give_options (fijo, fijos, call):
    options = []
    if fijos and not fijo:
        options.append([call[0], call[1] + 1])
    elif fijos and fijo:
        for card in all_cards_ranked:
            options.append([card, call[1] + 1])
    else:
        if call[0] == "aces":
            options.append([call[0], call[1] + 1])
            for card in all_cards_ranked:
                options.append([card, call[1] * 2]) #here we are going to have aces twice becuase of the statement earlier. We can ignore this because the chances of the "wrong" aces winning are none so it will be ignored.
        else:
            options.append(["aces", int(round((call[1]+1)/2 ,0))]) #include half of total for aces taking into account round up option.
            for card in all_cards_ranked:
                if all_cards_ranked[call[0]] < all_cards_ranked[card]: #take into account card rank
                    options.append([card, call[1]])
                else:
                    options.append([card, call[1] + 1])

    return options

def give_chances_options (fijo, fijos, call, num_players):
    chances = []
    options = give_options (fijo, fijos, call)
    for option in options:
        chances.append([option, calculate_chances_call(num_players, fijos, option)])
    chances.append(["dudo", 1.0 - calculate_chances_call(num_players, fijos, call)])
    return sorted(chances, key = itemgetter(1), reverse = True)


def play_dudo (fijo, fijos, call, num_players):
    chances = give_chances_options (fijo, fijos, call, num_players)
    for item in chances:
        item[1] = str(round(item[1]*100, 2)) + "%"
        print item
def combi (n, k):
    return math.factorial(n)/(math.factorial(k)*math.factorial(n-k))


play_dudo (False, False, ["queens", 2], 3)
#print calculate_chances_call(4, False, ["queens", 7])