o
    Z#a%                  
   @   s  d dl Z d dlZzd dlmZ dZe Zee W n ey8 Z zddl	m
Z
 e
e ZZW Y dZ[ndZ[ww e eZG dd dZG dd	 d	eZG d
d deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZi dedededed ed!ed"ed#ed$ed%ed&ed'ed(ed)ed*ed+eZejd,kreed-< eed.< eed/< eed0< eed1< need-< eed.< eed/< eed0< eed1< d5d3d4ZdS )6    N)FFIz|
    struct Pixel_RGBA {
        unsigned char r,g,b,a;
    };
    struct Pixel_I16 {
        unsigned char l,r;
    };
       )deferred_errorc                   @   s>   e Zd ZdddZdd Zdd Zdd	 ZeZeZd
d Z	dS )PyAccessFc                 C   s   t |jj}|| _td|d | _td|d | _td|d | _|jj	\| _
| _|| _|j| _| jjdkr=|j| _|   d S )Nzunsigned char **image8zint **image32imageP)dictimunsafe_ptrsreadonlyfficastr   r   r   sizexsizeysize_img_immodepalette_palette
_post_init)selfimgr   vals r   ./usr/lib/python3/dist-packages/PIL/PyAccess.py__init__2   s   zPyAccess.__init__c                 C   s   d S Nr   )r   r   r   r   r   E   s   zPyAccess._post_initc                 C   s   | j rtd|\}}|dk r| j| }|dk r| j| }| ||f\}}| jjdkrAt|tt	frAt
|dv rA| j|| j}| |||S )a0  
        Modifies the pixel at x,y. The color is given as a single
        numerical value for single band images, and a tuple for
        multi-band images

        :param xy: The pixel coordinate, given as (x, y). See
           :ref:`coordinate-system`.
        :param color: The pixel value.
        z%Attempt to putpixel a read only imager   r	   )      )r   
ValueErrorr   r   check_xyr   r   
isinstancelisttuplelenr   getcolorr   	set_pixel)r   xycolorxyr   r   r   __setitem__H   s   


zPyAccess.__setitem__c                 C   sJ   |\}}|dk r| j | }|dk r| j| }| ||f\}}| ||S )aq  
        Returns the pixel at x,y. The pixel is returned as a single
        value for single band images or a tuple for multiple band
        images

        :param xy: The pixel coordinate, given as (x, y). See
          :ref:`coordinate-system`.
        :returns: a pixel value for single band images, a tuple of
          pixel values for multiband images.
        r   )r   r   r#   	get_pixelr   r*   r,   r-   r   r   r   __getitem__e   s   

zPyAccess.__getitem__c                 C   sR   |\}}d|  kr| j k r#n tdd|  kr"| jk s'td td|S )Nr   zpixel location out of range)r   r   r"   r0   r   r   r   r#   {   s   zPyAccess.check_xyNF)
__name__
__module____qualname__r   r   r.   r1   putpixelgetpixelr#   r   r   r   r   r   1   s    
r   c                   @   (   e Zd ZdZdd Zdd Zdd ZdS )	_PyAccess32_2z7PA, LA, stored in first and last bytes of a 32 bit wordc                 O      t d| j| _d S Nzstruct Pixel_RGBA **r   r   r   pixelsr   argskwargsr   r   r   r         z_PyAccess32_2._post_initc                 C   s   | j | | }|j|jfS r   )r=   rar   r,   r-   pixelr   r   r   r/      s   z_PyAccess32_2.get_pixelc                 C   s2   | j | | }t|d d|_t|d d|_d S )Nr      r   )r=   minrB   rC   r   r,   r-   r+   rE   r   r   r   r)      s   z_PyAccess32_2.set_pixelNr3   r4   r5   __doc__r   r/   r)   r   r   r   r   r9      
    r9   c                   @   r8   )	_PyAccess32_3zARGB and friends, stored in the first three bytes of a 32 bit wordc                 O   r:   r;   r<   r>   r   r   r   r      rA   z_PyAccess32_3._post_initc                 C   s   | j | | }|j|j|jfS r   )r=   rB   gbrD   r   r   r   r/         z_PyAccess32_3.get_pixelc                 C   sH   | j | | }t|d d|_t|d d|_t|d d|_d|_d S )Nr   rF   r      r=   rG   rB   rM   rN   rC   rH   r   r   r   r)      s
   
z_PyAccess32_3.set_pixelNrI   r   r   r   r   rL      rK   rL   c                   @   r8   )	_PyAccess32_4z&RGBA etc, all 4 bytes of a 32 bit wordc                 O   r:   r;   r<   r>   r   r   r   r      rA   z_PyAccess32_4._post_initc                 C   s"   | j | | }|j|j|j|jfS r   )r=   rB   rM   rN   rC   rD   r   r   r   r/      s   z_PyAccess32_4.get_pixelc                 C   sR   | j | | }t|d d|_t|d d|_t|d d|_t|d d|_d S )Nr   rF   r   rP   r    rQ   rH   r   r   r   r)      s
   z_PyAccess32_4.set_pixelNrI   r   r   r   r   rR      rK   rR   c                   @   r8   )	
