_-Ph'$dZddlZdZdZdZdS)zJUtility Functions for reading and writing individual rows in BSR matrices.Nc|jd}t||z }|j|}|j|dz}||z}|j|||ddf}|j|||ddf|d|df}t jdt|ft j} d} t||D]]} ||j | z} |j| |ddfd}|j d} | |z| d| | | zf<| | z } ^| dd| dddffS)aReturn row i in BSR matrix A. Only nonzero entries are returned Parameters ---------- A : bsr_matrix Input matrix i : int Row number Returns ------- z : array Actual nonzero values for row i colindx Array of column indices for the nonzeros of row i Examples -------- >>> from numpy import array >>> from scipy.sparse import bsr_matrix >>> from pyamg.util.bsr_utils import bsr_getrow >>> indptr = array([0,2,3,6]) >>> indices = array([0,2,2,0,1,2]) >>> data = array([1,2,3,4,5,6]).repeat(4).reshape(6,2,2) >>> B = bsr_matrix( (data,indices,indptr), shape=(6,6) ) >>> Brow = bsr_getrow(B,2) >>> print(Brow[1]) [4 5] rN)dtype) blocksizeintindptrdatanonzeronpzeroslenint32rangeindicesshapereshape)Air BlockIndxrowstartrowend localRowIndxindyszcolindxcounterj coloffset increments T/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/util/bsr_utils.py bsr_getrowr"sl@ AIAiK  Ix "H Xik "Fy=L F8F?L!!!3 4 < < > >E x aaa/0q581CDAh3q66{"(333GG 8V $ $ail* q,)*2244Q7KN 2;e2C7GI-../9 99R  WQT] **c|jd}t||z }|j|}|j|dz}||z}|j|||ddf}||j|||ddf|d|df<dS)a[Set a scalar at each nonzero location in row i of BSR matrix A. Parameters ---------- A : bsr_matrix Input matrix i : int Row number x : float Scalar to overwrite nonzeros of row i in A Returns ------- A : bsr_matrix All nonzeros in row i of A have been overwritten with x. If x is a vector, the first length(x) nonzeros in row i of A have been overwritten with entries from x Examples -------- >>> from numpy import array >>> from scipy.sparse import bsr_matrix >>> from pyamg.util.bsr_utils import bsr_row_setscalar >>> indptr = array([0,2,3,6]) >>> indices = array([0,2,2,0,1,2]) >>> data = array([1,2,3,4,5,6]).repeat(4).reshape(6,2,2) >>> B = bsr_matrix( (data,indices,indptr), shape=(6,6) ) >>> bsr_row_setscalar(B,5,22) rrN)rrr r r rrxrrrrrrs r!bsr_row_setscalarr'>s> AIAiK  Ix "H Xik "Fy=L F8F?L!!!3 4 < < > >ECDAF8F?L!!! +,U1XuQx-?@@@r#c|jd}t||z }|j|}|j|dz}||z}|t |jf}|j|||ddf}||j|||ddf|d|df<dS)aSet the nonzeros in row i of BSR matrix A with the vector x. length(x) and nnz(A[i,:]) must be equivalent Parameters ---------- A : bsr_matrix Matrix assumed to be in BSR format i : int Row number x : array Array of values to overwrite nonzeros in row i of A Returns ------- A : bsr_matrix The nonzeros in row i of A have been overwritten with entries from x. x must be same length as nonzeros of row i. This is guaranteed when this routine is used with vectors derived form bsr_getrow Examples -------- >>> from numpy import array >>> from scipy.sparse import bsr_matrix >>> from pyamg.util.bsr_utils import bsr_row_setvector >>> indptr = array([0,2,3,6]) >>> indices = array([0,2,2,0,1,2]) >>> data = array([1,2,3,4,5,6]).repeat(4).reshape(6,2,2) >>> B = bsr_matrix( (data,indices,indptr), shape=(6,6) ) >>> bsr_row_setvector(B,5,array([11,22,33,44,55,66])) rrN) rrr __array__rmaxrr r r%s r!bsr_row_setvectorr+lsF AIAiK  Ix "H Xik "Fy=L s17||o..A F8F?L!!!3 4 < < > >ECDAF8F?L!!! +,U1XuQx-?@@@r#)__doc__numpyr r"r'r+r#r!r/sZPP4+4+4+n+E+E+E\4E4E4E4E4Er#