jF<dZddlZddlZddlZddlmZddlZddlZej de ee j ej ddddlZddlZddlZddlZddlZddlZddlZej ddddlZddlmZddlmZmZddlm Z m!Z!m"Z"m#Z#m$Z$e!d z Z%dd lm&Z&d Z'd d ddddddddddddddZ(iZ)gdZ*ddiZ+dZ,dZ-d d!iZ.gd"Z/d#Z0e1d$kr e0dSdS)%aCEvery working approach on every cloud: ../report/ablation.csv (resumes: rows already there are skipped). Clouds: ../pointclouds/__<10000|2048>.ply, every tag of make_clouds.py (clean, mesh-noise, per-point noise, variable density). Configurations: package fcp.encode (surf slicer, tracked-fit, fit_tol="auto") pointtest package slicer, cuts by the two-sided point test (pointfit.py) A the slicer's rings refitted to their points (ringfit.py) A+pointtest hybrid slicer topology repaired from the neighbouring layers, all fitted (hybrid.py) hybrid+pointtest hybrid-gated the same, a carried set tried only where the slicer's layer is suspicious hybrid-gated+pointtest thinwall thin walls (a parallel face within 1.5 spacings) added to the slicer's field as a band (thinwall.py) thinwall+pointtest eps one scale eps = max(spacing, 4 sigma) for the slicer and the minimum loop, point test (eps.py); sigma = the cloud's per-point noise, given (0 for clean, mesh-noise and density clouds) gwn the GWN slicer (../../fcp_2810_gwn/scripts/gwn.py, defaults, sigma given): inside / outside from the mollified generalized winding number; package cuts gwn+pointtest the same, cuts by the point test gwn-tree gwn with its two sums as tree codes (fastgwn.py; GPU): must match gwn local the same idea per point: eps_i = max(local spacing_i, 4 sigma) in the slicer's kernels (surf_local.py, CPU), point test Reference: the mesh the cloud was sampled from (the noisy mesh for noise*, else the clean mesh), mesh route. Errors against the clean mesh (the truth). env -u PYTHONPATH PYTHONWARNINGS=ignore .venv/bin/python ~/code/data/meshes/fcp_2710_tests/scripts/ablation.py N)Pathz/home/joao/code/fcp_devel/apiz0/home/joao/code/data/meshes/fcp_2810_gwn/scripts)PN)SIGMASlabel)OBJREPORTROOTUPstatsz ablation.csv)TAGS)i'ic0tj|d|S)Ncudadeviceup)fcpencodePrs >/home/joao/code/data/meshes/fcp_2710_tests/scripts/ablation.pyr6sSZ&R@@@c.tj||SN)r)pointfitencode_pointtestrs rrr7sx8rBBBrc.tj||Srrfencode_ringfitrs rrr8sr(r222rc0tj||dSN pointtest)rcutsrrs rrr9s!21+!N!N!Nrc.tj||Srhy encode_hybridrs rrr:sB,Q2666rc0tj||dSr"r&rs rrr;sb&6qRk&R&R&Rrc0tj||dS)NT)rgater&rs rrr<s""21$"G"G"Grc2tj||ddS)Nr#T)rr$r+r&rs rrr=sB,sb0r:::rc0tj||dSr"r.rs rrr?s(:1+(V(V(VrcFtj||tdSNsigma)rr4)epsmod encode_epsCTXrs rrr@s*13w<HHHrcFtj||tdSr3) surf_local encode_localr7rs rrrAs:213w<PPPrcFtj||tdSr3gwn encode_gwnr7rs rrrBsbG EEErcHtj||tddS)Nr4r#)rr4r$r<rs rrrCs3>!#g,U`#a#a#arc`t||tdSr3) _gwn_treer>r7rs rrrDs#ikk44Q2S\4RRr)packager#Az A+pointtesthybridzhybrid+pointtestz hybrid-gatedzhybrid-gated+pointtestthinwallzthinwall+pointtestepslocalr=z gwn+pointtestgwn-tree) pn0p6_10000 pn1p2_10000 dgrad_10000 dstripe_10000 clean_2048 pn0p6_2048 noise1_10000 dgrad_2048 pn1p2_2048noise0p6_10000rHc4dt|DS)Nch|]F\}}dt|ttzfD]}|g|ddRGS) clean_10000_)_OTHERlenrsplit).0iocs r z_sample..Ns] t t ttq!M[abcfijpfqfqbq[rKs t taQ "#q!! " " t t t tr) enumerate)objectss r_samplerbMs t ty/A/A t t ttrclts!ddl}|dtd<tdS)zvr4||t=| ftAtCtvrY| )tEtGj$| d|j%} tMj'} || |}tMj'|z }tGj(|d}|tS|||nt+dt+df\}}|j%r#tj*d|j%Dntj+d}tYtE|j%t[d|j%Dt]|dk-||t]|dut_|d}n#t`$r}tYdddt+dt+ddt_tMj'|z d}tcd||| | te|d Yd}~nd}~wwxYw|3tYd"||| | | d!||4tc||| d  ddddS#1swxYwYdS)#NcJh|] }|d|d|d|df!S)rirjrkrl)r[rs rr_zmain..`s1cccAQx[!G*ak1X;Gcccra)newlinemesh)forceiPrW)seed)rrrWnoisemeshesrVz.plyc4i|]}dt||S)pn)r)r[vs r zmain..ms& @ @ @eAhh! @ @ @rrhr4 pointcloudsrr)rnanc&g|]\}}}||z dzS)rWrw)r[rVh0h1s r zmain..}s&-S-S-SiaRb2gk-S-S-Srrc3*K|]\}}}|dkVdS)rNrw)r[srVs r zmain..~s.DbDbwqRSUVQUDbDbDbDbDbDbr)rmrnrorqrrrsrtFAILEDT)flush)rirjrkrlrprw)5setOUTexistsopencsv DictReader DictWriterFIELDS writeheaderrtrimeshloadnpasarraysampler getr startswithr rfloatlinalgnormextentsr7SIZESverticesCONFIGSitemsstrSAMPLEDrblistrYrrrmtime perf_counterdecoder arrayzerosdictsumintround Exceptionprintreprwriterowr)donefnewwnamegrefrtagsrcrpnrcfgf_enctdsecrdrqrrspansrowes rmainr\s 55D zz||d XXZZ d1ccQTQ_`aQbQbcccD d d d d d d d d d d d d d d djjll C #r " "$4a N1f % %   MMOOO 4 4D SYf555A*QXXf1X5566Ci((B 4 4[^[i[ijq[r[rygl4(?5H5Hs5H5H5H#HPVWWWWxy @ @R @ @ @ D DS# N NRWXZXaXfXfghgpXqXqRrRrrG 44A 7<}0D$G^G^QTG^G^WXG^G^G^0^#_#_#hiiA&-mmoo"" U #s1vvs3t;;$'>>tS#a&&.AQUVYQZQZI[I[.[.[$$,(+CJs6b,Q,Q,Q,W(X(XI -// T %a A"&"3"5"5"9C # 1V < < rsk4 33ttH~~,--...2333 EFFF $$$$$$$$55555555555555 ~@@BB 2 2NN66RRGGcc::VV H H P P E EaaRR  "  8 8 8 t uuun { { {*4*4*4Z zDFFFFFr