o "gD @s dZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z zddl Z Wn e y9dZ Ynwddl mZddlTddl mZejadadadadad ad ad adadadagad ad ada ia!d a"da#da$da%t%gia&d a'ga(d a)ga*da+d a,ga-ga.ga/d d Z0dddZ1de_2iZ3dD]Z4ddie3e4<qdD]Z4ddie3e4<qiZ5iZ6dD]Z7e7de5e7<e7e6e7d<qddZ8ddZ9ddZ:ddZ;eZ?eZ@eZAeej=d?fZZe -m ,--ignore-contains Functions: crackfortran, crack2fortran The following Fortran statements/constructions are supported (or will be if needed): block data,byte,call,character,common,complex,contains,data, dimension,double complex,double precision,end,external,function, implicit,integer,intent,interface,intrinsic, logical,module,optional,parameter,private,public, program,real,(sequence?),subroutine,type,use,virtual, include,pythonmodule Note: 'virtual' is mapped to 'dimension'. Note: 'implicit integer (z) static (z)' is 'implicit static (z)' (this is minor bug). Note: code after 'contains' will be ignored until its scope ends. Note: 'common' statement is extended: dimensions are moved to variable definitions Note: f2py directive: f2py is read as Note: pythonmodule is introduced to represent Python module Usage: `postlist=crackfortran(files)` `postlist` contains declaration information read from the list of files `files`. `crack2fortran(postlist)` returns a fortran code to be saved to pyf-file `postlist` has the following structure: *** it is a list of dictionaries containing `blocks': B = {'block','body','vars','parent_block'[,'name','prefix','args','result', 'implicit','externals','interfaced','common','sortvars', 'commonvars','note']} B['block'] = 'interface' | 'function' | 'subroutine' | 'module' | 'program' | 'block data' | 'type' | 'pythonmodule' | 'abstract interface' B['body'] --- list containing `subblocks' with the same structure as `blocks' B['parent_block'] --- dictionary of a parent block: C['body'][]['parent_block'] is C B['vars'] --- dictionary of variable definitions B['sortvars'] --- dictionary of variable definitions sorted by dependence (independent first) B['name'] --- name of the block (not if B['block']=='interface') B['prefix'] --- prefix string (only if B['block']=='function') B['args'] --- list of argument names if B['block']== 'function' | 'subroutine' B['result'] --- name of the return value (only if B['block']=='function') B['implicit'] --- dictionary {'a':,'b':...} | None B['externals'] --- list of variables being external B['interfaced'] --- list of variables being external and defined B['common'] --- dictionary of common blocks (list of objects) B['commonvars'] --- list of variables used in common blocks (dimensions are moved to variable definitions) B['from'] --- string showing the 'parents' of the current block B['use'] --- dictionary of modules used in current block: {:{['only':<0|1>],['map':{:,...}]}} B['note'] --- list of LaTeX comments on the block B['f2pyenhancements'] --- optional dictionary {'threadsafe':'','fortranname':, 'callstatement':|, 'callprotoargument':, 'usercode':|, 'pymethoddef:' } B['entry'] --- dictionary {entryname:argslist,..} B['varnames'] --- list of variable names given in the order of reading the Fortran code, useful for derived types. B['saved_interface'] --- a string of scanned routine signature, defines explicit interface *** Variable definition is a dictionary D = B['vars'][] = {'typespec'[,'attrspec','kindselector','charselector','=','typename']} D['typespec'] = 'byte' | 'character' | 'complex' | 'double complex' | 'double precision' | 'integer' | 'logical' | 'real' | 'type' D['attrspec'] --- list of attributes (e.g. 'dimension()', 'external','intent(in|out|inout|hide|c|callback|cache|aligned4|aligned8|aligned16)', 'optional','required', etc) K = D['kindselector'] = {['*','kind']} (only if D['typespec'] = 'complex' | 'integer' | 'logical' | 'real' ) C = D['charselector'] = {['*','len','kind','f2py_len']} (only if D['typespec']=='character') D['='] --- initialization expression string D['typename'] --- name of the type if D['typespec']=='type' D['dimension'] --- list of dimension bounds D['intent'] --- list of intent specifications D['depend'] --- list of variable names on which current variable depends on D['check'] --- list of C-expressions; if C-expr returns zero, exception is raised D['note'] --- list of LaTeX comments on the variable *** Meaning of kind/char selectors (few examples): D['typespec>']*K['*'] D['typespec'](kind=K['kind']) character*C['*'] character(len=C['len'],kind=C['kind'], f2py_len=C['f2py_len']) (see also fortran type declaration statement formats below) Fortran 90 type declaration statement format (F77 is subset of F90) ==================================================================== (Main source: IBM XL Fortran 5.1 Language Reference Manual) type declaration = [[]::] = byte | character[] | complex[] | double complex | double precision | integer[] | logical[] | real[] | type() = * | ([len=][,[kind=]]) | (kind=[,len=]) = * | ([kind=]) = comma separated list of attributes. Only the following attributes are used in building up the interface: external (parameter --- affects '=' key) optional intent Other attributes are ignored. = in | out | inout = comma separated list of dimension bounds. = [[*][()] | [()]*] [// | =] [,] In addition, the following attributes are used: check,depend,note TODO: * Apply 'parameter' attribute (e.g. 'integer parameter :: i=2' 'real x(i)' -> 'real x(2)') The above may be solved by creating appropriate preprocessor program, for example. N)Path) __version__)*)symbolicfix cCspdadadadadadadadadada ga da t gia da dadagaiadadadagagagagadadS)Nrrrrr r ) strictf77sourcecodeformquietverbosetabchar pyffilename f77modulename skipemptyendsignorecontains dolowercasedebug groupcounter grouplist neededmodule expectbeginskipblocksuntil usermodules f90modulevars gotnextfilefilepositiontextcurrentfilename skipfunctions skipfuncs onlyfuncs include_pathsprevious_contextr%r%Y/var/www/html/ecg_monitoring/venv/lib/python3.10/site-packages/numpy/f2py/crackfortran.pyreset_global_f2py_varss4 r'cCs0tsdSts|rtjttj|dSdSN)rr sysstdoutwriter)lineflagr%r%r&outmesss r.2zabcdefghopqrstuvwxyz$_typespecrealijklmninteger)3intdoublefloatcharshortlongvoidcasewhilereturnsignedunsignediffortypedefsizeofunionstructstaticregisternewbreakdogotoswitchcontinueelseinlineexterndeleteconstautolenrankshapeindexslensize_imaxminflenfshapestringcomplex_double float_doublestdinstderrr*typedefault_bncC(|tvrtd|t|ft|S|S)Nz&rmbadname1: Replacing "%s" with "%s". )badnameserrmessnamer%r%r& rmbadname1   rlcCdd|DS)NcSg|]}t|qSr%)rl.0_mr%r%r& zrmbadname..r%namesr%r%r& rmbadnamerwcCrg)Nz+undo_rmbadname1: Replacing "%s" with "%s". ) invbadnamesrirjr%r%r&undo_rmbadname1rmrzcCrn)NcSror%)rzrpr%r%r&rsrtz"undo_rmbadname..r%rur%r%r&undo_rmbadnamerxr{z-\*-\s*fortran\s*-\*-z-\*-\s*f90\s*-\*-z-\*-\s*fix\s*-\*-z[^c*]\s*[^\s\d\t])z.f90z.f95z.f03z.f08)z.forz.ftnz.f77z.fcCstdur t|j}nFtdtj|}t|d0}| |}| t j r*d}n| t j t jfr6d}n| t jt jfrBd}nd}Wdn1sNwYt|||dS) aNEnsures that filename is opened with correct encoding parameter. This function uses charset_normalizer package, when available, for determining the encoding of the file to be opened. When charset_normalizer is not available, the function detects only UTF encodings, otherwise, ASCII encoding is used as fallback. N rbz UTF-8-SIGzUTF-32zUTF-16ascii)encoding)charset_normalizer from_pathbestrr\ospathgetsizeopenread startswithcodecsBOM_UTF8 BOM_UTF32_LE BOM_UTF32_BEBOM_LEBOM_BE)filenamemodernbytesfhandlerawr%r%r&openhook,s     rcCs"d}t|jtvr d}t|du}|}d}t|r d}nt|r(d}d}|dkrg|ro|ddkrU|rU|d8}|ddkrJt |d d sR|d d d krUd}n*|}|dkrw|s.Wd |SWd |SWd |SWd |SWd |S1swY|S)z(Check if file is in free format Fortran.FTrr!r Nr &) rsuffixlowerCOMMON_FREE_EXTENSIONSrreadline _has_f_header_has_f90_headerstrip_free_f90_start)fnameresultrr,nr%r%r&is_free_formatKsF  ,     rc Cs^|s tttttttttf }|gkrdSt}d}d}d}t dt j }t d} t d} t d} |r8|ddd \}} d gd d t d D} dat j|td} z|}Wntyv}ztd|d|d|dd}~ww|s{n$|rda|ada|} dadatjtd}ttjtvrt|st|sdan ttrt|sdatrt ant!at"dt#tttrdpdf|$%dd }|dks|ddvrn |dd}|dkrd}t&|d\}}|d 7}|dddkrt&|d|ddd\}}d}|'dkrtdkrnd}qQtdkr|dd vrF|ddd!krAd"|dd}d}nd}d}qQtrVt(|d#krV|dd#}|d| vretd$t#||rktrt(|dkr|dd ks||d%d}d}d}n| )|}|r|*d&}|r|| )|*d&}d}d}nd"|dd}|r|}n|}|}|}nytdkr-|s|d'kr| )|r|d(} |}|st+d)n ||}| )|rnq|,}| )|}|r|*d&}|r|| )|*d&}d}d}n|r"|rt-|s|n|}n|}|}|}|du}nt.d*t#td+|/dt| fa|)|}|r|*d,}tj0|r\t1||dd-n@tj2tgt3}d}|D]}tj4||}tj0|rd}t1||dd-nqi|st"d.t#|tj54|fn|||} qR|r|}n|}|}d+|/dt| fa|)|}|r|*d,}tj0|rt1||dd-n@tj2tgt3}d}|D]}tj4||}tj0|rd}t1||dd-nq|s t"d.t#|tj54|fn||da|6|r"|dddS|\ aaaaaaaaadS)/z Read fortran codes from files and 1) Get rid of comments, line continuations, and empty lines; lower cases. 2) Call dowithline(line) on every line. 3) Recursively call itself when statement "include ''" is met. NFr z*\s*include\s*(\'|")(?P[^\'"]*)(\'|")z(?P.*)&\s*\Zz(\s*&|)(?P.*)z.*?'''r )r r  cSror%)strrpr%r%r&rsrtz#readfortrancode.. )rTzreadfortrancode: reading #z failed with zu. It is likely that installing charset_normalizer package will help f2py determine the input file encoding correctly.rrrfreez Reading file %s (format:%s%s) z,strict z rrz!f2pyr)rcrCrf2pyz Hzxreadfortrancode: Found non-(space,digit) char in the first column. Are you sure that this code is in fix form? line=%sr,.pyf z.Unexpected end of file when reading multiline z6Flag sourcecodeform must be either 'fix' or 'free': %szLine #%d in %s:"%s" rk) dowithlineistopzAreadfortrancode: could not find include file %s in %s. Ignoring. )7rrrr r beginpatternr rrrecompileIrange fileinput FileInputrrUnicodeDecodeError Exceptionrlineno isfirstlinerrsplitextrrrCOMMON_FIXED_EXTENSIONSr_has_fix_headerrbeginpattern77beginpattern90r.repr expandtabsreplacesplit_by_unquotedrrTmatchgrouprirstripiscstyledirective ValueError filelinenoisfilereadfortrancodedirnamer#joinpathsepclose)ffilerr saveglobalslocaldolowercasecont finallinell includelinecont1cont2 mline_markl1 spacedigitsfinlmsgextis_f2py_directiverl_ origfinallinerlcmfn include_dirs foundfileinc_dirfn1r%r%r&res            (                rzN\s*(?P%s(?=\s*(\b(%s)\b)))\s*(?P(\b(%s)\b))\s*(?P%s)\s*\Zzqcharacter|logical|integer|real|complex|double\s*(precision\s*(complex|)|complex)|type(?=\s*\([\w\s,=(*)]*\))|byte.*rdz|static|automatic|undefined)z([a-z]+[\w\s(=*+-/)]*?|)functionrrbegin)z [a-z\s]*? subroutinerrzprogram|block\s*dataz@|module(?!\s*procedure)|python\s*module|(abstract|)\s*interface|z type(?!\s*\()z\end|endprogram|endblockdata|endmodule|endpythonmodule|endinterface|endsubroutine|endfunctionendzDend\s*(if|do|where|select|while|forall|associate|critical|enum|team)z[\w]*?endifzmodule\s*proceduremoduleprocedure)r implicitrrr)r dimension|virtualrr dimension)r externalrrr)r optionalrrr)r requiredrrr)r publicrrr)r privaterrr)r intrinsicrrr)r intent|depend|note|checkrz \s*\(.*?\).*intent)r parameterrz\s*\(.*r)r datarrr)r callrrr)r entryrrr)r callfunrrr)r commonr rr )r user rr )r containsr r r )r formatr rr )r Kthreadsafe|fortranname|callstatement|callprotoargument|usercode|pymethoddefr rf2pyenhancementsz2\s*(?P''')(?P.*?)(?P''')\s*\Z multilinec Csvtdt|@r Jdtdjdt|dt|ddd}||}|r7|}|d |d fS|d fS) z Splits the line into (line[:i], line[i:]), where i is the index of first occurrence of one of the characters not within quotes, or len(line) if no such index exists z"'zcannot split by unquoted quoteszR\A(?P({single_quoted}|{double_quoted}|{not_quoted})*)(?P{char}.*)\Zz[^"'{}]z[{}]z('([^'\\]|(\\.))*')z("([^"\\]|(\\.))*")) not_quotedr7 single_quoted double_quotedbeforeafterr )setrrr escaper groupdict)r, charactersrrdr%r%r&rs rcCsBg}t|dD]}dD]}||d}q ||q d|S)N@,@z(),r,)markoutercommasplitrappendr)argslinearrr%r%r& _simplifyargss   r!z"\s*(?P\b[a-z]+\w*\b)\s*=.*z \s*(?P\b[a-z]+\w*\b)\s*=.*zH\s*bind\(\s*(?P[^,]+)\s*,\s*name\s*=\s*"(?P[^"]+)"\s*\)cCst|d\}}|rCtd|sCtd|sC|dks!Jt|t|d\}}|r[^"]*)\b%s\b\s*@\(@(?P[^@]*)@\)@.*\Zrargszcallfun %s(%s) result (%s)rzcallfun %s(%s)z8crackline: could not resolve function call for line=%s. rrz"crackline:%d: No pattern for line rz>crackline: groupcounter(=%s) is nonpositive. Check the blocks.thiszGcrackline: End group %s does not match with previous Begin group %s %srrrr );rf2pyenhancementspatternrmultilinepatternr crackliner groupname groupcacherrrrr.rdimensionpatternexternalpattern intentpatternoptionalpatternrequiredpatternparameterpattern datapattern publicpatternprivatepatternintrinsicpattern endifpattern endpattern formatpatternrfunctionpatternsubroutinepatternimplicitpattern typespattern commonpattern callpattern usepatterncontainspattern entrypatternmoduleprocedurepatterncrackline_re_1ryrmarkouterparenrrr!callfunpattern analyzelinerrrendswithr$rrrrrr)r,resetr has_semicolonsemicolon_lineflr-patrre_1rkm1m2r r%r%r&r-s4                  $    r-cCsdd}d}|D])}|dkr|d}|dkr|d}qn|dkr+|d}|dkr+|d}q||}q|S)Nr r(r@(@)@)@r%)r,rfrr%r%r&rHWs  rHrcCsd}d}t||d\}}||7}|rT|d|kr&|dkr&|d|d7}n||d7}|ddkr7|d7}n |ddkrA|d8}t|dd|d\}}||7}|s|r_Jt|||f|S)Nr r()@rTrrV)rr)r,commarrXrrr%r%r&ris"     rcCs|dddd}|S)NrUrTrWrV)r)r,rr%r%r&unmarkouterparen|sr\cCs|si}|s|S||ur|St|D]g}|dkr'|s ||vr&||||<q|dkr9||D]}t|||}q/q|dkrFt||||}q|dkrSt||||}q|dvrd|s]||vrc||||<q|dkriq|dvrttd|qtd t||S) Nr0attrspec kindselector charselector)=typenamenote)rcheckrrrdependz"appenddecl: "%s" not implemented. z-appenddecl: Unknown variable definition key: )listkeys setattrspecsetkindselectorsetcharselectorrirr)decldecl2forcekrr%r%r& appenddecls>     rnzD\s*(?P(@\(@.*?@\)@|\*[\d*]+|\*\s*@\(@.*?@\)@|))(?P.*)\Zz[(?:,(?P[\w(),]+))?(::)?(?P\b[a-z$_][\w$]*\b)(?:\((?P[\w,]*)\))?\Zz\s*(?P\b[\w$]+\b)\s*(@\(@\s*(?P[\w\s,]*)\s*@\)@|)\s*((result(\s*@\(@\s*(?P\b[\w$]+\b)\s*@\)@|))|(bind\s*@\(@\s*(?P(?:(?!@\)@).)*)\s*@\)@))*\s*\ZzF\s*(?P(operator|assignment))@\(@\s*(?P[^)]+)\s*@\)@\s*\Zz9\s*(?P\b[\w$]+\b)\s*@\(@\s*(?P.*)\s*@\)@\s*\Zz4([-+]?(?:\d+(?:\.\d*)?|\d*\.\d+))[dD]((?:[-+]?\d+)?)zA([-+]?((?:\d+(?:\.\d*)?|\d*\.\d+))[eE]((?:[-+]?\d+)?)|(\d+\.\d*))zintent\s*\(.*?\bcallback\bcCs&|dgD] }t|rdSqdS)Nr]rr)get_intentcallbackpatternr)vdeclr r%r%r&_is_intent_callbacks  rrcCsld|}t|}t|||r1|d}|r$dd|dDng}|d||dfSdgdfS)Nr attributescSg|]}|qSr%)r)rqr r%r%r&rsrtz*_resolvetypedefpattern..rrkparams)rrtypedefpatternrprintr)r,rRattrsr%r%r&_resolvetypedefpatterns    rycCsRtdtj}||}d}|r%|d}|d|||d}||fS)NzObind\(\s*(?P[^,]+)(?:\s*,\s*name\s*=\s*["\'](?P[^"\']+)["\']\s*)?\)r)rrrsearchrstartr)r,patternrbind_statementr%r%r&parse_name_for_binds   r~cCst|\}}t|}t|}|r |d|d|d|fSt|}|r;|dd|dd}|gddfSt|}|rN|d|dddfSdgddfS)Nrkr)rschemerTrV)r~rHnameargspatternrroperatorpatterncallnameargspattern)r, bind_cnamerRrkr%r%r&_resolvenameargspatterns      rcGCs|d}|dkr datrZ|dvrZtsZtdkrZtjt dd}t d|da tdad t t<it t<gtt<gt td <it td <d t td <|t td <dt td<da|dvr |}td|tjrnd}ntd|tjryd}n td|tjrd}|dkrt|d\}}}t|dt td |<g}d} n t|d\}}} } |dur|dkrd}nd}|dvrt d||tfa|rtddt| d D}ng}d|vrd|vr|dd|vst d!d} d} |d"vr1d} d#t tvrdS|t td#vr dSttD] } | d |krdSq|t td$vr*dSd%d&d"|}trKtd'krKtdkrKtd(ad} |d)vrKd} tdait t<gtt<| rtdkrht d*ttdd+t t<dt td <tt td <dt td<gt td <gt td,<gt td$<it td <tdait t<gtt<| rtdkrt d-tdd.t t<d.t td <d/t td <d0t tddt tdd ft td<gt td <gt td,<gt td$<it td <tdait t<gtt<|t t<|t td <|sd1|d2d3}|d4t td5<t|t td <| t td6<tdkr3tt td<n/trLtd7krLd0t tddtft td<nd0t tddt tdd ft td<tt t D]}t t|sxt t|=qj|t td#<gt td <gt td,<gt td$<it td <it td8<|dkrgt td9<|d"vr|t td(d,vrt td(d,!|t"#t td(d t td <zt td |t td |d:$d;=Wn t%yYnw|dt td=|d><|d?r-|d?t td=|d <zt(t td |t td(d dt td |<Wn t%yRYnw|d@kr| r| t td vr|| kst(t td |t td | t td |<z t td(d$!|Wn t%yYnw|d&krt)d|d4d2|}|rt*|d|d\}}}}t+|||||d"vrttd!t tttttdd'd <tt=tdattd!t tttttdd'd <tt=tdadSdS|d8krNt|d\}}} }|durL|r0tdAdt| d D}ng}| dus=Jt| |t td8|<d8|tfadSdS|dkrtt*||d\}}}}t+||||}|durrdB|tfadSdS|dCvrDt td }|d,}|-dD}|dkr|dEkrt.|-dFd(}|d|ddD||dd}|-dD}||ddDkrd#t tvrt dG|d|d|f|dH/t td#}|dkrd}d}n|d|,}||d(d}t| d }t0|dkr|d}t dIdH/|ddd}dJdt| d DD]}t1|}|sK|dKvr9d}nt2|3t dL|t|fq&t|d }|dKvrh|dMksa|dNkrh||d7}||vrqi||<|dOkr}||d}|dEkr|d|}t4|r|t td#vrtdkrdPt td(d vrt dQ|t td krt dR|t td ft td#!|n t5dS|nt5dT||dUvr|}d:||vr||d:!|n|g||d:<|d;kr,t td d kr t dVq&|t td#vrq&d,t tvr#gt td,<t td,!||}q&|t td <|durBdB|tfadSdS|dWkr\dXd|d dHDt tdY<dS|dZkrt td }|d,dd'}d}t| d D]}zd[d| d\D\}}Wnt%yt d]||fYq{wt6|}t|}||vri||<d\||vr||d\|kst d^|||d\|ft7||}|r"|8d_d`kr t|}t9:|D]}t||;|<dadb||;|<<qd/|}n|8d_dckr"|dddadbdHdd}zt=|i|} Wnt>t?t@fyI}!zt5de||!fWYd}!~!q{d}!~!wwt| ||d\<d:||vrc||d:!dZndZg||d:<|}q{|t td <|durdB|tfadSdS|dfkr|d,dgkrdt tdf<dS|drdft tvrt tdf}"ni}"|"durt dhi}"t|d d D]}i}#tdi|tj}|st dj|qtA|d}$|$st dk|qt*|$d|$d\}}}}tB||\}%}&}'||#d_<|%|#dl<|&|#dm<|'|#dn<t|# D] }|#|s'|#|=qt|d d D]\}(do|(vrZzdpd|( doD\})}*Wnt%yYt dq|(Yq3w|(,})}*t0|)t0|*krpdksynt dr|(q3tCtD|)tD|*dD] }+|#|"tE|+<qq3q|"t tdf<dSdS|dsk rg}d},d}-d}.d}/d}0|dD]i}1|0s|1dtkr|/ }/|1dukr|/r|.d}.q|1dvkr|0d}0n |1dwkr|0d}0|.dkr|,|1},q|.dkr|-|1}-q|.d(k r|,,},|,FdH r |,dd,},|!|,|-g|1},d}-d}.q|.d(k r:|,,},|,FdH r3|,dd,},|!|,|-gt t8d i}2d}|D]*}3|3d,|3d,|3d<|3d<|3dFdH rl|3ddd|3d<|3dFdv r~t dx|3d qFtGtdydt|3d d DD]\}4} | Fdv rt dx|  qdz|3dv rt d{|3d q|2H| i|2| 8d_}5tI|2| }6|5dck rtJd||3dn|3d dH}7z|6 rd}Kd~/|7n|7|4}8WnbtL yMtMdd|7D rd}.d}?d}@|D]:}1|1duk r|.d}. q|.d7k r|?,}?|? sd}?|>!|?|@g|.d(}.d}?d}@|.d( r|?|1}? q|@|1}@ q|?,}?|? sd}?|>!|?|@gi}Adt tv rt td}A|>D].}1|1d|Av rg|A|1d<ddt|1d d DD]}| r,|A|1d!| q q|At td<d|?tfadS|dk rtd|dtj}| r|3}Bdt tv rait td<|d }it td|<d}Cd|Bv r|Bddu rd|Bv r|Bddu rd}C|Ct td|d<dd|Bd dHD}i}D|D]9}3d\|3v rtd|3tj}$|$ r|$d,|D|$d,<n t dt|3n|3|D|3<|Dt td|d< qdS dSt2|3t ddS|dv r;dt tv rit td<t td}E|ddk r/d|Ev r/tQ|EdtR r#|Edg|Ed<|Ed!|ddS|d|E|d<dS|dk ratdu rNt rLt ddSt}FtSt |Ftdd(|ddStdk rrt2|3t ddSdS)a Reads each line in the input file in sequence and updates global vars. Effectively reads and collects information from the input file to the global variable groupcache, a dictionary containing info about each part of the fortran module. At the end of analyzeline, information is filtered into the correct dict keys, but parameter values and dimensions are not yet interpreted. r*rN)rrrrdr.rzBanalyzeline: no group yet. Creating program group with name "%s". programr#r$r%rkfromskyfrom)rrrz block\s*data block datazpython\s*module python modulezabstract\s*interfaceabstract interfacerdr)r] _BLOCK_DATA_r ) interfacerrz2analyzeline: No name/args pattern found for line. cSrtr%rrqxr%r%r&rs zanalyzeline..rz.*?)\s*(\(\s*(?P[a-z-, ]+)\s*\)\s*|)\ZzDanalyzeline: could not extract info of implicit statement part "%s" zManalyzeline: could not extract types pattern of implicit statement part "%s" r^r_ra-cSrtr%rrr%r%r&rsgrtzZanalyzeline: expected "-" instead of "%s" in range list of implicit statement z^analyzeline: expected "-" instead of "%s" in range list of implicit statement (2) r'/rTrVz>analyzeline: implied-DO list "%s" is not supported. Skipping. cSrtr%rrr%r%r&rsrtrz=Comment line in declaration "%s" is not supported. Skipping. z\(.*?\)z(/{}/), css|]}d|vVqdS)rNr%)rqrr%r%r& zanalyzeline..rz=analyzeline: changing init expression of "%s" ("%s") to "%s" r z//_BLNK_cSrtr%rrr%r%r&rsrtr zR\A\s*(?P\b\w+\b)\s*((,(\s*\bonly\b\s*:|(?P))\s*(?P.*))|)\s*\ZrenotonlyonlycSrtr%rrr%r%r&rsrtz7\A\s*(?P\b\w+\b)\s*=\s*>\s*(?P\b\w+\b)\s*\Zlocalz0analyzeline: Not local=>use pattern found in %s mapz0analyzeline: Could not crack the use statement. )rrusercodez-analyzeline: No context for multiline block. z+analyzeline: No code implemented for line. )Trr$rrrrrbasenamerrr.rr.r/rrrrrrydictrrwrremoverrrrrrlrerfrcopydeepcopyrWrrzcrackline_bindlangrnr@cracktypespec0 updatevarsrfindrHrrT namepatternrwrrpriget_parametersdetermineexprtypero real16patternfinditerr{reval SyntaxError NameError TypeErrortypespattern4implicit cracktypespecrordchrr enumerate setdefault getdimensionfindallr  IndexErroranyextendr4r isinstancerappendmultiline)Grr;r,r%newnamerkrxrr)r bindcline needmodule needinterfaceitrmbindcdattr0selectorattredecl last_nameriplchrrRapinitexprruttvrimplrjrS kindselect charselectrarbegcendcodlilrXfcinprr$ridxvtypevdimmatchesnew_val expanded_listr multipliervalue current_valclbnol commonkeymmisonlyrrgcr%r%r&rJs                                                                 $                                   " 2             ",    (" "  ,                          rJcCs:d|vri|d<|d}||vrg||<|||dS)Nf2pymultilinesr)r context_namemlrr%r%r&r2srcCsd}d}td|tjrd}ntd|tjrd}n|}tt|}|s/tddS|}t | D] }t ||||<q9|dvrP|d}|d}| d }|d kri|d|}||d d}||||fS) Nzdouble\s*complexdouble complexzdouble\s*precisionzdouble precisionz>cracktypespec0: no kind/char_selector pattern found for line. )rr3logicalr1 characterrdr*rrrr&) rrrrr selectpatternrHr.rrerfr\r)r0rrrrRrrmrr%r%r&r<s0   rz)\s*(?P\b\w+\b)\s*(?P.*)\s*\ZzB\s*(\(\s*(kind\s*=)?\s*(?P.*)\s*\)|\*\s*(?P.*?))\s*\Zz2\s*(\((?P.*)\)|\*\s*(?P.*))\s*\Zz\s*(kind\s*=\s*(?P.*?)\s*(@,@\s*len\s*=\s*(?P.*)|)|(len\s*=\s*|)(?P.*?)\s*(@,@\s*(kind\s*=\s*|)(?P.*)|(f2py_len\s*=\s*(?P.*))|))\s*\Zz\s*(@\(@\s*(?!/)\s*(?P.*?)\s*@\)@\s*\*\s*(?P.*?)|(\*\s*(?P.*?)|)\s*(@\(@\s*(?!/)\s*(?P.*?)\s*@\)@|))\s*(=\s*(?P.*?)|(@\(@|)/\s*(?P.*?)\s*/(@\)@|)|)\s*\ZcCs|}t|dkr |S|d}tdt|dD]}||dkr2||ddvs1||ddvr2q|||}q||d}|S)Nrrrz ()[]{}=+-/* r )rrTr)exprexpr2rr%r%r& removespacescs   rcCs~d}d}d}d}|D]2}|dkr|dvr||7}|}q |s#|dvr#|}||kr+| }n |dkr6|r6|d7}q ||7}|}q |S) a( The function replace all spaces in the input variable line which are surrounded with quotation marks, with the triplet "@_@". For instance, for the input "a 'b c'" the function returns "a 'b@_@c'" Parameters ---------- line : str Returns ------- str r FN\)r r")rr r@_@r%)r,fragmentinside current_quoteescapedrr%r%r&markinnerspacesrs&  rc Cs8d}t||\}}}|rFddt|dD}g}td} |D]"} | s&q!| | } | r>| d} | | t| d} | | q!|}ddt|dD} g}| D]"}ddtt t |d d d DD] }|rw| | d d qjqV|D]}t |} | std t|q{t| d}i}|ttdvrttd|}d|v}|r||d<n|r||dkstd||d|fd|vrt||d<n:|rt|D]0}||dvr|||d|kstd|||d|||fqt|||d|<qd|vr'|r'|r||d<nEtd||fn<|rct|D]2}||dvrV|||d|ksVtd|||d|||fq0t|||d|<q0d|vrm||d<n|r|d|kstd||d|fd|vrt||d<n8|r|D]} | |dvr|d | qn t||d<t||d<t||d<||d<t||d<d|dpgvr|ttdvrdttvrgttd<ttd || drmtt| d}|rb|}dD]}||ddur"||d||<||d=q t|D]}||dur=t||||<q*||=q*d |vrd!|vr|d d"kr]|d!|d <|d!=nG|d#krd|vsl|dspi|d<d |dvr||dd =|d |dd$<|d =n|d!d%|d |d!<|d =td&|||||d!fd |vr|d'vrd|vs|dsi|d<|d |dd$<|d =n*|d#krd|vs|dsi|d<d |dvr|dd =|d |dd$<|d =d(|vrd)|vr|d)|d(kstd*||d)|d(fn|d(|d)<d!|vrad+|d!}d|vs.|ds4|g|d<n9|d ||dD]!}|dd,d-kr_||kr_|dd.=td/|||fnq?n td0|| dt|D] }||s}||=qs|ttd|<d1ttvrttd1 ||}q{|S)2z Returns last_name, the variable name without special chars, parenthesis or dimension specifiers. Alters groupcache to add the name, typespec, attrspec (and possibly value) of current variable. NcSrtr%rrr%r%r&rsrtzupdatevars..rz(?P[a-zA-Z]+)r{cSrtr%rrr%r%r&rsrtcSrtr%rrr%r%r&rsrtrr[z@ @r z;updatevars: no name pattern found for entity=%s. Skipping. rkr$r0zIupdatevars: attempt to change the type of "%s" ("%s") to "%s". Ignoring. r^zVupdatevars: attempt to change the kindselector "%s" of "%s" ("%s") to "%s". Ignoring. r_zEupdatevars:%s: attempt to change empty charselector to %r. Ignoring. zVupdatevars: attempt to change the charselector "%s" of "%s" ("%s") to "%s". Ignoring. razMupdatevars: attempt to change the typename of "%s" ("%s") to "%s". Ignoring. r]rr)r'r)rTarrayinit2rTrr rrrz-updatevars: "%s %s" is mapped to "%s %s(%s)" rr3rr1rr`zTupdatevars: attempt to change the init expression of "%s" ("%s") to "%s". Ignoring. dimension(%s) rr z5updatevars:%s: attempt to change %r to %r. Ignoring. z?updatevars: could not crack entity declaration "%s". Ignoring. r)rrrrrrrrrTrrrrrr.rrlr/rrrerfrirolenarraypatternrHrr\)r0rr] entitydeclrrrrarrr rselel1re1enamernot_has_typespecrmrRd1lkdmdm1r%r%r&rs<    (     $  $     (                    rc Csd}d}d}|r|dvrLt|}|stdt|dS|}|d|d<|d=t|D] }||s9||=q0t|D] \}}t|||<q@n|dkrt |}|satdt|dS|}|d|d<|d=|drt t |d}|}d D]}||d r||d ||<||||<||d =q|d dur|d |d <|d=t|D] }||s||=qt|D] \}}t|||<qn'|d krt d |t j }|r|d}ntdt||ntdt||||fS)Nrz4cracktypespec: no kindselector pattern found for %s kind2rrz4cracktypespec: no charselector pattern found for %s charlenlenkind)rTkindrf2py_lenrdz\s*\(\s*(?P\w+)\s*\)rkz'cracktypespec: no typename found in %s z'cracktypespec: no selector used for %s )r^rr.rrrerfitemsrlr_lenkindpatternrrrr) r0rrrrarmrr&r!r%r%r&r@s|               rcCs|si}|s|Sd|vr|g|d<|S|r|d|||dvr$|S|dkr7d|dvr7|d||S|dkrJd|dvrJ|d||S|dkr]d|dvr[|d||S|dkrpd|dvrn|d||S|d||S)Nr]rF automaticrrr)rjrrlr%r%r&rg|s6      rgcCZ|si}|s|Sd|vr||d<|St|D]}|s"||dvr*|||d|<q|S)Nr^rerfrjselrlrmr%r%r&rhsrhcCr,)Nr_r-r.r%r%r&risriunknowncCsd|vr|dS|S)Nrkr%)r%r0r%r%r& getblocknamesr1cCs0z d|d|dfaWdStyYdSw)Nz In: %s:%s rrk)rr)r%r%r%r& setmesstexts  r2cCs2i}d|vr t|d}d|vr||d|S)N parent_blockr ) get_usedictupdate)r%usedictr%r%r&r4s  r4c Cs|duri}t|}|s|St|D]J\}}|}|tvr,td||dfqt|}t|}|s7q|r?td|t|D]\}}||vrYtdt |t |f|||<qEq|S)Nz0get_useparameters: no module %s info used by %s rkz,get_useparameters: mapping for %s not impl. zEget_useparameters: overriding parameter %s with value from module %s ) r4rer)rrr.rorrir) r% param_mapr6usenamemappingmvarsrurmrr%r%r&get_useparameterss4   r;c sts|St|trfdd|D}|St|td|dfddur,t|dur_d|vr_|d}t|D] }||}d|vr^|d}d|vr^|d}|vr^||d<q>fd d|d D} | |d <|S) Ncg|] }t|ddqSr)tabr7 postcrack2)rqgr7r>r%r&rszpostcrack2.. %sBlock: %s rkrr$r^r'cr<r=r?rqbrBr%r&rsrCr#)rrrer2r.r;rf) r%r>r7retr$rvarr'valnew_bodyr%rBr&r@s6    r@cCs8t|tr3g}g}|D]#}t|t||dd}d|vr)d|dvr)||q ||q ||St|t|tsHd|vrHtdt|d|vr]|ddks]td||dfd t |}t |}t ||d <t |d |d <| d r||d }t|||d|d <g}d|vr|d}t|D] }d|vr||qni}d} d|vr|d} | drg} d|vr|d} t|d } | r| d} nd} | |vrd} d| | f|vr| d} d| | f|vsd| | f} dgi| dd}|dD]u}|| vr\g}d}|d D]I}|d}|ddkrRd} |d D]}| d} d|vr;|d|kr;t|}|d | =nq|rR|d sJ|d |=| | |=nq |d |q|| vrpt| |sp| ||d |<q|d s{|d r| |d<d|gi| |dd}i|| <t||r||d<|S)zi TODO: function return values determine expression types if in argument list rr>rkrr%z0postcrack: Expected block dictionary instead of rrDrr$sortvarsr)r#r r r'r(__user__routinesunknown__user__routinesrz%s_%ir_user_interface)r%r#r$rkr r)r%r#r$rkr()rrer2 postcrackrrrrr. analyzeargs analyzecommon analyzevars sortvarnamesro analyzebodyrfrrW isexternalr)r%r)r>greturetrA userisdefineduseblockrmrkr(r:mnamerrredefjrFbbmblockr%r%r&rP s                    rPcCsg}g}t|D]}d||vr||dr||q ||q t|}d}|r|d}d}|ddD]}|||dvrFd}nq8|rm|dd|g}|d}||krltdd|d||} |Sn|||dd}t|}d}|s,|S)NrdrrzTsortvarnames: failed to compute dependencies because of cyclic dependencies between rr)rerfrrTrir)r$indepdeprrrrOwr%r%r&rTlsF    rTcCs~t|s|Sg}t|dD]}g}|d|D]}td|tj}|rg}|dr.*?)\b\s*(\((?P.*?)\)|)\s*\ZdimscSrtr%rrr%r%r&rsrz!analyzecommon..rrkr$r]rrzNanalyzecommon: failed to extract "[()]" from "%s" in common /%s/. commonvars) hascommonrerfrrrrrrrlrrrri)r%rdrmcomvarsrrrcrr%r%r&rRsR        rRcCs<t|dd|dD}g}|dD]}||d<|ddvrY|dur,|d|vr,q|d }|d|vr?t|d|dtvrFqtrO|dtvrOqt|d d d |d <n|}t|||dd}|ddvrx|dsx|dsxd|vrxq|d dddkrt |q|ddkr|dt |d<||q|S)NcSs*i|]\}}d|vsd|dvr||qS)r]rr%)rqkeyrr%r%r& s zanalyzebody..r$r#r3r%rrkr)z T as_interfacesaved_interfacerrKrrrrr pythonmoduler) r2r)rfr!rr"crack2fortrangenrProrrr)r%r)r> maybe_privater#rFas_r%r%r&rUsF         rUcCst|t}i}d|vrO|ddur&d}tdkr"tdt|d||fSt|dD] }|d|ddvrD|d|||<q.|d|d||<q.||fS)Nrrz6buildimplicitrules: no implicit rules for routine %s. rkr0)rFr+)r2defaultimplicitrulesrr.rrerfro)r% implicitrules attrrulesrmr%r%r&buildimplicitruless  rscCs,t|||}t|ttfvr|Std|)z2 Like `eval` but returns only integers and floats zr=%r)rrdr4r6r)rrArrr%r%r&myevals  rtz \A\b\w+\b\Zc Cs ztt|ii}d|dfWStyYnwt|r"dd|fSt|}|D]}t||kr1q(td|d|r>> getlincoef('2*x + 1', {'x'}) (2, 1, 'x') >>> getlincoef('3*x + x*2 + 2 + 1', {'x'}) (5, 3, 'x') >>> getlincoef('0', {'x'}) (0, 0, None) >>> getlincoef('0*x', {'x'}) (0, 0, 'x') >>> getlincoef('x*x', {'x'}) (None, None, None) This can be tricked by sufficiently complex expressions >>> getlincoef('(x - 0.5)*(x - 1.5)*(x - 1)*x + 2*x + 3', {'x'}) (2.0, 3.0, 'x') rNrz\w\s*\([^)]*\b\b(?P.*?)\b\b(?P.*)z%s(%s)%srrg?g?)NNN) r4rtrgetlincoef_re_1rrTrrzrrr) rxsetrlen_errQrrReerFr c2r%r%r& getlincoef s                   r}z\b[a-z][\w$]*\bcCs||vrW||dg}d||vr5t||s5t||dD]}||vr4||vr4||kr4||q!|ddD]}||gpHt|||D] }||vrT||qIq;n tdt|g}|||<|S)Nrdr`z,_get_depend_dict: no dependence info for %s )roisstring word_patternrr_get_depend_dictr.r)rkr$depswordswordrbr%r%r&rN s(    rcCs*t|}i}|D]}t|||q |Sr()rerfr)r$rv depend_dictrr%r%r&_calc_depend_dictd s rcstg}tD]}|s|||=q rBtD]\}}fdd|D}|s;|||=q#||<q#sfdd|DS)Ncg|]}|vr|qSr%r%)rqr)rr%r&rsu z$get_sorted_names..crr%r%)rqrkr$r%r&rs{ r)rrerfrr))r$rvrklstnew_lstr%)rr$r&get_sorted_namesl s    rcCs@|ddvr |dd}t|rdSt|rdSd|dS) Nr'"rr zkind(rV)rr real8pattern)r_r%r%r& _kind_func~ s     rcCsLd|}|dkr dS|dkrdS|dkrdS|dkrd S|dd kr$d Sd S) Nrrir&lrlrr r%)rrr%r%r&_selected_int_kind_func s rcCsZ|dkrdS|dkr dSt}|dr|dkrdSd S|dkr%dS|dkr+dSd S) Nrrr) aarch64alphaarm64 loongarchmipspowerppcriscvs390xsparc!rr )platformmachinerr)prradixrr%r%r&_selected_real_kind_func s  rc Cs:t|}t|}dtfdtfdtffD] \}}||vr!|||<qg}t|D]}d||vr=d||dvr=||q(tdtj}tdtj} tdtj} |D]D}d ||vr||d } t ||ry| } d D]} | j | } qq| d | } | d | } d } d||vrd||dvrt | }| d||ddd} t | |k} | s| | s| d}t |dkrd|dd |d d} t||rt| }t| D]}t| || dd|||<qd|} nt||rtd| ddd||dDp"dgd}t| r.d} nt| r6d} z t| |||d||<Wn"tyd}z| ||<td|d|dWYd}~nd}~wwt||r|t||tr|t ||||<| }||kr||||<qUt!||td|d qU|S)!Nr'selected_int_kindselected_real_kindr]rz \bkind\s*\(\s*(?P.*)\s*\)z-\bselected_int_kind\s*\(\s*(?P.*)\s*\)z4\bselected_(int|real)_kind\s*\(\s*(?P.*)\s*\)r`))z.false.False)z.true.Truez kind("\1")zselected_int_kind(\1)Fr^rr rr rrzAget_parameters[TODO]: implement evaluation of complex expression rcSs$g|]}|dr|dqS)r)r removeprefixrrqrr%r%r&rs sz"get_parameters..rrr)dimspeczget_parameters: got "" on zget_parameters:parameter z does not have value?! )"rrrrrrrrr islogicalrrsubrTrrrisdoublererrr{r iscomplexr.rzr param_evalrr~rr4rrw)r$ global_paramsrug_paramsrkfunc param_namesrkind_reselected_int_kind_reselected_kind_rerrepl is_replaced orig_v_lenv_rrrrnlr%r%r&r s                &      "   rcCs|dvrdSt||S)N)z(:)(*)rr) _eval_scalar)lengthrur%r%r& _eval_length s rz\d+_c Cst|r |dd}zt|i|}t|trtnt|}W|Stttfy-|YSt yL}zt d||t | fWYd}~|Sd}~ww)Nrrz,"%s" in evaluating %r (available names: %s) ) _is_kind_numberrrrrrrrrrrirerf)rrurr%r%r&r s rc> st|t|\}}t|d|ddkr#|dvr#i|d<d|dvrbd=d|ddvrb|ddd}ttdd|d DBD]}d D]}||vr`t|i||<qOqKg}|d }|D]}z |||WqjtyYqjwt D] }||vr||qt t |} i} t d j} t D].} | | } | r| | | }z| |Wqtyt d | t jj| |<Yqwq|D]}|dt| vrt|||d|<d|vrnd|vr d|dvsn|r[|d}t|| D]>}|dkr-|||dkr-q||vr@|||||<q|dkrX|||D] }t|||<qKqn||d vrntdt||dfd|vrd|dvr|dd}z tt|i| }Wn tyYnw||dd<d|vrd|dvr؈|dd}z tt|i| }Wn tyYnw||dd<i}d|vrQ|d}|g|d<d\}}}}}|D]K}|dddkr|dddd}na|dddkr)|dddd}nK|dddkr?|dddd}n5|dd d!krU|d ddd}n|dd"d#krk|d"ddd}n t|||<|rd|vrg|d<d$d%t|d&DD]}|d'd}||dvr|d|qd}|r|d(d)}|d*d+}d#|vr|g|d#<n |d#|d}|durd|vrg|d<td,d%t|d&DD]}||dvr |d|qd}|durEd!|vr!g|d!<d-d%t|d&DD]}||d!vrA|d!|q-d}q|rQd|vrQg|d<td.d%t|d&DD]}zt|| }Wntt tfytd/|d+Ynw|d0krd0nd1}||kr|g}nt|d0d2}t!|d3krd1|vrd1g}d1}t!|dkr|d|krd4|dg}t!|d3krFt"t#j$j%|\}}||d}|j&t#j'j(d5}i}|dD]Z} t#)| } |*| r@z|+| \}}!||!fd6d7}"t|,}#|#-|!,Wnt.y,}$zd}"t|,}#WYd}$~$nd}$~$wwtfd8d|#D}%|"t|%f|| <q|||<|d|qed!|vr(d |vr(||d vr(|dg}&g}'t/t0t1t2|}(t3|rt4|dD]:\})}||durn|(rl5D]\} \}*}+fd9d:t},| |,| |&vsd;| vsd| vrȐq|*dur| |,vrd<}-|*t#)d=|d>|)d?}.|.j&t#j'j(d5}.|.| d;<|g|+| d<d!| vrd=|d>|)d@|g| d!<n4dA}-d| vr g| d<dB| dvr2| ddB|&| |'d=|d>|)d@|| dg}/dC|/vsadD|/vsa|/|-r^dDndC|/rj|/| d<qdur5D]K\} \}*}+| dg}%| dgD]}0|06drd7|0}0|%8|0dEddFq|%rtt|%| d<||%vr|&| quqnNt9|rd|vrd1|dvrt:|dd1| }1|1|dd1<n#d|dvrt:|dd| }1|dd=|1|dd1<|'r|'|d!<|&r(tt|&|d<d;|vrd|vrd|2} | rt?| @dK| @dL\}6}7}}8tA|6|7\}9}:};|6|d<z|dGr5|6|dGd<Wn ty@Ynw|9rfd|9vr`z t|9di| |9d<Wn ty_Ynw|9|d<|:ro|:|d<|;rx|;|dM<|4rt|dI|<|5rt|dJ|<qtdNt|dHq|ddOvrbdP|vrt|d |dP}.r#rr)z[A-Za-z][\w$]*z.*\b%s\brr0r undefinedzCanalyzevars: typespec of variable %s is not defined in routine %s. r_rTr^r')NNNNNNrrrr rrrdrrcrrbcSrtr%rrr%r%r&rs rtzanalyzevars..rrz\n\nz z\n rcSrtr%rrr%r%r&rs rtcSrtr%rrr%r%r&rs rtcSrtr%rrr%r%r&rs rtz4analyzevars: could not parse dimension for variable :rz@:@r&1)languagecSs |||Sr(r%)rr rFr%r%r&solve_v s zanalyzevars..solve_vc3s |] }|jvr|jVqdSr()rrrr%r&r s cs:|dggdD]}||vr||||q dS)Nr)roadd)rrv1)coeffs_and_deps compute_depsr%r&r s   z!analyzevars..compute_depsr`Fzshape(rrVz) == Tinrrrrrrpure recursiver*rraz'analyzevars: prefix (%s) were not used )rrlrrrdr)rrrd)Dr2rsrrrgrorKeyErrorrerfrr;rrrr{rrrr.rrrrreverserrrrrw param_parserrrTrrExprparsetostringLanguager as_symbolr  linear_solvesymbolsr5 RuntimeErrorl_or isintent_inisintent_inoutisintent_inplaceisarrayrr)rrrr~risscalarrrnr@rrrisintent_callback isintent_aux)>r%rqrrgenrrmsvarsr)r ru dep_matches name_matchrrln0rdimension_exprsrdimrrdrcrbrtmprdim_charrr d2dsizesolver_and_depsrrFr all_symbolsrv_depsn_depsn_checks n_is_inputrsolverrall_deps is_requiredrv_attraarprpr1ispureisrecr0rrrrra neededvarsrkr%)rrr$r&rS* s   "                               &                L                         rSz\A[a-z]+[\w$]*\Zc Cs|dur.z t|||}W|Sty-}z|}td|d|dWYd}~|Sd}~wwt|dksA|ddt|ddkrItd|d |dd d }t|dkr|d d }t|dkrvt|d |}tdt|d}n"t|d |}t|d|} tt|t| d}ntd|d| dr|ddn|d }g} |D]-} zt| ||} Wnty}ztd|d| dWYd}~nd}~ww| | qt t || }|S)a Creates a dictionary of indices and values for each parameter in a parameter array to be evaluated later. WARNING: It is not possible to initialize multidimensional array parameters e.g. dimension(-3:1, 4, 3:5) at this point. This is because in Fortran initialization through array constructor requires the RESHAPE intrinsic function. Since the right-hand side of the parameter declaration is not executed in f2py, but rather at the compiled c/fortran extension, later, it is not possible to execute a reshape of a parameter array. One issue remains: if the user wants to access the array parameter from python, we should either 1) allow them to access the parameter array using python standard indexing (which is often incompatible with the original fortran indexing) 2) allow the parameter array to be accessed in python as a dictionary with fortran indices as keys We are choosing 2 for now. Nzparam_eval: got "rrr&rrYzparam_eval: dimension z can't be parsedr rrrz.param_eval: multidimensional array parameters z not supportedz(/r) rrr.rTrrrrr4rrrzip) rrrurrrdimrangeboundlbounduboundv_evalitemr%r%r&r sH &   " rcCsd|vr*|d|d}||dd|d}tt||}t|||S||vr4t||S|D]+}td|dtj}||}|ra| dt||| d}||}|sJq6|S) aRecursively parse array dimensions. Parses the declaration of an array variable or parameter `dimension` keyword, and is called recursively if the dimension for this array is a previously defined parameter (found in `params`). Parameters ---------- d : str Fortran expression describing the dimension of an array. params : dict Previously parsed parameters declared in the Fortran source file. Returns ------- out : str Parsed dimension expression. Examples -------- * If the line being analyzed is `integer, parameter, dimension(2) :: pa = (/ 3, 5 /)` then `d = 2` and we return immediately, with >>> d = '2' >>> param_parse(d, params) 2 * If the line being analyzed is `integer, parameter, dimension(pa) :: pb = (/1, 2, 3/)` then `d = 'pa'`; since `pa` is a previously parsed parameter, and `pa = 3`, we call `param_parse` recursively, to obtain >>> d = 'pa' >>> params = {'pa': 3} >>> param_parse(d, params) 3 * If the line being analyzed is `integer, parameter, dimension(pa(1)) :: pb = (/1, 2, 3/)` then `d = 'pa(1)'`; since `pa` is a previously parsed parameter, and `pa(1) = 3`, we call `param_parse` recursively, to obtain >>> d = 'pa(1)' >>> params = dict(pa={1: 3, 2: 5}) >>> param_parse(d, params) 3 rTNrrVrvrwrr) rrfindr4rrrrrrr)rrudnameddimsrWrrQrr%r%r&r s,9    rc Cs|}t| }|r^t|\}}t||d|}d}|D]} | } | tjtjvr,d} || }q|ddkr<|d}n|d}|}||dvsN||dvr^|d}||dvsN||dvsN||vr~d } |t| |vrx| d } |t| |vsl|t| }|r||d|<|S||dvr||dvr|d||d|<ni|d|<d |vr||d |d vrt |d|d |d|<|S) Nr$e_rr r_er)rrr'r(r) analyzeargs_re_1rrsrrr_ascii_lowercasedigitsrrg) r r%r)orig_a a_is_exprrqrratnarrmr%r%r& expr2nameB sD         rcCst|t|\}}d|vrg|d<g}|dD] }t|||}||q||d<d|vrLt|dD]\}}|D]}||dvrJi|d|<q\w+)|)\Zz*\A[+-]?[\d.]+[-\d+de.]*(_(?P\w+)|)\Zz \A\(.*\)\Zz\A(?P\w+)\s*\(.*?\)\s*\ZcCsPt|tr ddiSt|trddiSt|trddiSt|tr"|Stt|)Nr0r3r1r)rr4r6rrAssertionErrorr)rr%r%r&_ensure_exprdict s     rcCs||vr t||S|}t|rddiSt|}|r5d|vr1|dr1tdt|ddiSt |}|rSd|vrO|drOtdt|ddiSdD]%}dd t ||d  d |d DD]}||vryt||SqiqUi}t |rt |d d ||}n.t|}|r|d}t |d||}|rd|vr|d=|s|d|vrt||dS|ddvrdddidS|stdt||S)Nr0rrkz:determineexprtype: selected kind types not supported (%s) r3r1)+rrrcSrtr%rrr%r%r&rs rtz%determineexprtype..rrZrr r]rrrr)r0r_z>determineexprtype: could not determine expressions (%s) type. )rrdetermineexprtype_re_1rdetermineexprtype_re_2rrr.rdetermineexprtype_re_3rrdetermineexprtype_re_4rdetermineexprtype_re_5)rr$rulesroprrrnr%r%r&r sX      (       rrFcCst|d}t|tr3|D]#}|r'|ddvr'|dtvrq tr'|dtvr'q |t|||d}q |Sd}d}d}|d}|dkrCdSg} d|vrM|d}d|vr||d} |dD]} t| || } t| | sl| | qY|dd ksu| r|d d | }d} d |vrt|d  D]} d | |t | |d | f} q| dgdd}|d krd|vr| d|rd| |t d ||f} d}d|vrt|d|t }d}d|vrt|d|t }|dkrd}d}d|vrd|d}|d| vr| |dt|d|t |d}t||d| |t |d} d}d|vr&|s&d|d}d|vrLd}t|dD]\} }d||t | d |f}q5||}|dkrX|dkrXd}d|||||||| || |||||f}|S)Nr r%rrkrirr)r$rz(%s)rrz %s%s%s %srcallbackz%s%sintent(%s) %sr r rrz result (%s)r#rz! in %srz%s%sentry %s(%s)rrz%%s%s%s %s%s%s %s%s%s%s%s%s%send %s %s)r2rrer!r"rmrrrrrfrror use2fortrancommon2fortran vars2fortranr))r%r>rjrGrArrkr) blocktypeargslr$r rrm intent_lstr r rr#mess entry_stmtsrr%r%r&rm s            rmc CsVd}t|D] }|dkrd||d||f}qd|||d||f}q|S)Nr rz %s%scommon %srz%s%scommon /%s/ %s)rerfr)r r>rGrmr%r%r&r! s r!cCs d}t|D]{}d|||f}||ikr&|r%|ddkr%|dd}qd||vr6||dr6d|}d||vru||drud}t||dD]&}|||d|krdd |||f}d}qNd |||||d|f}d}qN|r|ddkr|dd}q|S) Nr z %s%suse %s,r rrz%s only:rr%s%s%sz %s%s%s=>%sr-)r r>rGrrrmr%r%r&r  s*   r c CsR|d}g}|D]}z td|}Wn tyYqw||r&||q|S)Nrz isintent_%s)globalsrr)rHrrGrrXr%r%r&true_intent_list/ s  r*c Cst|d}g}|D] }||dvr||q d|vr7|dD]}||vr0||vr/||q td|q d|vrB||d|sVt|D] }||vrU||qJ|D],}d||vr||dD]}||vrd||vr|||dvrtd||fqgd|vr||dvrt||rd |||f}d |||f}t||rd |||f}||vrd ||vrqXd } |dD]} || dkr| ddkrd} nq| rqX||vrt|t d|qX||dkr|ddks| drqXd ||vr(d||vrd||dvr||vrd |||f}qXt||t d|qX||d } | dkrDd||vrDd| ||df} i} d||vrT||d} n d||vra||d} d| vr| ddvrvd| | df} n6d | | df} n-d!| vrd"| | d!f} d#| vrd$| | d#f} nd%| } n d#| vrd&| | d#f} d'} d||vrd(d)||dD}|rd*|vrd+|vr| d+|rd,| d- |f} d-} d.||vrd/| | d- ||d.f} d-} d0||vrt ||}|rd1| | d- |f} d-} d2||vr*d3| | d- ||d2f} d-} d||vrAd4| | d- ||df} d-} d5||vrx||d5}||d d6vrpzt|}d7|j|jf}Wn tyoYnwd8| ||f} nd9| |f} d:||| f}qX|S);Nr r$rdz7vars2fortran: Confused?!: "%s" is not defined in vars. rrdzHvars2fortran: Warning: cross-dependence between variables "%s" and "%s" r'z%s%sintent(callback) %sz%s%sexternal %sz%s%soptional %sr0rr#rkr%rrz/vars2fortran: No definition for argument "%s". rr]rz-vars2fortran: No typespec for argument "%s". rdraz%s(%s)r^r_r)rrz%s*(%s)z%s*%srTz %s(len=%sr'z %s,kind=%s)z%s)z %s(kind=%s)rcSsg|]}|dvr|qS))rr%)rqrr%r%r&rs sz vars2fortran..z intent(in)z intent(out)z%s, %srrz%s%sdimension(%s)rz%s%sintent(%s)rcz %s%scheck(%s)z%s%sdepend(%s)r`)rrz(%s,%s)z %s :: %s=%sz%s :: %sr()r2rrirrerfr isoptionalshowr.rorrr*rr1imagr)r%r$r)r>rjrGnoutr rrrFvardefrrrrrr%r%r&r"= s       $                     r"cCsztddt|ttddgattd}tddtD]}td|jddt||}qtddt |}t|S)NzReading fortran codes... rzPost-processing... z"Applying post-processing hooks... z rzPost-processing (stage 2)... ) r.rr-rrPrpost_processing_hooks__name__traverser@)filespostlisthookr%r%r& crackfortran s       r6cCs$t|d}d}dt}|||S)NrzE! -*- f90 -*- ! Note: the context of this file is case sensitive. z ! This file was auto-generated with f2py (version:%s). ! See: ! https://web.archive.org/web/20140822061353/http://cens.ioc.ee/projects/f2py2e )rm f2py_version)r%pyfheaderfooterr%r%r& crack2fortran s  r;cCs(t|tot|dkot|dttfS)Nr&r)rtuplerTr4r)objr%r%r&_is_visit_pair s  r>cOs<t|r,|ddkr |S||||g|Ri|}|dur%t|s#J|S|}|\}}nd|f}d}t|trbg}t|D]#\} } t| | f|g|R||g|d|\} } | dur`|| q=n2t|trt}|D]"\} } t| | f|g|R||g|d|\}}|dur|||<qnn|}|dur|S||fS)aTraverse f2py data structure with the following visit function: def visit(item, parents, result, *args, **kwargs): """ parents is a list of key-"f2py data structure" pairs from which items are taken from. result is a f2py data structure that is filled with the return value of the visit function. item is 2-tuple (index, value) if parents[-1][1] is a list item is 2-tuple (key, value) if parents[-1][1] is a dict The return value of visit must be None, or of the same kind as item, that is, if parents[-1] is a list, the return value must be 2-tuple (new_index, new_value), or if parents[-1] is a dict, the return value must be 2-tuple (new_key, new_value). If new_index or new_value is None, the return value of visit is ignored, that is, it will not be added to the result. If the return value is None, the content of obj will be traversed, otherwise not. """ rr3N)parentsr)r>rrerr2rrr))r=visitr?rr)kwargs new_resultparent result_keyrWr new_indexnew_itemrgnew_key new_valuer%r%r&r2 sZ          r2c Os:|d\}}|\}}dd} |dvr#|dddksJ|dd} n|d kr8|d ddks1J|d d} nd } d } | d urU|} | D] \} } t| rS| | | } qFn(|d kr}|d dd} |} | D]\} } t| r|td | dd| | } qg| d ur| |krtd|d|d|d| d d|| fSd S)aPreviously, Fortran character was incorrectly treated as character*1. This hook fixes the usage of the corresponding variables in `check`, `dimension`, `=`, and `callstatement` expressions. The usage of `char*` in `callprotoargument` expression can be left unchanged because C `character` is C typedef of `char`, although, new implementations should use `character*` in the corresponding expressions. See https://github.com/numpy/numpy/pull/19388 for more information. r cSs0td|d||}td|d||}|S)Nz[*]\s*\bruz\b\s*[\[]\s*0\s*[\]])rr)varnamerr%r%r& fix_usage=s z8character_backward_compatibility_hook..fix_usage)rrcrr$rr`rN callstatementz (? `z` )r) ischaracterrrr.)rr?rr)rA parent_key parent_valuergrrJ vars_dictrHrIvdr%r%r&%character_backward_compatibility_hook+sL  rR__main__rz-quietz-verboser&z-fixz?Use option -f90 before -fix if Fortran 90 code is in fix form. z-skipemptyendsz--ignore-containsz-f77z-f90rz-hz-showz-mrzUnknown option %s z OSError: a Warning: You have specified module name for non Fortran 77 code that should not need one (expect if you are scanning F90 code for non module blocks but then you should use flag -skipemptyends and also be sure that the files do not contain programs without program statement). z Writing fortran code to file %s rb)r)r)r)r0r()r N)Nr )r )NN)rr)rF)r F)__doc__r)r_rrrrrrpathlibrr ImportErrorr rauxfuncsrversionr7r r r rrrrrrrrrrrrrrr/rrr.r#rr"r$rr!r rr'r. _MAXCACHErprrhryrrlrwrzr{rrrzrrrrrrrrrr,rbeforethisafter fortrantypesr@rr=r> groupbegins77r groupbegins90r groupendsr;endifsr:moduleproceduresrFr?r0r1r3r4r7r8r9r2r5r6rBrErIrArCrDr<Sr+r,rr!rGcrackline_bind_1rr-rHrr\rnrrvrrrrrrprrryr~rrJrrrr^r_r*rrrrrrgrhrir1r2r4r;r@rPrTrRrUrsrtrxr}rrrrrrrrrrrrSr rrrrQrrrrrrrrmr!r r*r"r0r6r;r>r2rRrr1r3funcsrXf2f3 showblocklistargvrrirrrOSErrordetailr4r8r+r%r%r%r&sZ       &  c                           W %* <     `" +- H  e  > P% - K  B <                H