gcdZddlZddlZddlZddlZddlmZddlZddl Z ddl m Z ddl mZmZmZddlmZddlmZmZmZddlmZdd lmZdd lmZmZdd lmZd Z d Z!dZ"dZ#dZ$d&dZ%d'dZ&d'dZ'dZ(d'dZ)d'dZ*d'dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2d Z3d!Z4d"Z5d#Z6e d(d$Z7ddej8fd%Z9dS))z2Read and write schedule and schedule instructions.N)BytesIO) QiskitError)librarychannels instructions) ScheduleBlock)formatscommon type_keys)value)QpyError)Kernel Discriminator)!ignore_pulse_deprecation_warningsctj|}tj||i}tj|}||S)N)r read_type_keyr read_valuer ScheduleChannelretrieve)file_objversiontype_keyindex channel_clss T/granary/edev/quantum/lib/python3.11/site-packages/qiskit/qpy/binary_io/schedules.py _read_channelr"sL#H--H  Xw 3 3E+44X>>K ;u  ctjtjtj|tj}||j}tj |tj }tj||i}tj|||j|jS)N)samplesnameepsilonlimit_amplitude)r WAVEFORM_makestructunpack WAVEFORM_PACKread WAVEFORM_SIZE data_sizer data_from_binarynploadr rrWaveformr! amp_limited)rrheader samples_rawrr s r_read_waveformr2+s   # #  ! MM'/ 0 0  F -- 011K%k27;;G  Hgr 2 2D   *    rc`|dkrFt|5}tj|t||cdddS#1swxYwYdS|dkrFt|5}tj|t||cdddS#1swxYwYdSt j||||S)zWraps `value.loads_value` to deserialize binary data to dictionary or list objects which are not supported by `value.loads_value`. Dr deserializerrvectorsNl)rr read_mapping _loads_obj read_sequencer loads_value)r binary_datarr7 containers rr:r:>sQ4 [ ! ! Y&"WV]                   T   [ ! ! Y'"WV]                    ;IIIs#AAA"B  BBctj|t|i}tj||i}t dd|i|SNr5r )r r9r:r rrrrparamsr s r _read_kernelrDPsT   F  Hgr 2 2D  & &t &v & &&rctj|t|i}tj||i}t dd|i|Sr@)r r9r:r rrrBs r_read_discriminatorrF[sT   F  Hgr 2 2D  - -d -f - --rFc|dkrdStj|}|rtj|Sddlm}|st d|tj}||}tj |S)Nrr) parse_exprzThis payload can not be loaded unless you set ``trust_payload`` to True, as it's using sympy for serialization of symbolic expressions which is insecure.) zlib decompressr load_symengine_payloadsympyrHr decodeENCODEsymsympify) expr_bytes use_symengine trust_inputrHexpr_txtexprs r_loads_symbolic_exprrVfsSt,,J !,Z888$$$$$$   $$V]33z(##{4   rc tjj}tj}tj}|t j|||}||j tj }t||j |}t||j|} t||j|} tj|t"j|i} gd} d} || vrt'j| d| d<t'j| d| d<t-jd\}}|||t-jt,j|zz}t7jd|d t:d } t#j||i}t#j||i}| dkr!t?j ||| ||j!|| | S| d kr/t?j"||| d| d| ||j!|| | StGd | d)NrSr6rr7)GaussianGaussianSquareDragConstant SymbolicPulseampanglez amp, anglez9Library pulses with complex amp are no longer supported. > with complex amp was converted to (amp,angle) representation.ScalableSymbolicPulse pulse_typeduration parametersr r"envelope constraintsvalid_amp_conditions rdrer_r`rfr r"rgrhriUnknown class '')$r SYMBOLIC_PULSEr$SYMBOLIC_PULSE_PACKSYMBOLIC_PULSE_SIZEr%r&r( type_sizerMr rNrV envelope_sizeconstraints_sizevalid_amp_conditions_sizer9r r<r,r`absrOsymbolssubsexpIwarningswarn UserWarningrrr^r/rbNotImplementedError)rrrSmakepacksizer0rdrgrhrirflegacy_library_pulses class_name_amp_anglerer s r_read_symbolic_pulserzs  ! 'D  &D  &D T  MM$    F v/0077 FFJ#HMM&2F$G$GU`aaaH& f-..KK0 f677[$& JONN J*** hz%'899 7F:e#455 5{<00 f==tcgcefn.E.E'EFF  Z Z Z Z    - '266H  Hgr 2 2D_$$$!!".#!5     . . .,!5!W%!".#!5    ""AJ"A"A"ABBBrc tjj}tj}tj}|t j|||}||j tj }||j  tj } t||j||} t||j||} t||j||} tj|t$j|i} t%j||i}t%j||i}|dkr!t+j| || ||j| | | S|dkrt1| dt2rXt5j| d| d<t5j| d| d<t;jd| d t>t+j | || d| d| ||j| | | StCd |d ) NrXrYr^rcrbr_r`z@ScalableSymbolicPulse with complex amp are no longer supported. rarjrkrl)"r SYMBOLIC_PULSE_V2r$SYMBOLIC_PULSE_PACK_V2SYMBOLIC_PULSE_SIZE_V2r%r&r(class_name_sizerMr rNrprVrqrrrsr9r r<rrr^r/ isinstancecomplexr,r`rtryrzr{rbr|)rrrRrSr}r~rr0rrdrgrhrirfrer s r_read_symbolic_pulse_v6rs  $ *D  )D  )D T  MM$    F v566==fmLLJv/0077 FFJ# f*++] H' f-.. ;K0 f677T_$& J'266H  Hgr 2 2D_$$$!!".#!5     . . . j' 1 1 "$(:e+<"="=Jw  "z%'8 9 9Ju  M^^^^    ,!5!W%!".#!5    ""AJ"A"A"ABBBrc tj|}tj|tj|i}t j|}t |}t||_ |S)NrY) r rr;r r<r ScheduleAlignmentrobject__new__tuple_context_params)rrrcontext_params context_clsinstances r_read_alignment_contextrsv#H--H)& N -66x@@K~~k**H$^44H Orc|tjjkrtj|t |S|tjjkrA|dkrtj|t||Stj|t|||S|tjj krtj|t|S|tjj kr| tj S|tjjkrtj|t|S|tjjkrtj|t"|St%j|||iS)Nr)rrS)rrRrS)r ScheduleOperandr#r r+r2rmrrCHANNELr OPERAND_STRrMrNKERNELrD DISCRIMINATORrFr r<)r data_bytesrrRrSs r_loads_operandr&si9,555&z>7SSSS9,;;; Q;;*0'{ *'+'  9,444&z='RRRR9,888  ///9,333&      9,:::&       Xz7B ? ??rctj|}|tjjkrt |||||Stj|t|||}tj ||i}t tj |}t||_||_d|_|S)NrX)r6rrRrS)r rr ProgramSCHEDULE_BLOCKread_schedule_blockr;rr rrrScheduleInstructionrr _operands_name_hash) rrmetadata_deserializerrRrSroperandsr rs r _read_elementrJs#H--H9$333" g4mQ\    ### H  Hgr 2 2D~~i;DDXNNOOHxHHNHN Orc|tjjkrdS|tjjkrt j|t|||Std|d)N)r6rrrSzKLoaded schedule reference item is neither None nor ScheduleBlock. Type key zd is not valid data type for a reference items. This data cannot be loaded. Please check QPY version.) r ValueNULLrrr r+rr )rrrrrSs r_loads_reference_itemrcs9?'''t9$333& ,"7#       @ @ @ @  rctj|}tj||t j||j|dSNr)r rassignr write_type_keyr write_valuer)rdatarrs r_write_channelrvsK(//55H (H--- h G<<<<<r=s rrrs~#t7 YY /)  "CJPW    $,,..K  / / / / / / / / / / / / / / / [  C  7 YY /)  !"SZQX    $,,..K  / / / / / / / / / / / / / / / [   g6666s#2A""A&)A&2CCCc|j}|j}tj||t|t j|||dSNrrr rCr rrr rrrrr rCs r _write_kernelrI 9D [F (Fz[bcccc hg666666rc|j}|j}tj||t|t j|||dSrrrs r_write_discriminatorrrrc<|dS|r!|dd}nbddlm}m}t |t jstd|||tj }tj |S)Nrr)sreprrPzInvalid ParameterExpression) __reduce__rLrrPrrOBasicrencoder rNrIcompress)rUrRrQrrPs r_dumps_symbolic_exprrs |s@__&&q)!, (((((((($ ** =;<< <U774==))00?? = $ $$rc |jjtj}|jtj}t |j|}t |j|}t |j |}tj tj t|t|t|t|t||j} || ||||||||||tj||jt&j|t'j||j|t'j||j|dS)N)rrrr) __class____name__rr rNrdrrgrhrir%r~r rrrrr_paramsr rrrer ) rrrRrclass_name_bytespulse_type_bytesenvelope_bytesconstraints_bytesvalid_amp_conditions_bytes header_bytess r_write_symbolic_pulsers~.55fmDD--fm<<)$-GGN,T-=}MM!5d6OQ^!_!_;&   N  &'' L NN<    NN#$$$ NN#$$$ NN>""" NN$%%% NN-...  $   h w???? h 7;;;;;;rctj|}tj||tj||jtj|dS)N)rrr) r rrr rrrr r)rcontextrrs r_write_alignment_contextrs_*11'::H (H--- 72u?PZarct|tjr/tjj}t j|t|}nwt|tj r0tjj }t j|t||}n-t|tj r.tjj}t j|t|}nt|t r1tjj}|t j}nt|t(r.tjj}t j|t,|}n\t|t.r.tjj}t j|t2|}nt5j||\}}||fS)Nr)rRr)rrr.r rr#r rrr^rmrrChannelrrstrrrrNrrrrrrr r)operandrRrrrs r_dumps_operandrs'7+,,K,5*7OWUUU GW2 3 3K,;* *-QX    GX- . . K,4*7NGTTT GS ! ! K,8^^FM22 GV $ $K,3*7M7SSS G] + +K,:*74HRYZZZ $0'JJJ* Z rct|tr:tj|tjjt|||||dStj |}tj||tj ||j t||tj||j|dS)Nr)rrrRr)rrr rr rrwrite_schedule_blockrrrrrr rr )relementmetadata_serializerrRrrs r_write_elementrs'=)) Ch (9(HIIIXw0C]\cdddddd077@@h111 %%'      (GL'BBBBBBrc|tjj}d}n.tjj}t j|t||}||fS)Nr)rrrr)r rrrrr rr)schedulerrrrs r_dumps_reference_itemrsT?' $3*+ 3    Z rc|dkrtd|dtjt jtj|tj}||j  tj }||j }tj||}t!||} t#||| } t%|jD]-} t)|||||} | | d.|d krtj|t.||| } i}| D]=\}}|6t3|t6jj}|||<>|r| |d| S) aRead a single ScheduleBlock from the file like object. Args: file_obj (File): A file like object that contains the QPY binary data. version (int): QPY version. metadata_deserializer (JSONDecoder): An optional JSONDecoder class that will be used for the ``cls`` kwarg on the internal ``json.load`` call used to deserialize the JSON payload used for the :attr:`.ScheduleBlock.metadata` attribute for a schedule block in the file-like object. If this is not specified the circuit metadata will be parsed as JSON with the stdlib ``json.load()`` function using the default ``JSONDecoder`` class. use_symengine (bool): If True, symbolic objects will be serialized using symengine's native mechanism. This is a faster serialization alternative, but not supported in all platforms. Please check that your target platform is supported by the symengine library before setting this option, as it will be required by qpy to deserialize the payload. trust_input (bool): if set to ``False`` (the default), :class:`.ScheduleBlock` objects in the payload that were serialized using ``sympy`` are not allowed and will error. This is because the ``sympy`` parsing uses :func:`eval`, which can allow for arbitrary code execution. The flag should only be set to ``True`` if you trust the QPY payload you are loading. Returns: ScheduleBlock: The schedule block object from the file. Raises: TypeError: If any of the instructions is invalid data format. QiskitError: QPY version is earlier than block support. z QPY version z does not support ScheduleBlock.)cls)r metadataalignment_contextrXT)inplace)rr6rrrS)rr SCHEDULE_BLOCK_HEADERr$r%r&SCHEDULE_BLOCK_HEADER_PACKr(SCHEDULE_BLOCK_HEADER_SIZE name_sizerMr rN metadata_sizejsonloadsrrrange num_elementsrappendr9ritemsrsplitr Reference key_delimiterassign_references)rrrrRrSrr metadata_rawrrblock_ block_elmflat_key_refdictref_dictkey_strr composite_keys rrr sF{{RRRRSSS  ( . .  . MM'< = =    D == ( ( / / > >D==!344Lz,,ABBBH%h88G  !   E 4$ % %..! g4mQ\     Y ----!||!.."7#    !1!7!7!9!9 3 3 GX# %gmmL4J4X&Y&Y Z Z *2'  <  # #Hd # ; ; ; Lrc,tj|jd|tj}|jtj}tjt|t|t|}tj tj g|R}| || || |t||j||jD]} t#|| |||i} |jD].\} } t(jj| } | | | </t j|| t2||dS)a*Write a single ScheduleBlock object in the file like object. Args: file_obj (File): The file like object to write the circuit data in. block (ScheduleBlock): A schedule block data to write. metadata_serializer (JSONEncoder): An optional JSONEncoder class that will be passed the :attr:`.ScheduleBlock.metadata` dictionary for ``block`` and will be used as the ``cls`` kwarg on the ``json.dump()`` call to JSON serialize that dictionary. use_symengine (bool): If True, symbolic objects will be serialized using symengine's native mechanism. This is a faster serialization alternative, but not supported in all platforms. Please check that your target platform is supported by the symengine library before setting this option, as it will be required by qpy to deserialize the payload. version (int): The QPY format version to use for serializing this circuit block Raises: TypeError: If any of the instructions is invalid data format. ),:) separatorsr)rrrr)rrrrrN)rdumpsrrr rNr r rrr%r~rrrr_blocksr_reference_managerrrr r joinrr)rrrrRrr block_name header_rawr0rrref_keysrrs rrrps(z%.ZEXYYY`` H""6=11J.j//(mmZZJ [; Ij I I IF NN6 NN: NN8Xu'>PPPP]aa  x,?X_`````#6<<>>--((6;;HEE$,!!  (/ r)FF)F)NFF):__doc__rr%rIryiornumpyr, symenginerOqiskit.exceptionsr qiskit.pulserrrqiskit.pulse.scheduler qiskit.qpyr r r qiskit.qpy.binary_ior qiskit.qpy.exceptionsr qiskit.pulse.configurationrrqiskit.utils.deprecate_pulserrr2r:rDrFrVrrrrrrrrrrrrrrrrrr QPY_VERSIONrrArrr-s98 ))))))8888888888//////1111111111&&&&&&******<<<<<<<<JJJJJJ&JJJ$'''...!!!!(MCMCMCMC`GCGCGCGCT$!@!@!@!@H2&=== < < <777,777777 % % %<<<@   6CCC"    #TYLLL#"L`*.UFL^666666r