g(hdZddlmZddlmZddlmZddlmZm Z ddl m Z m Z GddZ d S) a Channel event manager for pulse schedules. This module provides a `ChannelEvents` class that manages a series of instructions for a pulse channel. Channel-wise filtering of the pulse program makes the arrangement of channels easier in the core drawer function. The `ChannelEvents` class is expected to be called by other programs (not by end-users). The `ChannelEvents` class instance is created with the class method ``load_program``: .. code-block:: python event = ChannelEvents.load_program(sched, DriveChannel(0)) The `ChannelEvents` is created for a specific pulse channel and loosely assorts pulse instructions within the channel with different visualization purposes. Phase and frequency related instructions are loosely grouped as frame changes. The instantaneous value of those operands are combined and provided as ``PhaseFreqTuple``. Instructions that have finite duration are grouped as waveforms. The grouped instructions are returned as an iterator by the corresponding method call: .. code-block:: python for t0, frame, instruction in event.get_waveforms(): ... for t0, frame_change, instructions in event.get_frame_changes(): ... The class method ``get_waveforms`` returns the iterator of waveform type instructions with the ``PhaseFreqTuple`` (frame) at the time when instruction is issued. This is because a pulse envelope of ``Waveform`` may be modulated with a phase factor $exp(-i \omega t - \phi)$ with frequency $\omega$ and phase $\phi$ and appear on the canvas. Thus, it is better to tell users in which phase and frequency the pulse envelope is modulated from a viewpoint of program debugging. On the other hand, the class method ``get_frame_changes`` returns a ``PhaseFreqTuple`` that represents a total amount of change at that time because it is convenient to know the operand value itself when we debug a program. Because frame change type instructions are usually zero duration, multiple instructions can be issued at the same time and those operand values should be appropriately combined. In Qiskit Pulse we have set and shift type instructions for the frame control, the set type instruction will be converted into the relevant shift amount for visualization. Note that these instructions are not interchangeable and the order should be kept. For example: .. code-block:: python sched1 = Schedule() sched1 = sched1.insert(0, ShiftPhase(-1.57, DriveChannel(0)) sched1 = sched1.insert(0, SetPhase(3.14, DriveChannel(0)) sched2 = Schedule() sched2 = sched2.insert(0, SetPhase(3.14, DriveChannel(0)) sched2 = sched2.insert(0, ShiftPhase(-1.57, DriveChannel(0)) In this example, ``sched1`` and ``sched2`` will have different frames. On the drawer canvas, the total frame change amount of +3.14 should be shown for ``sched1``, while ``sched2`` is +1.57. Since the `SetPhase` and the `ShiftPhase` instruction behave differently, we cannot simply sum up the operand values in visualization output. It should be also noted that zero duration instructions issued at the same time will be overlapped on the canvas. Thus it is convenient to plot a total frame change amount rather than plotting each operand value bound to the instruction. ) annotations) defaultdict)Iterator)pulsecircuit)PhaseFreqTuplePulseInstructionceZdZdZejjejjejjfZ ejj ejj ejj ejj fZddZedd ZddZddZddZed dZdS)! ChannelEventszChannel event manager. waveformsdict[int, pulse.Instruction]frames"dict[int, list[pulse.Instruction]]channelpulse.channels.ChannelcZ||_||_||_d|_d|_d|_dS)aCreate new event manager. Args: waveforms: List of waveforms shown in this channel. frames: List of frame change type instructions shown in this channel. channel: Channel object associated with this manager. N) _waveforms_framesr _init_phase_init_frequency_dt)selfr rrs Z/granary/edev/quantum/lib/python3.11/site-packages/qiskit/visualization/pulse_v2/events.py__init__zChannelEvents.__init__hs8$  "programpulse.SchedulecBi}tt}||gjD]\\}}t ||jr|jdkr&|||<,t ||jr|||]t|||S)aLoad a pulse program represented by ``Schedule``. Args: program: Target ``Schedule`` to visualize. channel: The channel managed by this instance. Returns: ChannelEvents: The channel event manager for the specified channel. )channelsr) rlistfilter instructions isinstance_waveform_groupduration _frame_groupappendr )clsrrr rt0insts r load_programzChannelEvents.load_programs T""  ::G ( (HB$ 344 (=A%% $ " D#"233 (r !!$'''Y888rdtfloatinit_frequency init_phasec<|pd|_|pd|_|pd|_dS)zSetup system status. Args: dt: Time resolution in sec. init_frequency: Modulation frequency in Hz. init_phase: Initial phase in rad. g?rN)rrr)rr-r/r0s r set_configzChannelEvents.set_configs/9-4%,rreturnIterator[PulseInstruction]c#Kt|jdd}t|jd}|j}|j}|D]|\}}d}t |dkrn|dd|kr\|\}} t | || \}}t |dkr|dd|k\t|tj r1t|d |jD}t|tj r1t|d |jD}t!||} t|t"jr|j}t)||j| ||V~d S) z-Return waveform type instructions with frame.c|dSNrxs rz-ChannelEvents.get_waveforms.. !A$rT)keyreversec|dSr7r8r9s rr;z-ChannelEvents.get_waveforms..s 1rr=Fr frame_changesphase frequencyci|]}|dSrr8.0params r z/ChannelEvents.get_waveforms..)Q)Q)Qu%)Q)Q)Qrci|]}|dSrGr8rHs rrKz/ChannelEvents.get_waveforms..1]1]1]u%1]1]1]rN)sortedritemsrrrlenpopr _calculate_current_framer$rParameterExpressionr.bind parametersrrPlayis_parameterizedr r) rsorted_frame_changessorted_waveformsrDrEr*r+ is_opaque_rCframes r get_waveformszChannelEvents.get_waveformss%dl&8&8&:&:X\]]]!$/"7"7"9"9~~NNN ( ( I IHBI*++a//4H4LQ4OSU4U4U#7#;#;#=#= =#0#I#I"/u $J$$ y*++a//4H4LQ4OSU4U4U%!<== Tejj)Q)Q@P)Q)Q)QRRSS)W%@AA `!)..1]1] H\1]1]1]"^"^__ "5)44E$ ++ : J7799 "2txiHH H H H H+ I Irc#hKt|jd}|j}|j}|D]\}}d}|}|}t |||\}}t||z ||z } t|tj r3t| d|j D}d}t|tj r3t| d|j D}d}t||j| ||VdS) zEReturn frame change type instructions with total frame change amount.c|dSr7r8r9s rr;z1ChannelEvents.get_frame_changes..r<rr@FrBci|]}|dSrGr8rHs rrKz3ChannelEvents.get_frame_changes..rLrTci|]}|dSrGr8rHs rrKz3ChannelEvents.get_frame_changes..rNrN)rOrrPrrr rSrr$rrTr.rUrVr r) rrYrDrEr*rCr[ pre_phase pre_frequencyr]s rget_frame_changeszChannelEvents.get_frame_changess[ &dl&8&8&:&:OOO ( !5 R R B II%M,EE+5I F   E9 #59#4i-6OPPE%!<== !ejj)Q)Q@P)Q)Q)QRRSS )W%@AA !!)..1]1] H\1]1]1]"^"^__  "2tx yQQ Q Q Q Q) R RrrC$list[pulse.instructions.Instruction]rDrEtuple[float, float]cV|D]}t|tjjr|j})t|tjjr ||jz }St|tjjr|j}zt|tjjr ||jz }||fS)aCalculate the current frame from the previous frame. If parameter is unbound phase or frequency accumulation with this instruction is skipped. Args: frame_changes: List of frame change instructions at a specific time. phase: Phase of previous frame. frequency: Frequency of previous frame. Returns: Phase and frequency of new frame. ) r$rr# SetFrequencyrEShiftFrequencySetPhaserD ShiftPhase)r)rCrDrE frame_changes rrSz&ChannelEvents._calculate_current_frames"* , ,L,(:(GHH ,(2 L%*<*KLL ,\33 L%*<*EFF ,$*L%*<*GHH ,++irN)r r rrrr)rrrr)r-r.r/r.r0r.)r3r4)rCrfrDr.rEr.r3rg)__name__ __module__ __qualname____doc__rr#rWDelayAcquirer%rirjrkrlr'r classmethodr,r2r^rerSr8rrr r Ys      "O ' ) # % L0999[92 - - - -IIII>RRRR<   [   rr N)rq __future__r collectionsrcollections.abcrqiskitrr#qiskit.visualization.pulse_v2.typesrr r r8rrrzsCCH#"""""######$$$$$$!!!!!!!!PPPPPPPPe e e e e e e e e e r