
    e@                    @   d Z ddlmZ ddlZddlZddlZddlmZmZ ddl	m
Z
mZmZ ddlmZ ddlmZ dd	lmZ d
Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Zej4                  e   ZeD ]#  Z eedej>                  z   ej@                         % eD ]#  Z eedej>                  z   ej@                         % eD ]#  Z eedej>                  z   ej@                         % eD ]#  Z eedej>                  z   ej@                         % ejB                  Z"ejF                  Z$ejF                  ejJ                  z  Z&ejN                  Z(ejN                  ejJ                  z  Z)ejT                  Z+ejX                  Z-ej\                  ej^                  z  ej`                  z  ejb                  z  Z2ejf                  Z4ejj                  Z6ejn                  Z8ejr                  Z:ej^                  Z;ej`                  Z<ejz                  Z>ej~                  ej                  z  ZAej~                  ZBej                  ej                  z  ZEej                  ej                  z  ZGej                  ej                  z  ZIej                  ej                  z  ZKej                  ej                  z  ZMej                  ej                  z  ZOej                  ZQej                  ZSej                  ZUej                  ZWej                  ZYej                  Z[ej                  Z]ej                  Z_ej                  Zaej                  Zcej                  Zeej                  Zgej                  Ziej                  Zk G d dej                        Zl G d dej                        Znd Zod  Zp e
j                  elj                  elep        e
j                  d!en        e
j                  elj                  eo        e
j                  elj                  d"       y)#a  
A Pillow loader for .dds files (S3TC-compressed aka DXTC)
Jerome Leclanche <jerome@leclan.ch>

Documentation:
https://web.archive.org/web/20170802060935/http://oss.sgi.com/projects/ogl-sample/registry/EXT/texture_compression_s3tc.txt

The contents of this file are hereby released in the public domain (CC0)
Full text of the CC0 license:
https://creativecommons.org/publicdomain/zero/1.0/
    )annotationsN)IntEnumIntFlag   )Image	ImageFileImagePalette)i32le)o8)o32leiDDS c                  ,    e Zd ZdZdZdZdZdZdZdZ	dZ
