a Df@sdZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z ddl ZddlmZddlmZddlmZmZmZdd lmZdd lmZdd lmZmZmZdd lm Z d dl!m!Z!erddl"m#Z#ddlm$Z$ddl%m&Z&ddl'm(Z(Gdddej)Z*dddddZ+d;dddddd d!Z,dd"d/d0d1d2d3d4Z2ed?dd6d7d8d9d:Z3dS)@z* Utilities for manipulating bokeh models. ) annotationsN)contextmanager) TYPE_CHECKINGAnyIterableListOptionalSet) Serializer)Document)ColumnDataChangedEventDocumentPatchedEventModelChangedEvent) PatchJson) DataModel)ColumnDataSourceFlexBoxModel) patch_doc)state)HoldPolicyType)DocumentChangedEvent)Message)Commc@s0eZdZdZdddddZdddddZd S) comparable_arrayz1 Array subclass that allows comparisons. rbool)otherreturncCstj||ddSNT)Z equal_nannpZ array_equalselfrr$[/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/panel/io/model.py__eq__*szcomparable_array.__eq__cCstj||dd Srr r"r$r$r%__ne__-szcomparable_array.__ne__N)__name__ __module__ __qualname____doc__r&r'r$r$r$r%r%srzList[DocumentChangedEvent]None)eventsrcCsp|D]f}tt|ddtr$d|j_qt|trt|jtrt|jt j rt |jj |jj |j}||_||_qdS)zp Patch events applies patches to events that are to be dispatched avoiding various issues in Bokeh. hintN) isinstancegetattrr r.colsrmodelrnewr!ndarrayrshapeZdtypeZserializable_new)r-eZ new_arrayr$r$r%monkeypatch_events0s  r7Tr rz$Optional[List[DocumentChangedEvent]]zMessage[Any] | None)docbinaryr-rcs|durt|jj}|rtjr"dSdd|Ds8dStt|jj|dt  d}t }t |d|i|}fdd|jjD|j_|j fdd |rjD]}||q|S) z` Returns a json diff required to update an existing plot with the latest plot data. NcSsg|]}t|tr|qSr$)r/r ).0eventr$r$r% Pzdiff..)Z referencesdeferred)r-Z use_bufferscsg|]}|vr|qSr$r$)r:r6) patch_eventsr$r%r<Xr=cs | S)N)Zhas_ref)r2) serializerr$r%Yr=zdiff..)list callbacksZ _held_eventsr_holdr7r modelsZsynced_referencesrencoderZ create_headerZ flush_syncedbuffersZ add_buffer)r8r9r-Z patch_jsonheadermsgbufferr$)r?r@r%diffDs$    rKrzDocument | NonezSet[Model] | None)objreplaceskiprcCsZt}|dtiD]@}|r&||vr&q|j}d|_|r@|||rJ||_||q|S)zI Removes the document from any previously displayed bokeh object typeN)setselectrdocumentZ _document remove_rootadd)rLrMrNrEr2Zprev_docr$r$r%rS_s   rSF)rLr8holdrNcCs2t||d}|||jjdur.|r.||S)zX Adds a model to the supplied Document removing it from any existing Documents. )rNN)rSZadd_rootrC hold_valuerU)rLr8rUrNrEr$r$r% add_to_docps   rWcCs|didgD]}|ddks|ddkr6q|d|ddi}t|ts\q|d iD]0\}}t|trld d t|D|d |<qlqd S) zf Required for handling messages containing JSON serialized typed array from the frontend. contentr-kindZ ModelChangedattrdataidr2r3cSsg|] \}}|qSr$r$)r:_vr$r$r%r<r=z!patch_cds_msg..N)getZ select_oner/ritemsdictsorted)r2rIr;Zcdscolvaluesr$r$r% patch_cds_msg{s  re) childrentextnameZtoolbarZ renderersZbelowcenterleftrightintzOptional[Iterable[str]]str)rLdepthignoredrc Cs2|dur t}ddlm}t||r.|t}d}t|j}t|j dd }g}|D]b\}} ||vrjqXt| t rdt| j} nt | } t | dkr| ddd } |d || fqXd |} t|tr|d j|d 7}|jD]$} |tt| |dddd7}q|d| 7}n|dj|| d7}|S)zl Returns a string repr for a bokeh model, useful for recreating panel objects using pure bokeh. N)ViewableF)Zinclude_defaultsz%s()z...z%s=%sz, z{cls}(children=[ )clsr)rnz, z z], %s)z{cls}({props}))rtprops)_DEFAULT_IGNORED_REPRZviewablerqr/Zget_rootr rOr(rbZproperties_with_valuesr`rreprlenappendjoinrformatrftextwrapindent bokeh_repr) rLrnrorqrrtZ propertiesrukr^Z props_reprZ child_objr$r$r%r~s4          "r~combinerz Comm | None)r8policycommc cs|ptj}|durdVdS|jj}zZ|dur8|n ||dVW|rX||j_q|durvddlm}||||n6|r||j_n&|durddlm}||||0dS)a Context manager that holds events on a particular Document allowing them all to be collected and dispatched when the context manager exits. This allows multiple events on the same object to be combined if the policy is set to 'combine'. Arguments --------- doc: Document The Bokeh Document to hold events on. policy: HoldPolicyType One of 'combine', 'collect' or None determining whether events setting the same property are combined or accumulated to be dispatched when the context manager exits. comm: Comm The Comm to dispatch events on when the context manager exits. Nr)push) rZcurdocrCrVZunholdrUrDZnotebookr)r8rrZheldrr$r$r%rUs,          rU)TN)NN)FN)rN)rN)4r+ __future__rr| contextlibrtypingrrrrrr numpyr!Zbokeh.core.serializationr Zbokeh.documentr Zbokeh.document.eventsr r rZbokeh.document.jsonrZ bokeh.modelrZ bokeh.modelsrrrZ!bokeh.protocol.messages.patch_docrrZbokeh.core.enumsrrZbokeh.protocol.messagerZ pyviz_commsrr4rr7rKrSrWre frozensetrvr~rUr$r$r$r%s:                $