biTdZddlmZddlmZddlmcmZ dZ dZ dee dffdZ y) a, Copyright 2013 Steven Diamond 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. )TupleNc|dk(rtj|}d|jdz}g}|df}tjdgdgdgff|}tj ||d}|}|jstj |d}n%tj |d|jdf}||jdf}|tj|||gz }t|jdk(r%tj ||jddf}||jdf}|dz dgz} td|} t|dz } tj| | | ff|} tj | |d} ||jdf}|tj| ||gz }tjtj|gS)a\Formats all the row/column cones for the solver. Parameters ---------- t: The scalar part of the second-order constraint. X: A matrix whose rows/columns are each a cone. axis: Slice by column 0 or row 1. Returns ------- list A list of LinLeqConstr that represent all the elementwise cones. r?Tsparse)rr) lu transposeshapesp csc_array create_constreshapemul_exprlenrange create_geqsum_expr) tXaxis cone_sizeterms mat_shapet_matt_vec mul_shapeval_arrrow_arrcol_arrX_mats V/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/constraints/utilities.py format_axisr#s qy LLOAGGAJI EAI LL3%1#s,i 8E OOE9T :E E 77 1f% 1q!''!*o.EKKN+I bkk% 2 33E 177|q JJq1771:q/ *AGGAJ'I1}se#GAy!GIaK G LL'GW#56 BE OOE9T :EAGGAJ'I bkk%I. //E MM"++e, - ..cFt|}||djdz|djdf}g}t|D]7\}}t|||}|j t j ||9t jt j|gS)aFormats all the elementwise cones for the solver. Parameters ---------- vars_ : list A list of the LinOp expressions in the elementwise cones. Returns ------- list A list of LinLeqConstr that represent all the elementwise cones. r) rr enumerateget_spacing_matrixappendr rrr)vars_spacingrrivarmats r"format_elemwiser.Is%jGq**E!HNN1,=>I EE",3 GQ7 R[[c*+, MM"++e, - ..r$r .c g}g}g}t|dD];}|jd|j||z|z|j|=tj|||ff|}t j ||dS)a~Returns a sparse matrix LinOp that spaces out an expression. Parameters ---------- shape : tuple (rows in matrix, columns in matrix) spacing : int The number of rows between each non-zero. offset : int The number of zero rows at the beginning of the matrix. Returns ------- LinOp A sparse matrix constant LinOp. rrTr)rr(r r r r)r r*offsetrrr var_rowr-s r"r'r'bs"GGGq? sww/0w  ,,'7!34e r:s7$$ -/`/24eCHo4r$