o jg:6@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZgd Zd d Zdd dZdddZddZd ddZddZd!ddZd!ddZddZdS)"z1Primitive circuit operations on quantum circuits.)reduce)default_sort_key)Tuple)Mul)Symbolsympify)numbered_symbols)Gate) kmp_tablefind_subcircuitreplace_subcircuitconvert_to_symbolic_indicesconvert_to_real_indices random_reduce random_insertcCsd}d}g}|d|d|t|krF||d||kr.|d}|||d}n|dkr7||}n |d|d}|t|ks|S)zBuild the 'partial match' table of the Knuth-Morris-Pratt algorithm. Note: This is applicable to strings or quantum circuits represented as tuples. r)appendlen)wordposcndtablerZ/var/www/html/zoom/venv/lib/python3.10/site-packages/sympy/physics/quantum/circuitutils.pyr s          r cCst|tr|j}t|tr|j}t|dkst|t|kr dS|dkr(t|}|}d}t|}|||krg|||||krE|d}n||||}||dkrW||nd}|t|kra|S|||ks6dS)aFinds the subcircuit in circuit, if it exists. Explanation =========== If the subcircuit exists, the index of the start of the subcircuit in circuit is returned; otherwise, -1 is returned. The algorithm that is implemented is the Knuth-Morris-Pratt algorithm. Parameters ========== circuit : tuple, Gate or Mul A tuple of Gates or Mul representing a quantum circuit subcircuit : tuple, Gate or Mul A tuple of Gates or Mul to find in circuit start : int The location to start looking for subcircuit. If start is the same or past end, -1 is returned. end : int The last place to look for a subcircuit. If end is less than 1 (one), then the length of circuit is taken to be end. Examples ======== Find the first instance of a subcircuit: >>> from sympy.physics.quantum.circuitutils import find_subcircuit >>> from sympy.physics.quantum.gate import X, Y, Z, H >>> circuit = X(0)*Z(0)*Y(0)*H(0) >>> subcircuit = Z(0)*Y(0) >>> find_subcircuit(circuit, subcircuit) 1 Find the first instance starting at a specific position: >>> find_subcircuit(circuit, subcircuit, start=1) 1 >>> find_subcircuit(circuit, subcircuit, start=2) -1 >>> circuit = circuit*subcircuit >>> find_subcircuit(circuit, subcircuit, start=2) 4 Find the subcircuit within some interval: >>> find_subcircuit(circuit, subcircuit, start=2, end=2) -1 rrr) isinstancerargsrr )circuit subcircuitstartendrindexrrrrr 8s( 8      r NcCs|dkrd}t|tr|j}t|tr|j}t|tr|j}n|dur%d}t|||d}|dkrH|d|}||t|t|}|||}|S)aReplaces a subcircuit with another subcircuit in circuit, if it exists. Explanation =========== If multiple instances of subcircuit exists, the first instance is replaced. The position to being searching from (if different from 0) may be optionally given. If subcircuit cannot be found, circuit is returned. Parameters ========== circuit : tuple, Gate or Mul A quantum circuit. subcircuit : tuple, Gate or Mul The circuit to be replaced. replace : tuple, Gate or Mul The replacement circuit. pos : int The location to start search and replace subcircuit, if it exists. This may be used if it is known beforehand that multiple instances exist, and it is desirable to replace a specific instance. If a negative number is given, pos will be defaulted to 0. Examples ======== Find and remove the subcircuit: >>> from sympy.physics.quantum.circuitutils import replace_subcircuit >>> from sympy.physics.quantum.gate import X, Y, Z, H >>> circuit = X(0)*Z(0)*Y(0)*H(0)*X(0)*H(0)*Y(0) >>> subcircuit = Z(0)*Y(0) >>> replace_subcircuit(circuit, subcircuit) (X(0), H(0), X(0), H(0), Y(0)) Remove the subcircuit given a starting search point: >>> replace_subcircuit(circuit, subcircuit, pos=1) (X(0), H(0), X(0), H(0), Y(0)) >>> replace_subcircuit(circuit, subcircuit, pos=2) (X(0), Z(0), Y(0), H(0), X(0), H(0), Y(0)) Replace the subcircuit: >>> replacement = H(0)*Z(0) >>> replace_subcircuit(circuit, subcircuit, replace=replacement) (X(0), H(0), Z(0), H(0), X(0), H(0), Y(0)) rNr)r!r)rrrr r)rr replacerlocleftrightrrrr s 8     r cCs"i}|D] }t||||<q|SNr)mappingnew_mapkeyrrr_sympify_qubit_mapsr,cCst|tr|j}tddd}t|}i}dd}|dur+t|ts)d|}t||}|dur@t|tjs>d|}t||}|durUt|tsSd d |}t||}t |}||} d } |D]e} t| t rt | j|||d } | \} }}}| |||} n6t| t tfrt | |||d } | \} }}}| |||} n| | vr| | } nt|}| ||<|| | <|} t| t r| j| } | | f} qa| |||fS) aLReturns the circuit with symbolic indices and the dictionary mapping symbolic indices to real indices. The mapping is 1 to 1 and onto (bijective). Parameters ========== seq : tuple, Gate/Integer/tuple or Mul A tuple of Gate, Integer, or tuple objects, or a Mul start : Symbol An optional starting symbolic index gen : object An optional numbered symbol generator qubit_map : dict An existing mapping of symbolic indices to real indices All symbolic indices have the format 'i#', where # is some number >= 0. ir)prefixr!cSsdd}tt||S)NcSs|d|dfS)Nrrr)itemrrr szIconvert_to_symbolic_indices..create_inverse_map..)dictmapitems)symb_to_real_map rev_itemsrrrcreate_inverse_map sz7convert_to_symbolic_indices..create_inverse_mapNz+Expected Symbol for starting index, got %r.zExpected a generator, got %r.z$Expected dict for existing map, got z%r.r) qubit_mapr!gen)rrrr nextr TypeError __class__r1r,r rupdatetupler)seqr!r8r7 index_gencur_ndxndx_mapr6msginv_mapsym_seqr/resultsym_itemr*rrrrsn                rcCst|tr|j}t|tsd|}t|t|}d}|D].}t|tr+t|j|}nt|tt fr8t||}n||}t|trF|j |}||f}q|S)aReturns the circuit with real indices. Parameters ========== seq : tuple, Gate/Integer/tuple or Mul A tuple of Gate, Integer, or tuple objects or a Mul qubit_map : dict A dictionary mapping symbolic indices to real indices. Examples ======== Change the symbolic indices to real integers: >>> from sympy import symbols >>> from sympy.physics.quantum.circuitutils import convert_to_real_indices >>> from sympy.physics.quantum.gate import X, Y, H >>> i0, i1 = symbols('i:2') >>> index_map = {i0 : 0, i1 : 1} >>> convert_to_real_indices(X(i0)*Y(i1)*H(i0)*X(i1), index_map) (X(0), Y(1), H(0), X(1)) z$Expected dict for qubit_map, got %r.r) rrrr1r:r,r rr=rr;)r>r7rBreal_seqr/ real_itemrrrrMs"       rcCspddlm}|s |St|tr|j}t|}||}|r1|t|}||}t||dkr/n|s|St ||S)aShorten the length of a quantum circuit. Explanation =========== random_reduce looks for circuit identities in circuit, randomly chooses one to remove, and returns a shorter yet equivalent circuit. If no identities are found, the same circuit is returned. Parameters ========== circuit : Gate tuple of Mul A tuple of Gates representing a quantum circuit gate_ids : list, GateIdentity List of gate identities to find in circuit seed : int or list seed used for _randrange; to override the random selection, provide a list of integers: the elements of gate_ids will be tested in the order given by the list r _randranger) sympy.core.randomrJrrr flatten_idsrpopr r )rgate_idsseedrJids randranger-idrrrrs     rcCshddlm}|s |St|tr|j}||}|t|d}||t|}t|}||||<t|S)aInsert a circuit into another quantum circuit. Explanation =========== random_insert randomly chooses a location in the circuit to insert a randomly selected circuit from amongst the given choices. Parameters ========== circuit : Gate tuple or Mul A tuple or Mul of Gates representing a quantum circuit choices : list Set of circuit choices seed : int or list seed used for _randrange; to override the random selections, give a list two integers, [i, j] where i is the circuit location where choice[j] will be inserted. Notes ===== Indices for insertion should be [0, n] if n is the length of the circuit. rrIr)rKrJrrrrlistr=)rchoicesrOrJrQr%choicerrrrs   rcCs$dd}t||g}|jtd|S)NcSs|t|jtdS)Nr+)sortedequivalent_idsr)accan_idrrrr0szflatten_ids..rV)rsortr)rPcollapserrrrLs  rL)rr)Nr)NNNr()__doc__ functoolsrsympy.core.sortingrsympy.core.containersrsympy.core.mulrsympy.core.symbolrsympy.core.sympifyrsympy.utilitiesr sympy.physics.quantum.gater __all__r r r r,rrrrrLrrrrs&         XU a 5 2 /