bidZddlZddlZddlmcmZddlmZ ddl m Z GddejZ eZGdd 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. N)performance_utils) unique_listceZdZdZedZejdZdZ dZ dZ ddZ dd Z d efd Zd Zd ZddZdZej(d efdZd efdZd efdZej6d efdZy) Canonicalz= An interface for objects that can be canonicalized. cft|jdk(s td|jdS)Nz6'expr' is ambiguous, there should be only one argumentr)lenargs ValueErrorselfs T/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/utilities/canonical.pyexprzCanonical.exprs,499~"UV Vyy|c"|jS)zThe graph implementation of the object stored as a property. Returns: A tuple of (affine expression, [constraints]). ) canonicalizer s rcanonical_formzCanonical.canonical_form$s  ""rct|jDcgc]}|jD]}|c}}Scc}}w)z> HsBc |in|}t||vr|t|S| |j}n#t|t|jk(sJ|j}|t |||zSt ||S)aGReturns a shallow copy of the object. Used to reconstruct an object tree. Parameters ---------- args : list, optional The arguments to reconstruct the object. If args=None, use the current args of the object. Returns ------- Expression )idr r get_datatype)r r r datas rr%zCanonical.copyKs&-R: d8z !bh' ' <99Dt9DII. ..}}  4:t - -4:t$ $rreturncy)z Determines whether the current atom is implemented in C++. This method should be overridden in derived atom classes that are not implemented in C++. Tr s r _supports_cppzCanonical._supports_cppgs rc"|jS)a Called by copy.copy() Creates a shallow copy of the object, that is, the copied object refers to the same leaf nodes as the original object. Non-leaf nodes are recreated. Constraints keep their .id attribute, as it is used to propagate dual variables. Summary: ======== Leafs: Same object Constraints: New object with same .id Other expressions: New object with new .id )r%r s r__copy__zCanonical.__copy__nsyy{rct|dd}| ||vr||St|5tj||}dddt|ddt j }|_||<|S#1swY8xYw)z Called by copy.deepcopy() Creates an independent copy of the object while maintaining the relationship between the nodes in the expression tree. r*N)getattrDefaultDeepCopyContextManagerr%deepcopyluget_idr*)r memocvxpy_idnewnew_ids r __deepcopy__zCanonical.__deepcopy__}s 4t,  H$4> !.t4 0mmD$/ 0tT4(4 DNJ  0 0s A22A;cy)zvReturns info needed to reconstruct the object besides the args. Returns ------- list Nr0r s rr+zCanonical.get_datasrc:td|jDS)zaReturns all the atoms present in the args. Returns ------- list c3JK|]}|jD]}|ywN)atoms).0ratoms r z"Canonical.atoms..s"MCM4M4Ms!#)rr r s rrCzCanonical.atomssM499MMMrc|j}|j}|jD]0}t||j }|xr|j }2||d}|S)z Aggregates and caches metrics for expression trees in self.args. So far metrics include the maximum dimensionality ('max_ndim') and whether all sub-expressions support C++ ('all_support_cpp'). )max_ndimall_support_cpp)ndimr1r max _max_ndim_all_support_cpp)r rH cpp_supportrmetricss r_aggregate_metricszCanonical._aggregate_metricssk99((* 99 AC8S]]_5H%@#*>*>*@K A !* rc(|jdS)z@The maximum number of dimensions of the sub-expression. rHrPr s rrLzCanonical._max_ndims&&(44rc(|jdS)zS Returns True if all sub-expressions support C++, False otherwise. rIrRr s rrMzCanonical._all_support_cpps&&():;;rctrB)NotImplementedErrorr s r__str__zCanonical.__str__s !##rrB)NN)r.N)__name__ __module__ __qualname____doc__propertyrpulazyproprrrrr#r%boolr1r3r>r+rC compute_oncedictrPintrLrMabcabstractmethodstrrVr0rrrrs [[##C H G ?%8t $N__D(535 <$<  $$$rr) metaclassc"eZdZdZdZdZdZy)r6ze override custom __deepcopy__ implementation and call copy.deepcopy's implementation instead c ||_d|_yrB)itemr7)r rhs r__init__z&DefaultDeepCopyContextManager.__init__s  rct|jdt|_|jturd|j_yy)Nr>)r5rh_MISSINGr7r>r s r __enter__z'DefaultDeepCopyContextManager.__enter__s3 >8D == (%)DII " )rcv|jtur'|j|j_t|_yyrB)r7rkrhr>)r exc_typeexc_valexc_tbs r__exit__z&DefaultDeepCopyContextManager.__exit__s* == (%)]]DII "$DM )rN)rWrXrYrZrirlrqr0rrr6r6s* %rr6)rZrbr%cvxpy.lin_ops.lin_utilslin_ops lin_utilsr8cvxpy.utilitiesrr\cvxpy.utilities.deterministicrABCMetarobjectrkr6r0rrrysD  $$35k$#++k$\ 8%%r