Ë
    ñukia	  ã                   ót   — d dl Zd dlmZ d dlm Z d dlmZ d dlm	Z	 d dl
mZmZ deded	efd
„Zdeded	efd„Zy)é    N)Úlax)Únumpy)Ú_const)Úpromote_args_inexact)ÚArrayÚ	ArrayLikeÚxÚcÚreturnc                 óÂ  — t        d| |«      \  } }t        j                  t        j                  |t        |d«      «      t        j                  |t        |d«      «      z  t        j                  t        j                  | t        | d«      «      t        j                  | t        | t        j                  dz  «      «      z  t        j                  d||z  z
  «      t        j                  dt        j                  z  «      z
  t        j                  d||z  z   d|z  t        j                  | «      z  z
  «      z
  t        j                   «      t        j                  «      S )a#  Wrapped Cauchy log probability distribution function.

  JAX implementation of :obj:`scipy.stats.wrapcauchy` ``logpdf``.

  The wrapped Cauchy probability distribution function is given by

  .. math::

     f(x, c) = \frac{1-c^2}{2\pi(1+c^2-2c\cos x)}

  for :math:`0<c<1`, and where normalization is on the domain :math:`0\le x\le 2\pi`.

  Args:
    x: arraylike, value at which to evaluate the PDF
    c: arraylike, distribution shape parameter

  Returns:
    array of logpdf values.

  See Also:
    :func:`jax.scipy.stats.wrapcauchy.pdf`
  zwrapcauchy.logpdfr   é   é   )r   ÚjnpÚwherer   ÚgtÚ
_lax_constÚltÚgeÚleÚnpÚpiÚlogÚcosÚinfÚnan©r	   r
   s     úZ/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/scipy/stats/wrapcauchy.pyÚlogpdfr      sý   € ô. 
Ð1°1°aÓ	8$€!€QÜ	‰Ü‡FFˆ1Œj˜˜AÓÓ¤#§&¡&¨¬J°q¸!Ó,<Ó"=Ñ=Ü‡IIÜ	‡ffˆQ”
˜1˜aÓ Ó!¤C§F¡F¨1¬j¸¼B¿E¹EÀA¹IÓ.FÓ$GÑGÜ	‡ggˆa!a‘%‰iÓœ3Ÿ7™7 1¤r§u¡u¡9Ó-Ñ-´·±¸¸AÀ¹E¹	ÀAÈÁEÌCÏGÉGÐTUËJÑDVÑ8VÓ0WÑWÜ	‡vv€góô
 ‡FFó
ð ó    c                 ó@   — t        j                  t        | |«      «      S )a  Wrapped Cauchy probability distribution function.

  JAX implementation of :obj:`scipy.stats.wrapcauchy` ``pdf``.

  The wrapped Cauchy probability distribution function is given by

  .. math::

     f(x, c) = \frac{1-c^2}{2\pi(1+c^2-2c\cos x)}

  for :math:`0<c<1`, and where normalization is on the domain :math:`0\le x\le 2\pi`.

  Args:
    x: arraylike, value at which to evaluate the PDF
    c: arraylike, distribution shape parameter

  Returns:
    array of pdf values.

  See Also:
    :func:`jax.scipy.stats.wrapcauchy.logpdf`
  )r   Úexpr   r   s     r   Úpdfr"   ;   s   € ô. 
‰”˜˜1“Ó	Ðr   )r   r   Újax._srcr   r   Újax._src.lax.laxr   r   Újax._src.numpy.utilr   Újax._src.typingr   r   r   r"   © r   r   ú<module>r(      sN   ðó å Ý !Ý 1Ý 4ß ,ð ˆið  ˜Ið  ¨%ó  ðFˆ9ð ˜ð  uô r   