g*0dZddlmZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZmZdd lmZed ed iZejjejjdd ZdZejjejjddZdZdZdS)z Visualization function for a pass manager. Passes are grouped based on their flow controller, and coloured based on the type of pass. ) annotationsN) optionals)BaseController GenericPass)FlowControllerLinear) AnalysisPassTransformationPass)VisualizationErrorredblueFc"ddl}|st}|}d}d}t|jD]1\}} t | ||||\} }}|| 2t|||} | S)a Draws the pass manager. This function needs `pydot `__, which in turn needs `Graphviz `__ to be installed. Args: pass_manager (PassManager): the pass manager to be drawn filename (str): file path to save image to style (dict or OrderedDict): keys are the pass classes and the values are the colors to make them. An example can be seen in the DEFAULT_STYLE. An ordered dict can be used to ensure a priority coloring when pass falls into multiple categories. Any values not included in the provided dict will be filled in from the default dict raw (Bool) : True if you want to save the raw Dot output not an image. The default is False. Returns: PIL.Image or None: an in-memory representation of the pass manager. Or None if no image was generated or PIL is not installed. Raises: MissingOptionalLibraryError: when nxpd or pydot not installed. VisualizationError: If raw=True and filename=None. Example: .. code-block:: from qiskit import QuantumCircuit from qiskit.transpiler import generate_preset_pass_manager from qiskit.visualization import pass_manager_drawer pm = generate_preset_pass_manager(optimization_level=0) pass_manager_drawer(pm) rN) pydot DEFAULT_STYLEDot enumerateto_flow_controllertasks draw_subgraph add_subgraph make_output) pass_managerfilenamestylerawrgraph component_id prev_nodeindexcontroller_groupsubgraphoutputs e/granary/edev/quantum/lib/python3.11/site-packages/qiskit/visualization/pass_manager_visualization.pypass_manager_drawerr$ sHLLL  IIKKE LI#,\-L-L-N-N-T#U#U%%,9 lE9e- - ),  8$$$$ X . .F Mc|D]\}}t||r|cStD]\}}t||r|cSdS)Nblack)items isinstancer)pssrtypcolors r#_get_node_colorr-]skkmm U c3   LLL $))++ U c3   LLL  7r%c ddl}ttd|j}|st}|}d}d}d} |D]} t || } | |t| dt| d} | j D]3} t| ||| |\}}} | ||dz }4| | t|||}|S)a Draws the staged pass manager. This function needs `pydot `__, which in turn needs `Graphviz `__ to be installed. Args: pass_manager (StagedPassManager): the staged pass manager to be drawn filename (str): file path to save image to style (dict or OrderedDict): keys are the pass classes and the values are the colors to make them. An example can be seen in the DEFAULT_STYLE. An ordered dict can be used to ensure a priority coloring when pass falls into multiple categories. Any values not included in the provided dict will be filled in from the default dict raw (Bool) : True if you want to save the raw Dot output not an image. The default is False. Returns: PIL.Image or None: an in-memory representation of the pass manager. Or None if no image was generated or PIL is not installed. Raises: MissingOptionalLibraryError: when nxpd or pydot not installed. VisualizationError: If raw=True and filename=None. Example: .. code-block:: %matplotlib inline from qiskit.providers.fake_provider import GenericBackendV2 from qiskit.transpiler.preset_passmanagers import generate_preset_pass_manager pass_manager = generate_preset_pass_manager(3, GenericBackendV2(num_qubits=5)) pass_manager.draw() rNc |duS)N)ss r#z,staged_pass_manager_drawer..s 1D=r% helveticalfontnamelabel labeljustr )rlistfilterexpanded_stagesrrgetattrClusterstrrrrrr)rrrrrstagesrridxrststage stagegraphr r!r"s r#staged_pass_manager_drawerrDks4HLLL&00,2NOO P PF  IIKKE L CI + + b))  s2wwCPRGG_bccJ$)$<$<$>$>$D   4A$lE9c551, ''111q   z * * * X . .F Mr%c ddl}d|d}t|tr%t|ts||jjz }|t|d|d}|dz }t|trt|d g}n=t|tr|g}n$t|ttfr|}n|||fSg} |D]G} t| tr| | j 2| | H| D]} t| tr(|t|d dd d } nT|t|t!| |dtt#| jd } || |dz }t'j| j} | ddd} | drt-| dnd}t/| D]~\}}d}|t-| |z krd}||ddd|d|}|||dz }|||| |r)|||| | }|||fS)zDraw subgraph.rN[z] r3r4r5r rkzNested flow controller rectangle)r,r6r7shapesoliddashedr' ellipse)r,r6fontsizer7rIr)rr)rr __class____name__r=r>r<rr9tupleextendrappendNoder-typeadd_nodeinspectgetfullargspec__init__lenradd_edgeEdge)r rrrr@rr7r!r flatten_taskstasknodearg_specargs num_optional arg_indexargnd_style input_nodes r#rrsGLLL KKKE"N33:J.==: $.799}}S..E]`}aaHAL"N33 1('266 $k 2 21!" $tUm 4 41 y00M'' d0 1 1 '   , , , ,   & & & &88 dN + + ::L!!$.! DD::L!!%dE22$$t**-..! D $ )$-88{122+3A;=s8A;'''A (oo < >     7## '  ( JJx ' ' '                  sA>DDD)NNF)__doc__ __future__rrtrXrq qiskit.utilsrrmqiskit.passmanager.base_tasksrr#qiskit.passmanager.flow_controllersrqiskit.transpiler.basepassesrr exceptionsr r HAS_GRAPHVIZrequire_in_call HAS_PYDOTr$r-rDrrr0r%r#rsd#""""" 000000EEEEEEEEDDDDDDIIIIIIII******u&8&A  ( %888&%)(8v    ( %EEE&%)(EP`-`-`-Fr%