bidZddlZddlmZmZddlZdeedffdZdZ dee fd Z dd Z dd e fd Zde fd ZdZdZde fdZdeedfdeedffdZdZde fdZdZddZy)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)OptionalTupleshape.c t|}td|Dr tdt|dk(r tdt d|D}t||z }|t|kDr td|t|kr2t t |tdddgt||z zz}t dt||tt|DS) aCheck if the key is a valid index. Args: key: The key used to index/slice. shape: The shape (rows, cols) of the expression. Returns: The key as a tuple of slices. Raises: Error: Index/slice out of bounds. c3K|]z}t|txsdt|txrRt|jtxs6t|jtxst|j t|ywN) isinstancefloatslicestartstopstep).0ks T/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/utilities/key_utils.py zvalidate_key..)sm 6-. a  (Jq%$8%' qww & & qvvu % & qvvu % ( 6sBBzfloat is an invalid index type.rzAn index cannot be empty.c3&K|] }|d yw)Nrelems rrzvalidate_key..1s74$,Q7sz Too many indices for expression.Nc3>K|]\}}}t|||ywr) format_slice)rslcdimis rrzvalidate_key..9sa{sCc3*as) to_tupleany IndexErrorlensumtuplelistr ziprange)keyr none_countslicess r validate_keyr)s 3-C  625 66:;; 3x1}4557377J X "F E ;<< #e* DItT4!8 93u:;N OOP ac#ueTWX]T^N_>`a aac,t|tr|S|fS)z'Convert key to tuple if necessary. )r r"r&s rrr<s#u v r*returnc |yt|trt|jd}|dk(r t d|dkDrkt j tt|jd|d|}t j tt|j||d|}nxt j tt|j|dz |d|dz }t j tt|j| dz |dd|dz }t|||St|}t||}d|cxkr|krnnt||dzdStd|||fz)aConverts part of a key into a slice with a start and step. Uses the same syntax as numpy. Args: key_val: The value to convert into a slice. dim: The length of the dimension being sliced. Returns: A slice with a start and step. Nrrzstep length cannot be 0Tz3Index %i is out of bounds for axis %i with size %i.) r r to_intr ValueErrornpclipwrap_neg_indexr r r)key_valraxisrr r orig_key_vals rrrEsO GU #gllA& 1967 7 AXGGN6'--+CSI1cRE77>&s*CSI1cRDGGN6'--Q+GMrSVWXSXYE77>&tAv*FTRTVX[\]X]^DUD$''g  s3  # 'A+q1 1EW[]`Haa r*c ||St|S)z+Convert everything but None to an int. )int)valnone_vals rr0r0js {3xr*neg_stepc,||dkr |r|dk(s||z }|S)zConverts a negative index into a positive index. Args: index: The index to convert. Can be None. dim: The length of the dimension being indexed. rr/r)indexrr<s rr4r4ss' UQY "   Lr*c"t||dzdS)zConverts an index to a slice. Args: idx: int The index. Returns: slice A slice equivalent to the index. rN)r )idxs rindex_to_slicerAs c!eT ""r*c,t|rt|jS|j|jfDcgc] }t |}}|j )|j dk7r|dd|dd|j S|dd|dScc}w)z$Converts a slice into a string. rr:)is_single_indexstrr r none_to_emptyr)rr: endpointss r slice_to_strrHss399~03 388/DEs#EIE xxA &q\9Q<BB#A, ! 55 FsBc |y|S)z&Converts None to an empty string. r)r:s rrFrFs { r*c|jd}n |j}|jduxr,|jduxr|j|z|jk\S)z/Is the slice equivalent to a single index? Nr)rr r )rrs rrDrDsX xxxx 99D  %  % DCHH$%r*ct|}g}tt|D]}|||jdt t j ||j||jz ||jz }|dkDs"|t|k\st||ts|jt|dt|S)zFinds the dimensions of a sliced expression. Args: key: The key used to index/slice. shape: The shape (row, col) of the expression. Returns: The dimensions of the expression as (rows, cols). rr)rr%r appendr9r2ceilr r rr r maxr")r&orig_keyrdimsrsizes rrrs!H D 3u: * q6> KKNrwwA c!fll :CFKKGHIDax1H -HQK1O CaL) * ;r*c&td|DS)z4Converts a key (i.e. two slices) into a string. c32K|]}t|ywr)rHrs rrzto_str..s4d#4s)r"r,s rto_strrUs 44 44r*ct|D]:}t|tjtfr$t j |r:yy)z>Does the key contain a list, ndarray, or logical ndarray? TF)rr numbersNumberr r2isscalar)r&rs ris_special_slicerZs= 4'..%!89R[[=N r*c6g}|D]}t|tjtfr|j t |>t|t r|j t|i|j t|dj|S)z2Converts a special key to a string representation.z, ) r r2ndarrayr#rMpprint_sequencer rHrEjoin)r&key_strsrs rspecial_key_to_strr`suH $ a"**d+ , OOOA. / 5 ! OOLO , OOCF # $ 99X r*ct||kDr,|dz}t|d|dddzt|| dddzSt|S)z9Shorten the sequence (array or list) for pretty-printing.Nr/z, ..., r)r rE)seq max_elemshalfs rr]r]sU 3x)A~3u:s#i/#c4%&k2B122FFF s8Or*r)F))__doc__rWtypingrrnumpyr2r9r)rr rr0boolr4rArHrFrDrrUrZr`r]rr*rrks$" bU38_bB""J   #5 # 6 %D %c3hE#s(O,5 T r*