Count_gates.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "b55cd187",
   "metadata": {},
   "source": [
    "Count gates in transpiled MCT circuit\n",
    "==============================="
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "1d15159a",
   "metadata": {},
   "outputs": [],
   "source": [
    "import pytest\n",
    "from pytest import raises\n",
    "from circuits import SteaneCodeLogicalQubit\n",
    "#from qiskit import QuantumCircuit, QuantumRegister, ClassicalRegister, execute, Aer\n",
    "from qiskit import execute, Aer\n",
    "from qiskit.compiler import transpile\n",
    "import numpy as np\n",
    "\n",
    "from helper_functions import (\n",
    "    flip_code_words,\n",
    "    string_reverse,\n",
    "    strings_AND_bitwise,\n",
    "    string_ancilla_mask,\n",
    "    correct_qubit,\n",
    "    flip_code_words,\n",
    "    count_valid_output_strings,\n",
    "    mean_of_list,\n",
    "    calculate_standard_error, \n",
    "    correct_qubit,\n",
    "    print_time,\n",
    "    get_parity_check_matrix,\n",
    "    get_codewords\n",
    "    )\n",
    "\n",
    "SINGLE_GATE_SET = ['id', 'ry', 'rx']\n",
    "TWO_GATE_SET = ['rxx']\n",
    "BASIS_GATE_SET = SINGLE_GATE_SET + TWO_GATE_SET\n",
    "TEST_X_QUBIT = 4\n",
    "SHOTS = 10      #Number of shots to run    \n",
    "SIMULATOR = Aer.get_backend('qasm_simulator')\n",
    "\n",
    "parity_check_matrix = get_parity_check_matrix()\n",
    "codewords = get_codewords()\n",
    "\n",
    "SPACE = ' '\n",
    "DATA_QUBITS = 7\n",
    "NOISE = False"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "41b4c758",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Current Time = 09:42:19\n",
      "Processing without MCT\n",
      "Transpiled gate statistics OrderedDict([('rx', 387), ('ry', 183), ('rxx', 142), ('measure', 17), ('barrier', 12), ('reset', 6)])\n",
      "Gate statistics before transpilation OrderedDict([('cx', 39), ('cz', 31), ('measure', 17), ('h', 15), ('reset', 13), ('barrier', 12), ('ccx', 12), ('x', 1)])\n",
      "Current Time = 09:42:21\n",
      "Results for error introduced on qubit index 0\n",
      "The simulated result after error correction are {'0000 100 000 0000000': 2, '0000 100 000 0011110': 2, '0000 100 000 0101101': 1, '0000 100 000 0110011': 2, '0000 100 000 1001011': 1, '0000 100 000 1010101': 2}\n",
      "The error rate is 0.0000\n",
      "\n",
      "Current Time = 09:42:21\n",
      "Processing with MCT\n",
      "Transpiled gate statistics OrderedDict([('rx', 952), ('ry', 433), ('rxx', 312), ('measure', 13), ('barrier', 12), ('reset', 6)])\n",
      "Gate statistics before transpilation OrderedDict([('x', 37), ('h', 29), ('cx', 20), ('mcx_gray', 14), ('reset', 13), ('measure', 13), ('barrier', 12), ('cz', 12)])\n",
      "Current Time = 09:42:24\n",
      "Results for error introduced on qubit index 0\n",
      "The simulated result after error correction are {'100 000 0101101': 1, '100 000 0110011': 3, '100 000 1001011': 1, '100 000 1100110': 3, '100 000 1111000': 2}\n",
      "The error rate is 0.0000\n",
      "\n"
     ]
    }
   ],
   "source": [
    "for MCT in {True, False}:    \n",
    "    print_time()\n",
    "    if MCT:\n",
    "        qubit = SteaneCodeLogicalQubit(1, parity_check_matrix, codewords, extend_ancilla = False)\n",
    "        print(\"Processing with MCT\")\n",
    "    else:\n",
    "        qubit = SteaneCodeLogicalQubit(1, parity_check_matrix, codewords, extend_ancilla = True)\n",
    "        print(\"Processing without MCT\")\n",
    "    qubit.set_up_logical_zero()\n",
    "    qubit.barrier()\n",
    "    index = 0\n",
    "    qubit.force_X_error(index)   #force X error for testing\n",
    "    #qubit.force_Z_error(1)   #force Z error for testing\n",
    "    qubit.barrier()\n",
    "    qubit.set_up_ancilla()\n",
    "    qubit.barrier()\n",
    "    if MCT:     #Multi Control Toffoli (MCT) gates\n",
    "        qubit.correct_errors(mct = True)\n",
    "    else:\n",
    "        qubit.correct_errors()\n",
    "    qubit.barrier()\n",
    "\n",
    "    if NOISE:\n",
    "        noise_model = get_noise(MEASURE_NOISE, SINGLE_GATE_DEPOLARISING, TWO_GATE_DEPOLARISING,\n",
    "                           SINGLE_GATE_SET, TWO_GATE_SET)\n",
    "    qubit.logical_measure_data()\n",
    "    qubit.logical_measure_ancilla()\n",
    "    qubit.barrier()\n",
    "    qt = transpile(qubit, basis_gates = BASIS_GATE_SET)\n",
    "    print('Transpiled gate statistics', qt.count_ops())\n",
    "    print('Gate statistics before transpilation', qubit.count_ops())\n",
    "    if NOISE:\n",
    "        result = execute(qt, SIMULATOR, noise_model = noise_model, shots = SHOTS).result()\n",
    "    else:\n",
    "        result = execute(qt, SIMULATOR, shots = SHOTS).result()\n",
    "    counts = result.get_counts(qt)\n",
    "    print_time()\n",
    "    print(f'Results for error introduced on qubit index {index}')\n",
    "    print ('The simulated result after error correction are', counts)\n",
    "    if MCT:\n",
    "        count_valid, count_invalid, _ = count_valid_output_strings(counts, codewords, 2)\n",
    "    else:\n",
    "        count_valid, count_invalid, _ = count_valid_output_strings(counts, codewords, 3)\n",
    "    error_rate = count_invalid / SHOTS\n",
    "    print(f'The error rate is {error_rate:.4f}')\n",
    "    print_time\n",
    "    print()"
   ]
  }
 ],
 "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.8.10"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}