xf!bdZddlmZddlmZddlmZmZmZddl Z ddl m Z dgZ d Zddefd Zy) z&Byrd-Omojokun Trust-Region SQP method.)eye) projections)modified_dogleg projected_cgbox_intersectionsN)normequality_constrained_sqpcFtj|\}t|S)N)npshapespeye)xns ]/usr/lib/python3/dist-packages/scipy/optimize/_trustregion_constr/equality_constrained_sqp.pydefault_scalingr s !BA 8Oc > d}d}d}d}d}d}d}d}d}d}d}tj|\}| %tj|tj } |$tj|tj}tj|}| }| }|}|} |}!|}"||}#t |"| \}$}%}&|%j |  }'|||'}(t| |"jj |'ztj})t|!d kDrt|!tjnd }*d d d d }+d },|| ||,|)|*|||+sTt|"|&|!||z|| z||z}-|(j |-| z}.tj|!}/tj|d ztjj|-d zz }0| |-z }1||-z }2t|(|.|$|&|/|0|1|2\}3}+|-|3z}4d|(j |4j |4z| jj |4z}5|"j |4|!z}6t|!t|6z }7td |7}7|}8|5d kDr|5d|z |7zz }9t||9}|5 ||7zz}:||t|!zz};||#j |4z}<||<\}=}>|=|t|>zz}?|;|?z }@|@|:z }A|A|krt|-|t|3zkrs|&j |> }Bt!|4|B| |\}C}D}E||#j |4|D|Bzzz}F||F\}G}H|G|t|Hzz}I|;|Iz }J|J|:z }K|Er K|k\rF}<G}=H}>K}AA|k\rt|t|4z|}nZA|k\rt|t|4z|}n<A|kr7d|z dAz z }L|Lt|4z}M|M||zk\r||z}nM||zk\rM}n||z}A|k\r|<}|=|>}!}||\} }"||}#t |"| \}$}%}&|%j |  }'|||'}(d },t| |"jj |'ztj})t|!d kDrt|!tjnd }*n|8}d},|| ||,|)|*|||+sT|| fS)auSolve nonlinear equality-constrained problem using trust-region SQP. Solve optimization problem: minimize fun(x) subject to: constr(x) = 0 using Byrd-Omojokun Trust-Region SQP method described in [1]_. Several implementation details are based on [2]_ and [3]_, p. 549. References ---------- .. [1] Lalee, Marucha, Jorge Nocedal, and Todd Plantenga. "On the implementation of an algorithm for large-scale equality constrained optimization." SIAM Journal on Optimization 8.3 (1998): 682-706. .. [2] Byrd, Richard H., Mary E. Hribar, and Jorge Nocedal. "An interior point algorithm for large-scale nonlinear programming." SIAM Journal on Optimization 9.4 (1999): 877-900. .. [3] Nocedal, Jorge, and Stephen J. Wright. "Numerical optimization" Second Edition (2006). g333333?g?g:0yE>g@g@g?g?g?rF)niter stop_cond hits_boundarygؗҜ>"3Eq3H!HI"}"}"1c1a#1#'/ G GquuQxnnQ//!##''!*<EE!HQJQ$011E5!" Q )a.>-EFK';/G /.>WT!W_,QUU1X'/$wtF|';;*,??*-@@ 7 7 8}tBx/ /vA/1hIOAq)a!A#g&E)%0LE5!&e)rvs8,%$KK % &'+&*%4Hr