^Mh]7^ddlZddlZddlmZejejZdZ dZ dZ dS)N)get_arrays_tolc l|rNt|tsJt|tjr |jdksJt j|sJt|tjr|j|jksJt|tjr|j|jksJt|tsJt|tsJt||}tj ||ksJtj || ksJtj |r|dksJtj |d}tj|d}t||||||\}} t| | fd||||\} } t!| t!| kr|n| } |r\tj || ksJtj | |ksJtj| d|zksJ| S)a' Maximize approximately the absolute value of a quadratic function subject to bound constraints in a trust region. This function solves approximately .. math:: \max_{s \in \mathbb{R}^n} \quad \bigg\lvert c + g^{\mathsf{T}} s + \frac{1}{2} s^{\mathsf{T}} H s \bigg\rvert \quad \text{s.t.} \quad \left\{ \begin{array}{l} l \le s \le u,\\ \lVert s \rVert \le \Delta, \end{array} \right. by maximizing the objective function along the constrained Cauchy direction. Parameters ---------- const : float Constant :math:`c` as shown above. grad : `numpy.ndarray`, shape (n,) Gradient :math:`g` as shown above. curv : callable Curvature of :math:`H` along any vector. ``curv(s) -> float`` returns :math:`s^{\mathsf{T}} H s`. xl : `numpy.ndarray`, shape (n,) Lower bounds :math:`l` as shown above. xu : `numpy.ndarray`, shape (n,) Upper bounds :math:`u` as shown above. delta : float Trust-region radius :math:`\Delta` as shown above. debug : bool Whether to make debugging tests during the execution. Returns ------- `numpy.ndarray`, shape (n,) Approximate solution :math:`s`. Notes ----- This function is described as the first alternative in Section 6.5 of [1]_. It is assumed that the origin is feasible with respect to the bound constraints and that `delta` is finite and positive. References ---------- .. [1] T. M. Ragonneau. *Model-Based Derivative-Free Optimization Methods and Software*. PhD thesis, Department of Applied Mathematics, The Hong Kong Polytechnic University, Hong Kong, China, 2022. URL: https://theses.lib.polyu.edu.hk/handle/200/12294. c| S)N)xcurvs e/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/_lib/cobyqa/subsolvers/geometry.pyz!cauchy_geometry..[s4477(皙?) isinstancefloatnpndarrayndiminspect signaturebindshapeboolrallisfiniteminimummaximum _cauchy_geomabslinalgnorm) constgradr xlxudeltadebugtolstep1q_val1step2q_val2steps ` r cauchy_geometryr. s=t 2%'''''$ ++> Q &&++D11111"bj))Dbh$*.D.D.D.D"bj))Dbh$*.D.D.D.D%'''''%&&&&&R$$vbCi     vbSDj!!!!!{5!!1eckkkk B  B B  B !dBE5IIME6      ME6KK3v;;..55ED 2vbDj!!!!!vdbj!!!!!y~~d##cEk1111 Krc<|rt|tsJt|tjr |jdksJt j||sJt|tjr!|jdkr|jd|j ksJt|tjr|j|jksJt|tjr|j|jksJt|tsJt|tsJt||}tj ||ksJtj || ksJtj |r|dksJtj|d}tj|d}tj|} |} tj|d} |tj k|jt* |zkz} |tj k|jt*|zkz} |tjk|jt*|zkz}|tjk|jt* |zkz}tjtj|| j| |j| z }tj|dtj }tjtj||jd}tjtj|| j| |j| z }tj|dtj}tjtj||jd}tjtj||j||j|z }tj|dtj }tjtj||jd}tjtj||j||j|z }tj|dtj}tjtj||jd}t5|jdD]}| |t*|zkrt1|| |z d}n2t1t7||||d}t7t1||||d}||dd|fz}||dd|f}|dkr|t* |zks|dkr$|t* |zkrt1| |z d}n tj}|dkr|t*|zks|dkr#|t*|zkrt7| |z d}n tj }t7||}t1| |}|||zzd|d zz|zz}|||zzd|d zz|zz}||kr8|||zzd|d zz|zz} t9| t9|kr|}| }||kr8|||zzd|d zz|zz}!t9|!t9|kr|}|!}t9|t9|krGt9|t9| kr'tj||dd|fz||} |} kt9|t9|krEt9|t9| kr%tj||dd|fz||} |} |r\tj || ksJtj | |ksJtj| d |zksJ| S) a Maximize approximately the absolute value of a quadratic function subject to bound constraints in a trust region. This function solves approximately .. math:: \max_{s \in \mathbb{R}^n} \quad \bigg\lvert c + g^{\mathsf{T}} s + \frac{1}{2} s^{\mathsf{T}} H s \bigg\rvert \quad \text{s.t.} \quad \left\{ \begin{array}{l} l \le s \le u,\\ \lVert s \rVert \le \Delta, \end{array} \right. by maximizing the objective function along given straight lines. Parameters ---------- const : float Constant :math:`c` as shown above. grad : `numpy.ndarray`, shape (n,) Gradient :math:`g` as shown above. curv : callable Curvature of :math:`H` along any vector. ``curv(s) -> float`` returns :math:`s^{\mathsf{T}} H s`. xpt : `numpy.ndarray`, shape (n, npt) Points defining the straight lines. The straight lines considered are the ones passing through the origin and the points in `xpt`. xl : `numpy.ndarray`, shape (n,) Lower bounds :math:`l` as shown above. xu : `numpy.ndarray`, shape (n,) Upper bounds :math:`u` as shown above. delta : float Trust-region radius :math:`\Delta` as shown above. debug : bool Whether to make debugging tests during the execution. Returns ------- `numpy.ndarray`, shape (n,) Approximate solution :math:`s`. Notes ----- This function is described as the second alternative in Section 6.5 of [1]_. It is assumed that the origin is feasible with respect to the bound constraints and that `delta` is finite and positive. References ---------- .. [1] T. M. Ragonneau. *Model-Based Derivative-Free Optimization Methods and Software*. PhD thesis, Department of Applied Mathematics, The Hong Kong Polytechnic University, Hong Kong, China, 2022. URL: https://theses.lib.polyu.edu.hk/handle/200/12294. rrrr)axis)r0initialN?@r)rrrrrrrrrsizerrrrrr zeros_liker r!infTTINY atleast_2d broadcast_tomax atleast_1drangeminrclip)"r"r#r xptr$r%r&r'r(r-q_vals_normi_xl_posi_xl_negi_xu_posi_xu_neg alpha_xl_pos alpha_xl_neg alpha_xu_neg alpha_xu_poskalpha_tr alpha_bd_pos alpha_bd_neg grad_step curv_stepalpha_quad_posalpha_quad_neg alpha_pos alpha_neg q_val_pos q_val_negq_val_quad_posq_val_quad_negs" r spider_geometryrYjsx 2%'''''$ ++> Q &&++D11111 sBJ ' ' A  !  ))))"bj))Dbh$*.D.D.D.D"bj))Dbh$*.D.D.D.D%'''''%&&&&&R$$vbCi     vbSDj!!!!!{5!!1eckkkk B  B B  B =  D E Y^^Ca^ ( (FbfW $!34HbfW !23HRV r 12HRV  23H= HN++H5hGL6,Q@@@L?2=#>#> ! MML= HN++H5hGL6,Q???L?2=#>#> ! MML= HN++H5hGL6,Q@@@L?2=#>#> ! MML= HN++H5hGL6,Q???L?2=#>#> ! MML 39Q< JJ !9te| # #56!9,c22HH 3|A Q@@#FF 3|A Q@@#FF 3qqq!t9$ DQQQTOO   TEI---CTEI--- )i!7==NNVN   D9,,,CD9,,, )i!7==NN fWN,//  <00 I ) )C)S.,@9,L L  I ) )C)S.,@9,L L  I % % 9,-++i78  >""S^^33* * I % % 9,-++i78  >""S^^33* * y>>S^^ + +IU0K0K79s111a4y0"b99DEE ^^c)nn , ,Y#e**1L1L79s111a4y0"b99DE 2vbDj!!!!!vdbj!!!!!y~~d##cEk1111 Krc|dk|dkz}|dk|dkz}tj|} ||| |<||| |<tj| |kr||z} tj|| } tj|dz| | | | zz } | t t | zkrt| | z d} nni| || z| | <| | |kz}| | |kz}tj|stj|sn ||| |<||| |<| ||zz} || z}|dkrjtj| }|t |zkrt||z d}nd}|| }|t |zkrt| |z d}n tj }|tj k| t |zkz}|tj k| t |zkz}tj ||| |z tj }tj ||| |z tj }t||}t|||}tj || z||}|||zzd|dzz|zz}ntj|}|}|r\tj ||ksJtj ||ksJtj|d|zksJ||fS)zM Same as `bound_constrained_cauchy_step` without the absolute value. rTr3)r1r2r) rr5r r!sqrtr8rr;anyr6r>r?r)r"r#r r$r%r&r'fixed_xlfixed_xu cauchy_stepworkingg_norm delta_reducedmurOrBrLrP alpha_quadi_xli_xualpha_xlalpha_xualpha_bdalphar-rAs r rr8su STCZ(HSTCZ(H-%%KxLKxLK y~~k""U**X% 7Y^^DM22FGs ['2['5JJJMs=11111/55#%W #5K +"23H+"23H6(## BF8,<,< $&xLK !$&xLK !(X"5 66G) 7.{"IC ,, D5L 56>3//HHHD%% uy( ( (iZ)3S99JJJbfW tby!89RV dRi 786"T([%66GGG6"T([%66GGGx**Hj(33wu{*B33 ))C%*,ros""""""rx\\\~KKK\KKKKKr