^j7 \ddlZddlmZddlZddlmZmZddlm Z ddl m Z ddl m Z ddlmZmZddlmZmZmZdd lmZGd d ZGd d ZdedefdZdedefdZGddZ d%dedededededef dZde efdZ!de ede"fdZ#de fd Z$de"ezfd!Z%d"ed#efd$Z&y)&N)Path)Geom_BSplineCurveGeom_BSplineSurface)#GeomConvert_CompCurveToBSplineCurve)gp_Pnt) Precision)TColgp_Array1OfPntTColgp_Array2OfPnt)TColStd_Array1OfIntegerTColStd_Array1OfRealTColStd_Array2OfReal)minimizeceZdZdZd defdZedefdZejdefdZdefdZ de fd Z de fd Z y ) Standard_Realzr A mutable wrapper for a float value, mimicking C++'s Standard_Real when used as an output parameter. valuec$t||_yNfloat_value)selfrs Q/opt/ringagent/.cad-venv/lib/python3.12/site-packages/ocp_gordon/internal/misc.py__init__zStandard_Real.__init__s El returnc|jSrrrs rrzStandard_Real.values {{r new_valuec$t||_yrr)rrs rrzStandard_Real.value!sI& rc|jSrrrs r __float__zStandard_Real.__float__%s {{rc"d|jdS)NzStandard_Real()rrs r__repr__zStandard_Real.__repr__(s }A..rc,t|jSr)strrrs r__str__zStandard_Real.__str__+s4;;rN) __name__ __module__ __qualname____doc__rrpropertyrsetterr"r'r%r(rrrrsq #e#u \\'u''5/#/  rrceZdZddededefdZdZdefdZdedefd Zded efd Z d efd Z d efd Z d efdZ dZ dZdefdZdefdZdefdZdZdZdZy) math_Vector lower_index upper_index init_valuec||kDr td||_||_tj||z dz|tj |_y)Nz.Lower index cannot be greater than upper indexdtype) ValueErrorr4r5npfullfloat64data)rr4r5r6s rrzmath_Vector.__init__0sH  $MN N&&GGK+59:RZZX rc:|j|jz dzSNr8)r5r4rs r__len__zmath_Vector.__len__7s$"2"22Q66rindexc |j|cxkr|jks+ntd|d|jd|jd|j||jz SNzIndex z out of range [z, ]r4r5 IndexErrorr?rrCs r__call__zmath_Vector.__call__:sk  E=T-=-==t/?/?.@4CSCSBTTUV yy!1!1122rrc |j|cxkr|jks+ntd|d|jd|jd||j||jz <yrErG)rrCrs rSetValuezmath_Vector.SetValueAsj  E=T-=-==t/?/?.@4CSCSBTTUV /4 %$***+rrc||Srr1rIs rValuezmath_Vector.ValueHs E{rc|jSr)r4rs rLowerzmath_Vector.LowerKrc|jSr)r5rs rUpperzmath_Vector.UpperNrQrct|Sr)lenrs rLengthzmath_Vector.LengthQs 4yrct|trt|t|k7r tdt|j|j }|j |j z|_|S)Nz2Vectors must be of the same dimension for addition isinstancer3rUr;r4r5r?rotherresults r__add__zmath_Vector.__add__TsX%-Tc%j1HQR RT--t/?/?@ii%**,  rct|trt|t|k7r tdt|j|j }|j |j z |_|S)Nz5Vectors must be of the same dimension for subtractionrXrZs r__sub__zmath_Vector.__sub__[sX%-Tc%j1HTU UT--t/?/?@ii%**,  rscalarcnt|j|j}|j|z|_|Sr)r3r4r5r?rr`r\s r__mul__zmath_Vector.__mul__bs/T--t/?/?@ii&(  rc$|j|Sr)rc)rr`s r__rmul__zmath_Vector.__rmul__gs||F##rc|dk(r tdt|j|j}|j|z |_|S)NrzCannot divide by zero)r;r3r4r5r?rbs r __truediv__zmath_Vector.__truediv__jsB Q;45 5T--t/?/?@ii&(  rct|trt|t|k7rytj|j |j S)NF)rYr3rUr< array_equalr?rr[s r__eq__zmath_Vector.__eq__qs7%-Tc%j1H~~dii44rc&|j| Sr)rkrjs r__ne__zmath_Vector.__ne__vs;;u%%%rcVd|jd|jd|jdS)Nzmath_Vector(lower=z, upper=z, data=r$)r4r5r?rs rr%zmath_Vector.__repr__ys4#D$4$4#5Xd>N>N=OwW[W`W`VaabccrNr))r+r,r-intrrrBrJrLrNrPrSrVr]r_rcrergrkrmr%r1rrr3r3/sYCYcYuY73c34c4%435 s  s e $u$%5 &drr3splinerc td|j}|j|td|j}|j |td|j }|j |td|j }|j|t|||||j|j}|S)zw Clone a B-spline curve. Args: spline: Original B-spline curve Returns: New B-spline r8) r NbPolesPolesr WeightsNbKnotsKnotsr MultiplicitiesrDegree IsPeriodic)rppolesweightsknotsmults new_splines r clone_bspliners q&.."2 3E LL#1fnn&67G NN7 !FNN$4 5E LL $Av~~'7 8E % # wufmmov7H7H7JJ rsurfacec td|jd|j}|j|t d|jd|j}|j |t d|j}|j|t d|j}|j|td|j}|j|td|j}|j|t|||||||j|j!|j#|j% }|S)z Clone a B-spline surface. Args: surface: Original B-spline surface Returns: New B-spline surface r8)r NbUPolesNbVPolesrsr rtr NbUKnotsUKnotsNbVKnotsVKnotsr UMultiplicitiesVMultiplicitiesrUDegreeVDegree IsUPeriodic IsVPeriodic)rrzr{u_knotsv_knotsu_multsv_mults new_surfaces rclone_bspline_surfacersD q'"2"2"4a9I9I9K LE MM%#1g&6&6&8!W=M=M=OPG OOG#1g&6&6&89G NN7#1g&6&6&89G NN7&a)9)9);td}|j|||S)Nr*)rr)rrrrs rGradientz-math_MultipleVarFunctionWithGradient.Gradients # {{1a##rcy)NTr1rs rrz+math_MultipleVarFunctionWithGradient.ValuessrN) r+r,r-rrorr3rboolrNrrr1rrrrsh S${$}$$$+$+$$$  +$rraFuncaX aTolerance aNbIterationseZEPSc  j dttf fd tjt d dzDcgc] }|| c}}d fd }||}t |dd|d|i} t D]2}|j|j|z| j|4| jScc}w) Nargsc td}tD](}|j|j|z||*t d}td}j |||st d||jtjtddzDcgc] }|| c}fScc}w)Nr8r*zfunction cannot evaluate at ) r3rangerLrPrr RuntimeErrorrr<array)rocp_argsiF_kG_k_vecr nb_variabless rfzmath_BFGS..fsq,/|$ =A   hnn.2DG < =C a.||Hc73!=dVDE Eyy"((aPQAQ8R#S1GAJ#STTT#Ss'B> r8c |j}|dk(sJdtj||ft}t |D]e}|t dt ||z}tj|}|||<||z\}} ||z \}} | | z d|zz |dd|f<gd||jzz}t tj tj|j d} |tj||| zzz} tjj| } d| | jzz} tjj| \}}tj ||d}|tj|z|jz}d||jzz}tjj#||S#tjj$rdt dtjtj tj|}tj|d|z z} Y$wxYw#tjj$rpt%d t dtjtj tj|}tj|d|z z} | cYSwxYw) a Estimate 2x2 inverse Hessian by finite-differencing the gradient and inverting. - x0: 1D array-like, length=2 (current point) - grad_func: function(x) -> gradient array-like shape (2,) - rel_eps: relative step size for finite differences (default sqrt(machine eps)) - reg: regularization added to Hessian diagonal before inversion Returns: 2x2 numpy array H0 (approx inverse Hessian) rz:This routine expects 2 parameters; adapt if you have more.r9g?g@Ng?:0yE>u3Hessian projection failed — fallback to identity.)sizer<zerosrrmaxabs zeros_likeTdiageyelinalginv LinAlgErrormeaneighclipcholeskyprint)x0rel_epsregnHjepsej_g_plusg_minus diag_scaleH_reginvHavg_curveigvalseigvecseigvals_clampedinvH_spdrs restimate_initial_inv_hessianz/math_BFGS..estimate_initial_inv_hessiansq GGAvSSSv HHaV5 )q 7ACSAZ00Cr"BBqE"r' IAv27JAw'C#I6AadG 7 1qss7O +//137 BFF1Iz!122 099=='DdTVVm$ !yy~~d3 GW!gggw=O!99GIIEHh34H II  x (9yy$$ 04 );!<=H66!9h/D  0,yy$$  G H4 );!<=H66!9h/DK  s'G BI"A=II"B K/.K/TBFGS hess_inv0)rjacmethodtoloptions)rr) rlistrr<rrrrLrPxsuccess) rrrrrrrrH0resrrs ` @@r math_BFGSrs$$&L UU U %B@D &b )B  bd6zKQSCT C< . BHHJNCEE!H-. ;;_>sC splinesc g}|D]s}g}td|jdzD]Q}|j|}|j|j |j |j fStd|jdzDcgc]}|j|}}td|jdzDcgc]}|j|}}td|jdzDcgc]}|j|}}|j|||||j|j|j|jdv|Scc}wcc}wcc}w)Nr8)rzr{r|r}degree is_periodicfirst_parameterlast_parameter)rrrPoleappendrYZWeightruKnot MultiplicityrxryFirstParameter LastParameter) robjrprzrpr{r|r}s rsave_bsplines_to_objectrRs\ C 24q&..*Q./ 0A AA LL!##%. / 0.31fnn6F6J-KL6==#LL).q&..2BQ2F)GHAQHH16q&..:JQ:N1OPA$$Q'PP " --/%002#)#8#8#:"("6"6"8   . J#MHPsE>F F file_pathcttj|z d5}tjt ||ddddy#1swYyxYw)Nwr)indent)openrhomejsondumpr)rrrs rsave_bsplines_to_filerpsH diikI%s +Aq )'2Aa@AAAs "A  Aobjsc g}|D]t}t|d}td|}t|D]#}|j|dzt |d|%t d|}t|D]}|j|dz|d|t|d}t d|}t|D]}|j|dz|d|t d|} t|D]}| j|dz|d|t|||| |d|d} d|vrHd |vrD|d| jk7s|d | jk7r| j|d|d |j| w|S) Nrzr8r{r|r}rrrr) rUr rrLrr r rrrSegmentr) rbsplinesrrrrzrr{ru knot_array mult_arrayr~s rload_bsplines_from_objectrus(*H&$c'l#"1g.w >V]]_fmmo6J }}#j) }}#j)    !C .v 6D 88FCt ,@AA    F Mr)i g-q=)'rpathlibrnumpyr<OCP.GeomrrOCP.GeomConvertrOCP.gpr OCP.Precisionr OCP.TColgpr r OCP.TColStdr r r scipy.optimizerrr3rrrrrorrrrr'rrrrr1rrr$s( ;?#= $   6KdKdb + 0A F0#60;N0f2 c /ccc c  c  cNT*;%<<A4(9#:AsA *D*Z +sTz + 1;Lr