hHdZddgZddlmZmZddlmZddlmZd dZ d Z d S) z+ Functions that operate on sparse matrices count_blocksestimate_blocksize)isspmatrix_csr csr_matrix)isspmatrix_csc)csr_count_blocksffffff?ct|st|st|}|jdkrdSd|cxkrdksnt dd|zdz }t |j}|j\}}|dzdkr |dzdkr|dt|dzz }nd }|d zdkr |d zdkr|d t|d zz }nd }||kr&||kr |d t|dzz }||krdSd S|dzdkr |dzdkr|dt|dzz } nd } | |krdS||krd S||krdSdS)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 )rr$)r)r r )rrrnnz ValueErrorfloatshaper) A efficiencyhigh_efficiencyrMNe22e33e66e44s Q/var/www/html/Sam_Eipo/venv/lib/python3.11/site-packages/scipy/sparse/_spfuncs.pyrr s 1  !2!2 qMMuzzu z    C    IJJJZ'3.O ,,C 'CAa1uzza!eqjjQa.../1uzza!eqjjQa.../ _!6!6R,q///0   55 q5A::!a%1**l1U3334CCC   5 :  5 :  55c6|\}}|dks|dkrtdt|r(|j\}}t|||||j|jSt |rt|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 positive) rrrrindptrindicesrrTr)r blocksizercrrs r rr>s CAa1uuA3444a5g!!Aa;;;   5AC1&&&JqMM)444r!N)r ) __doc____all___csrrr_cscr _sparsetoolsrrrr!r r/s . /,,,,,,,, ******0000f55555r!