Fj+<dZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl Z ddl ZddlmZeejjjZGddZGdd Zd Zed krejedSdS) a FreeExile PBR Texture & Normal Map Synthesizer Engine (100% Free & Open-Source) Converts 2D Diffuse Sprites into: 1. Tangent-Space Normal Maps (Sobel gradient filter + Height reconstruction) 2. Specular / Roughness Maps (Surface microfacet frequency analysis) 3. Emissive / Glow Maps (Elemental hue isolation for Fire/Thunder/Poison visual FX) 4. MaxRects Power-of-Two (2^n) Texture Atlas Packer with UV coordinate mapping Designed specifically to feed Apple Metal 120 FPS Shaders (SpritePBRNormal.metal). N)Path)DictListTupleOptional)Imagec VeZdZdZeddejdedejfdZe dd ejd e d ed e ejdejf dZ edd ejde dejfdZ edd ejde dejfdZ ededededeeeffdZd S)PBRTextureSynthesizerzSZero-cost, mathematically pure PBR material synthesizer for 2.5D Isometric Sprites.rgba iterationsreturncV|}|ddddddf}|dddddf}|dktj}t jtjd}t|D]}t j||}|dk|dkz} tj | snOtdD]<} t j|dddd| f|} | | |dddd| f| <=|}||ddddddf<||dddddf<|S)z Extends RGB colors of opaque pixels into transparent regions (alpha == 0) to completely eliminate black borders / alpha bleeding on dark backgrounds, while strictly preserving the original alpha channel. Nr)rr) copyastypenpuint8cv2getStructuringElement MORPH_RECTrangedilateany) r r resultrgbalphamaskkernel_ dilated_maskfringec dilated_cs EC:\Projects\FreeExile\tools\asset_pipeline\pbr_texture_synthesizer.pyapply_dilation_paddingz,PBRTextureSynthesizer.apply_dilation_paddings\QQQ2A2Xqqq!!!Qw !!"(++*3>6BBz""  A:dF33L"a'DAI6F6&>> 1XX 9 9Js111aaa7|V<< '0'8AAAqqq!G V$$DDqqq!!!RaRxqqq!!!Qw @rN diffuse_bgr intensity smooth_ksizerct|jdkr3|jddkr"||dddddf}|ddddddf}tj|tjt jdz }|dkrtj|||fd}tj |tj ddd}tj |tj ddd}| |z}| |z}t j |} t j |dz|dzz| dzz} t j | d } || z d zd z} || z d zd z} | | z d zd z} | dzt j}| dzt j}| dzt j}||dk}d ||<d ||<d ||<t j|||f}|S) aZ Generate Tangent-Space Normal Map (RGB where R=X, G=Y, B=Z) using Sobel gradients. Normal vector (Nx, Ny, Nz) mapped from [-1, 1] to [0, 255]. OpenCV BGR layout: Channel 0=B(Nz), Channel 1=G(Ny), Channel 2=R(Nx). Transparent pixels default to flat normal (128, 128, 255 in RGB -> [255, 128, 128] in BGR). rNo@rrksizegư>g?)lenshapercvtColorCOLOR_BGR2GRAYrrfloat32 GaussianBlurSobelCV_32F ones_likesqrtmaximumrdstack)r*r+r,rgraysobel_xsobel_ydxdydznormnxnynzb_chg_chr_ch trans_mask normal_bgrs r&generate_normal_mapz)PBRTextureSynthesizer.generate_normal_map9s { ! !Q & &;+z0SpriteAtlasPacker.pack_sprites..sd1gmTUFVr(T)keyreverser/rlrr.NrrzTexture Atlas size xz exceeded!r3) pixel_rectuv_rect)sorteditemsrzerosrr6r5 RuntimeErrorrr7COLOR_GRAY2BGRrur@max)rrsorted_spritesatlas_watlas_hatlasuv_mapcurr_xcurr_y row_heightpaddingnamerxhwchannelsrzrs r& pack_spriteszSpriteAtlasPacker.pack_spritess  1 1 3 39V9V`deee  '7A.bh??? '! ,! ,ID#9RaR=DAq'*39~~':':sy||HzG#g--*w.. zG##"#d#d#d.#d#d#deee1}}>AfVaZ' )::;;%]]cc S#BT0U0UAq 3bh???>@ie >U>UfVaZ' )::; &vq!4W$W$aZ7*aZ7* F4L a'k !FZ++JJf}r(N)r) rrrrrrrrrrrrdictrrr(r&rrsyWW55$sBJ"755X]^`^hjnortxoxjy^yXz555\555r(rc*tdtdtdtdz dz dz }|ddd}tj||d ftj }t j|d d d dt j|ddddt j|dddd|dz }t j t||tdt ||d}| D]-\}}td|dd|.tdt j|dt jt j|dt jt j|dt jd}t"|d !\}} |d"z } |d#z } t j t| |t'| d$d%&5} t)j| | d'(dddn #1swxYwYtd)| td*| td+d,S)-NzA=================================================================z7 FreeExile PBR Texture & Normal Map Synthesizer Engineclientassets generated_pbrTrir4r/rl)@r0)-r3): )Fr)rrr3)r( )rrr3r3zdemo_martial_hero.pngz&Generating PBR maps for demo sprite...hero_azure_bladez * z<10z: z2 Packing into 512x512 Power-of-Two Sprite Atlas...rnrorc) hero_diffuse hero_normal hero_emissivei)rzatlas_pbr_hero.pngzatlas_pbr_hero.jsonrzutf-8)encodingr.)indentz OK Atlas image generated: z OK UV mapping JSON saved: z. PBR Synthesis pipeline verified successfully!r)print PROJECT_ROOTrprrrrcircle rectanglervrrr rrupperrqrsrropenjsondump) test_outsize demo_sprite demo_input pbr_pathskvsprites_to_packr uv_coords atlas_path json_pathfs r&mainrs (OOO CDDD (OOOh&1OCH NN4$N/// D(D$?"(;;;KJ{Hb*=rBBBM+x3GLLLJ{Hb*rsQ   ............ tH~~%%''.5< VVVVVVVVr99999999x+ + + \ z CHTTVVr(