g1 DdZddlmZddlmZmZmZmZmZm Z m Z ddl Z ddl mZddlmZddlmZdeejjd eeefd ed eeeeejjffd Zdeejjd eeefd ed eeeeejjffd Zdeejjd eeefd ed eeeeejjffdZdeejjd eeefd ed eeeeejjffdZ ddeeefdeeeefde ed ejfdZde ejejfd ed efdZ de ejejfd ed efdZ!dS)a! A collection of functions that decide the layout of an output image. See :py:mod:`~qiskit.visualization.pulse_v2.types` for more info on the required data. There are 3 types of layout functions in this module. 1. layout.chart_channel_map An end-user can write arbitrary functions that output the custom channel ordering associated with group name. Layout function in this module are called with the `formatter` and `device` kwargs. These data provides stylesheet configuration and backend system configuration. The layout function is restricted to: ```python def my_channel_layout(channels: List[pulse.channels.Channel], formatter: Dict[str, Any], device: DrawerBackendInfo ) -> Iterator[Tuple[str, List[pulse.channels.Channel]]]: ordered_channels = [] # arrange order of channels for key, channels in my_ordering_dict.items(): yield key, channels ``` 2. layout.time_axis_map An end-user can write arbitrary functions that output the `HorizontalAxis` data set that will be later consumed by the plotter API to update the horizontal axis appearance. Layout function in this module are called with the `time_window`, `axis_breaks`, and `dt` kwargs. These data provides horizontal axis limit, axis break position, and time resolution, respectively. See py:mod:`qiskit.visualization.pulse_v2.types` for more info on the required data. ```python def my_horizontal_axis(time_window: Tuple[int, int], axis_breaks: List[Tuple[int, int]], dt: Optional[float] = None) -> HorizontalAxis: # write horizontal axis configuration return horizontal_axis ``` 3. layout.figure_title An end-user can write arbitrary functions that output the string data that will be later consumed by the plotter API to output the figure title. Layout functions in this module are called with the `program` and `device` kwargs. This data provides input program and backend system configurations. ```python def my_figure_title(program: Union[pulse.Waveform, pulse.Schedule], device: DrawerBackendInfo) -> str: return 'title' ``` An arbitrary layout function satisfying the above format can be accepted. ) defaultdict)ListDictAnyTupleIteratorOptionalUnionN)pulse)types)DrawerBackendInfochannels formatterdevicereturnc#Ktt}|D]*}|t||+g}|t jg}|t|d|t j g}|t|d|t j g}|t|d|drE|t j g} |t| d|D] }|j |gfV!dS)aLayout function for the channel assignment to the chart instance. Assign single channel per chart. Channels are grouped by type and sorted by index in ascending order. Stylesheet key: `chart_channel_map` For example: [D0, D2, C0, C2, M0, M2, A0, A2] -> [D0, D2, C0, C2, M0, M2, A0, A2] Args: channels: Channels to show. formatter: Dictionary of stylesheet settings. device: Backend configuration. Yields: Tuple of chart name and associated channels. c|jSNindexxs [/granary/edev/quantum/lib/python3.11/site-packages/qiskit/visualization/pulse_v2/layouts.pyz+channel_type_grouped_sort..w!'keyc|jSrrrs rrz+channel_type_grouped_sort..{rrc|jSrrrs rrz+channel_type_grouped_sort..rrcontrol.show_acquire_channelc|jSrrrs rrz+channel_type_grouped_sort..sagrN)rlisttypeappendgetr DriveChannelextendsortedControlChannelMeasureChannelAcquireChannelnameupper) rrrchan_type_dictchanordered_channelsd_chansc_chansm_chansa_chanss rchannel_type_grouped_sortr6Xs,!&&N00tDzz"))$////  !3R88GF70A0ABBBCCC  !5r::GF70A0ABBBCCC  !5r::GF70A0ABBBCCC/0H $$U%92>>w4E4E F F FGGG ((ioo$'''''((rc#NKtt}t}|D]D}|t||||jE|tj g}t|dd}|tj g}t|dd}|tj g}t|dd}|tj g} t| dd} g} t|D]7} t|dkr8|dj| kr'| |t|dkr8|dj| kr'| |t|dkr8|dj| kr'| ||d rKt| dkr8| dj| kr'| | 9| D] }|j|gfV!d S) a Layout function for the channel assignment to the chart instance. Assign single channel per chart. Channels are grouped by the same index and sorted by type. Stylesheet key: `chart_channel_map` For example: [D0, D2, C0, C2, M0, M2, A0, A2] -> [D0, D2, C0, C2, M0, M2, A0, A2] Args: channels: Channels to show. formatter: Dictionary of stylesheet settings. device: Backend configuration. Yields: Tuple of chart name and associated channels. c|jSrrrs rrz,channel_index_grouped_sort..AGrTrreversec|jSrrrs rrz,channel_index_grouped_sort..r9rc|jSrrrs rrz,channel_index_grouped_sort..r9rc|jSrrrs rrz,channel_index_grouped_sort..r9rrr!N)rr#setr$r%addrr&r r'r)r*r+r,lenpopr-r.) rrrr/indsr0r2u_chansr4r5r1inds rchannel_index_grouped_sortrGs,!&&N 55DtDzz"))$///   !3R88GW"3"3TBBBG  !5r::GW"3"3TBBBG  !5r::GW"3"3TBBBG  !5r::GW"3"3TBBBGd|| 7 7 w< [D0, D2, C0, C2, M0, M2, A0, A2] Args: channels: Channels to show. formatter: Dictionary of stylesheet settings. device: Backend configuration. Yields: Tuple of chart name and associated channels. c|jSrrrs rrz.channel_index_grouped_sort_u..r9rTr:c|jSrrrs rrz.channel_index_grouped_sort_u..r9rc|jSrrrs rrz.channel_index_grouped_sort_u..r9rc|jSrrrs rrz.channel_index_grouped_sort_u..r9rrrr?r!N)rr#r@r$r%rArr&r r'r)r+r,r*rBrCr(r-r.) rrrr/rDr0r2r4r5rEr1rFs rchannel_index_grouped_sort_urMsV.!&&N 55DtDzz"))$///   !3R88GW"3"3TBBBG  !5r::GW"3"3TBBBG  !5r::GW"3"3TBBBG  !5r::GW"3"3444Gd|| 7 7 w< [Q0, Q1, Q2] Args: channels: Channels to show. formatter: Dictionary of stylesheet settings. device: Backend configuration. Yields: Tuple of chart name and associated channels. c|dS)Nrrs rrz"qubit_index_sort..,s 1rrQdN) r rr, MemorySlot RegisterSlotrr# isinstanceget_qubit_indexr%r)items) rrr_removedqubit_channel_mapr0 sorted_mapqindchanss rqubit_index_sortr]s2 % ! #H $D))EE dH % %  &00667>>tDDDD)//11~~FFFJ!"" e$lllE!!!!!""r time_window axis_breaksdtc "|\}}|}|}g}d}|D]N\} } | |ks| |kr|| kr|| | z z}|| kr|| | z z}|| |z || | z z }Otjt|d|d} | } |D]'\} } | | z } tj| | k| | z| } (|r d}| |dzz} nd}d| D}t j||ftt| |||S)aLayout function for the horizontal axis formatting. Calculate axis break and map true time to axis labels. Generate equispaced 6 horizontal axis ticks. Convert into seconds if ``dt`` is provided. Args: time_window: Left and right edge of this graph. axis_breaks: List of axis break period. dt: Time resolution of system. Returns: Axis formatter object. rz Time (ns)eAzSystem cycle time (dt)cg|]}|dS)z.0frP).0vals r z"time_map_in_ns..bs:::#||:::r)windowaxis_mapaxis_break_poslabel) r%nplinspacemaxcopywherer HorizontalAxisdictzip)r^r_r`t0t1t0_shiftt1_shiftrjoffset_accumulationt0bt1baxis_loc axis_labeloffsetrkformatted_labels rtime_map_in_nsr2s"FBHHN))S 88sRxx  88 c !H 88 c !Hc$77888sSy({3x++Xq99HJQQSsXj3. V0CZPP  )b3h (::z:::O  (#c(O4455%    rprogramc:g}|d|j|jr |jdznd}|d|j|zdd|jrdnd|jd kr|d |jd |S) zLayout function for generating figure title. This layout writes program name, program duration, and backend name in the title. zName: rcg?z Duration: z.1f nsr`z no-backendz Backend: z, )r%r-r`duration backend_namejoin)rr title_strr`s r detail_titlerls I,gl,,---#I .S3B ['"2R"7[[[fi>YddUY[[\\\l**:V%8::;;; 99Y  rcdS)z5Layout function for generating an empty figure title.rP)rrs r empty_titlers 2rr)"__doc__ collectionsrtypingrrrrrr r numpyrlqiskitr qiskit.visualization.pulse_v2r )qiskit.visualization.pulse_v2.device_infor rChannelstrr6rGrMr]intfloatrqrWaveformSchedulerrrPrrrs>>@$#####DDDDDDDDDDDDDDDDDD//////GGGGGG/(5>)*/(7;CH~/(O`/( eCen45567/(/(/(/(d;(5>)*;(7;CH~;(O`;( eCen45567;(;(;(;(|<(5>)*<(7;CH~<(O`<( eCen45567<(<(<(<(~("5>)*("7;CH~("O`(" eCen45567("("("("X]a77sCx7/3E#s(O/D7JRSX/7 7777t % >? IZ _b    *u~u~=>HY^ar