y	)
DDSDr                  i   i   N)__name__
__module____qualname__CAPSHEIGHTWIDTHPITCHPIXELFORMATMIPMAPCOUNT
LINEARSIZEDEPTH     4/usr/lib/python3/dist-packages/PIL/DdsImagePlugin.pyr   r      s*    DFEEKKJEr    r   c                      e Zd ZdZdZdZy)DDSCAPSr   r   i  @ N)r   r   r   COMPLEXTEXTUREMIPMAPr   r    r!   r#   r#   )   s    GGFr    r#   c                  ,    e Zd ZdZdZdZdZdZdZdZ	dZ
y	)
DDSCAPS2i      i   r   i    i @  i   i    N)r   r   r   CUBEMAPCUBEMAP_POSITIVEXCUBEMAP_NEGATIVEXCUBEMAP_POSITIVEYCUBEMAP_NEGATIVEYCUBEMAP_POSITIVEZCUBEMAP_NEGATIVEZVOLUMEr   r    r!   r(   r(   /   s0    GFr    r(   c                  $    e Zd ZdZdZdZdZdZdZy)DDPFr   r   r       @   r   N)	r   r   r   ALPHAPIXELSALPHAFOURCCPALETTEINDEXED8RGB	LUMINANCEr   r    r!   r3   r3   ;   s     KEFO
CIr    r3   c                     e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5Z7d6Z8d7Z9d8Z:d9Z;d:Z<d;Z=d<Z>d=Z?d>Z@d?ZAd@ZBdAZCdBZDdCZEdDZFdEZGdFZHdGZIdHZJdIZKdJZLdKZMdLZNdMZOdNZPdOZQdPZRdQZSdRZTdSZUdTZVdUZWdVZXdWZYdXZZdYZ[dZZ\d[Z]d\Z^d]Z_d^Z`d_Zad`ZbdaZcdbZddcZeddZfdeZgdfZhdgZidhZjdiZkdjZldkZmdlZndmZodnZpdoZqdpZrdqZsdrZtdsZudtZvduZwdvZxdwZydxZzdyZ{yz){DXGI_FORMATr   r   r      r            r   	   
                                                                  r4   !   "   #   $   %   &   '   (   )   *   +   ,   -   .   /   0   1   2   3   4   5   6   7   8   9   :   ;   <   =   >   ?   r5   A   B   C   D   E   F   G   H   I   J   K   L   M   N   O   P   Q   R   S   T   U   V   W   X   Y   Z   [   \   ]   ^   _   `   a   b   c   d   e   f   g   h   i   j   k   l   m   n   o   p   q   r   s                  N)|r   r   r   UNKNOWNR32G32B32A32_TYPELESSR32G32B32A32_FLOATR32G32B32A32_UINTR32G32B32A32_SINTR32G32B32_TYPELESSR32G32B32_FLOATR32G32B32_UINTR32G32B32_SINTR16G16B16A16_TYPELESSR16G16B16A16_FLOATR16G16B16A16_UNORMR16G16B16A16_UINTR16G16B16A16_SNORMR16G16B16A16_SINTR32G32_TYPELESSR32G32_FLOATR32G32_UINTR32G32_SINTR32G8X24_TYPELESSD32_FLOAT_S8X24_UINTR32_FLOAT_X8X24_TYPELESSX32_TYPELESS_G8X24_UINTR10G10B10A2_TYPELESSR10G10B10A2_UNORMR10G10B10A2_UINTR11G11B10_FLOATR8G8B8A8_TYPELESSR8G8B8A8_UNORMR8G8B8A8_UNORM_SRGBR8G8B8A8_UINTR8G8B8A8_SNORMR8G8B8A8_SINTR16G16_TYPELESSR16G16_FLOATR16G16_UNORMR16G16_UINTR16G16_SNORMR16G16_SINTR32_TYPELESS	D32_FLOAT	R32_FLOATR32_UINTR32_SINTR24G8_TYPELESSD24_UNORM_S8_UINTR24_UNORM_X8_TYPELESSX24_TYPELESS_G8_UINTR8G8_TYPELESS
R8G8_UNORM	R8G8_UINT
R8G8_SNORM	R8G8_SINTR16_TYPELESS	R16_FLOAT	D16_UNORM	R16_UNORMR16_UINT	R16_SNORMR16_SINTR8_TYPELESSR8_UNORMR8_UINTR8_SNORMR8_SINTA8_UNORMR1_UNORMR9G9B9E5_SHAREDEXPR8G8_B8G8_UNORMG8R8_G8B8_UNORMBC1_TYPELESS	BC1_UNORMBC1_UNORM_SRGBBC2_TYPELESS	BC2_UNORMBC2_UNORM_SRGBBC3_TYPELESS	BC3_UNORMBC3_UNORM_SRGBBC4_TYPELESS	BC4_UNORM	BC4_SNORMBC5_TYPELESS	BC5_UNORM	BC5_SNORMB5G6R5_UNORMB5G5R5A1_UNORMB8G8R8A8_UNORMB8G8R8X8_UNORMR10G10B10_XR_BIAS_A2_UNORMB8G8R8A8_TYPELESSB8G8R8A8_UNORM_SRGBB8G8R8X8_TYPELESSB8G8R8X8_UNORM_SRGBBC6H_TYPELESS	BC6H_UF16	BC6H_SF16BC7_TYPELESS	BC7_UNORMBC7_UNORM_SRGBAYUVY410Y416NV12P010P016
OPAQUE_420YUY2Y210Y216NV11AI44IA44P8A8P8B4G4R4A4_UNORMP208V208V408SAMPLER_FEEDBACK_MIN_MIP_OPAQUE'SAMPLER_FEEDBACK_MIP_REGION_USED_OPAQUEr   r    r!   r=   r=   E   s~   GONNOLKK! ONMNMOLLKLKLIIHHNMJIJILIIIHIHKHGHGHHOOLINLINLINLIILIILNNN!#MIILINDDDDDDJDDDDDD	BDNDDD&)#.1+r    r=   c                     e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5Z7d6Z8d7Z9 e:d8      Z; e:d9      Z< e:d:      Z= e:d;      Z> e:d<      Z? e:d=      Z@ e:d>      ZA e:d?      ZB e:d@      ZC e:dA      ZD e:dB      ZE e:dC      ZF e:dD      ZG e:dE      ZH e:dF      ZI e:dG      ZJ e:dH      ZKyI)JD3DFMTr   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   r4   rY   rZ   r[   r\   r`   ra   rj   rk   rl   rt   ru   rv   rw   r5   rz   r}   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   t   u   v   w      s   UYVYs   RGBGs   YUY2s   GRGBs   DXT1s   DXT2s   DXT3s   DXT4s   DXT5s   DX10s   BC4Ss   BC4Us   BC5Ss   BC5Us   ATI1s   ATI2s   MET1N)Lr   r   r   r   R8G8B8A8R8G8B8X8R8G8B8R5G6B5X1R5G5B5A1R5G5B5A4R4G4B4R3G3B2A8A8R3G3B2X4R4G4B4A2B10G10R10A8B8G8R8X8B8G8R8G16R16A2R10G10B10A16B16G16R16r"  r!  L8A8L8A4L4V8U8L6V5U5X8L8V8U8Q8W8V8U8V16U16A2W10V10U10D16_LOCKABLED32D15S1D24S8D24X8D24X4S4D16D32F_LOCKABLED24FS8D32_LOCKABLES8_LOCKABLEL16
VERTEXDATAINDEX16INDEX32Q16W16V16U16R16FG16R16FA16B16G16R16FR32FG32R32FA32B32G32R32FCxV8U8A1A2B10G10R10_XR_BIASBINARYBUFFERi32UYVY	R8G8_B8G8r  	G8R8_G8B8DXT1DXT2DXT3DXT4DXT5DX10BC4SBC4UBC5SBC5UATI1ATI2MULTI2_ARGB8r   r    r!   r*  r*     s   GFHHFHHHF	BHHKHHFKLD	B	BDDDFHHFKL
CEEEG
CMFLK
CJGGLDGMDGMF	BLw<DGIw<DGIw<Dw<Dw<Dw<Dw<Dw<Dw<Dw<Dw<Dw<Dw<Dw<Dw<Lr    r*  DDSD_DDSCAPS_	DDSCAPS2_DDPF_c                       e Zd ZdZdZd Zd Zy)DdsImageFileDDSzDirectDraw Surfacec                   t        | j                  j                  d            sd}t        |      t	        j
                  d| j                  j                  d            \  }|dk7  rdt        |       }t        |      | j                  j                  |dz
        }t        |      dk7  rdt        |       d}t        |      t        j                  |      }t	        j
                  d	|j                  d
            \  }}}||f| _        d| j                  z   }t	        j
                  d	|j                  d
            \  }	}
}t	        j
                  d|j                  d             t	        j
                  d|j                  d            \  }}}}d}d }|t        j                  z  ra|t        j                  z  r
