g; dZddlZddlmZmZmZmZmZddlm Z m Z ddl m Z ddl mZmZdejdeeefd eeejejffd Z ddejdeeefd ee d eejfd Zd e_dejdeeefd eejfdZdejdeeefd eejfdZ ddejdeeefd ee d eejfdZd e_dejdeeefd eejfdZdejdeeefd eejfdZ dS)a A collection of functions that generate drawings from formatted input data. See :py:mod:`~qiskit.visualization.timeline.types` for more info on the required data. An end-user can write arbitrary functions that generate custom drawings. Generators in this module are called with the `formatter` kwarg. This data provides the stylesheet configuration. There are 4 types of generators in this module. 1. generator.gates In this stylesheet entry the input data is `types.ScheduledGate` and generates gate objects such as time buckets and gate name annotations. The function signature of the generator is restricted to: ```python def my_object_generator( gate: types.ScheduledGate, formatter: Dict[str, Any]) -> List[ElementaryData]: # your code here: create and return drawings related to the gate object. ``` If a generator object has the attribute ``accepts_program`` set to ``True``, then the generator will be called with an additional keyword argument ``program: QuantumCircuit``. 2. generator.bits In this stylesheet entry the input data is `types.Bits` and generates timeline objects such as zero line and name of bit associated with the timeline. The function signature of the generator is restricted to: ```python def my_object_generator( bit: types.Bits, formatter: Dict[str, Any]) -> List[ElementaryData]: # your code here: create and return drawings related to the bit object. ``` If a generator object has the attribute ``accepts_program`` set to ``True``, then the generator will be called with an additional keyword argument ``program: QuantumCircuit``. 3. generator.barriers In this stylesheet entry the input data is `types.Barrier` and generates barrier objects such as barrier lines. The function signature of the generator is restricted to: ```python def my_object_generator( barrier: types.Barrier, formatter: Dict[str, Any]) -> List[ElementaryData]: # your code here: create and return drawings related to the barrier object. ``` If a generator object has the attribute ``accepts_program`` set to ``True``, then the generator will be called with an additional keyword argument ``program: QuantumCircuit``. 4. generator.gate_links In this stylesheet entry the input data is `types.GateLink` and generates barrier objects such as barrier lines. The function signature of the generator is restricted to: ```python def my_object_generator( link: types.GateLink, formatter: Dict[str, Any]) -> List[ElementaryData]: # your code here: create and return drawings related to the link object. ``` If a generator object has the attribute ``accepts_program`` set to ``True``, then the generator will be called with an additional keyword argument ``program: QuantumCircuit``. Arbitrary generator function satisfying the above format can be accepted. Returned `ElementaryData` can be arbitrary subclasses that are implemented in the plotter API. N)ListUnionDictAnyOptional)QubitQuantumCircuit) CircuitError)typesdrawingsgate formatterreturnc  t|j}n#ttf$rd}YnwxYw |jjpd}n#t$rd}YnwxYw|jj||j|j|j |d tt|jj d}|d |jj|d}|j dkr|d||d|d d }|jjd krtjj}ntjj}t%j||j|j|j zgd |d zd|d zg|j|j||}nc|d||dddd}|d} |d} t%jtjj|j|j|jd| | |}|gS)aHGenerate time bucket or symbol of scheduled gate. If gate duration is zero or frame change a symbol is generated instead of time box. The face color of gates depends on the operand type. Stylesheet: - The `gate` style is applied for finite duration gate. - The `frame_change` style is applied for zero duration gate. - The `gate_face_color` style is applied for face color. Args: gate: Gate information source. formatter: Dictionary of stylesheet settings. Returns: List of `TextData` or `BoxData` drawings. zn/a, )namelabelbitst0durationunitary parameters color.gatescolor.default_gaterz layer.gatez alpha.gatezline_width.gate)zorder facecoloralpha linewidthdelayzbox_height.gate?) data_typexvalsyvalsbitmetastyleszlayer.frame_changeztext_size.frame_changecenterrcolorsizevahazunicode_symbol.frame_changezlatex_symbol.frame_change)r"r%xvalyvaltextlatexr')stroperand to_matrixAttributeErrorr rrrrrjoinmapparamsgetr BoxTypeDELAY SCHED_GATEr BoxData bit_positionTextData SymbolTypeFRAME) r rrrr&r*r'r"drawingunicode_symbol latex_symbols ^/granary/edev/quantum/lib/python3.11/site-packages/qiskit/visualization/timeline/generators.pygen_sched_gaterFqs?*dl,,..// L ) "+e  ! gMiiC)< = =>>  D m $ ( ():IFZ<[ \ \E }q -|,"#45    <  ' ' +II 0I"7DGdm34)$566iHY>Z8Z[ $+,     4567   ##@A !<= #&, $+,    9s&)??A A! A!programc @|jdkrd}d}n d}|d}|d|d|d|dd}d |jjd }|d |jj|}|jj}|} t |jd kr_%dd|jD} n&dfd|jD} |d| dz }| d| dz } g} |jjD]M} | t| d(#t$r| | YJwxYwd| } | r"|jjdkr|d| dz }| d| dz } |jdkr|d|jdz }| d|jdz } |jjdkrtj j }ntj j}tj||jd|jzz||j|j|| |}|gS)aGenerate gate name. Parameters and associated bits are also shown. Stylesheet: - `gate_name` style is applied. - `gate_latex_repr` key is used to find the latex representation of the gate name. Args: gate: Gate information source. formatter: Dictionary of stylesheet settings. program: Optional program that the bits are a part of. Returns: List of `TextData` drawings. rr(bottomlabel_offset.frame_changelayer.gate_namecolor.gate_nametext_size.gate_namer){\rm }latex_symbol.gatesNrc3^K|](}tt|ddV)dS)N)r2id).0r%s rE z%gen_full_gate_name..s7 H HsRWWbcc!2 H H H H H Hc3NK|]}|jV dSN)find_bitindex)rUr%rGs rErVz%gen_full_gate_name..s7 W WcG$4$4S$9$9$?!A W W W W W WrW[]z.2fr()r!r"r.r/r%r0r1r')rr3rr9lenrr6r8appendfloat ValueErrorr LabelTyper; GATE_NAMEr r?rr>)r rrGv_alignv_posr' default_name latex_name label_plain label_latexbits_strr8val params_strr"rBs ` rEgen_full_gate_namerps& }q56-.,-/0 F3T\.222L/044T\5F UUJ\&(K "K 49~~ ?yy H Hdi H H HHHHHyy W W W WTY W W WWWH8& 8& F|"$$ $ MMU3ZZ-- . . . . $ $ $ MMS( # # # # # $6""J)dl'722(:(((( (:((((  }q+4=++++ +4=++++  |G##O) O-  WsT]* *  Id' ( G 9s$D)) E  E Tc |jdkrd}d}n d}|d}|d|d|d|dd}d |jjd }|d |jj|}|jj}|}|jjd krtjj} ntjj} tj | |j d |jzz||j |j |||} | gS)ahGenerate gate name. Only operand name is shown. Stylesheet: - `gate_name` style is applied. - `gate_latex_repr` key is used to find the latex representation of the gate name. Args: gate: Gate information source. formatter: Dictionary of stylesheet settings. Returns: List of `TextData` drawings. rr(rIrJrKrLrMr)rNrOrPrr!r`) rr3rr9r rer;rfr r?rrr>) r rrgrhr'rirjrkrlr"rBs rEgen_short_gate_namerr*s$ }q56-.,-/0 F3T\.222L/044T\5F UUJ\&(K/K |G##O) O-  WsT]* *  Id' ( G 9rWr%c|d|d|d|dd}tjtjjtjjtjjgd|dzd|dzg|| }|gS) a Generate time slot of associated bit. Stylesheet: - `timeslot` style is applied. Args: bit: Bit object associated to this drawing. formatter: Dictionary of stylesheet settings. Returns: List of `BoxData` drawings. zlayer.timeslotzalpha.timeslotzline_width.timeslotzcolor.timeslot)rrrrr zbox_height.timeslotr!r"r#r$r%r')r r=r r:TIMELINEAbstractCoordinateLEFTRIGHT)r%rr'rBs rE gen_timeslotryfs,-+,45/0 F-(',e.F.LMi 566iH]>^8^_  G 9rWc 6|d|d|dddd}|6tjdd t|trd nd }d |d}n||}|jrI|jddj}d |jddjd|jddd}n't|trd nd }d |d|jd}tj tj j tjjd||||}|gS)a7Generate bit label. Stylesheet: - `bit_name` style is applied. Args: bit: Bit object associated to this drawing. formatter: Dictionary of stylesheet settings. program: Optional program that the bits are a part of. Returns: List of `TextData` drawings. zlayer.bit_namezcolor.bit_nameztext_size.bit_namer(rightr)Nz;bits cannot be accurately named without passing a 'program') stacklevelqcrNrOrz}_{rQr`)warningswarn isinstancerrZ registersrprefixr[r r?r reBIT_NAMErvrw)r%rrGr'rkrllocrBs rE gen_bit_namersU&,-+,./ F S`abbbb'U33)rrr'rBs rE gen_barrierrs?+O,3434?+ F.(z7:&Sk L- . G 9rWlinkc|d|j|d}|d|d|d|d|d}tj|j|j|}|gS) a\Generate gate link line. Line color depends on the operand type. Stylesheet: - `gate_link` style is applied. - The `gate_face_color` style is applied for line color. Args: link: Gate link object. formatter: Dictionary of stylesheet settings. Returns: List of `GateLinkData` drawings. rrzalpha.gate_linkzlayer.gate_linkzline_width.gate_linkzline_style.gate_linkr)rr.r')r9opnamer GateLinkDatarr)rrr*r'rBs rE gen_gate_linkrs$ m $ ( (i@T6U V VE,--.5656 F#PPPG 9rWrY)!__doc__rtypingrrrrrqiskit.circuitrr qiskit.circuit.exceptionsr qiskit.visualization.timeliner r ScheduledGater2r?r=rFrpaccepts_programrrBitsryrBarrierrrGateLinkrrrWrErsZZx333333333333330000000022222299999999Y  YCH~Y %!8#33 45YYYYz_cWW  W*.sCx.WCKNC[W ( WWWWt&*"9  9*.sCx.9 ( 9999xejT#s(^XEU@VD)-11 1CH~1n %1 (  1111h $ 4S>d8K\F]@4S>d8K`FarW