Gj$8dZddlmZddlZddlZddlmZddlZddlZddl Z ddl Z ddl m Z m Z mZmZmZddlmZddlZegdZdZdZ d0d1dZd2dZd3dZ d4d5d$Z d6d7d*Zd8d-Zd9d.Ze d/kre j!edSdS):z FreeExile ASTC 4x4 Mobile Texture Compression Engine (Apple Metal iOS). Converts Power-of-Two PNG Atlases into canonical 16-byte header ASTC 4x4 binary containers. Complies with Apple Metal MTLPixelFormatASTC_4x4_sRGB and MTLPixelFormatASTC_4x4_LDR. ) annotationsN)Path)DictAnyListOptionalUnion)Image)\widthintheightblock_xblock_yblock_zreturnbytesc|dd}|dd}|dd}t|||g}t|z|z|z|zS)a Generates canonical 16-byte ASTC container header. Byte 0..3: 0x13, 0xAB, 0xA1, 0x5C (Magic) Byte 4: blockdim_x (e.g. 4 for 4x4) Byte 5: blockdim_y (e.g. 4 for 4x4) Byte 6: blockdim_z (e.g. 1 for 2D texture) Byte 7..9: xsize (24-bit little-endian integer) Byte 10..12: ysize (24-bit little-endian integer) Byte 13..15: zsize (24-bit little-endian integer, 1 for 2D) little byteorder)to_bytesr ASTC_MAGIC) rrrrrw_bytesh_bytesd_bytesdimss =C:\Projects\FreeExile\tools\asset_pipeline\astc_compressor.pycreate_astc_headerr&st"nnQ(n33Gooa8o44GqH55G '7G, - -D  w & 07 :: header_bytesDict[str, Any]c|t|tkr(tdt|dtd|dd}|tk}|d}|d}|d}t|dd d }t|d d d }t|d d d }|dkr ||zdz |znd} |dkr ||zdz |znd} | | z|pdz} | t z} ||||||||| | | | t| zd S)z0Parses and validates a 16-byte ASTC file header.zHeader too short: z bytes (expected )rr rr rr) magic_validmagicrrrrrdepthblocks_xblocks_y total_blocksexpected_payload_bytesexpected_total_size)lenASTC_HEADER_SIZE ValueErrorr r from_bytesASTC_BLOCK_BYTEShex) r(r2r1rrrrrr3r4r5r6expected_data_sizes r%parse_astc_headerr@2st <+++ec,.?.?eeRbeeefff 1 EJ&K1oG1oG1oG NN<"-N B BE ^^LB/8^ D DF NN<2.(N C CE3:Q;;!#//AH4;aKK 1$00QHh&%*15L%(88#$"4/2DD  r'rgbacd}|dd}|dz|z}|dz|z}|dz|z}|dz|z} tjd|||| } || zS)a( Encodes a 16-byte (128-bit) ASTC Void-Extent LDR block per Khronos ASTC Specification. Void-extent blocks encode a constant RGBA color across the 4x4 texels with zero compression distortion. Bits 0..8: 0b111111100 (0x1FC: Void-extent mode) Bit 9: 0 (LDR mode) Bits 10..63: 0xFFFF across 4 coordinates (all texels covered) Bits 64..79: R (16-bit unorm: (r << 8) | r) Bits 80..95: G (16-bit unorm: (g << 8) | g) Bits 96..111: B (16-bit unorm: (b << 8) | b) Bits 112..127: A (16-bit unorm: (a << 8) | a) l}rrzMMM*.png source_dir output_dirpattern List[Path]ct|}|r!t|n|}|ddg}||D][}|rE||djz } t|| ||} || \|S)zIBatch compresses all matching PNG images in directory to ASTC containers.TrZrYrr) rrirnglobis_filerlnamerr) rrrrrsrcdstresultspng_file target_fileout_ps r%batch_compress_directoryrs z   " " $ $C(2 ;$z   " " $ $ $CIIdTI***GHHW%%""      " 4 4W = = BBK(;Y`aaaE NN5 ! ! ! Nr' astc_pathboolct|}|r"|jt krdSt |d5}|t }dddn #1swxYwYt|}|dsdS|j}||dkS)z\Verifies that an ASTC file is valid and matches expected header dimensions and payload size.FrbNr1r8) rrirjrwrxr:roreadr@)rprr(info actual_sizes r%verify_astc_filers Y!!A 88::),<<<u a0!vv.// 000000000000000 \ * *D  u&&(("K $45 55s*BBBc tjd}|dd|ddd|d td d |d td d |ddd|ddd|}t |j}|jr.t|}td|rdndd||rdndS|j s| rFt||j |j|j}tdt!|d|dSt#||j |j|j}td|d|d |jd!dS)"Nz@FreeExile ASTC 4x4 Mobile Texture Compressor for Apple Metal iOS) descriptioninputzInput PNG file or directory)helpz-oz--outputzOutput ASTC file or directoryz --block-xrzBlock width (default: 4))typedefaultrz --block-yzBlock height (default: 4)z--batch store_truezBatch process directory)actionrz--verifyzVerify ASTC file integrity[VALIDINVALIDz ] ASTC file: rrrz[OK] Batch compressed z ASTC textures from z[OK] Compressed z -> z (z bytes))argparseArgumentParser add_argumentr parse_argsrrverifyrprintbatchis_dirroutputrrr9rrwrx)parserargsinpvalidrouts r%mainrs  $1s t t tF &CDDD j/NOOO  #q?YZZZ  #q?Z[[[  ,=VWWW  <>Z[[[     D tz  C {! %% EU1'' EEEEFFF qqq  zSZZ\\*3 T\[_[ghhh Ns7||NNNNOOOq sDKt| \ \ \C HS H Hc H HSXXZZ-? H H HIII 1r'__main__)rrr) rrrrrrrrrrrr)r(rrr)) rArrBrrCrrDrrr)NrrrQ) rRrSrTrUrrrrrVrWrr)Nrrr) rrSrrUrrWrrrrrr)rrSrr)rr)"__doc__ __future__rrospathlibrrrrGrtsystypingrrrrr PILr numpyrzrr r:r=r&r@rPrrrr__name__exitr'r%rs #"""""   33333333333333U+++ , ,   ;;;;;0!!!!H%%%%8/3 IIIII\.2 * 6 6 6 6    6 z CHTTVVr'