d| _        d}n	d| _        d}t	        j
                  d| d|j                  |dz              }d|d||ffg| _        y |t        j$                  z  rB|dk(  r	d| _        n |dk(  r|t        j                  z  r	d| _        nd| d| }t        |      |t        j&                  z  r=d| _        t)        j*                  d| j                  j                  d            | _        n||t        j.                  z  rX|dz   }|t0        j2                  k(  rd| _        d| _        d}n>|t0        j6                  k(  rd| _        d | _        d!}n|t0        j8                  k(  rd| _        d"| _        d}n|t0        j:                  t0        j<                  fv rd| _        d#| _        d}n|t0        j>                  k(  rd| _        d$| _        d%}n|t0        j@                  t0        jB                  fv rd| _        d&| _        d%}ng|t0        jD                  k(  r*|d'z  }t	        j
                  d| j                  j                  d            \  }| j                  j                  d       |tF        jH                  tF        jJ                  fv rd| _        d(| _        d}n|tF        jL                  tF        jN                  fv rd| _        d#| _        d}n|tF        jP                  tF        jR                  fv rd| _        d&| _        d%}nf|tF        jT                  k(  rd| _        d$| _        d%}nA|tF        jV                  k(  rd| _        d)| _        d*}n|tF        jX                  k(  rd| _        d+| _        d*}n|tF        jZ                  tF        j\                  tF        j^                  fv r3d| _        d,| _        d-}|tF        j^                  k(  rd.| j`                  d/<   n|tF        jb                  tF        jd                  tF        jf                  fv r*d| _        |tF        jf                  k(  rId.| j`                  d/<   n9d0| }ti        |      d1t        |       }ti        |      d2| }ti        |      |r+tk        jl                  d3||| j4                  f      g| _        y tk        jl                  d4|d|xs | jn                        g| _        y )5Nr   znot a DDS filez<I|   zUnsupported header size x   zIncomplete header: z bytesz<3IrE   r   r   z<11Ird   <4IrI   r   RGBAr:   r>   <Idds_rgbr   LLAzUnsupported bitcount z for Pr)   rh  r   rj  r   rl  BC4rp  r?   BC5rM   BC1BC6Hr@   BC6HSBC7rA   g]tE?gammazUnimplemented DXGI format zUnimplemented pixel format zUnknown pixel format flags bcnraw)8_acceptfpreadSyntaxErrorstructunpackreprOSErrorlenioBytesIO_sizesizer3   r:   r6   _modetiler;   r9   r	   r  paletter8   r*  rh  pixel_formatrj  rl  ro  rr  rp  rq  rs  rm  r=   r   r   r   r   r  r  r  r  r  r  r  r  infor   r   r   NotImplementedErrorr   _Tilemode)selfmsgheader_sizeheader_bytesheaderflagsheightwidthextentspitchdepthmipmapspfsizepfflagsfourccbitcountnrawmode
