bi dZddlmZddlmZddlZddlmZ ddl m Z ddl m Z ddlmZmZddlmZdd lmZd Zd Zd Zd ZdZdefdZdZGddZy)a, Copyright 2018 Akshay Agrawal Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. ) defaultdict)TupleN)reshape)vec)NonNegNonPos)Zero)canonInterfacecr|jd}|jd}t||z |jSNr constr_id)argsrr inequalitylhsrhss U/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/reductions/utilities.pylower_ineq_to_nonposr5 //! C //! C #)z';'; <<cr|jd}|jd}t||z |jSr )rrrrs rlower_ineq_to_nonnegr$rrcr|jd}|jd}t||z |jSr )rr r)equalityrrs rlower_equalityr*s5 -- C -- C c X%7%7 88rcFt|j |jS)Nr)rexprr)nonposs r nonpos2nonnegr!0s 6;;,&*:*: ;;rc|j}|jj}tj||jd}|d}t j |jd}t||t|dz|d}|gfS)NF)orderrcsc)format) _select_matshapenprsizesp eye_arrayr)rr select_mat final_shape select_vecargidentitylowereds rspecial_index_canonr34s{!!J""((KJ sCJ q'C||CHHU3Hhz*SC-@@+UXYG B;rreturnc&td|DS)Nc3^K|]%}|jD]}|j'ywN)r is_affine).0constrr0s r z"are_args_affine..@s3'6 ++'}}''s+-)all) constraintss rare_args_affiner>?s '[' ''rcltt}|D]}|t|j|!|S)a?Organize the constraints into a dictionary keyed by constraint names. Parameters --------- constraints : list of constraints Returns ------- dict A dict keyed by constraint types where dict[cone_type] maps to a list of exactly those constraints that are of type cone_type. )rlisttypeappend)r= constr_mapcs rgroup_constraintsrEDs:T"J &47""1%& rcbeZdZdZd dededdfdZd deddfdZd d ejd ede fd Z y) ReducedMataUtility class for condensing the mapping from parameters to problem data. For maximum efficiency of representation and application, the mapping from parameters to problem data must be condensed. It begins as a CSC sparse matrix matrix_data, such that multiplying by a parameter vector gives the problem data. The row index array and column pointer array are saved as problem_data_index, and a CSR matrix reduced_mat that when multiplied by a parameter vector gives the values array. The ReducedMat class caches the condensed representation and provides a method for multiplying by a parameter vector. This class consolidates code from ParamConeProg and ParamQuadProg. Attributes ---------- matrix_data : SciPy CSC sparse matrix A matrix representing the mapping from parameter to problem data. var_len : int The length of the problem variable. quad_form: (optional) if True, consider quadratic form matrix P var_len quad_formr4NcX||_||_||_d|_d|_d|_yr7) matrix_datarHrI reduced_matproblem_data_indexmapping_nonzero)selfrKrHrIs r__init__zReducedMat.__init__ms6& " #' $r keep_zerosc|jy|jtj|jjdk7rLt j |j|j|j\}}}}||_|||f|_ n|j|_d|_ |r=|j0t j|j|j|_ yyy)zCache computed attributes if not present. Parameters ---------- keep_zeros: (optional) if True, store explicit zeros in A where parameters are affected. Nr) rKrLr)prodr(r reduce_problem_data_tensorrHrIrMrNA_mapping_nonzero_rows)rOrQrLindicesindptrr(s rcachezReducedMat.cache|s    #     #wwt''--.!3"==(($,,H4 Wfe$/ +2FE*B'#'#3#3 *.' $..6#1#H#H  $,,$0D 7:r param_vec with_offsetctj|j||j|j||j S)aWraps get_matrix_from_tensor in canonInterface. Parameters ---------- param_vec: flattened parameter vector with_offset: (optional) return offset. Defaults to True. Returns ------- A tuple (A, b), where A is a matrix with `var_length` columns and b is a flattened NumPy array representing the constant offset. If with_offset=False, returned b is None. ) nonzero_rowsrZrM)r get_matrix_from_tensorrLrHrNrM)rOrYrZs rr]z!ReducedMat.get_matrix_from_tensors=44   i--##66 8 8r)F)T) __name__ __module__ __qualname____doc__intboolrPrXr)ndarrayrr]rrrGrGWsS* $S $T $d $000:8 88Y^8rrG)ra collectionsrtypingrnumpyr) scipy.sparsesparser+cvxpy.atoms.affine.reshapercvxpy.atoms.affine.vecrcvxpy.constraints.nonposrrcvxpy.constraints.zeror cvxpy.cvxcore.pythonr rrrr!r3rcr>rErGrerrrps] $.&3'/= = 9 <'D' &T8T8r