j? dZddlZddlmZddlmZmZmZm Z dejde de dejfd Z Gd d ZGd d Zdejdejdeee ffdZedk(redededd\ZZej,dedef\ZZddej2dej4zdzezez zej6dej4zdzezez zzdej8eedz z dzeedz z dzz dz zzdej:eedz z dkzzdej:eedz z dkzzZej>eddjAejBZededeed !Z"e"jGeZ$e"jKd"ejMd"Z'ee'Z(e(jSZ*eee*Z+ed#e+ejXded$z edzd%zZ-ejXded$z edzd%zZ.ej^e.e-d&'\Z0Z1e(jee1e0Z3ed(e3jhd)e(jkd*d+d*d+d,d,-Z6ed.e6jhd/ed0d1D]|Z7ee7!Z8e8jGed23Z9ee9Z:e:jSZ;eee;Zd?e>4841 ==ceZdZdZ ddeedeedefdZede jdedefd Z dde jd e de fd Zd eeeffd Zed eeefde fdZy) SkissEncoderu Retro-compresses an explicit H×W image into a finite parameter tensor θ. The fitting exploits the separable Kronecker structure: min_Θ || I - U Θ Vᵀ ||_F² solved via the normal equations: A Θ Bᵀ = Uᵀ I V, where A = UᵀU, B = VᵀV. Nr epsilon reg_lambdacZ| | td||_||_||_d|_y)aq Args: K: Manual critical rank (number of Fourier modes per axis). If None, epsilon must be provided. epsilon: Fraction of spectral energy to retain (0 < epsilon < 1). Used to auto-estimate K if K is not given. reg_lambda: Tiny ridge for numerical stability near degenerate modes. NzSpecify either K or epsilon.) ValueErrorK_manualr%r&theta)selfr r%r&s r __init__zSkissEncoder.__init__;s5 9;< <  $%) r"imager cl|jdk(r|jd}n|}|j\}}tjj |j tj}tj|j}tj|ddd}tj|}|ddz} ttj|d|z | zdz} ttjtj| dz d z } t!| dS) z6Auto-estimate critical rank from FFT energy retention.rNg-q=?rr)ndimmeanrrfftfft2astyperabsflattensortcumsumint searchsortedceilsqrtmax) r-r%grayHWFmagvalsr;totaln_keepK_ests r _estimate_KzSkissEncoder._estimate_KLs ::?::1:%DDzz1 FFKK BJJ/ 0ffQi!wws|DbD!4r U"R__VcGmu-DEFJBGGRWWV_s2c9:;5!}r"verbosec |jdk(r |dddddf}|j\}}}|jtj}|j |j n|j ||j}t|t||dzdz }t|d}tj|tj}tj|tj} t||t|} t| |t|} | j| z} | j| z} |jdkDrd| |jtj| jdzz } | |jtj| jdzz } tj |d|zdzd|zdzftj}t#|D]w}| j|dddd|fz| z}tj$j'| |}tj$j'| |jj}|||<y||||f|t|t|dt|jd|_|rF||z|z}|j*}||z }t-d|d |d |t-d |d |d |d |dd |j(S)u= Compress an image into the skiss residue θ. Args: image: ndarray, shape (H, W) or (H, W, C). Values typically in [0, 255]. verbose: Print compression statistics. Returns: theta: Serializable dict containing the spectral residue. r0Nrrr real_fourier_separable) coefficientsr rchannelsperiod_yperiod_xbasisr&z[SkissEncoder] Image: ×z[SkissEncoder] K=z | Params: z | Pixels: z | Compression: z.1f)r3rr7rrr)rJr%minr@rr!floatTr&eyezerosrangelinalgsolver*sizeprint)r+r-rKrBrCCIr y_coordsx_coordsUVABcoeffscRHSZTheta_c orig_size param_sizeratios r encodezSkissEncoder.encode_s ::?!Q*%E++1a LL $!]]6DMMDA% & 1I99Qbjj199Qbjj1 )1eAh ? (1eAh ? CC!G CC!G ??Q  266!''!*#55 5A 266!''!*#55 5A 1a!eaiQ32::Fq A##!Q' "Q&C 3'Aiiooa-//GF1I  #Vaa-0    A IJ *E *1#Rs"QC8 9 %aS j\B&K'9%BH Izzr"pathc|j tdtjt |fi|jy)z(Save theta as a compressed .npz archive.Nz%Nothing to save. Call encode() first.)r* RuntimeErrorrsavez_compressedstr)r+ros r savezSkissEncoder.saves3 :: FG G CI44r"ctjt|d}|jDcic]'\}}||jdk(r|j n|)}}}|Scc}}w)zLoad a theta dict from disk.F) allow_pickler)rloadrsitemsr3item)rodatarvr*s r load_thetazSkissEncoder.load_thetasWwws4yu5AENA1QVVX!3NN Os,A%)NNgư>)T)__name__ __module__ __qualname____doc__rr<rUr, staticmethodrndarrayrJboolrrnrrsrrtr|r"r r$r$1sLP%)*(3-*%*"*"2::#$FBJJFFFP5sDy)5 sDy)dr"r$cLeZdZdZdefdZ ddejdejdee de d ejf d Z dd ejd ejdee d ejfd Z ddee e e fd ejfdZ dedededede de d ejfdZy) SkissDecoderu Generates pixels from the skiss residue θ at arbitrary coordinates. The explicit pixel path is never reconstructed unless explicitly requested. r*c||_t|d|_|d|_t|d|_t |d|_t |d|_|d\|_|_ y)Nr rNrOrPrQr) r*r<r rfr^rUrPrQH_origW_orig)r+r*s r r,zSkissDecoder.__init__si U3ZN+ U:&'eJ/0 eJ/0 #(>  T[r"Nxychannel chunk_sizer ctj|tj}tj|tj}|j|jk7r t d|j}|j }|j }|j }||n||jfz} tj||n ||jftj} td||D]} t| |z|} || | } || | }t| |j|j}t||j|j}|0|j|}tj ||z|zd| | | t|jD]5}|j|}tj ||z|zd| | | |f<7| j#| S)uU Evaluate the parametric Fourier surface at arbitrary real coordinates. Uses chunked processing to remain memory-efficient for large query sets. Args: x: ndarray of arbitrary shape (float, in units of original pixel grid). y: ndarray, same shape as x. channel: None → return all channels (..., C). int → return single channel (same shape as x). chunk_size: Number of points to process per batch. Returns: values: ndarray of generated pixel intensities. r z"x and y must have identical shape.rrr)rrrrr(ravelr\r^rXrYrTr!r rQrPrfsumreshape)r+rrrr orig_shapeflat_xflat_yn_pts out_shapeoutiendbatch_xbatch_yV_batchU_batchThetargs r generatezSkissDecoder.generates JJq + JJq + 77agg AB BWW  ")"5J: ;Q hh 3u%PRPZPZ[q%, PAa*ne,CQsmGQsmG37DFFDMMRG27DFFDMMRG" G,VVWu_$?aHAc tvvPA KKNE$&FFGeOw+FQ$OC#qMP P"{{9%%r"rar`c~tj|tj}tj|tj}t||j|j }t||j|j }|!|j|}||z|jzSt|t|}}tj|||jftj} t|jD]-} |j| }||z|jz| dddd| f</| S)a Memory-efficient reconstruction for grid-aligned coordinates. Exploits the separable property: I = U @ Theta @ V.T Args: x_coords: 1D array of x-coordinates (width W'). y_coords: 1D array of y-coordinates (height H'). channel: None for all channels, or int for specific channel index. Returns: values: (H', W', C) or (H', W') array. r N) rrrr!r rQrPrfrVlenrXr^rY) r+rar`rrcrbrH_primeW_primerrgs r generate_gridzSkissDecoder.generate_grids::hbjj9::hbjj9 )4664== I (4664== I  KK(Eu9qss? ""8}c(mWG((GWdff5RZZHC466] / A 5y133Aq!G  /Jr"rc||j|jf}|\}}tj|tj}tj|tj}|j ||S)z\ Reconstruct a full pixel grid. If shape is None, uses original dimensions. r )rrrrrr)r+rrBrCr`ras r reconstructzSkissDecoder.reconstructs_ =[[$++.E199Qbjj199Qbjj1!!(H55r"x_minx_maxy_miny_maxres_xres_yctj|||}tj|||}|j||S)zO Generate a continuous viewport patch at arbitrary resolution. )rlinspacer) r+rrrrrrrar`s r reconstruct_patchzSkissDecoder.reconstruct_patch%s; ;;ueU3;;ueU3!!(H55r")Ni)N)r}r~rrrr,rrrr<rrrrrUrrr"r rrs 2d2OS#*.&"**.&.&hsm.& .&/1zz.&b04bjjBJJ'}8: B 6%S/!: 6bjj 66u6U6!&6/46!$6-0657ZZ6r"roriginal reconstructedc|jtj}|jtj}ttj||z dz}|dk(r td}nE|j dkDrdnd}dtj |tj|z z}|||dkDdS) zDReturn MSE and PSNR between original and skiss-reconstructed images.r0rinfr2go@g4@g>@)msepsnrskiss_complete)r7rrrUr4r@log10r?)rrrrmax_vals r skiss_metricsr4srzz*H!((4M M1a78 9C axU|#<<>C/%Sbhhw566t DDr"__main__zF======================================================================u'Skiss-Fourier Image Codec — Self-Test)rg`@gP@r0r/g@@g @g0@z Synthetic test image: rS)r zskiss_residue.npzz Full reconstruction metrics: rg?ij)indexingz Super-resolution sample shape: u3 (generated at 2× native resolution from θ alone)2Pi,)rrrrrrzViewport patch shape: z+ (arbitrary resolution from skiss manifold)z' --- Progressive Skiss Transmission ---)rr0rF)rKr#u ✅ Solidu ⚠️ Liquidu❌ Gasz K=2dz | Params=rN5dz | PSNR=z6.2fz dB | z* Done. Residue saved to: skiss_residue.npz)?rnumpyrpathlibrtypingrrrrrr<rUr!r$rrsrr}r]rBrCmgridrrrrrexpr8test_imgclipr7uint8encoderrnr*rtr| theta_loadeddecoderrrecmetricsrfine_xfine_ymeshgrid fine_y_grid fine_x_gridr super_resrrpatchK_testenc_tmpth_tmpdec_tmprec_tmpmstatusr\rr"r rs //>>>U>rzz>2BBRs6s6t EBJJ Erzz Ed3PU:FV E" z (O 34 (O DAq 88AaC1H DAq  BEE A )A-. .BEE A 8IA8M1N N O 1qs7Q,!ac'A56-HI I J &"&&QqS/A% & ' &"&&QqS/A% &  ' rwwxC(//9H $QCr! -.R G NN8 $E LL$% **+>?L<(G    CHc*G +G9 56R[[AE1q5 )C /F R[[AE1q5 )C /F*r{{66DIK  k:I -ioo->?? @A  % %Bb"-0 & =E "5;;-07 89 45";(%8v&%%' (G , !& B& TV_\e VBK|F>,B,G,G+KL64(: ;; 78{r"