mask_countmasksoffsetdxgi_formats                         r!   _openzDdsImageFile._openI  s   tww||A'"Cc""tTWW\\!_=#,T+->,?@C#,ww||K!O4|#'L(9':&AC#,L)%}}UFKKODvuV_
499$ &eV[[_ Eugffkk"o. -3MM%R,Q)TXX)))#

"

MMAj\"3V[[a5PQE#Wa(E1BCDDIt~~%1} 
RGd.>.>$>!
-hZuWIFcl"t+++DJ'++FDGGLL4FGDLt{{" 1_F$#
$*!6;;&#
$*!6;;&#
$*!FKK55 
$)!6;;&"
$*!FKK55"
$)!6;;&"!'tTWW\\!_!ER )),,#  "(DJ(-D%A [%=%={?T?T$UU!$DJ(-D%A [%=%={?T?T$UU!&DJ(-D%A K$9$99!&DJ(.D%A K$9$99!&DJ(.D%A K$9$99!&DJ(/D%A ,,))..% 
 "(DJ(-D%A"k&@&@@-4		'* 11..33% 
 "(DJ"k&E&EE-4		'*6{mDC-c223DL>B)#../y9C%c**wD<M<M8NODI #G<PtyyQRDIr    c                     y )Nr   )r  poss     r!   	load_seekzDdsImageFile.load_seek  s    r    N)r   r   r   formatformat_descriptionr  r  r   r    r!   rz  rz  E  s    F-KSZr    rz  c                      e Zd ZdZd Zy)DdsRgbDecoderTc           	     "   | j                   \  }}g }g }|D ]V  }d}|dk7  r(||dz   z	  |dz   z  |k(  r|dz  }||dz   z	  |dz   z  |k(  r|j                  |       |j                  ||z	         X t               }|dz  }	t        |      | j                  j
                  | j                  j                  z  t        |      z  k  rt        j                  | j                  j                  |	      d      }
t        |      D ]0  \  }}|
|z  }|t        t        |||   z	  ||   z  dz              z  }2 t        |      | j                  j
                  | j                  j                  z  t        |      z  k  r| j                  t        |             y)Nr   r   r   little   )r   )argsappend	bytearrayr  statexsizeysizeint
from_bytesfdr  	enumerater   
set_as_rawbytes)r  bufferr  r  mask_offsetsmask_totalsmaskr  data	bytecountvalueimasked_values                r!   decodezDdsRgbDecoder.decode  s   ))%  	/DFqyvz*vz:dBaKF vz*vz:dB'tv~.	/ {M	$i$****TZZ-=-==E
JJNN477<<	#:HEE$U+ 4$t|,,q/9[^KsRS  $i$****TZZ-=-==E
JJ 	d$r    N)r   r   r   	_pulls_fdr  r   r    r!   r  r    s    Ir    r  c                   | j                   dvrd| j                    d}t        |      | j                   d   dk(  }| j                   d   dk(  r(t        j                  }| j                   }|rg d}n\g d	}nWt        j                  }| j                   d d d   }g d
}|r/| j                         \  }}	}
}t        j                  d|||	|
f      } |r|t        j                  z  }|j                  |rdnd       t        j                  t        j                  z  t        j                  z  t        j                  z  t        j                  z  }t!        | j#                               dz  }| j$                  |z  dz   dz  }|j'                  t)        t*              t-        j.                  dd|| j0                  | j$                  |dd      z   t-        j.                  dgd z   t-        j.                  dd|d|      z   t-        j.                  dg| z   t-        j.                  dt2        j4                  dddd      z          t7        j8                  | |t7        j:                  dd| j<                  z   d|ddf      g       y )N)r:   r  r  r  zcannot write mode z as DDSr  Ar   r  )r  r  r  )      ~ r  r  )i   i   r  r  r  r   rA   z<7Ir}  11I)r   r   r   r   r   r   r   r   r   r   r   r  r4   z<5Ir  r  r   )r  r  r3   r;   r:   splitr   merger6   r  r   r   r   r   r   r   r  getbandsr  writeo32	DDS_MAGICr  packr  r#   r%   r   _saver  r  )imr  filenamer  alphapixel_flagsr  	rgba_maskrgbar  r  r  s                  r!   r  r    s    	ww00"277)73clGGBK3E	wwqzSnn''<I<Ihh''$B$-8	JAq!QVaAq\2Bt'''5Za0II#djj04::=@P@PPE2;;=!A%HXX 1$*EHHI
++IIHH	

	
 ++e