_PyAccess8z%1, L, P, 8 bit images stored as uint8c                 O      | j | _d S r   )r   r=   r>   r   r   r   r         z_PyAccess8._post_initc                 C      | j | | S r   r=   r   r,   r-   r   r   r   r/         z_PyAccess8.get_pixelc                 C   H   zt |d| j| |< W d S  ty#   t |d d| j| |< Y d S w )NrF   r   rG   r=   	TypeErrorr   r,   r-   r+   r   r   r   r)      
   z_PyAccess8.set_pixelNrI   r   r   r   r   rS      
    rS   c                   @   r8   )	_PyAccessI16_Nz0I;16 access, native bitendian without conversionc                 O   r:   )Nzunsigned short **r   r   r   r=   r>   r   r   r   r      rA   z_PyAccessI16_N._post_initc                 C   rV   r   rW   rX   r   r   r   r/      rY   z_PyAccessI16_N.get_pixelc                 C   rZ   )N  r   r[   r]   r   r   r   r)      r^   z_PyAccessI16_N.set_pixelNrI   r   r   r   r   r`      r_   r`   c                   @   r8   )	_PyAccessI16_LzI;16L access, with conversionc                 O   r:   Nzstruct Pixel_I16 **ra   r>   r   r   r   r      rA   z_PyAccessI16_L._post_initc                 C   s   | j | | }|j|jd  S N   r=   lrB   rD   r   r   r   r/      rO   z_PyAccessI16_L.get_pixelc                 C   sV   | j | | }zt|d}W n ty   t|d d}Y nw |d@ |_|d? |_d S )Nrb   r   rF      )r=   rG   r\   rh   rB   rH   r   r   r   r)         
z_PyAccessI16_L.set_pixelNrI   r   r   r   r   rc      rK   rc   c                   @   r8   )	_PyAccessI16_BzI;16B access, with conversionc                 O   r:   rd   ra   r>   r   r   r   r      rA   z_PyAccessI16_B._post_initc                 C   s   | j | | }|jd |j S re   rg   rD   r   r   r   r/      rO   z_PyAccessI16_B.get_pixelc                 C   sV   | j | | }zt|d}W n ty   t|d d}Y nw |d? |_|d@ |_d S )Nrb   r   ri   rF   )r=   rG   	Exceptionrh   rB   rH   r   r   r   r)      rj   z_PyAccessI16_B.set_pixelNrI   r   r   r   r   rk      rK   rk   c                   @   r8   )	_PyAccessI32_Nz"Signed Int32 access, native endianc                 O   rT   r   r   r=   r>   r   r   r   r   
  rU   z_PyAccessI32_N._post_initc                 C   rV   r   rW   rX   r   r   r   r/     rY   z_PyAccessI32_N.get_pixelc                 C   s   || j | |< d S r   rW   r]   r   r   r   r)     s   z_PyAccessI32_N.set_pixelNrI   r   r   r   r   rm     r_   rm   c                   @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )_PyAccessI32_Swapz,I;32L/B access, with byteswapping conversionc                 O   rT   r   rn   r>   r   r   r   r     rU   z_PyAccessI32_Swap._post_initc                 C   s\   t d|}t d|}|d |d |d |d f\|d< |d< |d< |d< t d|d S )Nzint *zunsigned char *r    rP   r   r   )r   newr   )r   iorigcharsr   r   r   reverse  s   4z_PyAccessI32_Swap.reversec                 C   s   |  | j| | S r   rt   r=   rX   r   r   r   r/      rA   z_PyAccessI32_Swap.get_pixelc                 C   s   |  || j| |< d S r   ru   r]   r   r   r   r)   #  s   z_PyAccessI32_Swap.set_pixelN)r3   r4   r5   rJ   r   rt   r/   r)   r   r   r   r   ro     s    ro   c                   @   r8   )	
_PyAccessFz32 bit float accessc                 O   r:   )Nzfloat **r<   r>   r   r   r   r   *  rA   z_PyAccessF._post_initc                 C   rV   r   rW   rX   r   r   r   r/   -  rY   z_PyAccessF.get_pixelc                 C   s<   z
|| j | |< W d S  ty   |d | j | |< Y d S w )Nr   )r=   r\   r]   r   r   r   r)   0  s
   z_PyAccessF.set_pixelNrI   r   r   r   r   rv   '  r_   rv   1Lr	   LALaPARGBLABHSVYCbCrRGBARGBaRGBXCMYKFIlittlezI;16zI;16LzI;16BzI;32LzI;32BFc                 C   s.   t | jd }|std| j d S || |S )NzPyAccess Not Implemented: %s)mode_mapgetr   loggerdebug)r   r   access_typer   r   r   rp   \  s
   
rp   r2   )loggingsyscffir   defsr   cdefImportErrorex_utilr   	getLoggerr3   r   r   r9   rL   rR   rS   r`   rc   rk   rm   ro   rv   r   	byteorderrp   r   r   r   r   <module>   s   
Q	


