a a@sJdZddgZddlmZmZddlmZddlmZd ddZ d dZ d S) z+ Functions that operate on sparse matrices count_blocksestimate_blocksize)isspmatrix_csr csr_matrix)isspmatrix_csc)csr_count_blocksffffff?c Cs^t|st|st|}|jdkr&dSd|kr:dksDntdd|d}t|j}|j\}}|ddkr|ddkr|dt|d}nd }|d dkr|d dkr|d t|d }nd }||kr||kr|d t|d}||krdSd Snb|ddkr(|ddkr(|dt|d} nd } | |kr:dS||krHd S||krVdSdSdS)zAttempt to determine the blocksize of a sparse matrix Returns a blocksize=(r,c) such that - A.nnz / A.tobsr( (r,c) ).nnz > efficiency )rrg?z.efficiency must satisfy 0.0 < efficiency < 1.0g@)r r g )r r $)r)r r N)rrrnnz ValueErrorfloatshaper) AZ efficiencyZhigh_efficiencyrMNZe22Ze33Ze66Ze44rd/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/scipy/sparse/spfuncs.pyr s<       cCsr|\}}|dks|dkr tdt|rH|j\}}t|||||j|jSt|r`t|j||fStt ||SdS)z]For a given blocksize=(r,c) count the number of occupied blocks in a sparse matrix A rzr and c must be positiveN) rrrrZindptrindicesrrTr)r blocksizercrrrrrr>s N)r) __doc____all__ZcsrrrZcscrZ _sparsetoolsrrrrrrrs    3