*y
*	+ ++eRa
:	; ++e
(i
(	) ++eW__aAq
9	:$ OO
B(8!gq!_MNr    c                    | d d dk(  S )Nr   s   DDS r   )prefixs    r!   r  r  /  s    "1:  r    r  z.dds)v__doc__
__future__r   r  r  sysenumr   r    r   r   r	   _binaryr
   rd  r   r   r  r  r   r#   r(   r3   r=   r*  modulesr   moduleitemsetattrnamer  r8   
DDS_FOURCCr:   DDS_RGBr6   DDS_RGBAr;   DDS_LUMINANCEDDS_LUMINANCEAr7   	DDS_ALPHAr9   DDS_PAL8r   r   r   r   DDS_HEADER_FLAGS_TEXTUREr   DDS_HEADER_FLAGS_MIPMAPr   DDS_HEADER_FLAGS_VOLUMEr   DDS_HEADER_FLAGS_PITCHr   DDS_HEADER_FLAGS_LINEARSIZE
DDS_HEIGHT	DDS_WIDTHr%   DDS_SURFACE_FLAGS_TEXTUREr$   r&   DDS_SURFACE_FLAGS_MIPMAPDDS_SURFACE_FLAGS_CUBEMAPr*   r+   DDS_CUBEMAP_POSITIVEXr,   DDS_CUBEMAP_NEGATIVEXr-   DDS_CUBEMAP_POSITIVEYr.   DDS_CUBEMAP_NEGATIVEYr/   DDS_CUBEMAP_POSITIVEZr0   DDS_CUBEMAP_NEGATIVEZrh  DXT1_FOURCCrj  DXT3_FOURCCrl  DXT5_FOURCCr   DXGI_FORMAT_R8G8B8A8_TYPELESSr   DXGI_FORMAT_R8G8B8A8_UNORMr   DXGI_FORMAT_R8G8B8A8_UNORM_SRGBr  DXGI_FORMAT_BC5_TYPELESSr  DXGI_FORMAT_BC5_UNORMr  DXGI_FORMAT_BC5_SNORMr  DXGI_FORMAT_BC6H_UF16r  DXGI_FORMAT_BC6H_SF16r  DXGI_FORMAT_BC7_TYPELESSr  DXGI_FORMAT_BC7_UNORMr  DXGI_FORMAT_BC7_UNORM_SRGBrz  	PyDecoderr  r  r  register_openr  register_decoderregister_saveregister_extensionr   r    r!   <module>r+     s  
 # 	  
 ! , , !  ! 	7 g w 7 y2' y2xI W I Z 
X	 5DFGdii'45 8DFJ*DJJ78 9DFK$))+TZZ89 5DFGdii'45 [[

((88d&&&$"2"22JJ	99t{{2TZZ?$BRBRR ** **  "oo [[
JJ	#OO "??W^^; #OO  ((8+E+EE  ((8+E+EE  ((8+E+EE  ((8+E+EE  ((8+E+EE  ((8+E+EE kkkkkk + = = (77 "-"A"A &33 #-- #-- #-- #-- &33 #-- (77 R9&& RjI'' B1h!   L''w ?   y- 0   L'' /   ,,f 5r    