a =Df@sdZddlmZddlZeeZddlmZddl m Z m Z m Z m Z mZmZdZddd d d Zd'd d d d ddddZddddZddddddZd(d d ddddZddd ddZd)d d d d dddd Zed!Zd"d#d$d%d&ZdS)*a Provide a set of decorators useful for repeatedly updating a a function parameter in a specified way each time the function is called. These decorators can be especially useful in conjunction with periodic callbacks in a Bokeh server application. Example: As an example, consider the ``bounce`` forcing function, which advances a sequence forwards and backwards: .. code-block:: python from bokeh.driving import bounce @bounce([0, 1, 2]) def update(i): print(i) If this function is repeatedly called, it will print the following sequence on standard out: .. code-block:: none 0 1 2 2 1 0 0 1 2 2 1 ... ) annotationsN)partial)AnyCallableIterableIteratorSequenceTypeVar)bouncecosinecountforcelinearrepeatsinez Sequence[int]zpartial[Callable[[], None]])sequencereturncs.tdddfdd }ttt|dS)a Return a driver function that can advance a "bounced" sequence of values. .. code-block:: none seq = [0, 1, 2, 3] # bounce(seq) => [0, 1, 2, 3, 3, 2, 1, 0, 0, 1, 2, ...] Args: sequence (seq) : a sequence of values for the driver to bounce intircs6t|\}}|ddkr"|S|dSdS)Nr)divmod)rdivmodNrZ/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/bokeh/driving.pyf[s zbounce..frlenrr _advancerrrrrr Lsr rfloat)wAphioffsetrcs8ddlmdddfdd }ttt|dS)a Return a driver function that can advance a sequence of cosine values. .. code-block:: none value = A * cos(w*i + phi) + offset Args: w (float) : a frequency for the cosine driver A (float) : an amplitude for the cosine driver phi (float) : a phase offset to start the cosine driver with offset (float) : a global offset to add to the driver values r)cosr%rcs|SNrrr'r*r)r(r&rrrrszcosine..fr )mathr*rr r#r&r'r(r)rrr-rr cs r rcCstttdddS)z@ Return a driver function that can advance a simple count. cSs|Sr+r)xrrrzzcount..r rr r#rrrrr vsr zCallable[[Any], None]z Iterator[Any]zCallable[[], None])rrrcsddfdd }|S)z Return a decorator that can "force" a function with an arbitrary supplied generator Args: sequence (iterable) : generator to drive f with Returns: decorator Noner0cstdSr+)nextrrrrrwrapperszforce..wrapperr)rrr8rr7rr |s r )mbrcs&dddfdd }ttt|dS)z Return a driver function that can advance a sequence of linear values. .. code-block:: none value = m * i + b Args: m (float) : a slope for the linear driver x (float) : an offset for the linear driver r%rcs |Sr+rr,r:r9rrrszlinear..fr r4)r9r:rrr;rrs rcs.tdddfdd }ttt|dS)a  Return a driver function that can advance a repeated of values. .. code-block:: none seq = [0, 1, 2, 3] # repeat(seq) => [0, 1, 2, 3, 0, 1, 2, 3, 0, 1, ...] Args: sequence (seq) : a sequence of values for the driver to bounce rrcs |Sr+rr,rrrrszrepeat..fr r!r$rrrrs rcs8ddlmdddfdd }ttt|dS)a Return a driver function that can advance a sequence of sine values. .. code-block:: none value = A * sin(w*i + phi) + offset Args: w (float) : a frequency for the sine driver A (float) : an amplitude for the sine driver phi (float) : a phase offset to start the sine driver with offset (float) : a global offset to add to the driver values r)sinr%rcs|Sr+rr,r'r)r(r<r&rrrszsine..fr )r.r<rr r#r/rr=rrs rTzCallable[[int], T]z Iterable[T])rrccsd}||V|d7}qdS)z Yield a sequence generated by calling a given function with successively incremented integer values. Args: f (callable) : The function to advance Yields: f(i) where i increases each call rrNr)rrrrrr#s  r#)rrr)rr)rrr)__doc__ __future__rlogging getLogger__name__log functoolsrtypingrrrrrr __all__r r r r rrrr>r#rrrrs