GATE/game7.py at main · learningdungeon/GATE · 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
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
import random
from qiskit import QuantumCircuit
from qiskit.quantum_info import Statevector

# --- CORE ENGINES ---

def run_guided_engine(num_ions, target_state, player_moves):
    """Guided Mode: Automatically handles H-gates and Diffuser."""
    qc = QuantumCircuit(num_ions)
    qc.h(range(num_ions))
    for move in player_moves:
        if move.startswith('X'):
            idx = int(move[1:])
            qc.x(idx)
        if move == 'LOCK':
            if num_ions == 2: qc.cz(0, 1)
            elif num_ions == 3: qc.ccz(0, 1, 2)
            else:
                qc.h(num_ions-1); qc.mcx(list(range(num_ions-1)), num_ions-1); qc.h(num_ions-1)
    # Automatic Diffuser
    qc.h(range(num_ions)); qc.x(range(num_ions))
    qc.h(num_ions-1); qc.mcx(list(range(num_ions-1)), num_ions-1); qc.h(num_ions-1)
    qc.x(range(num_ions)); qc.h(range(num_ions))

    state = Statevector.from_instruction(qc)
    return state.probabilities_dict(), qc

def run_scientist_engine(num_ions, player_moves):
    """Scientist Mode: Pure manual control of every gate."""
    qc = QuantumCircuit(num_ions)
    for move in player_moves:
        if move == 'H_ALL': qc.h(range(num_ions))
        elif move.startswith('X'):
            idx = int(move[1:])
            qc.x(idx)
        elif move == 'LOCK':
            if num_ions == 2: qc.cz(0, 1)
            elif num_ions == 3: qc.ccz(0, 1, 2)
            else:
                qc.h(num_ions-1); qc.mcx(list(range(num_ions-1)), num_ions-1); qc.h(num_ions-1)
        elif move == 'DIFFUSE':
            qc.h(range(num_ions)); qc.x(range(num_ions))
            qc.h(num_ions-1); qc.mcx(list(range(num_ions-1)), num_ions-1); qc.h(num_ions-1)
            qc.x(range(num_ions)); qc.h(range(num_ions))

    state = Statevector.from_instruction(qc)
    return state.probabilities_dict(), qc

# --- VISUALS ---

def display_trap(num_ions, moves):
    print("\n--- ❄️  VACUUM CHAMBER STATUS ---")
    ions = []
    for i in range(num_ions):
        x_count = sum(1 for m in moves if m == f"X{i}")
        state = "|1>" if x_count % 2 != 0 else "|0>"
        ions.append(f"( Ion_{i}:{state} )")
    print("  " + " <--> ".join(ions))
    print("---------------------------------")

# --- GAME MODES ---

def play_guided():
    ions = 2
    while ions <= 4:
        target = format(random.randint(0, (2**ions)-1), f'0{ions}b')
        threshold = 0.95 if ions == 2 else 0.75 if ions == 3 else 0.45
        moves = []
        print(f"\n🚀 GUIDED MISSION: Isolate |{target}>")
        while True:
            display_trap(ions, moves)
            cmd = input(f"CONTROL (X0-{ions-1}, LOCK, RUN, CLEAR) > ").upper().strip()
            if cmd == 'RUN':
                probs, _ = run_guided_engine(ions, target, moves)
                win_prob = probs.get(target[::-1], 0)
                if win_prob >= threshold:
                    print(f"✨ SUCCESS! Probability: {win_prob*100:.1f}%"); ions += 1; break
                else: print(f"❌ WEAK SIGNAL: {win_prob*100:.1f}%")
            elif cmd == 'CLEAR': moves = []
            elif cmd in [f"X{i}" for i in range(ions)] + ["LOCK"]: moves.append(cmd)

def play_scientist():
    ions = 3
    target = format(random.randint(0, (2**ions)-1), f'0{ions}b')
    moves = []
    print(f"\n🧪 SCIENTIST MODE: Manual Build for |{target}>")
    print("Sequence required: H_ALL -> X-Sandwich -> LOCK -> X-Sandwich -> DIFFUSE")
    while True:
        cmd = input("ION_COMMAND (H_ALL, X0-2, LOCK, DIFFUSE, RUN, CLEAR) > ").upper().strip()
        if cmd == 'RUN':
            probs, qc = run_scientist_engine(ions, moves)
            print(qc.draw(output='text'))
            win_prob = probs.get(target[::-1], 0)
            if win_prob > 0.7: print("🏆 MISSION ACCOMPLISHED!"); break
            else: print(f"⚠️  SIGNAL WEAK ({win_prob*100:.1f}%)")
        elif cmd == 'CLEAR': moves = []
        else: moves.append(cmd)

# --- MAIN INTERFACE ---

def main():
    while True:
        print("\n" + "="*40)
        print("⚛️  OPEN QUANTUM DESIGN: ION MAZE  ⚛️")
        print("="*40)
        print("1. Beginner Mode (Guided Oracle)")
        print("2. Scientist Mode (Manual Algorithm)")
        print("3. Exit")
        choice = input("\nSelect your level (1-3): ")
        if choice == '1': play_guided()
        elif choice == '2': play_scientist()
        elif choice == '3': break

if __name__ == "__main__":
    main()