a RG5dPã@sšddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z ddd d d d d d ddddddddddœZGdd„de ƒZddd„Zdd„ZdS) é)ÚSet)ÚBasicÚS)ÚLambda)Ú CodePrinter)Ú precedence©ÚreduceÚabsÚsinÚcosÚtanÚacosÚasinÚatanÚceilÚfloorÚsignÚexpÚlogÚaddÚsubÚmulÚpow)ÚAbsr r r rrrÚatan2Úceilingrrrrrrrrc@s eZdZdZeƒZdZdZdddddddd d d iddddd d œZifdd„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„ZeZeZd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„Zd2d3„Zd4d5„Zd6d7„Zd\óz+GLSLPrinter.indent_code..cs g|]}ttt|jˆƒƒƒ‘qSr:)ÚintÚanyÚmapÚendswithrR)Ú inc_tokenr:r;rU^rVcs g|]}ttt|jˆƒƒƒ‘qSr:)rWrXrYÚ startswithrR)Ú dec_tokenr:r;rU_rVr)rLÚ z%s%s)Ú isinstanceÚstrrIÚ splitlinesÚjoinÚ enumerateÚappend) r8ÚcodeZ code_linesÚtabZincreaseZdecreaseÚprettyÚlevelÚnrTr:)r]r[r;rIQs&    zGLSLPrinter.indent_codec Cs˜|jd}|jd}|r"|jdkn|jdk}||kr<| ¡n|}|jd}|jd}|j|j}d ||¡} |jdkr„| |d¡S|jdkr|jdkr|r|jdkrÆd  |j|j|d d d ¡S|j|jkrîd  |j|j|dddd¡Sd |j|j|j|dddd¡Sn„tj|j vr8d | |j||ddd¡S|jdsfd | |j||ddd|j|j¡S|jdr”d ||j|j|j||dd d¡SdS)Nr&r'ér)r(ú{}[{}]rézvec{}{}rNrP)ÚrowstartÚrowendz mat{}({})ú, rL)Úrowseprmrnz mat{}x{}({})ú{}({})r%z{}( {} ) /* a {}x{} matrix */z{}[{}][{}]( {} )zfloat[]() Ú _settingsÚrowsÚcolsÚ transposerBÚ_printÚtablerÚOneÚshape) r8Úmatr&r'Z column_vectorÚAr)r(Ú array_sizeÚarray_constructorr:r:r;Ú_print_MatrixBaselsV        ÿ  ÿÿÿþþ ý  þzGLSLPrinter._print_MatrixBasecCs | |¡SrH)Ú_print_not_supported)r8rzr:r:r;Ú_print_SparseRepMatrixšsz"GLSLPrinter._print_SparseRepMatrixcs:|jd}|r|j\‰}n |j\}‰‡fdd„t|ƒDƒS)Nr'c3s$|]}tˆƒD]}||fVqqdSrH)Úrange)rSÚiÚj©rsr:r;Ú ¤rVz7GLSLPrinter._traverse_matrix_indices..)rrryr)r8rzr'rtr:r„r;Ú_traverse_matrix_indicesžs    z$GLSLPrinter._traverse_matrix_indicesc Cs¢|jd}|jd}|jd}|r>|jj\}}|j|j}}n|jj\}}|j|j}}| |j¡} |rŠ|dkrx|dks||rŠd | ||¡Sd | |||¡SdS)Nr%r)r'rlz {}[{}][{}]rk)rrÚparentryrƒr‚rvrB) r8ÚexprÚnestr)r'rtrsr‚rƒZpntr:r:r;Ú_print_MatrixElement¦s      z GLSLPrinter._print_MatrixElementcshd ‡fdd„|Dƒ¡}ˆjd}ˆjd}t|ƒ}d ||¡}|dkrX|rXd ||¡Sd  ||¡SdS) Nroc3s|]}ˆ |¡VqdSrH©rv©rSÚitem©r8r:r;r…¸rVz*GLSLPrinter._print_list..r)r(rkrlz vec{}({})rq)rbrrÚlenrB)r8rˆÚlr)r(r|r}r:rŽr;Ú _print_list·s     zGLSLPrinter._print_listc CsZg}g}d}|D]@}| || |j¡| |j¡| |jd¡dœ¡| d¡q||fS)NzAfor (int %(varble)s=%(start)s; %(varble)s<%(end)s; %(varble)s++){rj)ZvarbleÚstartÚendrO)rdrvÚlabelÚlowerÚupper)r8ÚindicesZ open_linesZ close_linesZ loopstartr‚r:r:r;Ú_get_loop_opening_endingÆs  ý  z$GLSLPrinter._get_loop_opening_endingc s²|ˆjvrŒˆj|}d}t|tƒr(|}n|D]\}}||ƒr,qBq,|dur®z|‡fdd„|DƒŽWStyˆd |ˆ |d¡¡YS0n"t|tƒr¤ˆ ||Ž¡Sˆ |¡SdS)Ncsg|]}ˆ |d¡‘qS)r)Ú parenthesizerŒrŽr:r;rUßrVz9GLSLPrinter._print_Function_with_args..rqro) r5r_r`Ú TypeErrorrBÚ stringifyrrvr)r8ÚfuncÚ func_argsÚ cond_funcÚcondr:rŽr;Ú_print_Function_with_argsÓs       z%GLSLPrinter._print_Function_with_argsc s"ddlm}|jdjdkr$tdƒ‚g}| |¡rÊt|jƒD]‚\}\}}|dkrf| dˆ |¡¡n:|t |jƒdkrŒ|dkrŒ| d¡n| d ˆ |¡¡ˆ |¡}| |¡| d ¡q.z: ( %s )z: ú rP) Úsympy.codegen.astr¡ÚargsrŸÚ ValueErrorÚhasrcrdrvrrbrˆ) r8rˆr¡rJr‚r£r¤Úcode0ÚecpairsÚ last_liner:rŽr;Ú_print_Piecewiseès(         þzGLSLPrinter._print_PiecewisecCs | |j¡SrH)rvr”©r8rˆr:r:r;Ú _print_Idx szGLSLPrinter._print_IdxcCs`|j}tj}tj}tt|jƒƒD]"}||j||7}|||9}q d |  |j j ¡|  |¡¡S)Nrk) ryrÚZerorxÚreversedrÚrankr—rBrvÚbaser”)r8rˆÚdimsÚelemÚoffsetr‚r:r:r;Ú_print_Indexed s þzGLSLPrinter._print_IndexedcCsŽt|ƒ}|jdkr$d| |j|¡S|jdkr>d| |j¡Sz| t|jƒ¡}Wntyp| |j¡}Yn0| d| |j¡|f¡SdS)Nr¢z1.0/%sgà?zsqrt(%s)r)rrr™r³rvrršr )r8rˆÚPRECr£r:r:r;Ú _print_Pows    þzGLSLPrinter._print_PowcCs tt|ƒƒSrH)r`rr®r:r:r;Ú _print_int*szGLSLPrinter._print_intcCsd |j|j¡S)Nz {}.0/{}.0)rBr=Úqr®r:r:r;Ú_print_Rational-szGLSLPrinter._print_RationalcCs,| |j¡}| |j¡}|j}d |||¡S)Nz{} {} {})rvÚlhsÚrhsÚrel_oprB)r8rˆÚlhs_codeÚrhs_codeÚopr:r:r;Ú_print_Relational0s  zGLSLPrinter._print_Relationalcs¾ˆjdrtjˆ||dS| ¡}dd„}‡fdd„‰|dd„|ƒ\}}|rrt‡fd d„t‡fd d„|ƒƒ}}nˆ ˆjd ¡}}|rºt‡fd d„t‡fd d„|ƒƒ}ˆ d||f¡}|S)Nr#)r*cst‡fdd„|ggfƒS)Ncs4ˆ|ƒr|d|g|dfS|d|d|gfS)Nrrjr:)ÚxÚy©r=r:r;Ú=rVz;GLSLPrinter._print_Add..partition..r)r=rr:rÆr;Ú partition<sz)GLSLPrinter._print_Add..partitioncsˆ d||f¡S)Nr©r ©ÚaÚbrŽr:r;r>sz#GLSLPrinter._print_Add..addcSs| ¡SrH)Úcould_extract_minus_sign)Úargr:r:r;rÇArVz(GLSLPrinter._print_Add..cs ˆ||ƒSrHr:rÊ©rr:r;rÇCrVcs ˆ |¡SrHr‹©ÚtrŽr:r;rÇCrVr$cs ˆ||ƒSrHr:rÊrÏr:r;rÇIrVcs ˆ | ¡SrHr‹)rirŽr:r;rÇIrVr)rrrÚ _print_AddÚas_ordered_termsr rYrvr )r8rˆr*ÚtermsrÈÚnegÚposÚsr:)rr8r;rÒ6s  & zGLSLPrinter._print_Addc sVˆjdrtjˆ|fi|¤ŽS| ¡}‡fdd„‰t‡fdd„t‡fdd„|ƒƒ}|S)Nr#csˆ d||f¡S)NrrÉrÊrŽr:r;rSsz#GLSLPrinter._print_Mul..mulcs ˆ||ƒSrHr:rÊ)rr:r;rÇWrVz(GLSLPrinter._print_Mul..cs ˆ |¡SrHr‹rÐrŽr:r;rÇWrV)rrrÚ _print_MulÚas_ordered_factorsr rY)r8rˆÚkwargsrÔr×r:)rr8r;rØOs    zGLSLPrinter._print_Mul)N)"Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚsetÚ_not_supportedÚ printmethodÚlanguageÚ_default_settingsr3r>r@rDrGrKrIr~r€r†rŠr‘Ú _print_tupleÚ _print_Tupler˜r r­r¯r·r¹rºr¼rÃrÒrØr:r:r:r;rs\ï .  "  rNcKst|ƒ ||¡S)a¾Converts an expr to a string of GLSL code Parameters ========== expr : Expr A SymPy expression to be converted. assign_to : optional When given, the argument is used for naming the variable or variables to which the expression is assigned. Can be a string, ``Symbol``, ``MatrixSymbol`` or ``Indexed`` type object. In cases where ``expr`` would be printed as an array, a list of string or ``Symbol`` objects can also be passed. This is helpful in case of line-wrapping, or for expressions that generate multi-line statements. It can also be used to spread an array-like expression into multiple assignments. use_operators: bool, optional If set to False, then *,/,+,- operators will be replaced with functions mul, add, and sub, which must be implemented by the user, e.g. for implementing non-standard rings or emulated quad/octal precision. [default=True] glsl_types: bool, optional Set this argument to ``False`` in order to avoid using the ``vec`` and ``mat`` types. The printer will instead use arrays (or nested arrays). [default=True] mat_nested: bool, optional GLSL version 4.3 and above support nested arrays (arrays of arrays). Set this to ``True`` to render matrices as nested arrays. [default=False] mat_separator: str, optional By default, matrices are rendered with newlines using this separator, making them easier to read, but less compact. By removing the newline this option can be used to make them more vertically compact. [default=', '] mat_transpose: bool, optional GLSL's matrix multiplication implementation assumes column-major indexing. By default, this printer ignores that convention. Setting this option to ``True`` transposes all matrix output. [default=False] array_type: str, optional The GLSL array constructor type. [default='float'] precision : integer, optional The precision for numbers such as pi [default=15]. user_functions : dict, optional A dictionary where keys are ``FunctionClass`` instances and values are their string representations. Alternatively, the dictionary value can be a list of tuples i.e. [(argument_test, js_function_string)]. See below for examples. human : bool, optional If True, the result is a single string that may contain some constant declarations for the number symbols. If False, the same information is returned in a tuple of (symbols_to_declare, not_supported_functions, code_text). [default=True]. contract: bool, optional If True, ``Indexed`` instances are assumed to obey tensor contraction rules and the corresponding nested loops over indices are generated. Setting contract=False will not generate loops, instead the user is responsible to provide values for the indices in the code. [default=True]. Examples ======== >>> from sympy import glsl_code, symbols, Rational, sin, ceiling, Abs >>> x, tau = symbols("x, tau") >>> glsl_code((2*tau)**Rational(7, 2)) '8*sqrt(2)*pow(tau, 3.5)' >>> glsl_code(sin(x), assign_to="float y") 'float y = sin(x);' Various GLSL types are supported: >>> from sympy import Matrix, glsl_code >>> glsl_code(Matrix([1,2,3])) 'vec3(1, 2, 3)' >>> glsl_code(Matrix([[1, 2],[3, 4]])) 'mat2(1, 2, 3, 4)' Pass ``mat_transpose = True`` to switch to column-major indexing: >>> glsl_code(Matrix([[1, 2],[3, 4]]), mat_transpose = True) 'mat2(1, 3, 2, 4)' By default, larger matrices get collapsed into float arrays: >>> print(glsl_code( Matrix([[1,2,3,4,5],[6,7,8,9,10]]) )) float[10]( 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 ) /* a 2x5 matrix */ The type of array constructor used to print GLSL arrays can be controlled via the ``array_type`` parameter: >>> glsl_code(Matrix([1,2,3,4,5]), array_type='int') 'int[5](1, 2, 3, 4, 5)' Passing a list of strings or ``symbols`` to the ``assign_to`` parameter will yield a multi-line assignment for each item in an array-like expression: >>> x_struct_members = symbols('x.a x.b x.c x.d') >>> print(glsl_code(Matrix([1,2,3,4]), assign_to=x_struct_members)) x.a = 1; x.b = 2; x.c = 3; x.d = 4; This could be useful in cases where it's desirable to modify members of a GLSL ``Struct``. It could also be used to spread items from an array-like expression into various miscellaneous assignments: >>> misc_assignments = ('x[0]', 'x[1]', 'float y', 'float z') >>> print(glsl_code(Matrix([1,2,3,4]), assign_to=misc_assignments)) x[0] = 1; x[1] = 2; float y = 3; float z = 4; Passing ``mat_nested = True`` instead prints out nested float arrays, which are supported in GLSL 4.3 and above. >>> mat = Matrix([ ... [ 0, 1, 2], ... [ 3, 4, 5], ... [ 6, 7, 8], ... [ 9, 10, 11], ... [12, 13, 14]]) >>> print(glsl_code( mat, mat_nested = True )) float[5][3]( float[]( 0, 1, 2), float[]( 3, 4, 5), float[]( 6, 7, 8), float[]( 9, 10, 11), float[](12, 13, 14) ) Custom printing can be defined for certain types by passing a dictionary of "type" : "function" to the ``user_functions`` kwarg. Alternatively, the dictionary value can be a list of tuples i.e. [(argument_test, js_function_string)]. >>> custom_functions = { ... "ceiling": "CEIL", ... "Abs": [(lambda x: not x.is_integer, "fabs"), ... (lambda x: x.is_integer, "ABS")] ... } >>> glsl_code(Abs(x) + ceiling(x), user_functions=custom_functions) 'fabs(x) + CEIL(x)' If further control is needed, addition, subtraction, multiplication and division operators can be replaced with ``add``, ``sub``, and ``mul`` functions. This is done by passing ``use_operators = False``: >>> x,y,z = symbols('x,y,z') >>> glsl_code(x*(y+z), use_operators = False) 'mul(x, add(y, z))' >>> glsl_code(x*(y+z*(x-y)**z), use_operators = False) 'mul(x, add(y, mul(z, pow(sub(x, y), z))))' ``Piecewise`` expressions are converted into conditionals. If an ``assign_to`` variable is provided an if statement is created, otherwise the ternary operator is used. Note that if the ``Piecewise`` lacks a default term, represented by ``(expr, True)`` then an error will be thrown. This is to prevent generating an expression that may not evaluate to anything. >>> from sympy import Piecewise >>> expr = Piecewise((x + 1, x > 0), (x, True)) >>> print(glsl_code(expr, tau)) if (x > 0) { tau = x + 1; } else { tau = x; } Support for loops is provided through ``Indexed`` types. With ``contract=True`` these expressions will be turned into loops, whereas ``contract=False`` will just print the assignment expression that should be looped over: >>> from sympy import Eq, IndexedBase, Idx >>> len_y = 5 >>> y = IndexedBase('y', shape=(len_y,)) >>> t = IndexedBase('t', shape=(len_y,)) >>> Dy = IndexedBase('Dy', shape=(len_y-1,)) >>> i = Idx('i', len_y-1) >>> e=Eq(Dy[i], (y[i+1]-y[i])/(t[i+1]-t[i])) >>> glsl_code(e.rhs, assign_to=e.lhs, contract=False) 'Dy[i] = (y[i + 1] - y[i])/(t[i + 1] - t[i]);' >>> from sympy import Matrix, MatrixSymbol >>> mat = Matrix([x**2, Piecewise((x + 1, x > 0), (x, True)), sin(x)]) >>> A = MatrixSymbol('A', 3, 1) >>> print(glsl_code(mat, A)) A[0][0] = pow(x, 2.0); if (x > 0) { A[1][0] = x + 1; } else { A[1][0] = x; } A[2][0] = sin(x); )rÚdoprint)rˆÚ assign_tor9r:r:r;Ú glsl_codeZsLrècKstt|fi|¤ŽƒdS)zpPrints the GLSL representation of the given expression. See GLSLPrinter init function for settings. N)Úprintrè)rˆr9r:r:r;Ú print_glsl'srê)N)ÚtypingrÚtSetÚ sympy.corerrÚsympy.core.functionrZsympy.printing.codeprinterrÚsympy.printing.precedencerÚ functoolsr r5rrèrêr:r:r:r;Ús:     ï? N