o ÒtBh½ã@sJdZddgZddlmZmZddlmZddlmZd dd„Z d d„Z d S) z+ Functions that operate on sparse matrices Ú count_blocksÚestimate_blocksizeé)Úisspmatrix_csrÚ csr_matrix)Úisspmatrix_csc)Úcsr_count_blocksçffffffæ?c CsTt|ƒs t|ƒs t|ƒ}|jdkrdSd|kr dks%tdƒ‚tdƒ‚d|d}t|jƒ}|j\}}|ddkrK|ddkrK|dt|dƒ}nd }|d dkrc|d dkrc|d t|d ƒ}nd }||kr~||kr~|d t|dƒ}||kr|dSd S|ddkr”|ddkr”|dt|dƒ} nd } | |krœdS||kr¢d S||kr¨dSdS)z—Attempt 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 )rrrÚnnzÚ ValueErrorÚfloatÚshaper) ÚAÚ efficiencyÚhigh_efficiencyrÚMÚNÚe22Úe33Úe66Úe44©rúl/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/scipy/sparse/_spfuncs.pyr s@ ÿ   cCsn|\}}|dks |dkrtdƒ‚t|ƒr$|j\}}t|||||j|jƒSt|ƒr0t|j||fƒStt |ƒ|ƒS)z]For a given blocksize=(r,c) count the number of occupied blocks in a sparse matrix A rzr and c must be positive) rrrrÚindptrÚindicesrrÚTr)rÚ blocksizeÚrÚcrrrrrr>s N)r) Ú__doc__Ú__all__Ú_csrrrÚ_cscrÚ _sparsetoolsrrrrrrrÚs    3