l j7>dZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z ddl m Z ddlmZmZmZmZddlmZmZmZddlZddlmZmZerdd lmZdd lmZdd l m!Z!dd l"m#Z#dd l$m%Z%gdZ&hdZ'GddZ(Gddej)Z*d"dZ+d#d!Z,dS)$zUtilities for scene caching.) annotationsN)CallableHashableIterableSequence) perf_counter) FunctionTypeMappingProxyType MethodType ModuleType) TYPE_CHECKINGAnyoverload)configlogger) Animation)Camera)Mobject) OpenGLCamera)Scene)KEYS_TO_FILTER_OUTget_hash_from_play_callget_json> background original_id pixel_arraypixel_array_to_cairo_contextceZdZdZeZdZdZeddZ e ddd Z ed dZ ed!dZ ed"dZ e d#d$dZdS)% _MemoizeraEImplements the memoization logic to optimize the hashing procedure and prevent the circular references within iterable processed. Keeps a record of all the processed objects, and handle the logic to return a place holder instead of the original object if the object has already been processed by the hashing logic (i.e, recursively checked, converted to JSON, etc..). This class uses two signatures functions to keep a track of processed objects : hash or id. Whenever possible, hash is used to ensure a broader object content-equality detection. APiclstype[_Memoizer]returnNonec8|jdSN)_already_processedclearr!s N/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/utils/hashing.pyreset_already_processedz!_Memoizer.reset_already_processed8s $$&&&&&F is_methodboolrcdfd }|S)adDecorator to handle the arguments that goes through the decorated function. Returns the value of ALREADY_PROCESSED_PLACEHOLDER if the obj has been processed, or lets the decorated function call go ahead. Parameters ---------- is_method Whether the function passed is a method, by default False. funcCallable[[Any], Any]r#rc"rfdSfdS)Nc<|fdS)Nc|Sr&)objr0selfs r*z^_Memoizer.check_already_processed_decorator..layer....PsdCr,default_function_handle_already_processed)r7r6r!r0s` r*r8zL_Memoizer.check_already_processed_decorator..layer..Ns0)F)F%@%@%@%@%@*G**r,c2|S)Nr9r;)r6r!r0s r*r8zL_Memoizer.check_already_processed_decorator..layer..Rss<.layerJsA YXXXX Xr,)r0r1r#rr5)r!r-r>s`` r*!check_already_processed_decoratorz+_Memoizer.check_already_processed_decorator<s5 Y Y Y Y Y Y Y r,r6rc0||dS)aChecks if obj has been already processed. Returns itself if it has not been, or the value of ALREADY_PROCESSED_PLACEHOLDER if it has. Marks the object as processed in the second case. Parameters ---------- obj The object to check. Returns ------- Any Either the object itself or the placeholder. c|Sr&r5xs r*r8z3_Memoizer.check_already_processed..gsAr,r;r!r6s r*check_already_processedz!_Memoizer.check_already_processedVs",,S++>>>r,cn||d||tddS)zMarks an object as processed. Parameters ---------- obj The object to mark as processed. c|Sr&r5rBs r*r8z-_Memoizer.mark_as_processed..rsQr,c|Sr&r5rBs r*r8z-_Memoizer.mark_as_processed..ssar,F) memoizing)r<_returnidrDs r*mark_as_processedz_Memoizer.mark_as_processedis8 %%c;;777{{3KK5{AAAr,r:r1 str | Anyc(t|ttttfr |d|jfvr|St|t r- ||t|S#t$rYnwxYw||t|Sr&) isinstanceintfloatstrcomplexALREADY_PROCESSED_PLACEHOLDERrrJhash TypeErrorrK)r!r6r:s r*r<z#_Memoizer._handle_already_processedus      $ ABBBJ c8 $ $  {{3.>???      {{3$4555s A(( A54A5Tobj_to_membership_signCallable[[Any], int] default_funcrIc(||}||jvr|jS|rktjsEt |j|jkr(t jdt jd|j|||S)NzIt looks like the scene contains a lot of sub-mobjects. Caching is sometimes not suited to handle such large scenes, you might consider disabling caching with --disable_caching to potentially speed up the rendering process.z\You can disable this warning by setting disable_caching_warning to True in your config file.) r'rTrdisable_caching_warninglenTHRESHOLD_WARNINGrwarningadd)r!r6rWrYrIobj_membership_signs r*rJz_Memoizer._returns54S99 #"8 8 84 4  <2 .//33HHH6 3  " & &': ; ; ;|C   r,N)r!r"r#r$)F)r!r"r-r.r#r)r!r"r6rr#r)r!r"r6rr#r$)r!r"r6rr:r1r#rM)T) r!r"r6rrWrXrYr1rIr.r#rM)__name__ __module__ __qualname____doc__setr'rTr] classmethodr+r?rErLr<rJr5r,r*rr%s  %)!'''['05[2???[?$ B B B[ B666[66  !!!![!!!r,rc^eZdZddZeddZedd Zd Zdfd ZxZS)_CustomEncoderr6rr#ct|tst|ttfrt j|}it j|jt j|j}t|D] }t||tr||=! t j |}n#ttf$rd}YnwxYw| ||dSt|tjrA|jdkr'tj|d}dt%|St%|St'|dr3|j}t|t*rdS| |St|tjrt/|St1t3|S)aThis method is used to serialize objects to JSON format. If obj is a function, then it will return a dict with two keys : 'code', for the code source, and 'nonlocals' for all nonlocalsvalues. (including nonlocals functions, that will be serialized as this is recursive.) if obj is a np.darray, it converts it into a list. if obj is an object with __dict__ attribute, it returns its __dict__. Else, will let the JSONEncoder do the stuff, and throw an error if the type is not suitable for JSONEncoder. Parameters ---------- obj Arbitrary object to convert Returns ------- Any Python object that JSON encoder will recognize )code nonlocalsi)drmzTRUNCATED ARRAY: __dict__ MappingProxy)rOr r r inspectgetclosurevarscopyglobalsrllist getsourceOSErrorrV_cleaned_iterablenpndarraysizeresizereprhasattrrnr uint8rPrRtype)r7r6cvarscvardictirktemps r*defaultz_CustomEncoder.defaults,3 ++"    &2 2 " *3//EQ$)EM22Qdi6P6PQH(^^ $ $hqk:66$   (--Y'       ))4h*O*OPP P RZ ( ( x$iZ00649966699  S* % % ._key_to_hashs-:dj.AAAHHJJKK Kr,lstrrc,dgt|z}t|D]o\}}t|}t |t t fr |}n#t |tr |}n|}|||<p|Sr&)r\ enumeraterrErOrttupledict)rprocessed_listrel new_value_iter_check_dict_iter_check_lists r*rz:_CustomEncoder._cleaned_iterable.._iter_check_lists"Vc#hh.N"3 . .266r::b4-00# 0 0 4 4IID))# 0 0 4 4II "I$-q!!! !r,dctrci}|D]\}}t|}|tvr)t |t t ttfs| |}n|}t |tr |}n*t |ttfr |}n|}|||<|Sr&) itemsrrErrOrRrPrQr.rrtr) rprocessed_dictkvk_newrrrrs r*rz:_CustomEncoder._cleaned_iterable.._iter_check_dictsN  2 2155a88***!!c3t%<==!-(LOOEEEa&&" 0 0 3 3IID%=11" 0 0 3 3II !I(1u%%! !r,z3'iterable' is neither an iterable nor a dictionary.N)rrr#rP)rrr#r)rrr#r)rOrtrrrV)r7rrrrs @@@r*rwz _CustomEncoder._cleaned_iterables L L L L " " " " " " " " " " " " " " "( hu . . S##H-- - $ ' ' S##H-- -QRR Rr,rRc&t|t|ttt fr4t ||St |S)aOverriding of :meth:`JSONEncoder.encode`, to make our own process. Parameters ---------- obj The object to encode in JSON. Returns ------- :class:`str` The object encoder with the standard json process. ) rrLrOrrtrsuperrrw)r7r6 __class__s r*rz_CustomEncoder.encode*sl ##C((( cD$. / / ?77>>$"8"8"="=>> >ww~~c"""r,)r6rr#r)rrr#r)rrr#rr6rr#rR)rarbrcrrrwr __classcell__)rs@r*rhrhs::::xJJJXJ PPPXP5S5S5Sn##########r,rhr6rr#rRc8tj|tS)zRecursively serialize `object` to JSON using the :class:`CustomEncoder` class. Parameters ---------- obj The dict to flatten Returns ------- :class:`str` The flattened object r))rrrh)r6s r*rr=s :c~ . . ..r, scene_objectr camera_objectCamera | OpenGLCameraanimations_listIterable[Animation]current_mobjects_listIterable[Mobject]ctjdt}t|t |}dt |tD}d|D}d|||fD\}} } |d| d| } t} tjddt| |z d d ittjd d | i| S) a4Take the list of animations and a list of mobjects and output their hashes. This is meant to be used for `scene.play` function. Parameters ----------- scene_object The scene object. camera_object The camera object used in the scene. animations_list The list of animations. current_mobjects_list The list of mobjects. Returns ------- :class:`str` A string concatenation of the respective hashes of `camera_object`, `animations_list` and `current_mobjects_list`, separated by `_`. z Hashing ...c,g|]}t|Sr5r.0rCs r* z+get_hash_from_play_call..lsRRRAHQKKRRRr,rc,g|]}t|Sr5rrs r*rz+get_hash_from_play_call..ms!M!M!M!(1++!M!M!Mr,c3|K|]7}tjt|V8dSr&)rrr|r)rjson_vals r* z*get_hash_from_play_call..nsR;;  4>>((**++;;;;;;r,_zHashing done in %(time)s s.timeNzHash generated : %(h)sh) rdebugrrrLrsortedrRr+) rrrrt_start camera_jsonanimations_list_jsoncurrent_mobjects_list_json hash_camerahash_animationshash_current_mobjects hash_completet_ends r*rrMs'6 LnnG  ---=))KRRS1Q1Q1QRRR!M!M7L!M!M!M;;$&:rs""""""""  BBBBBBBBBBBBHHHHHHHHHHHH//////////(((((((((333333******------;;;;;;'''''' G G GH!H!H!H!H!H!H!H!VJ#J#J#J#J#T%J#J#J#Z / / / / ++++++r,