a VŸ$c‚~ã@sZddlmZmZmZmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZGdd„de ƒZ!e!ƒZ"Gdd„de ƒZ#Gd d „d e#ƒZ$Gd d „d ej%j&ƒZ'e(ƒa)d d„Z*dd„Z+e ¡Z,da-ej% .¡Z/Gdd„dej j0ƒZ1Gdd„dej j0ƒZ2dS)é)ÚdivisionÚabsolute_importÚwith_statementÚprint_functionÚunicode_literals) ÚPY2Ú basestringÚbchrÚbordÚchrÚopenÚpystrÚrangeÚroundÚstrÚtobytesÚunicode)ÚOptionalNc@seZdZdd„ZdS)Ú StoreDeletedcCstrdSdSdS)NsdeletedÚdeleted)r©Úself©rúrenpy/rollback.pyÚ __reduce__1szStoreDeleted.__reduce__N)Ú__name__Ú __module__Ú __qualname__rrrrrr/src@seZdZdZdZdS)ÚSlottedNoRollbackaº :doc: norollback class Instances of classes inheriting from this class do not participate in rollback. The difference between this and :class:`NoRollback` is that this class does not have an associated dictionary, hence can be used with ``__slots__`` to reduce memory usage. Objects reachable through an instance of a NoRollback class only participate in rollback if they are reachable through other paths. rN)rrrÚ__doc__Ú __slots__rrrrr?s rc@seZdZdZdS)Ú NoRollbacka :doc: norollback class Instances of this class, and classes inheriting from this class, do not participate in rollback. Objects reachable through an instance of a NoRollback class only participate in rollback if they are reachable through other paths. N)rrrrrrrrr!Nsr!cs eZdZdZ‡fdd„Z‡ZS)ÚAlwaysRollbackz¿ This is a revertible object that always participates in rollback. It's used when a revertable object is created by an object that doesn't participate in the rollback system. cs0tt|ƒ |¡}tjj}|dur,|jt|ƒ=|S©N)Úsuperr"Ú__new__ÚrenpyÚgameÚlogÚmutatedÚid)ÚclsÚargsÚkwargsrr(©Ú __class__rrr%as  zAlwaysRollback.__new__)rrrrr%Ú __classcell__rrr.rr"Zsr"cCsJ|r |ƒt|ƒ}||vrdS|||<t|tƒr4dSzbt|ddƒ}|durvt|ƒ ¡D]\}}||vrVt|||ƒqVnt|ƒ ¡D]}t|||ƒq‚Wnty¨Yn0zt |ƒr¾t|t ƒrÄWdSWntyÚYdS0z|  ¡D]}t|||ƒqæWntyYn0z | ¡D]}t|||ƒqWntyDYn0dS)zÖ @param obj: The object that was reached. `reachable` A map from id(obj) to int. The int is 1 if the object was reached normally, and 0 if it was reached, but inherits from NoRollback. NÚnosave) r*Ú isinstanceÚNOROLLBACK_TYPESÚgetattrÚvarsÚitemsÚreachedÚvaluesÚ ExceptionÚlenrÚ__iter__)ÚobjÚ reachableÚwaitÚidobjr1ÚkÚvrrrr7nsB        r7cCsn| ¡D]}t|||ƒqtjjD]F}t|j||ƒt|j||ƒ|jD]}| ¡D]}t|||ƒqTqHq"dS)a Marks everything reachable from the variables in the store or from the context info objects as reachable. `store` A map from variable name to variable value. `reachable` A dictionary that will be filled in with a map from id(obj) to obj. N)r8r7r&r'ÚcontextsÚinfoÚmusicÚ dynamic_stack)Ústorer=r>rAÚcÚdrrrÚ reached_vars¬s    rIcsLeZdZdZdZdZdZ‡fdd„Zdd„Zd d „Z d d „Z d d„Z ‡Z S)ÚRollbacka< Allows the state of the game to be rolled back to the point just before a node began executing. @ivar context: A shallow copy of the context we were in before we started executing the node. (Shallow copy also includes a copy of the associated SceneList.) @ivar objects: A list of tuples, each containing an object and a token of information that, when passed to the rollback method on that object, causes that object to rollback. @ivar store: A list of updates to store that will cause the state of the store to be rolled back to the start of node execution. This is a list of tuples, either (key, value) tuples representing a value that needs to be assigned to a key, or (key,) tuples that mean the key should be deleted. @ivar checkpoint: True if this is a user-visible checkpoint, false otherwise. @ivar purged: True if purge_unreachable has already been called on this Rollback, False otherwise. @ivar random: A list of random numbers that were generated during the execution of this element. éNFcsptt|ƒ ¡tj ¡ ¡|_g|_d|_g|_ d|_ i|_ i|_ d|_ d|_d|_d|_ttf|_td7adS)NFé)r$rJÚ__init__r&r'ÚcontextÚ rollback_copyÚobjectsÚpurgedÚrandomÚforwardÚstoresÚ delta_ebcÚretain_after_loadÚ checkpointÚhard_checkpointÚ not_greedyÚ generationÚserialÚ identifierrr.rrrMís zRollback.__init__cCs†|dkrVdii|_|jD]<}t|ƒdkr@|\}}||jd|<q|\}t|jd|<q|dkrdd|_|dkrt|j|_|dkr‚i|_dS)NérFéFérK)rTrFr:rrVrWrXrU)rÚversionÚir@rArrrÚ after_upgrades   zRollback.after_upgradec Cs&|jr dSd|_|j ¡D]*}| ¡D]\}}|tur&t|||ƒq&qt|jj||ƒ|jjD]}| ¡D]}t|||ƒqjq^t|jj   ¡||ƒg}d}t ƒ} |r d}|j D]V\} } t | ƒ} | | vs°| |vrÒq°|  | ¡t| tƒrèq°d}| | | f¡t| ||ƒq°q |j dd…=|j  |¡dS)aT Adds objects that are reachable from the store of this rollback to the set of reachable objects, and purges information that is stored about totally unreachable objects. Returns True if this is the first time this method has been called, or False if it has already been called once before. FTN)rQrTr8r6rr7rNrCrEÚ scene_listsÚget_all_displayablesÚsetrPr*Úaddr2r3ÚappendÚextend) rr=r>ÚchangesÚ_krArHÚ new_objectsÚobjects_changedÚseenÚoÚrbÚid_orrrÚpurge_unreachable)s<       zRollback.purge_unreachablecCsîtjj}t|jƒD]\}}|dur| |¡q|j ¡D]P\}}| |d¡}|durVq8| ¡D](\}}|t ur~||vr†||=q^|||<q^q8|j  ¡D],\}}| |d¡}|dur²q”|j |8_ q”t   |j¡tjj ¡tjj |j¡dS)zo Reverts the state of the game to what it was at the start of the previous checkpoint. N)r&ÚpythonÚ store_dictsÚreversedrPÚ _rollbackrTr6ÚgetrrUÚever_been_changedÚrngÚpushbackrRr'rBÚpoprgrN)rrsr<ÚrollÚnamerirFÚvaluerrrÚrollbackhs*      zRollback.rollbackcCs tjj ¡tjj |j¡dS)zr This rolls back only the control information, while leaving the data information intact. N)r&r'rBrzrgrNrrrrÚrollback_controlŽs zRollback.rollback_control) rrrrÚ __version__r\rYrMrbrqr~rr0rrr.rrJËs &?&rJcsüeZdZdZdZgd¢ZdZdZ‡fdd„Zdd „Z d d „Z d7d d „Z dd„Z d8dd„Z dd„Zd9dd„Zdd„Zdd„Zdd„Zd:dd„Zdd „Zd;d!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Zdd1d2„Zd3d4„Zd5d6„Z‡Z S)?Ú RollbackLoga­ This class manages the list of Rollback objects. @ivar log: The log of rollback objects. @ivar current: The current rollback object. (Equivalent to log[-1]) @ivar rollback_limit: The number of steps left that we can interactively rollback. Not serialized: @ivar mutated: A dictionary that maps object ids to a tuple of (weakref to object, information needed to rollback that object) rK)Ú old_storer)Úidentifier_cacheNFcshtt|ƒ ¡g|_d|_i|_d|_d|_d|_d|_ g|_ i|_ d|_ t  ¡d|_d|_d|_dS)NrFT)r$rrMr(Úcurrentr)Úrollback_limitÚrollback_is_fixedÚcheckpointing_suspendedÚfixed_rollback_boundaryrSr‚Úrolled_forwardrxÚresetÚretain_after_load_flagÚdid_interactionÚforce_checkpointrr.rrrM°szRollbackLog.__init__cCsi|_d|_dS)NF)r)r‰rrrrÚafter_setstateÐszRollbackLog.after_setstatecCs||dkrdt|jƒi|_|dkr,d|_d|_|dkr:d|_|dkrx|jrxd}|j|j d…D]}|jr^|d7}q^||_dS) Nr]rFr^Fr_rKrrL)rerwr†rˆr‹r…r(rX)rr`ÚnrblrorrrrbÔs zRollbackLog.after_upgradecCs(d|_tj ¡}|jsdSd}|r(d}n0|jr4d}n$|jdurX|jjrLd}n |jjrXd}|r`dSd|_|jdur||  d¡n tj   ¡t |j ƒtjjkr¦|j  d¡q†t |j ƒdkrÎ|j djj|jkrÎd|_|jrì|jsì|jjj|_d|_tƒ|_|j|j_|j  |j¡|j ¡dtj_d|_dS)z~ Called before a node begins executing, to indicate that the state needs to be saved for rollbacking. NTFrr]éþÿÿÿ)rƒr&r'rNr~rŒr„rWrVÚcompleterrÚ begin_storesr:r(ÚconfigÚrollback_lengthrzrˆr†rSrJr‹rgr)ÚclearÚ revertableÚ mutate_flagr‰)rÚforcerNÚignorerrrÚbeginêsD         zRollbackLog.begincCs|jD]}|j ||¡qdS)zk Replaces references to the `old` ast node with a reference to the `new` ast node. N)r(rNÚ replace_node)rÚoldÚnewrarrrr›*s zRollbackLog.replace_nodec Csä|jr|jddd|_tjj ¡D].\}}| |¡}|r$|\|jj|<|jj |<q$t dƒD]‚}|jj dd…=z\|j  ¡D]F\}}|durŒqz|\}} |ƒ} | dur¤qz|   | ¡} |jj  | | f¡qzWqàWq\tyÜYq\0q\dS)ap Called after a node is finished executing, before a save begins, or right before a rollback is attempted. This may be called more than once between calls to begin, and should always be called after an update to the store but before a rollback occurs. `begin` Should be true if called from begin(). F)Úhardr_N)rrWr&rrrsr6Ú get_changesr„rTrUrrPr)Ú _compressrgÚ RuntimeError) rršr|ÚsdÚdeltaÚ_irjrAÚrefÚcleanr<Ú compressedrrrr‘3s,       zRollbackLog.completecCs|i}tjj ¡D]B\}}|jD]2}||vr@||||d|<qt||d|<qqttjjdd…ƒD]}|  |¡qh|S)a Return a map giving the current roots of the store. This is a map from a variable name in the store to the value of that variable. A variable is only in this map if it has ever been changed since the init phase finished. Ú.rLN) r&rrrsr6rwrrtr'rBÚpop_dynamic_roots)rÚrvÚ store_namer¢r|rarrrÚ get_rootsgs  zRollbackLog.get_rootscCsbtjtjjttjtfa i}t |||ƒ|j dd…}|  ¡|D]}|  ||¡s@qVq@| ¡dS)a This is called to purge objects that are unreachable from the roots from the object rollback lists inside the Rollback entries. This should be called immediately after complete(), so that there are no changes queued up. N)ÚtypesÚ ModuleTyper&rrÚ StoreModulerÚioÚIOBaseÚtyper3rIr(Úreverserqr•)rÚrootsr>r=Úrevlograrrrrq}s   zRollbackLog.purge_unreachablecCs|jr dSdSdS)NTF)rSrrrrÚ in_rollback™szRollbackLog.in_rollbackcCs|jSr#)r†rrrrÚin_fixed_rollbackŸszRollbackLog.in_fixed_rollbackcCs*|jr&|jd\}}|jjj|kr&|SdS)z; Returns the current forward info, if any. rN)rSr„rN)rr|ÚdatarrrÚ forward_info¢s zRollbackLog.forward_infoTcCs>|jr d}|rd|_|jjr dStj ¡js0dSd|j_|rz|jjsz|j tj j kr`|j d7_ |dkrrd|j_ n||j_|  ¡rÐ|jrÐ|jd\}}|jjj|krº||j_|j d¡n||j_|jdd…=nj|dur:|jr2|jd\}}|jjj|kr&||kr&|s|jr&|j d¡n |jdd…=||j_dS)z˜ Called to indicate that this is a checkpoint, which means that the user may want to rollback to just before this node. FNTrLrYr)r‡r‹r„rWr&r'rNr~rXr…r“Úhard_rollback_limitrYr·rSrzr‰)rr¸Ú keep_rollbackržÚfwd_nameÚfwd_datarrrrW°sF    ÿþþ zRollbackLog.checkpointcCs ||_dS)z{ Called to temporarily suspend checkpointing, so any rollback will jump to prior to this statement N)r‡)rÚflagrrrÚsuspend_checkpointingész!RollbackLog.suspend_checkpointingcCs&d|_dtj ¡_|r"|jdd…=dS)zr Called to indicate that the user should not be able to rollback through this checkpoint. rTN)r…r&r'rNrr()rÚpurgerrrÚblockñs zRollbackLog.blockcCsLtjjjrdSd|_d|j_t|jƒD]}|j r4q)r‘r¬rqr„rQ)rr>r´rrrÚfreezeðs zRollbackLog.freezecCsdS)za Called to indicate that we will not be restoring from the frozen state. NrrrrrÚdiscard_freezeszRollbackLog.discard_freezec CsÄtjj ¡|tj_tj ¡tj  ¡tjj }|  ¡D]d\}}d|vrZ|  dd¡\}}nd}||vrhq8||}|j  |¡|tur”||vrœ||=q8|||<q8tj dd¡}|jdd||ddd S) a} Used to unfreeze the game state after a load of this log object. This call will always throw an exception. If we're lucky, it's the one that indicates load was successful. @param roots: The roots returned from freeze. @param label: The label that is jumped to in the game script after rollback has finished, if it exists. r¨rLrFÚ_greedy_rollbackFrT)r˜rÞrßràN)r&rÂÚscreenÚbefore_restartr'r(rrÚ clean_storesÚ translationÚinit_translationrsr6ÚrsplitrwrfrÚsessionrzr~) rr´rÞrsr|r}r«rFrßrrrÚunfreeze s&     zRollbackLog.unfreezecCs~|jdurdS|j}d}i|_t|jƒD]P}|jdurTtjj |j j ¡rT||j|j<|j rb|d7}|j rp|d8}|s(qzq(dS)NrL) rƒr…rtr(r\r&r'rÑrÒrNr„rXrW)rr…rÝrarrrÚbuild_identifier_cache:s   z"RollbackLog.build_identifier_cachecCs| ¡|j |d¡Sr#)rñrƒrv)rr\rrrÚget_identifier_checkpointsSsz&RollbackLog.get_identifier_checkpoints)F)F)N)NFT)F)FNTFTN)N)N)!rrrrr€r1rƒrrMrŽrbršr›r‘r¬rqr¶r·r¹rWr¿rÁrVrÅrÆrÏr~rærçrðrñròr0rrr.rr˜s<  @ 4  9  =  1r)3Ú __future__rrrrrÚ renpy.compatrrr r r r r rrrrrÚtypingrÚmarshalrRÚweakrefÚreÚsysÚtimer°r­ÚcopyregÚ functoolsr&Úobjectrrrr!r–ÚRevertableObjectr"Útupler3r7rIrZr[Ú DetRandomrxÚObjectrJrrrrrÚs68   > N