a V$c\@sxddlmZmZmZmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZddlZddlZddlmZddlZddlZddlmZmZhdZeddeeDZehd 8Zhd Zd d heBeBZ eZ!eZ"d dZ#ddZ$ddZ%Gddde&Z'dZ(dZ)dZ*Gddde&Z+Gddde&Z,Gdddej-Z.Gddde&Z/e/Z0e/Z1dZ2d d!Z3d"d#Z4dS)$)divisionabsolute_importwith_statementprint_functionunicode_literals) PY2 basestringbchrbordchropenpystrrangeroundstrtobytesunicodeN) py_compile)loadsdumps>FalseTrueNoneccs|]}|ds|VqdS)__N) startswith.0irrenpy/py3analysis.py *r > Zcreditshelpprintnextinputr enumeratelicense copyrightreversedmap memoryview>R SnowBlossomzui.jumps LiveCompositeText CharacterMotionPushMovezui.invokesinnewcontextSolidDynamicCharacter__renpy__dict__ImageReferenceComposeTransitionFixedzrenpy.version_tupleVBoxAlphaFade MoveFactorycolorMoveDragRevolveDissolvezrenpy.version_onlyFlattenRotoZoomMultipleTransition RevolveInOutGridPanMoveTransition ImageDissolvez ui.returns ZoomInOutImageLiveTilez ui.gamemenus__renpy__set__zui.jumpsoutofcontextAt_absolutezrenpy.version_string AlphaMask AlphaBlend Animation__renpy__list__Window Particles Transform Pixellate ADVCharacterConditionSwitchParameterizedTextz renpy.licenseMoveInzui.callsinnewcontextViewportCropMoveMoveOutNullSizeZoom AlphaDissolveHBoxMovieSpeakerColorOldMoveTransition_pPause SubTransitionzrenpy.version_name FontGroup FactorZoom ADVSpeaker ShowingSwitchZoomFrameLiveCropDynamicDisplayablePositionconfigstylecCs|tvrt|dS)at :doc: const Declares a variable in the store to be constant. A variable is constant if nothing can change its value, or any value reached by indexing it or accessing its attributes. Variables must remain constant out of define, init, and translate python blocks. `name` A string giving the name of the variable to declare constant. N) not_constants constantsaddnamerrrconst^srzcCs"t|t|t|dS)z :doc: const Declares a name in the store to be not constant. This undoes the effect of calls to :func:`renpy.const` and :func:`renpy.pure`. `name` The name to declare not constant. N)rvdiscardpure_functionsrurwrxrrr not_constps  r}cCs\|}t|ts&|j}|j}|d|}|dr<|dd}|tvrXt|t||S)a8 :doc: const Declares a function as pure. A pure function must always return the same value when it is called with the same arguments, outside of define, init, and translate python blocks. `fn` The name of the function to declare pure. This may either be a string containing the name of the function, or the function itself. If a string is passed and the function is inside a module, this string should contain the module name with the dot. Returns `fn`, allowing this function to be used as a decorator. .zstore.N) isinstancer__name__ __module__rrur|rwrv)fnrymodulerrrpures      rc@seZdZdZddZdS)Controlz Represents control flow. `const` True if this statement always executes. `loop` True if this corresponds to a loop. `imagemap` True if this control is in a non-constant imagemap. cCs||_||_||_dSN)rzloopimagemapselfrzrrrrr__init__szControl.__init__N)rr __qualname____doc__rrrrrrs rc@s>eZdZdddZddZddZdd Zd d Zd d ZdS)DeltaSetNcCsB||_|dur(t|j|_t|j|_nt|_t|_d|_dS)zc Represents a set that stores its contents as differences from a base set. NF)basesetaddedremovedchanged)rrcopyrrrrs zDeltaSet.__init__cCsH||jvr|j|d|_n&||jvrD||jvrD|j|d|_dSNT)rr{rrrrwrvrrrrws    z DeltaSet.addcCsH||jvr|j|d|_n&||jvrD||jvrD|j|d|_dSr)rr{rrrrwrrrrr{s    zDeltaSet.discardcCs||jvp||jvo||jvSr)rrrrrrr __contains__szDeltaSet.__contains__cCs t|j|Sr)rrrrrrrsz DeltaSet.copyccs2|jD]}||jvr|Vq|jD] }|Vq"dSr)rrrrrrrr__iter__s    zDeltaSet.__iter__)N) rrrrrwr{rrrrrrrrs    rc@s|eZdZdZdddZddZd d d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ddZddZddZdS)!Analysiszb Represents the result of code analysis, and provides tools to perform code analysis. NcCsP||_i|_tt|_tt|_tt|_tt |_ t ddd|_ |j g|_ dS)NTF) parentchildrenrru not_constantlocal_constantslocal_constantalways_constantsglobal_constantr|rcontrol control_stack)rrrrrrs    zAnalysis.__init__cCs*||jvr|j|St|}||j|<|Sr)rr)r identifierrvrrr get_childs    zAnalysis.get_childTFcCs.t|jjo |||jp||_|j|jdSr)rrrzrrappendrrrr push_controlszAnalysis.push_controlcCs|j}|jd|_|S)N)rpopr)rrrrr pop_control"s  zAnalysis.pop_controlcCs|jjr tStSdS)zH Returns NOT_CONST if we're in a non-constant imagemap. N)rr NOT_CONST GLOBAL_CONSTrrrrr'szAnalysis.imagemapcCs0t|j}||D]}d|_|jrq,qdS)zu Call this to indicate the current loop is being exited by the continue or break statements. FN)listrreverserzr)rlrrrr exit_loop1s  zAnalysis.exit_loopcCsf|jD]}|s dSq |jjs>|jjs>|jjs>|jjrbd|j_d|j_d|j_d|j_dSdS)z Returns True if we've reached a fixed point, where the analysis has not changed since the last time we called this function. FT)rvaluesat_fixed_pointrrrrr|rrrrr@s zAnalysis.at_fixed_pointcCs2||jvr.|j||j||j|dS)z= Marks `name` as a potential local constant. N)rrrwrr{r|rryrrr mark_constantXs   zAnalysis.mark_constantcCs4|j||j||j||j|dS)z: Marks `name` as definitely not-constant. N)rrwr|r{rrrrrrmark_not_constantbs   zAnalysis.mark_not_constantcs6fddfddfdd|S)z Returns true if `node` is constant for the purpose of screen language. Node should be a python AST node. Screen language ignores object identity for the purposes of object equality. cst|tjrt}|j}n>t|tjrJ|j\}}|durV|d|j}n |dfS|jvrht|fS|j vrzt |fS|j vrt |fS||fSdS)a5 Check nodes that make up a name. This returns a pair: * The first element is True if the node is constant, and False otherwise. * The second element is None if the node is constant or the name is not known, and the name otherwise. Nr~) rastNamerid Attributevalueattrrrrr LOCAL_CONSTnoderzry) check_name check_noderrrrvs      z(Analysis.is_constant..check_namecs&t|}|stStfdd|DS)z? Checks a list of nodes for constness. c3s|]}|VqdSrrrrrrr r!z.check_nodes..)rrmin)nodesrrr check_nodessz)Analysis.is_constant..check_nodescsD|dur tSt|tjrtSt|tjr2|jSt|tjrH|jSt|tjrjt |j |j St|tj r|j St|tjrt |j|j|jSt|tjrt |j|jSt|tjr|jSt|tjrt |j |jSt|tjr^|j\}}|tks:|jvr>tSt |jdd|jDSt|tjrt |j|jSt|tj r|jSt|tj!tj"fr|dSt|tj#rt |j|j$St|tj%r|jSt|tj&tj'fr|jSt|tj(r@t |j)|j*|j+StS)zz When given `node`, part of a Python expression, returns how const the expression is. Ncss|] }|jVqdSr)rrrrrr r!z;Analysis.is_constant..check_node..r),rrrConstantBoolOpr NamedExprrBinOprleftrightUnaryOpoperandIfExptestbodyorelseDictkeysSeteltsCompare comparatorsCallfuncr|rargskeywordsFormattedValue format_spec JoinedStrrr SubscriptsliceStarredListTupleSliceloweruppersteprrrrrrrrs               z(Analysis.is_constant..check_noderrrrrr is_constantms  rzAnalysis.is_constantcCs$t|\}}|rtS||SdS)z| Compiles `expr` into an AST node, then returns the result of self.is_constant called on that node. N)ccacheast_eval_literalrr)rexprrliteralrrris_constant_exprszAnalysis.is_constant_exprcCs*t|}t|}|D]}||qdS)z> Performs analysis on a block of python code. N)rast_exec PyAnalysisvisit)rcoderarrrrpython#s zAnalysis.pythoncCsTtt|_|jD]\}}||q|jdur:||j|jdurP||jdS)z8 Analyzes the parameters to the screen. N)rrvr parametersrextraposextrakw)rrry_defaultrrrr/s     zAnalysis.parameters)N)TFF)rrrrrrrrrrrrrrrrrrrrrrs     *  rc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&S)'rz This analyzes Python code to determine which variables should be marked const, and which should be marked non-const. cCs ||_dSr)analysis)rrrrrrHszPyAnalysis.__init__cCsXt|jtjr|j|jn6t|jtjrT|jjj rF|j |jn|j|jdSr) rctxrAugStorerrrStorerrzrrrrr visit_NameMs  zPyAnalysis.visit_NamecCs4|j|j}|j|d|||jdSNFrrrr generic_visitrrrrzrrrvisit_NamedExprXs zPyAnalysis.visit_NamedExprcCs|j|jdSrrrryrrrrvisit_FunctionDefcszPyAnalysis.visit_FunctionDefcCs|j|jdSrr rrrrvisit_AsyncFunctionDeffsz!PyAnalysis.visit_AsyncFunctionDefcCs|j|jdSrr rrrrvisit_ClassDefiszPyAnalysis.visit_ClassDefcCs4|j|j}|j|d|||jdSrrrrrr visit_Assignrs zPyAnalysis.visit_AssigncCs&|jdd|||jdSrrrrrrrrrvisit_AugAssign{s zPyAnalysis.visit_AugAssigncCs4|j|j}|j|d|||jdSrrrrrrvisit_AnnAssigns zPyAnalysis.visit_AnnAssigncCsX|j|j}|jj|dd|jjj}|||jjj|krJ|||jdSNT)rzr)rriterrrrzrrrrrz old_constrrr visit_Fors   zPyAnalysis.visit_ForcCs ||Sr)rrrrrvisit_AsyncForszPyAnalysis.visit_AsyncForcCsX|j|j}|jj|dd|jjj}|||jjj|krJ|||jdSr)rrrrrrzrrrrrr visit_Whiles   zPyAnalysis.visit_WhilecCs4|j|j}|j|d|||jdSr)rrrrrrrrrrvisit_Ifs zPyAnalysis.visit_IfcCsL|j|j}||j|j|d|jdur>||j|jdSr)rr context_exprrr optional_varsrrrrrvisit_withitems    zPyAnalysis.visit_withitemcCs$|jd|||jdSrrrrrr visit_Matchs  zPyAnalysis.visit_MatchcCs.|jD]}|jr|j|jq||dSr)handlersryrrr)rrrrrr visit_Trys zPyAnalysis.visit_TrycCs|jdSrrrrrrr visit_BreakszPyAnalysis.visit_BreakcCs|jdSrrrrrrvisit_ContinueszPyAnalysis.visit_ContinueN)rrrrrrrr r r r rrrrrrrrrr r!rrrrrBs&      rc@s0eZdZdZddZddZddZdd Zd S) CompilerCachezR Objects of this class are used to cache the compiliation of Python code. cCsi|_i|_d|_d|_dS)NFr)ast_eval_cacheast_exec_cacheupdatedversionrrrrrszCompilerCache.__init__cCst|tjjr|j}|j}nd}d}|||f}|j|d}|durt|ddd}zt |d}Wnt yzd}Yn0||f}||j|<d|_ |t j|<|S)z5 Compiles an expression into an AST. NevalTast_nodeF) rrenpyrPyExprfilename linenumberr#getr literal_eval Exceptionr% new_ccache)rrr,r-keyrrrrrrs&      zCompilerCache.ast_eval_literalcCs||dS)Nr)r)rrrrrast_evalszCompilerCache.ast_evalcCsjt|tjjr||j|jf}n |ddf}|j|d}|dur\t|ddd}||j|<d|_ |t j|<|S)z/ Compiles a block into an AST. NexecTr() rr*rr+r,r-r$r.rr%r1)rrr2rrrrrs   zCompilerCache.ast_execN)rrrrrrr3rrrrrr"s  "r"zcache/py3analysis.rpybcCstjjjrdSzjtjt"}tt | }Wdn1sD0Y|j t j krvt j |j t j|jWntyYn0dSr)r*gamercompileloaderloadCACHE_FILENAMErzlib decompressreadr&rr#updater$r0)fcrrr load_cache4s 0  r@cCs~tjs dStjrdSzRtttdd}ttj t d}| |Wdn1sZ0YWnt yxYn0dS)NTwb)rr%r*macappr:compressrr1r r7get_pathr9writer0)datar>rrr save_cacheCs, rH)5 __future__rrrrr renpy.compatrrr r r r r rrrrrbuiltinsr* renpy.pythonrrr:renpy.compat.picklerrrrdirr|rvrurrzr}robjectrrrrrr NodeVisitorrr"rr1r9r@rHrrrrsB8  )#5P#J