Mermin–Peres-Game.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Mermin Peres Magic Square Game"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 109,
   "metadata": {},
   "outputs": [],
   "source": [
    "from qiskit import *#\n",
    "import numpy as np\n",
    "from qiskit.test.mock import *"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 110,
   "metadata": {},
   "outputs": [],
   "source": [
    "def check_result(results, col, row):\n",
    "    print_tut = True\n",
    "    for i in results: \n",
    "        \n",
    "        if print_tut:\n",
    "            explain(i, col, row)\n",
    "            print_tut = False\n",
    "\n",
    "        print(\"Quantum circuit solution\")\n",
    "        print(i+\"\\n\")\n",
    "        i = np.array(list(i.split())).astype(int)[::-1]\n",
    "        # bring array in correct i.e. from incoming y2,y1,x2,x1 -> x1,x2,y1,y2\n",
    "        a = i[0:2]\n",
    "        a_sum = np.sum(a)\n",
    "        a3 = int(np.mod(a_sum+1,2))\n",
    "        a = np.append(a,a3)\n",
    "        winning_alice = np.mod(a_sum + a3, 2) == 1\n",
    "        # compute x3 by x1+x2+1 mod 2 and save to array\n",
    "\n",
    "        b = i[2:]\n",
    "        b_sum = np.sum(b)\n",
    "        b3 = int(np.mod(b_sum,2))\n",
    "        b = np.append(b,b3)\n",
    "        winning_bob = np.mod(b_sum + b3, 2) == 0   \n",
    "        # compute y3 by y1+y2 mod 2 and save to array\n",
    "\n",
    "        # compute overall winning condition -> alice at index of bob's row and vice versa\n",
    "        print(\"Alice's selection. Column:\"+str(col)+\"\\n\"+str(a.reshape(3,1))+\"\\n\")\n",
    "        print(\"Bob's circuit solution. Row:\"+str(row)+\"\\n\"+str(b)+\"\\n\")\n",
    "        \n",
    "        if(winning_bob & winning_alice & (a[row-1]==b[col-1])):\n",
    "            print(\"Victory!\")    \n",
    "            print(\"The following magic square wins the game!\")\n",
    "            k = np.empty((3,3,))\n",
    "            k[:] = np.nan\n",
    "            k[:,col-1] = a\n",
    "            k[row-1,:] = b\n",
    "            print(k)\n",
    "        else:\n",
    "            print(\"The Game is lost!\")\n",
    "        print(\"------------------------------------------\")  \n",
    "        print(\"\\n\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 123,
   "metadata": {},
   "outputs": [],
   "source": [
    "def explain(result, col, row):\n",
    "    print(\"Resultant quantum circuit solution:\"+\"\\n\")\n",
    "    print(result+\"\\n\")\n",
    "    example = np.array(list(result.split())).astype(int)[::-1]\n",
    "    print(\"Bring array in correct bit order i.e. from incoming y2,y1,x2,x1 -> x1,x2,y1,y2\"+\"\\n\")\n",
    "    print(str(example)+\"\\n\")\n",
    "    print(\"Compute missing x3 with x1+x2+1 mod 2 for Alice and save to array for further processing\")\n",
    "    print(\"Compute missing y3 with y1+y2 mod 2 for Bob and save to array for further processing\"+\"\\n\")\n",
    "   \n",
    "    a = example[0:2]\n",
    "    a_sum = np.sum(a)\n",
    "    a3 = int(np.mod(a_sum+1,2))\n",
    "    a = np.append(a,a3)\n",
    "    winning_alice = np.mod(a_sum + a3, 2) == 1\n",
    "\n",
    "    b = example[2:]\n",
    "    b_sum = np.sum(b)\n",
    "    b3 = int(np.mod(b_sum,2))\n",
    "    b = np.append(b,b3)\n",
    "    winning_bob = np.mod(b_sum + b3, 2) == 0   \n",
    "\n",
    "    print(\"Alice's column solution\"+\"\\n\"+str(a.reshape(3,1))+\"\\n\" + \"Winning conditions fullfilled (Odd quantitude of 1)?: \"+str(winning_alice)+\"\\n\")\n",
    "    print(\"Bob's row solution\"+\"\\n\"+str(b)+\"\\n\"+ \"Winning conditions fullfilled (Even quantitude of 1)?: \"+str(winning_bob)+\"\\n\")\n",
    "    print(\"Create matrix to show results\"+\"\\n\")\n",
    "    k = np.empty((3,3,))\n",
    "    k[:] = np.nan\n",
    "    k[:,col-1] = a\n",
    "    k[row-1,:] = b\n",
    "    print(k)\n",
    "    print(\"------------------------------------------\")  \n",
    "    print(\"------------------------------------------\")  "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 124,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "def get_state_accuracy(counts): # funciton to calculate state accuracy\n",
    "    expected_counts = 0\n",
    "    for state in counts.keys():\n",
    "        if state in expected_states:\n",
    "            expected_counts = expected_counts + counts[state]\n",
    "    state_accuracy = expected_counts / shots\n",
    "    return str(state_accuracy*100)+\"%\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 125,
   "metadata": {},
   "outputs": [],
   "source": [
    "#The following function sets up the initial entanglement between Alice and Bob\n",
    "def share_bell_state(qc,a,b,c,d): \n",
    "    qc.h(a)\n",
    "    qc.h(b)\n",
    "    qc.cx(a,c)\n",
    "    qc.cx(b,d)\n",
    "    \n",
    "# The following functions represent the U(gamma) and V(gamma) controlled Cliffords.\n",
    "def U(qc,gamma,a,b):\n",
    "    if gamma==1:\n",
    "        qc.h(a)\n",
    "        qc.i(b)\n",
    "    elif gamma==2:\n",
    "        qc.swap(a,b)\n",
    "        qc.h(a)\n",
    "        qc.i(b)\n",
    "\n",
    "    elif gamma==3:\n",
    "        qc.cx(a,b)\n",
    "        qc.h(a)\n",
    "        qc.i(b)\n",
    "\n",
    "def V(qc,gamma,a,b):\n",
    "    if gamma==1:\n",
    "        qc.h(a)\n",
    "        qc.h(b)\n",
    "    if gamma==2:\n",
    "        qc.swap(a,b)\n",
    "    elif gamma==3:\n",
    "        qc.z(a)\n",
    "        qc.z(b)\n",
    "        qc.cz(a,b)\n",
    "        qc.h(a)\n",
    "        qc.h(b)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 127,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Alice and Bob, select a column or row value of 1, 2, or 3\n",
      "Alice choice (column):\n",
      "Bob choice (row):\n",
      "        ┌───┐           ░ ┌───┐ ░ ┌─┐         \n",
      "q534_0: ┤ H ├──■────────░─┤ H ├─░─┤M├─────────\n",
      "        ├───┤  │        ░ ├───┤ ░ └╥┘┌─┐      \n",
      "q534_1: ┤ H ├──┼────■───░─┤ I ├─░──╫─┤M├──────\n",
      "        └───┘┌─┴─┐  │   ░ ├───┤ ░  ║ └╥┘┌─┐   \n",
      "q535_0: ─────┤ X ├──┼───░─┤ H ├─░──╫──╫─┤M├───\n",
      "             └───┘┌─┴─┐ ░ ├───┤ ░  ║  ║ └╥┘┌─┐\n",
      "q535_1: ──────────┤ X ├─░─┤ H ├─░──╫──╫──╫─┤M├\n",
      "                  └───┘ ░ └───┘ ░  ║  ║  ║ └╥┘\n",
      "c160: 1/═══════════════════════════╩══╬══╬══╬═\n",
      "                                   0  ║  ║  ║ \n",
      "                                      ║  ║  ║ \n",
      "c161: 1/══════════════════════════════╩══╬══╬═\n",
      "                                      0  ║  ║ \n",
      "                                         ║  ║ \n",
      "c162: 1/═════════════════════════════════╩══╬═\n",
      "                                         0  ║ \n",
      "                                            ║ \n",
      "c163: 1/════════════════════════════════════╩═\n",
      "                                            0 \n",
      "Resultant quantum circuit solution:\n",
      "\n",
      "0 0 1 0\n",
      "\n",
      "Bring array in correct bit order i.e. from incoming y2,y1,x2,x1 -> x1,x2,y1,y2\n",
      "\n",
      "[0 1 0 0]\n",
      "\n",
      "Compute missing x3 with x1+x2+1 mod 2 and save to array for further processing\n",
      "Compute missing y3 with y1+y2 mod 2 and save to array for further processing\n",
      "\n",
      "Alice's column solution\n",
      "[[0]\n",
      " [1]\n",
      " [0]]\n",
      "Winning conditions fullfilled (Odd quantitude of 1)? :True\n",
      "\n",
      "Bob's row solution\n",
      "[0 0 0]\n",
      "Winning conditions fullfilled (Even quantitude of 1)? :True\n",
      "\n",
      "Create matrix to show results\n",
      "\n",
      "[[ 0.  0.  0.]\n",
      " [ 1. nan nan]\n",
      " [ 0. nan nan]]\n",
      "------------------------------------------\n",
      "------------------------------------------\n",
      "Quantum circuit solution\n",
      "0 0 1 0\n",
      "\n",
      "Alice's selection. Column:1\n",
      "[[0]\n",
      " [1]\n",
      " [0]]\n",
      "\n",
      "Bob's circuit solution. Row:1\n",
      "[0 0 0]\n",
      "\n",
      "Victory!\n",
      "The following magic square wins the game!\n",
      "[[ 0.  0.  0.]\n",
      " [ 1. nan nan]\n",
      " [ 0. nan nan]]\n",
      "------------------------------------------\n",
      "\n",
      "\n",
      "Quantum circuit solution\n",
      "0 1 1 1\n",
      "\n",
      "Alice's selection. Column:1\n",
      "[[1]\n",
      " [1]\n",
      " [1]]\n",
      "\n",
      "Bob's circuit solution. Row:1\n",
      "[1 0 1]\n",
      "\n",
      "Victory!\n",
      "The following magic square wins the game!\n",
      "[[ 1.  0.  1.]\n",
      " [ 1. nan nan]\n",
      " [ 1. nan nan]]\n",
      "------------------------------------------\n",
      "\n",
      "\n",
      "Quantum circuit solution\n",
      "0 1 0 1\n",
      "\n",
      "Alice's selection. Column:1\n",
      "[[1]\n",
      " [0]\n",
      " [0]]\n",
      "\n",
      "Bob's circuit solution. Row:1\n",
      "[1 0 1]\n",
      "\n",
      "Victory!\n",
      "The following magic square wins the game!\n",
      "[[ 1.  0.  1.]\n",
      " [ 0. nan nan]\n",
      " [ 0. nan nan]]\n",
      "------------------------------------------\n",
      "\n",
      "\n",
      "Quantum circuit solution\n",
      "1 0 1 0\n",
      "\n",
      "Alice's selection. Column:1\n",
      "[[0]\n",
      " [1]\n",
      " [0]]\n",
      "\n",
      "Bob's circuit solution. Row:1\n",
      "[0 1 1]\n",
      "\n",
      "Victory!\n",
      "The following magic square wins the game!\n",
      "[[ 0.  1.  1.]\n",
      " [ 1. nan nan]\n",
      " [ 0. nan nan]]\n",
      "------------------------------------------\n",
      "\n",
      "\n",
      "Quantum circuit solution\n",
      "1 1 0 1\n",
      "\n",
      "Alice's selection. Column:1\n",
      "[[1]\n",
      " [0]\n",
      " [0]]\n",
      "\n",
      "Bob's circuit solution. Row:1\n",
      "[1 1 0]\n",
      "\n",
      "Victory!\n",
      "The following magic square wins the game!\n",
      "[[ 1.  1.  0.]\n",
      " [ 0. nan nan]\n",
      " [ 0. nan nan]]\n",
      "------------------------------------------\n",
      "\n",
      "\n",
      "Quantum circuit solution\n",
      "0 0 0 0\n",
      "\n",
      "Alice's selection. Column:1\n",
      "[[0]\n",
      " [0]\n",
      " [1]]\n",
      "\n",
      "Bob's circuit solution. Row:1\n",
      "[0 0 0]\n",
      "\n",
      "Victory!\n",
      "The following magic square wins the game!\n",
      "[[ 0.  0.  0.]\n",
      " [ 0. nan nan]\n",
      " [ 1. nan nan]]\n",
      "------------------------------------------\n",
      "\n",
      "\n",
      "Quantum circuit solution\n",
      "1 1 1 1\n",
      "\n",
      "Alice's selection. Column:1\n",
      "[[1]\n",
      " [1]\n",
      " [1]]\n",
      "\n",
      "Bob's circuit solution. Row:1\n",
      "[1 1 0]\n",
      "\n",
      "Victory!\n",
      "The following magic square wins the game!\n",
      "[[ 1.  1.  0.]\n",
      " [ 1. nan nan]\n",
      " [ 1. nan nan]]\n",
      "------------------------------------------\n",
      "\n",
      "\n",
      "Quantum circuit solution\n",
      "1 0 0 0\n",
      "\n",
      "Alice's selection. Column:1\n",
      "[[0]\n",
      " [0]\n",
      " [1]]\n",
      "\n",
      "Bob's circuit solution. Row:1\n",
      "[0 1 1]\n",
      "\n",
      "Victory!\n",
      "The following magic square wins the game!\n",
      "[[ 0.  1.  1.]\n",
      " [ 0. nan nan]\n",
      " [ 1. nan nan]]\n",
      "------------------------------------------\n",
      "\n",
      "\n",
      "{'0 0 1 0': 1041, '0 1 1 1': 1007, '0 1 0 1': 1033, '1 0 1 0': 1014, '1 1 0 1': 1039, '0 0 0 0': 1036, '1 1 1 1': 1028, '1 0 0 0': 994}\n"
     ]
    }
   ],
   "source": [
    "#We will ask the user to select the row and column on Alice and Bob's behalf.\n",
    "print(\"Alice and Bob, select a column or row value of 1, 2, or 3\") \n",
    "print(\"Alice choice (column):\")\n",
    "alpha = int(input())\n",
    "print(\"Bob choice (row):\")\n",
    "beta = int(input())\n",
    "\n",
    "#Create the quantum register and the classical register to store our final bit values\n",
    "aliceQR = QuantumRegister(2)\n",
    "x1CR = ClassicalRegister(1)\n",
    "x2CR = ClassicalRegister(1)\n",
    "bobQR = QuantumRegister(2)\n",
    "y1CR = ClassicalRegister(1)\n",
    "y2CR = ClassicalRegister(1)\n",
    "\n",
    "#Create the circuit\n",
    "magicsquare_circuit = QuantumCircuit(aliceQR,bobQR,x1CR,x2CR,y1CR,y2CR)\n",
    "\n",
    "#Generate the Bell state on the circuit\n",
    "share_bell_state(magicsquare_circuit,0,1,2,3)\n",
    "\n",
    "magicsquare_circuit.barrier()\n",
    "\n",
    "#Draw the rest of the circuit based on Alice and Bob's selection   \n",
    "if 4>alpha>0 and 4>beta>0:\n",
    "    U(magicsquare_circuit,alpha,0,1) \n",
    "    V(magicsquare_circuit,beta,2,3)\n",
    "    magicsquare_circuit.barrier()\n",
    "    magicsquare_circuit.measure(0,0)\n",
    "    magicsquare_circuit.measure(1,1)\n",
    "    magicsquare_circuit.measure(2,2)\n",
    "    magicsquare_circuit.measure(3,3)\n",
    "    \n",
    "    magicsquare_circuit.draw(output='mpl')\n",
    "    print(magicsquare_circuit)\n",
    "\n",
    "    backend = BasicAer.get_backend('qasm_simulator') # define the backend\n",
    "    job = execute(magicsquare_circuit, backend, shots=8192) # run the job simulation\n",
    "    results = job.result().get_counts()\n",
    "    check_result(results, alpha, beta)\n",
    "    print(results)\n",
    "\n",
    "else:\n",
    "    print(\"please enter values of 1, 2, or 3\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}