biP5dZddlmZddlZddlmZddlZddlm Z ddl m Z ddl mZddlmZmZddlmZdd lmZmZGd d Zy) a+ Copyright, the CVXPY authors 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. ) annotationsN)List)canonInterface)TensorRepresentation)NO_OPLinOp) InverseData)replace_quad_formsrestore_quad_formscVeZdZddZdZdZdZ d dZ d dZy) CoeffExtractorc|j|_|j|_|j|_|j|_|j |_|j |_||_y)N) var_offsetsid_mapx_length var_shapes param_shapes param_to_size param_id_map canon_backend)self inverse_datars Z/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/utilities/coeff_extractor.py__init__zCoeffExtractor.__init__&s[".. $-- &11(55)77(55*c t|tr|}n|g}t|Dcgc]}|jc}sJt |Dcgc]}|j c}}|Dcgc]}|j d}}tj||j|j|j|j||jScc}wcc}wcc}w)aExtract problem data tensor from an expression that is reducible to A*x + b. Applying the tensor to a flattened parameter vector and reshaping will recover A and b (see the helpers in canonInterface). Parameters ---------- expr : Expression or list of Expressions. The expression(s) to process. Returns ------- SciPy CSR matrix Problem data tensor, of shape (constraint length * (variable length + 1), parameter length + 1) r) isinstancelistallis_dppsumsizecanonical_formrget_problem_matrixrrrrr)rexpr expr_listenum_rowsop_lists raffinezCoeffExtractor.affine/s$ dD !II 21AHHJ2333 212309:11##A&::001515151C1C151B1B19151C1C E E32:sC  C'Cc h |jsJtj|j\}}}}|jdg}t j ||||j|j|j|j}|dgddf} |ddddfj} |jd} i} |jD]} | j|vr| j}||djdj}||d}||d}| |||zddf}||dj}|j Jdt#j$|r5t'|t"j(s|j j+}nt#j(|j }|dk(r|ddk7}|j,dddf|dd|fz}t/j0| |}t3|j5dt/j6|j8t;|t/j6|j<t;|t/j>|t;|j,|j}ny|j<|j8k(jAsJd ||z}t/jB|\}}|||f}t3|||jE||j}|| vr$d | |vr| |d |z| |d <b|| |d <ltG| |<|| |d <|jd| jdf}| dk(rt/jH|| |d <t#j(gggff| | |d <|| jd}t/jJ|| jtL }| j| vrU| dk(r(| | jd xx| |||zddfz cc<Y| | jd xx||||zddfz cc<tG| | j<| dk(r | |||zddf| | jd <||||zddf| | jd <| | fS)zn Assumes quadratic forms all have variable arguments. Affine expressions can be anything. rNzFP matrix must be instantiated before calling extract_quadratic_coeffs.F)orderz?Only diagonal P matrices are supported for multiple quad forms.Pq)shapedtype)'r r get_var_offsets variablesr#rr$rrr"rtoarrayr3idargsr1valuespissparser coo_matrixtocoodatanparangerflattentilerowlencolrepeatrnonzerocopydictzerosprodint)r affine_expr quad_forms affine_id_mapaffine_offsetsraffine_var_shapesr) param_coeffsconstantc num_paramscoeffsvarvar_idorig_id var_offsetvar_sizec_partr1 nonzero_idxsr@ param_idxsP_tup scaled_c_partparamx_idx_row param_idx_colc_valsr3s rextract_quadratic_coeffsz'CoeffExtractor.extract_quadratic_coeffsPs!!###  ' ' (=(=(? @ C ~x1B--a01%889A9G9=9K9K9=9J9J9D9I9I9=9K9K M  a( "a ( ( *!''*  ((*W ^C vv#$V,Q/44Q7::*6215 (03:j&991<=v&q)++GG'\[\';;q>*Q *F A agg.Aq=$*!9>L66!T'?VA|O-DDD!#:!6|!DJ0 3 /s:7s:7 *c!&&k: EEEQUUN//1ZYZ1%&JM46JJ}4M1NM#NM$ABF0&&++-% Ef$fWo-06w0Du0Lw,/4w,'+fF7O+0F7OC(WWQZ4E!Q/1xxw,/1}}b2r(^SX/Yw,+366215 77#4SVV#appendr ValueError merge_P_list merge_q_list)rr%rootrPrXrUoffsetsrWP_listq_listP_height P_entriesrZ_r3r"r1r2s r quad_formzCoeffExtractor.quad_forms UDJJ3(b1  8819CE 499Q<4**,(2E2Ea2HI^^A&    IFAOOF+E775,DC6&>$96N3'QVV[[( (994,GC6&>$96N3'?$ !34A rB8ntZ6HIA MM!  MM!   H' * t}} $89 9   fh ;   fh ;!t rc0d}|D]j}|j\}}||k(sJ|j|jusJ|xj|z c_|xj|z c_||f|_||z }ltj|}|j |S)aZConceptually we build a block diagonal matrix out of all the Ps, then flatten the first two dimensions. eg P1 P2 We do this by extending each P with zero blocks above and below. Args: P_list: list of P submatrices as TensorRepresentation objects. P_entries: number of entries in the merged P matrix. P_height: number of rows in the merged P matrix. num_params: number of parameters in the problem. Returns: A CSC sparse representation of the merged P matrix. r)r3rErGrcombineflatten_tensor) rrtrvrWoffsetr1mncombineds rrpzCoeffExtractor.merge_P_list s, A77DAq6M655% %% EEVOE EEVOE*AG aKF (//7&&z22rc|dk(rOtj|}tj||jg}tj|Stj||gz}tj|S)afStack q with constant offset as last row. Args: q_list: list of q submatrices as SciPy sparse matrices or NumPy arrays. constant: The constant offset as a CSC sparse matrix. num_params: number of parameters in the problem. Returns: A CSR sparse representation of the merged q matrix. r-)rAvstackr8r< csr_array)rrurUrWr2s rrqzCoeffExtractor.merge_q_list2sf" ? &!A 1h..012A<<? " &H:-.A<<? "rN)rz str | NonereturnNone)rtzList[TensorRepresentation]rvrNrWrNr sp.csc_array)ruzList[sp.spmatrix | np.ndarray]rUrrWrNrz sp.csr_array) __name__ __module__ __qualname__rr*rfryrprqrrr r $sr+EB@ D7r%3.%3%3 %3  %3N#.## #  #rr )__doc__ __future__rrktypingrnumpyrA scipy.sparsesparser<cvxpy.cvxcore.pythonrcvxpy.lin_ops.canon_backendrcvxpy.lin_ops.lin_oprrcvxpy.reductions.inverse_datar "cvxpy.utilities.replace_quad_formsr r r rrrrs8 #/<-5e#e#r