U
    ڲg                     @  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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   7/tmp/pip-unpacked-wheel-xela2va7/PIL/FtexImagePlugin.pyr   A   s   r   c                   @  s2   e Zd ZdZdZddddZdddd	d
ZdS )FtexImageFileZFTEXzTexture File Format (IW2:EOC)None)returnc                 C  s0  t | jdsd}t|td| jd td| jd| _td| jd\}}d| _|dkspttd| jd\}}| j	| td| jd\}| j|}|t
jkrd| _d	d
| j ddfg| _n:|t
jkrdd
| j ddfg| _ndt| }t|| j  t|| _d S )N   znot an FTEX filez<iz<2i   RGBr   ZRGBAZbcn)r   r   r   raw)r   r   r   z$Invalid texture compression format: )_acceptfpreadSyntaxErrorstructunpack_size_modeAssertionErrorseekr   r   sizeZtiler   repr
ValueErrorcloser   )selfmsgZmipmap_countZformat_countformatwhereZmipmap_sizedatar   r   r   _openJ   s*    

zFtexImageFile._openint)posr   c                 C  s   d S )Nr   )r%   r,   r   r   r   	load_seekj   s    zFtexImageFile.load_seekN)r	   r
   r   r'   format_descriptionr*   r-   r   r   r   r   r   F   s    r   bytesbool)prefixr   c                 C  s   | d d t kS )Nr   )MAGIC)r1   r   r   r   r   n   s    r   z.ftcz.ftu)__doc__
__future__r   r   enumr   ior    r   r   r2   r   r   r   Zregister_openr'   Zregister_extensionsr   r   r   r   <module>   s   5(