
    e                         d dl Z d dlZi Zd Zd Z e j                  d      d        Zd Z e j                  d      d        Z	y)	    Nc                     t        | j                        D ]&  }| |   |z
  |z  }t        t        ||      |      ||<   ( y N)rangesizeminmax)xxscaleoffsetvminvmaxyyivals           ;/usr/lib/python3/dist-packages/pyqtgraph/functions_numba.pyrescale_clip_sourcer      sA    277^ *!uv~&CTND)1*    c                 R   t        j                  | |      }| j                  j                  |j                  j                  f}t        j                  |      }|8 t        j                  |d    d|d    dgdd      t              }|t        |<    || |||d   |d   |	       |S )
Ndtyper   z[:],f8,f8,f8,f8,   z[:]z(n),(),(),(),()->(n)Tnopython)out)	np
empty_liker   namerescale_functionsgetnumbaguvectorizer   )datar
   r   r   clipdata_outkeyfuncs           r   rescaleDatar'      s    }}T/H::??HNN//
0C  %D|u  Ax'Axs34" /0 "&#ufd1gtAwH=Or   Tr   c                    d|j                   d   dz
  }}t        | j                   d         D ]U  }t        | j                   d         D ]8  }| ||f   |z
  |z  }	t        t        |	|      |      }	|t	        |	         |||f<   : W y )Nr   r   )shaper   r   r   int)
r"   r
   r   lutr   r   r   rcr   s
             r   _rescale_and_lookup1d_functionr.      s    CIIaL1$$D4::a=! &tzz!}% 	&A1:&%/Cc#tnd+CCHC1I	&&r   c                     t        j                  | |j                        }t        | t	        |      t	        |      ||       |S )Nr   )r   r   r   r.   float)r"   r
   r   r+   r$   s        r   rescale_and_lookup1dr1   !   s5     }}T3H"4uuV}c8TOr   c                 .    t        j                  | |      S r   )r   take)r+   r"   s     r   
numba_taker4   (   s     773r   )
r    numpyr   r   r   r'   jitr.   r1   r4    r   r   <module>r8      s^      *
 D& & D r   