biLdZddlmZmZddlZddlmcmZ ddl m Z ddl mZmZGddZeZdefdZd5d eed ffd Zd5d eed ffd Zd6d eed fd efdZdefdZdefdZdZdZdZdZd eed ffdZd eed ffdZdZ d eed ffdZ!d eed ffdZ"dZ#d eed ffdZ$d eed ffdZ%d7d eed ffdZ&dZ'd eed ffdZ(d eed ffd Z)d5d!Z*d eed ffd"Z+d8d#efd$Z,d8d#efd%Z-d&Z.d eed ffd'Z/d eed ffd(Z0d8d eed fd)eefd*Z1d+Z2d7d,Z3d7d-Z4d7d.Z5d/Z6d0Z7d1Z8d2Z9d3Z:d4Z;y)9a 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. THIS FILE IS DEPRECATED AND MAY BE REMOVED WITHOUT WARNING! DO NOT CALL THESE FUNCTIONS IN YOUR CODE! )OptionalTupleN) LinEqConstr LinLeqConstrceZdZdZddZy)CounterzaA counter for ids. Attributes ---------- count : int The current count. Ncd|_y)N)count)selfs R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/lin_ops/lin_utils.py__init__zCounter.__init__'s  )returnN)__name__ __module__ __qualname____doc__rrr rrs rrrcXtj}txjdz c_|S)zdReturns a new id and updates the id counter. Returns ------- int A new id. r ) ID_COUNTERr )new_ids r get_idr.s%  F Mrshape.cf| t}tjtj|g|S)aCreates a new internal variable. Parameters ---------- shape : tuple The (rows, cols) dimensions of the variable. var_id : int The id of the variable. Returns ------- LinOP A LinOp representing the new variable. )rloLinOpVARIABLE)rvar_ids r create_varr ;s)~ 88BKKF 33rcf| t}tjtj|g|S)zWraps a parameter. Parameters ---------- shape : tuple The (rows, cols) dimensions of the operator. Returns ------- LinOP A LinOp wrapping the parameter. )rrrPARAM)rparam_ids r create_paramr$Os*8 88BHHeR 22rsparsec|dk(r+tj}tj|s)|d}n#|rtj}ntj }tj ||g|S)agWraps a constant. Parameters ---------- value : scalar, NumPy matrix, or SciPy sparse matrix. The numeric constant to wrap. shape : tuple The (rows, cols) dimensions of the constant. sparse : bool Is the constant a SciPy sparse matrix? Returns ------- LinOP A LinOp wrapping the constant. r r )rr)r SCALAR_CONSTnpisscalar SPARSE_CONST DENSE_CONSTr)valuerr%op_types r create_constr/asS$ //{{5!$KE //.. 88GUB ..rct|jdk(xs(tj|jtdk(S)zReturns whether a LinOp is a scalar. Parameters ---------- operator : LinOp The LinOp to test. Returns ------- True if the LinOp is a scalar, False otherwise. r)dtyper )lenrr)prodintoperators r is_scalarr7s1 x~~ ! # Nrwwx~~S'IQ'NNrc|jtjtjtjtj fvS)zReturns whether a LinOp is constant. Parameters ---------- operator : LinOp The LinOp to test. Returns ------- True if the LinOp is a constant, False otherwise. )typerr(r+r,r"r5s r is_constr:s5 ==R____^^XX' ''rchtjtj|dj|dS)zAdd linear operators. Parameters ---------- operators : list A list of linear operators. Returns ------- LinOp A LinOp representing the sum of the operators. rN)rrSUMr) operatorss r sum_exprr>s' 88BFFIaL.. 4 @@rcdtjtj|j|gdS)zNegate an operator. Parameters ---------- expr : LinOp The operator to be negated. Returns ------- LinOp The negated operator. N)rrNEGrr5s r neg_exprrAs# 88BFFHNNXJ ==rc.t|t|gS)a-Difference of linear operators. Parameters ---------- lh_op : LinOp The left-hand operator in the difference. rh_op : LinOp The right-hand operator in the difference. Returns ------- LinOp A LinOp representing the difference of the operators. r>rAlh_oprh_ops r sub_exprrGs UHUO, --rcTtjj|j|j\}}}tj|j ||j |j}tj|j ||j |j}|||fS)avPromote arguments for multiplication. Parameters ---------- lh_op : LinOp The left-hand operator in the multiplication. rh_op : LinOp The right-hand operator in the multiplication. Returns ------- LinOp Promoted left-hand operator. LinOp Promoted right-hand operator. tuple Shape of the product )urmul_shapes_promoterrr9argsdata)rErFlh_shaperh_shapers r promote_lin_ops_for_mulrOs&!" : : U[[!"Hh HHUZZ5::ZZ !E HHUZZ5::ZZ !E % rcPtjtj||g|S)a;Multiply two linear operators, with the constant on the left. Parameters ---------- lh_op : LinOp The left-hand operator in the product. rh_op : LinOp The right-hand operator in the product. Returns ------- LinOp A linear operator representing the product. )rrMULrErFrs r mul_exprrSs 88BFFEE7E 22rcPtjtj||g|S)apMultiply two linear operators, with the constant on the right. Parameters ---------- lh_op : LinOp The left-hand operator in the product. rh_op : LinOp The right-hand operator in the product. shape : tuple The shape of the product. Returns ------- LinOp A linear operator representing the product. )rrRMULrRs r rmul_exprrV" 88BGGUUGU 33rct|j|j}tjtj||g|S)a(Multiply two linear operators elementwise. Parameters ---------- lh_op : LinOp The left-hand operator in the product. rh_op : LinOp The right-hand operator in the product. Returns ------- LinOp A linear operator representing the product. )maxrrrMUL_ELEMrRs r multiplyr[s3  U[[ )E 88BKK 77rcPtjtj||g|S)aPKronecker product of two matrices, where the right operand is a Variable Parameters ---------- lh_op : LinOp The left-hand operator in the product. rh_op : LinOp The right-hand operator in the product. Returns ------- LinOp A linear operator representing the Kronecker product. )rrKRON_RrRs r kron_rr^' 88BIIuugu 55rcPtjtj||g|S)aOKronecker product of two matrices, where the left operand is a Variable Parameters ---------- lh_op : LinOp The left-hand operator in the product. rh_op : LinOp The right-hand operator in the product. Returns ------- LinOp A linear operator representing the Kronecker product. )rrKRON_LrRs r kron_lrb9r_rcdtjtj|j|g|S)aDivide one linear operator by another. Assumes rh_op is a scalar constant. Parameters ---------- lh_op : LinOp The left-hand operator in the quotient. rh_op : LinOp The right-hand operator in the quotient. shape : tuple The shape of the quotient. Returns ------- LinOp A linear operator representing the quotient. )rrDIVrrDs r div_exprreKs#& 88BFFEKK%% 88rcPtjtj||gdS)aPromotes a scalar operator to the given shape. Parameters ---------- operator : LinOp The operator to promote. shape : tuple The dimensions to promote to. Returns ------- LinOp A linear operator representing the promotion. N)rrPROMOTEr6rs r promoteria 88BJJz4 88rcNtjtj||gS)z,Broadcasts operators to a common shape. )rr BROADCAST_TOr=rs r broadcast_tornss 88BOOUIr ::rcVtjtj||g||gS)zSum the entries of an operator. Parameters ---------- expr : LinOp The operator to sum the entries of. shape : tuple The shape of the sum. Returns ------- LinOp An operator representing the sum. rL)rr SUM_ENTRIES)r6raxiskeepdimss r sum_entriesrtys$ 88BNNEH:T8ztranspose..s:(..+:srp)r2rtuplerr TRANSPOSE)r6axesrs` r transposers_  8>>Q <NN4R4(E:T::Exx ehZtfEErcPtjtj||gdS)a Reshapes an operator. Parameters ---------- operator : LinOp The operator to reshape. shape : tuple The (rows, cols) of the reshaped operator. Returns ------- LinOp LinOp representing the reshaped expression. N)rrRESHAPErhs r reshaperrjrkc|jdt|z}||f}tjtj||g|S)aConverts a vector to a diagonal matrix. Parameters ---------- operator : LinOp The operator to convert to a diagonal matrix. k : int The offset of the diagonal. Returns ------- LinOp LinOp representing the diagonal matrix. r)rabsrrDIAG_VEC)r6rrowsrs r diag_vecrs? >>! s1v %D 4LE 88BKK A 66rc|jdt|z df}tjtj||g|S)aConverts the diagonal of a matrix to a vector. Parameters ---------- operator : LinOp The operator to convert to a vector. k : int The offset of the diagonal. Returns ------- LinOp LinOp representing the matrix diagonal. rr )rrrrDIAG_MAT)r6rrs r diag_matrs;^^A Q ' +E 88BKK A 66rc|jd|jdz}||jdz dzdf}tjtj||gdS)zVectorized upper triangular portion of a square matrix. Parameters ---------- operator : LinOp The matrix operator. Returns ------- LinOp LinOp representing the vectorized upper triangle. rr rN)rrr UPPER_TRI)r6entriesrs r upper_trirsXnnQq 11Gq))A -q 1E 88BLL%(T ::rcNtjtj||dS)aConcatenates operators horizontally. Parameters ---------- operator : list The operators to stack. shape : tuple The (rows, cols) of the stacked operators. Returns ------- LinOp LinOp representing the stacked expression. N)rrHSTACKrms r hstackr) 88BIIui 66rcNtjtj||dS)aConcatenates operators vertically. Parameters ---------- operator : list The operators to stack. shape : tuple The (rows, cols) of the stacked operators. Returns ------- LinOp LinOp representing the stacked expression. N)rrVSTACKrms r vstackr;rrrrcPtjtj|||gS)aoConcatenate operators on axis. Parameters ---------- operator : list The operators to concatenate. shape : tuple The (rows, cols) of the concatenated operators. axis : int, optional The axis along which the operators will be joined. Returns ------- LinOp LinOp representing the stacked expression. )rr CONCATENATE)r=rrrs r concatenaterLs" 88BNNE9tf ==rc6||St|t|gS)z,Returns the operator in the constraint. rCrDs r get_constr_exprr_s" } 011rc`| t}t||}t|||jS)aqCreates an internal equality constraint. Parameters ---------- lh_term : LinOp The left-hand operator in the equality constraint. rh_term : LinOp The right-hand operator in the equality constraint. constr_id : int The id of the CVXPY equality constraint creating the constraint. Returns ------- LinEqConstr )rrrrrErF constr_idexprs r create_eqris0 H 5% (D tY 44rc`| t}t||}t|||jS)apCreates an internal less than or equal constraint. Parameters ---------- lh_term : LinOp The left-hand operator in the <= constraint. rh_term : LinOp The right-hand operator in the <= constraint. constr_id : int The id of the CVXPY equality constraint creating the constraint. Returns ------- LinLeqConstr )rrrrrs r create_leqrs0 H 5% (D i 55rcH| t|}tt|||S)asCreates an internal greater than or equal constraint. Parameters ---------- lh_term : LinOp The left-hand operator in the >= constraint. rh_term : LinOp The right-hand operator in the >= constraint. constr_id : int The id of the CVXPY equality constraint creating the constraint. Returns ------- LinLeqConstr )rAr)rErFrs r create_geqrs&   huoui 88rc|jtjk(r|j|jfgSg}|j D]}|t |z }|S)zGet a list of the variables in the operator and their shapes. Parameters ---------- operator : LinOp The operator to extract the variables from. Returns ------- list A list of (var id, var shape) pairs. )r9rrrLrrK get_expr_vars)r6vars_args r rrsV}} #/00== (C ]3' 'E ( rc.|jtjk(r|jj Sg}|j D]}|t |z }t|jtjr|t |jz }|S)zGet a list of the parameters in the operator. Parameters ---------- operator : LinOp The operator to extract the parameters from. Returns ------- list A list of parameter objects. ) r9rr"rL parametersrKget_expr_params isinstancer)r6paramsrs r rrsz}} }}''))== +C oc* *F + hmmRXX . ohmm4 4F rct||j}t|||j|jS)aOCreates a copy of the constraint modified according to func. Parameters ---------- constr : LinConstraint The constraint to modify. func : function Function to modify the constraint expression. Returns ------- LinConstraint A copy of the constraint with the specified changes. )rr9rr)constrfuncrs r copy_constrrs1  D 4<f.. ==rc>|jtjk(r|j|vr||jSg}|jD]}|j t ||tj|j|j||jS)a@Replaces the given variables in the expression. Parameters ---------- expr : LinOp The expression to replace variables in. id_to_new_var : dict A map of id to new variable. Returns ------- LinOp An LinOp identical to expr, but with the given variables replaced. ) r9rrrLrKappendreplace_new_varsrr)r id_to_new_varnew_argsrs r rrs yyBKKDII$>TYY''99 C OO m4  xx 4::xCCrc8|j}| td|S)a*Wrapper on accessing a parameter. Parameters ---------- param : Parameter The parameter whose value is being accessed. Returns ------- The numerical value of the parameter. Raises ------ ValueError Raises error if parameter value is None. z$Problem has missing parameter value.)r- ValueError)paramvals r check_param_valrs#" ++C {?@@ rc|jtjk(r)tt |j |j Sg}|jD]}|jt|t|j tjr{|j jtjk(rT|j }t|j tsJt |j }t||j }n |j }tj|j|j ||S)zReplaces parameters with constant nodes. Parameters ---------- expr : LinOp The expression to replace parameters in. Returns ------- LinOp An LinOp identical to expr, but with the parameters replaced. ) r9rr"r/rrLrrKrreplace_params_with_constsrrr)rrr data_lin_oprrLs r rrs yyBHHODII6 CC99 =C OO6s; < = dii *tyy~~/I))Kk//7 77!+"2"23C[%6%67D99Dxx 4::x>>rr)F)NN)r)rArGrOrSrVr[r^rbrerirnrtrwr{r~rrrrrrrrrrrrrrrrrrrrr rs%$#!!C  Y   4eCHo4(3c3h3$/uS#X//< O4 O'$'$ A > .$83%S/3$45c?4(8&6c3h6$6c3h6$9,9U38_9$;5c?; NsCxN$ 8 75c?7(4eCHo4(F69U38_9$7#7(7#7&;$7U38_7$7U38_7">%S/>#>&25,6,9*,2>&D40?r