Encoded-Copy.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Qconfig loaded from /Users/Atsushi/Documents/DOMath/Jupyter/quantum_rb/Qconfig.py.\n",
      "['508504678146834637680148014804676805167802768450467680563780258451467681276836518', '54716806805827804547804582586780128023481367680460547824504802578605308', '25048136781246768450678047683780454781345467845476845781450845048715805476830658', '3783607834637681345768245787126845768051768124678036768135478136768154780467687638', '78514676845484560768045168027834638638157815467845482567681567834638430208', '54726815046768454676814504678371805768451476845783468156815046768354768456078268028', '15047681238054863787680467826825048038057183678134678026824578753068', '06768025676813768347280637680454637683548451480246854808134546768245676814504676815178', '0637681504678045148054676813481460788354676804514768268384576834676802476864158']\n",
      "15 0\n",
      "\n",
      "got\n",
      "Status @ 0 seconds\n",
      "{'job_id': None, 'status': <JobStatus.INITIALIZING: 'job is being initialized'>, 'status_msg': 'Job is initializing. Please, wait a moment.'}\n",
      "Status @ 120 seconds\n",
      "{'job_id': '5b4a0c691deae900384940a7', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 21}\n",
      "Status @ 240 seconds\n",
      "{'job_id': '5b4a0c691deae900384940a7', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 21}\n",
      "Status @ 360 seconds\n",
      "{'job_id': '5b4a0c691deae900384940a7', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 21}\n",
      "Status @ 480 seconds\n",
      "{'job_id': '5b4a0c691deae900384940a7', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 21}\n",
      "Status @ 600 seconds\n",
      "{'job_id': '5b4a0c691deae900384940a7', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 18}\n",
      "Status @ 720 seconds\n",
      "{'job_id': '5b4a0c691deae900384940a7', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 9}\n",
      "{'job_id': '5b4a0c691deae900384940a7', 'status': <JobStatus.DONE: 'job has successfully run'>, 'status_msg': 'job has successfully run'}\n",
      "here\n",
      "nice\n",
      "0000: 366\n",
      "1111: 312\n",
      "1010: 2\n",
      "0101: 7\n",
      "0011: 19\n",
      "1100: 37\n",
      "1001: 7\n",
      "0110: 8\n",
      "1000: 13\n",
      "0111: 54\n",
      "0001: 15\n",
      "1110: 68\n",
      "0010: 19\n",
      "1101: 48\n",
      "0100: 16\n",
      "1011: 33\n",
      "{'0000': 366, '0001': 15, '0010': 19, '0011': 19, '0100': 16, '0101': 7, '0110': 8, '0111': 54, '1000': 13, '1001': 7, '1010': 2, '1011': 33, '1100': 37, '1101': 48, '1110': 68, '1111': 312}\n",
      "00\n",
      "0.8944591029023746\n",
      "got\n",
      "Status @ 0 seconds\n",
      "{'job_id': None, 'status': <JobStatus.INITIALIZING: 'job is being initialized'>, 'status_msg': 'Job is initializing. Please, wait a moment.'}\n",
      "Status @ 120 seconds\n",
      "{'job_id': '5b4a0fb8fc5378003bf3b865', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 45}\n",
      "Status @ 240 seconds\n",
      "{'job_id': '5b4a0fb8fc5378003bf3b865', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 18}\n",
      "Status @ 360 seconds\n",
      "{'job_id': '5b4a0fb8fc5378003bf3b865', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 18}\n",
      "Status @ 480 seconds\n",
      "{'job_id': '5b4a0fb8fc5378003bf3b865', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 3}\n",
      "{'job_id': '5b4a0fb8fc5378003bf3b865', 'status': <JobStatus.DONE: 'job has successfully run'>, 'status_msg': 'job has successfully run'}\n",
      "here\n",
      "nice\n",
      "0000: 408\n",
      "1111: 315\n",
      "1010: 2\n",
      "0101: 6\n",
      "0011: 19\n",
      "1100: 29\n",
      "1001: 2\n",
      "0110: 12\n",
      "1000: 17\n",
      "0111: 51\n",
      "0001: 18\n",
      "1110: 50\n",
      "0010: 15\n",
      "1101: 49\n",
      "0100: 11\n",
      "1011: 20\n",
      "{'0000': 408, '0001': 18, '0010': 15, '0011': 19, '0100': 11, '0101': 6, '0110': 12, '0111': 51, '1000': 17, '1001': 2, '1010': 2, '1011': 20, '1100': 29, '1101': 49, '1110': 50, '1111': 315}\n",
      "00\n",
      "0.9117276166456494\n",
      "got\n",
      "Status @ 0 seconds\n",
      "{'job_id': None, 'status': <JobStatus.INITIALIZING: 'job is being initialized'>, 'status_msg': 'Job is initializing. Please, wait a moment.'}\n",
      "Status @ 120 seconds\n",
      "{'job_id': '5b4a2020a2e0a8003dc5cf1a', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 12}\n",
      "Status @ 240 seconds\n",
      "{'job_id': '5b4a2020a2e0a8003dc5cf1a', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 3}\n",
      "{'job_id': '5b4a2020a2e0a8003dc5cf1a', 'status': <JobStatus.DONE: 'job has successfully run'>, 'status_msg': 'job has successfully run'}\n",
      "here\n",
      "nice\n",
      "0000: 402\n",
      "1111: 309\n",
      "1010: 6\n",
      "0101: 9\n",
      "0011: 22\n",
      "1100: 27\n",
      "1001: 5\n",
      "0110: 9\n",
      "1000: 32\n",
      "0111: 49\n",
      "0001: 30\n",
      "1110: 34\n",
      "0010: 24\n",
      "1101: 39\n",
      "0100: 10\n",
      "1011: 17\n",
      "{'0000': 402, '0001': 30, '0010': 24, '0011': 22, '0100': 10, '0101': 9, '0110': 9, '0111': 49, '1000': 32, '1001': 5, '1010': 6, '1011': 17, '1100': 27, '1101': 39, '1110': 34, '1111': 309}\n",
      "00\n",
      "0.9011406844106464\n",
      "got\n",
      "Status @ 0 seconds\n",
      "{'job_id': None, 'status': <JobStatus.INITIALIZING: 'job is being initialized'>, 'status_msg': 'Job is initializing. Please, wait a moment.'}\n",
      "Status @ 120 seconds\n",
      "{'job_id': '5b4a21921deae9003849417e', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 6}\n",
      "{'job_id': '5b4a21921deae9003849417e', 'status': <JobStatus.DONE: 'job has successfully run'>, 'status_msg': 'job has successfully run'}\n",
      "here\n",
      "nice\n",
      "0000: 408\n",
      "1111: 317\n",
      "1010: 4\n",
      "0101: 7\n",
      "0011: 21\n",
      "1100: 28\n",
      "1001: 3\n",
      "0110: 12\n",
      "1000: 16\n",
      "0111: 55\n",
      "0001: 13\n",
      "1110: 52\n",
      "0010: 18\n",
      "1101: 39\n",
      "0100: 11\n",
      "1011: 20\n",
      "{'0000': 408, '0001': 13, '0010': 18, '0011': 21, '0100': 11, '0101': 7, '0110': 12, '0111': 55, '1000': 16, '1001': 3, '1010': 4, '1011': 20, '1100': 28, '1101': 39, '1110': 52, '1111': 317}\n",
      "00\n",
      "0.90625\n",
      "got\n",
      "Status @ 0 seconds\n",
      "{'job_id': None, 'status': <JobStatus.INITIALIZING: 'job is being initialized'>, 'status_msg': 'Job is initializing. Please, wait a moment.'}\n",
      "Status @ 120 seconds\n",
      "{'job_id': '5b4a227ffc5378003bf3b8cd', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 6}\n",
      "{'job_id': '5b4a227ffc5378003bf3b8cd', 'status': <JobStatus.DONE: 'job has successfully run'>, 'status_msg': 'job has successfully run'}\n",
      "here\n",
      "nice\n",
      "0000: 419\n",
      "1111: 309\n",
      "1010: 2\n",
      "0101: 6\n",
      "0011: 28\n",
      "1100: 31\n",
      "1001: 7\n",
      "0110: 9\n",
      "1000: 21\n",
      "0111: 47\n",
      "0001: 41\n",
      "1110: 32\n",
      "0010: 25\n",
      "1101: 23\n",
      "0100: 14\n",
      "1011: 10\n",
      "{'0000': 419, '0001': 41, '0010': 25, '0011': 28, '0100': 14, '0101': 6, '0110': 9, '0111': 47, '1000': 21, '1001': 7, '1010': 2, '1011': 10, '1100': 31, '1101': 23, '1110': 32, '1111': 309}\n",
      "00\n",
      "0.8976572133168927\n",
      "got\n",
      "Status @ 0 seconds\n",
      "{'job_id': None, 'status': <JobStatus.INITIALIZING: 'job is being initialized'>, 'status_msg': 'Job is initializing. Please, wait a moment.'}\n",
      "Status @ 120 seconds\n",
      "{'job_id': '5b4a2370a2e0a8003dc5cf29', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 6}\n",
      "{'job_id': '5b4a2370a2e0a8003dc5cf29', 'status': <JobStatus.DONE: 'job has successfully run'>, 'status_msg': 'job has successfully run'}\n",
      "here\n",
      "nice\n",
      "0000: 394\n",
      "1111: 294\n",
      "1010: 7\n",
      "0101: 12\n",
      "0011: 24\n",
      "1100: 40\n",
      "1001: 5\n",
      "0110: 3\n",
      "1000: 19\n",
      "0111: 35\n",
      "0001: 22\n",
      "1110: 47\n",
      "0010: 23\n",
      "1101: 57\n",
      "0100: 12\n",
      "1011: 30\n",
      "{'0000': 394, '0001': 22, '0010': 23, '0011': 24, '0100': 12, '0101': 12, '0110': 3, '0111': 35, '1000': 19, '1001': 5, '1010': 7, '1011': 30, '1100': 40, '1101': 57, '1110': 47, '1111': 294}\n",
      "00\n",
      "0.883183568677792\n",
      "got\n",
      "Status @ 0 seconds\n",
      "{'job_id': None, 'status': <JobStatus.INITIALIZING: 'job is being initialized'>, 'status_msg': 'Job is initializing. Please, wait a moment.'}\n",
      "Status @ 120 seconds\n",
      "{'job_id': '5b4a2463e036a500391bdbfc', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 3}\n",
      "Status @ 240 seconds\n",
      "{'job_id': '5b4a2463e036a500391bdbfc', 'status': <JobStatus.QUEUED: 'job is queued'>, 'status_msg': 'job is queued', 'queue_position': 3}\n",
      "{'job_id': '5b4a2463e036a500391bdbfc', 'status': <JobStatus.DONE: 'job has successfully run'>, 'status_msg': 'job has successfully run'}\n",
      "here\n",
      "nice\n",
      "0000: 376\n",
      "1111: 339\n",
      "1010: 3\n",
      "0101: 4\n",
      "0011: 20\n",
      "1100: 26\n",
      "1001: 2\n",
      "0110: 9\n",
      "1000: 20\n",
      "0111: 42\n",
      "0001: 29\n",
      "1110: 44\n",
      "0010: 29\n",
      "1101: 50\n",
      "0100: 15\n",
      "1011: 16\n",
      "{'0000': 376, '0001': 29, '0010': 29, '0011': 20, '0100': 15, '0101': 4, '0110': 9, '0111': 42, '1000': 20, '1001': 2, '1010': 3, '1011': 16, '1100': 26, '1101': 50, '1110': 44, '1111': 339}\n",
      "00\n",
      "0.9178433889602053\n",
      "15 1\n",
      "\n",
      "15 2\n",
      "\n",
      "15 3\n",
      "\n",
      "Time elapsed:  7002.592554092407\n"
     ]
    }
   ],
   "source": [
    "# Author: Joey Li, Atsushi Hu, Becca Shapiro\n",
    "# Date: 7/20/18\n",
    "\n",
    "# Implements real randomized benchmarking procedure on the [[4,2,2]] code\n",
    "# Draws fixed number of gates from R(2) group and concatenates them to make a circuit then inverts the gates\n",
    "# Data gathered is used to compute infidelity rate for procedure\n",
    "# See paper for more details\n",
    "\n",
    "from qiskit import QuantumCircuit, ClassicalRegister, QuantumRegister, QISKitError\n",
    "from qiskit import available_backends, register, execute, get_backend\n",
    "from qiskit.tools.visualization import circuit_drawer, plot_histogram, plot_state\n",
    "from qiskit.tools.qi.qi import state_fidelity\n",
    "import random\n",
    "import pickle\n",
    "import time\n",
    "import datetime\n",
    "import xlsxwriter\n",
    "\n",
    "#setup\n",
    "import sys, getpass\n",
    "try:\n",
    "    sys.path.append(\"../../\") # go to parent dir\n",
    "    import Qconfig\n",
    "    qx_config = {\n",
    "        \"APItoken\": Qconfig.APItoken,\n",
    "        \"url\": Qconfig.config['url']}\n",
    "    print('Qconfig loaded from %s.' % Qconfig.__file__)\n",
    "except:\n",
    "    APItoken = getpass.getpass('Please input your token and hit enter: ')\n",
    "    qx_config = {\n",
    "        \"APItoken\": APItoken,\n",
    "        \"url\":\"https://quantumexperience.ng.bluemix.net/api\"}\n",
    "    print('Qconfig.py not found in qiskit-tutorial directory; Qconfig loaded using user input.')\n",
    "    \n",
    "#matlab stuff\n",
    "import matplotlib.pyplot as plt\n",
    "%matplotlib inline\n",
    "import numpy as np\n",
    "from numpy import matrix\n",
    "from math import pi\n",
    "\n",
    "#set \n",
    "QX_URL = \"https://quantumexperience.ng.bluemix.net/api\"\n",
    "QX_TOKEN = \"06704f99fc6dd68ab5ae2663f0861723406de8b4553cacdabdf500292c937717442daa90c2ffa19109cfb9114adb7c87711a08860d34d1da9332ac664e4e24a7\"\n",
    "\n",
    "#register token and url\n",
    "register(QX_TOKEN, QX_URL)\n",
    "\n",
    "backend = 'ibmqx4'\n",
    "\n",
    "# prepare a state of all zeros\n",
    "def prepareZeros(qc):\n",
    "    qc.h(qr[1])\n",
    "    qc.barrier()\n",
    "    qc.cx(qr[1],qr[0])\n",
    "    qc.barrier()\n",
    "    qc.cx(qr[1],qr[2])\n",
    "    qc.barrier()\n",
    "    qc.cx(qr[2],qr[3])\n",
    "    qc.barrier()\n",
    "\n",
    "    \n",
    "# compile into different states\n",
    "def compilation(qc,i,perm):\n",
    "    if i == 0: \n",
    "        return\n",
    "    elif i == 1:\n",
    "        qc.x(qr[perm[0]])\n",
    "        qc.x(qr[perm[2]])\n",
    "    elif i == 2:\n",
    "        qc.x(qr[perm[0]])\n",
    "        qc.x(qr[perm[1]])\n",
    "    elif i == 3:\n",
    "        qc.x(qr[perm[2]])\n",
    "        qc.x(qr[perm[1]])\n",
    "    qc.barrier()\n",
    "            \n",
    "# add phase gate\n",
    "def addPhase(qc):\n",
    "    qc.h(qr[0])\n",
    "    qc.h(qr[1])\n",
    "    qc.h(qr[2])\n",
    "    qc.h(qr[3])\n",
    "    qc.barrier()\n",
    "    qc.cx(qr[1],qr[0])\n",
    "    qc.barrier()\n",
    "    qc.s(qr[0])\n",
    "    qc.barrier()\n",
    "    qc.cx(qr[1],qr[0])\n",
    "    qc.barrier()\n",
    "    \n",
    "# remove phase gate\n",
    "def removePhase(qc,perm):\n",
    "    qc.cx(qr[perm[1]], qr[perm[0]])\n",
    "    qc.barrier()\n",
    "    qc.sdg(qr[perm[0]])\n",
    "    qc.barrier()\n",
    "    qc.cx(qr[perm[1]], qr[perm[0]])\n",
    "    qc.barrier()\n",
    "    qc.h(qr[3])\n",
    "    qc.h(qr[2])\n",
    "    qc.h(qr[1])\n",
    "    qc.h(qr[0])\n",
    "    qc.barrier()\n",
    "    \n",
    "# parse a string of gens\n",
    "def addCliffGens(qc, string, perm):\n",
    "    for char in string:\n",
    "        i = int(char)\n",
    "        if i==0:\n",
    "            qc.x(qr[perm[0]])\n",
    "            qc.x(qr[perm[2]])\n",
    "        elif i==1:\n",
    "            qc.x(qr[perm[0]])\n",
    "            qc.x(qr[perm[1]])\n",
    "        elif i==2:\n",
    "            qc.z(qr[perm[0]])\n",
    "            qc.z(qr[perm[1]])\n",
    "        elif i==3:\n",
    "            qc.z(qr[perm[0]])\n",
    "            qc.z(qr[perm[2]])\n",
    "        elif i ==4:\n",
    "            qc.h(qr[0])\n",
    "            qc.h(qr[1])\n",
    "            qc.h(qr[2])\n",
    "            qc.h(qr[3])\n",
    "            qc.barrier()\n",
    "        elif i == 5:\n",
    "            qc.s(qr[0])\n",
    "            qc.s(qr[1])\n",
    "            qc.s(qr[2])\n",
    "            qc.s(qr[3])\n",
    "            qc.barrier()\n",
    "        elif i==6:\n",
    "            swap(perm,0,1)\n",
    "        elif i==7:\n",
    "            swap(perm,0,2)        \n",
    "\n",
    "# find string for generators\n",
    "def findGenString(mat):\n",
    "    for key, value in signedGroup.items():\n",
    "        if np.array_equal(value, mat):\n",
    "            return key\n",
    "    print(\"OH NO\")\n",
    "    return None\n",
    "        \n",
    "# invert cliff gens \n",
    "def invertCliffGens(qc, genString, perm):\n",
    "    for char in genString[::-1]:\n",
    "        i = int(char)\n",
    "        if i==0:\n",
    "            qc.x(qr[perm[0]])\n",
    "            qc.x(qr[perm[2]])\n",
    "        elif i==1:\n",
    "            qc.x(qr[perm[0]])\n",
    "            qc.x(qr[perm[1]])\n",
    "        elif i==2:\n",
    "            qc.z(qr[perm[0]])\n",
    "            qc.z(qr[perm[1]])\n",
    "        elif i==3:\n",
    "            qc.z(qr[perm[0]])\n",
    "            qc.z(qr[perm[2]])\n",
    "        elif i ==4:\n",
    "            qc.h(qr[0])\n",
    "            qc.h(qr[1])\n",
    "            qc.h(qr[2])\n",
    "            qc.h(qr[3])\n",
    "        elif i == 5:\n",
    "            qc.sdg(qr[0])\n",
    "            qc.sdg(qr[1])\n",
    "            qc.sdg(qr[2])\n",
    "            qc.sdg(qr[3])\n",
    "        elif i==6:\n",
    "            swap(perm,0,1)\n",
    "        elif i==7:\n",
    "            swap(perm,0,2)\n",
    "    qc.barrier()\n",
    "\n",
    "# adds random real clifford\n",
    "def addRandCliff(qc, mat,perm):\n",
    "    genString = random.choice(list(r2group.keys()))\n",
    "    addCliffGens(qc,genString,perm)\n",
    "    qc.barrier()\n",
    "    return np.matmul(mat,r2group[genString])\n",
    "\n",
    "# used for reindexing qubits\n",
    "def swap(perm, a, b):\n",
    "    temp = perm[a]\n",
    "    perm[a] = perm[b]\n",
    "    perm[b] = temp\n",
    "    \n",
    "# converts 0-3 to binary strings\n",
    "def toBinString(num):\n",
    "    if num ==0:\n",
    "        return \"00\"\n",
    "    elif num == 1:\n",
    "        return \"01\"\n",
    "    elif num == 2:\n",
    "        return \"10\"\n",
    "    elif num == 3:\n",
    "        return \"11\"\n",
    "    else:\n",
    "        print(\"OH NO\")\n",
    "\n",
    "def printDataFormatted(data):\n",
    "    strings = [\"0000\", \"1111\", \"1010\", \"0101\", \"0011\", \"1100\", \"1001\", \"0110\", \"1000\", \"0111\", \"0001\", \"1110\"\n",
    "              , \"0010\", \"1101\", \"0100\", \"1011\"]\n",
    "    for string in strings:\n",
    "        try:\n",
    "            print(string+\": \"+str(data[string]))\n",
    "        except:\n",
    "            print(\"oops, none of \"+string)\n",
    "            pass\n",
    "    \n",
    "\n",
    "def writeToExcel(results,initialState,row,col):\n",
    "    states = [\"0000\", \"1111\", \"1010\", \"0101\", \"0011\", \"1100\", \"1001\", \"0110\", \"1000\", \"0111\", \"0001\", \"1110\"\n",
    "              , \"0010\", \"1101\", \"0100\", \"1011\"]\n",
    "    print(results)\n",
    "    worksheet.write(row,0,initialState)\n",
    "\n",
    "    col = 1\n",
    "    for state in states:\n",
    "        try:\n",
    "            worksheet.write(row,col,results[state])\n",
    "            \n",
    "        except:\n",
    "            print(state+\"does not exist\")\n",
    "            worksheet.write(row,col,0)\n",
    "        col = col+1\n",
    "    \n",
    "    return row+1\n",
    "\n",
    "def prepCircuitStrings(length, reps):\n",
    "    circuits = []\n",
    "    for i in range(reps):\n",
    "        circuit = \"\"\n",
    "        mat = r2group[\"\"]\n",
    "        for j in range(length-1):\n",
    "            genString = random.choice(list(r2group.keys()))\n",
    "            circuit += genString\n",
    "            circuit += \"8\"\n",
    "            mat = np.matmul(mat,r2group[genString])\n",
    "        #Find String\n",
    "        totalString = findGenString(mat)\n",
    "        \n",
    "        circuit += totalString[::-1]\n",
    "        circuit += \"8\"\n",
    "        \n",
    "        circuits.append(circuit)\n",
    "        print(circuit)\n",
    "    f = open(str(length)+\"-\"+str(9)+\"-circuits.pkl\", \"wb\")\n",
    "    pickle.dump(circuits,f)\n",
    "    f.close()\n",
    "    \n",
    "def loadAllCircuits(length,reps):\n",
    "    circPickle = open(str(length)+\"-\"+str(9)+\"-circuits.pkl\", \"rb\")\n",
    "    circuitStrings = pickle.load(circPickle)\n",
    "    return circuitStrings   \n",
    "        \n",
    "lengths = [15]\n",
    "reps = [7,0,0,0]\n",
    "initialStates = [0,1,2,3]\n",
    "shots = 1024\n",
    "phased = False\n",
    "shouldWrite = True\n",
    "\n",
    "pickle_in = open(\"r2group.pkl\", \"rb\")\n",
    "r2group = pickle.load(pickle_in)\n",
    "pickle_in2 = open(\"r2group.pickle\", \"rb\")\n",
    "signedGroup = pickle.load(pickle_in2)\n",
    "\n",
    "if shouldWrite:\n",
    "    workbook = xlsxwriter.Workbook(\"encoded_(\"+\n",
    "                                   \",\".join(map(str,lengths))+\n",
    "                                   \")_(\"+\n",
    "                                   \",\".join(map(str,reps))+\n",
    "                                   \")_\"+\n",
    "                                   str(int(phased))+\n",
    "                                   \"_\"+\n",
    "                                   str(datetime.datetime.now().strftime('%Y%m%d_%H%M%S'))+\n",
    "                                   \".xlsx\")\n",
    "    worksheet = workbook.add_worksheet()\n",
    "    \n",
    "    states = [\"Key\",\"0000\", \"1111\", \"1010\", \"0101\", \"0011\", \"1100\", \"1001\", \"0110\", \"1000\", \"0111\", \"0001\", \"1110\"\n",
    "              , \"0010\", \"1101\", \"0100\", \"1011\"]\n",
    "    for index in range(len(states)):\n",
    "        worksheet.write(0,index+1,states[index])\n",
    "    row = 1\n",
    "    col = 1\n",
    "    \n",
    "    \n",
    "# Reset this whenever you feel like it\n",
    "# prepCircuitStrings(6,9)\n",
    "\n",
    "\n",
    "startTime = time.time()\n",
    "\n",
    "for length in lengths:\n",
    "\n",
    "        \n",
    "    circuits = loadAllCircuits(lengths[0],reps[0])\n",
    "    print(circuits)\n",
    "    \n",
    "    for compiler in initialStates:\n",
    "        \n",
    "        '''\n",
    "        if shouldWrite:\n",
    "            worksheet.write(row,0, \"Length: \"+str(length))\n",
    "            row = row+1\n",
    "            '''\n",
    "        \n",
    "        print(str(length)+\" \"+str(compiler)+\"\\n\")\n",
    "\n",
    "        # run our trials\n",
    "        for i in range(reps[compiler]):\n",
    "\n",
    "            # run one trial of given length \n",
    "            mat = r2group[\"\"]\n",
    "            n = 4\n",
    "            qr = QuantumRegister(n)\n",
    "            cr = ClassicalRegister(n)\n",
    "            qc = QuantumCircuit(qr, cr)\n",
    "            \n",
    "            perm = {0: 0, 1: 1, 2: 2, 3: 3}\n",
    "            \n",
    "            # prepare encoded zeros\n",
    "            prepareZeros(qc) \n",
    "            \n",
    "            # optional: add phase\n",
    "            if phased:\n",
    "                addPhase(qc)\n",
    "            \n",
    "            # Deprecated\n",
    "            \n",
    "            '''\n",
    "            # add the first m-1 clifford gates\n",
    "            for j in range(length-1):\n",
    "                mat = addRandCliff(qc,mat,perm)\n",
    "            \n",
    "            #Find String\n",
    "            genString = findGenString(mat)\n",
    "            \n",
    "            print(genString)\n",
    "            \n",
    "            # invert everything\n",
    "            invertCliffGens(qc,genString, perm)'''\n",
    "            \n",
    "            circuit = circuits[i]\n",
    "            addCliffGens(qc,circuit,perm)\n",
    "            \n",
    "            # optional: remove phase\n",
    "            if phased:\n",
    "                removePhase(qc,perm)\n",
    "\n",
    "            # get the qc compiled into one of the 4 different states\n",
    "            compilation(qc,compiler, perm)\n",
    "            \n",
    "\n",
    "            # measure\n",
    "            for i in range(n):\n",
    "                qc.measure(qr[perm[i]], cr[i])      \n",
    "            \n",
    "            print(\"got\")\n",
    "            job = execute(qc, backend, shots = shots)\n",
    "            \n",
    "            #Print Job status every 120 seconds\n",
    "            lapse = 0\n",
    "            interval = 120\n",
    "            while not job.done:\n",
    "                print('Status @ {} seconds'.format(interval * lapse))\n",
    "                print(job.status)\n",
    "                time.sleep(interval)\n",
    "                lapse += 1\n",
    "            print(job.status)\n",
    "            print(\"here\")\n",
    "            result = job.result()\n",
    "            print(\"nice\")\n",
    "            data = result.get_counts(qc)\n",
    "            \n",
    "            # use when running on qx5 and when debugging\n",
    "            printDataFormatted(data)\n",
    "            \n",
    "            # clean the data\n",
    "            cleanData = {}\n",
    "            totalSuccessful = 0\n",
    "            for key in data:\n",
    "                ohoh = ['0000', '1111']\n",
    "                ohone = ['1010', '0101']\n",
    "                oneoh = ['1100', '0011']\n",
    "                oneone = ['0110', '1001']\n",
    "                if key not in ohoh+ohone+oneoh+oneone:\n",
    "                    continue\n",
    "                if key in ohoh:\n",
    "                    if \"00\" not in cleanData.keys():\n",
    "                        cleanData[\"00\"] = 0\n",
    "                    cleanData[\"00\"] += data[key]\n",
    "                    totalSuccessful += data[key]\n",
    "                elif key in ohone:\n",
    "                    if \"01\" not in cleanData.keys():\n",
    "                        cleanData[\"01\"] = 0\n",
    "                    cleanData[\"01\"] += data[key]\n",
    "                    totalSuccessful += data[key]\n",
    "                elif key in oneoh:\n",
    "                    if \"10\" not in cleanData.keys():\n",
    "                        cleanData[\"10\"] = 0\n",
    "                    cleanData[\"10\"] += data[key]\n",
    "                    totalSuccessful += data[key]\n",
    "                elif key in oneone:\n",
    "                    if \"11\" not in cleanData.keys():\n",
    "                        cleanData[\"11\"] = 0\n",
    "                    cleanData[\"11\"] += data[key]\n",
    "                    totalSuccessful += data[key]\n",
    "            \n",
    "            # set shouldWrite to false if running on simulator or don't actually care\n",
    "            if shouldWrite:\n",
    "                row = writeToExcel(data,compiler,row,col)\n",
    "            \n",
    "            # optional: get success rate\n",
    "            desired = toBinString(compiler)\n",
    "            print(desired)\n",
    "            print(cleanData[desired]/totalSuccessful)\n",
    "\n",
    "\n",
    "endTime = time.time()\n",
    "print(\"Time elapsed: \", endTime-startTime)\n",
    "\n",
    "if shouldWrite:\n",
    "    workbook.close()\n",
    "\n",
    "\n",
    "  "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "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.6.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}