jjHdZddlmZddlZddlZddlZddlmZddlZ ddl m Z ddl Z ddl m Z ddlm cmZddlZej$dddlmZddlmZmZmZdd lmZed z Zed z Zed z Zed z ZdZ e jBgdZ"e jBgdZ#dZ$dZ%dZ&dZ'dZ(Gdde jRZ*dZ+dZ,dZ-dZ.dZ/dZ0e1dk(re0yy)z`Prepared graph_6000 (5,925 actual nodes) weak-label diagnostic: normal baseline vs dihedral GNN.) annotationsN)Path)cKDTree)nnAgg)ROOTfixed_physical_normalizesplit_embedded_units)seed_everything dihedral_gnninput_processedmodeloutputsg>@) pos_norm_x pos_norm_y pos_norm_znormal_xnormal_ynormal_zvertex_area_mm2mean_incident_angle_radmean_curvature_abs_1_per_mm gaussian_curvature_abs_1_per_mm2 boundary_node feature_nodetopological_degreeanalytic_k1_abs_1_per_mmanalytic_k2_abs_1_per_mm"analytic_gaussian_signed_1_per_mm2cad_k1_signed_1_per_mmcad_k2_signed_1_per_mmcad_mean_H_signed_1_per_mmcad_gaussian_K_1_per_mm2cad_curvedness_1_per_mmcad_shape_index) length_mm direction_x direction_y direction_z normal_dot dihedral_rad boundary_edge feature_edgecttdz }|tjvr tjj d|ndddl}|S)Nu 전처리r)strrsyspathinsert cad_graphs)pr3s dihedral_gnn/run_experiment.pycgr6s9tK#((1B1-cP tdz }tj|d5}|jDcic]}||| }}ddddj d}|dj d}|dj d}|d j t }|d j d }|d j d} | j } | ddd fxxdzcc<| dddfxxtjzcc<t|dk7rtdt|tjtdz d} t| dj|d\} } || d| zjd}| dktj|dkDztj| d| ztj| d| z}tj|d k\dd}| d| |z}| d| |z}tj ||}tj"||}||zdz }||z}tj$||z||zzdz }dtjz tj&||z||z z}tj(||||||fj d}d ||<tj(||f}tj(|tj*|dddfddf}t,j/dd t1|}|j3|j d|t4j d!"tj6t,d#z fi|tj6t,d$z |j d|j d|j d|j d|j d|j d|| j d|j d% d&dt9|d'j:ddd(dt=|j?t9|jt9|jd)d*d+d,d-}t,d.z jAtCjD|dd/d01|||tj(tG|t4|j dfj d|| | |d'j d2|d3 Scc}w#1swYxYw)4zEOne prepared ~6k CAD graph only: requested 6,000, actual 5,925 nodes.$data_processed/graphs/graph_6000.npzF allow_pickleNpos_mmf4normalx feature_valid edge_indexi8 edge_attrrii%z+graph_6000 expected 5925 actual nodes, got z*analysis/results/cad_surface_curvature.npz points_mmworkersparameter_normalsg? k1_per_mm k2_per_mmg?gTparentsexist_okz0.8; both k values are multiplied by sign(dot) into prepared-normal convention, then ordered k1>=k2; invalid signs are maskedu[H=(k1+k2)/2; K=k1*k2; curvedness=sqrt((k1²+k2²)/2); shape-index=(2/pi)*atan2(k1+k2,k1-k2)zsign is meaningful only under this local oriented correspondence; use dihedral at sharp creases, where smooth curvature is undefined)coverage_fraction valid_count invalid_count valid_ruleformulascaveat)requested_node_budgetactual_node_counttriangle_count source_graphtopologynode_padding_or_duplicationsigned_curvature_mappingz#prepared_graph_5925_provenance.json ensure_asciiindentutf8encodingi4) posr>rawnodeedgeerawearVmeta)$rnploadfilesastypeboolcopypilen RuntimeErrorrquerysumabsisfinitewheremaximumminimumsqrtarctan2c_repeatINPUTmkdirdictupdateNAMESsavez_compressedintshapefloatmean write_textjsondumpsr )sourceakzvnr?maskrnrorpcdixdotokorientbk1k2HKCSextraaugrqs r5graphrsm  33ggf%(A0IA1Q4a0I(X;dak006A3t9LqRSTcRdRkRklpRqTwxzFxGxNxNOSxTrvZ[\gZhZoZoptZuUYy}yByByDvxEGHIJKHKELNRERELSUVWXYVYSZ\^\a\aSaSZFDL|&QRUVWRXQY$Z[[774< <%PV]^_`k^lVmVsVstu~VsWAQUQRSUGHIJK^I_`bIcGcFhFhijFkBEpqrspsvxv|v|}@vABDvDoEFHFQFQRST_R`acRdFeoefhfqfqrstr@ACrDfEoElnMOMUMUVY[\V\]_`cMdFLghitguvxgyz@g@efCDEPCQRTCUV\C\AB`b`j`jklmn`o]_sus}s}~@AsBprFHIKFKMNENCDQSTVQVOPY[Y`Y`bdegbghjkmhmbmopapYqWXuvwyw|w|u|~@~H~HIKLNINOQRTOT~UtUrs\^\a\abdeghijklmnobo\p\w\wx|\}V[IJ~CEGDG~HMOMRMRSTUZSZM[KLacafafgklnluluvxyz{yv@ A B C D lE gE aF \`T4(T!WSZZ!((4._crwr~r~@FsGZ6HIKI\I\]bc|]|ICBICDFDWDWX]^FXF^`^g^ghl^mEGENENOSETpqpxpxy}p~XYX`X`aeXf@GGHLM^_^f^fgk^lAVWV^V^_cVdpspzpz{p@ DA #cRSTZR[RaRabcRdNev\hIhm^cdfdkdkdm^n}@ACAGAGAI}J[^ac`c_h_h_j[kyO [ x B HIIJLQRwLwKCKCDHDNDNOSafnoDpz@KCKA!!2551I!E1RSWS^S^_cSd1d+e+l+lmq+rz~GKQS[\]c[d[k[klp[qy} ~~ 1J((sT TTTT%c |g}ttdz jdD]k}tj|d5}|j t |dj|jDcic]}||| c}fdddmt|dk7r td|Scc}w#1swYxYw)Nzdata_processed/pairedz*_cad_ply_targets.npzFr:run_idrMz requires exactly two paired runs) sortedrglobrrrsappendr/itemrtryrz)outr4rrs r5pairsr s $..445LM NQ wwqe$#**c!H+:J:J:L6M_`_f_fNg_fZ[qQRSTQUv_fNg5h*i$$ OHaKl#EFF Oh$$s5B23 B-B2-B22B; c|djdj}|djt}t|djt\}}|j }t jt|t}d|d|dfd|d |d ffD]Z\}}}td d D]E} || d } | |vs|| d z s|| d z ||j| <d ||j| <G\||z} t jtt j| |ddt j||d|jd|jdf|fS)N process_xr=process_feature_validprocess_feature_namesu 상부온도 heater_top_cheater_top_imputedu 하부온도heater_bottom_cheater_bottom_imputedrF02dr)rurwrvr r/tolistrrzerosryrangeindexr_r rasarray) r?rlobsnamesunitsimpprevalflagrrgoods r5procr&s {^4 %%'A.E,F,M,Md,S`tuvxOvPvWvWX[v\a]T_TYZ_didpdpdr^cwyww@CDG@HIMwNsv%a&7:N8OPR`abctauvwyPwQRRS\ST 2;a uQsG_15jT!A#Y3qs8s5;;q>2PQC Aztrain..@s!`ytxmpmnopaefgahtxsrkrnrMrrFgh㈵>)lr weight_decayr=)r )geArNr) pos_weightrg333333?)deltar皙?)epochtotalrisk_bce risk_dicesigned_deformation_hubermagnitude_deformation_huberr%)runrrrweak target_signedmap_distance_mmr)variant node_features edge_featureshistory best_loss best_epochruns).r rrzipryrrrrrr rrroptimAdamW parametersmap as_tensorrlongr| clamp_minclampr zero_gradr binary_cross_entropy_with_logitssigmoid huber_lossr}backwardsteprdetachr state_dictitemscloneload_state_dictevalno_gradnumpyr enumeraterr)1rrrepochsrx0e0nnamesenamesrdeformrpvrrr?rrprtrtdroptXEAPBRDpwhistbestepr bcedicedlmllossrrrr4rrjr*r4s1 r5trainrh?s va~&2bvaPT~6Fd6$[^`ytx`y[zVZVXYZ~ABCDIBJ~K{|NPNSNSTVWYTYNZLM]_]b]bcdekclmnoumvwxmxcx]y[\}}B}BCEFHCH}Iz|LNLULUVXV_V_`aVbcdLeJKikininostuovw{|}w~o~ifhCECHCHIOPQIRSYZ[S\I\C]@Bdghihohopqhrsus{s{|}s~BCEFGCHIdJ^cOTOZOZO`O`afaqaqaswyGKO`OLKNWZ[`[j[jlmnoprsus}s}~@AEsFGHkIWJMVMNOPQRSTUVMNMSMSMUKLXYX^X^X`VWchcrcrsucvaby~yHyHIKyLwxRUVWRXYZY^Y^Y`R`bcbgbgbibsbstubvQvP}P}B E P}PF MOL N G K T ` O S =R--/E!Aa!$!)K)KAfIVWce)fSlmoprwrrABCIAJsKLMsMrRrRrTpTUVpVY^YfYfghiogpYqYuYuYwxyx}x}xY@AYAoBmBgkFGFRFRSTU]S^_`gjFkCEopo{o{|}~C|DEFEJEJELSVoWln]`ae]efhikfk]klnoqlq]qX\rvrrrABEBJBJBLXZ[\X\ejkokvkvkxeyEJKNKUKUKWEXejkokvkvkxeyUZ[][d[d[fUgFKLNLULULWFXOYMNZ^ZeZefgZhwZQajAuO_O_OaOgOgOi.jOi!q1A1A1C/COi.jk tAw  mmoa!Aa(a5==6+C+I+I+KaPXkN_N_NabgNgAjklqjrjxjxjz|AkAhioYbcgYhiYhLULMNUORSTc1Q3!Qw1Q3!Qwa!QqSRSGn\`ab\ctz{|t}DuDW[\]W^hnopqrstqtvwxyvy{|u|o}~BCD~EhFGYhi d6l4PVA%VVct|d}tjd|djddz t d|djdt }tj |}tj |}tjj||dd|ddddftjj||dddtj|d d }t|ddD]\}} || dks||| <|ddd|fjj} |d j} i} t| D]q\} }|d|df|d|d f|d |dffD]L\} }| jttt | t |fgj!| Ns|d| }tj"|dddf|dddfz |ddd f|dddfz }|tj$tj&j)|dd dz}g}| D]\} }| j+tt| |fg}t|d k(r|j!d||j-dj||j-djt/tj0tj2||d||dzd dd|j!t|dk(rdndt|ddg}|D]}g}t5|d|D]\}\}}t7|dj9t:dj}g}dD]P}|j=|}|j!||d|rt/|d|ndt?|d|dR|j!i|jADcic]F\}}|dvs |tC|tjDr|j-djn|Hc}}|dj-dj|d j-d j|d!j-d j|d"j9t>j|d#j9t>j|d$|j!|d%|d%d&k(rd'nd(|d)|d*|d+}|d,ddttFd-zfj9t>}||ddd.fd/kDz} tjH|dtjH|ddd0fdkdtjH| d tjH|ddd-fdk\d1d}!|dj-dj|d2j-dj|ddd|fjj|ddd3fj-dj|ddd4fj-dj|ddd5fj-dj|ddd-fj-dj|ddd6fj-dj|j| j|!j|tj$|dz j-djd7 | ttJD"cic]6\}"}t;||d||"fj-dj8c}}"||d8|djdd9gd:d; Scc}}wcc}}"w)two_facerrN)status trianglesnormals theta_radboundarynonmanifold_or_missingz*face dihedral undefined; no theta invented)ro face_countpolicyr4r)u 가열시간u 가열온도u상부원단온도u하부원단온도rr)namevalueobserved)r*rrrr+rrrrrr) magnituderrrrkey_global_processr.rzDihedral-aware GNNzNormal GNN baseliner2)idlabelr2r4rlrmrgư>rJr>r ) rrrr"r%cad_signed_curvature_validcad_shape_index_validcad_shape_classmean_adjacent_dihedral_radz 4,500 of zA directed edges drawn; selected edge and all-edge mean are exact.)zRPrepared graph_6000: requested ~6k, actual 5,925 nodes; no padding or duplication.zSigned CAD curvature is gray/unknown outside the orientation-valid mask; shape index is additionally undefined at near-zero curvedness. H positive/negative follows prepared-normal convention, not universal outward convexity; K<0 is saddle.zLTwo NG scans and point-dilated weak labels only; no held-out generalization.zeNo validated registration, local thermal field, material/contact/FLC, stress/strain, or tear physics.) nodesrqedges node_featureincident_edgesincident_edge_featureincident_face_geometrymodelsedge_draw_note limitations)&ryrrlinspacerminrraddatfullrLTr setdefaulttuplerrcrossrlinalgnormgetroundrarccosrr5r rur/rrvrF isinstancendarrayrrr)#rinfosrrpickr%degreefirstrgrincfacesfmfirrtrifnfgfsrinforsrrr?rkeysrrrlvalid shapevalid shape_classrs# r5viewrHs9 qx=bkk!AfIOOA$6q$8T!F)//RSBT9U\_`gigogopqgrafz|{C{CDE{FsyGIGMGMGPGPQVWXY_W`abWcdefldmnopqnqdrGstvtztzt}t}~DEFGMENOPEQRStT[][b[bcdeghl[mUZ ail #SQq 1XaZa $vYqw  "65R u TRtAaDk1Q4!+qtAaDk 2ca2==vsSTvVYZ[V\oG^A_`b3c3j3jkm3n 2uXe_QqS#ac(!23qs8C!H3DERb"**UWU^U^UcUcdflmw{UcU|~BKCGCbGIDF SQqVVE&!A- $"WaZ J3r7==;K;R;R;T_abd_e_k_klm_n_u_u_wEJKMKTKTUWU\U\]_`bcd`e]fgijlmnjogp]pqstuUvKwExyz yyR!:9Q_bce_fq]^_   T"T&\$'ga1 a 78??D Ea H O O Q5WYRV Uq kk!nAT[[qQhOijkOl5;9J3KrvCGHIJaHbcdHeCf"gh V99RWXW^W^W`XW`PSPQRSdejWeW:a +CAGGAJ%%'JW`XRefglemesestueve}e}eOPQ_O`OfOfghOiOpOpOrEFGXEYE_E_`aEbEiEiEkABCWAXA_A_`dAeAlAlAnGHI`GaGhGhimGnGuGuGwMQRS (  --d9od9oWgFg.BnCPTU`PaiklmuXAfIaE 2 o.55d;eucRSTVRViX\nG]Jjljrjrtysyz{|~}E}EFIJKLNJNFOPQFQRSTVT\T\^h]hijkmkskstwxyz|x|t}@t@ABCDkETF}GkH^iYZ[`YaYgYghiYjYqYqYs~@H~I~O~OPQ~R~Y~Y~[defldmnoptntdudwdwd~d~d@orstuvsvowo}o}~o@oGoGoIY\]^_a]aYbYhYhijYkYrYrYtEHIJKMIMENETETUVEWE^E^E`~ABCDFBF~G~M~MNO~P~W~W~Ylopqrtptlul{l{|}l~lE lE lG e j e q e q e s L V L ] L ] L _ r } r D r D r F e j k m k u k u v | } ~ k  e  d F d F G H d I d P d P d R PS e h y BCIy JA Ky Jr u r s t u B E F G B H I J K Q I R S X Y Z S Z I [ I a I a b c I d I k I k I m B m y JA KegqwKTUVW]U^UdUdefUgThhiIjyxPyIyXA Ks _'$>_' ;_-c, |js tdtt}t }g}dD]}t ||||j |j\}}tjddtj|j|t|d|d|ddd t|d z |j|tjdd|Dcic]}|d | }}t!j"td z |d|d t!j$t|dt!j&|ddDcgc]}|d c}t!j(|ddDcgc]}|d c}t!j(|ddDcgc]}|d c}t!j(|ddDcgc]}|d c}t!j(|ddDcgc]}|d c}t!j(|ddDcgc]}|d c}t!j(|ddDcgc]}|d c}t!j*d tdz tdz tdz tdz t,dz t,dz t,dz fD]#} | j/s| j1%ddt|dt3|d j4d!d"dd#d$d%|Dcic]}|d |d|dd&c}d' } t,d(z j7t9j:| dd)*d+,td-z j7t9j:t=|||dd./d+,t?j@d01\} } |D cgc]A}| jC|d2D cgc]} | d3 c} |d2D cgc]} | d4 c} |d 5Cc} }| jEd6| jG| jId78| jKd9| jMtd:z d;<t?jN| t?j@d)d)d=1\} }tQ|D]\}}tQ|dD]\}}|||fjS|ddddf|dddd)f|dd)d>dd!?}|||fjK|d d@|ddA|||fjUdB| jW||||fC| jMtdDz d;<t?jN| tY|dd|ddDcgc]4\}}|d|dE|dE|dEdF|dEdFkD|dEdG|dEdGkDdH6}}}dIdJdKdLdMdNdOdP}dQddR||Dcgc]+}|d |dS|dTdU|d2|dDcgc]}|dE c}dV-c}}| | dW|dXgdY }tdZz j7t9j:|dd)*d+,|Scc}wcc}wcc}wcc}wcc}wcc}wcc}wcc}wcc}wcc} wcc} wcc} }wcc}}wcc}wcc}}w)[NzBPass --allow-diagnostic-weak-labels: targets are weak/unvalidated.)normal_baselinerTrOrkr/r0F)rEr. node_countr/r0production_validz_gnn.ptr.z*prepared_graph_5925_direct_predictions.npzr>rr4r*rrrrr) r cad_normalrr normal_baseline_risk_probabilitydihedral_risk_probabilitynormal_baseline_signed_mmdihedral_signed_mmnormal_baseline_magnitude_mmdihedral_magnitude_mmdirect_model_evaluationz#direct_original_cad_predictions.npzz!fixed_6000_direct_predictions.npzzdirect_original_predictions.pngz training.pngz#finest_prepared_cad_graph_11956.npzzfixed_6000_surface_graph.npzz"fixed_6000_surface_provenance.jsonr9rUrVrFz0same prepared graph_6000 with 5,925 actual nodeszusame GNN/seed/epochs/loss/process/threshold; excludes curvature, analytic curvature, boundary/ridge and face dihedralzTacos(clamp(dot(two adjacent face normals), -1,1)); undefined on boundary/nonmanifoldaA new CAD can execute with variable node count/topology, but this experiment fixes the ~6k preparation rule: independently simplify each new CAD with the same rule; node indices never correspond across designs; output is direct per-new-CAD-node. Generalization is unmeasured with one CAD/two runs. Use multi-design training, CAD-family held-out splits and OOD uncertainty before interpreting cross-design output.)rmrn) surface_graphr]r^r_!training_inference_viewer_outputsnode_density_is_risk_featurerdihedral_face_definition changed_cadfeatureszfeature_contract.jsonrMrdrgrhzviewer_data.json),:)re separators)rg @)figsizer1r$r%)r}logr#)alphazrsf"! .*..'JJ+c"33#g+E#i-c^aX]bjjmnrzzEF ~ T B"))BCC]y,lR Zr7