bidZddlmZmZddlZddlmZddl m cm Z ddl m cmZddlmZddlmZddlmZddlmZddlmZGd d eZdd efd Zd edej<fdZy)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. )ListTupleN)AffAtom)reshape)vec) Constraint) ExpressionceZdZdZd fd Zej dZd dZde e e ffdZ de fdZ de fdZ d d e e d fde ej eeffd ZxZS) upper_tria The vectorized strictly upper-triangular entries. The vectorization is performed by concatenating (partial) rows. For example, if :: A = np.array([[10, 11, 12, 13], [14, 15, 16, 17], [18, 19, 20, 21], [22, 23, 24, 25]]) then we have :: upper_tri(A).value == np.array([11, 12, 13, 16, 17, 21]) returnc,tt| |yN)superr __init__)selfexpr __class__s W/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/affine/upper_tri.pyrzupper_tri.__init__4s i'-ctj|djdd|djd}|d|S)zB Vectorize the strictly upper triangular entries. r)nkm)np triu_indicesshape)rvalues upper_idxs rnumericzupper_tri.numeric7s@ OOfQiooa&8AQRAST ay##rc|jdjdk(r9|jdjd|jdjdk7r tdy)z5Checks that the argument is a square matrix. rrz1Argument to upper_tri must be a 2-d square array.N)argsndimr ValueErrorrs rvalidate_argumentszupper_tri.validate_arguments?s^yy|  A%1););A)>$))A,BTBTUVBW)WC *XrcT|jdj\}}||dz zdzdfS)zA vector. rrr")r#r)rrowscolss rshape_from_argszupper_tri.shape_from_argsGs3YYq\'' dd1f q !$$rcy)z$Is the atom log-log convex? Tr&s ris_atom_log_log_convexz upper_tri.is_atom_log_log_convexMrcy)z%Is the atom log-log concave? Tr-r&s ris_atom_log_log_concavez!upper_tri.is_atom_log_log_concaveRr/rr.c6tj|dgfS)aVectorized strictly upper triangular entries. Parameters ---------- arg_objs : list LinExpr for each argument. shape : tuple The shape of the resulting expression. data : Additional data required by the atom. Returns ------- tuple (LinOp for objective, list of constraints) r)lur )rarg_objsrdatas rgraph_implementationzupper_tri.graph_implementationWs& Xa[)2..r)r Nr)__name__ __module__ __qualname____doc__rr numpy_numericr r'rintr+boolr.r1loLinOprrr6 __classcell__)rs@rr r s*. $$%sCx%   6:/$S#X/ rxxj)) */rr strictctj|}|js td|jdk7r t |d}|j d}|rd|zdzdzdzdz}nd|zdzdzdz dz}t|}||dzzdz|k(s||dz zdz|k(s td  |rdnd}tj|| \}}||z|z}tj|}tj|j} tj| ||ff||z|f } t| |z||fdj S) zReshapes a vector into an upper triangular matrix in row-major order. The strict argument specifies whether an upper or a strict upper triangular matrix should be returned. Inverts cp.upper_tri. zThe input must be a vector.rF)orderrg?r"z3The size of the vector must be a triangular number.)rr)r cast_to_const is_vectorr%r$rrr<rrarangeonessizesp csc_arrayrT) rrAellrrrow_idxcol_idxP_rowsP_colsP_valsPs rvec_to_upper_trirVmsZ  # #D )D >> 677 yyA~4s# **Q-C #gkc !A %! +#gkc !A %! + AA QK1  #qAE{a'73'>NOO AqA.GW [7 "F YYs^F WWV[[ !F fvv./As|DA 1t8aV3 / 1 11rrr c||dzzdz}tj|\}}||k7}tj||z|z|||z||zg}tjtj|tj||g}tj|j t }tj|||ff||z|fS)aI Returns a coefficient matrix A that creates a symmetric matrix when multiplied with a variable vector v. That is, (A @ v).reshape((n, n)) is a symmetric matrix. Parameters ---------- n : int The length of the matrix. Returns ------- sp.csc_array The coefficient matrix. rr")dtyperF) rr concatenaterIrJrKfloatrLrM)rentriesr)r*maskrPrQrs rupper_tri_to_fullr]s 1gqjG#JD$ 4cvxpy.lin_ops.lin_utils lin_utilsr3cvxpy.atoms.affine.affine_atomrcvxpy.atoms.affine.reshapercvxpy.atoms.affine.vecrcvxpy.constraints.constraintrcvxpy.expressions.expressionr r r=rVr<rMr]r-rrrlsf!!$$2.&33L/L/^#24#2LNNNr