Jj*pdZddlZddlZddlmZejddejde dddddl Z ddl Z ddl m Z ddlmZmZmZd Zd Zd Zd Zd Zd \ZZd\ZZd Zd ZdZdZiZiZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'ddZ(ddZ)dS)aWPrototype (bench only, the package is untouched): the slicer's topology by default, repaired from the neighbouring layers, every contour fitted to the points. Per layer three candidate sets of contours, all fitted to the layer's points (ringfit.py's slab: the points within a spacing of the plane that reach it along the surface within SHIFT; x(t), y(t) with N from the point count): S the slicer's rings, refitted (the default; never loses a layer); U the contours the layer below ended with, carried up and refitted (bottom-up sweep); D the same from the layer above (top-down sweep). A contour takes the points within REACH spacings whose oriented normal agrees with its solid-outward normal (cos >= FACE: a thin wall's two faces go to their own contours: B's idea), starts from its parameters on the guess curve (projection, not an angle around a centre: A's idea), drops the points more than TRIM spacings off its fit and refits. A carried set (U or D) replaces S on a layer when every contour in it has at least MIN_PTS points, lies within FIT spacings of them (p90) and has points within COVER_D spacings along at least COVER of its length, the set explains at least as many of the layer's points (within EXPLAIN_D spacings) as S minus SLACK, and its region does not exceed the slicer's grown by GROW spacings, and it keeps every trusted slicer ring (one that fits its points as well as the carried contours must: its own carried contour of the same kind covering MATCH of it, none merging two). With gate=True a carried set is tried only on a suspicious layer: one with a slicer ring that cannot be trusted or a solid sliver narrower than THIN spacings (a C, a wall fragment). When both pass, the one explaining more points; when none, S. A sweep carries the layer's final contours on (so a repaired layer seeds the next). from hybrid import encode_hybrid NcKDTreez/home/joao/code/fcp_devel/api/)_rings) PointSlicercentres check_layersg@g333333? g?)gffffff??)r gQ?g?g@c`tj|ddtj|ddz }tj|dddf|dddf ftjtj|dddz }|t j|dk|krdndzS)zMUnit solid-outward normals at the vertices of ring R (hole: R bounds a hole).rrNT)axiskeepdims-q=)nprollc_maximumlinalgnormrf signed_area)Rholetns list of dicts (ring, hole, n, fit, cover) or None per guess.cpg|]2}tjtj|tdz3S)?)r _resamplerasarrayfloat.0rss r zfit_set..>s5 I I IaR\"*Q..q 9 9 I I Ir cXg|]'\}}tjt||(S)rfulllen)r(iDs rr*zfit_set..?s.JJJA"'#a&&!,,JJJr c4g|]\}}t||Sr,)r)r(r0hs rr*zfit_set..@s&FFFDAq\!Q''FFFr rrrrNr#?Z)ringrrfitcovernpts)#r concatenate enumeratevstackziprquery_ball_pointREACHr-r.r%FACEargminrrnonzerorMIN_PTSappendrarc_targmaxrefitqueryTRIMsumallpirr$dictr& percentilemeanCOVER_D)guessholesQNQr)denseownnrmallvtreecandwhichpjsoutr/r0selPrCrdkeepdPdCs ` rfit_setrg:s  I I I I5 I I IE .JJ59I9IJJJ K KC )FFCu4E4EFFF G GC 9U  D 4==D  EAI . .D GCFFB  E4NN2   Z^^ R2a5T) * r77 N")BINN48ad?N$K$KLLME!H C%  OO1j%1*RZq-A-A)Ba)GHII!L s88g   JJt     cF HQKKc RYsax%8%88 9x1~~1 AJJ  Q   "D1H} 88::  $A8Aq25y71::+;+;A+>+>q+AACFFJKKDAq N1   !r~a'8'81'< = =$$B$A R\!TAX.. / / 5 5a 8 8 ; QZZ  bl1cAg66 7 7 : 4QU1XbmBPR>S>SVW>W8X8X#BGB'A+,=$>$>??c!ffNNN O O O O Jr c  |rt|sdSttjt tjfd|D|dtzkS)Ngcdg|],}tjtj|dz-Sr5rr$rr%r's rr*zexplained..gs2+`+`+`UVBLAa,P,P+`+`+`r r)r.r&rrPrr=rI EXPLAIN_D)ringsrTr)s `r explainedrnds As +`+`+`+`Z_+`+`+`!a!abbhhijkklmnr{~rAA B BBr cRt|otd|DS)Nc3K|];}|duo2|dtko!|dtko|dtkVzgood..ksU{{stATMnai7.BnqxSVn[\]d[ein[n{{{{{{r )boolrL)fss rgoodrxjs+ 88 {{{xz{{{{{{r c|rtj|ng}t|||||}dt||D}d|D}|||fS)zS: the slicer's rings refitted; a ring with too few points stays as the slicer drew it. -> rings, hole flags, and the rings that fit their points well (trusted: a carried set must keep each of them).c\g|])\}}||dntj|t*S)Nr7rr%r&)r(rtrs rr*zslicer_set..ss4 Z Z Zda!-QvYYRZ5-A-A Z Z Zr c|g|]9}|duo2|dtko!|dtko|dtk:S)Nr8r9r:)rqrr TRUST_PTSrss rr*zslicer_set..tsGppphiq}c5ScQwZ55HcQvYZcMcpppr )r_holesrgr>)slrTrUr)rSrwrmtrusteds r slicer_setrnsm 'BIbMMMRE UAr1 % %B Z ZcRTVXkk Z Z ZEppmopppG %  r c t|||D]\}}}|sdStj|}tjtjtj||ddgdd}|sDdttj |zt|dz t|zkrdSdS) zThe slicer's layer looks wrong: a ring it cannot trust (too few points, a poor fit, not backed along its length) or a solid sliver (width 2 area / perimeter under THIN spacings: a wall opened into a C, a wall fragment).TNrrr3r4rF) r>rr%rrdiffr=rKabsrrmaxTHIN)rmrSrr)rr2rpers r suspiciousrxsueW--1a 44 JqMMinnRWRY1RaR5z%:%:CCC!nLLPPRR QR^A..///#c5//AD1HLL44 5r c |st|d Sfd|D}d|D}i}t|d|d|dD]\} }|s tjt j|dz  fdt |D} | sdS| D].} || d d z|| <|| d krdS/d S) zEvery trusted slicer ring is matched by a carried ring of the same kind (MATCH of its length within COVER_D spacings), and no carried ring matches two of them (a merge).rcdg|],}tjtj|dz-Srjrkr's rr*z!keeps_trusted..s1CCCr|BJqMM3733CCCr c,g|]}t|Sr,r)r(xs rr*z!keeps_trusted..s ) ) )AWQZZ ) ) )r rmrSr5cg|]U\}}|ktj|dtzktkS|VS)r)rrPrIrQMATCH)r(r/rr2rSr)rs rr*z!keeps_trusted..sZrrrTQqQ27177ST::VW=\cfg\gKgChChlqCqCqqCqCqCqr FrrT)anyr>rr$rr%r<get) rmrSSr)samplestreesusedrtrhitr/r2rs ` ` @@r keeps_trustedrs4 %q|$$$$CCCCUCCCG ) ) ) ) )E D' AgJ) ==  1b   LAa 0 0rrrrrrrYu--rrr 55  Ahhq!nnq(DGAw{{uuu  4r Fc Fid}}|D]}||\} } d} |:t| tkr!|r||drt|d|d| | |} t| rt d| Dd| D|||rd| D} ddlm}||r|||nd}|duoW|| |t|zj d t|j d zk}|s+t| d | Dt| | | } | ||<| "| d ||d tz kr| n||}|dr|d|dfnd}|S)zWOne sweep: per layer the carried set (fitted) when it passes, with its explained share.Nrrrcg|] }|d Sr7r,rss rr*zsweep..s*A*A*A1V9*A*A*Ar cg|] }|d Srr,rss rr*zsweep..sCZCZCZRSAfICZCZCZr cg|] }|d Srr,rss rr*zsweep..s///q6///r )_regiong{Gz?rcg|] }|d Srr,rss rr*zsweep..s2I2I2I1V92I2I2Ir )rmrSrnrnrmrS)r.rDrgrxr neighbourr differencebufferGROWarearrNrnSLACK)loopsslabsr)orderrgater_prevkrTrUresrwrmrreg_slgrownfinals rsweeprsDC LLa2  A' 1 14 11Q4 CU 1a$q'1b!44BBxx mM*A*Ab*A*A*ACZCZWYCZCZCZ\]^_\`bcdd m//B///------.3Ah@q***Dd*Hwwu~~/H/HW[^_W_I`I`/a/a/fimpstztBGqHqHjH0HmU2I2Ib2I2I2IU^_dfgijUkUklllCA3{+;qtK?PSX?X+X+X^_`a^b38>Kgg//t Jr cudac tj}|ffd }|t_ |dkr"ddl} | jf|||d||t_St jf|||d||t_S#|t_wxYw)zcuts: None (the package's cut rule) or "pointtest" (pointfit.py's); gate: a carried set may replace a layer only where the slicer's layer is suspicious (`suspicious`); elsewhere the slicer's rings refitted.c ptt|}|jjt t jt |jjz |jj z| |jj zt |j fd| |D}fd|dzD}itt|D]ct|g|R\}}}t!|||t#||dt%|||<dt'||t)tt|}t'||t)tt|dz dd} g} t*dddt.tt|D]fd d |fd | ffD} | rt3| d n df\} } t*| xxdz cc<t!|d|r|dnd| r| dnd| | dt.<| fd| dDt7j| |jS)Nc,g|]}fd|DS)cZg|]'}tj|tddz(SNr4r{)r(Lscs rr*z.slices...s2<<.slices..s.UUU<<<<<<<UUUr c>g|]}tj|Sr,)rslab)r(zkNXXr)s rr*z1encode_hybrid..slices..s);;;2BA&&;;;r r4r)rmrSrrnrrr)rUr0cbg|]+\}}||ddtz k'||f,S)Nrn)r)r(rorrs rr*z1encode_hybrid..slices..snDDDtq!Q]WXYdWeijklimnyiz~CjCXCXCQFXCXCXCr rr0c|ddS)Nrrnr,)rs rz/encode_hybrid..slices..s1k1Br )keyrrm)slicerrrr0rrcg|]>}tj|ddz tj?Sr)rr%astypefloat32)r(rrs rr*z1encode_hybrid..slices..s>[[[ABQB/77 CC[[[r )r r posescalerr&rr%centerrotation world_normalsspacingrranger.rrNrnrrlistSTATSupdateDEBUGclearrrErslicesdevice)selflayerscrrrmrSrrr0newoptsnamepickrrrrr)rrars @@@@@@rrzencode_hybrid..slicessN L(( ) ) Y_ # #E * * Z5 ! !DI$4 4 8J J    ! !DI$6 6 $,  UUUUtzzRS}}UUU;;;;;;RU;;; s1vv I IA$.uQx$F%($FA$F$F$F !E5'e5'YW\^cde^fgh^iklMmMm#-eUGQ#G#GIIIAaDD %4c!ff #6#64 @ @ %4c!ffqj"b(A(A#B#BAt L L qA ### s1vv ] ]ADDDDDad c1Q4['ADDDDGK\T'B'BCCCCRUWXYZW[Q\JD$ $KKK1 KKK58qtG}QRSTQU@_!W [_/0t=ad7mmDPTU\P]___E!H JJ[[[[TRY][[[ \ \ \ \}S$+...r pointtestrN)uprr)rrpointfitencode_pointtestfcpencode) rarrrcutsrkworigrrs ` ` r encode_hybridrs  D"///////6 K" ;   OOO,8,QZ2fVZZWYZZ" z!G6&GGBGG! T !!!!sA. A..A<)F)NrrNF)*__doc__sysnumpyr scipy.spatialrpathinsert__file__rsplitrringfitrr fcp._slicerr r r@rArJrDrqrrrQrlrrr}rrrrrrgrnrxrrrrrr,r rrs, !!!!!!23338??3**1-... 9999999999     w 5       >>>'''TBBB |||!!!   ,*("("("("("("r