o 8Va@sdZddlZddlZddlmZmZddlmZmZddlm Z ddl m Z ddl mZddlmZdd lmZmZdd lmZdd lmZdd lmZGd ddeZGdddeZddZddZddZddZdS)zShor's algorithm and helper functions. Todo: * Get the CMod gate working again using the new Gate API. * Fix everything. * Update docstrings and reformat. N)MulS)logsqrt)igcd)continued_fraction_periodic) variations)Gate)Qubitmeasure_partial_oneshot)qapply)QFT) QuantumErrorc@s eZdZdS)OrderFindingExceptionN)__name__ __module__ __qualname__rr} q| dkrtd |t| d||}|S) a0Finds the period of a in modulo N arithmetic This is quantum part of Shor's algorithm. It takes two registers, puts first in superposition of states with Hadamards so: ``|k>|0>`` with k being all possible choices. It then does a controlled mod and a QFT to determine the order of a. g?r"cSsg|]}dqS)rr).0r:rrr szperiod_find..r rT)Z repetition)Z floatingPointz/Order finder returned 0. Happens with chance %f)r%mathZceilrr$rrr expandrr r r Z decompose isinstancerrr>rr=)r!r#epsilonrstartZfactorr)ZarrZ qbitArrayZcircuitr-registerr+r7grrrr4ys@        r4) r0rBr3ZsympyrrrrZsympy.core.numbersrZ sympy.ntheoryrr8Zsympy.utilities.iterablesrZsympy.physics.quantum.gater Zsympy.physics.quantum.qubitr r Zsympy.physics.quantum.qapplyr Zsympy.physics.quantum.qftr Zsympy.physics.quantum.qexprrrrr5r=r9r4rrrrs&        8