grovers_algorithm.py

#******************************************************************************#
#                                                                              #
#                        Main function for Grover's                            #
#                            search algorithm.                                 #
#                                                                              #
#******************************************************************************#

import sys
import numpy as np
from numpy.linalg import norm
import cmath
import matplotlib.pyplot as plt
import os

from src.grover import Grover

from src.sparse_matrix import SparseMatrix
from src.quantum_register import QuantumRegister
from src.quantum_operator import Operator
from src.operators import *

from src.oracle import *
from src.grover_phase import *


def main(args):
    if len(args) > 4 or len(args) < 3:
        print("Add the number of qubits and target state.")
        print("python grovers_algorithm.py n_qubits [1:'multiples-of' or" +
            "2:'exponents-of'] target-number optional(number of target states)")
        sys.exit()
    elif len(args) == 4 and type(int(args[3])) != np.int:
        print("Add the number of qubits and target state.")
        print("python grovers_algorithm.py n_qubits [1:'multiples-of or'" +
            "'2:exponents-of'] target-number optional(number of target states)")
        sys.exit()

    n_qubits = int(args[0])
    test = args[1]
    target_number = int(args[2])
    n_states = -1  # Default value assuming an unknown number of states.
    if len(args) == 4:
        n_states = int(args[3])

    p = Grover(n_qubits, target_number)
    p.build_quantum_register()
    p.init_register()
    p.init_reflection_matrix()
    p.gen_oracle(test)
    itterations = 1000
    for i in range(itterations):
        sys.stdout.write("Simulation progress: %.0f%%\r"
                                % ((100 * i / itterations)))
        sys.stdout.flush()  # Prints progress of simulation.
        p.build_quantum_register() # Reset Quantum Register to run Grovers on it again.
        p.init_register()
        result = p.run(n_states)
    p.plot_results()
    p.print_results(itterations)

if __name__ == '__main__':
    main(sys.argv[1:])