bi_/dZddlZddlZddlmZddlmZddl m cm Z dde fdZdde fdZdZdZd Zd Zd Zdd e de fd ZdZdZdZde 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. THIS FILE IS DEPRECATED AND MAY BE REMOVED WITHOUT WARNING! DO NOT CALL THESE FUNCTIONS IN YOUR CODE! N) fftconvolveis_abscN|jtjurs|j|vr3|r"t j ||jS||jSt j t j|jS|jtjur2t j t j|jSg}|jD]}|jt||| |r t||St||S)aiMultiply the expression tree by a vector. Parameters ---------- lin_op : LinOp The root of an expression tree. val_dict : dict A map of variable id to value. is_abs : bool, optional Multiply by the absolute value of the matrix? Returns ------- NumPy matrix The result of the multiplication. )typeloVARIABLEdatanpabsasmatrixzerosshapeNO_OPargsappendmul op_abs_mulop_mul)lin_opval_dictr eval_argsargs Q/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/lin_ops/tree_mat.pyrrs${{bkk! ;;( "vvhv{{344 ,,;;rxx 56 6  {{288FLL122 ;; 9C   Sh7 8 9 fi0 0&), ,c<|jtjur|j|iS|jtjuriS|r t ||}n t ||}g}|jD]}|jt||| t|S)apMultiply the transpose of the expression tree by a vector. Parameters ---------- lin_op : LinOp The root of an expression tree. value : NumPy matrix The vector to multiply by. is_abs : bool, optional Multiply by the absolute value of the matrix? Returns ------- dict A map of variable id to value. ) rrrr r op_abs_tmulop_tmulrrtmul sum_dicts)rvaluerresult result_dictsrs rrrHs${{bkk! U##    /FVU+F ;; ;C   S&& 9 : ;&&rcri}|D]/}|jD]\}}||vr |||z||<|||<1|S)zSums the dictionaries entrywise. Parameters ---------- dicts : list A list of dictionaries with numeric entries. Returns ------- dict A dict with the sum. )items)dictssum_dictrid_r s rrrks[H&"..* &JCh (  5 %  && Orc|jtjtjtjfvr|j }|S|jtj ury|jtjur t|}|S|jtjur|d }|S|jtjur t|j i}||dz}|S|jtjur t|j i}|d|z }|S|jtjurtj|d}|S|jtjur|j \}}|d||f}|S|jtj ur|dj"}|S|jtj$urt'||d}|S|jtj(ur'tj*|j,|dz}|S|jtj.ur/t1j2|d}tj4|}|St7d)a,Applies the linear operator to the arguments. Parameters ---------- lin_op : LinOp A linear operator. args : list The arguments to the operator. Returns ------- NumPy matrix or SciPy sparse matrix. The result of applying the linear operator. NrUnknown linear operator.)rr SCALAR_CONST DENSE_CONST SPARSE_CONSTr rSUMsumNEGMULrDIV SUM_ENTRIESr INDEX TRANSPOSETCONVconv_mulPROMOTEonesrDIAG_VECintf from_2D_to_1Ddiag Exception)rrr!coeffdivisorrow_slccol_slcvals rrrs' {{rHH> M;    T2 M1  q'. M-  FKK$tAw( M'  fkk2&a" M!  &Q M  !;;a')* M  $a M  &$q'* M  "&tAw. M  #  a) M233rcn|jtjtjtjfvr!t j |j}|S|jtjur|d}|S|jtjur!t|jid}||dz}|S|jtjur!t|jid}|d|z }|S|jtjurt||dd}|St||}|S)aBApplies the absolute value of the linear operator to the arguments. Parameters ---------- lin_op : LinOp A linear operator. args : list The arguments to the operator. Returns ------- NumPy matrix or SciPy sparse matrix. The result of applying the linear operator. rT)r)rrr*r+r,r r r r/r0rr1r6r7r)rrr!r?r@s rrrs  {{rHH $ M  a M  FKKT*tAw M  fkk2t,a M  &$q'$7 M% Mrc|jtjur|}|S|jtjur| }|S|jtjurCt |j i}tj|r||z}|S|j|z}|S|jtjurt |j i}||z }|S|jtjurDtjtj|jdj|z}|S|jtj urW|j \}}tjtj"|jdj}||||f<|S|jtj$ur|j}|S|jtj&ur3tj|jdj)|}|S|jtj*urItj,|}t/|tj0rtj2|d}|S|jtj4urt7||d}|St9d)a]Applies the transpose of the linear operator to the arguments. Parameters ---------- lin_op : LinOp A linear operator. value : NumPy matrix A numeric value to apply the operator's transpose to. Returns ------- NumPy matrix or SciPy sparse matrix. The result of applying the linear operator. rT) transposer))rrr-r/r0rr r isscalarr5r1r2r r9rrr3r r4r8dotr:r= isinstancematrixasarrayr6r7r>)rr r!r?r@rArBs rrrs={{bff@ M?  < M;  FKK$ ;;u 5[F2 M/WWU]F. M-  fkk2&w( M'  &RWWV[[^%9%9:;EA$ M#  !;;RXXfkk!n&:&:;<#(w  M  $ M  "a)--e4 M  # fbii (ZZ'*F M  &%48 M233rc|jtjur|}|S|jtjurDt |j id}t j|r||z}|S|j|z}|S|jtjurt |j id}||z }|S|jtjurt||dd}|St||}|S)aRApplies the linear operator |A.T| to the arguments. Parameters ---------- lin_op : LinOp A linear operator. value : NumPy matrix A numeric value to apply the operator's transpose to. Returns ------- NumPy matrix or SciPy sparse matrix. The result of applying the linear operator. T) rrr/r0rr r rGr5r1r6r7r)rr r!r?r@s rrr s{{bff M  FKKT* ;;u 5[F MWWU]F M  fkk2t,w M  &%t4 M' MrrFc"t|ji|}ttj||g\}}|r#t j |}t||dS|j|jk\rt||dSt||dS)aMultiply by a convolution operator. arameters ---------- lin_op : LinOp The root linear operator. rh_val : NDArray The vector being convolved. transpose : bool Is the transpose of convolution being applied? is_abs : bool Is the absolute value of convolution being applied? Returns ------- NumPy NDArray The convolution. valid)modefull) rr mapr; from_1D_to_2Dr flipudrsize)rrh_valrFrconstants rr7r7.s&6;;F+H4--&/ABHf99X&68':: ==FKK 'xf= =vxf= =rct|i}|jd|jdz}tj||dS)zReturns the constant term in the expression. Parameters ---------- lin_op : LinOp The root linear operator. Returns ------- NumPy NDArray The constant term as a flattened vector. rF)rrr reshape)rrV const_sizes r get_constantr\Ps>62H"8>>!#44J ::h C 00rcz|Dcgc]}t|j}}tj|Scc}w)zReturns the constant term for the constraints matrix. Parameters ---------- constraints : list The constraints that form the matrix. Returns ------- NumPy NDArray The constant term as a flattened vector. )r\exprr hstack) constraintsc constantss rget_constr_constantrcbs40;;!aff%;I; 99Y r~s$ $!! &-$&-R ' 'F00fD0f F>>d>D1$ $4$r