2021_QCB/main.py at main · iQuHACK/2021_QCB · 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
import time
import math
from view import *
from model import *

isGameOver = False
winner = False

def play_game(username, view, model):
    view.display_initial_message(username)
    while not isGameOver:
        move = view.accept_move()
        execute(model, move)
    view.final_message(username, winner)
#    exit(0)

def execute(model, move):
    global isGameOver
    if move == 'show circuit':
        model.draw_current_circuit()
        return
    type_hammer, hole = move[0], move[1]
    if type_hammer == 'c':
        print("Classical Smash!!!")
        isGameOver = True
        winner = classical_smash(model, hole)
    else:
        print('Quantum Smash!!!')
        quantum_smash(model, hole)

def classical_smash(model, hole):
    measured_hole = model.measureState()
    return (measured_hole==[0,0] and hole==1) or (measured_hole==[0,1] and hole==2) or (measured_hole==[1,0] and hole==3) or (measured_hole==[1,1] and hole==4)

def translate_hole(hole):
  if hole==1:
    return [0,0]
  if hole==2:
    return [0,1]
  if hole==3:
    return [1,0]
  if hole==4:
    return [1,1]

def quantum_smash(model, hole):
    #hint is a number or a complex number
    hint = model.getProbabilityOf(translate_hole(hole))
    view.display_quantum_message(hint, hole)
    gate_details = view.accept_quantum_gate()

    if gate_details[0] == 'r' or gate_details[0] == 'R':
        model.add_r_gate(math.radians(float(gate_details[1])), int(gate_details[2]))

    if gate_details[0] == 'cx' or gate_details[0] == 'CX':
        model.add_cnot(int(gate_details[1]), int(gate_details[2]))

    else:
        model.add_unitary(gate_details[0], int(gate_details[2]))

    model.draw_current_circuit()

def initialize_game():
    username = input("What is your name?")
    view = View()
    model = Model()
    return username, view, model

if __name__ == '__main__':
    x = input("Do you want to play Quantum Whack-a-Mole?")
    if x == 'Yes' or x=="yes":
        username, view, model = initialize_game()
        play_game(username, view, model)
    else:
        exit(0)