gB"ddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZddlmZmZddlmZmZmZmZmZmZmZmZmZmZmZdd l m!Z!dd l"m#Z#dId Z$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5dZ6d Z7d!Z8d"Z9d#Z:d$Z;d%Zd(Z?d)Z@d*ZAd+ZBd,ZCd-ZDd.ZEd/ZFd0ZGd1ZHd2ZId3ZJd4ZKd5ZLd6ZMd7ZNd8ZOd9ZPd:ZQd;ZRd<ZSd=ZTd>ZUd?ZVd@ZWdAZXdBZYdCZZdDZ[dEZ\dFZ]dGZ^dHS)J)StringIO)symbolsEqpiCatalanLambdaDummyEquality)Symbol)erf)Integral)Matrix MatrixSymbol) codegen make_routineCCodeGen C89CodeGen C99CodeGen InputArgument CodeGenErrorFCodeGenCodeGenArgumentListErrorOutputArgument InOutArgument)raises)implemented_functionfileFct}|||||||}||S)a5Wrapper for dump_fn. dump_fn writes its results to a stream object and this wrapper returns the contents of that stream as a string. This auxiliary function is used by many tests below. The header and the empty lines are not generated to facilitate the testing of the output. )rgetvalueclose)dump_fnroutinesprefixheaderemptyoutputsources ^/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/utilities/tests/test_codegen.py get_stringr*sFZZF GHfffe444 __  F LLNNN Mc td\ z ztt fdtt fdtdt   g}d|jD  gksJd|jDt t tt gksJtdt  g}d|jDtt t gksJd d l m }m }t|d d g\}}td d}|d|}tdt |||||||g}d|jD|j |j |gksJt z z ddf ddftdt   g}d|jD  gksJdS)Nza x y zc,tdgSNtestargument_sequencer)exprxzsr)z-test_Routine_argument_order..&s!\&$V.%.%.%r+cJtdtgSr.)rr)ar3r4yr5sr)r6z-test_Routine_argument_order..(s4\&"Q CC%&1I./././r+r/r0cg|] }|j Sname.0args r) z/test_Routine_argument_order..+ . . .#SX . . .r+c,g|]}t|Sr;typer>s r)rAz/test_Routine_argument_order.., / / /3T#YY / / /r+c,g|]}t|Sr;rDr>s r)rAz/test_Routine_argument_order../rFr+r IndexedBaseIdxABmTintegericg|] }|j Sr;r<r>s r)rAz/test_Routine_argument_order..7rBr+cg|] }|j Sr;r<r>s r)rAz/test_Routine_argument_order..;rBr+)rrrrr argumentsrrr sympy.tensorrIrJmaplabelr) rrIrJrKrLrMrPr8r3r4r9r5s @@@@@r)test_Routine_argument_orderr[#s##JAq!Q E19D #&%&%&%&%&%&%&&& #&/&/&/&/&/&/&/&/000VR4[[Q1aLIIIA . . . . .1aA, > > > > / /!+ / / /}nm4G G G G GVR4[[Q1IFFFA / /!+ / / /}m46 6 6 6 6.------- {S#J ' 'DAqT"""A C AVR!ad^^1ayIIIA . . . . .17AGQ2G G G G G AaCEAq!9q!Qi 0 0DVR4[[Q1aLIIIA . . . . .1aA, > > > > > >r+c\t}t|jg}|dksJdS)Nz$#include "file.h" #include rr*dump_ccode_genr(s r)test_empty_c_codera>s4||H  , ,F ? ? ? ? ? ? ?r+ct}t|jgd}|dddksJ|dddksJdS)NTr%RzR/****************************************************************************** *a* * * * See http://www.sympy.org/ for more information. * * * * This file is part of 'project' * ******************************************************************************/ #include "file.h" #include r]r_s r)test_empty_c_code_with_commentrfDst||H D 9 9 9F #2#;]     #$$<["      r+c\t}t|jg}|dksJdS)Nz9#ifndef PROJECT__FILE__H #define PROJECT__FILE__H #endif )rr*dump_hr_s r)test_empty_c_headerriVs4||H  , ,F S S S S S S Sr+ctd\}}}||z|z}td|}t}t|j|g}d}||ksJdS)Nx,y,zr/#include "file.h" #include double test(double x, double y, double z) { double test_result; test_result = z*(x + y); return test_result; } rrrr*r^r4r9r5r3routiner`r(expecteds r)test_simple_c_coderq\spgGAq! E19D64((G||H ' 3 3F   X      r+ctd\}}}||z|z}td|}t}t|j|g}d}||ksJdS)Nzif, typedef, whiler/z#include "file.h" #include double test(double if_, double typedef_, double while_) { double test_result; test_result = while_*(if_ + typedef_); return test_result; } )rrrr*r^rns r)test_c_code_reserved_wordsrsnsq*++GAq! EQ;D64((G||H ' 3 3F   X      r+ctdttz}t}t |j|g}dtjdz}||ksJdS)Nr/z#include "file.h" #include double test() { double test_result; double const Catalan = %s; test_result = pow(M_PI, Catalan); return test_result; } )rrrrr*r^evalfror`r(rps r)test_numbersymbol_c_coderxse62w;//G||H ' 3 3F   b H X      r+ctd\}}}||z}td||||g}t}t|j|g}d}||ksJdS)Nrkr/r0z#include "file.h" #include double test(double z, double x, double y) { double test_result; test_result = x + y; return test_result; } rmrns r)test_c_code_argument_orderrzswgGAq! q5D64Aq!9EEEG||H ' 3 3F   X      r+ctd\}}}||z|z}td|}t}t|j|g}d}||ksJdS)Nrkr/d#ifndef PROJECT__FILE__H #define PROJECT__FILE__H double test(double x, double y, double z); #endif )rrrr*rhrns r)test_simple_c_headerr}spgGAq! E19D64((G||H ' 3 3F  X      r+ctd\}}}||z|z}ddg}td|fdddd}||ksJdS) Nrk)file.crl)file.hr|r/CrFr%r&rr)r4r9r5r3rpresults r)test_simple_c_codegenrshgGAq! E19D  Hfd^S&e L L LF X      r+ctd\}}}||z|z}||z |z}td||gtttfddS)Nrkr/c0tjgSNr*rhr`rosr)r6z)test_multiple_results_c..HOgY!G!Gr+)rrrrrr4r9r5expr1expr2r`ros @@r)test_multiple_results_crsxgGAq! UAIE UAIE G||H <GGGGGHHHHHr+c2ttddS)Nc"tdgSNr/r2r;r+r)r6z#test_no_results_c..|FB77r+r ValueErrorr;r+r)test_no_results_cr :7788888r+cddlm}ddlm}ddlm}m}m}ddlm }m }ddl m }ddl m}m} m} m} m} m} t)d}d ||fd ||fd | |fd | |fd ||fd| |fd||fd||fd||fd||fd| |fd||fd||fd| |fd||fg}t+|dddd}|dddksJ|dddksJ|dddksJ|ddd ksJdS)!NrAbslogcoshsinhtanh)ceilingfloorsqrtacosasinatancossintanr4 test_fabs test_acos test_asin test_atan test_ceiltest_cos test_cosh test_floortest_logtest_lntest_sin test_sinh test_sqrttest_tan test_tanhC89rFrrrRa#include "file.h" #include double test_fabs(double x) { double test_fabs_result; test_fabs_result = fabs(x); return test_fabs_result; } double test_acos(double x) { double test_acos_result; test_acos_result = acos(x); return test_acos_result; } double test_asin(double x) { double test_asin_result; test_asin_result = asin(x); return test_asin_result; } double test_atan(double x) { double test_atan_result; test_atan_result = atan(x); return test_atan_result; } double test_ceil(double x) { double test_ceil_result; test_ceil_result = ceil(x); return test_ceil_result; } double test_cos(double x) { double test_cos_result; test_cos_result = cos(x); return test_cos_result; } double test_cosh(double x) { double test_cosh_result; test_cosh_result = cosh(x); return test_cosh_result; } double test_floor(double x) { double test_floor_result; test_floor_result = floor(x); return test_floor_result; } double test_log(double x) { double test_log_result; test_log_result = log(x); return test_log_result; } double test_ln(double x) { double test_ln_result; test_ln_result = log(x); return test_ln_result; } double test_sin(double x) { double test_sin_result; test_sin_result = sin(x); return test_sin_result; } double test_sinh(double x) { double test_sinh_result; test_sinh_result = sinh(x); return test_sinh_result; } double test_sqrt(double x) { double test_sqrt_result; test_sqrt_result = sqrt(x); return test_sqrt_result; } double test_tan(double x) { double test_tan_result; test_tan_result = tan(x); return test_tan_result; } double test_tanh(double x) { double test_tanh_result; test_tanh_result = tanh(x); return test_tanh_result; } ra#ifndef PROJECT__FILE__H #define PROJECT__FILE__H double test_fabs(double x); double test_acos(double x); double test_asin(double x); double test_atan(double x); double test_ceil(double x); double test_cos(double x); double test_cosh(double x); double test_floor(double x); double test_log(double x); double test_ln(double x); double test_sin(double x); double test_sinh(double x); double test_sqrt(double x); double test_tan(double x); double test_tanh(double x); #endif )$sympy.functions.elementary.complexesr&sympy.functions.elementary.exponentialr%sympy.functions.elementary.hyperbolicrrr#sympy.functions.elementary.integersrr(sympy.functions.elementary.miscellaneousr(sympy.functions.elementary.trigonometricrrrrrrrr)rrrrrrrrrrrrrrr4 name_exprrs r)test_ansi_math1_codegenrsA888888::::::HHHHHHHHHHDDDDDDDD======ZZZZZZZZZZZZZZZZ A cc!ff dd1gg dd1gg dd1gg ggajj! SSVV dd1gg uuQxx SSVV CCFF SSVV dd1gg dd1gg SSVV dd1ggI"Yve5 I I IF !9Q<8 # # # # !9Q< F    $ !9Q<8 # # # # !9Q< 0      r+c(ddlm}td\}}d|||fd||zfg}t|dddd }|ddd ksJ|dd d ksJ|d dd ksJ|d d dksJdS)Nratan2x,y test_atan2test_powrrFrrrRa*#include "file.h" #include double test_atan2(double x, double y) { double test_atan2_result; test_atan2_result = atan2(x, y); return test_atan2_result; } double test_pow(double x, double y) { double test_pow_result; test_pow_result = pow(x, y); return test_pow_result; } rz#ifndef PROJECT__FILE__H #define PROJECT__FILE__H double test_atan2(double x, double y); double test_pow(double x, double y); #endif rrrr)rr4r9rrs r)test_ansi_math2_codegenrs>>>>>> 5>>DAq uuQ{{# QTIYve5 I I IF !9Q<8 # # # # !9Q< N     !9Q<8 # # # # !9Q<       r+c ddlm}m}m}t d\}}}d||||z||zdzfd||||||||||z|zfg}t |ddd d }|ddd ksJ|dd d ksJ|d ddksJ|d d dksJdS)Nrrrrrktest1test2rrFrrrRa-#include "file.h" #include double test1(double x, double y, double z) { double test1_result; test1_result = pow(sin(x), 7) + 7*pow(sin(x), 6)*cos(y) + 7*pow(sin(x), 6)*tan(z) + 21*pow(sin(x), 5)*pow(cos(y), 2) + 42*pow(sin(x), 5)*cos(y)*tan(z) + 21*pow(sin(x), 5)*pow(tan(z), 2) + 35*pow(sin(x), 4)*pow(cos(y), 3) + 105*pow(sin(x), 4)*pow(cos(y), 2)*tan(z) + 105*pow(sin(x), 4)*cos(y)*pow(tan(z), 2) + 35*pow(sin(x), 4)*pow(tan(z), 3) + 35*pow(sin(x), 3)*pow(cos(y), 4) + 140*pow(sin(x), 3)*pow(cos(y), 3)*tan(z) + 210*pow(sin(x), 3)*pow(cos(y), 2)*pow(tan(z), 2) + 140*pow(sin(x), 3)*cos(y)*pow(tan(z), 3) + 35*pow(sin(x), 3)*pow(tan(z), 4) + 21*pow(sin(x), 2)*pow(cos(y), 5) + 105*pow(sin(x), 2)*pow(cos(y), 4)*tan(z) + 210*pow(sin(x), 2)*pow(cos(y), 3)*pow(tan(z), 2) + 210*pow(sin(x), 2)*pow(cos(y), 2)*pow(tan(z), 3) + 105*pow(sin(x), 2)*cos(y)*pow(tan(z), 4) + 21*pow(sin(x), 2)*pow(tan(z), 5) + 7*sin(x)*pow(cos(y), 6) + 42*sin(x)*pow(cos(y), 5)*tan(z) + 105*sin(x)*pow(cos(y), 4)*pow(tan(z), 2) + 140*sin(x)*pow(cos(y), 3)*pow(tan(z), 3) + 105*sin(x)*pow(cos(y), 2)*pow(tan(z), 4) + 42*sin(x)*cos(y)*pow(tan(z), 5) + 7*sin(x)*pow(tan(z), 6) + pow(cos(y), 7) + 7*pow(cos(y), 6)*tan(z) + 21*pow(cos(y), 5)*pow(tan(z), 2) + 35*pow(cos(y), 4)*pow(tan(z), 3) + 35*pow(cos(y), 3)*pow(tan(z), 4) + 21*pow(cos(y), 2)*pow(tan(z), 5) + 7*cos(y)*pow(tan(z), 6) + pow(tan(z), 7); return test1_result; } double test2(double x, double y, double z) { double test2_result; test2_result = cos(cos(cos(cos(cos(cos(cos(cos(x + y + z)))))))); return test2_result; } rz#ifndef PROJECT__FILE__H #define PROJECT__FILE__H double test1(double x, double y, double z); double test2(double x, double y, double z); #endif rrrrrexpandr)rrrr4r9r5rrs r)test_complicated_codegenr,ssHHHHHHHHHHgGAq! CCFFSSVVOcc!ff,q088::; ##cc##cc##cc##cc!a%!)nn*=*=&>&>"?"?@@AABBCCDIYve5 I I IF !9Q<8 # # # # !9Q<. 0 0 0 0 b !9Q<8 # # # # !9Q<       r+c ddlm}m}ddlm}|dd\}}|d}|d}|d }|d |}|d |} t d t ||||| f|| zfd ddd\\} } \} } | dksJd}| |dd||z| zzizks>| |dd| ||zzzizks*| |dd||z| zzizks| |dd| ||zzzizksJ| dksJ| dksJdS)NrrHrn mTrNrKr4r9rPj matrix_vectorC99rFrra#include "file.h" #include void matrix_vector(double *A, int m, int n, double *x, double *y) { for (int i=0; i>>>   H|qsQw'?@@ @ @ H|q1Q3w'?@@ @ @ H|qsQw'?@@ @ @ H|q1Q3w'?@@ @ @ @ @ >>>>        r+c ddlm}m}tddt\}}|d}|d}|||}d|jj|jd z}td t||||}t}t} t| j |g} | |ksJtt5t|j |gddddS#1swxYwYdS) NrrHi mTrOclsr4r9z#include "file.h" #include void test_dummies(int m_%(mno)i, double *x, double *y) { for (int i_%(ino)i=0; i_%(ino)i| |dd| | |zzzizks*| |dd| |z| zzizks| |dd| | |zzzizksJ|dksJ|dksJdS)NrrHrn m o pTrNrKshaper4r9rPrrrrFrra##include "file.h" #include void matrix_vector(double *A, int m, int n, int o, int p, double *x, double *y) { for (int i=o; i<%(upperi)s; i++){ y[i] = 0; } for (int i=o; i<%(upperi)s; i++){ for (int j=0; j>>> q5++ ,H H|qsQw'?@@ @ @ H|q1Q3w'?@@ @ @ H|qsQw'?@@ @ @ H|q1Q3w'?@@ @ @ @ @ >>>>        r+c Hddlm}ddlm}m}t d\}}}t d||||||g}t}||gddd}|ddd ksJd } |dd | ksJdS) Nrr rrrkfoor/Frtest.cz#include "test.h" #include double foo(double x, double *y) { (*y) = sin(x); double foo_result; foo_result = cos(x); return foo_result; } rR sympy.core.relationalr rrrrrrwrite r rrr4r9r5rZcrrps r)test_output_arg_crs......CCCCCCCCgGAq!UXXaQ00##a&&9::A A WWaS&eW < double foo(double if_, double *while_) { (*while_) = sin(if_); double foo_result; foo_result = cos(if_); return foo_result; } rRrrs r) test_output_arg_c_reserved_wordsr s......CCCCCCCCn%%GAq!UXXaQ00##a&&9::A A WWaS&eW < void c(double *A, double *b, double *out) { out[0] = A[0]*b[0] + A[1]*b[1] + A[2]*b[2]; out[1] = A[3]*b[0] + A[4]*b[1] + A[5]*b[2]; out[2] = A[6]*b[0] + A[7]*b[1] + A[8]*b[2]; } ) rrrrorV result_var_namer*r^) A_symb_symrKr rcgenrZrrps r)test_multidim_c_argument_csers a # #E a # #Eu Au A !A F - - -D S!A!&AKO!AKO dkA3v 6 6 6D   8      r+c ztd\}}}td\}}tddd}t|tdd|gg}t|||z|z}t|d|z}d|||gf} d} t | d dd d ||||||f } | d d} | | ksJdS) NrkB,CrKrRrTrSr/z#include "test.h" #include void test(double x, double *C, double z, double y, double *A, double *B) { (*C) = z*(x + y); A[0] = 1; A[1] = 2; A[2] = x; (*B) = 2*x; } rFr%r&r1rrrr rr) r4r9r5rLrrKrrexpr3rrprr(s r) test_ccode_results_named_orderedrsgGAq! 5>>DAqS!QA QAq {++ , ,E QQ " "E Q!  E%./I  YVE()1aAq'9;;;F AYq\F X      r+c tddd}tddd}tddd}tddd}dt||d ddft||dddft||ddd fgf}t|d dd d }|d d}d}||ksJdS)NrKrrTrLrRrDr/rrSrFra#include "test.h" #include void test(double *A, double *B, double *C, double *D) { B[0] = A[0]; B[1] = A[1]; B[2] = A[2]; C[0] = A[3]; C[1] = A[4]; C[2] = A[5]; D[0] = A[2]; D[1] = A[5]; D[2] = A[8]; D[3] = A[11]; D[4] = A[14]; } rr rrKrLrrrrr(rps r)test_ccode_matrixsymbol_slicer 8sS!QAS!QAS!QAS!QA(1a111g.."1a111g.."1a1g..01IYve5 I I IF AYq\F  " X      r+c <td\}}}}tddd}dt|t||zg||z||zzg||z|z|zgggf}t d}t ||dd }|d d}d } || ksJdS) Nza b c derTrRr/T)r F)r`r%r&rz#include "test.h" #include void test(double a, double b, double c, double d, double *e) { const double x0 = a*b; const double x1 = c*d; e[0] = x0; e[1] = x0 + x1; e[2] = x0*x1; } )rrr rrr) r8r rdr"r generatorrr(rps r)test_ccode_cser%Us##JAq!QS!QA(1fqseacAaCi[1Q3q57)-L&M&MNNOPIT"""I Y5 N N NF AYq\F   X      r+ctddd}d}t}t||dd|f}|dd}d}||ksJdS) Nr4rSrR)r/g?F)r`r%r&r1rz#include "test.h" #include double test(double *x) { double test_result; test_result = 1.0; return test_result; } )rrr)r4rr$rr(rps r)test_ccode_unused_array_argr'ismS!QAI I Y5bcae f f fF AYq\F   X      r+c tddd}tddd}tdd}d|d |dzf}t|d d d |||f }|d d}d }||ksJdS)NXrTrRYr5TrNtestBugrrF)languager%r&r1z#include "testBug.h" #include double testBug(double *X, double *Y, int z) { double testBug_result; testBug_result = X[0] + X[1]; return testBug_result; } )rrr)r)r*r5rrr(rps r) test_ccode_unused_array_arg_funcr-|sS1AS1Ad###AAaD1Q4K(I YU%\]_`bc[d e e eF AYq\F   X      r+c\t}t|jg}|dksJdSNrr*dump_f95r_s r)test_empty_f_coder3s.zzH )2 . .F R<<<<<."rr+)rrrrrrs @@r)test_multiple_results_frLsxgGAq! UAIE UAIE GzzH <GGGGGHHHHHr+c2ttddS)Nc"tdgSrr2r;r+r)r6z#test_no_results_f..&rr+rr;r+r)test_no_results_frO%rr+cddlm}ddlm}ddlm}m}m}ddlm }ddl m }m }m }m} m} m} t#d} d|| fd || fd || fd || fd | | fd || fd|| fd|| fd| | fd|| fd|| fd| | fd|| fg } t%| dddd}|dddksJd}|dd|ksJ|dddksJd}|dd|ksJdS)Nrrrrrrr4test_absrrrrrrrrrrrrrGrFrrHaREAL*8 function test_abs(x) implicit none REAL*8, intent(in) :: x test_abs = abs(x) end function REAL*8 function test_acos(x) implicit none REAL*8, intent(in) :: x test_acos = acos(x) end function REAL*8 function test_asin(x) implicit none REAL*8, intent(in) :: x test_asin = asin(x) end function REAL*8 function test_atan(x) implicit none REAL*8, intent(in) :: x test_atan = atan(x) end function REAL*8 function test_cos(x) implicit none REAL*8, intent(in) :: x test_cos = cos(x) end function REAL*8 function test_cosh(x) implicit none REAL*8, intent(in) :: x test_cosh = cosh(x) end function REAL*8 function test_log(x) implicit none REAL*8, intent(in) :: x test_log = log(x) end function REAL*8 function test_ln(x) implicit none REAL*8, intent(in) :: x test_ln = log(x) end function REAL*8 function test_sin(x) implicit none REAL*8, intent(in) :: x test_sin = sin(x) end function REAL*8 function test_sinh(x) implicit none REAL*8, intent(in) :: x test_sinh = sinh(x) end function REAL*8 function test_sqrt(x) implicit none REAL*8, intent(in) :: x test_sqrt = sqrt(x) end function REAL*8 function test_tan(x) implicit none REAL*8, intent(in) :: x test_tan = tan(x) end function REAL*8 function test_tanh(x) implicit none REAL*8, intent(in) :: x test_tanh = tanh(x) end function rRraAinterface REAL*8 function test_abs(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_acos(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_asin(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_atan(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_cos(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_cosh(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_log(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_ln(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_sin(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_sinh(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_sqrt(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_tan(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_tanh(x) implicit none REAL*8, intent(in) :: x end function end interface )rrrrrrrrrrrrrrrrrrr)rrrrrrrrrrrrr4rrrps r)test_intrinsic_math_codegenrR)s888888::::::HHHHHHHHHH======ZZZZZZZZZZZZZZZZ A SSVV dd1gg dd1gg dd1gg SSVV dd1gg SSVV CCFF SSVV dd1gg dd1gg SSVV dd1ggIYve5 I I IF !9Q<: % % % %@  F !9Q<8 # # # # !9Q<8 # # # #M  ` !9Q<8 # # # # # #r+c0ddlm}td\}}d|||fd||zfg}t|dddd }|ddd ksJd }|dd |ksJ|d dd ksJd}|d d |ksJdS)NrrrrrrGrFrrHzREAL*8 function test_atan2(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y test_atan2 = atan2(x, y) end function REAL*8 function test_pow(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y test_pow = x**y end function rRrainterface REAL*8 function test_atan2(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y end function end interface interface REAL*8 function test_pow(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y end function end interface r)rr4r9rrrps r)test_intrinsic_math2_codegenrTs>>>>>> 5>>DAq uuQ{{# QTIYve5 I I IF !9Q<: % % % %   !9Q<8 # # # # !9Q<8 # # # #  !9Q<8 # # # # # #r+c(ddlm}m}m}t d\}}}d||||z||zdzfd||||||||||z|zfg}t |ddd d }|ddd ksJd }|dd |ksJ|d ddksJd}|d d |ksJdS)NrrrkrrrrGrFrrHa{REAL*8 function test1(x, y, z) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z test1 = sin(x)**7 + 7*sin(x)**6*cos(y) + 7*sin(x)**6*tan(z) + 21*sin(x) & **5*cos(y)**2 + 42*sin(x)**5*cos(y)*tan(z) + 21*sin(x)**5*tan(z) & **2 + 35*sin(x)**4*cos(y)**3 + 105*sin(x)**4*cos(y)**2*tan(z) + & 105*sin(x)**4*cos(y)*tan(z)**2 + 35*sin(x)**4*tan(z)**3 + 35*sin( & x)**3*cos(y)**4 + 140*sin(x)**3*cos(y)**3*tan(z) + 210*sin(x)**3* & cos(y)**2*tan(z)**2 + 140*sin(x)**3*cos(y)*tan(z)**3 + 35*sin(x) & **3*tan(z)**4 + 21*sin(x)**2*cos(y)**5 + 105*sin(x)**2*cos(y)**4* & tan(z) + 210*sin(x)**2*cos(y)**3*tan(z)**2 + 210*sin(x)**2*cos(y) & **2*tan(z)**3 + 105*sin(x)**2*cos(y)*tan(z)**4 + 21*sin(x)**2*tan & (z)**5 + 7*sin(x)*cos(y)**6 + 42*sin(x)*cos(y)**5*tan(z) + 105* & sin(x)*cos(y)**4*tan(z)**2 + 140*sin(x)*cos(y)**3*tan(z)**3 + 105 & *sin(x)*cos(y)**2*tan(z)**4 + 42*sin(x)*cos(y)*tan(z)**5 + 7*sin( & x)*tan(z)**6 + cos(y)**7 + 7*cos(y)**6*tan(z) + 21*cos(y)**5*tan( & z)**2 + 35*cos(y)**4*tan(z)**3 + 35*cos(y)**3*tan(z)**4 + 21*cos( & y)**2*tan(z)**5 + 7*cos(y)*tan(z)**6 + tan(z)**7 end function REAL*8 function test2(x, y, z) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z test2 = cos(cos(cos(cos(cos(cos(cos(cos(x + y + z)))))))) end function rRra4interface REAL*8 function test1(x, y, z) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z end function end interface interface REAL*8 function test2(x, y, z) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z end function end interface r) rrrr4r9r5rrrps r)test_complicated_codegen_f95rV snHHHHHHHHHHgGAq! CCFFSSVVOcc!ff,q088::; ##cc##cc##cc##cc!a%!)nn*=*=&>&>"?"?@@AABBCCDIYve5 I I IF !9Q<: % % % %  < !9Q<8 # # # # !9Q<8 # # # #  $ !9Q<8 # # # # # #r+c ddlm}m}ddlm}|dd\}}t |d\}}}|d|}|d |} t d t||||| f|| zfd d d d \\} } \} } | dksJd}| |ddizks | |ddizksJ| dksJ| dksJdS)NrrHrzn,mTrNAxyrPrrrGrFrrHa~subroutine matrix_vector(A, m, n, x, y) implicit none INTEGER*4, intent(in) :: m INTEGER*4, intent(in) :: n REAL*8, intent(in), dimension(1:m, 1:n) :: A REAL*8, intent(in), dimension(1:n) :: x REAL*8, intent(out), dimension(1:m) :: y INTEGER*4 :: i INTEGER*4 :: j do i = 1, m y(i) = 0 end do do i = 1, m do j = 1, n y(i) = %(rhs)s + y(i) end do end do end subroutine r A(i, j)*x(j) x(j)*A(i, j)rainterface subroutine matrix_vector(A, m, n, x, y) implicit none INTEGER*4, intent(in) :: m INTEGER*4, intent(in) :: n REAL*8, intent(in), dimension(1:m, 1:n) :: A REAL*8, intent(in), dimension(1:n) :: x REAL*8, intent(out), dimension(1:m) :: y end subroutine end interface )rWrIrJrrrXrrrs r) test_loopsr[GsL--------)))))) 75$ ' ' 'DAq+u%%GAq! C A C A") "QqT1QT71Q4<0015&V[#]#]#]JRY       * 8un55 5 5 E>222222 >>>>        r+chddlm}m}tddt\}}|d}|d}|||}d|jj|jd z}td t||||}t}t|j |g} | |ksJdS) NrrHrTrr4r9a>subroutine test_dummies(m_%(mcount)i, x, y) implicit none INTEGER*4, intent(in) :: m_%(mcount)i REAL*8, intent(in), dimension(1:m_%(mcount)i) :: x REAL*8, intent(out), dimension(1:m_%(mcount)i) :: y INTEGER*4 :: i_%(icount)i do i_%(icount)i = 1, m_%(mcount)i y(i_%(icount)i) = x(i_%(icount)i) end do end subroutine )icountmcountr) rWrIrJrr rYrrrrr*r2) rIrJrPrMr4r9rprZrrs r)test_dummy_loops_f95r_zs-------- 5$E 2 2 2DAq CA CA Aq A 7&!-@@ AH ^R!ad^^44A A aj1# & &D 8      r+c ddlm}m}ddlm}|dd\}}}}|d\}}} ||||||||f}|||||}|| |||} t dt | | ||zzfd d d d \\} } \} } | d ksJd}| |ddizks | |ddizksJ| dksJ| dksJdS)NrrHrzi,j,n,mTrNzA,x,yrrGrFrrHaasubroutine matrix_vector(A, m, n, x, y) implicit none INTEGER*4, intent(in) :: m INTEGER*4, intent(in) :: n REAL*8, intent(in), dimension(1:m, 1:n) :: A REAL*8, intent(in), dimension(1:n) :: x REAL*8, intent(inout), dimension(1:m) :: y INTEGER*4 :: i INTEGER*4 :: j do i = 1, m do j = 1, n y(i) = %(rhs)s + y(i) end do end do end subroutine rrYrZrainterface subroutine matrix_vector(A, m, n, x, y) implicit none INTEGER*4, intent(in) :: m INTEGER*4, intent(in) :: n REAL*8, intent(in), dimension(1:m, 1:n) :: A REAL*8, intent(in), dimension(1:n) :: x REAL*8, intent(inout), dimension(1:m) :: y end subroutine end interface r)rIrJrrPrrrMrKr4r9rrrrrps r)test_loops_InOutras--------))))))D111JAq!QggGAq! Ass1ayy##a))+,A Ass1ayy!A Ass1ayy!A") "QAaC..)5&e#U#U#UJRY       $ H~66 6 6 H~66 6 6 6 6 >>>>        r+c ddlm}m}ddlm}|dd\}}}}|d||f}|d }|d } |d ||d z f} |d |} t dt | | || | f|| zfdddd\\} } \}}ddt|dz td|zt|dz |z dz}| |ddizks | |ddizksJdSdS)NrrHrrTrNrKrr4r9rPrrrrGrFrasubroutine matrix_vector(A, m, n, o, p, x, y) implicit none INTEGER*4, intent(in) :: m INTEGER*4, intent(in) :: n INTEGER*4, intent(in) :: o INTEGER*4, intent(in) :: p REAL*8, intent(in), dimension(1:m, 1:p) :: A REAL*8, intent(in), dimension(1:n) :: x REAL*8, intent(out), dimension(1:%(iup-ilow)s) :: y INTEGER*4 :: i INTEGER*4 :: j do i = %(ilow)s, %(iup)s y(i) = 0 end do do i = %(ilow)s, %(iup)s do j = 1, n y(i) = %(rhs)s + y(i) end do end do end subroutine rrrR)riupilowziup-ilowrrYrZ)rWrIrJrrrrstrrs r)test_partial_loops_frfs.-------))))))D111JAq!Q C1v&&&A CA CA C!QUA C A") "QqT1QT71Q4<0015&V[#]#]#]JRY *1q5zzAE A NN   +H8 8un55 5 5 E>222222 6 522r+c Dddlm}ddlm}m}t d\}}}t d||||||g}t}||gddd}|ddd ksJ|dd d ksJdS) Nrr rrkrr/Frztest.f90rRz~REAL*8 function foo(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(out) :: y y = sin(x) foo = cos(x) end function ) rr rrrrrrr) r rrr4r9r5rZrrs r)test_output_arg_frhs......CCCCCCCCgGAq!UXXaQ00##a&&9::A A WWaS&eW < *>??G aj7) , ,D   "D 8d? " " 4":%%%%% # "%%r+ctd}d}|D]}||z }td|}t}t|j|g}d}||ksJdS)Nzx:20rr/a REAL*8 function test(x0, x1, x10, x11, x12, x13, x14, x15, x16, x17, x18, & x19, x2, x3, x4, x5, x6, x7, x8, x9) implicit none REAL*8, intent(in) :: x0 REAL*8, intent(in) :: x1 REAL*8, intent(in) :: x10 REAL*8, intent(in) :: x11 REAL*8, intent(in) :: x12 REAL*8, intent(in) :: x13 REAL*8, intent(in) :: x14 REAL*8, intent(in) :: x15 REAL*8, intent(in) :: x16 REAL*8, intent(in) :: x17 REAL*8, intent(in) :: x18 REAL*8, intent(in) :: x19 REAL*8, intent(in) :: x2 REAL*8, intent(in) :: x3 REAL*8, intent(in) :: x4 REAL*8, intent(in) :: x5 REAL*8, intent(in) :: x6 REAL*8, intent(in) :: x7 REAL*8, intent(in) :: x8 REAL*8, intent(in) :: x9 test = x0 + x1 + x10 + x11 + x12 + x13 + x14 + x15 + x16 + x17 + x18 + & x19 + x2 + x3 + x4 + x5 + x6 + x7 + x8 + x9 end function r:)r4r3symror`r(rps r)test_f_code_call_signature_wraprp sxA D  64((GzzH )G9 5 5FH8 X      r+c`td\ttfddS)Nzx,Xc0tdzfddS)Nr/f95r$)rr)r4sr)r6z!test_check_case..Js&!A#x!H!Hr+)rrrrts@@r)test_check_caseruHs6 5>>DAq <HHHHHIIIIIr+ctd}tdd} td||zfdddS#t$r:}|jddrt d Yd}~dSd}~wwxYw) Nr4T) my_assumptionr/rsr$rzFortran ignores case.z$This exception should not be raised!)rrrrl startswithAssertionError)x1x2r"s r)test_check_case_false_positiver|Ms B D ) ) )BIB11111 III 6!9   7 8 8 I !GHH H I I I I I IIs: A>/A99A>c,td\}}dd|zfg}t|ddd}t|dddd}|dd|ddksJd||zf}t|ddd}t|dddd}|dd|ddksJdt||g||z||z ggf}t|d dd}t|d ddd}|dd|ddksJdS) NrrrSrGFrrrRr)rrr)r4r9rr expresults r) test_c_fortran_omit_routine_namer[sG 5>>DAq1I Ye5 A A AF 5%UKKKI !9Q<9Q<? * * * *! I Ye5 A A AF 5%UKKKI !9Q<9Q<? * * * *A1ac 3445I Ye5 A A AF 5%UKKKI !9Q<9Q<? * * * * * *r+c@td\}}}||z}t||g|dgg}d||ff}t|dddd}|dd}d }|d } | d } | d} |d | iz}||ksJdS) Nrkr/rsFrrrRa2REAL*8 function test(x, y, z, out_%(hash)s) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z REAL*8, intent(out), dimension(1:2, 1:2) :: out_%(hash)s out_%(hash)s(1, 1) = x out_%(hash)s(2, 1) = z out_%(hash)s(1, 2) = y out_%(hash)s(2, 2) = 16 test = x + y end function r_hash)rrr splitlinessplit) r4r9r5e1e2rrr(rpr8r rs r)test_fcode_matrix_outputrmsgGAq! QB !Q!R! " "B"b"I Yve5 I I IF AYq\F   AA  A A$C63-'H X      r+c ztd\}}}td\}}tddd}t|tdd|gg}t|||z|z}t|d|z}d|||gf} t | ddd d ||||||f } | d d} d } | | ksJdS) NrkrrKrRrTrSr/rsFrra!subroutine test(x, z, y, C, A, B) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: z REAL*8, intent(in) :: y REAL*8, intent(out) :: C REAL*8, intent(out) :: B REAL*8, intent(out), dimension(1:1, 1:3) :: A C = z*(x + y) A(1, 1) = 1 A(1, 2) = 2 A(1, 3) = x B = 2*x end subroutine r) r4r9r5rLrrKrrrrrr(rps r) test_fcode_results_named_orderedrsgGAq! 5>>DAqS!QA QAq {++ , ,E QQ " "E Q!  E%./I Yve5()1aAq'9;;;F AYq\F  X      r+c tddd}tddd}tddd}tddd}dt||d ddft||dddft||dddfgf}t|d dd d }|d d}d }||ksJdS)NrKrSrTrLrRrrr/rrsFrasubroutine test(A, B, C, D) implicit none REAL*8, intent(in), dimension(1:2, 1:3) :: A REAL*8, intent(out), dimension(1:1, 1:3) :: B REAL*8, intent(out), dimension(1:1, 1:3) :: C REAL*8, intent(out), dimension(1:2, 1:1) :: D B(1, 1) = A(1, 1) B(1, 2) = A(1, 2) B(1, 3) = A(1, 3) C(1, 1) = A(2, 1) C(1, 2) = A(2, 2) C(1, 3) = A(2, 3) D(1, 1) = A(1, 3) D(2, 1) = A(2, 3) end subroutine rrs r)test_fcode_matrixsymbol_slicersS!QAS!QAS!QAS!QA(1a111g.."1a111g.."1a1g..01IYve5 I I IF AYq\F  " X      r+ctddd}d|dddff}t|dddd}|d d}d }|d}|d }|d}|d |iz}||ksJdS) NrKrSrTr/rRrsFrrzsubroutine test(A, out_%(hash)s) implicit none REAL*8, intent(in), dimension(1:2, 1:3) :: A REAL*8, intent(out), dimension(1:2, 1:1) :: out_%(hash)s out_%(hash)s(1, 1) = A(1, 2) out_%(hash)s(2, 1) = A(2, 2) end subroutine rr)rrrr)rKrrr(rpr8r rs r)&test_fcode_matrixsymbol_slice_autonamersS!QA111a4!I Yve5 I I IF AYq\F   AA  A A$C63-'H X      r+ctd\}}}}td||zfddd|f}|dd}d}||ksJd }td||z|zfd dd||f}|dd}||ksJdS) Nzx y z tfrGF)r%r& global_varsrrRzPREAL*8 function f(x) implicit none REAL*8, intent(in) :: x f = x*y end function z#include "f.h" #include double f(double x, double y) { double f_result; f_result = x*y + z; return f_result; } rr)r4r9r5trr(rps r)test_global_varsrs##JAq!Q c1Q3ZuE"#'''F AYq\F   X      c1Q3q5\3uE"#Q)))F AYq\F X      r+cddlm}ddlm}|dddii}t d\}}|||z}t |d g }d }t d |fd d |}|dd} | |ksJt |}|jd d}t d |fd d |}|dd} | |ksJdS)NrC99CodePrinter)expuser_functionsrfastexp)settingszx yz#include "fastexp.h")printerpreprocessor_statementsz#include "expr.h" #include "fastexp.h" double expr(double x, double y) { double expr_result; expr_result = fastexp(x + y); return expr_result; } r3F)r%r&r`rR)rz#include "expr.h" #include #include "fastexp.h" double expr(double x, double y) { double expr_result; expr_result = fastexp(x + y); return expr_result; } ) sympy.printing.crrrrrrrappend) rrrr4r9r3genrprr(s r)test_custom_codegenrs<//////::::::n'7%9K&LMMMG 5>>DAq 3q1u::D W.D-E G G GC  fd^E M M MF AYq\F X     W % % %C&&'=>>>  fd^E M M MF AYq\F X      r+c ddlm}Gdd|}td}|dz}ddg}td |fd d d d | }||ksJdS)NrrceZdZdZdS)*test_c_with_printer..CustomPrintercd||j||jS)Nzfastpow({}, {}))format_printbaser)selfr3s r) _print_Powz5test_c_with_printer..CustomPrinter._print_Pow+s>$++DKK ,B,B,0KK,A,ACC Cr+N)__name__ __module__ __qualname__rr;r+r) CustomPrinterr*s( C C C C Cr+rr4rT)rz#include "file.h" #include double test(double x) { double test_result; test_result = fastpow(x, 3); return test_result; } )rzP#ifndef PROJECT__FILE__H #define PROJECT__FILE__H double test(double x); #endif r/rrF)r%r&r)rrrr)rrr4r3rprs r)test_c_with_printerr's//////CCCCCCCC  A a4D  Hfd^SUVcVcVeVe f f fF X      r+cddl}d|jj_t dd}t dd}td||zfdddd }|dd }d }||ksJt d}t dd}td||zfdddd }|dd }d }||ksJd|jj_dS) NrTr4)realr9r/rsFrrRzsREAL*8 function test(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y test = x + y end function z{COMPLEX*16 function test(x, y) implicit none COMPLEX*16, intent(in) :: x REAL*8, intent(in) :: y test = x + y end function )sympy.utilities.codegen utilitiesrCOMPLEX_ALLOWEDr )sympyr4r9rr(rps r)test_fcode_complexrDs"""".2EO+sAsA fQqS\5&e L L LFQilF   X    s AsA fQqS\5&e L L LFQilF   8    .3EO+++r+N)rFF)_ior sympy.corerrrrrr rr rr 'sympy.functions.special.error_functionsr sympy.integrals.integralsrsympy.matricesrrrrrrrrrrrrrrsympy.testing.pytestrsympy.utilities.lambdifyrr*r[rarfrirqrsrxrzr}rrrrrrrrrrr rrr r%r'r-r3r5r7r;r=r@rBrErIrLrOrRrTrVr[r_rarfrhrmrprur|rrrrrrrrrr;r+r)rs>>>>>>>>>>>>>>>>******$$$$$$777777......////////('''''999999    ???6@@@ $TTT $$"$   , I I I999999x0AAAH'''T$$$.)))X$$$*$$$*.4:(&&     $    & & &$&6 I I I999o$o$o$d+$+$+$\;$;$;$|000f0...b,3,3,3^&&&&6%%%PJJJ I I I+++$:<<.8,,,\:44444r+