a ]Df, @sbdZddlmZddlZeeZddlZddl Z ddl m Z ddl m Z ddlmZmZddlZddlmZdd lmZd d lmZerddlmZddlZdd lmZe dd dddZdddddZe ej!e ej"e ej#e ej$e ej%e ej&e ej'e ej(e ej)h Z*e+ddZ,e-d dZ.ejj/dej0j1dZ2eed3dde*DdZdZ4ddddd Z5dd!dd"d#Z6d$d%dd&d'Z7d(d%dd)d*Z8d+d%dd,d-Z9d.d/d0d1d2Z:d3dd4d5Z;d3dd6d7Zee>jd=3d>de*Dde>_d.d.d0d?d@Z?dAd.dBdCdDZ@dEaAe ZBdS)Fz Functions for helping with serialization and deserialization of Bokeh objects. Certain NumPy array dtypes can be serialized to a binary format for performance and efficiency. The list of supported dtypes is: {binary_array_types} ) annotationsN) lru_cache)Lock) TYPE_CHECKINGAny)ID)settings)format_docstring) TypeGuardz set[type])returncCsPddl}tjtjtjh}||j||j||j |t |j |S)Nr) pandasdttimedatetimenp datetime64add Timestamp TimedeltaPeriodtypeNaT)pdresultre/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/bokeh/util/serialization.py_compute_datetime_types6s   rstrr)namer cCs|dkrtStdS)NZDATETIME_TYPES)rAttributeError)r rrr __getattr__Asr"ms)tz ccs|]}d|dVqdSz* ``np.z``Nr.0xrrr Yr*)Zbinary_array_types) array_encoding_disabledconvert_date_to_datetimeconvert_datetime_arrayconvert_datetime_typeconvert_timedelta_typeis_datetime_typeis_timedelta_type make_globally_unique_css_safe_idmake_globally_unique_idmake_idtransform_arraytransform_seriesz0TypeGuard[dt.time | dt.datetime | np.datetime64])objr cCstt}t||S)z Whether an object is any date, time, or datetime type recognized by Bokeh. Args: obj (object) : the object to test Returns: bool : True if ``obj`` is a datetime type )tupler isinstance)r8Z _dt_tuplerrrr1ns r1z(TypeGuard[dt.timedelta | np.timedelta64]cCst|tjtjfS)z Whether an object is any timedelta type recognized by Bokeh. Args: obj (object) : the object to test Returns: bool : True if ``obj`` is a timedelta type )r:r timedeltar timedelta64r8rrrr2}s r2zdt.datefloatcCs,tj|dddtjjitdS)z Convert a date object to a datetime Args: obj (date) : the object to convert Returns: datetime Ntzinfo)rr timetupletimezoneutcDT_EPOCH total_secondsr=rrrr-s r-zdt.timedelta | np.timedelta64cCsBt|tjr|dSt|tjr0t|tStd|dS)z Convert any recognized timedelta value to floating point absolute milliseconds. Args: obj (object) : the object to convert Returns: float : milliseconds @@zUnknown timedelta object: N) r:rr;rFrr<r> NP_MS_DELTA ValueErrorr=rrrr0s    r0zSAny | pd.Timestamp | pd.Timedelta | dt.datetime | dt.date | dt.time | np.datetime64cCsddl}||jurtjSt||jr2|jdSt||jrH|jdSt||j r^|jdSt|t j r|j t j jdt}|dSt|t jrt|St|tjr|t}t|tSt|t jr|jd|jd|jd|jdStd |dS) z Convert any recognized date, time, or datetime value to floating point milliseconds since epoch. Args: obj (object) : the object to convert Returns: float : milliseconds rNg.A)r@rAi<rGzunknown datetime object: )rrrnanr:r to_timestampvaluerrrrreplacerCrDrErFdater-rNP_EPOCHr>rHrhourminutesecond microsecondrI)r8rdiffZ epoch_deltarrrr/s(             (r/npt.NDArray[Any]znpt.NDArray[np.floating[Any]]arrayr cCsddddd}|jjdkr*||dS|jjdkrD||dS|jjd krt|d krt|d tjrz||dWStyYn0|S) a Convert NumPy datetime arrays to arrays to milliseconds since epoch. Args: array : (obj) A NumPy array of datetime to convert If the value passed in is not a NumPy array, it will be returned as-is. Returns: array rVrWcSs tt|tj|ddS)Nint64rG)rwhereZisnatrKastyperXrrrconvertsz'convert_datetime_array..convertMzdatetime64[us]mztimedelta64[us]Or)dtypekindr[lenr:rrO Exception)rXr]rrrr.s   ( r.rcCsNtrDt&td7atdtWdS1s80YntSdS)a Return a new unique ID for a Bokeh object. Normally this function will return simple monotonically increasing integer IDs (as strings) for identifying Bokeh objects within a Document. However, if it is desirable to have globally unique for every object, this behavior can be overridden by setting the environment variable ``BOKEH_SIMPLE_IDS=no``. Returns: str r pN)r Z simple_ids_simple_id_lock _simple_idrr4rrrrr5s .r5cCstttS)z Return a globally unique UUID. Some situations, e.g. id'ing dynamically created Divs in HTML documents, always require globally unique IDs. Returns: str )rruuiduuid4rrrrr4s r4cCs>d}td|D]}t}|dr|SqtdtS)a Return a globally unique CSS-safe UUID. Some situations, e.g. id'ing dynamically created Divs in HTML documents, always require globally unique IDs. ID generated with this function can be used in APIs like ``document.querySelector("#id")``. Returns: str drzbk-)ranger4isalphar)Zmax_iterZ_iidrrrr3s   r3boolcCs |jtvS)z Determine whether an array may be binary encoded. The NumPy array dtypes that can be encoded are: {binary_array_types} Args: array (np.ndarray) : the array to check Returns: bool )raBINARY_ARRAY_TYPESr\rrrr,.sr,z ccs|]}d|dVqdSr&rr'rrrr*Br+cCst|}dddddd}|jttjkr:||tj}n|jttjkrX||tj}t|tjj rr| tj }|j dst |}|S)z Transform a ndarray into a serializable ndarray. Converts un-serializable dtypes and returns JSON serializable format Args: array (np.ndarray) : a NumPy array to be transformed Returns: ndarray rVz type[Any])rXrar cSs:t|}t||jk|j|kBr(|S|j|ddSdS)Nunsafe)Zcasting)rZiinfoanyminmaxr[)rXrainforrr _cast_if_canWs z%transform_array.._cast_if_canZ C_CONTIGUOUS)r.rarrYint32Zuint64uint32r:maZ MaskedArrayZfilledrKflagsZascontiguousarray)rXrurrrr6Es     r6zApd.Series[Any] | pd.Index[Any] | pd.api.extensions.ExtensionArray)seriesr cCs,ddl}t||jr |j}n|}|S)z Transforms a Pandas series into serialized form Args: series (pd.Series) : the Pandas series to transform Returns: ndarray rN)rr:Z PeriodIndexrLvaluesZto_numpy)rzrvalsrrrr7ks   r7i)C__doc__ __future__rlogging getLogger__name__logrrrh functoolsr threadingrtypingrrnumpyrZ core.typesrr stringsr Z numpy.typingZnptrrZtyping_extensionsr rr"raZbool_Zuint8Zint8Zuint16int16rwrvfloat32float64rorrPr<rH fromtimestamprCrDrEjoin__all__r1r2r-r0r/r.r5r4r3r,r6r7rgrfrrrrsh                       3 &