+ju~ddlZddlZddlZddlZddlmZddlZddlZddl m Z ddl m Z m ZmZddlmZddlmZddlZddlZddlmZmZmZddlmZdd lmZd Z d Z!d Z"d ej#de$de$fdZ%de&de'e(e(e(ffdZ)dZ*de+fdZ,dZ-dZ.dZ/dZ0de+fdZ1dZ2dZ3dZ4e dz Z5de+fdZ6idd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d d0d"d1d$d2d&d3d(d4d*d5d,d6d.Z7d7d8d9d:Z8dd;dZ9d?Z:d@Z;dAe+fdBZdAe+fdEZ?ee?dFZ@dAe+de&fdGZAd ej#de$fdHZBdIZCdJZDdKZEdLZFGdMdNejGZHGdOdPejGZIdQZJdRZKdSZLd}dWZMd~dXZNdYZOdZZPd[ZQ dd]ZRGd^d_ejGZSGd`daejGZTGdbdcejGZUGdddeejGZVdfZWdgeXdej#fdhZYGdidjejGZZdkZ[dTddej\fdlZ]dmZ^dnZ_ddoZ`dpZaddrZbdduZcdvZddd ej#fdwZeGdxdyejGZfGdzd{eZgdegfd|ZhdS)N)BytesIO)override)map_coordinatesminimummaximum) coo_matrix)connected_components)ComfyExtensionIOTypes)pack_variable_mesh_batch) PromptServergmBP ?cttj|dk|dz |ddzdz dzS)N?ܵ?ףp= )@r)\(?zG?333333@torchwhere clamp_mincs A/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_gaussian_splat.py_srgb_to_linearrs8 ;qG|QY!++a..52HE1QVY0Y Z ZZcttj|dk|dzd|ddzzdz S)Ngsåi?rrrg?rrrs r_linear_to_srgbrs8 ;qI~q5y%!++a..W:U2UX]2] ^ ^^rgireturnc|j,t|j|n|jjdS)N)countsintitem positionsshape)r r!s r _real_lenr*!s9&'h&:3qx{!! " " " @QRS@TTrhcdtdkrdStfddDS)N#)r/r/c3TK|]"}t||dzddz V#dS)o@N)r&).0r!r+s r z_hex_to_rgb..+s>@@QqQwZ$$u,@@@@@@r)rr1)lstriplentuple)r+s`r _hex_to_rgbr:&sI  A 1vv{{ @@@@i@@@ @ @@rcd}||kr#|dd||zdz}tj||S)Nr$)numelrquantile)xqlims r _quantilerB.sO Cwwyy3 %%c!A%% & >!Q  rc|tj}|jd}|dkrt dtj|}|tj}|dddddf} |ddddddfddd |d} |tj |d dd} tj | d | z z } tj |tj d } |tj} gd d tdDzdt| jdDzdgzdtdDzdtdDz}tj |d|D}ttt tj||| | | | | gd|dd<dd|dz}|dd|Ddzz }|d|zS)a<Serialize render-ready gaussian tensors as a binary 3DGS .ply. positions (N,3) world; scales (N,3) linear; rotations (N,4) quat wxyz; opacities (N,1) in [0,1]; sh (N,K,3) SH coefficients. Activated values are inverted to the standard 3D gaussian splat storage convention (log scale, logit opacity). r SplatToFile3D: gaussian is emptyNr$r1ư>g !??:0yE>min)r?yznxnynzcg|]}d|S)f_dc_r4r!s r z'_gaussian_ply_bytes..Ks...  ...rcg|]}d|S)f_rest_rRrSs rrTz'_gaussian_ply_bytes..Ls>>>! ! >>>ropacitycg|]}d|S)scale_rRrSs rrTz'_gaussian_ply_bytes..Ms===qlqll===rcg|]}d|S)rot_rRrSs rrTz'_gaussian_ply_bytes..Ms@^@^@^PQ@^@^@^rr6cg|]}|dfS)f4rRr4as rrTz'_gaussian_ply_bytes..Ns!;!;!;1d)!;!;!;rdtypeaxisz$ply format binary_little_endian 1.0 zelement vertex  c3"K|] }d|dV dS)zproperty float reNrRr_s rr5z&_gaussian_ply_bytes..Rs.==!----======rz end_header ascii)cpunumpyastypenpfloat32r) ValueError zeros_like transposereshapecliplograngeemptylistmapr9 concatenatejoinencodetobytes)r(scales rotations opacitiesshxyznnormalsff_dcf_restopscalerotattrselementsheaders r_gaussian_ply_bytesr6s  --//   ! ! ( ( 4 4C ! AAvv;<<<mC  G  ++A QQQ111W:D qqq!""aaax[ " "1a + + 3 3Ar : :F    ' ' 3 3 ; ;Aq A A F FtX V VB cBh B F6::<<%%''..rz::??D?II J JE --//   ! ! ( ( 4 4C . . ...U1XX.../>>uV\!_'='=>>>?{>=E!HH===>A_@^UZ[\U]U]@^@^@^_Ex!;!;U!;!;!;<<>DsJAq Q QC    ' ' 3 3 ; ;Aq A A F Fq! L LB 8BNC91555= > > E Ebh O OD^S%-A 6 6 6 = =e D DF (1-~> ? ? ?CCHCHJ + , ,F"F1I"F1I T61a((( T61a((( T62q))) T62q))) T62q))) VVRc3777 UFBS111455G T7Aq))) T7Aq))) T7B,C(CDDD T7B*** T7B*** T7B*** ==5>> )J 66riNGSPr1rg333333?cN |tj}|jd}|dkrt ddtz}tjtj ||zddtj }|dztj } tj | dz| dz dz| d z dzgd  |d tj} tj |tj |d zddtj} |dddddftjtzdz} tj | dz t z dzd zddtj} tj|tjd}tj |dzdzddtj}|tj}|tj|dddz }||dddfdkxxdzcc<tj |ddddfdzdzddtj}t)d }t+jd|dt.t+jd|dt0t+jd|d|d|d<t|d<d|d<d|d<t3|| z| z| z|z|z}t7j|S) zSerialize gaussian tensors as a gzip-compressed .spz (Niantic v2, SH degree 0, base color only). positions (N,3) world; scales (N,3) linear; rotations (N,4) wxyz; opacities (N,1) in [0,1]; sh (N,K,3). rrDr$iiirr2rErc r3Nrg& .>rI$@0@Trrr6rG_@rr )rirjrkrlrmr)rn_SPZ_FRACTIONAL_BITSrrrint32uint32stackrqrr_SPZ_COLOR_SCALErsrrrrr _SPZ_MAGIC _SPZ_VERSIONrr{gzipcompress)r(r|r}r~rrrfixedqiquposalpharcolslnscbrrotbrraws r_gaussian_spz_bytesrs --//   ! ! ( ( 4 4C ! AAvv;<<< % %E #+&& M B B I I"( S SB x-   * *B (BIa4/"(d1BC" M M M U UVWYZ [ [ b bceck l lC HY]]__**,,33BJ??GGJJUR S S X XYZ\_ ` ` g ghjhp q qE QQQ111W+//   ! ! # # * *2: 6 6 z%_parse_ply_gaussian..s#a&-- ++rr?rKrLscale_0scale_1scale_2rU{Gz?rot_0rot_1rot_2rot_3)r$rrrrXrGf_dc_0f_dc_1f_dc_2c3DK|]}|d|VdS)rWN) startswith)r4rs rr5z&_parse_ply_gaussian..s3CCQ1<< +B+BCqCCCCCCrcRt|ddS)N_rE)r&split)ss rrz%_parse_ply_gaussian..s%SVWXW^W^_bWcWcdfWgShShr)keyc&g|] }|SrRrR)r4rrs rrTz'_parse_ply_gaussian..s!---1!!A$$---rredgreenbluer3)findrndecoder8 splitlinesr r&append _PLY_DTYPESrl frombufferrbnamesrexpfullrmrtilearrayonessortedr)rqrprxrzeros)rendrbodyrprops in_vertexlineprrrrrrXdcrestrkkrrrs @@r_parse_ply_gaussianr+s ))M " "C QwwHIII $3$Z  w 2 2F ]## #Dd4!8m$//AAQ6DU)5E!!## : : JJLL   Q48  !(> > >jqtjjjkk k Q49  !(I "AaD  qTZ  I tv~~ !^___ LL!A$k!A$&7 78 9 9 9 -bhuooU4 H H HC IOE++++A A (AAcFFAAcFFAAcFF+Q / /CErx9qq||QQy\\ JANNOOAbj11%11W::qqzz11W::qqzz"RTUVVWWgbh|||RZ881a&AA5>%5G5GcS2611Y<<-00011RWUVXZXbMcMcG5 Xqq{{AAhKK8=q A ACC%CCCIhIhiii  -------q11Aq1$B !QQ''11!Q::AAAAtQQQJ 3Q77BBAAAtQQQJBB % 28QQuXXqqzz11V99$EqIIEQ R R Xq!Qi , , sGR ''rc bt|dzdkrtdtj|tjgd}|dtjdz }t|dtjdz dz }|d tj|d tj||ddd ft|dddd ffS) Nrrz8File3DToSplat: .splat size is not a multiple of 32 bytes))rrrU)rrrU)rrr6)quatrr6rr3r-g`@rrrU) r8rnrlrrbrkrmrcopyr)rrrrs r_parse_splat_gaussianr/ s 4yy2~STTT -bh(N(N(NOO P PC v;  bj ) )E 1D c&k((44u<E F FC J  bj ) )3w<+>+>rz+J+JC AJOO  }T!!!RaR%[99 ;;rc  '|ddkr!td|dd|dtjd|dd}tjd|dd}|tvrtd |t|\}}}}}}g'd |d zz} t |D]} d | d zz} tjd|| dzd} tjd|| dzd} tjd|| d zd}tjd|| d zd}tjd|| dzd}tjd|| dzd}tjd|| dzdp|}tjd|| dzd}tjd|| dzd}t tjd|| dzdd}||z|z|z||zz}|dz}||z|z}| |z}| dkr|dkrdnd\}}d|dfd|dfd|dfdg}|r(|d|dkrdn |dkrdnd|ftj |tj || |}|d  tj d!z }t|d tj }|d tj } |dkr'|d tj }!ntj |d|dz| |zd"d}"tj| tj}#||z}$t#|$| }%tj|%|z|#d|%<| |$krRtj |d#|| }&tj|tj|z|&d| |$z |#|$d<|d tj |z |d$z |z z|"|#z}!'|!| ||dddft+|ddddff| || z|zz } 'std%t-'fd&t d'DS)(Nrz+File3DToSplat: unsupported .ksplat version .r$rr6rrz1File3DToSplat: invalid .ksplat compression level rrrrz..Ns 5 5 5!1 5 5 5rN)rlrx)r4r!partss @rr5z)_parse_ksplat_gaussian..NsCII1 5 5 5 5u 5 5 566IIIIIIr)rnr unpack_from_KSPLAT_COMPRESSIONrt_KSPLAT_SH_COMPONENTSgetrrlrrbrkrmrrqruint64rJarangerepeatr.rr9)(r max_sectionslevelbcbsbrbcolbshc default_rangebaser socntsec_max bucket_size bucket_count block_size bucket_store scale_range full_bucketspartial_buckets sh_componentsbytes_per_splat meta_bytes buckets_store data_basectftfieldsrcolfrrrbucketsidx full_splatsnflengthsr8s( @r_parse_ksplat_gaussianrbs7 Aw!||ZtAwZZQUVWQXZZ[[[%dD!44Q7L  tT2 . .q 1E '''TUTTUUU,?,F)BBdM E ,% %D < ':': AH_ tR!V44Q7$T4a88;(tR!V<>qA 'dBG<>qA (tR"W==a@QM )$b2g>>qA  ,T4bAA!D-11&2DT4QSVXQX2Y2YZ[2\^_`` r'B,- 0DD$q( $|3j@ =( 77',zz^^~FBQ''2q)9E2q>K]^F o teqjjUUQR uuX\_lmnnn-bhv&6&6c)TTTCw<&&rz22U:DSZ..rz::;;CL'' 33Ezz(m**2:66-eX>XY Z Z Z ')M99 A?@@@ IIIIaIII I IIrc  tj|}tjd|ddtkrt dtjd|dd}tjd|dd}|d}d}|dkrUt j|d |d z| t j  |d }||d zz }n|d vrt j|t j |dz|  |d d t j }|ddz|ddzz|dz}t j |dz|dz |}|d|zz t j }||dzz }nt d|t j|t j || t j dz } ||z }t j|t j |d z|  |d t j } ||d zz }| dz dz tzdz} t j|t j |d z|  |d t j } ||d zz }t j| dz dz } |d kr.t j|t j |dz|  |dt j }|dddf|dddfdzz|dddfdzz|ddd fdzz}|dz d z}t j|dft j }|t j|t j}}dD]}||k}|dzt j}|dz dz}t j ||dz |}dt'jdz |dz z}t j |dk| |}|||||f<|t j |||zd z }t jt jd|z dd|t j||f<t/t j|ddd f|dddf|dddf|dddfgd}nt j|t j |d z|  |d t j }|d!z dz }t jt jd|dzdz dd}t/t j|dddf|gd}|| || t7| fS)"Nrrz'File3DToSplat: invalid .spz (bad magic)r6rrr2r$r2rUrr.)r1rUr.r1.r$.rir<z(File3DToSplat: unsupported .spz version r3rrrr1r)rUr1r$rirrGr/r)r decompressrr:rrnrlrrkrmrqrr>rrrr r.rmathsqrtrrr?rrsumrxr)rrversionr frac_bitsoffrbvrcbrsbrqbcombinedlargestr@ remainingsumsqcompactivevaluesignvalrxqws r_parse_spz_gaussianrQs /$  C $Q''*j88BCCC sA..q1G4a((+ABI C!||mCa!eC@@@GG SS[[\]_`aa q1u  F   M#rxq1uS A A A I I!QPQ R R Y YZ\Zb c c vY"_6a 01V9 < HQ\1y=! 4 4AN#++BJ77 q1u MGMMNNN M#rxq = = = D DRZ P PSX XE1HC sBHAE# > > > F Fq! L L S STVT^ _ _B1q5LC : / /# 5C sBHAE# > > > F Fq! L L S STVT^ _ _B1q5LC F29t# $ $E!|| ]3Ac B B B J J1a P P W WXZX` a aaaad8r!!!Q$x1}-AAAqDR@Bqqq!tHPRNSr>Q& HaVRZ ( (#==??BHQ ,C,C5   6 6DW_F&..rz::ENa'Db)DDI1%%%-8C(419sdC00C!&kAfdlO RXfcCi55 5EE#%7273;4+H+H#I#I")A,,  1QQQT7AaaadGQqqq!tWa1g"FJJKK ]3Ac B B B J J1a P P W WXZXb c c %Z#  GBGC27--"2"22At<< = =111d7R(8!<<== sE=#5#5 55r)plysplatksplatspzc|dddkrdS|dddkrdSt|dkr|ddkr|ddkrd St|d zdkrd Std ) NrUsplyrr1srrr$rrrz=File3DToSplat: could not determine splat format from contents)r8rn)rs r_detect_splat_formatrs BQBx6u BQBx;u 4yyA~~$q'Q,,47a<<x 4yy2~w T U UUrct||}fd}||j|d|f}||j|d|dddftzdzdd}||j|d|fd}||j|d|f}||j|d|f} ||||| fS)NcF|tjS)Ndevicerb)torrm)r`rs rrz _gaussian_item..s144vU]4;;rrrr$rE) r*r(rrclampr~rqr|r}) r r!rr!rrrrXrrs ` r_gaussian_itemrs Aq//C ; ; ; ;B "Q[DSD ! " "C 2ad1dsdAqqq=! " "S (3 . 5 5a ; ;CbQW%&&..r22G Bqx4C4 ! !E "Q[DSD ! " "C WeS ((rc||dddz }|d\}}}}tjdd||z||zzzz d||z||zz zd||z||zzzd||z||zzzdd||z||zzzz d||z||zz zd||z||zz zd||z||zzzdd||z||zzzz g ddddS) NrETdimkeepdimrr$r1rrU)rrunbindrrrq)r@r~r?rKrLs r _quat_to_matrs4 AFFr4F ( ( 2 25 9 99A"JAq!Q ; AQQ a!ea!em!4a1q51q5=6I QUQU]Qa!ea!em!44a1q51q5=6I QUQU]Q!a%!a%-0!a1q51q5=6I2I     wr1a  !rcF|d\}}}}|d\}}}} tj||z||zz ||zz || zz ||z||zz|| zz||zz ||z|| zz ||zz||zz|| z||zz||zz ||zzgdS)NrEr)rrr) r`roawaxayazbwbxbybzs r _quat_mulrsXXb\\NBBXXb\\NBB ; R"r'BG#b2g- R"r'BG#b2g- R"r'BG#b2g- R"r'BG#b2g-      rctjtj|||gdz tjtj|||gdz }}t j|d|dddgtj}t j|dd|ddgtj}t j|ddd|dgtj}tt|||S)Nr5rr/rar$r1)rlcosradianssinrtensorrmr)rxryrzrr qxqyqzs r_euler_to_quatrs 6"*b"b\**S0 1 126"*b"b\:R:RUX:X3Y3YqA qtQqT3,EM B B BB qtS!A$,EM B B BB qtS#qt,EM B B BB Yr2&& + ++rc v|d|d|d}}}|d|d}}|d|d}}|d|d } }tjd |z|z|zd |z|z |z d |z |z|z d |z |z |zgd } tjtj| d ||z ||z || z gd tj||z | d || z||zgd tj||z || z| d||zgd tj|| z ||z||z| dgd gd} | d } tj| d| d| jdzddddf} | | d ddz S)N).rr).r$r$).r1r1).r1r$).r$r1).rr1).r1r).r$r).rr$r$rErfrerd).rU.NN)r$r6.rTrr)rrargmaxgatherexpandr)rr)mm00m11m22m21m12m02m20m10m01q2candselr@s r _mat_to_quatrsiL!I,) cC|Qy\C|Qy\C|Qy\C a#gmc)1s7S=3+>#gmc)1s7S=3+>@AC E EB ; RZsC#IsSyA2FF S3Y6 C#IsSyA2FF S3Yc 2f:sSyA2FF S3Yc 39bjA2FF     D ))B--C T2s?3::39v;MNNOOPSUVXYXYXYPYZA qvv"dv++55e<< <.s(e&r';A'>'>??EEGGrr$rE) get_bytesrlower_GAUSSIAN_PARSERSr=rr SPLATrqr r) rrrfmtparserrrrrXrtrs rrzFile3DToSplat.executes!!##$"++--"&&s++\/@AUVZA[A[/\'-vd||$UC" G G AcFF4L AeHHTN AcFF4L AgJJ  q"a ( ( AbEE$K   }U###rN) rrrrrr rr rrrRrrrrs_  [ &$u|$ $$$[$$$rrctj|tj|}}tj|tj|tj|tj|f\}}}}t j|d|ggd| d|gg|} t jgdd|| gd||gg|} | | zS)Nrrr$rrr$rr)rirrrrr) yaw_deg pitch_degrrKr&cysycpspRyRxs r_view_matrix_tr s < $,y"9"9qAXa[[$(1++tx{{DHQKKGNBB Ar{IIIQ|..sh%,aeeCoo 6 6quuS#9O9O8ORWXYX]X]^acfXgXgRhRhQhiruvvvrcXtj|d|d |d gSNrr$r1rrrps rrz_camera_basis..0s'5;!qteadU344rcV||dz S)NrH)rrrs rrz_camera_basis..1s!!affhh((...r quaternionr~rGr?r/rKrLrrr$r1r) r=rrrrridegreesatan2asinmaxrJr) camera_inforrtgtreyer mvrr@qwxyzRrightupfwdyawpitchWs ` r _camera_basisr&(s* z2.. "0M0MCvvvvA!C&&!!C&&C 4 4B..A %%A+ eAEE#sOO44eAEE#sOO6L6L#AEE#sOO44eAEE#sOO6L6LNVY[[[ t % %a (""Qqqq!tW++ Qrr1QQQT7(||__aAaaadG8 ooFE2s** !FSL//C ,tz5Q==.%A--@@ A AC L3tSeCFmm-D-D#E#EFF G GEsE3''A !adAaD ((rcR||z }||dz }tjgd|}||dkrtjgd|}tj||}||dz }tj||}tj|||gd}t|ddS) NrHr/rGr/r+?)r/r/rGr$rr) rrrrdotabsrcrossrr)rr rrLup0r?rKrs r_lookat_quat_wxyzr.Bs6A AFFHH  t $ $$A ,s 3 3 3CuuSzz~~%l???3777 3""A AFFHH  t $ $$A 1a  A Q1I1%%%A $  ##rrG perspectiver/c tj|tj|}tj|tj|}t|||} |rzt j|} tjt j| dz ddt j| dz g|} t| d| dd} d} | || |t| dt| dt| dt| dd t|t|t|d S) Nrbrr1r/rrct|dt|dt|ddSNrr$r1)r?rKrLrrs rrz%_lookat_camera_info..Y.%!++E!A$KKeAaDkkJJrr$rUr?rKrLr~rr rfov cameraTypezoom) r as_tensorrmr.rirrrrrrstr) rr r8rr: camera_typerollrrr@r`rrs r_lookat_camera_infor?Ps2 /(%- D D DC /& c B B BC#sC((A , L   \48AE??Cdhq1uooFs S S S agr$x ( ( + J JCCCCHH %ad %!++E!A$KKV[\]^_\`VaVabb::S-=-=uT{{ T TTrc tj|tj|}d|D\}}} } tj| ||| gtj|} | | dz } t | dd} || dddfz } d}|||| t| dt| dt| dt| dd t|t|t|d S) Nr1c34K|]}t|VdSrr4)r4rs rr5z$_quat_camera_info..bs(221eAhh222222rrHrr1ct|dt|dt|ddSr3r4rs rrz#_quat_camera_info..gr5rr$rUr6r7) rr;rmrrrrrr<)r quat_xyzwr8rr:r=rrrrqwrrrrs r_quat_camera_inforE_s+ /(%- D D DC22 222NBB L"b"b)s K K KE EJJLL**400 0EU4[!!!$A !!!Q$-C J JCCCCHH %eAheE!HooERWXYRZOObghmnohpbqbqrr::S-=-=uT{{ T TTrctj|tj|}}tj|tj|tj|tj|f\}} } } t j| | z| | |zg|} d} t | ||| zz | |||S)NrcXtj|d|d |d gSrrrs rrz$_orbit_camera_info..ss'%+qtadUQqTE233r)rirrrrrr?)r#r$distancer8 pivot_splatrrKr&rrrr fwd_splatrs r_orbit_camera_inforKms <  dl511qAXa[[$(1++tx{{DHQKKGNBB rcBhBG4SAAAI33A qqx)/C!CDDaa nnVY[^ _ __rc ^tj|}tj|tj|}}fd}||di||di}}t j|d|ggd| d|gg} || ||z zz} |dpi} t jt| dd t| d dt| d dt| d dg} t jtj|d z dtj|d z dg} t| d| dd}d}i||| t|dt|d t|dt|dddS)Nc tjt|ddt|ddt|ddgSr r r s rrz(_orbit_camera_info_yaw..{sb%,aeeCoo 6 6aeeCoo8N8NPUVWV[V[\_adVeVePfPfgpstttrrr r/r(rrr~rGr?rKrLr1rct|dt|dt|ddSr3r4)rs rrz(_orbit_camera_info_yaw..r5rr$rUr6)rr) rirrrr=rrrr)r angle_degrr`casarprrrnew_posr@qcurrqnrs ` r_orbit_camera_info_yawrUws  YA Xa[[$(1++BttttAqR001111[__Xr5R5R3S3SC C}oooS"~Fs S S SBB#)$$G %%+A <quuS#//quuS#1G1GquuS#//quuS#1G1GIQT V V VD txATXa!e__cB3 O O OB 2d8T$Z ( ( +B J JC hk hss7|| %be 5A<tj/|>|>d%&\}?}@|<0|@d|?dftj1|9d'z|;2tg|:dz (}A|j d}BtQt)d)tSd|Bd*z}Ctg|:}Dtj4|}Etj5d|B|Cdz67}Ftj8|Btj7+}Gtj!|B|G|E<tj9|F|Gd,-dz 2d|Cdz }H|H|Dz|Az}Itj4|I}E|I|E}I|1|E6|2|E6}K}J|&|E|'|E|(|E}(}'}&|)|E|*|E|+|E}+}*})|'dz|(dz|*dz}N}M}L|,|E|-|E|.|E|%|Ef\},}-}.}%||E||E}}|r||End}O|r|/|End}Pr||E||E||Efnd.\}Q}R}S|J|K|&|)|+|L|M|N|g }Ttj |Tr|(|*|Q|R|Sgn|,|-|.|%gztj:tj$dtj7+|Idd|Iddk;<dzg}U|I|U}V|U=|Bgz}W|V|Dz=}Xtj>|V|Dd/0=}Yd1t|CD}Zttg|XD];}[|Z|Y|[0|W|[|W|[dz|X|[f<fd2}\| d k}]tj$|df}^tj@|f}_tj$|f}`tj$|f}atj$|df}b|]rtj$|fnd}c|rtj$|fnd}d|rtj$|dfnd}e|rtj$|fnd}f|rtj$|dfnd}gd}ht|CD]}#|Z|#}i|is|`A|aA|bA|]r|cA|r|fA|r|gA|iD]\}j}k}l|<|l\}m}ntSd3d4|mj dz}ot|j|k|oD]}pt)|p|oz|k}q|\|p|q|m|n\}r}s|rd|sd}t}r|`Bd|r|t|aBd|rtjC|s  d|]r|sD| n|s}u|bBd|r|udddddf||p|qdddfzdd|]r*|cBd|r|ud|r7|fBd|r|s|O|p|qdfzd|rH|gBd|r|sdddddf|P|p|qdddfzdddtjE|a z }v|_|vz}w|]r|cn|`}x|^F|wdddf|b|x&ddddfz |s|r_|`&d}y|r|dF|w|f|yz |r-|eF|wdddf|g|ydddfz |_Gd|vz |#d5zd6kr6t|w)dkr|hdz }h|hdkrnd}hd|_z }z|z}{ d7vr|{dknd}||r,|d|z&dz nd}}d}~|r|edHdddd}|zdd}t|d8t|d8}}||&dz dHddd}||%dd,9&dz }~ dkrtj$}t||Krz|}|||}||)}q}ptj#|||q|}z |q|pz &dz dd|}|dddddfLd}ni d:kr>|~| gd;zdzdzdd}|||dddddfz}n%|^d} dkrr| gd<}||%z }d|~|z dzdzdd}|d=d>|z|zzdddddfz}nZ| dkrTt| }|~dddddf dd}|dd?|zz d?|z|zzdddddfz}|F|_d|}|rt|}|ddN|{ddNfS)@NcFtj|tjS)Nr1)rr;rm)r`rs rrz"_render_gaussian..s%/!5=EEErrr$rEr4depth)normalclaycdkrdntjd}|tjfS)Nr4rUrr)rrr r)imgbg_trheightidevidtype render_stylewidths rbackground_onlyz)_render_gaussian..background_onlysk/;w/F/Fdkk&%+++EKX^`eghqtLuLuLuvvdF##U[tSY%Z%Z%ZZZrr8A@r:rGr9rforthorc3(K|] }|V dSrrR)r4r`keeps rr5z#_render_gaussian..s'+d+dAdG+d+d+d+d+d+drrdg?r1r)dim1dim2MbP?rUrrrFr@rHrr2gףp= ?c g|] }|k| SrRrR)r4LKs rrTz$_render_gaussian..s 0 0 0A!a%%a%%%r)r2@rijindexinggUUUUUU?)rrzr1T)r NNNfloor) rounding_modecg|]}gSrRrR)r4r s rrTz$_render_gaussian..s''''''rc  -dd||dfd}|dd\ }}}}} } } } } ||dddfz}||dddfz}|dk|/kz|dkz|+kz},r |dd\}}}}}|(z .z |z }|)z .z |z }| }| |z}tj| |z||| |}tj||z|||tj| |z||}tj|||||}tj|||z| ddd}n|dd\}}} }!|(z *z |)z *z }#}"tj| |"tj| ||"}$|$|#tj| ||#}$|$| |"z|#}$tj| |"||#|}%tj|!|%|%z|$ddd}| tjd|zz|zdd}&| d+dz /z| d/dz z}'|'|&fS) NrrrrE)rzrWr)r$) rraddcmuladdcmul_addcdivr clamp_min_rclamp_longr)0lohioxoycolscxr_cyr_a00a11a22b01b02b12opapxpyvalidc02c12mxmymzrrra22rzinxrSrdsrr@su0su1su2musdxdydsiddsimurr^cx0cy0rrhis_orthopstackr rls0 rrz_render_gaussian..splats aaaBn%,,Q//8D 1R55D'0:od###dG(J"44M8IA8MK[[[[[[[[[[[ y|q   "/ S"A"ACC UB$a((A 9 C  q)).Q / / 74C --4 5 5D;??<4455;;==HHQQHBJBB 9D+d+d+d+db"b#wX]_b=c+d+d+d(BBWeS x{a   voc4(( UF  a 48DL,=,=,A#B#BBTIAqy&1*HC 4<$$ $BELLNN111dAAA:..",,q!2D2DDG))r);;@@DDtKQQQPTVZ][^c^ghiru^v^v^vvvG b"b\2 & &B   ' " "B AAAqqq$J AAAq )D4i__R Fqqq!QwKAAAq!GbAqkcCqqq!QwKAAAq!GbAqkcC++b//E5%<bhqk#6665<<;K;KKLEKAAAqsF as;;; RXa[!Qs 3 3 3B E v|))++55d;;<< <q2vsQV|B111a7 111a7 Rxq2v}$cAFTM&9B#$t8aR"Wt{{}}-D 111a7 R1a[#$t8aR"Wt{{}}-D 111a7 R1a[ <",,q!,, ,D111a7mT!!!Q']DAqM!qA1umQ# 5577!a%?JJ1MMRRTTTG 7$$T**//11 1F CBE6**b0113q$)IfV[D\D\DaDaDcDc:d:d3e3e f fggA1 0 0 0 0 0 0A3 6F|F3emDDDH E 77lA2q1uS FFFS4888B bjjnnbjjnn56666 _Vy18 < < C CF VW C X XF  A Sc!Q$Y'' ( ( ) )B VB M"  E ^Aq"q& 5 5 5 ; ; = = B B D DF ;q 3 7 7 7D,q---DK!&$d;;;a?GG2PQ6RRG B, C M#  E e*C%y  "U)//"3"3CJE CJcCJE CJcCQwaq$D"',e eElFSXM"YE5%5>3u:SG" ,2e99D% /CJJ4E?G_2e9biE;;M_E5%3S#tT4 AF [h#qCeUE#B#BUZ\achjpTqr s sFY AUZDDDs122wRUVYWYVYRZGZFcFcFeFeFmFmFoForsFst u uF VB ]]__s "FBw  GyRw777>>@@H''U2YY'''I 3w<< NN(1+%%vay&Q-&LMMMM< sNE ;ay - - -D Jws + + +E K , , ,Ek4'#...G ;ay - - -D/4 >5;ws + + + +$D/9 C5;ws + + + +tD1< F5;ay - - - -$D0: DEK , , , ,E2= GEKq # . . . .4E EBii//}       JJLLL   KKMMM   KKMMM" i iND$2YFBT:!455BD$++ i ib$''"U2r2r22 U++b//5==+<+>6;;2t;<<FFtLLLw Kc 2 2 2    at_((**IdO,?,?,A,AB DBNrBw6I6I$6O6O#O"V"VWXZ["\"\^_``A111d m""65!44  ! !,,,---3c9@@AFFDAAAt$$ll65!,, 6 ! !%%%&&BbggiiB2**2...4;;AqAABr B.111d ;;CC  " "'((AaaaAg&&--a33ES1WsQw6111d CCCll5==::GDD  '!#&&C 99Q??  dF + +TZZ1-=-=-@-@v-N-N NNrcNeZdZedZe ddejfdZdS) RenderSplatctjddgdddtjdtjddd d d tjd dd d d tjdddddtjdddddddtjdddddd !tjd"d#d#d$ddd%tjd&d#d#dd'dd(tjd)gd*d+,tjd-d./tjd0dd12tj d3dd42g tj d56tj d76g8S)9Nr7z Render Splat)zsplat to imagez render splatzgaussian turntablerabRender a gaussian splat as an image with an anisotropic EWA rasterizer (oriented elliptical splats, antialiased, depth-sorted front-to-back). The camera comes from a camera_info input (Load / Preview 3D, or a Create Camera Info node); leave it empty to auto-frame the splat. Set frames greater than 1 for a turntable batch of images to feed a Video node.rrlrryrrdefaultrJrsteprhframesr$iz-1, 0, 1 = single still image; >1 = turntable, the camera orbits over a full 360 turn (works with any camera_info). Negative value orbits the other way.r:rJrrrrGg?g@g?TzhMultiplier on each splat's projected footprint (lower = crisper points, higher = softer/fuller surface).r:rJrr;advancedrrr5 @rzSharpen overlapping splats: 1.0 = physically-correct blend; higher biases each pixel toward its dominant (nearest) splat for crisper texture, without shrinking splats or opening gaps. Non-physical above 1.r:rJrr;rrr/ruzDiffuse shading from a light at the camera (headlight), using the splat surfel normals: darkens surfaces that turn away from view to reveal form/curvature. 0 = flat albedo, 1 = strongest shading.opacity_thresholdrz@Cull gaussians with opacity below this (removes faint floaters).rk)r4rdrbrczrWhat the image output shows: color, clay (neutral-albedo shaded), depth (near=bright), normal (OpenGL normal map).r backgroundz#000000)r:bg_imagezOptional background plate composited behind the splat (overrides the solid background colour). Resized to the render size; a batch is used per frame, a single image for all. color/clay only.)optionalrrzCamera to render from - a Load3D / Preview3D camera or a Create Camera Info node. If empty, the splat is auto-framed from a default 3/4 view.imagermaskr) r rrrIntFloatrColorImage Load3DCamerarMaskrs rrzRenderSplat.define_schemasy!'SSS@ w'' Wd1 MM XtA NN Xqd&stt}csRVae(JKKy#3Cc(abb2CScX\gk(QRR2CScX\gk'ikk~7[7[7[(Z[[|Y??zD(RSS%%md/w&xx9@X__'_::BGNNX^N<_<_`S* * * * rNr"c t| } d}| m| dddfdd}tj|||dd}|dddd}t t|pd}|dkrdnd }gg}}tj }|j j d|z}|dkrtj |nd}d}t|j j dD]}t|||\}}}}}|dkr1||k}||||||||||f\}}}}}| }||j dr|dnt!jd| } |j dr:t%|| z d d d nt!jd | }!t-|!t/jt/jddz dzz }"t5dd|"d| |}t|D]}#|dkr|nt7||dz|#z|z |}$||||j dzn| }%t9|||||||||%|$|||  \}&}'||&||'|dz }||dt?j t!j!|t!j!|S)N.rUrEr$bicubicdisabledrrrGrrgGz?-C6?rnr1g?>@v@)rrrk)"r:movedimrutilscommon_upscalerr+r&rrr(r) ProgressBarrtrrrr rBrrrrrirrrKrUr5rupdater rr)(rrrlrhr<rrrrCrDrkrrErbg_imgsbin_frames orbit_dirimgsmasksrtotalpbarrr!rrrXrrrqbase_camr3extentdistfrcam_frbg_krfrHs( rrzRenderSplat.executesz $ $  #(#++B22B++Bvy*UUBjjB''--a33Gs6{{##(q"QJJDDC "e'88::%a(8316u{&&u--- u,Q/00 # #A,:5!V,L,L )Cguc 1$$"3303D 3t9gdmUZ[_U`befjbk0k-S'5#"H(+ ! W!%+aPV:W:W:WZ]ZcdeZf@)S6\$7$7B$7$?$?FFPPQUVVV$|C???Vtx T0B0BQ0F'G'G#'MNOO-dD$ffUUHoo # #&.!mm((5h E@QTV@VYa@acijj8?8Kwq7=#3344QS,S#wsESY[fhlnt5?g@S)ANrizCreate Camera Info)zcamera positionzmake camera infoz orbit camerazlook at camera3dzBuild a camera_infoMode 'orbit' aims with yaw/pitch/distance around the target; 'look_at' places the camera at world position. Coordinates are the viewer's world space (right-handed,Y-up).modeorbitr#rnvrTrGr9r$rSg@Vg@V@rH@rg@@z Camera distance from the target.rBlook_at position_xg@z4Camera position in world space (right-handed, Y-up). position_y position_zrquat_xr/rrtquat_yquat_zquat_wzUCamera world-rotation quaternion (three.js: looks down local -Z). Normalized for you.zYHow to define the camera: orbit angles, an explicit position, or a position + quaternion.rtarget_xTzLook-at point (orbit pivot / aim). In orbit mode, move it to pan/translate the whole camera. Ignored in quaternion mode. Defaults to the origin.r?target_y)r:rJrr;r@target_zr>gfgf@z)Camera roll about the view axis, degrees.r8g^@z"Vertical field of view in degrees.r:Y@zNDigital zoom (focal-length multiplier). >1 zooms in without moving the camera.r=r/ orthographiczYProjection used by Render Splat: perspective (foreshortening) or orthographic (parallel).rrr) r r DynamicComborOptionrJrrMrrs rrzCreateCameraInfo.define_schemasy&-dddG%%fO**7udEX[\\w%TX[\\z3Df[_/Q'SS5 O**9|SgSY`d/e'gg|SgSY`dee|SgSY`dee 7 O**<|SgSY`d/e'gg|SgSY`dee|SgSY`deex$CV[\\x$CV[\\x$CV[\\x$CV[0G'HH :  70w1&xx2z3GVZei(kllz3GVZeijjz3GVZeijjvsEPS'RTTudS'KMMvs%d'wyy}}n6U(CDDK'P_+++GGHa1 1 1 1 rrGr/r"c tj} |d} | dkr\|d|d|dg} |d|d|d|d g} tjt | | || || S|||g} | d krt j|d t j|d }}t j|t j |t j|t j |f\}}}}|d}|||z|zz|||zz|||z|zzg} n|d|d|dg} tjt| | || |||S)Nrlrrqrrrsrtrurvrw)r:r=rmr#r$rH)r:r=r>) rrrr rrErirrrr?)rrlrxryrzr>r8r:r=rkindrr-r rKr&rrrrrs rrzCreateCameraInfo.execute&s~$5577F| <  \*D,>\@RSHNDNDNDNSD=!28T3RVdo!p!p!pqq qHh/ 7??<U ,,dl4=.I.IqA!Xa[[$(1++tx{{DHQKKONBBZ A 1r6B;.1r60A8aRTfWYkCYZHH\*D,>\@RSH}063RVdovz{{{|||rNrGr/rrRrrririsf2 2 [2 h}}lnly}}}[}}}rricJeZdZedZedejfdZdS)TransformSplatc,tjddgdddtjdtjddd d d tjd dd d d tjddd d d tjddddd tjddddd tjddddd tjddd d d tjddd d d tjddd d d g tjdgS)NrzTransform Splat)z move splatz rotate splatz scale splatzgaussian transformrzvTranslate, rotate, and scale a gaussian splat. Non-uniform scale also reshapes every individual splat, slower process.r translate_xr/gYr{rr9 translate_y translate_zrotate_xrnrTrGrotate_yrotate_zscale_xscale_yscale_zrr)r rrrrJrrs rrzTransformSplat.define_schema:s_y$*^^^bw''}cv5W[\\}cv5W[\\}cv5W[\\z3FTWXXz3FTWXXz3FTWXXy#4UQUVVy#4UQUVVy#4UQUVV X__'_::;'    rr"c |j} | j| j} } t|||| | }t |dd}t j|| | g| | }|dddf|z}t j|||g| | }| |jz|z}|| cxkr| krinnf|j |z}t| |j |j }|| dddz }n]t |j dd}|j dd }||dddddfz|dd z}||z|jz}t j|\}}|t jt j|dkd d d z}|d|j j}t1||j j}t3j||||j|jt;|dd}t=j|S)Nrrr1rETrrr6rUrrrGrr%r%) r(rrbrrrrrrr|r expand_asr}rrrqrrpreighrdetrjr)rr rr~rgetattrr r)rrrrrrrrrrrrrdtq_rotrDArr(r|r}rgs2r+lamVouts rrzTransformSplat.executeRso*ciRx8<<??sRT?UU t % %a ( L'7G4Bs K K K aaagJN L+{K@SV W W W!#IM g ( ( ( ( ( ( ( ( (\G+F!%//%/"B"BEOTTI!INNr4N$H$H$R$RSX$Y$YYIIeo55b!<<==B%%b!,,3355B111dAAA:&",,r2*>*>>Cc'AC-C\&&s++FCEK 0 0 3 3a 7sCCOTTA]]1%%**,,44U\5GHHF$Q//0EFFIk)VY$+E8T$B$BDDD}S!!!rNrrRrrrr9sW  [ ."LNM"""["""rrcJeZdZedZedejfdZdS) GetSplatCountc tjddgdddtjdgtjdtjdgtjjg S) NrzGet Splat Count)z splat countzgaussian countznumber of splatsz splat inforz5Returns the number of splats summed across the batch.rrr)rrrrrrrhidden)r rrrrrIHidden unique_idrs rrzGetSplatCount.define_schemaqs|y#*^^^OHNN7++,X__'_::V]]]88I'(    rr"ctfdtjjdD}|jjr.t j|dd|jjtj |S)Nc38K|]}t|VdSr)r*)r4r!rs rr5z(GetSplatCount.execute..s-QQAIeQ''QQQQQQrr,z splats) rkrtr(r)rrrinstancesend_progress_textr r)rrrs ` rrzGetSplatCount.executesQQQQu7LQ7O1P1PQQQQQ :  `  ! 4 45H5H5H5H#*J^ _ _ _}UE***rNrrRrrrrpsV   [  +r}+++[+++rrc|djdd}|dt||g|R}t|D]\}}|||d|jdf<|SNrr$)r) new_zerosr8 enumerate)itemsrtailrr!rs r _pad_stackrs{ 8>!"" D (  c%jj!3d33 4 4C%    1A{ {N Jr gaussiansc 4d|D}|std|djjd}|D]>}|jjd|kr&td|d|jjdd?td|D}ggggggf\}}}}} } t |D](} gggggf\} } }}}|D]+}t || }| |j| d|f| |j| d|f||j| d|f||j | d|f|j | d|f}|jd|krQtj || |jd||jdz |jd gd }||-|tj | |tj | |tj ||tj || tj || |d jd*t| }d}tt| dkr1tj| |djjtj }t'jt+||t+||t+||t+||t+| || S)Ncg|]}||SrrRr4r s rrTz$_merge_gaussians..s 0 0 0!-!---rz!MergeSplat: no gaussians to mergerz!MergeSplat: batch size mismatch (z vs z).c3:K|]}|jjdVdS)r1N)rr)rs rr5z#_merge_gaussians..s***! 1 ******rr$r1rrErr)rnr(r)rrtr*rr|r}r~rrrrr8setrrr>r rr)rgsror max_kpos_bscl_brot_bop_bsh_brar!pos_iscl_irot_iop_ish_ir!rrr%s r_merge_gaussiansrsk 0 0Y 0 0 0B ><=== 1a A bb ; Q 1 $ $``` HYZ[H\```aa a % **r*** * *E/12r2r2/E,E5%tW 1XX++*,b"b"*<'ueT4  AAq//C LLQW- . . . LL!TcT'* + + + LLQW- . . . KK AttG, - - -a#gBx{U""YBLL!ebhqk>QSUS[\]S^$_$_`fghhh KKOOOO UYu%%&&& UYu%%&&& UYu%%&&& EIdOO$$$ EIdOO$$$uRyq)**** G A F 3w<<1gbeo.DEKXXX ;z%++Zq-A-A:eUVCWCW$T1--z$/B/B6 S S SSrcbeZdZedZedejjdejfdZ dS) MergeSplatc :tjtjdddd}tjddgddd tjd | gtjd g S)Nrr1r)prefixrJrrz Merge Splats)z union splatzdensify gaussianz combine splatzmerge gaussianrzConcatenate any number of gaussian splats into one. Unioning several decodes of the same latent at different seeds densifies the surface, this can improve surface quality when meshing.splats)templaterr)r AutogrowTemplatePrefixrrrr)rrs rrzMergeSplat.define_schemas++BHNN7,C,CGYZ`b+ccy 'aaazK%%h%@@AX__'_::;    rrr"cd|D}|stdtjt |S)Ncg|]}||SrrR)r4rps rrTz&MergeSplat.execute..s : : :AAMaMMMrz'MergeSplat: connect at least one splat.)valuesrnr rr)rrrs rrzMergeSplat.executesN : : : : : HFGG G}-b11222rN) rrrrrr rTyperrrRrrrrsb   [  3R[-3"-333[333rrcF||dddz }|d\}}}}tjdd||z||zzzz d||z||zz zd||z||zzzd||z||zzzdd||z||zzzz d||z||zz zd||z||zz zd||z||zzzdd||z||zzzz g dddd}d |d dzz }tjd |||S) Nr$TrrrEr1rrUrGrHznij,nj,nkj->nik)rrrrrrqeinsum) rr-r@r~r?rKrLrinv_s2s r_inverse_covariancersh tyyQy--77>>>A"JAq!Q  AQQ a!ea!em!4a1q51q5=6I QUQU]Qa!ea!em!44a1q51q5=6I QUQU]Q!a%!a%-0!a1q51q5=6I2I    gb!Q  5??4((A- -F <)1fa 8 88rc  d|z} |d| z } |d| z}|| z |z d}t j|| z |z dz\}}}t||}t jd|dz|z  dt|}|dk}t j ||z|z|}t j ||z|zd || }|rt j ||z|z|| nd}|j dd}}tdt|dzD]=}||kd d}|dkrbmrWg"@rfrerd.N)"medianaminamaxrrrrrrrrr&r r)rtrr=r?rmrrrqrrrrrorrrrJrirjr)-rrXrr-rreskernelr color_sharpenchunkprogress col_dtypepadrrvoxelrrdzsinvkreqrvolcolvolwcolrdonerrrrnstgiccr^rquadwgtiiixiyizrwpcolnorms- r_splat_densityrs   C !s B !s B2g]]__s " - -d 3 3E*b2g.//4466:BBDDJBB ud + +D :cEJJrNN*U2 3 3 8 8 : : @ @CKK P PD S E +b2gl6 2 2 2C [b2qy I I IFHM W5;rBw|F) D D D DSWDilAtA 1c&kkAo & &55qy!!4!003 99;;!   lA2q1uV5=IIIk%.c3FFFKKSSTVXYZZ399;;.. 5 5BRU ]#BRBGu$++--aaaqqqj9CIECcEk!R4 ^3A< 11d2hBBD"d(#eit &<&<>D NN1dCKKOO 4 4 4+09'''cB   a9 BaaaK8H(H'Q'QRTVW'X'X'['[\e'f'f g g g G4B):):9)E)EFFF BHHJJ D#S$Q"233444) 5*%tt#..r2r::G ;;r2r " "FNN2r2q$A$A7BFFHHNNL\L\^cdi^j^j jjrcftj|dddf|dddfg}tj|dddf|dddfg}tj||j} ||}||||}}|}|d|||dd|d|||dd|d||dd|d||dd||}tj||r|S|})Nrr$r1rTr include_self)rrr?rclonescatter_reduce_equal) facesnvr`rorgpgagbnews r_connected_components_gpursN  5A;aaad ,--A 5A;aaad ,--A R ---A  qTA1Bggii AqtRdCCC AqtRdCCC Aq"f4@@@ Aq"f4@@@#h ;sA   H  rc tj||}tj||}|jd}tjt ||d\}}t |dz} ||dddf} tj|| |k} t | dkrNtj| | } t tj | | | d } ||dddf||dddf||dddf}}}tj | | d| |tj||zd}|dddfdd }tj| d ft%d |d||d d }tj| d ft%d |d||dd }d|| || z z}||| |z kd||| |zkdz}|tj|tj|| kztj| || kz}| |z} || | }|jddkr|dd|ddfStj|}tj|fdtj|}tj|jd|||<||||fS)NrrT)return_inverser$ minlengthrEr1rUinfrrz-infrrRr1)rr;r)uniquerr&rbincountrkr new_tensorrr  scatter_add_rr,rrrrallr{r?r>rirj)vertsr min_vertsrvtrZrr labelncompflabelrqfcountruv0v1v2cvolidx3cmincmaxtolenclosedinnerfaces_kusedremaps r_clean_components_gpur#s v . . .B v . . .B !B|5b"==dSSSHAu   q E 2aaad8_F >%5 1 1 1Y >D 488::%888ek$0A0A"0E0EFFMMOOPP111a4\2bAh<Bqqq!tHB{5000==a"u|OaOabdfhOiOiJiInInoqIrIrssQQQW~$$R++z5!*eEll6BBBRRSTVZ\^`fuyRzzz5!*eFmmFCCCSSTUW[]_agvzS{{d7md7m388:::DMC//44Q7744=SVCV;V:[:[\]:^:^^EJt,, 4=0I0IIJel[`ioNpNpNpt{N{| fG}Q1RaRy%)## < D Jub F C C CE,tz!}V<<>   ! !5>#5#5#7#7#=#=#?#? ??rc |_tj|s@tj|d|jzdzkrt |||St }tj|ddddgf|ddddgf|ddddgfgd}tttj t ||dddf|dddfff||fd\}|dddf}tj | |k}| dkrtj || } tj|| d } |dddf|dddf|dddf} } } tj |tjd | tj| | | }tj|tjfd t)dDd}tjfdt)dDd}d|| || z z}||| |z kd||| |zkdz}|tj|tj|| kztj|| kz}||z}|||}t |dkr dd|fStj|}tj|d tj}tjt |||<|||fS)Nrrrr$r1r)F)directedrrEzij,ij->i)weightsrcFg|]}tdd|fSr) _ndi_minimumr4r`cidxrr s rrTz%_clean_components..S1RRRAeAAAqDk5$??RRRrrUcFg|]}tdd|fSr) _ndi_maximumr&s rrTz%_clean_components..Tr(rrR)rr is_device_cpuget_free_memorysizerr8rlrxr rrrrkrrrr,r?rrtrr r{rrr>)r rr rrerrrqrrurrrrrrrrrrrr'rs` @@r_clean_componentsr/As%"8"F"Fv"N"N  " 2 26 : :R%*_q=P P P$UE9fEEE UB aaa!Qi(%Aq6 *:E!!!aV).nse K K Kr2r "R"Wq[."R"Wq[."R"Wq[.@ A K K Kr)rrrrrr$r$rrrr$r$rr$rr$r$r$r$r$rr$rUr6r9r.rFrr5rrr6r7r8r9r:rirfrerdrrrrGrEr1c||dkz|dkz|dkz|dkz}t|sdS||||||||||f\}}}}}tj|||tj|||}} tj|||gd tj|||gddSr)rrrrrr) crsolr`rorrrp2p4rs remitz_surface_nets..emitsa1f a(AF3qAv>EIIKK    FE(AeHah%#e*L1aCS!Q''S!Q)?)?B U[!RQ//000 U[!QQ//00000r)rrrmr)r;rlr r>r&rkrrrtrr+rrlerprrr8rrrr?rirjrk)*rrBroriginrorigin_trucs8any_inall_inryracoffsofffedgese0e1oe0oe1csteplocrcicvalcslrrr,r)rptslocalr vidr@r`rrrrr4s* @@@@@r _surface_netsrWbs@ &&em& 4 4CJBBvfEMJJJH Xfbj ) )28FBH+E+E FE Avva266 E\F K K K K K K KJ K K KCVc!f_s1v %A .Q 7#a& @3q6 ICPQF RF Vc!f_s1v %A .Q 7#a& @3q6 ICPQF RF vg F VZZ\\  B Qww    ' 'B <IIIyyy)))"IIIyyy)))EMS U U UD 775= ! !D L1a&1a&1a&1a&1a&1a&a&1a&1a&1a&1a&1a&JRX Z Z ZE 111a4[%1+BBxbC E CAr5!! G G 2: tQQQ& '$t* 42f:r&z2f:56emaaaed111b5kBQQQUs111b5z)9588GGR K e+ebjE-A5?SXZ]C^C^ _ _ e efgij k kjc1Y<00 C%K$$Q''%))A,,*@*@*E*EEFFFF!$SAEIc1   3q6E em,,u4= =E UB *b1fb1fb1f-rV T T TC,rVDDDCK E11111 qax261R!V8+,ADfQrT1R!V8QrAvX-. . QQQ"q&!BF( " #SAb1fHaQh)>%? QQQ"q&!BF( " #SAb1fHaQh)>%?AAA qax261R!V8+,ADfQrAvXqtQrAvX-. . QrAvXqqq!BF( " #S26111aQh)>%? QrAvXqqq!BF( " #S26111aQh)>%?AAA qax261R!V8+,ADfQrAvXqax2-. . QrAvXqax " #S261R!V8QQQ)>%? QrAvXqax " #S261R!V8QQQ)>%?AAA  99;;     % %bj 1 159UA3F3F3J3J3L3L3R3R3T3T3[3[\^\d3e3e eerrzc8tj||\}}|tj}|dd|ddzdz}tj|}tj||z}|d|z }|tj|dk|dz }|d|z tj|dk|dz } ||z|| z dzz} d| |dk|dkz<t |ttj| S) N)binsrEr$rrrGr1r) rl histogramrkrcumsumrrr&r) rrYhistrJcentersr~rwfmbmfvar_bs r _otsu_levelrbs ,vD111KD% ;;rz " "DSbSzE!""I%,G $A 4'> " "B 2B bhq1ua%% %B R&2+"q&"c22 2B Fb2g!^ #E"&E16bAg  RYu--../ 0 00rr(\c|dks&t|dkst|dkr|St|}tj|ddddgf|ddddgf|ddddgfgd}tj||dddddfgd}ttjt|tj|dddf|dddfff||f}d|jdd<tjtj | d dd tjdddf}| tj} tt|D],} ||fD]%} | tj| || z|z | z zz} &-tj| S)Nrr$r1rEr!rG)r8rlrxrrrmtocsrrrrrrkravelrkrtr&r) r ritersrrrr.adjdegrpr facs r_taubin_smoothrks zzSZZ1__E a UB aaa!Qi(%Aq6 *:E!!!aV).sampsydd6ll""1aA..t4::<< H  + +CJU]mq + r rAAAq!QQQrrrH) r)rr;rmrrrrrirj)rrr rBrrrrrr orgrr-r rvnumdenrus ` @r_sample_vertex_colours_gpurzsIJBB vU] C C CB /&u} E E EC s(e B <R V5= I I ID bD1H'',,,s2A ;!!!Q$111a4!AAAqD'2B 7 78H ID       $v,,C $wy! " "C #--%% % ( , , . . 4 4 6 66rc 8 %&'| | nd't||} |j|d| f| tj} |j|d| f| tj} |j|d| f| tj}|j|d| fd| tj}|j |d| dddf| tjtzdz dd}||k}| || |||||||f\} } }}}| j ddkrdSt| || ||||| |'fd \}}}}}tj|  o:tj| d |zd zk}|r?||}}dx%}n|t.j%|t.j}~~'d t1|t1|}}|||d zk}|dkrdSt3t5t7||z|d ||z zz|d ||z zz }| }tj| sctj| d |zd zkr(tjd}|}t;|||||\} }!~'d|dkr(t=|!dkrt?| |!|| \} }!t=| dkst=|!dkrdStA| |!|} 'd|rtC||| ||| }"nr| |z |z j"&t/j#%&fdtIdDd}#tK|&dd}$|#t/j&|$dddddfz }"'dt/j&|"dd}"t/j'|"dk|"dz |"dzdz dzt.j}"t/j(| t/j)gdt.jz} tj*| tj*|!tj*|"fS)NcdSrrR)r s rrz#_gaussian_to_mesh..sDrrrErrr$c d|zS)Ng?rR)rreps rrz#_gaussian_to_mesh..sCCPTWXPXMMr)rrr.r6皙?rtrFrig?gffffff?cFg|]}td|fddS).r$nearestrrl)r)r4r colvol_npcoordss rrTz%_gaussian_to_mesh..+s5nnn`a #q&(96QZ[[[nnnrrUrrrHrGrrrrrrra)+r*r(rrrmr|r}r~rqrrrr)rrrr+r,r=rihalfrjrkrlrrJrrbrrWr8r/rkrzrrrtrrrrrrr)(r r!rrtaubin level_bias min_component min_opacityrrrr!rrr-rXrrqrrrrBr colour_gpu colvol_cpu colnorm_cpu colnorm_npvminvmaxoccrBsn_devr rrrxryrrr~s( @@@r_gaussian_to_meshrs*((C Aq//C +a#g  ! !u} ! E EC HQW  em D DE ;q$3$w  " "& " F FDk!TcT'"**2..11u}1UUG 44C4AAA  ! !u} ! E E Kc Q X XYZ\] ^ ^C k !D%(Yd T$ZQUX[\`Xa%a"Cgs y|qt*8gudTWY\^dflGTBYBYBYBY+[+[+['C&%+99&AAAKeF\FlFlmsFtFtwx{~|E|E|G|GxGJKxKGKJ>"(**,, 0B0B0D0DK !%% JJJJLL&&((// ;; [[]]((**11"*== CIIIswwyy!!5#3#3$D cD4K C yy{{at K  1 122Z?PTW[P[H\A\]]ttd{++ - -E F  ! / / 7 7EU*-hnnnnnejklememnnnprssj& JJJBGCt,,QQQW55CHHH '#q!  C (3'>3;#+1F30N O O V VWYWa b bC  2C2C2C2:)V)V)V!V W WE  U # #U%5e%<%Ns>S>S TTrcJeZdZedZedejfdZdS) SplatToMeshctjddgdddtjdtjddd d d d tjdddddtjddddddtjdddddd tjd d!dd"d#dd$%tjd&d'dd(ddd)%tjd*dd(d+d,d- gtjd./g0S)1NrzExtract Mesh from Splat)z splat to meshzgaussian surface netsz splat surfacez mesh splatrz.Extract a coloured mesh from a gaussian splat.r resolutioniryir2zqDensity-grid resolution along the longest axis. Higher = finer surface, more VRAM/time (grows with resolution^3).rBrr9r$rzMax splat half-width in voxels. Each gaussian is rasterized over a window sized to its own 3-sigma, capped here - small surfels stay cheap, large ones aren't truncated. Raise if sparse splats leave gaps.r>smoothr<TzTaubin mesh-smoothing iterations. Smooths the surface without shrinking it (volume-preserving), unlike blurring the density. 0 = raw surface.)r:rJrr@rrBrr/r5rzIso-surface level. Auto-picked by Otsu; this biases it (1.0 = auto, lower = fatter/more-connected surface, higher = thinner/tighter).rii2zDrop connected components smaller than this many vertices (0 = keep all). Removes detached floater blobs and the inner shell of the double wall.r?rg{Gz?rGz2Ignore gaussians fainter than this before meshing.rrArzCrisp up the vertex texture: 1.0 = physically-correct blend; higher biases each voxel's colour toward its dominant gaussian instead of averaging neighbours (de-smears the texture). Colour only - geometry is unchanged.meshrr)r rrrrIrJMeshrrs rrzSplatToMesh.define_schema<sy!2dddHw'' \3BCb&QRR XqaQ&\]] XqaRD&jkkw#D(cdd _cqfSU`d&noo}dSWbf'[]]#TW(rss)2W^^^889?    rr"c tj} |jjd} dtj| zggg} } } t| D]}|zffd }t|||||||||| | }|^tj d|tj dtj dtj tj d}}}n|\}}}| || || ||dzzt!jt%| | | d S) Nrr~c |ttt|ddzzS)Nr/rG)update_absoluter&rJr)rrIraprecs rrz%SplatToMesh.execute..is=$*>*>tc#cRSUXkk[^J_J_bfJfFgFg?g*h*hrzBSplatToMesh: splat %d produced no surface; emitting an empty mesh.r1rar$T)colorsunlit)rrrr(r)rVrXrtrrrrr r>rrr rr )rrrrrrBrrrrroverts_lfaces_lcolors_lr!rqrrprrrars @@rrzSplatToMesh.execute`s'88:: O !! ${&&q4x00%'R(q 1 1A !DhhhhhhB#E1j&&%Q^`kmz}CEGHHC{ dfghhh+f--u{6/U/U/UW\WbciWjWja11a NN1    NN1    OOA     !a%4 0 0 0 0}5gwx_cdddeeerNrrRrrrr;s`! ! [! Ffmomzfff[fffrrcLeZdZedeeejfdZdS)GaussianExtensionr"cjKtttttt t tgSr)rrr7rirrrr)selfs r get_node_listzGaussianExtension.get_node_listys#}k;K^z;8 8rN) rrrrrvtyper ComfyNoderrRrrrrxsE 8T$r|*<%=888X888rrc"KtSr)rrRrrcomfy_entrypointrs   r)rGr/r/r)rGr/r4r)rz)rrc)irrririorrjrlrtyping_extensionsr scipy.ndimagerrr%rr* scipy.sparserscipy.sparse.csgraphr comfy.model_managementr comfy.utilscomfy_api.latestr r r comfy_extras.nodes_save_3dr serverrrrrrr&r*r<r9rr:rBrrrrrrrrrrrrrr;r<rrr+r/rbrrrrrrrrrrrrr&r.r?rErKrUr_r5r7rirrrrvrrrfloat16rrrr/rWrbrkrzrrrrrRrrrs   &&&&&&[[[[[[[[[[######5555556666666666??????[[[___UUUUUUU A3A5u!45AAAA   77777D#+75+7+7+7+7b  :(((((\PvtPWdPGTP8TP5RVPX^`dPP&P.4dPQSS Vu V V V V V )ek )c ) ) ) )!!!   ,,,===(NNNNNBLNNN@$$$$$$$$$$BL$$$$$$N)))4 $ $ $ T T T T T T T T```hhh$   GNOOOODWDWDWDWDWD",WDWDWDtF}F}F}F}F}r|F}F}F}R4"4"4"4"4"R\4"4"4"n+++++BL+++0$S$S$S$S$S$SN333333330 9 9 9WZaept"]/k/k/k/kd(@@@<%%%%BPfPfPff 1 1 1 1####$777(NUNUNUNUNUNUb:f:f:f:f:f",:f:f:fz88888888 1r