o
    n¨ iÏ  ã                   @  s–   d Z ddlmZ ddlZddlmZ ddlmZ ddlm	Z	m
Z
 dZG d	d
„ d
eƒZG dd„ de
j
ƒZddd„Ze	 ejee¡ e	 ejddg¡ dS )a  
A Pillow loader for .ftc and .ftu files (FTEX)
Jerome Leclanche <jerome@leclan.ch>

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/

Independence War 2: Edge Of Chaos - Texture File Format - 16 October 2001

The textures used for 3D objects in Independence War 2: Edge Of Chaos are in a
packed custom format called FTEX. This file format uses file extensions FTC
and FTU.
* FTC files are compressed textures (using standard texture compression).
* FTU files are not compressed.
Texture File Format
The FTC and FTU texture files both use the same format. This
has the following structure:
{header}
{format_directory}
{data}
Where:
{header} = {
    u32:magic,
    u32:version,
    u32:width,
    u32:height,
    u32:mipmap_count,
    u32:format_count
}

* The "magic" number is "FTEX".
* "width" and "height" are the dimensions of the texture.
* "mipmap_count" is the number of mipmaps in the texture.
* "format_count" is the number of texture formats (different versions of the
same texture) in this file.

{format_directory} = format_count * { u32:format, u32:where }

The format value is 0 for DXT1 compressed textures and 1 for 24-bit RGB
uncompressed textures.
The texture data for a format starts at the position "where" in the file.

Each set of texture data in the file has the following structure:
{data} = format_count * { u32:mipmap_size, mipmap_size * { u8 } }
* "mipmap_size" is the number of bytes in that mip level. For compressed
textures this is the size of the texture data compressed with DXT1. For 24 bit
uncompressed textures, this is 3 * width * height. Following this are the image
bytes for that mipmap level.

Note: All data is stored in little-Endian (Intel) byte order.
é    )ÚannotationsN)ÚIntEnum)ÚBytesIOé   )ÚImageÚ	ImageFiles   FTEXc                   @  s   e Zd ZdZdZdS )ÚFormatr   r   N)Ú__name__Ú
__module__Ú__qualname__ÚDXT1ÚUNCOMPRESSED© r   r   úN/home/app/PyTorch/.pytorch/lib/python3.10/site-packages/PIL/FtexImagePlugin.pyr   A   s    r   c                   @  s(   e Zd ZdZdZddd„Zdd	d
„ZdS )ÚFtexImageFileZFTEXzTexture File Format (IW2:EOC)ÚreturnÚNonec                 C  s6  t | j d¡ƒsd}t|ƒ‚t d| j d¡¡ t d| j d¡¡| _t d| j d¡¡\}}|dks5J ‚t d| j d¡¡\}}| j |¡ t d| j d¡¡\}| j |¡}|tj	krnd| _
t dd	| j d
d¡g| _n!|tjkr„d| _
t dd	| j d
d¡g| _ndt|ƒ› }t|ƒ‚| j ¡  t|ƒ| _d S )Né   znot an FTEX filez<iz<2ié   r   ZRGBAZbcn)r   r   r   )r   ÚRGBÚrawz$Invalid texture compression format: )Ú_acceptÚfpÚreadÚSyntaxErrorÚstructÚunpackÚ_sizeÚseekr   r   Ú_moder   Z_TileÚsizeZtiler   ÚreprÚ
ValueErrorÚcloser   )ÚselfÚmsgZmipmap_countZformat_countÚformatÚwhereZmipmap_sizeÚdatar   r   r   Ú_openJ   s*   


zFtexImageFile._openÚposÚintc                 C  s   d S ©Nr   )r$   r*   r   r   r   Ú	load_seeki   s   zFtexImageFile.load_seekN)r   r   )r*   r+   r   r   )r	   r
   r   r&   Úformat_descriptionr)   r-   r   r   r   r   r   F   s
    
r   ÚprefixÚbytesr   Úboolc                 C  s
   |   t¡S r,   )Ú
startswithÚMAGIC)r/   r   r   r   r   m   s   
r   z.ftcz.ftu)r/   r0   r   r1   )Ú__doc__Ú
__future__r   r   Úenumr   Úior   Ú r   r   r3   r   r   r   Zregister_openr&   Zregister_extensionsr   r   r   r   Ú<module>   s    5
'