`jcdZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl ZdZddZddZdd Zdd Zdd Zdd Zdd Zd dZd!dZd"dZd#d$dZd%dZd&dZd'dZd(dZedfd)dZd*dZ e!dk(re yy)+a{Prepare topology-faithful, multi-resolution graphs from a triangle CAD mesh. The requested resolutions are *vertex/node budgets*. Open3D's quadric error decimator takes triangle budgets, so this program deliberately records both the requested node budget and the triangle budget passed to the decimator. The resulting number of graph nodes is always measured, never inferred. ) annotationsN)Path)Any)i.ipii iii,dcVttjjdS)N)r__file__resolveparents전처리/cad_graphs.py_rootrs > ! ! # + +A ..r ctjtjj}t |t j dk(rdz Sdz S)NDarwinii)resource getrusage RUSAGE_SELF ru_maxrssfloatplatformsystem)rsss r_peak_memory_mbrsG   X11 2 < ?? ##%$$&$$&%%'! Kr ctj|jtjj }tj|j tj j }tj|jtjj }t|dddf}tj||tj||dkD}|||fS)Ndtype-q=outwhere) r!asarrayr7float64copyr8int64vertex_normalsr/divide zeros_like)rAr.fnnlens r _mesh_arraysrT=s 4== 388:A 4>>2779A 4&&bjj9>>@A a=D !D !Tr}}Q/te|DA a7Nr c X t|}|}tj|ddddgf|ddddgf|ddddgffd}tj|d}tj|ddd\}}} | dk(} tj ||dddf||dddfz ||dddf||dddfz } t | dddf} tj| | tj| | dkD } tj| | | fd} tjt|d ftj }tjj||| tjt|d tj }tjt|d tj }t|D]!\}}||dkr|||<||dks|||<#tjt|tj }|dk\}tj tj"| ||| ||zj%dd d||<| | dk7z|tj&d kDz}tj|dddf|dddff}tj|dddf|dddff}||||z }t |}tj||dddftj||dddfdkD }tj(||||||zj%dtj*|dtj*| dtj*|df}tj,||j/tj0}tj2||fj/tj0t5|tj,|| j7|j/t8tj,||j7|j/t8||j/tj0| dS)z>Build edges solely from triangle topology (never spatial KNN).Nrrr)T)r*return_inverse return_countsrFrGrDr(g>@) minlength) edge_index edge_attr boundary_node feature_nodedegree unique_edges edge_counts)r6r! concatenatesortuniquecrossr/rOrPzerosrKaddatfullrM enumeratearccosclipr-deg2rad column_stacktilebincountr#int32vstackr&ravelbool)r7facesnormalsrRtridirected undirectedkeysinversecounts boundary_u face_normalflenedge_face_normalsum_nfirstsecondhalf_iedge_idihedralboth feature_usrcdstdeltalength directionattrsr_s r _topologyrFs H A C~~s1q!f9~s1q!f9~s1q!f9~NUVWH*JIIjq]abD'61J((8C1I.#ad)1DDhsSTVWSWyFY\dehijlmimen\oFopK k "1d7 +D))K2==3MUY\aUabK~~{K&MTUV HHc$i^2:: 6EFFIIeW./ GGCIr 2E WWSY"(( 3F#G, =1 "E&M F^a #F6N - xxD 4H Q;DYYrww(8t(EHXY_`dYeHf(f'k'klm'nprtuvwHTNfk*hD9I.IJI ..$q!t*d1a4j1 2C ..$q!t*d1a4j1 2C SMHSM )E  F %4bmmE6JRXYZ\`Y`RadiRijI OOVY 1L0Q0QRS0TWWXq1277:q3I277S\^_K`b cE [[ * 1 1"(( ;Fiic +22288<7SX>T*%5%;%;%=KRRSWX DO$9$9$;qIPPQUV$++bhh*?PV  r c t|}||dddf||dddfz ||dddf||dddfz ||dddf||dddfz }}}ttj|| }|dz}tj|tj } tj j| |jtj|dz dg} d D]7\} } ||dd| f||dddfz ||dd| f||dddfz }} 9tdD]}|dd|dzdzf|dd|dzdzf} } || ||dd|fz || ||dd|fz }} t| t|z}| jtjtjtj| |zjd|tj ||d kD d dtj|tj }t#| D]-\}}tj j||dd|f|/tj$|d tj&dtj&z}tj||z | tj|| dkD }tj|dftj }t#| D]\}}tjdtj(|tj*|tj,tj(|dkD }|dd|dzdzf|dd|dzdzf}}||||z }tj j|||dddf|ztj j|||dddf |ztjt|d| ztj|| dkD }| dkD|ddk\ztj.|ztj.|z}tj0| |tj2|ddz |tj,||d |d|df} tj.| }|ddddfxx|dddfzcc<t5| |j7t8gdfS)zSFinite, intrinsic-ish local features; curvature is unsigned due to STL orientation.NrrrV?rD@rY))rrV)rVr)rrgV瞯D((1BJJ/KFFIIk1779biis A&>? F(11a4zAa1gJ&!AqD' QqAwZ(?1)(VaZ1$$%qVaZ1,<)<'=1ta!V) o%qta!V) o'=11 1 - bii 1q5++a.%R\\Z_M`hmpuhu(vxz|} ~ A  "**-I ( "&&))IqF|S*Q( XXd?+RUUAI >Fyy)+[bhhqkQ\_dQdeH ((Aq6 ,C ( iiRVVC[bmmC.@rvvVY{H[^cHcdVaZ1$$%qVaZ1,<)<'=1 aD1Q4K  #q#ag,*+  #q3q$w<-!+, ) yyC# *;!T_bgTghH"U*tH~/BCbkkRZF[[^`^i^ijr^ssO i"**T(^Q2O&OQY)4+@$~BVX\]eXfh iA KKNE !QqS&M_QW--M 1:u||D),T TTr ctj|fd}|D]2\}}|t||t|}}||k7s.||<4tt |Dchc] }|| c}Scc}w)NcJ||k7r||<|}||k7r|S)Nr )r%parents rfindz_components..finds:Qi1nvay)F1Ivay1Qi1nr )r!arangeintr6r)rRr`rr%rrrs @r _componentsrst YYq\F1CF|T#a&\1 6q6!9 q*AQ* ++*s%A;c tj|jtj|j}}t tj ||dddf||dddfz ||dddf||dddfz dz}|j t||d||jz }|j||j|}}||zdkD} d|| z || <d|| z || <|||df|dddf|||df|||dfz zz|dddf|||df|||dfz zzS)zODeterministic area-uniform triangle samples, rather than vertex-biased samples.NrrrVrT)sizereplacep) r!rJr7r8r/rechoicer6r-random) rAcountrngr.rQr%idsr1r2flips r_area_uniform_pointsrsd ::dmm $bjj&@qA288Aa1gJ1QT73QqAwZ!AadG*5LMNQSSA **SV%QUUW* EC ZZ  5 1B 7Q;D1RX:4!BtH*r$x QsAvY<"QW+1S!V9!CF) )DE E1d7 WXYZ[^`a[aYbWcfghijmopjphqfrWrHs ssr c ddl}tjjd}t |||}t |||} |j j j}|j|j j jj||j j j}|j|j j jj|t||j|jj|jtj dj#z } t||j|jj|jtj dj#z } d} d }|| || d | fS#t$$r} ddlm} | tj*|j,j/|d\} }| tj*|j,j/|d\} }d t1| j2} Yd} ~ d} ~ wwxYw) zLBidirectional, area-uniform surface sampling queried against mesh triangles.rNi2(5pointszUOpen3D RaycastingScene closest point on triangles; deterministic area-uniform samplescKDTreer(workersz2fallback nearest vertices (NOT surface distance): ct|jttj|dt|j t t |dS)N_)mean_mmp95_mmmax_mmsamples)rmeanr! percentilemaxrr6)rs rstatsz'_sample_surface_distance..statsscU1668_bmmTUWYFZ@[glmnmrmrmtguCFGJKLGMCNOOr )source_to_simplifiedsimplified_to_source)r2r!r default_rngrtgeometryRaycastingScene add_triangles TriangleMesh from_legacyr/compute_closest_pointscoreTensorr#r$numpy Exception scipy.spatialrrJr7querytype__name__) source_meshsimplified_mesh sample_countr@r src_points simp_pointstarget_starget_o d_forward d_reversemethodexcr_rs r_sample_surface_distancers6 ))   )C%k<EJ& cJK [55>>113X5K5KCEENNLgLgLsLsuDME6F55>>113X5K5KCEENNLgLgLsLstMA6BzH,K,KCHHOO\f\m\mnpnxnx\yLz,{}E-F-L-L-N NO {X-L-LSXX__]h]o]oprpzpz]{M|-}G.H.N.N.P PQ h O$))$4eT]N^ _ag gg [)rzz/*B*BCDJJ:_aJb 1rzz+*>*>?@FF{\^F_ 1Ed3iFXFXEYZ [sFG## J,BJJcddlm}||j|d\}}|jtj |jtj fS)Nrrr(r)rrrr#r!rqr$)source_verticesr7rrixs r_source_correspondencersH% O $ * *8R * @EAr 99RXX  4 44r c Dtjt|dftj}tjt|dft}tj t|tj tj}||||dfSddlm}tj|d}hd}|j|js(|||dt|t|jz fS||d j|d \} } |d | |d | } } d|jvr|d| n#tj|tj } tj || zj#d}| dk|dkDz}tj$tj | tj | | | zf}tj&|ddd|dd|dddftj(|z|dd| |dd|||dfS)aMap supplied analytic samples only when explicitly provided for this mesh. Mapping is nearest sample correspondence, with a conservative distance mask; it never silently substitutes samples from the reference CAD for an alternate ``--mesh``. Signed principal values are retained as supplied, while the discrete unsigned channels remain available independently. rYNrrF) allow_pickle> k1_per_mm k2_per_mm points_mmzignored: missing rr(rrrparameter_normalsrrg?rrznearest supplied analytic samples; valid iff distance < 3 mm and abs(normal dot parameter_normal) > 0.8; principal values stored as magnitudes, Gaussian retains sign)r!rfr6r$rtrirrrloadissubsetfilessortedsetr full_likerr-rnr"r)r?r7rvvaluesrdistancerr%requiredrrk1k2sampled_normal alignmentcloseraws r_external_curvaturersXXs8}a("** 5FrxxXXYHZ\`?auwws8}bffbjj9H |FE8T99% 5)A6H   QWW %uh*;F8CPQPWPWLCXaqQAq$c)R'S !Q$1a4!AqD'SwV[gjvHKY^ _ x"3 45_Q~EVDWXcd e "3 '):):9)Er{{SWX[G\^`^i^ijnor^suwvAvABFGJvK % CS 1399S> 01r /1rs& J 8J # J5J c 0d|Dcgc]}}|d|d|djdj|dj|djj|djdjd c}i}|d z jd t j |d zdzdd}t jjtdz |}|jd|}|dz j|dycc}w)NrJr,rr$rur[r%)budgetactualposfaceedger%zviewer_data.jszwindow.CAD_GRAPH_DATA=),:) separators;utf8encodingu5 CAD graph resolutions
zdatas_2/viewer/vendorz../../datas_2/viewer/vendorz viewer.html) roundtolistT write_textjsondumpsosr?relpathrr)rJout_dirrOpayloadhtml vendor_paths r _write_viewerrls8lr slrfg &''8%9Q~EV m))!,335qz?P?P?R((//1+@T@TUV@W@^@^@` blr stG  ++,DtzzRYfpGq,qtw,wCI+J J!D ''//%',C"CWMK <<5{ CD }(((?ssBDc ddl}tj}|jddt |}t |\}} } |j d|jdzdz } ttj|jd|j dz } | dkr tdt|j} g}t|| j| dgd}|D]Y}tj}t t| tdt|dz}|j!|}|j#|j%|j't |\}}}t)|||}t+|||\}}}tj,|j/d t1|d kDz}|ddddfxx|dddfzcc<t3|||\}}}}|| z | z } tj4| |||fj7tj8}!tj4tj,| tj:|dddfd d ||f}"gd |zgdz}#t=||\}$}%t?||\}&}'t|j}(t|})|)|kD}*|*}+d},|+rD|j!d}-|-j%tt|-j@},|,|kD}+id|d|d|)dtt|d|*ddd|+r|,ndd|+rdndd|&d|'dtC|)|ddt|dd k(jEdt|ddkDjEd t|dddfjGd!t|dddfjGd"|d#|(|(| z | z tj|z tId$}.tjJ|d%|d&z |j7tj8| j7tj8|j7tj8|!|"tjL|#d'(|d)|d*tjLgd+d,(|jNj7tjP|$|%|tjR|)tT(-|d.jW|.|jW||)| ||d)|dddf|&d/\tt||d0<tt| |d1<| |d2<d3|d4<|r|jYr t|nd|d5<tj|z |d6<tI|d7<|d8z j[t]j^|d9d:;|xs tadNrT)r exist_okrVz mesh has zero bounding-box scale) center_mmscale_mm)source normalization resolutionsrrrrYr)) pos_norm_x pos_norm_y pos_norm_znormal_xnormal_ynormal_z)analytic_k1_abs_1_per_mmanalytic_k2_abs_1_per_mm"analytic_gaussian_signed_1_per_mm2requested_node_budgetdecimator_target_trianglesractual_trianglesnode_budget_exceedednode_budget_notezYmeasured output count may differ because QEM accepts a triangle target, not a node targetconfirmed_qem_node_floor floor_notez;QEM minimum triangle request still retained this many nodessurface_distance_mmsurface_distance_method componentsr`boundary_edgesranonmanifold_edgescurvature_valid_fraction!analytic_curvature_valid_fractionanalytic_curvature_mappingsurface_area_mm2)area_change_fraction elapsed_speak_process_memory_mbgraph_z.npzzrrrallr/rrnr#r$rrrr7rr-rrsavez_compressedrJrbrqrfrtrexistsrcrdrerrQrl)/ mesh_pathcurvature_pathrhbudgets results_dirr@startedrqsource_vsource_frcenterscale source_arearJindex requestedt0 target_trisimpler.rQrrbase_x base_valid base_names normal_good external_xexternal_validexternal_distance external_noter$rrnamesoriginal_indexcorrespondence_mmrdistance_methodrrUexceededfloor floor_nodesminimumrecords/ rpreparersB!G7===#M  "FLne z$FGG//12K#%F#i.yzGzGzIW\l]np@qu    XAs9q=/A(BC 33J?**,f.Q.Q.SU[UrUrUt#F+ 1fIaF4KT);Aq$)G& Jkk&)--a0Jv4F4KL 1ac6k!T'22GZ[iklntGuD N$5}fHe# OOXvvzB C J J2:: VX)> +VWY]V]J^`ahi@jlvyG)HI ^akkoMM,B8Q,O))$>qxxPRPZPZG[fnfufuvxwAwAgBKQKXKXY[YcYcKd}BJJW\djLkx|~JyKW[\gWh/1zz;mu{0|!"BHH!5^zKCT+-88F$+G  I m##F+ )VYalm~BCO~P_efgijfj_kyAB C_` #3x=1E sSVW_S`Oa59K3LJchizc{%oE !?MR`RgRgRiN(;osE $%#002Wrs#  A/Q P:"*Z$TN ,th*5 L@B.@$etQU@F s z46r