modified_code.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "authorship_tag": "ABX9TyONuEU/e7XinlYjIDK/9qqg",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/Sh0g0-1758/Zenith-zealots/blob/main/modified_code.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 2,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "CNq9GLY8NqSD",
        "outputId": "f9635ce6-3a89-4464-cac3-c5fa18caef22"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Collecting qiskit\n",
            "  Downloading qiskit-0.45.0-py3-none-any.whl (9.6 kB)\n",
            "Collecting qiskit-terra==0.45.0 (from qiskit)\n",
            "  Downloading qiskit_terra-0.45.0-cp38-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (6.3 MB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m6.3/6.3 MB\u001b[0m \u001b[31m37.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hCollecting rustworkx>=0.13.0 (from qiskit-terra==0.45.0->qiskit)\n",
            "  Downloading rustworkx-0.13.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (2.0 MB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m2.0/2.0 MB\u001b[0m \u001b[31m61.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: numpy<2,>=1.17 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit) (1.23.5)\n",
            "Collecting ply>=3.10 (from qiskit-terra==0.45.0->qiskit)\n",
            "  Downloading ply-3.11-py2.py3-none-any.whl (49 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m49.6/49.6 kB\u001b[0m \u001b[31m5.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: psutil>=5 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit) (5.9.5)\n",
            "Requirement already satisfied: scipy>=1.5 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit) (1.11.3)\n",
            "Requirement already satisfied: sympy>=1.3 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit) (1.12)\n",
            "Collecting dill>=0.3 (from qiskit-terra==0.45.0->qiskit)\n",
            "  Downloading dill-0.3.7-py3-none-any.whl (115 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m115.3/115.3 kB\u001b[0m \u001b[31m11.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: python-dateutil>=2.8.0 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit) (2.8.2)\n",
            "Collecting stevedore>=3.0.0 (from qiskit-terra==0.45.0->qiskit)\n",
            "  Downloading stevedore-5.1.0-py3-none-any.whl (49 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m49.6/49.6 kB\u001b[0m \u001b[31m6.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hCollecting symengine<0.10,>=0.9 (from qiskit-terra==0.45.0->qiskit)\n",
            "  Downloading symengine-0.9.2-cp310-cp310-manylinux2010_x86_64.whl (37.5 MB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m37.5/37.5 MB\u001b[0m \u001b[31m40.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit) (4.5.0)\n",
            "Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.10/dist-packages (from python-dateutil>=2.8.0->qiskit-terra==0.45.0->qiskit) (1.16.0)\n",
            "Collecting pbr!=2.1.0,>=2.0.0 (from stevedore>=3.0.0->qiskit-terra==0.45.0->qiskit)\n",
            "  Downloading pbr-5.11.1-py2.py3-none-any.whl (112 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m112.7/112.7 kB\u001b[0m \u001b[31m11.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy>=1.3->qiskit-terra==0.45.0->qiskit) (1.3.0)\n",
            "Installing collected packages: ply, symengine, rustworkx, pbr, dill, stevedore, qiskit-terra, qiskit\n",
            "Successfully installed dill-0.3.7 pbr-5.11.1 ply-3.11 qiskit-0.45.0 qiskit-terra-0.45.0 rustworkx-0.13.2 stevedore-5.1.0 symengine-0.9.2\n",
            "Collecting qiskit-aer\n",
            "  Downloading qiskit_aer-0.13.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (12.3 MB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m12.3/12.3 MB\u001b[0m \u001b[31m12.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: qiskit>=0.44.0 in /usr/local/lib/python3.10/dist-packages (from qiskit-aer) (0.45.0)\n",
            "Requirement already satisfied: numpy>=1.16.3 in /usr/local/lib/python3.10/dist-packages (from qiskit-aer) (1.23.5)\n",
            "Requirement already satisfied: scipy>=1.0 in /usr/local/lib/python3.10/dist-packages (from qiskit-aer) (1.11.3)\n",
            "Requirement already satisfied: qiskit-terra==0.45.0 in /usr/local/lib/python3.10/dist-packages (from qiskit>=0.44.0->qiskit-aer) (0.45.0)\n",
            "Requirement already satisfied: rustworkx>=0.13.0 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (0.13.2)\n",
            "Requirement already satisfied: ply>=3.10 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (3.11)\n",
            "Requirement already satisfied: psutil>=5 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (5.9.5)\n",
            "Requirement already satisfied: sympy>=1.3 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (1.12)\n",
            "Requirement already satisfied: dill>=0.3 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (0.3.7)\n",
            "Requirement already satisfied: python-dateutil>=2.8.0 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (2.8.2)\n",
            "Requirement already satisfied: stevedore>=3.0.0 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (5.1.0)\n",
            "Requirement already satisfied: symengine<0.10,>=0.9 in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (0.9.2)\n",
            "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (4.5.0)\n",
            "Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.10/dist-packages (from python-dateutil>=2.8.0->qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (1.16.0)\n",
            "Requirement already satisfied: pbr!=2.1.0,>=2.0.0 in /usr/local/lib/python3.10/dist-packages (from stevedore>=3.0.0->qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (5.11.1)\n",
            "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy>=1.3->qiskit-terra==0.45.0->qiskit>=0.44.0->qiskit-aer) (1.3.0)\n",
            "Installing collected packages: qiskit-aer\n",
            "Successfully installed qiskit-aer-0.13.0\n"
          ]
        }
      ],
      "source": [
        "!pip install qiskit\n",
        "!pip install qiskit-aer"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "from qiskit import QuantumCircuit, QuantumRegister, ClassicalRegister\n",
        "from qiskit.visualization import plot_histogram\n",
        "import numpy as np\n",
        "\n",
        "# Import the necessary functions for visualizing quantum states\n",
        "from qiskit.visualization import plot_bloch_multivector\n",
        "\n",
        "# Import the Statevector class from Qiskit's quantum information module\n",
        "from qiskit.quantum_info import Statevector\n",
        "import numpy as np\n",
        "\n",
        "\n",
        "# def matrix_gen(qubit_location, valid_qubits):\n",
        "#     # initialise the matrix with all -1\n",
        "#     matrix = np.full((4, 4), -1)\n",
        "\n",
        "#\n",
        "#     # set the qubit locations\n",
        "#     for qubit in qubit_location:\n",
        "#         if valid_qubits[qubit]:\n",
        "#             matrix[qubit_location[qubit][0]][qubit_location[qubit][1]] = qubit\n",
        "#\n",
        "#     return matrix\n",
        "\n",
        "def print_matrix(matrix):\n",
        "    for i in range(4):\n",
        "        for j in range(4):\n",
        "            print(matrix[i][j], end=\" \")\n",
        "        print()\n",
        "    print()\n",
        "\n",
        "\n",
        "def left_shift(matrix):\n",
        "    n = len(matrix)\n",
        "    # array of length n with 0\n",
        "    last_loc = np.zeros(n)\n",
        "\n",
        "    for i in range(n):\n",
        "        for j in range(n):\n",
        "            if matrix[i][j] != -1:\n",
        "                val = matrix[i][j]\n",
        "                matrix[i][j] = -1\n",
        "                # shift it to last-loc in that row\n",
        "                matrix[i][int(last_loc[i])] = val\n",
        "                last_loc[i] += 1\n",
        "\n",
        "    return matrix\n",
        "\n",
        "\n",
        "def right_shift(matrix):\n",
        "    n = len(matrix)\n",
        "    # array of length n with (n-1)\n",
        "    last_loc = np.full(n, n - 1)\n",
        "\n",
        "    for i in range(n):\n",
        "        for j in range(n - 1, -1, -1):\n",
        "            if matrix[i][j] != -1:\n",
        "                val = matrix[i][j]\n",
        "                matrix[i][j] = -1\n",
        "                # shift it to last-loc in that row\n",
        "                matrix[i][int(last_loc[i])] = val\n",
        "                last_loc[i] -= 1\n",
        "    return matrix\n",
        "\n",
        "\n",
        "def up_shift(matrix):\n",
        "    n = len(matrix)\n",
        "    # array of length n with 0\n",
        "    last_loc = np.zeros(n)\n",
        "\n",
        "    for i in range(n):\n",
        "        for j in range(n):\n",
        "            if matrix[j][i] != -1:\n",
        "                val = matrix[j][i]\n",
        "                matrix[j][i] = -1\n",
        "                # shift it to last-loc in that row\n",
        "                matrix[int(last_loc[i])][i] = val\n",
        "                last_loc[i] += 1\n",
        "\n",
        "    return matrix\n",
        "\n",
        "\n",
        "def down_shift(matrix):\n",
        "    n = len(matrix)\n",
        "    # array of length n with n-1\n",
        "    last_loc = np.full(n, n - 1)\n",
        "\n",
        "    for i in range(n):\n",
        "        for j in range(n - 1, -1, -1):\n",
        "            print(\"row: \", j, \"col: \", i)\n",
        "            if matrix[j][i] != -1:\n",
        "                val = matrix[j][i]\n",
        "                matrix[j][i] = -1\n",
        "                # shift it to last-loc in that row\n",
        "                matrix[last_loc[i]][i] = val\n",
        "                last_loc[i] -= 1\n",
        "\n",
        "    return matrix\n",
        "\n",
        "\n",
        "def shift(matrix, direction):\n",
        "    if direction == 0:\n",
        "        return left_shift(matrix)\n",
        "    elif direction == 1:\n",
        "        return right_shift(matrix)\n",
        "    elif direction == 2:\n",
        "        return up_shift(matrix)\n",
        "    elif direction == 3:\n",
        "        return down_shift(matrix)\n",
        "    else:\n",
        "        print(\"Invalid direction\")\n",
        "        return matrix\n",
        "\n",
        "\n",
        "# def operate_gate(qc_map,qc_loc,qc_matrix, direction, gates):\n",
        "\n",
        "def merge_left_wall(qc_loc, qc_matrix, state_vector_dict):\n",
        "    j = 0\n",
        "    k = 1\n",
        "    for i in range(len(qc_matrix)):\n",
        "        if qc_matrix[i][j] == -1 or qc_matrix[i][k] == -1:\n",
        "            continue\n",
        "        s1 = state_vector_dict[qc_matrix[i][j]]\n",
        "        s2 = state_vector_dict[qc_matrix[i][k]]\n",
        "        if s1 == s2:\n",
        "            # we will non-wall qubit\n",
        "            del qc_map[qc_matrix[i][k]]\n",
        "            qc_loc[qc_matrix[i][k]] = None\n",
        "            qc_matrix[i][k] = -1\n",
        "\n",
        "def merge_right_wall(qc_loc, qc_matrix, state_vector_dict):\n",
        "    j = len(qc_matrix) - 1\n",
        "    k = len(qc_matrix) - 2\n",
        "    for i in range(len(qc_matrix)):\n",
        "        if qc_matrix[i][j] == -1 or qc_matrix[i][k] == -1:\n",
        "            continue\n",
        "        s1 = state_vector_dict[qc_matrix[i][j]]\n",
        "        s2 = state_vector_dict[qc_matrix[i][k]]\n",
        "        if s1 == s2:\n",
        "            # we will non-wall qubit\n",
        "            del qc_map[qc_matrix[i][k]]\n",
        "            qc_loc[qc_matrix[i][k]] = None\n",
        "            qc_matrix[i][k] = -1\n",
        "\n",
        "def merge_up_wall(qc_loc, qc_matrix, state_vector_dict):\n",
        "    j = 0\n",
        "    k = 1\n",
        "    for i in range(len(qc_matrix)):\n",
        "        if qc_matrix[i][j] == -1 or qc_matrix[i][k] == -1:\n",
        "            continue\n",
        "        s1 = state_vector_dict[qc_matrix[j][i]]\n",
        "        s2 = state_vector_dict[qc_matrix[k][i]]\n",
        "        if s1 == s2:\n",
        "            # we will non-wall qubit\n",
        "            del qc_map[qc_matrix[k][i]]\n",
        "            qc_loc[qc_matrix[k][i]] = None\n",
        "            qc_matrix[k][i] = -1\n",
        "\n",
        "def merge_down_wall(qc_loc, qc_matrix, state_vector_dict):\n",
        "    j = len(qc_matrix) - 1\n",
        "    k = len(qc_matrix) - 2\n",
        "    for i in range(len(qc_matrix)):\n",
        "        if qc_matrix[i][j] == -1 or qc_matrix[i][k] == -1:\n",
        "            continue\n",
        "        s1 = state_vector_dict[qc_matrix[j][i]]\n",
        "        s2 = state_vector_dict[qc_matrix[k][i]]\n",
        "        if s1 == s2:\n",
        "            # we will non-wall qubit\n",
        "            del qc_map[qc_matrix[k][i]]\n",
        "            qc_loc[qc_matrix[k][i]] = None\n",
        "            qc_matrix[k][i] = -1\n",
        "\n",
        "\n",
        "def merge(qc_map, qc_loc, qc_matrix, curr_dir, state_vector_dict):\n",
        "    # first we update the state vector\n",
        "    for i in range(len(qc_map)):\n",
        "        state_vector_dict[i] = Statevector(qc_map[i])\n",
        "\n",
        "    # now depending on the direction we will iterate the wall and compare their state vectors and if same we will\n",
        "    # destroy the non-wall qubit\n",
        "    if curr_dir == 0:\n",
        "        merge_left_wall(qc_loc, qc_matrix, state_vector_dict)\n",
        "    elif curr_dir == 1:\n",
        "        merge_right_wall(qc_loc, qc_matrix, state_vector_dict)\n",
        "    elif curr_dir == 2:\n",
        "        merge_up_wall(qc_loc, qc_matrix, state_vector_dict)\n",
        "    elif curr_dir == 3:\n",
        "        merge_down_wall(qc_loc, qc_matrix, state_vector_dict)\n",
        "    else:\n",
        "        print(\"Invalid direction\")\n",
        "\n",
        "def matrix_gen(qubit_location):\n",
        "    # initialise the matrix with all -1\n",
        "    matrix = np.full((4, 4), -1)\n",
        "\n",
        "    # set the qubit locations\n",
        "    for qubit in qubit_location:\n",
        "        matrix[qubit_location[qubit][0]][qubit_location[qubit][1]] = qubit\n",
        "\n",
        "    return matrix\n",
        "\n",
        "def operate_gate(qc_map, qc_i,dir,gate):\n",
        "    gate = gate[dir]\n",
        "    if gate == 'H':\n",
        "        qc_map[qc_i].h(0)\n",
        "    elif gate == 'X':\n",
        "        qc_map[qc_i].x(0)\n",
        "    elif gate == 'I':\n",
        "        qc_map[qc_i].i(0)\n",
        "    elif gate == 'Z':\n",
        "        qc_map[qc_i].z(0)\n",
        "    else:\n",
        "        print(\"Invalid gate\")\n",
        "\n",
        "\n",
        "def gate_op(curr_dir, qc_map, qc_location, qc_matrix, wall_gate):\n",
        "\n",
        "    # now we will operate the gates\n",
        "    for i in range(len(qc_matrix)):\n",
        "        if curr_dir == 0:\n",
        "            # left\n",
        "            if qc_matrix[i][0] != -1:\n",
        "                operate_gate(qc_map, qc_matrix[i][0], curr_dir, wall_gate)\n",
        "        elif curr_dir == 1:\n",
        "            # right\n",
        "            if qc_matrix[i][len(qc_matrix) - 1] != -1:\n",
        "                operate_gate(qc_map, qc_matrix[i][len(qc_matrix) - 1], curr_dir, wall_gate)\n",
        "        elif curr_dir == 2:\n",
        "            # up\n",
        "            if qc_matrix[0][i] != -1:\n",
        "                operate_gate(qc_map, qc_matrix[0][i], curr_dir, wall_gate)\n",
        "        elif curr_dir == 3:\n",
        "            # down\n",
        "            if qc_matrix[len(qc_matrix) - 1][i] != -1:\n",
        "                operate_gate(qc_map, qc_matrix[len(qc_matrix) - 1][i], curr_dir, wall_gate)\n",
        "        else:\n",
        "            print(\"Invalid direction\")\n",
        "\n",
        "    qc_matrix = matrix_gen(qc_location)"
      ],
      "metadata": {
        "id": "IprEW5UvYsPe"
      },
      "execution_count": 8,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "last_direction = -1\n",
        "curr_direction = 0\n",
        "game_over = False\n",
        "\n",
        "start_game = curr_direction != last_direction\n",
        "start_game = start_game & game_over\n",
        "\n",
        "\n",
        "def process(qc_map, qc_loc, qc_matrix, dir, gates, state_vector):\n",
        "    # first we will shift the matrix\n",
        "    shift(qc_matrix, dir)\n",
        "\n",
        "    # now we will operate the gates\n",
        "    gate_op(dir, qc_map, qc_loc, qc_matrix, gates)\n",
        "\n",
        "    # now we will merge the qubits\n",
        "    merge(qc_map, qc_loc, qc_matrix, dir, state_vector)\n",
        "\n",
        "    shift(qc_matrix, dir)\n",
        "\n",
        "\n",
        "# Designing Level 1\n",
        "\n",
        "wall_gate = {\n",
        "    0: 'H',\n",
        "    1: 'X',\n",
        "    2: 'I',\n",
        "    3: 'Z'\n",
        "}\n",
        "\n",
        "qc1 = QuantumCircuit(1)\n",
        "qc2 = QuantumCircuit(1)\n",
        "\n",
        "qc1.h(0)  # Hadamard Gate on qubit |+>\n",
        "qc1.z(0)  # Setting the initial state to |->\n",
        "qc2.x(0)  # Setting the initial state to |1>\n",
        "\n",
        "# create a map of quantum circuits\n",
        "qc_map = {\n",
        "    0: qc1,\n",
        "    1: qc2\n",
        "}\n",
        "\n",
        "# now set location of these qubits\n",
        "\n",
        "qc_location = {\n",
        "    0: [3, 1],\n",
        "    1: [3, 3]\n",
        "}\n",
        "\n",
        "# valid_qc = [True, True]\n",
        "\n",
        "# now generate the matrix\n",
        "location_matrix = matrix_gen(qc_location)\n",
        "\n",
        "# let's see the matrix\n",
        "print(location_matrix)\n",
        "\n",
        "# the shift operation is ready\n",
        "\n",
        "# creating a state-vector dictionary\n",
        "state_vector_dict = {}\n",
        "state_vector_dict[0] = Statevector(qc_map[0])\n",
        "state_vector_dict[1] = Statevector(qc_map[1])\n",
        "\n",
        "print(state_vector_dict[0])\n",
        "print(state_vector_dict[1])\n",
        "\n",
        "process(qc_map, qc_location, location_matrix, 0, wall_gate, state_vector_dict)\n",
        "\n",
        "# wall_gate = {\n",
        "#     0: 'H',\n",
        "#     1: 'X',\n",
        "#     2: 'I',\n",
        "#     3: 'Z'\n",
        "# }\n",
        "\n",
        "# qc1.h(0)  # Hadamard Gate on qubit |+>\n",
        "# qc2.x(0)  # Setting the initial state to |1>\n",
        "\n",
        "# qc1_test = qc_map[0]\n",
        "# qc1_test.draw(output='mpl')\n",
        "\n",
        "\n",
        "print(state_vector_dict[0])\n",
        "\n",
        "print(location_matrix)\n"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "aMgDCChpY6-h",
        "outputId": "0cef1eeb-abdb-4cdd-8844-02197c2c1aa3"
      },
      "execution_count": 10,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "[[-1 -1 -1 -1]\n",
            " [-1 -1 -1 -1]\n",
            " [-1 -1 -1 -1]\n",
            " [-1  0 -1  1]]\n",
            "Statevector([ 0.70710678+0.j, -0.70710678+0.j],\n",
            "            dims=(2,))\n",
            "Statevector([0.+0.j, 1.+0.j],\n",
            "            dims=(2,))\n",
            "Statevector([0.+0.j, 1.+0.j],\n",
            "            dims=(2,))\n",
            "[[-1 -1 -1 -1]\n",
            " [-1 -1 -1 -1]\n",
            " [-1 -1 -1 -1]\n",
            " [ 0 -1 -1 -1]]\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "def gate_op(curr_dir, qc_map,qc_location, location_matrix, wall_gate) :\n",
        "  if curr_dir==0 :\n",
        "    for i in qc_location:\n",
        "      if qc_map[i]==None :\n",
        "        continue\n",
        "      if qc_location[i][1]==0:\n",
        "        if wall_gate[0]=='X':\n",
        "          qc_map[i].x(0)\n",
        "        if wall_gate[0]=='H':\n",
        "          qc_map[i].h(0)\n",
        "        if wall_gate[0]=='Z':\n",
        "          qc_map[i].z(0)\n",
        "  if curr_dir==1 :\n",
        "    for i in qc_location:\n",
        "      if qc_location[i][1]==3:\n",
        "        if wall_gate[1]=='X':\n",
        "          qc_map[i].x(0)\n",
        "        if wall_gate[1]=='H':\n",
        "          qc_map[i].h(0)\n",
        "        if wall_gate[1]=='Z':\n",
        "          qc_map[i].z(0)\n",
        "  if curr_dir==2 :\n",
        "    for i in qc_location:\n",
        "      if qc_location[i][0]==0:\n",
        "        if wall_gate[2]=='X':\n",
        "          qc_map[i].x(0)\n",
        "        if wall_gate[2]=='H':\n",
        "          qc_map[i].h(0)\n",
        "        if wall_gate[2]=='Z':\n",
        "          qc_map[i].z(0)\n",
        "\n",
        "  if curr_dir==3 :\n",
        "    for i in qc_location:\n",
        "      if qc_location[i][0]==3:\n",
        "        if wall_gate[3]=='X':\n",
        "          qc_map[i].x(0)\n",
        "        if wall_gate[3]=='H':\n",
        "          qc_map[i].h(0)\n",
        "        if wall_gate[3]=='Z':\n",
        "          qc_map[i].z(0)\n",
        "\n"
      ],
      "metadata": {
        "id": "8lBftb7tZ5Q-"
      },
      "execution_count": 6,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [],
      "metadata": {
        "id": "TxU3Rw3kb3Ci"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}