bi,dZddlmZmZmZddlmZddlm cm Z ddl m cm ZddlmZddlmZddlmZddlmZddlmZGdd 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. )AnyListTupleN)Atom)canonInterface)Constant)performance_utilsceZdZdZdZdeeeffdZdefdZdefdZ defdZ defdZ defd Z defd Z defd Zdefd Zdefd ZdefdZej&defdZej&defdZdeefdZy)AffAtomz' Abstract base class for affine atoms. Treturncztjj|jDcgc]}|c}Scc}w)zLBy default, the sign is the most general of all the argument signs. )usign sum_signsargs)selfargs Y/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/affine/affine_atom.pysign_from_argszAffAtom.sign_from_args!s+vv : :;; :s 8c:td|jDS)z%Is the expression imaginary? c3<K|]}|jywN)is_imag.0rs r z"AffAtom.is_imag..*6S3;;=6allrrs rrzAffAtom.is_imag&s6DII666c:td|jDS)z*Is the expression complex valued? c3<K|]}|jywr) is_complexrs rrz%AffAtom.is_complex..0s93>>#9ranyrr!s rr%zAffAtom.is_complex,s9tyy999r"cy)zIs the atom convex? Tr!s ris_atom_convexzAffAtom.is_atom_convex2r"cy)zIs the atom concave? Tr)r!s ris_atom_concavezAffAtom.is_atom_concave7r+r"cy)z;Is the composition non-decreasing in argument idx? Tr)ridxs ris_incrzAffAtom.is_incr<sr"cy)z;Is the composition non-increasing in argument idx? Fr)r/s ris_decrzAffAtom.is_decrBsr"c:td|jDS)Nc3<K|]}|jywr) is_quadraticrs rrz'AffAtom.is_quadratic..Is;#3##%;rrr!s rr6zAffAtom.is_quadraticHs;;;;r"c:td|jDS)aDoes the affine head of the expression contain a quadratic term? The affine head is all nodes with a path to the root node that does not pass through any non-affine atom. If the root node is non-affine, then the affine head is the root alone. c3<K|]}|jywr)has_quadratic_termrs rrz-AffAtom.has_quadratic_term..RsA3))+Arr&r!s rr9zAffAtom.has_quadratic_termKsAtyyAAAr"c:td|jDS)Nc3<K|]}|jywr)is_qpwars rrz"AffAtom.is_qpwa..Urrrr!s rr<zAffAtom.is_qpwaTs6DII666r"c:td|jDS)Nc3<K|]}|jywr)is_pwlrs rrz!AffAtom.is_pwl..Xs5C3::<5rrr!s rr?zAffAtom.is_pwlWs5499555r"ct|jD]J\}}|j|r|jr(|j |r|j rJyyz:Is the expression a positive semidefinite matrix? FT) enumeraterr1is_psdr3is_nsdrr0rs rrCzAffAtom.is_psd[P"$)), HC\\#&3::<\\#&3::< r"ct|jD]J\}}|j|r|jr(|j |r|j rJyyrA)rBrr3rCr1rDrEs rrDzAffAtom.is_nsderFr"cg}i}d}t|jD]t\}}|jr'|t|jj dgz }=|t j|j|gz }|||<||jz }v|}|j||j|j\}} tjdi} tjdi} tj|g||| | |j} |dz|jf} | j!| j#ddddf}g}d}|jD]k}|jr9|j| df}|dk(r|dgz }0|t%j&|dgz }L||jz}||||ddfgz }|}m|S)a+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. rN)rIrIfloat64)dtype)rBr is_constantrvaluecanonical_formlu create_varshapesizegraph_implementationget_datalo CONSTANT_IDrget_problem_matrixreshapetocscsp coo_matrix)rvalues fake_args var_offsetsoffsetr0r var_length fake_expr_ param_to_size param_to_col canon_matrR stacked_grad grad_liststart grad_shapestops r_gradz AffAtom._grados  !$)), #HC hsyy1@@CDD bmmCIIs;<< #) C #(("  # 00DJJ15B 1+ * "55 K     II  a+ ((/557Q?  99 C !hha1 '!$I"-- )"L!MMIsxx'l5:q=9::  r"N)__name__ __module__ __qualname____doc___allow_complexrboolrrr%r*r-r1r3r6r9r<r?perf compute_oncerCrDrrrlr)r"rr r s1N<dDj 1< 77 :D:   d d rs>$#!!$$!/05IdIr"