3j2TSSKrSSKrSSKJr SSKrSSKrSSKJr SSKJ r SSK J r S\ S\ S \ S \ S \ S \ \\ 44 S jrS\\ S\S \\ \\44Sjr\ "SSS9"SS\R&55rS\S \\\ 4SjrS\R,S\R.S \R,4SjrS\S\S\4SjrSrSrS rS!r\ "S"S#S9"S$S"\R&55rS%\R,S&\R,S'\R,S(\R,S)\R,S*\S+\S,\S-\S S4S.jrg)/N)BytesIO)io)Image) comfy_nodep0p1p2p3treturncXD-nXT-nSSUS-US*US-U--SUS-SUS-- SUS--US- U--US*SUS--SUS-- US-U---SSUS-US*US-U--SUS-SUS-- SUS--US- U--US*SUS--SUS-- US-U---S.$) Ng?xyrr)rr r r r t2t3s K/home/wildlama/comfy/ComfyUI/custom_nodes/ComfyUI-LTXVideo/sparse_tracks.py _catmull_romr s^ B B  3K3x"S'!Q& '2c7{QC[(1r#w;6C@BF G3x!bg+%BsG 3bg=C D  3K3x"S'!Q& '2c7{QC[(1r#w;6C@BF G3x!bg+%BsG 3bg=C D  control_points num_samplesc [U5S:Xa/$[U5S:Xa<USn[U5Vs/sH n[US5[US5S.PM" sn$[U5S:XaeUupE[U5Vs/sHJn[USUSUS- U-US- - -5[USUSUS- U-US- - -5S.PML sn$US/UQUSPn[U5S- n/n [U5HunXaS- - U-n [[ U 5US- 5n X- n [ X{X{S-X{S-X{S-U 5nU R [US5[US5S.5 Mw U $s snfs snf) z@Catmull-Rom spline interpolation matching the JS frontend logic.rrrrrrr)lenrangeroundminintrappend) rrp_abiptsn_segresultg_tsegl_ts r_interpolate_spliner1"s >a  >a 1 BG BTUBTQeAcFm%#-8BTUU >a ;'  (1S6QsVaf_$9[1_$MMN1S6QsVaf_$9[1_$MMN (    !  B~ B~b/A BC HqLE F ; !O$-#c(EAI&i 3Qw<1Ws7|S Q E!C&Maf >?  M'V s 'E4=AE9LTXVSparseTrackEditorzLTX Sparse Track Editor)name descriptionc `\rSrSrSr\S5r\S\S\S\S\ R4Sj5r S r g ) r2@zInteractive spline editor for drawing sparse motion tracks. Provides a canvas widget where users can draw and edit spline control points on top of a reference image. Outputs interpolated track coordinates compatible with LTXVDrawTracks. cb[R"SSS[RRSSS9[RRSSS S 9[RRS SS S 9[R RS SSSSS9/[RR S5/SS9$)Nr2Lightricks/motion_trackingzPInteractive spline editor for drawing sparse motion tracks on a reference image.imagez:Reference image displayed as the editor canvas background.)tooltip points_store[]zAJSON array of spline control points managed by the editor widget.)defaultr: coordinateszDJSON array of interpolated track coordinates produced by the editor.points_to_sampleyri'z1Number of points sampled along each spline curve.)r=r#maxr:tracksT)node_idcategoryr4inputsoutputsis_output_node)rSchemarInputStringIntOutputclss r define_schema#LTXVSparseTrackEditor.define_schemaIsyy+1(X " _   ! b   &O 0   * C" " rr;r>r?r cU(a[R"U5O/nU(a_[ U[ 5(aJ[ US[ 5(a2UVs/sHn[ Xd5PM nn[R"U5nOU(a US:waUnOSnUSR5R5S-R[R5n [R"U 5n [5n U R!U SSS9 ["R$"U R'55R)S5n [*R,"USU /0S 9$![R[4a /nGNTf=fs snf) Nrr<JPEGK)formatqualityzutf-8bg_image)ui)jsonloadsJSONDecodeError TypeError isinstancelistr1dumpscpunumpyastypenpuint8r fromarrayrsavebase64 b64encodegetvaluedecoder NodeOutput) rNr9r;r>r?splinessp interpolatedrB img_arrayimgbufimg_b64s rexecuteLTXVSparseTrackEditor.executeos* 2>djj.BG z'400Z D5Q5QPWXPW"/EPWLXZZ -F [D0 FF1X\\^))+c199"((C ooi(i VR0""3<<>299'B}}VgY(?@@!$$i0 G YsEE9E65E6rN __name__ __module__ __qualname____firstlineno____doc__ classmethodrOstrr$rrkrs__static_attributes__rrrr2r2@sb# # JAA A  A  AArrawcH[U[5(a[R"U5OUn[U[5(aI/nUH?nUR [U[5(a[R"U5OU5 MA Un/nU/nU(aUR 5n[U[5(a\[U5S:aM[US[5(a$SUS;aSUS;aUR U5 OURU5 U(aMU$)zAParse tracks from a JSON string, handling nested/wrapped formats.rrr) r]r|rYrZr^r%popr dictextend)r~parsed unwrappeditemrBstackobjs r _parse_tracksrs *3 4 4TZZ_#F&$ D   D#1F1FTZZ-D Q!F HE iik c4 SX\#a&$''C3q6McSVm c" S! % Mrratiosdevicec[R"URSSUS9nUS:*nXS-nXBUS4'SU- X#S4'US:US:*-nXS- S-nXbUS4'SX%S4'US:nXS- S-nSX'S4'SU- X'S4'U$)z^Vectorised age-ratio -> RGB [0..1] mapping on GPU. Gradient: blue -> green -> yellow -> red. rrrgUUUUUU?rrgUUUUUU?)torchzerosshape) rrcolorsm1tr1m2tr2m3tr3s r_age_color_batchrs [[a!F ;F 5B *q.C2q5MGFq5M 5.Vu_ -B :  "C2q5MFq5M %B :  "CFq5MGFq5M Mrwidthheightreference_short_sidecnX::a[X-U- 5nUnOUn[X-U- 5nX0- nXA- nX4XV4$)zACompute the higher render resolution that preserves aspect ratio.)r$)rrrrwrhscale_xscale_ys r_render_resolutionrsN  -6 7 ! ! .6 7jGkG 7 ##rr2i8LTXVDrawTrackszLTX Draw Sparse Tracksc `\rSrSrSr\S5r\S\S\S\S\ R4Sj5r S r g ) ruGPU-accelerated sparse track renderer. Renders circles at a high reference resolution and downscales with bilinear interpolation so circle sizes match the CPU version. All work — rasterisation, compositing and resize — stays on GPU. c,[R"SSS[RRSSSS9[RRSS S S S S S 9[RRSS S S S SS 9/[R R 5/S9$)Nrr8zxGPU-accelerated sparse track renderer. Rasterises circles at high resolution and downscales with bilinear interpolation.rBTzEJSON string of track coordinates (list of point lists with x/y keys).) multiliner:riri zOutput image width in pixels.)r=r#rAstepr:rzOutput image height in pixels.)rCrDr4rErF)rrHrJrIrKrrLrMs rrOLTXVDrawTracks.define_schemasyy$1N "c   ;  < 0!=! ! rrBrrr c [R"[RR5(aSOS5n[ U5nU(d<[R "SX2SU[R S9n[R"U5$[U5n[SU55n[X#[5upp[R "XxSUS9n [R "Xx[RUS 9n[U5Hkunn[R"UVs/sHnUS U -US U -/PM sn[R US 9nUXS[U524'S XS[U524'Mm S["-S-nUS-n[R$"UUS9U- n[R&"UUS S9unnUR)5R+5UR)5R+5-n[R "XU SUS9n[-U5GHn[SU[.- 5nUU- S-nU SS2UUS-24nUSS2UUS-24n[R$"US- SSU[R S9nSU[.- - n [0["[0- U --n![3U U5n"UR5SS5n#UR5S5n$U!R7S5R9US5R5S5n%U"R7S5R9USS5R5SS5n&U$R;S S9Sn'U'R<SS:XaGM@U#U'n(U%U'n)U&U'n*UR7S5R9US5R5S5n+U+U'R?S S9n,U(U,n(U)U,n)U*U,n*[AUUU(U)U*UUUX5 GM [RBRDRGURISSSS5X24SSS9RISSSS5n-U-S/SQ4n-[R"U-R 55$s snf)Ncudar`rrrdtypec38# UHn[U5v M g7f)N)r ).0r s r )LTXVDrawTracks.execute..s0AQsrr)rrrrTij)indexingrrg?)as_tuple) descendingbilinearF)sizemode align_corners.)rrr)%rrr is_availablerrhalfrrkr rAr_REF_SHORT_SIDEbool enumeratetensorfloat32 _MAX_RADIUSarangemeshgridfloatsquarer! _MAX_TRAIL _MIN_RADIUSrreshape unsqueezeexpandnonzerorargsort_rasterise_circlesnn functional interpolatepermute).rNrBrrrrblank num_tracks num_framesrrsxsypoint_xyvisr*trkr&coordsmax_dhalf_doffsetsoyoxtemplate_dist_sq render_framesr tau_minwindow active_xy active_visagesrradiirflat_xyflat_vis flat_radii flat_colorsidxr+rc flat_ages sort_orderouts. rrsLTXVDrawTracks.executes (?(?(A(AfuMv&KK6!F%**UE==' '[ 000 +E?K;;zqHkk* 6R'FAs\\589S!C&2+qv{+S9mmF '-H #c( ] #!%C:SX: (K!#!,,uV4v=4@B88:,,.1B1B1DD JB&I z"A!Q^,G[1_F Ga!eO!34IQ!a%/0J<< B6WD4*,,F;#<"FFE%ff5F''A.G!))"-H+22:rBJJ2NJ **1-44ZRHPPQSUVWK""D"1!4Cyy|q #,C3AC Aq)00R@HHLI"3//4/@Jj/C* A* A a #q!-=vub ?#Fhh!!--  ! !!Q1 - . '!Q1  #y.!}}SXXZ((u:sQ rNrurrrrrsV" " HH)SH)H)cH)bmmH)H)rframer+rrrrrHWc RURSn U S:XagURn X"-RU SS5n URS5U :*n USS2S4R 5R 5RU SS5n USS2S4R 5R 5RU SS5n[ R"XjS9RU5RSUS5n[ R"XjS9RU5RSSU5nX-RXU5nU U-RXU5nU US:-UU:-US:-UU:-nUUnUUnUU-U-R 5n[ R"X[ RS9RU SS5nURU5nUUn[ R"Xx-4SU S9nURSUUSSS 9 URXx5R 5nUS:nUUUUU'g) zStamp filled circles onto *frame* fully on-device. Uses ``scatter_reduce_`` with ``'amax'`` to resolve overlaps in painter's order (circles are expected oldest-first so the highest index = newest wins). rNrrrgamaxF)reduce include_self)rrviewrr"longrrsubrr expand_asfullscatter_reduce_)rr+rrrrrrrMrradii_sq circle_maskscxcy offsets_y offsets_xfyfxvalidflat_fyflat_fxflat_linj_mapflat_jpriority has_circles rrrJs" ! AAv ZZF ##Aq!,H#--a0HCCAuaPI U266v>CCAq%PI . 5 1B y. 5 1B B!G $Q /27 ;rAv FEiGiG! g%++-H LL ? D DQ1 ME OOE "E 5\Fzz15(D8H Q&eT}}Q"'')HQJx 34E*r) rgrYrrrarcrcomfy_api.latestPILrnodes_registryrrrr|rr^r$r1 ComfyNoder2rTensorrrrrrrrrrrrrrs  &Tt4Ed3PU:FV*J-0 $sCx.< (6OPJABLLJAQJAZstDJ/,U\\5<<ELL2 $c $3 $c $    !/GHv)R\\v)Iv)r35 <<35 35 <<35 LL 35 ll 35  35 35 35 35 35r