TixdZddlZddlmZddlmZeeej fZ ej ej j Z dZdS)a  This module provides the _project function that attempts to project the initial guess onto the feasible set. Adapted from the corresponding function in the PDFO package (https://www.pdfo.net) by Tom M. Ragonneau (https://ragonneau.github.io) and Zaikun Zhang (https://www.zhangzk.net). NLinearConstraint)OptimizeResultc d}t|tr|gn5t|dr|n"td| t jt jn0#t$r#td|wxYwtj dkr"td|j }t|tr|g}n5t|dr|}n"td| t j|t j}n0#t$r#td|wxYwt|j dks |j |kr"td |t|tr|g}n5t|dr|}n"td | t j|t j}n0#t$r#td |wxYwt|j dks |j |kr"td |t|trJd dht| kr#t|d ts*|d "td|d}|d >t jt j|fd|fd} t#| St%t jt j|d j|d jz t.rt j|| krt j||kr|d j} |d j|d jzdz } t j| | t j| z d\} } } } t jt j| z|fd|fd} t#| S|d O ddlm}ddlm }ddlm}|d }tt}}t jggt jggc|d<|d<|d!d|jj d|d<|d!d|jj d|d<t jgt jgc|d<|d<t jgt jgc|d<|d<tE|jj D]*}|j||j|krt j#|d|j||dzddffd|d<t j$|d|j|f|d<t j$|d|j|f|d<t j#|d|j||dzddffd|d<t j$|d|j|f|d<t j$|d|j|f|d<,|dj dkr3|dj dkr"|dj dkr|di||di|g}nG|dj dkr|dj dkr |di|}n|dj dkr |di|}nd}t j|d}t j|d}t j%||d&st j%|d|&s{t j'||d&sNt j%|&s't j%|&r |fdfd|||| St#S#tP$rt#cYSwxYwt#S)!aProjection of the initial guess onto the feasible set. Parameters ---------- x0: ndarray, shape (n,) The same as in prepdfo. lb: ndarray, shape (n,) The same as in prepdfo. ub: ndarray, shape (n,) The same as in prepdfo. constraints: dict The general constraints of the problem, defined as a dictionary with fields: linear: LinearConstraint The linear constraints of the problem. nonlinear: dict The nonlinear constraints of the problem. When ``_project`` is called, the nonlinear constraints are None. Returns ------- result: OptimizeResult The result of the projection. Authors ------- Tom M. RAGONNEAU (ragonneau.github.io) and Zaikun ZHANG (www.zhangzk.net) Dedicated to the late Professor M. J. D. Powell FRS (1936--2015). prima__len__z,{}: UNEXPECTED ERROR: x0 should be a vector.)dtypez5{}: UNEXPECTED ERROR: x0 should contain only scalars.rz,{}: UNEXPECTED ERROR: lb should be a vector.z5{}: UNEXPECTED ERROR: lb should contain only scalars.z={}: UNEXPECTED ERROR: the size of lb is inconsistent with x0.z,{}: UNEXPECTED ERROR: ub should be a vector.z5{}: UNEXPECTED ERROR: ub should contain only scalars.z={}: UNEXPECTED ERROR: the size of ub is inconsistent with x0.linear nonlinearNz6{}: UNEXPECTED ERROR: The constraints are ill-defined.g@xDr)axis)x)rcond)minimize)BoundsrAlbubc@tj|z |z dz S)Nr)npdotrx0_cs z/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/scipy/_lib/pyprima/common/_project.pyz_project..s "&T1t8*D*Dq*Hc|z S)Nrrs rrz_project..s_`cg_gr)jacbounds constraints)) isinstance scalar_typeshasattr ValueErrorformatrasarrayfloat64lenshapesizedictsetkeysrnanminnanmaxrall less_equalabsrrepsmaxminrlinalglstsqrscipy.optimizerrreshaperange concatenater_greaterany not_equal ImportError)x0rrr"invokerlenx0lb_cub_cmax_conx_projabxi_r ScipyBoundsScipyLinearConstraintr pc_args_ineq pc_args_eqiproject_constraintsax_ineqax_eqrs @r_projectrVsf @G"l##Yt Y  YGNNwWWXXXbz$bj111 bbbPWWX_``aaab 4:!GNNwWWXXX IE"l##Yt Y  YGNNwWWXXXbz$bj111 bbbPWWX_``aaab 4:!tyE11X__`ghhiii"l##Yt Y  YGNNwWWXXXbz$bj111 bbbPWWX_``aaab 4:!tyE11X__`ghhiii k4 ( (c(K1HCP[P`P`PbPbLcLc1c1cK13CDD2dHST\H]HeQXXY`aabbbG8$BItTl;;;TBKKK'''' R]26+h"7":[=R=U"UVVX[ \ \ ] ] ( F4LLWH $ $)@)@  ! #  " % H(=(@ @A EiooaRVAt__)>2:b>> 0L  4 035:b>>2:b>> 0L  4 069>** M M9Q<69Q<//(* S8I68TUVWXYVYTY[\[\[\T\K]7^ef(g(g(gL%)+|D/A69Q'>L_aaaa&---- * * *!D))) ) ) ) * D ! ! !!s= A44-B!4 E-F H::-I'.Pc cc98c9)__doc__numpyr_linear_constraintsrr:rintfloatgenericr$finfor)r5rVrrrr^s111111))))))UBJ' bhrz\"\"\"\"\"r