bidZddlmZddlmZmZddlZddlm Z ddlm Z ddl mZddlmZddlmZGd d eZeed Zy) 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. )wraps)ListTupleN)linalg)Atom)QuadForm) ConstraintceZdZdZdZdfd ZdZdeefdZ dZ ddZ de e d ffd Zde eeffd Zdefd Zdefd ZdefdZdefdZdefdZdefdZdefdZxZS) MatrixFracz tr X.T*P^-1*X Treturnc.tt| ||y)N)superr __init__)selfXP __class__s R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/matrix_frac.pyrzMatrixFrac.__init__!s j$(A.c|d}|d}|jdjrQtj|jj t j|j |}n=|jj t j|j |}t|jdk(r|jS|S)zReturns tr X.T*P^-1*X. r) args is_complexnpconjTdotLAinvlenshapetrace)rvaluesrrproducts rnumericzMatrixFrac.numeric$s 1I 1I 99Q< " " $ggajll&&rvvay155a8GccggbffQi(,,Q/G"%gmm"4"9w}}FwFrc(|jddz gS)z?Returns constraints describing the domain of the node. rr)rrs r_domainzMatrixFrac._domain0s ! !""rctj|d}|jdk(r |dddf}tj|d} tj|}tj |tj |z|}|jjd}tj|gj}tj ||}tj ||j}tj ||}|j }tj|jjdgj}||gS#tj$rddgcYSwxYw)a4 Gives the (sub/super)gradient of the atom w.r.t. each argument. Matrix expressions are vectorized, so the gradient is a matrix. Args: values: A list of numeric values for the arguments. Returns: A list of SciPy CSC sparse matrices or None. rrNF)order) rarrayndimrr r transposerravelsp csc_array LinAlgError)rr$rrP_invDXDPs r_gradzMatrixFrac._grad5s( HHVAY  66Q;!T' A HHVAY  FF1IE bll51115B#&Brd#%%Bq!BACCBE"B$$Brttzzz45688B8O~~ $<  s EE87E8c|jd}|jd}|jdk7s|jd|jdk7r td|jd|jdk7r tdy)z5Checks that the dimensions of x and P match. rrrz;The second argument to matrix_frac must be a square matrix.z:The arguments to matrix_frac have incompatible dimensions.N)rr.r" ValueError)rrrs rvalidate_argumentszMatrixFrac.validate_argumentsWs IIaL IIaL 66Q;!''!* 2M WWQZ1771: %L &r.ctS)z8Returns the (row, col) shape of the expression. )tupler(s rshape_from_argszMatrixFrac.shape_from_argses wrcy)zCReturns sign (is positive, is negative) of the expression. )TFr(s rsign_from_argszMatrixFrac.sign_from_argsjsrcy)zIs the atom convex? Tr?r(s ris_atom_convexzMatrixFrac.is_atom_convexosrcy)zIs the atom concave? Fr?r(s ris_atom_concavezMatrixFrac.is_atom_concavetrcy)z;Is the composition non-decreasing in argument idx? Fr?ridxs ris_incrzMatrixFrac.is_incryrErcy)z;Is the composition non-increasing in argument idx? Fr?rGs ris_decrzMatrixFrac.is_decr~rErcz|jdjxr|jdjS)z4Quadratic if x is affine and P is constant. rr)r is_affine is_constantr(s r is_quadraticzMatrixFrac.is_quadratics1yy|%%'FDIIaL,D,D,FFrc<|jdjS)z)Quadratic term if P is constant. r)rrNr(s rhas_quadratic_termzMatrixFrac.has_quadratic_termsyy|''))rcz|jdjxr|jdjS)zEQuadratic of piecewise affine if x is PWL and P is constant. rr)ris_pwlrNr(s ris_qpwazMatrixFrac.is_qpwas1yy|""$C1)A)A)CCr)r N)__name__ __module__ __qualname____doc___allow_complexrr&rr r)r7r:rintr=boolr@rBrDrIrKrOrQrT __classcell__)rs@rr r sN/ G#j)# D sCx dDj 1   d d GdG *D* DDrr ct|tjrDtj|}t ||tj |jzdz St||S)Ng@) isinstancerndarrayrr rrrr )rrinvPs r matrix_fracrasM!RZZ vvayD2774=??2c9::!Qr)rX functoolsrtypingrrnumpyr scipy.sparsesparser1rrcvxpy.atoms.atomrcvxpy.atoms.quad_formrcvxpy.constraints.constraintr r rar?rrrjsN !*3sDsDlz  r