a Re@sFdZddgZddlmZddlmZddlmZd ddZd dZ d S) z+ Functions that operate on sparse matrices count_blocksestimate_blocksize)issparse) csr_array)csr_count_blocksffffff?c Cs`t|r|jdvst|}|jdkr(dSd|kr efficiency )csccsr)rrg?z.efficiency must satisfy 0.0 < efficiency < 1.0g@)r r g )r r $)r)r r N)rformatrnnz ValueErrorfloatshaper) AZ efficiencyZhigh_efficiencyrMNZe22Ze33Ze66Ze44rI/var/www/sunrise/env/lib/python3.9/site-packages/scipy/sparse/_spfuncs.pyr s<       cCsz|\}}|dks|dkr tdt|rl|jdkrR|j\}}t|||||j|jS|jdkrlt|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) rrrrrZindptrindicesrTr)r blocksizercrrrrrr>s   N)r) __doc____all___baserZ_csrrZ _sparsetoolsrrrrrrrs     3