biddlmZddlmZddlZej ZejZejZ dZ dZ dZ dZdZd Zd Zd ZdZd Zd ZdZGddeZdZy))sparse) _scs_directNFiicb|jddr td|jddr&|jdtrddlm}|Sddlm}|S|jd dr%|jddr t d dd lm}|S|jdtrdd lm}|StS) NcudssFz1To use cuDSS set gpu=True and use_indirect=False.gpu use_indirectr)_scs_gpu) _scs_cudssmklzAMKL indirect solver not yet available, pass `use_indirect=False`.)_scs_mkl) _scs_indirect) pop ValueError_USE_INDIRECT_DEFAULTscsrrNotImplementedErrorrrr)stgsrrrrs G/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/scs/__init__.py_select_scs_modulers XXgu H II XXeU xx 56 o  XXeU xx&  M  O XXn34!  c"eZdZdZddZddZy)SCSc b||_|r|s tdd|vsd|vr tdd|vr td|d}|d}|d}||| tdtj|s t d |j d k(s%t jd |j}tj|r|j}tj|r|j}t|}t|}|js|j|j|j|j} } } |j ||fk7r td d \} } }d|vr|d}|tj|s t d|j ||fk7r td|j d k(s%t jd|j}tj"|djj$dkDrtj&|d }|js|j|j|j|j}} } t)|j}|j*||f| | | | | ||||f i|j|_y)zInitialize the SCS solver. @param data Dictionary containing keys `P`, `A`, `b`, `c`. @param cone Dictionary containing cone information. @param settings Settings as kwargs, see docs. z Missing data or cone informationbcz(Missing one of b, c from data dictionaryAzMissing A from data dictionaryNzIncomplete data specificationz#A is required to be a sparse matrixcsczJConverting A to a CSC (compressed sparse column) matrix; may take a while.zA shape not compatible with b,c)NNNPz#P is required to be a sparse matrixz!P shape not compatible with A,b,czJConverting P to a CSC (compressed sparse column) matrix; may take a while.r r)format) _settingsrrissparse TypeErrorr&warningswarntocsctodenselenhas_sorted_indices sort_indicesdataindicesindptrshapetrilsizetriurr_solver)selfr1conesettingsr#r!r"mnAdataAindicesAcolptrPdataPindicesPcolptrr%_scss r__init__z SCS.__init__>stDN t 9 :: $#T/ A BB $ 7 88 S A S A S AyAI 6 77 ??1  ; << 88u mm  '')a q ))+a q ))+a AA AA  nn vvqyy!((W8Eww1a& 8 99/E8W d{ s)a q!?@ @ 77q!f >? ?xx5 --* ggi! ;;q"  " " ' '! +kk!E*!## .. #$66199ahhx dnn -D488 A       .. DLrNc>|jj||||S)aSolve the optimization problem. @param warm_start Whether to warm-start. By default the solution of the previous problem is used as the warm-start. The warm-start can be overridden to another value by passing `x`, `y`, `s` args. @param x Primal warm-start override. @param y Dual warm-start override. @param s Slack warm-start override. @return dictionary with solution with keys: 'x' - primal solution 's' - primal slack solution 'y' - dual solution 'info' - information dictionary (see docs) )r8solve)r9 warm_startxyss rrGz SCS.solves" <<  j!Q 22rc<|jj||y)aUpdate the `b` vector, `c` vector, or both, before another solve. After a solve we can reuse the SCS workspace in another solve if the only problem data that has changed are the `b` and `c` vectors. @param b New `b` vector. @param c New `c` vector. N)r8update)r9r!r"s rrMz SCS.updates LL1r)TNNN)NN)__name__ __module__ __qualname__rErGrMrrrr<sTl3& rrc t||fi|}dx}x}}d|vr|d}d|vr|d}d|vr|d}|jd|||S)NrIrJrKT)rHrIrJrK)rrG)r1r:r;solverrIrJrKs rrGrGsk tT &X &&!a!D[ S AD[ S AD[ S A a1 55r)scipyrrrr*version __version__ sizeof_int__sizeof_int__ sizeof_float__sizeof_float__rINFEASIBLE_INACCURATEUNBOUNDED_INACCURATESIGINTFAILED INDETERMINATE INFEASIBLE UNBOUNDED UNFINISHEDSOLVEDSOLVED_INACCURATErobjectrrGrQrrrfs!k!!# ''')+;++-       @t&tp 6r