gD'dZddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZGdd ejZdddZdS)z@Parser for mathematical string expressions returned by backends.) annotationsN)Callable)Any) PulseError)ParameterExpressionc reZdZUdZidejdejdejdejdej dej dej d ej d ej d ejd ejd ejdejdejdejdejdejdejiZded< ejejejejejej ej!ej"ej#ej$iZ% ej&ej'ej(ej)iZ* d:d;dZ+e,dd&Z0d?d)Z1d@d+Z2dAd.Z3dBd1Z4dCd4Z5dDd7Z6d8Z7d9S)EPulseExpressionz/Expression parser to evaluate parameter values.acosacoshasinasinhatanatanhcoscoshexploglog10sinsinhsqrttantanhpiezdict[str, Callable | float] _math_opsFsourcestr | ast.Expressionpartial_bindingboolcD||_i|_t|_t |t jr||_nB t j|d|_n%#t$r}t|d|d}~wwxYw| |jdS)zCreate new evaluator. Args: source: Expression of equation to evaluate. partial_binding: Allow partial bind of parameters. Raises: PulseError: When invalid string is specified. eval)modez is invalid expression.N) _partial_binding _locals_dictset_params isinstanceast Expression_treeparse SyntaxErrorrvisit)selfrrexs I/granary/edev/quantum/lib/python3.11/site-packages/qiskit/pulse/parser.py__init__zPulseExpression.__init__Bs!0,.!$ fcn - - MDJJ M YvF;;;  M M M F!C!C!CDD"L M 4:sA!! B+A>>Breturn list[str]cNt|jS)z\Get parameters. Returns: A list of parameters in sorted order. )sortedr'copy)r/s r1paramszPulseExpression.params[s dl''))****complex | ast.Expression | PulseExpressionct|jjtjr|jjjS|j|r%t|j |D]\}}||j|<|rt| D]_\}}||j vr3||jvr ||j|<"t|j j d|dt|j j d|d||j}t|jtjs4|jrt!||jStd|j d|jjS)a;Evaluate the expression with the given values of the expression's parameters. Args: *args: Variable length parameter list. **kwargs: Arbitrary parameters. Returns: Evaluated value. Raises: PulseError: When parameters are not bound. z# got multiple values for argument ''z% got an unexpected keyword argument 'z Parameters z are not all bound.)r(r+bodyr)Constantvaluer%clearzipr8itemsr __class____name__r.r$r )r/argskwargskeyvalexprs r1__call__zPulseExpression.__call__ds djos| 4 4 ):?( ( !!!  - T22 - -S),!#&&  "LLNN  S$+%%$"33314)#..(#~6aa[^aaa%>2__Y\___zz$*%%$)S\22 Q$ Q&tT-BCCC !Ot{!O!O!OPPPyr9oprast.ASTopr_dictdictcomplexc|D]\}}t||r||cStd|jjd)aCHelper method to apply operators. Args: opr: Operator of node. opr_dict: Mapper from ast to operator. *args: Arguments supplied to operator. Returns: Evaluated value. Raises: PulseError: When unsupported operation is specified. z Operator  is not supported.)rBr(rrCrD)rKrMrEop_typeop_funcs r1 _match_opszPulseExpression._match_opssj!) 0 0 & & GW#w'' &w~%%% &OS]%;OOOPPPr9nodeast.Expressioncltj|}||j|_|S)zEvaluate children nodes of expression. Args: node: Expression to evaluate. Returns: Evaluated value. )r7r.r=)r/rUtmp_nodes r1visit_Expressionz PulseExpression.visit_Expressions+9T?? 8=11 r9 ast.Constantc|S)z}Return constant value as it is. Args: node: Constant. Returns: Input node. r/rUs r1visit_ConstantzPulseExpression.visit_Constants  r9ast.Nameast.Name | ast.ConstantcH|j|jvr9tj|j|j}tj||S|j|jvr|j|j}t |ts] t|}|j s|j }n>#t$r1}td|jd|j|jd|d}~wwxYwtj|}tj||S|j |j|S)zEvaluate name and return ast.Constant if it is bound. Args: node: Name to evaluate. Returns: Evaluated value. Raises: PulseError: When parameter value is not a number. zInvalid parameter value z = z is specified.N)idrr)r> copy_locationr%r(rrOimagreal ValueErrorrr'add)r/rUrH_valr0s r1 visit_NamezPulseExpression.visit_Names9 7dn $ $,t~dg677C$S$// / W) ) )$TW-Dd$788 "4==D9)#y!$%47%%t?PQUQX?Y%%% ,t$$C$S$// / !!! s>B C&,CC ast.UnaryOpast.UnaryOp | ast.ConstantcNtj|}||j|_t|jtjrRt j||j|j|jj }t j ||S|S)zEvaluate unary operation and return ast.Constant if operand is bound. Args: node: Unary operation to evaluate. Returns: Evaluated value. ) r7r.operandr(r)r>rTop _unary_opsr?rcr/rUrHs r1 visit_UnaryOpzPulseExpression.visit_UnaryOps|yzz$,// dlCL 1 1 0,ttwI[\\]]C$S$// / r9 ast.BinOpast.BinOp | ast.Constantctj|}||j|_||j|_t |jt jr|t |jt jr]t j||j|j |jj |jj }t j ||S|S)zEvaluate binary operation and return ast.Constant if operands are bound. Args: node: Binary operation to evaluate. Returns: Evaluated value. ) r7r.leftrightr(r)r>rTrn _binary_opsr?rcrps r1 visit_BinOpzPulseExpression.visit_BinOpsyJJty)) ZZ ++ di . . 0:dj#,3W3W 0,)949?DJL\]]C$S$// / r9ast.Callast.Call | ast.Constantct|jtjst dt j|}fd|jD|_td|jDr|jjj vrt d|jjdd|jD}j |jj|}|j s|j }tj |}tj ||S|S)aEvaluate function and return ast.Constant if all arguments are bound. Args: node: Function to evaluate. Returns: Evaluated value. Raises: PulseError: When unsupported or unsafe function is specified. zUnsafe expression is detected.c:g|]}|Sr\)r.).0argr/s r1 z.PulseExpression.visit_Call..s#:::TZZ__:::r9c3JK|]}t|tjVdS)N)r(r)r>r}r~s r1 z-PulseExpression.visit_Call..s.BBz#s|,,BBBBBBr9z Function rQcg|] }|j Sr\)r?rs r1rz.PulseExpression.visit_Call..s4443SY444r9)r(funcr)Namerr7rEallrbrrdrer>rc)r/rU_argsrhrHs` r1 visit_CallzPulseExpression.visit_Calls$)SX.. ?=>> >y:::: ::: BB BBB B B 0y|4>11 !MTY\!M!M!MNNN44$)444E/4>$),/7D9 !y,t$$C$S$// / r9c:td|jj)NzUnsupported node: )rrCrDr]s r1 generic_visitzPulseExpression.generic_visitsGdn.EGGHHHr9NF)rrrr )r3r4)r3r:)rKrLrMrNr3rO)rUrVr3rV)rUrZr3rZ)rUr_r3r`)rUrjr3rk)rUrrr3rs)rUryr3rz)8rD __module__ __qualname____doc__cmathr r r r rrrrrrrrrrrrrrr__annotations__r)AddoperatorrgSubsubMultmulDivtruedivPowpowrwUAddposUSubnegror2propertyr8rJ staticmethodrTrYr^rirqrxrrr\r9r1r r s99. ..  .  .  .  . uy.  . uy. uy. . uy.  .  . uy.  !." eh#.$ UW%..I(    (, !  K#(HL#(HLAJ!2+++X+))))VQQQ\Q&        B &6IIIIIr9r Frstrrr r3cfgd}|D]\}}|||}t||S)a"Safe parsing of string expression. Args: source: String expression to parse. partial_binding: Allow partial bind of parameters. Returns: PulseExpression: Returns a expression object. Example: expr = 'P1 + P2 + P3' parsed_expr = parse_string_expr(expr, partial_binding=True) # create new PulseExpression bound_two = parsed_expr(P1=1, P2=2) # evaluate expression value1 = bound_two(P3=3) value2 = bound_two(P3=4) value3 = bound_two(P3=5) ))znumpy.)znp.r)zmath.r)zcmath.r)replacer )rrsubsmatchrs r1parse_string_exprrsH. H G GD,, ss++ 6? 3 33r9r)rrrr r3r )r __future__rr)r7rrcollections.abcrtypingrqiskit.pulse.exceptionsrqiskit.circuitrNodeTransformerr rr\r9r1rsGF""""""  $$$$$$............IIIIIc)IIID4444444r9