biIdZddlZddlmZmZddlZddlmZ ddl m cm Z ddlm cmZddlmZddlmZddlmZddlmZGddeZGd d 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. N)ListTuple)AffAtom) Constraint) is_param_freeceZdZdZdfd Zej dZddZde fdZ de fdZ de e e ffdZde e e ffd Zde fd Zde fd Z dd e e d fde ej&eeffdZxZS)conva 1D discrete convolution of two vectors. The discrete convolution :math:`c` of vectors :math:`a` and :math:`b` of lengths :math:`n` and :math:`m`, respectively, is a length-:math:`(n+m-1)` vector where .. math:: c_k = \sum_{i+j=k} a_ib_j, \quad k=0, \ldots, n+m-2. Parameters ---------- lh_expr : Constant A constant 1D vector or a 2D column vector. rh_expr : Expression A 1D vector or a 2D column vector. returncbtjdttt|||y)Nz)conv is deprecated. Use convolve instead.)warningswarnDeprecationWarningsuperr __init__)selflh_exprrh_expr __class__s R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/affine/conv.pyrz conv.__init__2s$ ACUV dD"7G4ctttj|}t j |d|d}|dj dk(s|dj dk(r |dddfS|S)!Convolve the two values. rN)listmapintf from_2D_to_1Dnpconvolvendim)rvalues flat_valuesoutputs rnumericz conv.numeric6se3t116:; [^[^< !9>>Q &)..A"5!T'? "Mrc|jdjr|jdjs td|jdjs tdy)KChecks that both arguments are vectors, and the first is constant. rrz1The arguments to conv must resolve to a 1-d array,The first argument to conv must be constant.N)args is_vector ValueError is_constantrs rvalidate_argumentszconv.validate_argumentsAs\yy|%%'tyy|/E/E/GPQ Qyy|'')KL L*rctjjr,|jd}|j xr t |S|jdj SzIs the atom convex? ruscopesdpp_scope_activer)r,rrcs ris_atom_convexzconv.is_atom_convexIN 88 $ $ & ! A==?7}Q'7 799Q<++- -rc"|jSzIs the atom concave? r7r-s ris_atom_concavezconv.is_atom_concaveT""$$rc|jdj}|jdj}||zdz }|jdjdk(s|jdjdk(r|dfS|fS)0The sum of the argument dimensions - 1. rrr)r)sizer!)r lh_length rh_length output_lengths rshape_from_argszconv.shape_from_argsYsxIIaL%% IIaL%% !I-1 99Q<   !TYYq\%6%6!%;!1% %!# #rcvtjj|jd|jdSzSame as times. rrr2signmul_signr)r-s rsign_from_argszconv.sign_from_argsd)vvtyy|TYYq\::rc<|jdjSz;Is the composition non-decreasing in argument idx? rr) is_nonnegridxs ris_incrz conv.is_incriyy|%%''rc<|jdjSz;Is the composition non-increasing in argument idx? rr) is_nonposrPs ris_decrz conv.is_decrnrSrshape.c@tj|d|d|gfSatConvolve two vectors. Parameters ---------- arg_objs : list LinExpr for each argument. shape : tuple The shape of the resulting expression. data : Additional data required by the atom. Returns ------- tuple (LinOp for objective, list of constraints) rrlur rarg_objsrYdatas rgraph_implementationzconv.graph_implementations$& Xa[%8"==rr NN)__name__ __module__ __qualname____doc__rr numpy_numericr%r.boolr7r<rintrDrJrRrXloLinOprrra __classcell__)rs@rr r s$5 M . .%% $sCx $;dDj 1; (d( (d( 6:>$S#X> rxxj)) *>rr ceZdZdZej dZddZdefdZ defdZ de e e ffdZ de eeffd Zdefd Zdefd Z dd e e d fde ej$eeffdZy)r a 1D discrete convolution of two vectors. The discrete convolution :math:`c` of vectors :math:`a` and :math:`b` of lengths :math:`n` and :math:`m`, respectively, is a length-:math:`(n+m-1)` vector where .. math:: c_k = \sum_{i+j=k} a_ib_j, \quad k=0, \ldots, n+m-2. Matches numpy.convolve Parameters ---------- lh_expr : Constant A constant scalar or 1D vector. rh_expr : Expression A scalar or 1D vector. c:tj|d|dS)rrr)rr )rr"s rr%zconvolve.numerics{{6!9fQi00rr Nc|jdjdkr|jdjdks td|jdjs tdy)r'rrz+The arguments to conv must be scalar or 1D.r(N)r)r!r+r,r-s rr.zconvolve.validate_argumentss`yy|  A%TYYq\->->!-CJK Kyy|'')KL L*rctjjr,|jd}|j xr t |S|jdj Sr0r1r5s rr7zconvolve.is_atom_convexr8rc"|jSr:r;r-s rr<zconvolve.is_atom_concaver=rcx|jdj}|jdj}||zdz fS)r?rr)r)r@)rrArBs rrDzconvolve.shape_from_argss=IIaL%% IIaL%% I%)++rcvtjj|jd|jdSrFrGr-s rrJzconvolve.sign_from_argsrKrc<|jdjSrMrNrPs rrRzconvolve.is_incrrSrc<|jdjSrUrVrPs rrXzconvolve.is_decrrSrrY.c@tj|d|d|gfSr[r\r^s rrazconvolve.graph_implementationrbrrcrd)rerfrgrhrrir%r.rjr7r<rrkrDrJrRrXrlrmrrrarrr r s, 11 M . .%% ,sCx,;dDj 1; (d( (d( 6:>$S#X> rxxj)) *>rr )rhr typingrrnumpyrcvxpy.interface interfacercvxpy.lin_ops.lin_oplin_opslin_oprlcvxpy.lin_ops.lin_utils lin_utilsr]cvxpy.utilities utilitiesr2cvxpy.atoms.affine.affine_atomrcvxpy.constraints.constraintr%cvxpy.expressions.constants.parameterrr r ryrrrsM !!$$23?g>7g>T^>w^>r