o wTh@sFdZddgZddlmZddlmZddlmZd ddZd dZ d S) z+ Functions that operate on sparse matrices count_blocksestimate_blocksize)issparse) csr_array)csr_count_blocksffffff?c CsVt|r |jdvs t|}|jdkrdSd|kr!dks&tdtdd|d}t|j}|j\}}|ddkrL|ddkrL|dt|d }nd }|d dkrd|d dkrd|d t|d }nd }||kr||kr|dt|d}||kr}dSd S|ddkr|ddkr|dt|d} nd } | |krdS||krd S||krd SdS)zAttempt to determine the blocksize of a sparse matrix Returns a blocksize=(r,c) such that - A.nnz / A.tobsr( (r,c) ).nnz > efficiency )csccsr)rrg?z.efficiency must satisfy 0.0 < efficiency < 1.0g@)r r g )r r $)r)r r )rformatrnnz ValueErrorfloatshaper) A efficiencyhigh_efficiencyrMNe22e33e66e44r V/home/air/segue/gemini/back/venv/lib/python3.10/site-packages/scipy/sparse/_spfuncs.pyr s@    cCsz|\}}|dks |dkrtdt|r6|jdkr)|j\}}t|||||j|jS|jdkr6t|j||fStt ||S)z]For a given blocksize=(r,c) count the number of occupied blocks in a sparse matrix A rzr and c must be positiver r) rrrrrindptrindicesrTr)r blocksizercrrr r r!r>s   N)r) __doc____all___baser_csrr _sparsetoolsrrrr r r r!s     3