g$dZddlmZddlmZddlmZmZmZm Z m Z ddl Z ddl Z ddlmZddlmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZedZ d%dZ!d&dZ"d'dZ#d(d)d"Z$d*d#Z%d+d$Z&dS),z=This module contains common utils for disjoint coupling maps.) annotations) defaultdict)ListCallableTypeVarDictUnionN) DAGOpNode)QubitBarrierClbit) DAGCircuit) DAGOutNode) CouplingMap)Target)TranspilerError) vf2_utilsTdagrcomponents_sourceUnion[Target, CouplingMap]run_func&Callable[[DAGCircuit, CouplingMap], T]returnList[T]ct|tr|d}n|}|}t |dkrR||dkrtd||dgSt|}t||}g}| D]\}} || } | D]} || }| |j | |j|jD]} | | |jD]} | | | ||j |j|| ||fވfd|D}|S)zz6run_pass_over_connected_components..Hs) L L L}w88GT " " L L L) isinstancerbuild_coupling_mapconnected_componentslen num_qubitssizer separate_dagmap_componentsitemspop add_qubitsr! add_clbitsr"qregsadd_qregcregsadd_cregcomposeappend)rrr coupling_mapcmap_componentsdag_componentsmapped_componentsout_component_pairs cmap_indexdagsr& dag_indexqregcregress ` r("run_pass_over_connected_componentsrH!s #V,,)(;;t;TT ( "7799O ?q  >>  oa05577 7 7!2 oa01122!#&&N&~GG-3355KK D !, G GI +C   sz * * *   sz * * *  ' '  &&&&  ' '  &&&& OOC 3:O F F F F""G_Z-H#IJJJJ L L L L8K L L LC Jr*r?List[DAGCircuit]r>List[CouplingMap]Dict[int, List[int]]cdt|Dtt}tt|ddD]\}}tt t |fddD]_}||kr?||||xx|zcc<n`td|S)zReturns a map where the key is the index of each connected component in cmap_components and the value is a list of indices from dag_components which should be placed onto it.c>i|]\}}|t|jSr$)r.graph)r%indexr's r( z"map_components..Qs&XXXkeT5#dj//XXXr*c6|dS)Nr)r/)xs r(z map_components..Us11B1Br*T)keyreversec|S)Nr$)rO free_qubitss r(rSz map_components..Xs ;u;Mr*r ) enumeraterlistsortedranger.r/r<r)r?r> out_mappingrDrrBrWs @r(r2r2Ls0 YXY=W=WXXXKd##K .!!'B'BD 3! #o&& ' '-M-M-M-MW[    J ~~;z#:::J'..y999J'''3>>+;+;;'''; "2  r*c |tD]}t|j}|dkr|jr#|jjdt j}ndt j}t}| dt|Dt|D]5}| td||j |fd6| ||dS) zAMutate an input dag to split barriers into single qubit barriers.r_uuid= _none_uuid=c*g|]}tSr$)r )r%_s r(r)z"split_barriers..ssAAA!eggAAAr*labelF)qargscheckN)op_nodesr r.rdrcopuuiduuid4rr5r[apply_operation_backr!substitute_node_with_dag)rnoder/ barrier_uuid split_dagis r(split_barriersrphs% W%%66__ ??  : 8"gmAA4:<<AALL777LLL AAuZ/@/@AAABBBz""  A  * *... '*, +     $$T95555!66r*T retain_uuidboolcdt|jD}i}|tD]}|jrd|jvr|j}n||vr||}t |jt |jz}|st|jtr|j drd}n^t|jtr?d|jvr6d |j ddd}n|}n|}t||} | ||g| |} | ||<|||<!dS)zLMutate input dag to combine barriers with UUID labels into a single barrier.ci|]\}}|| Sr$r$)r%rObits r(rPz$combine_barriers..sHHHJE3S%HHHr*r^r_Nrb) rXr!rfr rcr.rdr+str startswithjoinsplitreplace_block_with_op) rrq qubit_indicesuuid_maprlrm other_noder/rcnew_opnew_nodes r(combine_barriersr}swHH)CJ2G2GHHHM%'H W%%.. : .4:%%#z x''%l3  !122S__D ")!$*c22-tz7L7L]7[7[- $#DJ44-TZ9O9O ( dj.>.>x.H.H".M N N ,(E 599944dJ5GQ^__)1&&)-&-..r*c bt|tr|d}n|}d|D}|D]}d}t |D]0\}}||jdj|vr|}n1||jdj||vr\td||jdjd||jdjd dS) aC Check that the layout of the dag does not require connectivity across connected components in the CouplingMap Args: dag: DAGCircuit to check. components_source: Target to check against. Raises: TranspilerError: Chosen layout is not valid for the target disjoint connectivity. TrcZg|](}t|j)Sr$)setrNnodes)r%rRs r(r)z8require_layout_isolated_to_component..s*XXXqc!'--//**XXXr*NrrzThe circuit has an invalid layout as two qubits need to interact in disconnected components of the coupling map. The physical qubit z" needs to interact with the qubit z( and they belong to different components) r+rr,r- two_qubit_opsrXfind_bitrdrOr)rrr=component_setsinstcomponent_indexro component_sets r($require_layout_isolated_to_componentrs^#V,,)(;;t;TT ( XXL4U4U4W4WXXXN!!##   ). 9 9   A}||DJqM**0MAA"#B << 1 & & ,N?4S S S!e<< 1 ..4eedjm44:eee  T  r*ct|tj|\}}}tj|}g}|D]#}|fd|D$t |j}g}|D]9} |} | j || z d| _ | D]D} | | j r(| | j| j | jdEg} | jD]k\} } t'| | }t+|t,r/t+|jt0r| | l| j| t5| || ;t5|d|S)z6Separate a dag circuit into it's connected components.c h|] }| Sr$r$)r%rR qubit_maps r( zseparate_dag..s A A A!1 A A Ar*rF)re)rq)rprbuild_interaction_graphrxweakly_connected_componentsr<rr!copy_empty_like remove_qubits global_phasetopological_op_nodes issupersetrdrjrgcargs input_mapr3next successorsr+rwirer remove_clbitsr)rim_graphra__r-component_qubits componentr!decomposed_dags dag_qubitsnew_dagrl idle_clbitsru succ_noders @r(r1r1s3!*!B3!G!GHaB9(CC)CC  A A A Ay A A ABBBB __FO&(( %%''v 233 ,,.. [ [D$$TZ00 [,,TWdj$*TY,ZZZ  *0022 ( (ICW//5566I)Z00 (Z PU5V5V (""3'''{++!!!w''''Se,,,, r*)rrrrrrrr)r?rIr>rJrrK)rr)T)rrrqrr)rrrr)rrrrI)'__doc__ __future__r collectionsrtypingrrrrr rh rustworkxrqiskit.dagcircuitr qiskit.circuitr r r qiskit.dagcircuit.dagcircuitrqiskit.dagcircuit.dagnoderqiskit.transpiler.couplingrqiskit.transpiler.targetrqiskit.transpiler.exceptionsrqiskit.transpiler.passes.layoutrrrHr2rprrr1r$r*r(rsDC""""""######77777777777777 ''''''0000000000333333000000222222++++++888888555555 GCLL((((V86666*.....:Dr*