bidZddlmZddlmZmZmZddlZddl m cm Z ddl m cmZddlmZddlmZddlmZd ddZGd d eZGd d eZy)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. ) annotations)ListTupleUnionN)AffAtom)vec) ConstraintcHtj|}|jrtt |d|S|j dk(rM|j d|j dk(r.t||j dksJdt||Std)aExtracts the diagonal from a matrix or makes a vector a diagonal matrix. Parameters ---------- expr : Expression or numeric constant A vector or square matrix. k : int Diagonal in question. The default is 0. Use k>0 for diagonals above the main diagonal, and k<0 for diagonals below the main diagonal. Returns ------- Expression An Expression representing the diagonal vector/matrix. F)orderrzOffset out of bounds.z9Argument to diag must be a 1-d array or 2-d square array.) r cast_to_const is_vectordiag_vecrndimshapeabsdiag_mat ValueError)exprks R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/affine/diag.pydiagrs$   &D ~~D,a00 aDJJqMTZZ]:1v 1 %>'>>%a  TUUceZdZdZddfd ZddZddZddZddZdZ ddZ dd Z dd Z dd Z dd Z d dd ZxZS)rz)Converts a vector into a diagonal matrix.c:||_tt||yN)rsuperr__init__selfrr __class__s rr zdiag_vec.__init__< h&t,rc|jgSrrr"s rget_datazdiag_vec.get_data@ xrcR|jdjdk(s tdy)z.Checks that the argument is a vector. rrz(Argument to diag_vec must be a 1-d arrayNargsrrr's rvalidate_argumentszdiag_vec.validate_argumentsCs/yy|  A%: &rcyz$Is the atom log-log convex? Tr's ris_atom_log_log_convexzdiag_vec.is_atom_log_log_convexKrcyz%Is the atom log-log concave? Tr0r's ris_atom_log_log_concavez diag_vec.is_atom_log_log_concavePr2rcJtj|d|jS)z)r"r?_s rr@zdiag_mat.shape_from_argss2))A,$$a DFF wrc|jdjxs|jdjxr|jdk(S)zIs the expression nonnegative?r)r,rGrHrr's rrGzdiag_mat.is_nonnegs= ! &&(ADIIaL,?,?,ARtvvQR{RrcLtj|d|jgfS)aExtracts the diagonal of a matrix. 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)rNrrrOs rrRzdiag_mat.graph_implementationrSrrTrUrYrZr[)rWz np.ndarray)rWz Tuple[int]rr])r^r_r`rar r(r-r1r5r numpy_numericr;r@rGrRrbrcs@rrrs`-   ,, S 6:6.6 *6rrrT)rrVrWzUnion['diag_mat', 'diag_vec'])ra __future__rtypingrrrnumpyr8cvxpy.lin_ops.lin_oplin_opslin_oplocvxpy.lin_ops.lin_utils lin_utilsrNcvxpy.atoms.affine.affine_atomrcvxpy.atoms.affine.vecrcvxpy.constraints.constraintr rrrr0rrr|sN#%%!!$$2&3V8N6wN6b?6w?6r